add file_server and file_client tests for scheduler/socket

1 files changed, 33 insertions, 0 deletions

diff --git a/tests/modules/socket/sched_tcp/file_server.lua b/tests/modules/socket/sched_tcp/file_server.lua
new file mode 100644
index 000000000..904b3db8e
--- /dev/null
+++ b/tests/modules/socket/sched_tcp/file_server.lua
@@ -0,0 +1,33 @@
+import("core.base.socket")
+import("core.base.scheduler")
+
+function _session(sock, filepath)
+
+    local file = io.open(filepath, 'rb')
+    if file then
+        local send = sock:sendfile(file, {block = true})
+        print("%s: send %s %d bytes!", sock, filepath, send)
+        file:close()
+    end
+    sock:close()
+end
+
+function _listen(addr, port, filepath)
+
+    local sock = socket.bind(addr, port)
+    sock:listen(20)
+    print("%s: listening %s:%d ..", sock, addr, port) 
+    while true do 
+        local sock_client = sock:accept()
+        if sock_client then
+            print("%s: accepted", sock_client) 
+            scheduler.co_start(_session, sock_client, filepath) 
+        end
+    end
+    sock:close()
+end
+
+function main(filepath)
+    scheduler.co_start(_listen, "127.0.0.1", 9090, filepath)
+    scheduler.runloop()
+end