improve pushpointer

1 files changed, 13 insertions, 11 deletions

diff --git a/tests/modules/socket/tcp/echo_client.lua b/tests/modules/socket/tcp/echo_client.lua
index f3d6d9bae..6c7d67068 100644
--- a/tests/modules/socket/tcp/echo_client.lua
+++ b/tests/modules/socket/tcp/echo_client.lua
@@ -5,17 +5,19 @@ function main()
     local port = 9001
     print("connect %s:%d ..", addr, port)
     local sock = socket.connect(addr, port)
-    print("%s: connected!", sock)
-    local count = 0
-    while count < 10000 do
-        local send = sock:send("hello world..", {block = true})
-        if send > 0 then
-            sock:recv(13, {block = true})
-        else
-            break
+    if sock then
+        print("%s: connected!", sock)
+        local count = 0
+        while count < 10000 do
+            local send = sock:send("hello world..", {block = true})
+            if send > 0 then
+                sock:recv(13, {block = true})
+            else
+                break
+            end
+            count = count + 1
         end
-        count = count + 1
+        print("%s: send ok, count: %d!", sock, count)
+        sock:close()
     end
-    print("%s: send ok, count: %d!", sock, count)
-    sock:close()
 end