|
@@ -1,35 +1,61 @@
|
|
|
+/*
|
|
|
+ mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
|
|
|
+ Copyright (C)2013-2016 Philemon <mmgen-py@yandex.com>
|
|
|
+
|
|
|
+ This program is free software: you can redistribute it and/or modify it under
|
|
|
+ the terms of the GNU General Public License as published by the Free Software
|
|
|
+ Foundation, either version 3 of the License, or (at your option) any later
|
|
|
+ version.
|
|
|
+
|
|
|
+ This program is distributed in the hope that it will be useful, but WITHOUT
|
|
|
+ ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
|
+ FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
|
|
|
+ details.
|
|
|
+
|
|
|
+ You should have received a copy of the GNU General Public License along with
|
|
|
+ this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
+*/
|
|
|
+
|
|
|
#include <Python.h>
|
|
|
#include <secp256k1.h>
|
|
|
|
|
|
static PyObject * priv2pub(PyObject *self, PyObject *args) {
|
|
|
- const unsigned char * privkey;
|
|
|
- const int klen;
|
|
|
- const int compressed;
|
|
|
- if (!PyArg_ParseTuple(args, "t#I", &privkey, &klen, &compressed))
|
|
|
- return NULL;
|
|
|
- if (klen != 32) return NULL;
|
|
|
- secp256k1_pubkey pubkey;
|
|
|
- size_t pubkeyclen = compressed == 1 ? 33: 65;
|
|
|
- unsigned char pubkeyc[pubkeyclen];
|
|
|
+ const unsigned char * privkey;
|
|
|
+ const int klen;
|
|
|
+ const int compressed;
|
|
|
+ if (!PyArg_ParseTuple(args, "t#I", &privkey, &klen, &compressed))
|
|
|
+ return NULL;
|
|
|
+ if (klen != 32) {
|
|
|
+ PyErr_SetString(PyExc_ValueError, "Private key length not 32 bytes");
|
|
|
+ return NULL;
|
|
|
+ }
|
|
|
+ secp256k1_pubkey pubkey;
|
|
|
+ size_t pubkeyclen = compressed == 1 ? 33: 65;
|
|
|
+ unsigned char pubkeyc[pubkeyclen];
|
|
|
static secp256k1_context *ctx = NULL;
|
|
|
if (ctx == NULL) {
|
|
|
/* puts ("Initializing context"); */
|
|
|
ctx = secp256k1_context_create(SECP256K1_CONTEXT_SIGN | SECP256K1_CONTEXT_VERIFY);
|
|
|
}
|
|
|
- if (secp256k1_ec_pubkey_create(ctx, &pubkey, privkey) != 1) return NULL;
|
|
|
- if (secp256k1_ec_pubkey_serialize(ctx, pubkeyc, &pubkeyclen, &pubkey,
|
|
|
- compressed == 1 ? SECP256K1_EC_COMPRESSED: SECP256K1_EC_UNCOMPRESSED) != 1)
|
|
|
- return NULL;
|
|
|
- return Py_BuildValue("s#", pubkeyc,pubkeyclen);
|
|
|
+ if (secp256k1_ec_pubkey_create(ctx, &pubkey, privkey) != 1) {
|
|
|
+ PyErr_SetString(PyExc_RuntimeError, "Public key creation failed");
|
|
|
+ return NULL;
|
|
|
+ }
|
|
|
+ if (secp256k1_ec_pubkey_serialize(ctx, pubkeyc, &pubkeyclen, &pubkey,
|
|
|
+ compressed == 1 ? SECP256K1_EC_COMPRESSED: SECP256K1_EC_UNCOMPRESSED) != 1) {
|
|
|
+ PyErr_SetString(PyExc_RuntimeError, "Public key serialization failed");
|
|
|
+ return NULL;
|
|
|
+ }
|
|
|
+ return Py_BuildValue("s#", pubkeyc,pubkeyclen);
|
|
|
}
|
|
|
|
|
|
static PyMethodDef secp256k1Methods[] = {
|
|
|
- {"priv2pub", priv2pub, METH_VARARGS, "Generate pubkey from privkey using libsecp256k1"},
|
|
|
- {NULL, NULL, 0, NULL} /* Sentinel */
|
|
|
+ {"priv2pub", priv2pub, METH_VARARGS, "Generate pubkey from privkey using libsecp256k1"},
|
|
|
+ {NULL, NULL, 0, NULL} /* Sentinel */
|
|
|
};
|
|
|
|
|
|
PyMODINIT_FUNC initsecp256k1(void) {
|
|
|
- PyObject *m;
|
|
|
- m = Py_InitModule("secp256k1", secp256k1Methods);
|
|
|
- if (m == NULL) return;
|
|
|
+ PyObject *m;
|
|
|
+ m = Py_InitModule("secp256k1", secp256k1Methods);
|
|
|
+ if (m == NULL) return;
|
|
|
}
|