58 lines
1.5 KiB
Nim
58 lines
1.5 KiB
Nim
import std/strutils
|
|
|
|
import pkg/chronos
|
|
import pkg/stew/byteutils
|
|
import pkg/testground_sdk
|
|
|
|
testground(client):
|
|
let
|
|
myId = await client.signalAndWait("setup", client.testInstanceCount)
|
|
myIp = client.testSubnet.split('.')[0..1].join(".") & ".1." & $myId
|
|
serverIp = client.testSubnet.split('.')[0..1].join(".") & ".1.1"
|
|
|
|
await client.updateNetworkParameter(
|
|
NetworkConf(
|
|
network: "default",
|
|
ipv4: some myIp & "/24",
|
|
enable: true,
|
|
callback_state: "network_setup",
|
|
callback_target: some client.testInstanceCount,
|
|
routing_policy: "accept_all",
|
|
)
|
|
)
|
|
|
|
await client.waitForBarrier("network_setup", client.testInstanceCount)
|
|
|
|
let
|
|
payload = client.param(string, "payload")
|
|
count = client.param(int, "count")
|
|
printResult = client.param(bool, "printResult")
|
|
|
|
if myId == 1: # server
|
|
let
|
|
server = createStreamServer(
|
|
initTAddress(myIp & ":5050"), flags = {ReuseAddr})
|
|
|
|
connection = await server.accept()
|
|
|
|
for _ in 0 ..< count:
|
|
doAssert (await connection.write(payload.toBytes())) == payload.len
|
|
|
|
connection.close()
|
|
|
|
else: # client
|
|
let
|
|
connection = await connect(initTAddress(serverIp & ":5050"))
|
|
|
|
var
|
|
buffer = newSeq[byte](payload.len)
|
|
|
|
for _ in 0 ..< count:
|
|
await connection.readExactly(addr buffer[0], payload.len)
|
|
doAssert string.fromBytes(buffer) == payload
|
|
|
|
connection.close()
|
|
|
|
if printResult:
|
|
client.recordMessage("Hourray " & $myId & "!")
|