1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
|
#!/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"
|