aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorMatthew Wild <mwild1@gmail.com>2009-08-18 15:58:44 +0100
committerMatthew Wild <mwild1@gmail.com>2009-08-18 15:58:44 +0100
commita440c6f67a906c0d1476d6f557d928b6f124296f (patch)
tree3d434d16ce2744043080155e7d15e1887e281570
parentfa96706886852a40f6b8404ffd0f3cd92a75968d (diff)
downloadprosody-a440c6f67a906c0d1476d6f557d928b6f124296f.tar.gz
prosody-a440c6f67a906c0d1476d6f557d928b6f124296f.zip
net.server: Add checking for nil sockets to prevent traceback
-rw-r--r--net/server.lua20
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