/* mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution Copyright (C)2013-2017 Philemon 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 . */ #include #include 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) { 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) { 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 */ }; PyMODINIT_FUNC initsecp256k1(void) { PyObject *m; m = Py_InitModule("secp256k1", secp256k1Methods); if (m == NULL) return; }