improve udp demo

2 files changed, 14 insertions, 5 deletions

diff --git a/tests/modules/socket/udp/echo_client.lua b/tests/modules/socket/udp/echo_client.lua
index 2c628cc92..f46ae4716 100644
--- a/tests/modules/socket/udp/echo_client.lua
+++ b/tests/modules/socket/udp/echo_client.lua
@@ -1,10 +1,15 @@
 import("core.base.socket")
 
-function main()
+function main(data)
     local addr = "127.0.0.1"
     local port = 9001
     local sock = socket.udp()
-    local send = sock:sendto("hello world..", addr, port, {block = true})
+    local send = sock:sendto(data or "hello xmake!", addr, port, {block = true})
     print("%s: send to %s:%d %d bytes!", sock, addr, port, send)
+    local recv, data, peer_addr, peer_port = sock:recvfrom(8112, {block = true})
+    if recv > 0 then
+        print("%s: recv %d bytes from %s:%d", sock, recv, peer_addr, peer_port)
+        data:dump()
+    end
     sock:close()
 end
diff --git a/tests/modules/socket/udp/echo_server.lua b/tests/modules/socket/udp/echo_server.lua
index 98cf72f7c..352c2a6db 100644
--- a/tests/modules/socket/udp/echo_server.lua
+++ b/tests/modules/socket/udp/echo_server.lua
@@ -1,13 +1,17 @@
 import("core.base.socket")
 
 function main()
+    local addr = "127.0.0.1"
+    local port = 9001
     local sock = socket.udp()
+    sock:bind(addr, port)
     while true do 
-        print("%s: recv ..", sock)
-        local recv, data, addr, port = sock:recv(8192, {block = true})
-        print("%s: recv: %d bytes from: %s:%d", sock, recv, addr, port)
+        print("%s: recv in %s:%d ..", sock, addr, port)
+        local recv, data, peer_addr, peer_port = sock:recvfrom(8192, {block = true})
+        print("%s: recv %d bytes from: %s:%d", sock, recv, peer_addr, peer_port)
         if data then
             data:dump()
+            sock:sendto(data:str(), peer_addr, peer_port)
         end
     end
     sock:close()