From 0c13b5994f8512985cf989a8c44d7c2a6933f414 Mon Sep 17 00:00:00 2001
From: Victor Wagner <vitus@wagner.pp.ru>
Date: Fri, 7 Aug 2015 15:38:59 +0300
Subject: [PATCH] Fixed broken engine.set_default

---
 ctypescrypto/engine.py | 64 +++++++++++++++++++++++++++++++-----------
 setup.py               |  2 +-
 2 files changed, 49 insertions(+), 17 deletions(-)

diff --git a/ctypescrypto/engine.py b/ctypescrypto/engine.py
index c2858e6..f5bdb75 100644
--- a/ctypescrypto/engine.py
+++ b/ctypescrypto/engine.py
@@ -5,29 +5,61 @@ from ctypes import c_void_p, c_char_p, c_int
 from ctypescrypto import libcrypto
 from ctypescrypto.exception import LibCryptoError
 
-__all__ = ['default', 'set_default']
+__all__ = ['default', 'set_default', 'Engine']
 
 default = None
 
-def set_default(engine):
+class Engine(object):
     """
-    Loads specified engine and sets it as default for all
+    Represents Openssl loadable module (engine).
+    Allows to create PKey objects from private keys stored
+    in the token, accessed by engine
+    """
+    def __init__(self, engine_id, **kwargs):
+        eng = libcrypto.ENGINE_by_id(engine_id)
+        if eng is None:
+            # Try load engine
+            eng = libcrypto.ENGINE_by_id("dynamic")
+            if  eng is None:
+                raise LibCryptoError("Cannot get 'dynamic' engine")
+            if not libcrypto.ENGINE_ctrl_cmd_string(eng, "SO_PATH",
+                                                    engine_id, 0):
+                raise LibCryptoError("Cannot execute ctrl cmd SO_PATH")
+            if not libcrypto.ENGINE_ctrl_cmd_string(eng, "LOAD", None, 0):
+                raise LibCryptoError("Cannot execute ctrl cmd LOAD")
+        if eng is None:
+            raise ValueError("Cannot find engine " + engine)
+        for cmd, value in kwargs.items():
+            if not libcrypto.ENGINE_ctrl_cmd_string(eng, cmd, value, 0):
+                raise LibCryptoError("Cannot execute ctrl cmd %s" % cmd)
+        if not libcrypto.ENGINE_init(eng):
+            raise LibCryptoError("Cannot initialize engine")
+        self.ptr = eng
+
+    def private_key(self, key_id, ui_method = None, ui_data=None):
+        from ctypescrypto.pkey import PKey
+        if ui_method is None:
+            ui_ptr = libcrypto.UI_OpenSSL()
+        else:
+            ui_ptr = ui_method.ptr
+        pkey = libcrypto.ENGINE_load_private_key(self.ptr, key_id, ui_ptr,
+                                                 ui_data)
+        if pkey is None:
+            raise LibCryptoError("Cannot load private key")
+        return PKey(ptr=pkey, cansign=True)
+
+def set_default(eng, algorithms=0xFFFF):
+    """
+    Sets specified engine  as default for all
     algorithms, supported by it
+
+    For compatibility with 0.2.x if string is passed instead
+    of engine, attempts to load engine with this id
     """
+    if not isinstance(eng,Engine):
+        eng=Engine(eng)
     global default
-    eng = libcrypto.ENGINE_by_id(engine)
-    if eng is None:
-        # Try load engine
-        eng = libcrypto.ENGINE_by_id("dynamic")
-        if  eng is None:
-            raise LibCryptoError("Cannot get 'dynamic' engine")
-        if not libcrypto.ENGINE_ctrl_cmd_string(eng, "SO_PATH", engine, 0):
-            raise LibCryptoError("Cannot execute ctrl cmd SO_PATH")
-        if not libcrypto.ENGINE_ctrl_cmd_string(eng, "LOAD", None, 0):
-            raise LibCryptoError("Cannot execute ctrl cmd LOAD")
-    if eng is None:
-        raise ValueError("Cannot find engine " + engine)
-    libcrypto.ENGINE_set_default(eng, c_int(0xFFFF))
+    libcrypto.ENGINE_set_default(eng.ptr, c_int(algorithms))
     default = eng
 
 # Declare function result and arguments for used functions
diff --git a/setup.py b/setup.py
index bed60db..0b85ab9 100644
--- a/setup.py
+++ b/setup.py
@@ -33,7 +33,7 @@ class MyTests(distutils.cmd.Command):
 
 setup(
     name="ctypescrypto",
-    version="0.3.0",
+    version="0.3.1",
     description="CTypes-based interface for some OpenSSL libcrypto features",
     author="Victor Wagner",
     author_email="vitus@wagner.pp.ru",
-- 
2.39.5