1 files changed, 90 insertions, 0 deletions
diff --git a/src/SWIG/_aes.i b/src/SWIG/_aes.i
new file mode 100644
index 0000000..ea36167
--- /dev/null
+++ b/src/SWIG/_aes.i
@@ -0,0 +1,90 @@
+/* Copyright (c) 1999-2004 Ng Pheng Siong. All rights reserved. */
+/* $Id$ */
+
+%{
+#include <openssl/evp.h>
+
+#if OPENSSL_VERSION_NUMBER >= 0x0090800fL
+#include <openssl/aes.h>
+#endif
+
+/* 
+// 2004-10-10, ngps: 
+// CTR mode is not included in the default OpenSSL build.
+// To use the AES CTR ciphers, link with your own copy of OpenSSL.
+*/
+#ifdef HAVE_AES_CTR
+extern EVP_CIPHER const *EVP_aes_128_ctr(void);
+extern EVP_CIPHER const *EVP_aes_192_ctr(void);
+extern EVP_CIPHER const *EVP_aes_256_ctr(void);
+#endif
+%}
+
+%apply Pointer NONNULL { AES_KEY * };
+
+%constant int AES_BLOCK_SIZE = AES_BLOCK_SIZE;
+
+%inline %{
+AES_KEY *aes_new(void) {
+    AES_KEY *key;
+    
+    if (!(key = (AES_KEY *)PyMem_Malloc(sizeof(AES_KEY)))) {
+        PyErr_SetString(PyExc_MemoryError,
+                        "Insufficient memory for AES key.");
+        return NULL;
+    }
+    return key;
+}   
+
+void AES_free(AES_KEY *key) {
+    PyMem_Free((void *)key);
+}
+
+/* 
+// op == 0: encrypt
+// otherwise: decrypt (Python code will supply the value 1.)
+*/
+PyObject *AES_set_key(AES_KEY *key, PyObject *value, int bits, int op) { 
+    char *vbuf; 
+    Py_ssize_t vlen;
+
+    if (PyBytes_AsStringAndSize(value, &vbuf, &vlen) == -1)
+        return NULL;
+
+    if (op == 0) 
+        AES_set_encrypt_key((const unsigned char *)vbuf, bits, key);
+    else
+        AES_set_decrypt_key((const unsigned char *)vbuf, bits, key);
+    Py_RETURN_NONE;
+}
+
+/* 
+// op == 0: encrypt
+// otherwise: decrypt (Python code will supply the value 1.)
+*/
+PyObject *AES_crypt(const AES_KEY *key, PyObject *in, int outlen, int op) {
+    char *buf;
+    Py_ssize_t len;
+    unsigned char *out;
+    PyObject *res;
+
+    if (PyBytes_AsStringAndSize(in, &buf, &len) == -1)
+        return NULL;
+
+    if (!(out=(unsigned char *)PyMem_Malloc(outlen))) {
+        PyErr_SetString(PyExc_MemoryError, "AES_crypt");
+        return NULL;
+    }
+    if (op == 0)
+        AES_encrypt((const unsigned char *)buf, out, key);
+    else
+        AES_decrypt((const unsigned char *)buf, out, key);
+    res = PyBytes_FromStringAndSize((char*)out, outlen);
+    PyMem_Free(out);
+    return res;
+}
+
+int AES_type_check(AES_KEY *key) {
+    return 1;
+}
+%}