about summary refs log tree commit diff stats
path: root/core
diff options
context:
space:
mode:
Diffstat (limited to 'core')
0 files changed, 0 insertions, 0 deletions
 

generated by cgit-pink 1.4.1-2-gfad0 (git 2.36.2.497.gbbea4dcf42) at 2025-07-15 10:13:03 +0000


 


='n29' href='#n29'>29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90


# It is a reproduction of the 'tnewasyncudp' test code, but using a high level
# of asynchronous procedures. Output: "5000"
import asyncdispatch, asyncnet, nativesockets, net, strutils

var msgCount = 0
var recvCount = 0

const
  messagesToSend = 100
  swarmSize = 50
  serverPort = 10333

var
  sendports = 0
  recvports = 0

proc saveSendingPort(port: Port) =
  sendports = sendports + int(port)

proc saveReceivedPort(port: Port) =
  recvports = recvports + int(port)

proc launchSwarm(serverIp: string, serverPort: Port) {.async.} =
  var i = 0

  while i < swarmSize:
    var sock = newAsyncSocket(nativesockets.AF_INET, nativesockets.SOCK_DGRAM,
                              Protocol.IPPROTO_UDP, false)

    bindAddr(sock, address = "127.0.0.1")

    let (null, localPort) = getLocalAddr(sock)

    var k = 0
    
    while k < messagesToSend:
      let message = "Message " & $(i * messagesToSend + k)

      await asyncnet.sendTo(sock, serverIp, serverPort, message)

      let (data, fromIp, fromPort) = await recvFrom(sock, 16384)

      if data == message:
        saveSendingPort(localPort)

        inc(recvCount)

      inc(k)
    
    close(sock)

    inc(i)

proc readMessages(server: AsyncSocket) {.async.} =
  let maxResponses = (swarmSize * messagesToSend)

  var i = 0
  
  while i < maxResponses:
    let (data, fromIp, fromPort) = await recvFrom(server, 16384)

    if data.startswith("Message ") and fromIp == "127.0.0.1":
      await sendTo(server, fromIp, fromPort, data)

      inc(msgCount)

      saveReceivedPort(fromPort)

    inc(i)

proc createServer() {.async.} =
  var server = newAsyncSocket(nativesockets.AF_INET, nativesockets.SOCK_DGRAM, Protocol.IPPROTO_UDP, false)
  
  bindAddr(server, Port(serverPort), "127.0.0.1")

  asyncCheck readMessages(server)

asyncCheck createServer()
asyncCheck launchSwarm("127.0.0.1", Port(serverPort))

while true:
  poll()

  if recvCount == swarmSize * messagesToSend:
    break

assert msgCount == swarmSize * messagesToSend
assert sendports == recvports

echo msgCount