-n

add __pairs iterator to X509_NAME binding

1 files changed, 52 insertions, 0 deletions

diff --git a/openssl.c b/openssl.c
index c2396ee..a3103b0 100644
--- a/openssl.c
+++ b/openssl.c
@@ -659,6 +659,57 @@ static int xn_all(lua_State *L) {
 } /* xn_all() */
 
 
+static int xn__next(lua_State *L) {
+        X509_NAME *name = checksimple(L, lua_upvalueindex(1), X509_NAME_CLASS);
+        X509_NAME_ENTRY *entry;
+        ASN1_OBJECT *obj;
+        const char *id;
+        char txt[256];
+        int i, n, nid, len;
+
+        lua_settop(L, 0);
+
+        i = lua_tointeger(L, lua_upvalueindex(2));
+        n = X509_NAME_entry_count(name);
+
+        while (i < n) {
+                if (!(entry = X509_NAME_get_entry(name, i++)))
+                        continue;
+
+                obj = X509_NAME_ENTRY_get_object(entry);
+                nid = OBJ_obj2nid(obj);
+
+                if (nid != NID_undef && ((id = OBJ_nid2sn(nid)) || (id = OBJ_nid2ln(nid)))) {
+                        lua_pushstring(L, id);
+                } else {
+                        if (0 > (len = OBJ_obj2txt(txt, sizeof txt, obj, 1)))
+                                return throwssl(L, "x509.name:__pairs");
+
+                        lua_pushlstring(L, txt, len);
+                }
+
+                len = ASN1_STRING_length(X509_NAME_ENTRY_get_data(entry));
+                lua_pushlstring(L, (char *)ASN1_STRING_data(X509_NAME_ENTRY_get_data(entry)), len);
+
+                break;
+        }
+
+        lua_pushinteger(L, i);
+        lua_replace(L, lua_upvalueindex(2));
+
+        return lua_gettop(L);
+} /* xn__next() */
+
+static int xn__pairs(lua_State *L) {
+        lua_settop(L, 1);
+        lua_pushinteger(L, 0);
+
+        lua_pushcclosure(L, &xn__next, 2);
+
+        return 1;
+} /* xn__pairs() */
+
+
 static int xn__gc(lua_State *L) {
         X509_NAME **ud = luaL_checkudata(L, 1, X509_NAME_CLASS);
 
@@ -689,6 +740,7 @@ static const luaL_Reg xn_methods[] = {
 };
 
 static const luaL_Reg xn_metatable[] = {
+        { "__pairs",    &xn__pairs },
         { "__gc",       &xn__gc },
         { "__tostring", &xn__tostring },
         { NULL,         NULL },