3 files changed, 92 insertions, 31 deletions

diff --git a/misc/python/src/block.cpp b/misc/python/src/block.cpp
new file mode 100644
index 000000000..d68300ce6
--- /dev/null
+++ b/misc/python/src/block.cpp
@@ -0,0 +1,79 @@
+/*************************************************
+* Boost.Python module definition                 *
+* (C) 1999-2006 The Botan Project                *
+*************************************************/
+
+#include <botan/botan.h>
+using namespace Botan;
+
+#include <boost/python.hpp>
+namespace python = boost::python;
+
+/* Encrypt or decrypt */
+std::string process(const std::string& in, BlockCipher* cipher,
+                    void (BlockCipher::* proc)(const byte[], byte[]) const)
+   {
+   if(in.size() != cipher->BLOCK_SIZE)
+      throw Invalid_Argument("Input for cipher " + cipher->name() +
+                             " must be " + to_string(cipher->BLOCK_SIZE) +
+                             " bytes");
+
+   const byte* in_data = (const byte*)in.data();
+
+   SecureVector<byte> out(cipher->BLOCK_SIZE);
+   (cipher->*proc)(in_data, out);
+
+   return std::string((const char*)out.begin(), out.size());
+   }
+
+class Py_BlockCipher
+   {
+   public:
+      u32bit block_size() const { return cipher->BLOCK_SIZE; }
+      u32bit keylength_min() const { return cipher->MINIMUM_KEYLENGTH; }
+      u32bit keylength_max() const { return cipher->MAXIMUM_KEYLENGTH; }
+      u32bit keylength_mod() const { return cipher->KEYLENGTH_MULTIPLE; }
+
+      bool valid_keylength(u32bit kl) const
+         {
+         return cipher->valid_keylength(kl);
+         }
+
+      std::string encrypt(const std::string& in) const
+         {
+         return process(in, cipher, &BlockCipher::encrypt);
+         }
+      std::string decrypt(const std::string& in) const
+         {
+         return process(in, cipher, &BlockCipher::decrypt);
+         }
+
+      void set_key(const OctetString& key)
+         {
+         cipher->set_key(key);
+         }
+
+      Py_BlockCipher(const std::string& name)
+         {
+         cipher = get_block_cipher(name);
+         }
+      ~Py_BlockCipher()
+         {
+         delete cipher;
+         }
+   private:
+      BlockCipher* cipher;
+   };
+
+void export_block_ciphers()
+   {
+   python::class_<Py_BlockCipher>("BlockCipher", python::init<std::string>())
+      .add_property("block_size", &Py_BlockCipher::block_size)
+      .add_property("keylength_min", &Py_BlockCipher::keylength_min)
+      .add_property("keylength_max", &Py_BlockCipher::keylength_max)
+      .add_property("keylength_mod", &Py_BlockCipher::keylength_mod)
+      .def("valid_keylength", &Py_BlockCipher::valid_keylength)
+      .def("set_key", &Py_BlockCipher::set_key)
+      .def("encrypt", &Py_BlockCipher::encrypt)
+      .def("decrypt", &Py_BlockCipher::decrypt);
+   }
diff --git a/misc/python/src/core.cpp b/misc/python/src/core.cpp
index 07252e720..5669a5b83 100644
--- a/misc/python/src/core.cpp
+++ b/misc/python/src/core.cpp
@@ -3,14 +3,15 @@
 * (C) 1999-2006 The Botan Project                *
 *************************************************/
 
-#include <botan/init.h>
-#include <botan/symkey.h>
+#include <botan/botan.h>
 using namespace Botan;
 
 #include <boost/python.hpp>
 namespace python = boost::python;
 
+extern void export_block_ciphers();
 extern void export_filters();
+extern void export_pk();
 extern void export_x509();
 
 BOOST_PYTHON_MODULE(_botan)
@@ -22,12 +23,14 @@ BOOST_PYTHON_MODULE(_botan)
       .def(python::init< python::optional<std::string> >())
       .def(python::init< u32bit >())
       .def("__str__", &OctetString::as_string)
-      .add_property("length", &OctetString::length);
+      .def("__len__", &OctetString::length);
 
    python::enum_<Cipher_Dir>("cipher_dir")
       .value("encryption", ENCRYPTION)
       .value("decryption", DECRYPTION);
 
+   export_block_ciphers();
    export_filters();
+   export_pk();
    export_x509();
    }
diff --git a/misc/python/test.py b/misc/python/test.py
index 9896777d4..20c2a9119 100755
--- a/misc/python/test.py
+++ b/misc/python/test.py
@@ -2,36 +2,15 @@
 
 import sys, botan
 
-def encrypt(input):
-    cipher_key = botan.SymmetricKey("AABB")
-    print cipher_key.length
-    cipher_key = botan.SymmetricKey("AABBCCDD")
-    print cipher_key.length
-
-    cipher = botan.Filter("ARC4", key = cipher_key)
-    
-    pipe = botan.Pipe(cipher, botan.Filter("Hex_Encoder"))
-
-    pipe.start_msg()
-    pipe.write(input)
-    pipe.end_msg()
-
-    str = pipe.read_all()
-    print str
-    return str
-
-def decrypt(input):
-    pipe = botan.Pipe(botan.Filter("Hex_Decoder"),
-                      botan.Filter("ARC4",
-                                   key = botan.SymmetricKey("AABBCCDD")))
-
-    pipe.process_msg(input)
-    return pipe.read_all()
+class PyAlgo(botan.Algorithm):
+    def name(self):
+        return "PyAlgo"
+    def clear(self):
+        print "clearing"
 
 def main():
-    ciphertext = encrypt("hi chappy")
-    print ciphertext
-    print decrypt(ciphertext)
+    alg = PyAlgo()
+    botan.print_algo(alg)
 
 if __name__ == "__main__":
     sys.exit(main())