56 lines
1.6 KiB
Nim
56 lines
1.6 KiB
Nim
# Nim-WebRTC
|
|
# Copyright (c) 2023 Status Research & Development GmbH
|
|
# Licensed under either of
|
|
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
|
# * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
|
# at your option.
|
|
# This file may not be copied, modified, or distributed except according to
|
|
# those terms.
|
|
|
|
import sequtils
|
|
import chronos, chronicles
|
|
import webrtc_connection
|
|
|
|
logScope:
|
|
topics = "webrtc udp"
|
|
|
|
type
|
|
UdpConn* = ref object of WebRTCConn
|
|
udp: DatagramTransport
|
|
remote: TransportAddress
|
|
recvData: seq[seq[byte]]
|
|
recvEvent: AsyncEvent
|
|
|
|
method init(self: UdpConn, conn: WebRTCConn, addrss: TransportAddress) {.async.} =
|
|
await procCall(WebRTCConn(self).init(conn, addrss))
|
|
|
|
proc onReceive(udp: DatagramTransport, address: TransportAddress) {.async, gcsafe.} =
|
|
let msg = udp.getMessage()
|
|
echo "\e[33m<UDP>\e[0;1m onReceive\e[0m: ", udp.getMessage().len()
|
|
self.remote = address
|
|
self.recvData.add(msg)
|
|
self.recvEvent.fire()
|
|
|
|
self.recvEvent = newAsyncEvent()
|
|
self.udp = newDatagramTransport(onReceive, local = addrss)
|
|
|
|
method close(self: UdpConn) {.async.} =
|
|
self.udp.close()
|
|
if not self.conn.isNil():
|
|
await self.conn.close()
|
|
|
|
method write(self: UdpConn, msg: seq[byte]) {.async.} =
|
|
echo "\e[33m<UDP>\e[0;1m write\e[0m"
|
|
await self.udp.sendTo(self.remote, msg)
|
|
|
|
method read(self: UdpConn): Future[seq[byte]] {.async.} =
|
|
echo "\e[33m<UDP>\e[0;1m read\e[0m"
|
|
while self.recvData.len() <= 0:
|
|
self.recvEvent.clear()
|
|
await self.recvEvent.wait()
|
|
result = self.recvData[0]
|
|
self.recvData.delete(0..0)
|
|
|
|
method getRemoteAddress*(self: UdpConn): TransportAddress =
|
|
self.remote
|