82 lines
2.1 KiB
Nim
82 lines
2.1 KiB
Nim
|
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 =
|
||
|
|
when defined secure:
|
||
|
|
TlsHttpServer.create(
|
||
|
|
address = address,
|
||
|
|
tlsPrivateKey = tlsPrivateKey,
|
||
|
|
tlsCertificate = tlsCertificate,
|
||
|
|
handler = handler,
|
||
|
|
flags = flags,
|
||
|
|
tlsFlags = tlsFlags,
|
||
|
|
tlsMinVersion = tlsMinVersion,
|
||
|
|
tlsMaxVersion = tlsMaxVersion)
|
||
|
|
else:
|
||
|
|
HttpServer.create(
|
||
|
|
address = address,
|
||
|
|
handler = handler,
|
||
|
|
flags = flags)
|
||
|
|
|
||
|
|
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)
|