mirror of
https://github.com/brunoos/luasec.git
synced 2025-05-14 18:31:50 +02:00
Add shutdown example
This commit is contained in:
parent
bccb1f74f0
commit
c3176c5678
85
samples/shutdown/client.lua
Normal file
85
samples/shutdown/client.lua
Normal 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()
|
47
samples/shutdown/server.lua
Normal file
47
samples/shutdown/server.lua
Normal 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()
|
Loading…
x
Reference in New Issue
Block a user