diff options
author | Matthew Wild <mwild1@gmail.com> | 2009-08-18 15:58:44 +0100 |
---|---|---|
committer | Matthew Wild <mwild1@gmail.com> | 2009-08-18 15:58:44 +0100 |
commit | a440c6f67a906c0d1476d6f557d928b6f124296f (patch) | |
tree | 3d434d16ce2744043080155e7d15e1887e281570 | |
parent | fa96706886852a40f6b8404ffd0f3cd92a75968d (diff) | |
download | prosody-a440c6f67a906c0d1476d6f557d928b6f124296f.tar.gz prosody-a440c6f67a906c0d1476d6f557d928b6f124296f.zip |
net.server: Add checking for nil sockets to prevent traceback
-rw-r--r-- | net/server.lua | 20 |
1 files changed, 13 insertions, 7 deletions
diff --git a/net/server.lua b/net/server.lua index 069db50f..65567d15 100644 --- a/net/server.lua +++ b/net/server.lua @@ -483,13 +483,19 @@ wrapconnection = function( server, listeners, socket, ip, serverport, clientport end
end
local _sendbuffer = function( ) -- this function sends data
- local buffer = table_concat( bufferqueue, "", 1, bufferqueuelen )
- local succ, err, byte = send( socket, buffer, 1, bufferlen )
- local count = ( succ or byte or 0 ) * STAT_UNIT
- sendtraffic = sendtraffic + count
- _sendtraffic = _sendtraffic + count
- _ = _cleanqueue and clean( bufferqueue )
- --out_put( "server.lua: sended '", buffer, "', bytes: ", tostring(succ), ", error: ", tostring(err), ", part: ", tostring(byte), ", to: ", tostring(ip), ":", tostring(clientport) )
+ local succ, err, byte, buffer, count;
+ local count;
+ if socket then
+ buffer = table_concat( bufferqueue, "", 1, bufferqueuelen )
+ succ, err, byte = send( socket, buffer, 1, bufferlen )
+ count = ( succ or byte or 0 ) * STAT_UNIT
+ sendtraffic = sendtraffic + count
+ _sendtraffic = _sendtraffic + count
+ _ = _cleanqueue and clean( bufferqueue )
+ --out_put( "server.lua: sended '", buffer, "', bytes: ", tostring(succ), ", error: ", tostring(err), ", part: ", tostring(byte), ", to: ", tostring(ip), ":", tostring(clientport) )
+ else
+ succ, err, count = false, "closed", 0;
+ end
if succ then -- sending succesful
bufferqueuelen = 0
bufferlen = 0
|