nim-libp2p/libp2p/muxers/mplex/mplex.nim

92 lines
3.2 KiB
Nim
Raw Normal View History

2019-09-03 20:40:51 +00:00
## Nim-LibP2P
## Copyright (c) 2018 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.
2019-09-04 01:42:00 +00:00
import tables, sequtils
2019-09-03 20:40:51 +00:00
import chronos
2019-09-04 03:08:51 +00:00
import ../../varint, ../../connection,
../../vbuffer, ../../protocol,
../../stream/bufferstream,
../../stream/lpstream, ../muxer,
coder, types, channel
2019-09-03 20:40:51 +00:00
2019-09-04 03:08:51 +00:00
type
2019-09-03 20:40:51 +00:00
Mplex* = ref object of Muxer
2019-09-04 01:42:00 +00:00
remote*: Table[int, Channel]
local*: Table[int, Channel]
2019-09-03 20:40:51 +00:00
currentId*: int
maxChannels*: uint
streamHandler*: StreamHandler
2019-09-04 03:08:51 +00:00
proc newMplexUnknownMsgError(): ref MplexUnknownMsgError =
2019-09-03 20:40:51 +00:00
result = newException(MplexUnknownMsgError, "Unknown mplex message type")
2019-09-04 01:42:00 +00:00
proc getChannelList(m: Mplex, initiator: bool): var Table[int, Channel] =
if initiator:
2019-09-03 20:40:51 +00:00
result = m.remote
else:
result = m.local
2019-09-04 01:42:00 +00:00
proc newStreamInternal*(m: Mplex,
initiator: bool = true,
chanId: int):
Future[Channel] {.async, gcsafe.} =
2019-09-03 20:40:51 +00:00
## create new channel/stream
2019-09-04 01:42:00 +00:00
let id = if initiator: m.currentId.inc(); m.currentId else: chanId
2019-09-04 06:40:11 +00:00
result = newChannel(id, m.connection, initiator)
2019-09-04 01:42:00 +00:00
m.getChannelList(initiator)[id] = result
2019-09-03 20:40:51 +00:00
2019-09-04 01:42:00 +00:00
proc newStreamInternal*(m: Mplex): Future[Channel] {.gcsafe.} =
result = m.newStreamInternal(true, 0)
proc handle*(m: Mplex): Future[void] {.async, gcsafe.} =
try:
while not m.connection.closed:
let (id, msgType) = await m.connection.readHeader()
let initiator = bool(ord(msgType) and 1)
case msgType:
of MessageType.New:
let channel = await m.newStreamInternal(false, id.int)
2019-09-04 02:28:03 +00:00
channel.handlerFuture = m.streamHandler(newConnection(channel))
2019-09-04 01:42:00 +00:00
of MessageType.MsgIn, MessageType.MsgOut:
let channel = m.getChannelList(initiator)[id.int]
let msg = await m.connection.readLp()
await channel.pushTo(msg)
of MessageType.CloseIn, MessageType.CloseOut:
let channel = m.getChannelList(initiator)[id.int]
2019-09-04 06:40:11 +00:00
await channel.closeRemote()
2019-09-04 01:42:00 +00:00
of MessageType.ResetIn, MessageType.ResetOut:
let channel = m.getChannelList(initiator)[id.int]
await channel.reset()
else: raise newMplexUnknownMsgError()
except Exception as exc:
#TODO: add proper loging
discard
finally:
await m.connection.close()
2019-09-03 20:40:51 +00:00
proc newMplex*(conn: Connection,
streamHandler: StreamHandler,
maxChanns: uint = MaxChannels): Mplex =
new result
result.connection = conn
result.maxChannels = maxChanns
result.streamHandler = streamHandler
2019-09-04 01:42:00 +00:00
result.remote = initTable[int, Channel]()
result.local = initTable[int, Channel]()
2019-09-03 20:40:51 +00:00
2019-09-04 01:42:00 +00:00
method newStream*(m: Mplex): Future[Connection] {.async, gcsafe.} =
let channel = await m.newStreamInternal()
await m.connection.writeHeader(channel.id, MessageType.New, 0)
result = newConnection(channel)
2019-09-03 20:40:51 +00:00
2019-09-04 01:42:00 +00:00
method close*(m: Mplex) {.async, gcsafe.} =
await allFutures(@[allFutures(toSeq(m.remote.values).mapIt(it.close())),
allFutures(toSeq(m.local.values).mapIt(it.close()))])
2019-09-03 20:40:51 +00:00
await m.connection.close()