diff options
| -rw-r--r-- | src/openssl.c | 52 |
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 }, }; |