#!/usr/bin/env lua
local regress = require "regress"
local cqueues = require "cqueues"
local cs = require "cqueues.socket"
local openssl_ctx = require "openssl.ssl.context"
local cli_ctx, srv_ctx
local call_check = 0
cli_ctx = regress.getsslctx("TLS", false, false)
regress.check(cli_ctx.addCustomExtension, "Custom extension support not available")
local function c_add_ext(ssl, ext_type, context) -- luacheck: ignore 212
call_check = call_check + 1
return "from the client"
end
local function c_parse_ext(ssl, ext_type, context, data) -- luacheck: ignore 212
call_check = call_check + 2
assert(data == "from the server")
return true
end
cli_ctx:addCustomExtension(5000,
openssl_ctx.EXT_CLIENT_HELLO +
openssl_ctx.EXT_TLS1_2_SERVER_HELLO +
openssl_ctx.EXT_TLS1_3_SERVER_HELLO
, c_add_ext, c_parse_ext)
srv_ctx = regress.getsslctx("TLS", true)
local function s_add_ext(ssl, ext_type, context) -- luacheck: ignore 212
call_check = call_check + 4
return "from the server"
end
local function s_parse_ext(ssl, ext_type, context, data) -- luacheck: ignore 212
call_check = call_check + 8
assert(data == "from the client")
return true
end
srv_ctx:addCustomExtension(5000,
openssl_ctx.EXT_CLIENT_HELLO +
openssl_ctx.EXT_TLS1_2_SERVER_HELLO +
openssl_ctx.EXT_TLS1_3_SERVER_HELLO
, s_add_ext, s_parse_ext)
local srv, cli = regress.check(cs.pair(cs.SOCK_STREAM))
local main = regress.check(cqueues.new())
main:wrap(function ()
regress.check(cli:starttls(cli_ctx))
end)
main:wrap(function ()
regress.check(srv:starttls(srv_ctx))
end)
regress.check(main:loop())
regress.check(call_check == 15, "callback count doesn't match")
regress.say "OK"
