improve udp demo

1 files changed, 7 insertions, 3 deletions

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()