Bind SSL_CTX_set_alpn_protos and SSL_get0_alpn_selected

1 files changed, 52 insertions, 0 deletions

diff --git a/src/openssl.c b/src/openssl.c
index 6e3039f..ab19410 100644
--- a/src/openssl.c
+++ b/src/openssl.c
@@ -4516,6 +4516,44 @@ static int sx_setEphemeralKey(lua_State *L) {
         return 1;
 } /* sx_setEphemeralKey() */
 
+static int sx_setAlpnProtos(lua_State *L) {
+        SSL_CTX *ctx = checksimple(L, 1, SSL_CTX_CLASS);
+        size_t len;
+        const char *tmp;
+        unsigned protos_len = 0;
+        luaL_Buffer B;
+        luaL_checktype(L, 2, LUA_TTABLE);
+        luaL_buffinit(L, &B);
+
+        while (1) {
+                protos_len++;
+                lua_rawgeti(L, 2, protos_len);
+                switch (lua_type(L, -1)) {
+                        case LUA_TNIL:
+                                goto done;
+                        case LUA_TSTRING:
+                                break;
+                        default:
+                                return luaL_argerror(L, 2, "array of strings expected");
+                }
+                tmp = luaL_checklstring(L, -1, &len);
+                luaL_argcheck(L, len <= UCHAR_MAX, 2, "proto string too long");
+                luaL_addchar(&B, (unsigned char)len);
+                luaL_addlstring(&B, tmp, len);
+                lua_pop(L, 1);
+        }
+done:
+        luaL_pushresult(&B);
+
+        if (0 != SSL_CTX_set_alpn_protos(ctx, (const unsigned char*)lua_tostring(L, -1), protos_len)) {
+                lua_pushnil(L);
+                return 1;
+        }
+
+        lua_pushboolean(L, 1);
+
+        return 1;
+} /* sx_setAlpnprotos */
 
 static int sx__gc(lua_State *L) {
         SSL_CTX **ud = luaL_checkudata(L, 1, SSL_CTX_CLASS);
@@ -4540,6 +4578,7 @@ static const luaL_Reg sx_methods[] = {
         { "setPrivateKey",    &sx_setPrivateKey },
         { "setCipherList",    &sx_setCipherList },
         { "setEphemeralKey",  &sx_setEphemeralKey },
+        { "setAlpnProtos",    &sx_setAlpnProtos },
         { NULL, NULL },
 };
 
@@ -4790,6 +4829,18 @@ static int ssl_getClientVersion(lua_State *L) {
         return 1;
 } /* ssl_getClientVersion() */
 
+static int ssl_getAlpnSelected(lua_State *L) {
+        SSL *ssl = checksimple(L, 1, SSL_CLASS);
+        const unsigned char *data;
+        unsigned len;
+        SSL_get0_alpn_selected(ssl, &data, &len);
+        if (0 == len) {
+                lua_pushnil(L);
+        } else {
+                lua_pushlstring(L, data, len);
+        }
+        return 1;
+} /*ssl_getAlpnSelected */
 
 static int ssl__gc(lua_State *L) {
         SSL **ud = luaL_checkudata(L, 1, SSL_CLASS);
@@ -4814,6 +4865,7 @@ static const luaL_Reg ssl_methods[] = {
         { "setHostName",      &ssl_setHostName },
         { "getVersion",       &ssl_getVersion },
         { "getClientVersion", &ssl_getClientVersion },
+        { "getAlpnSelected",  &ssl_getAlpnSelected },
         { NULL,            NULL },
 };