1 files changed, 62 insertions, 0 deletions
diff --git a/pycryptopp/_pycryptoppmodule.cpp b/pycryptopp/_pycryptoppmodule.cpp
new file mode 100644
index 0000000..1679a73
--- /dev/null
+++ b/pycryptopp/_pycryptoppmodule.cpp
@@ -0,0 +1,62 @@
+#include <Python.h>
+
+#include "publickey/ecdsamodule.hpp"
+#include "publickey/rsamodule.hpp"
+#include "hash/sha256module.hpp"
+#include "cipher/aesmodule.hpp"
+
+/* from Crypto++ */
+#ifdef DISABLE_EMBEDDED_CRYPTOPP
+#include <cryptopp/config.h>
+#else
+#include <embeddedcryptopp/config.h>
+#endif
+
+PyDoc_STRVAR(_pycryptopp__doc__,
+"_pycryptopp -- Python wrappers for a few algorithms from Crypto++\n\
+\n\
+from pycryptopp import publickey\n\
+from pycryptopp.publickey import ecdsa\n\
+from pycryptopp.publickey import rsa\n\
+from pycryptopp import cipher\n\
+from pycryptopp.cipher import aes\n\
+from pycryptopp import hash\n\
+from pycryptopp.hash import sha256");
+
+static PyMethodDef _pycryptopp_functions[] = {
+    {"rsa_generate", reinterpret_cast<PyCFunction>(rsa_generate), METH_KEYWORDS, const_cast<char*>(rsa_generate__doc__)},
+    {"rsa_create_verifying_key_from_string", reinterpret_cast<PyCFunction>(rsa_create_verifying_key_from_string), METH_KEYWORDS, const_cast<char*>(rsa_create_verifying_key_from_string__doc__)},
+    {"rsa_create_signing_key_from_string", reinterpret_cast<PyCFunction>(rsa_create_signing_key_from_string), METH_KEYWORDS, const_cast<char*>(rsa_create_signing_key_from_string__doc__)},
+    {NULL, NULL, 0, NULL}  /* sentinel */
+};
+
+#ifndef PyMODINIT_FUNC /* declarations for DLL import/export */
+#define PyMODINIT_FUNC void
+#endif
+PyMODINIT_FUNC
+init_pycryptopp(void) {
+    PyObject *module;
+
+    module = Py_InitModule3("_pycryptopp", _pycryptopp_functions, _pycryptopp__doc__);
+    if (!module)
+      return;
+
+    PyObject* version;
+
+    /* a tuple of (Crypto++ version, extra-version) */
+    #ifdef CRYPTOPP_EXTRA_VERSION
+    version = Py_BuildValue("is", CRYPTOPP_VERSION, CRYPTOPP_EXTRA_VERSION);
+    #else
+    version = Py_BuildValue("iO", CRYPTOPP_VERSION, Py_None);
+    #endif
+
+    int succ = PyModule_AddObject(module, "cryptopp_version", version);
+    if (succ != 0)
+        return;
+
+
+    init_ecdsa(module);
+    init_rsa(module);
+    init_sha256(module);
+    init_aes(module);
+}