improve udp demo

1 files changed, 7 insertions, 2 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