net/server.lua

Sat, 27 Dec 2008 21:20:09 +0000

author
Matthew Wild <mwild1@gmail.com>
date
Sat, 27 Dec 2008 21:20:09 +0000
changeset 658
1952fdcf1017
parent 657
7f1946174d4b
child 659
c424bec771d9
permissions
-rw-r--r--

Fix specifying ports in config, and SSL support

528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
1 --[[
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
2
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
3 server.lua by blastbeat of the luadch project
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
4
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
5 re-used here under the MIT/X Consortium License
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
6
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
7 Modifications (C) 2008 Matthew Wild, Waqas Hussain
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
8 ]]--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
9
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
10 ----------------------------------// DECLARATION //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
11
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
12 --// constants //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
13
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
14 local STAT_UNIT = 1 / ( 1024 * 1024 ) -- mb
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
15
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
16 --// lua functions //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
17
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
18 local function use( what ) return _G[ what ] end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
19
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
20 local type = use "type"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
21 local pairs = use "pairs"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
22 local ipairs = use "ipairs"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
23 local tostring = use "tostring"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
24 local collectgarbage = use "collectgarbage"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
25
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
26 --// lua libs //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
27
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
28 local table = use "table"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
29 local coroutine = use "coroutine"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
30
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
31 --// lua lib methods //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
32
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
33 local table_concat = table.concat
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
34 local table_remove = table.remove
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
35 local string_sub = use'string'.sub
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
36 local coroutine_wrap = coroutine.wrap
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
37 local coroutine_yield = coroutine.yield
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
38 local print = print;
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
39 local out_put = function () end --print;
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
40 local out_error = print;
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
41
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
42 --// extern libs //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
43
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
44 local luasec = select(2, pcall(require, "ssl"))
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
45 local luasocket = require "socket"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
46
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
47 --// extern lib methods //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
48
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
49 local ssl_wrap = ( luasec and luasec.wrap )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
50 local socket_bind = luasocket.bind
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
51 local socket_select = luasocket.select
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
52 local ssl_newcontext = ( luasec and luasec.newcontext )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
53
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
54 --// functions //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
55
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
56 local loop
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
57 local stats
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
58 local addtimer
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
59 local closeall
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
60 local addserver
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
61 local firetimer
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
62 local closesocket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
63 local removesocket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
64 local wrapserver
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
65 local wraptcpclient
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
66 local wrapsslclient
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
67
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
68 --// tables //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
69
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
70 local listener
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
71 local readlist
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
72 local writelist
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
73 local socketlist
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
74 local timelistener
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
75
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
76 --// simple data types //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
77
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
78 local _
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
79 local readlen = 0 -- length of readlist
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
80 local writelen = 0 -- lenght of writelist
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
81
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
82 local sendstat= 0
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
83 local receivestat = 0
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
84
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
85 ----------------------------------// DEFINITION //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
86
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
87 listener = { } -- key = port, value = table
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
88 readlist = { } -- array with sockets to read from
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
89 writelist = { } -- arrary with sockets to write to
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
90 socketlist = { } -- key = socket, value = wrapped socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
91 timelistener = { }
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
92
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
93 stats = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
94 return receivestat, sendstat
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
95 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
96
658
1952fdcf1017 Fix specifying ports in config, and SSL support
Matthew Wild <mwild1@gmail.com>
parents: 657
diff changeset
97 wrapserver = function( listener, socket, ip, serverport, mode, sslctx, wrapper_function ) -- this function wraps a server
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
98
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
99 local dispatch, disconnect = listener.listener, listener.disconnect -- dangerous
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
100
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
101 local wrapclient, err
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
102
630
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
103 out_put("Starting a new server on "..tostring(serverport).." with ssl: "..tostring(sslctx));
658
1952fdcf1017 Fix specifying ports in config, and SSL support
Matthew Wild <mwild1@gmail.com>
parents: 657
diff changeset
104 out_put(traceback())
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
105 if sslctx then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
106 if not ssl_newcontext then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
107 return nil, "luasec not found"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
108 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
109 if type( sslctx ) ~= "table" then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
110 out_error "server.lua: wrong server sslctx"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
111 return nil, "wrong server sslctx"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
112 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
113 sslctx, err = ssl_newcontext( sslctx )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
114 if not sslctx then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
115 err = err or "wrong sslctx parameters"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
116 out_error( "server.lua: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
117 return nil, err
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
118 end
658
1952fdcf1017 Fix specifying ports in config, and SSL support
Matthew Wild <mwild1@gmail.com>
parents: 657
diff changeset
119 end
1952fdcf1017 Fix specifying ports in config, and SSL support
Matthew Wild <mwild1@gmail.com>
parents: 657
diff changeset
120
1952fdcf1017 Fix specifying ports in config, and SSL support
Matthew Wild <mwild1@gmail.com>
parents: 657
diff changeset
121 if wrapper_function then
1952fdcf1017 Fix specifying ports in config, and SSL support
Matthew Wild <mwild1@gmail.com>
parents: 657
diff changeset
122 wrapclient = wrapper_function
1952fdcf1017 Fix specifying ports in config, and SSL support
Matthew Wild <mwild1@gmail.com>
parents: 657
diff changeset
123 elseif sslctx then
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
124 wrapclient = wrapsslclient
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
125 else
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
126 wrapclient = wraptcpclient
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
127 end
658
1952fdcf1017 Fix specifying ports in config, and SSL support
Matthew Wild <mwild1@gmail.com>
parents: 657
diff changeset
128
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
129 local accept = socket.accept
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
130 local close = socket.close
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
131
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
132 --// public methods of the object //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
133
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
134 local handler = { }
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
135
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
136 handler.shutdown = function( ) end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
137
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
138 --[[handler.listener = function( data, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
139 return ondata( handler, data, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
140 end]]
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
141 handler.ssl = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
142 return sslctx and true or false
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
143 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
144 handler.close = function( closed )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
145 _ = not closed and close( socket )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
146 writelen = removesocket( writelist, socket, writelen )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
147 readlen = removesocket( readlist, socket, readlen )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
148 socketlist[ socket ] = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
149 handler = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
150 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
151 handler.ip = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
152 return ip
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
153 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
154 handler.serverport = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
155 return serverport
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
156 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
157 handler.socket = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
158 return socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
159 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
160 handler.receivedata = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
161 local client, err = accept( socket ) -- try to accept
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
162 if client then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
163 local ip, clientport = client:getpeername( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
164 client:settimeout( 0 )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
165 local handler, client, err = wrapclient( listener, client, ip, serverport, clientport, mode, sslctx ) -- wrap new client socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
166 if err then -- error while wrapping ssl socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
167 return false
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
168 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
169 out_put( "server.lua: accepted new client connection from ", ip, ":", clientport )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
170 return dispatch( handler )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
171 elseif err then -- maybe timeout or something else
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
172 out_put( "server.lua: error with new client connection: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
173 return false
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
174 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
175 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
176 return handler
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
177 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
178
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
179 wrapsslclient = function( listener, socket, ip, serverport, clientport, mode, sslctx ) -- this function wraps a ssl cleint
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
180
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
181 local dispatch, disconnect = listener.listener, listener.disconnect
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
182
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
183 --// transform socket to ssl object //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
184
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
185 local err
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
186 socket, err = ssl_wrap( socket, sslctx ) -- wrap socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
187 if err then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
188 out_put( "server.lua: ssl error: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
189 return nil, nil, err -- fatal error
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
190 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
191 socket:settimeout( 0 )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
192
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
193 --// private closures of the object //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
194
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
195 local writequeue = { } -- buffer for messages to send
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
196
630
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
197 local eol, fatal_send_error, wants_closing
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
198
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
199 local sstat, rstat = 0, 0
519
cccd610a0ef9 Insert copyright/license headers
Matthew Wild <mwild1@gmail.com>
parents: 374
diff changeset
200
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
201 --// local import of socket methods //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
202
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
203 local send = socket.send
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
204 local receive = socket.receive
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
205 local close = socket.close
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
206 --local shutdown = socket.shutdown
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
207
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
208 --// public methods of the object //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
209
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
210 local handler = { }
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
211
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
212 handler.getstats = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
213 return rstat, sstat
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
214 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
215
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
216 handler.listener = function( data, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
217 return listener( handler, data, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
218 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
219 handler.ssl = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
220 return true
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
221 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
222 handler.send = function( _, data, i, j )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
223 return send( socket, data, i, j )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
224 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
225 handler.receive = function( pattern, prefix )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
226 return receive( socket, pattern, prefix )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
227 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
228 handler.shutdown = function( pattern )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
229 --return shutdown( socket, pattern )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
230 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
231 handler.close = function( closed )
630
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
232 if eol and not fatal_send_error then
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
233 -- There is data in the buffer, and we haven't experienced
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
234 -- an error trying to send yet, so we'll flush the buffer now
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
235 handler._dispatchdata();
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
236 if eol then
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
237 -- and there is *still* data in the buffer
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
238 -- we'll give up for now, and close later
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
239 wants_closing = true;
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
240 return;
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
241 end
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
242 end
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
243 close( socket )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
244 writelen = ( eol and removesocket( writelist, socket, writelen ) ) or writelen
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
245 readlen = removesocket( readlist, socket, readlen )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
246 socketlist[ socket ] = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
247 out_put "server.lua: closed handler and removed socket from list"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
248 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
249 handler.ip = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
250 return ip
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
251 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
252 handler.serverport = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
253 return serverport
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
254 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
255 handler.clientport = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
256 return clientport
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
257 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
258
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
259 handler.write = function( data )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
260 if not eol then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
261 writelen = writelen + 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
262 writelist[ writelen ] = socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
263 eol = 0
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
264 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
265 eol = eol + 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
266 writequeue[ eol ] = data
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
267 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
268 handler.writequeue = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
269 return writequeue
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
270 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
271 handler.socket = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
272 return socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
273 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
274 handler.mode = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
275 return mode
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
276 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
277 handler._receivedata = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
278 local data, err, part = receive( socket, mode ) -- receive data in "mode"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
279 if not err or ( err == "timeout" or err == "wantread" ) then -- received something
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
280 local data = data or part or ""
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
281 local count = #data * STAT_UNIT
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
282 rstat = rstat + count
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
283 receivestat = receivestat + count
562
04ee161d936b Comment out debug logging for now
Matthew Wild <mwild1@gmail.com>
parents: 561
diff changeset
284 --out_put( "server.lua: read data '", data, "', error: ", err )
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
285 return dispatch( handler, data, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
286 else -- connections was closed or fatal error
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
287 out_put( "server.lua: client ", ip, ":", clientport, " error: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
288 handler.close( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
289 disconnect( handler, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
290 writequeue = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
291 handler = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
292 return false
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
293 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
294 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
295 handler._dispatchdata = function( ) -- this function writes data to handlers
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
296 local buffer = table_concat( writequeue, "", 1, eol )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
297 local succ, err, byte = send( socket, buffer )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
298 local count = ( succ or 0 ) * STAT_UNIT
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
299 sstat = sstat + count
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
300 sendstat = sendstat + count
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
301 out_put( "server.lua: sended '", buffer, "', bytes: ", succ, ", error: ", err, ", part: ", byte, ", to: ", ip, ":", clientport )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
302 if succ then -- sending succesful
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
303 --writequeue = { }
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
304 eol = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
305 writelen = removesocket( writelist, socket, writelen ) -- delete socket from writelist
630
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
306 if wants_closing then
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
307 handler.close();
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
308 end
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
309 return true
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
310 elseif byte and ( err == "timeout" or err == "wantwrite" ) then -- want write
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
311 buffer = string_sub( buffer, byte + 1, -1 ) -- new buffer
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
312 writequeue[ 1 ] = buffer -- insert new buffer in queue
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
313 eol = 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
314 return true
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
315 else -- connection was closed during sending or fatal error
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
316 fatal_send_error = true;
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
317 out_put( "server.lua: client ", ip, ":", clientport, " error: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
318 handler.close( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
319 disconnect( handler, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
320 writequeue = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
321 handler = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
322 return false
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
323 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
324 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
325
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
326 -- // COMPAT // --
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
327
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
328 handler.getIp = handler.ip
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
329 handler.getPort = handler.clientport
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
330
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
331 --// handshake //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
332
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
333 local wrote
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
334
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
335 handler.handshake = coroutine_wrap( function( client )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
336 local err
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
337 for i = 1, 10 do -- 10 handshake attemps
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
338 _, err = client:dohandshake( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
339 if not err then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
340 out_put( "server.lua: ssl handshake done" )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
341 writelen = ( wrote and removesocket( writelist, socket, writelen ) ) or writelen
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
342 handler.receivedata = handler._receivedata -- when handshake is done, replace the handshake function with regular functions
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
343 handler.dispatchdata = handler._dispatchdata
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
344 return dispatch( handler )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
345 else
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
346 out_put( "server.lua: error during ssl handshake: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
347 if err == "wantwrite" then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
348 if wrote == nil then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
349 writelen = writelen + 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
350 writelist[ writelen ] = client
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
351 wrote = true
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
352 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
353 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
354 coroutine_yield( handler, nil, err ) -- handshake not finished
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
355 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
356 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
357 _ = err ~= "closed" and close( socket )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
358 handler.close( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
359 disconnect( handler, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
360 writequeue = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
361 handler = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
362 return false -- handshake failed
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
363 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
364 )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
365 handler.receivedata = handler.handshake
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
366 handler.dispatchdata = handler.handshake
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
367
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
368 handler.handshake( socket ) -- do handshake
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
369
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
370 socketlist[ socket ] = handler
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
371 readlen = readlen + 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
372 readlist[ readlen ] = socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
373
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
374 return handler, socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
375 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
376
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
377 wraptlsclient = function( listener, socket, ip, serverport, clientport, mode, sslctx ) -- this function wraps a tls cleint
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
378
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
379 local dispatch, disconnect = listener.listener, listener.disconnect
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
380
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
381 --// transform socket to ssl object //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
382
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
383 local err
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
384
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
385 socket:settimeout( 0 )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
386 --// private closures of the object //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
387
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
388 local writequeue = { } -- buffer for messages to send
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
389
630
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
390 local eol, fatal_send_error, wants_closing
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
391
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
392 local sstat, rstat = 0, 0
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
393
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
394 --// local import of socket methods //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
395
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
396 local send = socket.send
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
397 local receive = socket.receive
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
398 local close = socket.close
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
399 --local shutdown = socket.shutdown
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
400
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
401 --// public methods of the object //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
402
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
403 local handler = { }
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
404
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
405 handler.getstats = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
406 return rstat, sstat
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
407 end
519
cccd610a0ef9 Insert copyright/license headers
Matthew Wild <mwild1@gmail.com>
parents: 374
diff changeset
408
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
409 handler.listener = function( data, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
410 return listener( handler, data, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
411 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
412 handler.ssl = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
413 return false
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
414 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
415 handler.send = function( _, data, i, j )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
416 return send( socket, data, i, j )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
417 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
418 handler.receive = function( pattern, prefix )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
419 return receive( socket, pattern, prefix )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
420 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
421 handler.shutdown = function( pattern )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
422 --return shutdown( socket, pattern )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
423 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
424 handler.close = function( closed )
630
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
425 if eol and not fatal_send_error then
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
426 -- There is data in the buffer, and we haven't experienced
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
427 -- an error trying to send yet, so we'll flush the buffer now
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
428 handler._dispatchdata();
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
429 if eol then
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
430 -- and there is *still* data in the buffer
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
431 -- we'll give up for now, and close later
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
432 wants_closing = true;
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
433 return;
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
434 end
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
435 end
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
436 close( socket )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
437 writelen = ( eol and removesocket( writelist, socket, writelen ) ) or writelen
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
438 readlen = removesocket( readlist, socket, readlen )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
439 socketlist[ socket ] = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
440 out_put "server.lua: closed handler and removed socket from list"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
441 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
442 handler.ip = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
443 return ip
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
444 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
445 handler.serverport = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
446 return serverport
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
447 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
448 handler.clientport = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
449 return clientport
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
450 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
451
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
452 handler.write = function( data )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
453 if not eol then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
454 writelen = writelen + 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
455 writelist[ writelen ] = socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
456 eol = 0
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
457 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
458 eol = eol + 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
459 writequeue[ eol ] = data
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
460 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
461 handler.writequeue = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
462 return writequeue
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
463 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
464 handler.socket = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
465 return socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
466 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
467 handler.mode = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
468 return mode
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
469 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
470 handler._receivedata = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
471 local data, err, part = receive( socket, mode ) -- receive data in "mode"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
472 if not err or ( err == "timeout" or err == "wantread" ) then -- received something
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
473 local data = data or part or ""
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
474 local count = #data * STAT_UNIT
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
475 rstat = rstat + count
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
476 receivestat = receivestat + count
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
477 --out_put( "server.lua: read data '", data, "', error: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
478 return dispatch( handler, data, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
479 else -- connections was closed or fatal error
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
480 out_put( "server.lua: client ", ip, ":", clientport, " error: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
481 handler.close( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
482 disconnect( handler, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
483 writequeue = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
484 handler = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
485 return false
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
486 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
487 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
488 handler._dispatchdata = function( ) -- this function writes data to handlers
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
489 local buffer = table_concat( writequeue, "", 1, eol )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
490 local succ, err, byte = send( socket, buffer )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
491 local count = ( succ or 0 ) * STAT_UNIT
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
492 sstat = sstat + count
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
493 sendstat = sendstat + count
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
494 out_put( "server.lua: sended '", buffer, "', bytes: ", succ, ", error: ", err, ", part: ", byte, ", to: ", ip, ":", clientport )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
495 if succ then -- sending succesful
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
496 --writequeue = { }
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
497 eol = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
498 writelen = removesocket( writelist, socket, writelen ) -- delete socket from writelist
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
499 if handler.need_tls then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
500 out_put("server.lua: connection is ready for tls handshake");
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
501 handler.starttls(true);
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
502 end
630
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
503 if wants_closing then
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
504 handler.close();
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
505 end
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
506 return true
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
507 elseif byte and ( err == "timeout" or err == "wantwrite" ) then -- want write
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
508 buffer = string_sub( buffer, byte + 1, -1 ) -- new buffer
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
509 writequeue[ 1 ] = buffer -- insert new buffer in queue
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
510 eol = 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
511 return true
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
512 else -- connection was closed during sending or fatal error
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
513 fatal_send_error = true; -- :(
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
514 out_put( "server.lua: client ", ip, ":", clientport, " error: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
515 handler.close( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
516 disconnect( handler, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
517 writequeue = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
518 handler = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
519 return false
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
520 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
521 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
522
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
523 handler.receivedata, handler.dispatchdata = handler._receivedata, handler._dispatchdata;
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
524 -- // COMPAT // --
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
525
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
526 handler.getIp = handler.ip
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
527 handler.getPort = handler.clientport
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
529 --// handshake //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
530
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
531 local wrote, read
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
532
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
533 handler.starttls = function (now)
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
534 if not now then out_put("server.lua: we need to do tls, but delaying until later"); handler.need_tls = true; return; end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
535 out_put( "server.lua: attempting to start tls on "..tostring(socket) )
567
ae7f6167d780 Re-commit TLS fix
Matthew Wild <mwild1@gmail.com>
parents: 566
diff changeset
536 local oldsocket = socket;
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
537 socket, err = ssl_wrap( socket, sslctx ) -- wrap socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
538 out_put("sslwrapped socket is "..tostring(socket));
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
539 if err then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
540 out_put( "server.lua: ssl error: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
541 return nil, nil, err -- fatal error
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
542 end
567
ae7f6167d780 Re-commit TLS fix
Matthew Wild <mwild1@gmail.com>
parents: 566
diff changeset
543 socket:settimeout(0);
ae7f6167d780 Re-commit TLS fix
Matthew Wild <mwild1@gmail.com>
parents: 566
diff changeset
544
ae7f6167d780 Re-commit TLS fix
Matthew Wild <mwild1@gmail.com>
parents: 566
diff changeset
545 -- Add the new socket to our system
ae7f6167d780 Re-commit TLS fix
Matthew Wild <mwild1@gmail.com>
parents: 566
diff changeset
546 socketlist[ socket ] = handler
ae7f6167d780 Re-commit TLS fix
Matthew Wild <mwild1@gmail.com>
parents: 566
diff changeset
547 readlen = readlen + 1
ae7f6167d780 Re-commit TLS fix
Matthew Wild <mwild1@gmail.com>
parents: 566
diff changeset
548 readlist[ readlen ] = socket
ae7f6167d780 Re-commit TLS fix
Matthew Wild <mwild1@gmail.com>
parents: 566
diff changeset
549
ae7f6167d780 Re-commit TLS fix
Matthew Wild <mwild1@gmail.com>
parents: 566
diff changeset
550 -- Remove traces of the old socket
ae7f6167d780 Re-commit TLS fix
Matthew Wild <mwild1@gmail.com>
parents: 566
diff changeset
551 readlen = removesocket( readlist, oldsocket, readlen )
ae7f6167d780 Re-commit TLS fix
Matthew Wild <mwild1@gmail.com>
parents: 566
diff changeset
552 socketlist [ oldsocket ] = nil;
ae7f6167d780 Re-commit TLS fix
Matthew Wild <mwild1@gmail.com>
parents: 566
diff changeset
553
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
554 send = socket.send
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
555 receive = socket.receive
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
556 close = socket.close
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
557 handler.ssl = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
558 return true
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
559 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
560 handler.send = function( _, data, i, j )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
561 return send( socket, data, i, j )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
562 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
563 handler.receive = function( pattern, prefix )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
564 return receive( socket, pattern, prefix )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
565 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
566
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
567 handler.starttls = nil;
564
779ab5b99e13 Don't say we need TLS after we've already started the handshake
Matthew Wild <mwild1@gmail.com>
parents: 563
diff changeset
568 handler.need_tls = nil
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
569
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
570 handler.handshake = coroutine_wrap( function( client )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
571 local err
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
572 for i = 1, 10 do -- 10 handshake attemps
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
573 _, err = client:dohandshake( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
574 if not err then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
575 out_put( "server.lua: ssl handshake done" )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
576 writelen = ( wrote and removesocket( writelist, socket, writelen ) ) or writelen
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
577 handler.receivedata = handler._receivedata -- when handshake is done, replace the handshake function with regular functions
566
Matthew Wild <mwild1@gmail.com>
parents: 564 565
diff changeset
578 handler.dispatchdata = handler._dispatchdata;
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
579 return true;
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
580 else
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
581 out_put( "server.lua: error during ssl handshake: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
582 if err == "wantwrite" then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
583 if wrote == nil then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
584 writelen = writelen + 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
585 writelist[ writelen ] = client
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
586 wrote = true
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
587 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
588 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
589 coroutine_yield( handler, nil, err ) -- handshake not finished
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
590 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
591 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
592 _ = err ~= "closed" and close( socket )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
593 handler.close( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
594 disconnect( handler, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
595 writequeue = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
596 handler = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
597 return false -- handshake failed
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
598 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
599 )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
600 handler.receivedata = handler.handshake
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
601 handler.dispatchdata = handler.handshake
562
04ee161d936b Comment out debug logging for now
Matthew Wild <mwild1@gmail.com>
parents: 561
diff changeset
602
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
603 handler.handshake( socket ) -- do handshake
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
604 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
605 socketlist[ socket ] = handler
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
606 readlen = readlen + 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
607 readlist[ readlen ] = socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
608
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
609 return handler, socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
610 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
611
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
612 wraptcpclient = function( listener, socket, ip, serverport, clientport, mode ) -- this function wraps a socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
613
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
614 local dispatch, disconnect = listener.listener, listener.disconnect
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
615
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
616 --// private closures of the object //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
617
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
618 local writequeue = { } -- list for messages to send
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
619
630
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
620 local eol, fatal_send_error, wants_closing
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
621
620
9f9f69d67edb Make wraptcpclient set timeout to 0, and add it to the list of exported functions from server.lua
Matthew Wild <mwild1@gmail.com>
parents: 581
diff changeset
622 socket:settimeout(0);
9f9f69d67edb Make wraptcpclient set timeout to 0, and add it to the list of exported functions from server.lua
Matthew Wild <mwild1@gmail.com>
parents: 581
diff changeset
623
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
624 local rstat, sstat = 0, 0
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
625
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
626 --// local import of socket methods //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
627
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
628 local send = socket.send
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
629 local receive = socket.receive
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
630 local close = socket.close
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
631 local shutdown = socket.shutdown
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
632
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
633 --// public methods of the object //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
634
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
635 local handler = { }
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
636
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
637 handler.getstats = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
638 return rstat, sstat
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
639 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
640
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
641 handler.listener = function( data, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
642 return listener( handler, data, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
643 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
644 handler.ssl = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
645 return false
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
646 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
647 handler.send = function( _, data, i, j )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
648 return send( socket, data, i, j )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
649 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
650 handler.receive = function( pattern, prefix )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
651 return receive( socket, pattern, prefix )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
652 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
653 handler.shutdown = function( pattern )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
654 return shutdown( socket, pattern )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
655 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
656 handler.close = function( closed )
630
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
657 if eol and not fatal_send_error then
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
658 -- There is data in the buffer, and we haven't experienced
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
659 -- an error trying to send yet, so we'll flush the buffer now
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
660 handler.dispatchdata();
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
661 if eol then
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
662 -- and there is *still* data in the buffer
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
663 -- we'll give up for now, and close later
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
664 wants_closing = true;
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
665 return;
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
666 end
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
667 end
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
668 _ = not closed and shutdown( socket )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
669 _ = not closed and close( socket )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
670 writelen = ( eol and removesocket( writelist, socket, writelen ) ) or writelen
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
671 readlen = removesocket( readlist, socket, readlen )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
672 socketlist[ socket ] = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
673 out_put "server.lua: closed handler and removed socket from list"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
674 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
675 handler.ip = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
676 return ip
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
677 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
678 handler.serverport = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
679 return serverport
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
680 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
681 handler.clientport = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
682 return clientport
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
683 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
684 handler.write = function( data )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
685 if not eol then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
686 writelen = writelen + 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
687 writelist[ writelen ] = socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
688 eol = 0
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
689 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
690 eol = eol + 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
691 writequeue[ eol ] = data
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
692 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
693 handler.writequeue = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
694 return writequeue
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
695 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
696 handler.socket = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
697 return socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
698 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
699 handler.mode = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
700 return mode
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
701 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
702
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
703 handler.receivedata = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
704 local data, err, part = receive( socket, mode ) -- receive data in "mode"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
705 if not err or ( err == "timeout" or err == "wantread" ) then -- received something
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
706 local data = data or part or ""
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
707 local count = #data * STAT_UNIT
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
708 rstat = rstat + count
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
709 receivestat = receivestat + count
562
04ee161d936b Comment out debug logging for now
Matthew Wild <mwild1@gmail.com>
parents: 561
diff changeset
710 --out_put( "server.lua: read data '", data, "', error: ", err )
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
711 return dispatch( handler, data, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
712 else -- connections was closed or fatal error
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
713 out_put( "server.lua: client ", ip, ":", clientport, " error: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
714 handler.close( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
715 disconnect( handler, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
716 writequeue = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
717 handler = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
718 return false
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
719 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
720 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
721
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
722 handler.dispatchdata = function( ) -- this function writes data to handlers
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
723 local buffer = table_concat( writequeue, "", 1, eol )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
724 local succ, err, byte = send( socket, buffer )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
725 local count = ( succ or 0 ) * STAT_UNIT
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
726 sstat = sstat + count
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
727 sendstat = sendstat + count
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
728 out_put( "server.lua: sended '", buffer, "', bytes: ", succ, ", error: ", err, ", part: ", byte, ", to: ", ip, ":", clientport )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
729 if succ then -- sending succesful
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
730 --writequeue = { }
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
731 eol = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
732 writelen = removesocket( writelist, socket, writelen ) -- delete socket from writelist
630
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
733 if wants_closing then
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
734 handler.close();
e9336adb66e5 Fix data loss when closing connection with a large write queue. Thanks albert :)
Matthew Wild <mwild1@gmail.com>
parents: 620
diff changeset
735 end
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
736 return true
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
737 elseif byte and ( err == "timeout" or err == "wantwrite" ) then -- want write
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
738 buffer = string_sub( buffer, byte + 1, -1 ) -- new buffer
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
739 writequeue[ 1 ] = buffer -- insert new buffer in queue
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
740 eol = 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
741 return true
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
742 else -- connection was closed during sending or fatal error
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
743 fatal_send_error = true; -- :'-(
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
744 out_put( "server.lua: client ", ip, ":", clientport, " error: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
745 handler.close( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
746 disconnect( handler, err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
747 writequeue = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
748 handler = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
749 return false
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
750 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
751 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
752
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
753 -- // COMPAT // --
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
754
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
755 handler.getIp = handler.ip
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
756 handler.getPort = handler.clientport
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
757
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
758 socketlist[ socket ] = handler
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
759 readlen = readlen + 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
760 readlist[ readlen ] = socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
761
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
762 return handler, socket
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
763 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
764
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
765 addtimer = function( listener )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
766 timelistener[ #timelistener + 1 ] = listener
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
767 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
768
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
769 firetimer = function( listener )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
770 for i, listener in ipairs( timelistener ) do
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
771 listener( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
772 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
773 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
774
658
1952fdcf1017 Fix specifying ports in config, and SSL support
Matthew Wild <mwild1@gmail.com>
parents: 657
diff changeset
775 addserver = function( listeners, port, addr, mode, sslctx, wrapper_function ) -- this function provides a way for other scripts to reg a server
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
776 local err
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
777 if type( listeners ) ~= "table" then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
778 err = "invalid listener table"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
779 else
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
780 for name, func in pairs( listeners ) do
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
781 if type( func ) ~= "function" then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
782 --err = "invalid listener function"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
783 break
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
784 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
785 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
786 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
787 if not type( port ) == "number" or not ( port >= 0 and port <= 65535 ) then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
788 err = "invalid port"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
789 elseif listener[ port ] then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
790 err= "listeners on port '" .. port .. "' already exist"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
791 elseif sslctx and not luasec then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
792 err = "luasec not found"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
793 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
794 if err then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
795 out_error( "server.lua: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
796 return nil, err
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
797 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
798 addr = addr or "*"
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
799 local server, err = socket_bind( addr, port )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
800 if err then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
801 out_error( "server.lua: ", err )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
802 return nil, err
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
803 end
658
1952fdcf1017 Fix specifying ports in config, and SSL support
Matthew Wild <mwild1@gmail.com>
parents: 657
diff changeset
804 local handler, err = wrapserver( listeners, server, addr, port, mode, sslctx, wrapper_function ) -- wrap new server socket
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
805 if not handler then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
806 server:close( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
807 return nil, err
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
808 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
809 server:settimeout( 0 )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
810 readlen = readlen + 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
811 readlist[ readlen ] = server
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
812 listener[ port ] = listeners
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
813 socketlist[ server ] = handler
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
814 out_put( "server.lua: new server listener on ", addr, ":", port )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
815 return true
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
816 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
817
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
818 removesocket = function( tbl, socket, len ) -- this function removes sockets from a list
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
819 for i, target in ipairs( tbl ) do
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
820 if target == socket then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
821 len = len - 1
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
822 table_remove( tbl, i )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
823 return len
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
824 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
825 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
826 return len
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
827 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
828
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
829 closeall = function( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
830 for sock, handler in pairs( socketlist ) do
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
831 handler.shutdown( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
832 handler.close( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
833 socketlist[ sock ] = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
834 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
835 writelist, readlist, socketlist = { }, { }, { }
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
836 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
837
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
838 closesocket = function( socket )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
839 writelen = removesocket( writelist, socket, writelen )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
840 readlen = removesocket( readlist, socket, readlen )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
841 socketlist[ socket ] = nil
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
842 socket:close( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
843 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
844
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
845 loop = function( ) -- this is the main loop of the program
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
846 --signal_set( "hub", "run" )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
847 repeat
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
848 local read, write, err = socket_select( readlist, writelist, 1 ) -- 1 sec timeout, nice for timers
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
849 for i, socket in ipairs( write ) do -- send data waiting in writequeues
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
850 local handler = socketlist[ socket ]
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
851 if handler then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
852 handler.dispatchdata( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
853 else
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
854 closesocket( socket )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
855 out_put "server.lua: found no handler and closed socket (writelist)" -- this should not happen
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
856 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
857 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
858 for i, socket in ipairs( read ) do -- receive data
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
859 local handler = socketlist[ socket ]
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
860 if handler then
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
861 handler.receivedata( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
862 else
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
863 closesocket( socket )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
864 out_put "server.lua: found no handler and closed socket (readlist)" -- this can happen
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
865 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
866 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
867 firetimer( )
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
868 until false
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
869 return
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
870 end
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
871
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
872 ----------------------------------// BEGIN //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
873
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
874 ----------------------------------// PUBLIC INTERFACE //--
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
875
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
876 return {
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
877
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
878 add = addserver,
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
879 loop = loop,
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
880 stats = stats,
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
881 closeall = closeall,
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
882 addtimer = addtimer,
620
9f9f69d67edb Make wraptcpclient set timeout to 0, and add it to the list of exported functions from server.lua
Matthew Wild <mwild1@gmail.com>
parents: 581
diff changeset
883 wraptcpclient = wraptcpclient,
658
1952fdcf1017 Fix specifying ports in config, and SSL support
Matthew Wild <mwild1@gmail.com>
parents: 657
diff changeset
884 wrapsslclient = wrapsslclient,
528
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
885 wraptlsclient = wraptlsclient,
5fabb5aeed9d Fix GPL'ing MIT/X licensed code :)
Matthew Wild <mwild1@gmail.com>
parents: 519
diff changeset
886 }

mercurial