1 files changed, 138 insertions, 0 deletions
diff --git a/src/crypto/engine/engine.c b/src/crypto/engine/engine.c
new file mode 100644
index 0000000..5b8cf1c
--- /dev/null
+++ b/src/crypto/engine/engine.c
@@ -0,0 +1,138 @@
+/* Copyright (c) 2014, Google Inc.
+ *
+ * Permission to use, copy, modify, and/or distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
+ * SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
+ * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
+ * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */
+
+#include <openssl/engine.h>
+
+#include <string.h>
+
+#include <openssl/dh.h>
+#include <openssl/dsa.h>
+#include <openssl/ec_key.h>
+#include <openssl/err.h>
+#include <openssl/mem.h>
+#include <openssl/rsa.h>
+#include <openssl/thread.h>
+
+
+struct engine_st {
+  DH_METHOD *dh_method;
+  DSA_METHOD *dsa_method;
+  RSA_METHOD *rsa_method;
+  ECDSA_METHOD *ecdsa_method;
+};
+
+ENGINE *ENGINE_new(void) {
+  ENGINE *engine = OPENSSL_malloc(sizeof(ENGINE));
+  if (engine == NULL) {
+    return NULL;
+  }
+
+  memset(engine, 0, sizeof(ENGINE));
+  return engine;
+}
+
+void ENGINE_free(ENGINE *engine) {
+  if (engine->dh_method != NULL) {
+    METHOD_unref(engine->dh_method);
+  }
+
+  OPENSSL_free(engine);
+}
+
+/* set_method takes a pointer to a method and its given size and sets
+ * |*out_member| to point to a copy of it. The copy is |compiled_size| bytes
+ * long and has zero padding if needed. */
+static int set_method(void **out_member, const void *method, size_t method_size,
+                      size_t compiled_size) {
+  void *copy = OPENSSL_malloc(compiled_size);
+  if (copy == NULL) {
+    return 0;
+  }
+
+  memset(copy, 0, compiled_size);
+
+  if (method_size > compiled_size) {
+    method_size = compiled_size;
+  }
+  memcpy(copy, method, method_size);
+
+  METHOD_unref(*out_member);
+  *out_member = copy;
+
+  return 1;
+}
+
+int ENGINE_set_DH_method(ENGINE *engine, const DH_METHOD *method,
+                         size_t method_size) {
+  return set_method((void **)&engine->dh_method, method, method_size,
+                    sizeof(DH_METHOD));
+}
+
+DH_METHOD *ENGINE_get_DH_method(const ENGINE *engine) {
+  return engine->dh_method;
+}
+
+int ENGINE_set_DSA_method(ENGINE *engine, const DSA_METHOD *method,
+                         size_t method_size) {
+  return set_method((void **)&engine->dsa_method, method, method_size,
+                    sizeof(DSA_METHOD));
+}
+
+DSA_METHOD *ENGINE_get_DSA_method(const ENGINE *engine) {
+  return engine->dsa_method;
+}
+
+int ENGINE_set_RSA_method(ENGINE *engine, const RSA_METHOD *method,
+                         size_t method_size) {
+  return set_method((void **)&engine->rsa_method, method, method_size,
+                    sizeof(RSA_METHOD));
+}
+
+RSA_METHOD *ENGINE_get_RSA_method(const ENGINE *engine) {
+  return engine->rsa_method;
+}
+
+int ENGINE_set_ECDSA_method(ENGINE *engine, const ECDSA_METHOD *method,
+                            size_t method_size) {
+  return set_method((void **)&engine->ecdsa_method, method, method_size,
+                    sizeof(ECDSA_METHOD));
+}
+
+ECDSA_METHOD *ENGINE_get_ECDSA_method(const ENGINE *engine) {
+  return engine->ecdsa_method;
+}
+
+void METHOD_ref(void *method_in) {
+  struct openssl_method_common_st *method = method_in;
+
+  if (method->is_static) {
+    return;
+  }
+
+  CRYPTO_add(&method->references, 1, CRYPTO_LOCK_ENGINE);
+}
+
+void METHOD_unref(void *method_in) {
+  struct openssl_method_common_st *method = method_in;
+
+  if (method == NULL || method->is_static) {
+    return;
+  }
+
+  if (CRYPTO_add(&method->references, -1, CRYPTO_LOCK_ENGINE) == 0) {
+    OPENSSL_free(method);
+  }
+}
+
+OPENSSL_DECLARE_ERROR_REASON(ENGINE, OPERATION_NOT_SUPPORTED);