Add shutdown example

This commit is contained in:
Bruno Silvestre 2025-05-08 15:51:38 -03:00
parent bccb1f74f0
commit c3176c5678
2 changed files with 132 additions and 0 deletions

View File

@ -0,0 +1,85 @@
--
-- Test the conn:shutdown() function
--
-- Public domain
--
local socket = require("socket")
local ssl = require("ssl")
local params = {
mode = "client",
protocol = "tlsv1_2",
key = "../certs/clientAkey.pem",
certificate = "../certs/clientA.pem",
cafile = "../certs/rootA.pem",
verify = {"peer", "fail_if_no_peer_cert"},
options = "all",
}
-- Wait until socket is ready (for reading or writing)
local function wait(peer)
-- What event blocked us?
local err = peer:want()
print("Want? ", err)
if err == "read" then
socket.select({peer}, nil)
elseif err == "write" then
socket.select(nil, {peer})
elseif err == "nothing" then
return
else
peer:close()
os.exit(1)
end
end
-- Send data
local function send(peer, data)
local offset = 1
while true do
local succ, msg, part = peer:send(data, offset)
if succ then break end
if part then
offset = offset + part
wait(peer)
end
end
end
-- Start the TCP connection
local peer = socket.tcp()
peer:setoption('tcp-nodelay', true)
assert(peer:connect("127.0.0.1", 8888))
peer = assert(ssl.wrap(peer, params))
local ctx = assert(ssl.newcontext(params))
peer:settimeout(0.3)
print("*** Handshake")
while true do
local succ, msg = peer:dohandshake()
if succ then break end
wait(peer)
end
print("*** Send data")
for i = 1, 10 do
send(peer, string.rep('1', 8192))
end
print("*** Shutdown")
while true do
local succ, msg = peer:shutdown()
if succ then break end
print(succ, msg)
if msg ~= "inprogress" then
wait(peer)
end
end
print("*** Done")
peer:close()

View File

@ -0,0 +1,47 @@
--
-- Public domain
--
local socket = require("socket")
local ssl = require("ssl")
local params = {
mode = "server",
protocol = "any",
key = "../certs/serverAkey.pem",
certificate = "../certs/serverA.pem",
cafile = "../certs/rootA.pem",
verify = {"peer", "fail_if_no_peer_cert"},
options = "all",
}
local ctx = assert(ssl.newcontext(params))
local server = socket.tcp()
server:setoption('reuseaddr', true)
assert(server:bind("127.0.0.1", 8888))
server:listen()
while true do
local peer = server:accept()
peer:setoption('tcp-nodelay', true)
print("*** New connection")
peer = assert( ssl.wrap(peer, ctx) )
print("*** Handshake")
assert( peer:dohandshake() )
print("*** Receive")
while true do
local str = peer:receive(1024)
if not str then break end
socket.sleep(0.1)
end
print("*** Done")
peer:close()
end
server:close()