Merge branch 'custom-ext' of https://github.com/kunkku/luaossl into kunkku-custom-ext

3 files changed, 110 insertions, 0 deletions

diff --git a/src/GNUmakefile b/src/GNUmakefile
index 240a773..f988855 100644
--- a/src/GNUmakefile
+++ b/src/GNUmakefile
@@ -95,6 +95,7 @@ MODS$(1)_$(d) = \
         $$(DESTDIR)$(3)/openssl/x509/altname.lua \
         $$(DESTDIR)$(3)/openssl/x509/chain.lua \
         $$(DESTDIR)$(3)/openssl/x509/crl.lua \
+        $$(DESTDIR)$(3)/openssl/x509/extension.lua \
         $$(DESTDIR)$(3)/openssl/x509/store.lua \
         $$(DESTDIR)$(3)/openssl/pkcs12.lua \
         $$(DESTDIR)$(3)/openssl/ssl/context.lua \
diff --git a/src/openssl.c b/src/openssl.c
index ee1fd89..7bd9c61 100644
--- a/src/openssl.c
+++ b/src/openssl.c
@@ -84,6 +84,7 @@
 #define PKEY_CLASS       "EVP_PKEY*"
 #define X509_NAME_CLASS  "X509_NAME*"
 #define X509_GENS_CLASS  "GENERAL_NAMES*"
+#define X509_EXT_CLASS   "X509_EXTENSION*"
 #define X509_CERT_CLASS  "X509*"
 #define X509_CHAIN_CLASS "STACK_OF(X509)*"
 #define X509_CSR_CLASS   "X509_REQ*"
@@ -1819,6 +1820,98 @@ int luaopen__openssl_x509_altname(lua_State *L) {
 
 
 /*
+ * X509_EXTENSION - openssl.x509.extension
+ *
+ * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
+
+static int xe_new(lua_State *L) {
+        lua_settop(L, 3);
+
+        X509_EXTENSION **ud = prepsimple(L, X509_EXT_CLASS);
+
+        char *name = (char *) luaL_checkstring(L, 1);
+        char *value = (char *) luaL_checkstring(L, 2);
+
+        CONF *conf = NULL;
+        X509V3_CTX *ctx = NULL;
+        X509_EXTENSION *ext = NULL;
+
+        if (!lua_isnil(L, 3)) {
+                char *cdata = (char *) luaL_checkstring(L, 3);
+                BIO *bio = getbio(L);
+                if (BIO_puts(bio, cdata) < 0)
+                        goto error;
+
+                if (!(conf = NCONF_new(NULL)))
+                        goto error;
+                if (!NCONF_load_bio(conf, bio, NULL))
+                        goto error;
+
+                ctx = (X509V3_CTX *) malloc(sizeof (X509V3_CTX));
+                X509V3_set_nconf(ctx, conf);
+        }
+
+        if (!(*ud = X509V3_EXT_nconf(conf, ctx, name, value)))
+                goto error;
+
+        if (conf) {
+                free(ctx);
+                NCONF_free(conf);
+        }
+
+        return 1;
+
+        error:
+        if (ctx)
+                free(ctx);
+        if (conf)
+                NCONF_free(conf);
+
+        return throwssl(L, "x509.extension.new");
+} /* xe_new() */
+
+
+static int xe_interpose(lua_State *L) {
+        return interpose(L, X509_EXT_CLASS);
+} /* xe_interpose() */
+
+
+static int xe__gc(lua_State *L) {
+        X509_EXTENSION **ud = luaL_checkudata(L, 1, X509_EXT_CLASS);
+
+        X509_EXTENSION_free(*ud);
+        *ud = NULL;
+
+        return 0;
+} /* xe__gc() */
+
+
+static const luaL_Reg xe_methods[] = {
+        { NULL,  NULL },
+};
+
+static const luaL_Reg xe_metatable[] = {
+        { "__gc", &xe__gc },
+        { NULL,   NULL },
+};
+
+
+static const luaL_Reg xe_globals[] = {
+        { "new",       &xe_new },
+        { "interpose", &xe_interpose },
+        { NULL,        NULL },
+};
+
+int luaopen__openssl_x509_extension(lua_State *L) {
+        initall(L);
+
+        luaL_newlib(L, xe_globals);
+
+        return 1;
+} /* luaopen__openssl_x509_extension() */
+
+
+/*
  * X509 - openssl.x509.cert
  *
  * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
@@ -2511,6 +2604,19 @@ static int xc_setBasicConstraintsCritical(lua_State *L) {
 } /* xc_setBasicConstraintsCritical() */
 
 
+static int xc_addExtension(lua_State *L) {
+        X509 *crt = checksimple(L, 1, X509_CERT_CLASS);
+        X509_EXTENSION *ext = checksimple(L, 2, X509_EXT_CLASS);
+
+        if (!X509_add_ext(crt, ext, -1))
+                throwssl(L, "x509.cert:addExtension");
+
+        lua_pushboolean(L, 1);
+
+        return 1;
+} /* xc_addExtension() */
+
+
 static int xc_isIssuedBy(lua_State *L) {
         X509 *crt = checksimple(L, 1, X509_CERT_CLASS);
         X509 *issuer = checksimple(L, 2, X509_CERT_CLASS);
@@ -2726,6 +2832,7 @@ static const luaL_Reg xc_methods[] = {
         { "setBasicConstraint",  &xc_setBasicConstraint },
         { "getBasicConstraintsCritical", &xc_getBasicConstraintsCritical },
         { "setBasicConstraintsCritical", &xc_setBasicConstraintsCritical },
+        { "addExtension",  &xc_addExtension },
         { "isIssuedBy",    &xc_isIssuedBy },
         { "getPublicKey",  &xc_getPublicKey },
         { "setPublicKey",  &xc_setPublicKey },
@@ -4967,6 +5074,7 @@ static void initall(lua_State *L) {
         addclass(L, PKEY_CLASS, pk_methods, pk_metatable);
         addclass(L, X509_NAME_CLASS, xn_methods, xn_metatable);
         addclass(L, X509_GENS_CLASS, gn_methods, gn_metatable);
+        addclass(L, X509_EXT_CLASS, xe_methods, xe_metatable);
         addclass(L, X509_CERT_CLASS, xc_methods, xc_metatable);
         addclass(L, X509_CSR_CLASS, xr_methods, xr_metatable);
         addclass(L, X509_CRL_CLASS, xx_methods, xx_metatable);
diff --git a/src/openssl.x509.extension.lua b/src/openssl.x509.extension.lua
new file mode 100644
index 0000000..7043f45
--- /dev/null
+++ b/src/openssl.x509.extension.lua
@@ -0,0 +1 @@
+return require('_openssl.x509.extension')