mirror of
https://github.com/status-im/nim-websock.git
synced 2025-01-24 16:29:16 +00:00
5af418f850
* add accepts api * Fix partial frame handling and allow extensions to hijack the flow (#56) * moving files around * wip * wip * move tls example into server example * add tls functionality * rename * rename * fix tests * move extension related files to own folder * use trace instead of debug * export extensions * rework partial frame handling and closing * rework status codes as distincts * logging * re-enable extensions processing for frames * enable all test for non-tls server * remove tlsserver * remove offset to mask - don't think we need it * pass sessions extensions when calling send/recv * adding encode/decode extensions flow test * move server/client setup to helpers * proper frame order execution on decode * fix tls tests * fix merge * add tls support for `accept` call * fix tests to use accepts & cb
101 lines
2.6 KiB
Nim
101 lines
2.6 KiB
Nim
{.push raises: [Defect].}
|
|
|
|
import std/[strutils, random]
|
|
import pkg/[
|
|
chronos,
|
|
chronos/streams/tlsstream,
|
|
httputils,
|
|
chronicles,
|
|
stew/byteutils]
|
|
|
|
import ../ws/ws
|
|
import ./keys
|
|
|
|
let
|
|
WSSecureKey* = TLSPrivateKey.init(SecureKey)
|
|
WSSecureCert* = TLSCertificate.init(SecureCert)
|
|
|
|
const WSPath* = when defined secure: "/wss" else: "/ws"
|
|
|
|
proc rndStr*(size: int): string =
|
|
for _ in 0..<size:
|
|
add(result, char(rand(int('A') .. int('z'))))
|
|
|
|
proc rndBin*(size: int): seq[byte] =
|
|
for _ in 0..<size:
|
|
add(result, byte(rand(0 .. 255)))
|
|
|
|
proc waitForClose*(ws: WSSession) {.async.} =
|
|
try:
|
|
while ws.readystate != ReadyState.Closed:
|
|
discard await ws.recv()
|
|
except CatchableError:
|
|
trace "Closing websocket"
|
|
|
|
proc createServer*(
|
|
address = initTAddress("127.0.0.1:8888"),
|
|
tlsPrivateKey = WSSecureKey,
|
|
tlsCertificate = WSSecureCert,
|
|
handler: HttpAsyncCallback = nil,
|
|
flags: set[ServerFlags] = {ServerFlags.TcpNoDelay, ServerFlags.ReuseAddr},
|
|
tlsFlags: set[TLSFlags] = {},
|
|
tlsMinVersion = TLSVersion.TLS12,
|
|
tlsMaxVersion = TLSVersion.TLS12): HttpServer
|
|
{.raises: [Defect, HttpError].} =
|
|
try:
|
|
let server = when defined secure:
|
|
TlsHttpServer.create(
|
|
address = address,
|
|
tlsPrivateKey = tlsPrivateKey,
|
|
tlsCertificate = tlsCertificate,
|
|
flags = flags,
|
|
tlsFlags = tlsFlags,
|
|
tlsMinVersion = tlsMinVersion,
|
|
tlsMaxVersion = tlsMaxVersion)
|
|
else:
|
|
HttpServer.create(
|
|
address = address,
|
|
flags = flags)
|
|
|
|
when defined accepts:
|
|
proc accepts() {.async, raises: [Defect].} =
|
|
try:
|
|
let req = await server.accept()
|
|
await req.handler()
|
|
except TransportOsError as exc:
|
|
error "Transport error", exc = exc.msg
|
|
|
|
asyncCheck accepts()
|
|
else:
|
|
server.handler = handler
|
|
server.start()
|
|
|
|
return server
|
|
except CatchableError as exc:
|
|
raise newException(Defect, exc.msg)
|
|
|
|
proc connectClient*(
|
|
address = initTAddress("127.0.0.1:8888"),
|
|
path = WSPath,
|
|
protocols: seq[string] = @["proto"],
|
|
flags: set[TLSFlags] = {TLSFlags.NoVerifyHost, TLSFlags.NoVerifyServerName},
|
|
version = WSDefaultVersion,
|
|
frameSize = WSDefaultFrameSize,
|
|
onPing: ControlCb = nil,
|
|
onPong: ControlCb = nil,
|
|
onClose: CloseCb = nil,
|
|
rng: Rng = nil): Future[WSSession] {.async.} =
|
|
let secure = when defined secure: true else: false
|
|
return await WebSocket.connect(
|
|
address = address,
|
|
flags = flags,
|
|
path = path,
|
|
secure = secure,
|
|
protocols = protocols,
|
|
version = version,
|
|
frameSize = frameSize,
|
|
onPing = onPing,
|
|
onPong = onPong,
|
|
onClose = onClose,
|
|
rng = rng)
|