Software `/` code `/` prosody

Annotate

net/server_select.lua @ 5524:e9090966c803

util.prosodyctl: Initialize storagemanager on the host before initializing usermanager. This fixes brokenness when the auth provider opens the store on load (as they all do since eeea0eb2602a) (thanks nulani)

author	Matthew Wild <mwild1@gmail.com>
date	Sat, 27 Apr 2013 13:11:03 +0100
parent	5492:5d0cc5de5c51
child	5531:483f795f6f99

rev	line source
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	1 --
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	2 -- server.lua by blastbeat of the luadch project
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	3 -- Re-used here under the MIT/X Consortium License
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	4 --
2925 692b3c6c5bd2 Merge 0.6->0.7 Matthew Wild <mwild1@gmail.com> parents: 2923 2740 diff changeset	5 -- Modifications (C) 2008-2010 Matthew Wild, Waqas Hussain
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	6 --
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	7
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	8 -- // wrapping luadch stuff // --
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	9
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	10 local use = function( what )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	11 return _G[ what ]
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	12 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	13
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	14 local log, table_concat = require ("util.logger").init("socket"), table.concat;
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	15 local out_put = function (...) return log("debug", table_concat{...}); end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	16 local out_error = function (...) return log("warn", table_concat{...}); end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	17
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	18 ----------------------------------// DECLARATION //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	19
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	20 --// constants //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	21
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	22 local STAT_UNIT = 1 -- byte
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	23
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	24 --// lua functions //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	25
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	26 local type = use "type"
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	27 local pairs = use "pairs"
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	28 local ipairs = use "ipairs"
3685 63efe4eefa07 net.server_select: Made some globals local. Waqas Hussain <waqas20@gmail.com> parents: 3682 diff changeset	29 local tonumber = use "tonumber"
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	30 local tostring = use "tostring"
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	31
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	32 --// lua libs //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	33
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	34 local os = use "os"
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	35 local table = use "table"
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	36 local string = use "string"
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	37 local coroutine = use "coroutine"
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	38
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	39 --// lua lib methods //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	40
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	41 local os_difftime = os.difftime
3682 d12141cbbaa0 net.server_select: Make changes required for sub-second timer precision. Waqas Hussain <waqas20@gmail.com> parents: 3543 diff changeset	42 local math_min = math.min
d12141cbbaa0 net.server_select: Make changes required for sub-second timer precision. Waqas Hussain <waqas20@gmail.com> parents: 3543 diff changeset	43 local math_huge = math.huge
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	44 local table_concat = table.concat
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	45 local string_sub = string.sub
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	46 local coroutine_wrap = coroutine.wrap
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	47 local coroutine_yield = coroutine.yield
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	48
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	49 --// extern libs //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	50
2597 40a174e8cdf6 net.server_select: Fix traceback with no LuaSec available (thanks Nolan) Matthew Wild <mwild1@gmail.com> parents: 2582 diff changeset	51 local luasec = use "ssl"
40a174e8cdf6 net.server_select: Fix traceback with no LuaSec available (thanks Nolan) Matthew Wild <mwild1@gmail.com> parents: 2582 diff changeset	52 local luasocket = use "socket" or require "socket"
3682 d12141cbbaa0 net.server_select: Make changes required for sub-second timer precision. Waqas Hussain <waqas20@gmail.com> parents: 3543 diff changeset	53 local luasocket_gettime = luasocket.gettime
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	54
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	55 --// extern lib methods //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	56
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	57 local ssl_wrap = ( luasec and luasec.wrap )
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	58 local socket_bind = luasocket.bind
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	59 local socket_sleep = luasocket.sleep
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	60 local socket_select = luasocket.select
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	61
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	62 --// functions //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	63
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	64 local id
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	65 local loop
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	66 local stats
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	67 local idfalse
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	68 local closeall
3686 27d3e2a2a248 net.server_select: Made another global local. Waqas Hussain <waqas20@gmail.com> parents: 3685 diff changeset	69 local addsocket
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	70 local addserver
5288 2777f34adbd3 net.server_select: Fix global access Kim Alvefur <zash@zash.se> parents: 5280 diff changeset	71 local addtimer
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	72 local getserver
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	73 local wrapserver
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	74 local getsettings
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	75 local closesocket
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	76 local removesocket
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	77 local removeserver
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	78 local wrapconnection
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	79 local changesettings
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	80
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	81 --// tables //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	82
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	83 local _server
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	84 local _readlist
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	85 local _timerlist
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	86 local _sendlist
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	87 local _socketlist
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	88 local _closelist
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	89 local _readtimes
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	90 local _writetimes
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	91
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	92 --// simple data types //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	93
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	94 local _
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	95 local _readlistlen
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	96 local _sendlistlen
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	97 local _timerlistlen
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	98
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	99 local _sendtraffic
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	100 local _readtraffic
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	101
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	102 local _selecttimeout
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	103 local _sleeptime
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	104
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	105 local _starttime
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	106 local _currenttime
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	107
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	108 local _maxsendlen
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	109 local _maxreadlen
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	110
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	111 local _checkinterval
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	112 local _sendtimeout
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	113 local _readtimeout
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	114
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	115 local _timer
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	116
5331 ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	117 local _maxselectlen
ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	118 local _maxfd
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	119
3685 63efe4eefa07 net.server_select: Made some globals local. Waqas Hussain <waqas20@gmail.com> parents: 3682 diff changeset	120 local _maxsslhandshake
63efe4eefa07 net.server_select: Made some globals local. Waqas Hussain <waqas20@gmail.com> parents: 3682 diff changeset	121
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	122 ----------------------------------// DEFINITION //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	123
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	124 _server = { } -- key = port, value = table; list of listening servers
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	125 _readlist = { } -- array with sockets to read from
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	126 _sendlist = { } -- arrary with sockets to write to
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	127 _timerlist = { } -- array of timer functions
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	128 _socketlist = { } -- key = socket, value = wrapped socket (handlers)
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	129 _readtimes = { } -- key = handler, value = timestamp of last data reading
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	130 _writetimes = { } -- key = handler, value = timestamp of last data writing/sending
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	131 _closelist = { } -- handlers to close
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	132
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	133 _readlistlen = 0 -- length of readlist
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	134 _sendlistlen = 0 -- length of sendlist
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	135 _timerlistlen = 0 -- lenght of timerlist
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	136
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	137 _sendtraffic = 0 -- some stats
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	138 _readtraffic = 0
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	139
3775 f3f2a7810108 net/server_select.lua: Reduce select() timeout back to 1s Matthew Wild <mwild1@gmail.com> parents: 3687 diff changeset	140 _selecttimeout = 1 -- timeout of socket.select
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	141 _sleeptime = 0 -- time to wait at the end of every loop
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	142
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	143 _maxsendlen = 51000 * 1024 -- max len of send buffer
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	144 _maxreadlen = 25000 * 1024 -- max len of read buffer
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	145
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	146 _checkinterval = 1200000 -- interval in secs to check idle clients
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	147 _sendtimeout = 60000 -- allowed send idle time in secs
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	148 _readtimeout = 6 * 60 * 60 -- allowed read idle time in secs
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	149
5331 ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	150 _maxfd = luasocket._SETSIZE or 1024 -- We should ignore this on Windows. Perhaps by simply setting it to math.huge or something.
ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	151 _maxselectlen = luasocket._SETSIZE or 1024 -- But this still applies on Windows
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	152
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	153 _maxsslhandshake = 30 -- max handshake round-trips
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	154
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	155 ----------------------------------// PRIVATE //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	156
5331 ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	157 wrapserver = function( listeners, socket, ip, serverport, pattern, sslctx ) -- this function wraps a server -- FIXME Make sure FD < _maxfd
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	158
5331 ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	159 if socket:getfd() >= _maxfd then
ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	160 out_error("server.lua: Disallowed FD number: "..socket:getfd())
ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	161 socket:close()
ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	162 return nil, "fd-too-large"
ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	163 end
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	164
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	165 local connections = 0
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	166
4388 eded80beab58 net.server_select: Never call onincoming with nil data when a listener doesn't have an onconnect handler Matthew Wild <mwild1@gmail.com> parents: 4353 diff changeset	167 local dispatch, disconnect = listeners.onconnect, listeners.ondisconnect
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	168
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	169 local accept = socket.accept
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	170
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	171 --// public methods of the object //--
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	172
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	173 local handler = { }
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	174
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	175 handler.shutdown = function( ) end
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	176
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	177 handler.ssl = function( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	178 return sslctx ~= nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	179 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	180 handler.sslctx = function( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	181 return sslctx
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	182 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	183 handler.remove = function( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	184 connections = connections - 1
5280 1514a6147e85 net.server_select: Pause servers while they are full Kim Alvefur <zash@zash.se> parents: 5279 diff changeset	185 if handler then
1514a6147e85 net.server_select: Pause servers while they are full Kim Alvefur <zash@zash.se> parents: 5279 diff changeset	186 handler.resume( )
1514a6147e85 net.server_select: Pause servers while they are full Kim Alvefur <zash@zash.se> parents: 5279 diff changeset	187 end
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	188 end
4904 5d2ff4627b72 net.server_select: Don't close client connections when closing a server (it wasn't working anyway) Matthew Wild <mwild1@gmail.com> parents: 4903 diff changeset	189 handler.close = function()
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	190 socket:close( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	191 _sendlistlen = removesocket( _sendlist, socket, _sendlistlen )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	192 _readlistlen = removesocket( _readlist, socket, _readlistlen )
4675 80f2d952d4c5 net.server_select: Remove server from _server table when closing Matthew Wild <mwild1@gmail.com> parents: 4429 diff changeset	193 _server[ip..":"..serverport] = nil;
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	194 _socketlist[ socket ] = nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	195 handler = nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	196 socket = nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	197 --mem_free( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	198 out_put "server.lua: closed server handler and removed sockets from list"
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	199 end
5330 0a0ca2eb991b net.server_select: Be less aggressive about server pause Kim Alvefur <zash@zash.se> parents: 5289 diff changeset	200 handler.pause = function( hard )
5279 e8f7962056f5 net.server_select: Add pause and resume methods Kim Alvefur <zash@zash.se> parents: 5197 diff changeset	201 if not handler.paused then
e8f7962056f5 net.server_select: Add pause and resume methods Kim Alvefur <zash@zash.se> parents: 5197 diff changeset	202 _readlistlen = removesocket( _readlist, socket, _readlistlen )
5330 0a0ca2eb991b net.server_select: Be less aggressive about server pause Kim Alvefur <zash@zash.se> parents: 5289 diff changeset	203 if hard then
0a0ca2eb991b net.server_select: Be less aggressive about server pause Kim Alvefur <zash@zash.se> parents: 5289 diff changeset	204 _socketlist[ socket ] = nil
0a0ca2eb991b net.server_select: Be less aggressive about server pause Kim Alvefur <zash@zash.se> parents: 5289 diff changeset	205 socket:close( )
0a0ca2eb991b net.server_select: Be less aggressive about server pause Kim Alvefur <zash@zash.se> parents: 5289 diff changeset	206 socket = nil;
0a0ca2eb991b net.server_select: Be less aggressive about server pause Kim Alvefur <zash@zash.se> parents: 5289 diff changeset	207 end
5279 e8f7962056f5 net.server_select: Add pause and resume methods Kim Alvefur <zash@zash.se> parents: 5197 diff changeset	208 handler.paused = true;
e8f7962056f5 net.server_select: Add pause and resume methods Kim Alvefur <zash@zash.se> parents: 5197 diff changeset	209 end
e8f7962056f5 net.server_select: Add pause and resume methods Kim Alvefur <zash@zash.se> parents: 5197 diff changeset	210 end
5330 0a0ca2eb991b net.server_select: Be less aggressive about server pause Kim Alvefur <zash@zash.se> parents: 5289 diff changeset	211 handler.resume = function( )
5279 e8f7962056f5 net.server_select: Add pause and resume methods Kim Alvefur <zash@zash.se> parents: 5197 diff changeset	212 if handler.paused then
5330 0a0ca2eb991b net.server_select: Be less aggressive about server pause Kim Alvefur <zash@zash.se> parents: 5289 diff changeset	213 if not socket then
0a0ca2eb991b net.server_select: Be less aggressive about server pause Kim Alvefur <zash@zash.se> parents: 5289 diff changeset	214 socket = socket_bind( ip, serverport );
0a0ca2eb991b net.server_select: Be less aggressive about server pause Kim Alvefur <zash@zash.se> parents: 5289 diff changeset	215 socket:settimeout( 0 )
0a0ca2eb991b net.server_select: Be less aggressive about server pause Kim Alvefur <zash@zash.se> parents: 5289 diff changeset	216 end
5279 e8f7962056f5 net.server_select: Add pause and resume methods Kim Alvefur <zash@zash.se> parents: 5197 diff changeset	217 _readlistlen = addsocket(_readlist, socket, _readlistlen)
e8f7962056f5 net.server_select: Add pause and resume methods Kim Alvefur <zash@zash.se> parents: 5197 diff changeset	218 _socketlist[ socket ] = handler
e8f7962056f5 net.server_select: Add pause and resume methods Kim Alvefur <zash@zash.se> parents: 5197 diff changeset	219 handler.paused = false;
e8f7962056f5 net.server_select: Add pause and resume methods Kim Alvefur <zash@zash.se> parents: 5197 diff changeset	220 end
e8f7962056f5 net.server_select: Add pause and resume methods Kim Alvefur <zash@zash.se> parents: 5197 diff changeset	221 end
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	222 handler.ip = function( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	223 return ip
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	224 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	225 handler.serverport = function( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	226 return serverport
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	227 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	228 handler.socket = function( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	229 return socket
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	230 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	231 handler.readbuffer = function( )
5331 ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	232 if _readlistlen >= _maxselectlen or _sendlistlen >= _maxselectlen then
5280 1514a6147e85 net.server_select: Pause servers while they are full Kim Alvefur <zash@zash.se> parents: 5279 diff changeset	233 handler.pause( )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	234 out_put( "server.lua: refused new client connection: server full" )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	235 return false
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	236 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	237 local client, err = accept( socket ) -- try to accept
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	238 if client then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	239 local ip, clientport = client:getpeername( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	240 local handler, client, err = wrapconnection( handler, listeners, client, ip, serverport, clientport, pattern, sslctx ) -- wrap new client socket
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	241 if err then -- error while wrapping ssl socket
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	242 return false
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	243 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	244 connections = connections + 1
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	245 out_put( "server.lua: accepted new client connection from ", tostring(ip), ":", tostring(clientport), " to ", tostring(serverport))
5492 5d0cc5de5c51 net.server_select: Don't call onconnect twice for SSL connections Matthew Wild <mwild1@gmail.com> parents: 5491 diff changeset	246 if dispatch and not sslctx then -- SSL connections will notify onconnect when handshake completes
4388 eded80beab58 net.server_select: Never call onincoming with nil data when a listener doesn't have an onconnect handler Matthew Wild <mwild1@gmail.com> parents: 4353 diff changeset	247 return dispatch( handler );
eded80beab58 net.server_select: Never call onincoming with nil data when a listener doesn't have an onconnect handler Matthew Wild <mwild1@gmail.com> parents: 4353 diff changeset	248 end
eded80beab58 net.server_select: Never call onincoming with nil data when a listener doesn't have an onconnect handler Matthew Wild <mwild1@gmail.com> parents: 4353 diff changeset	249 return;
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	250 elseif err then -- maybe timeout or something else
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	251 out_put( "server.lua: error with new client connection: ", tostring(err) )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	252 return false
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	253 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	254 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	255 return handler
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	256 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	257
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	258 wrapconnection = function( server, listeners, socket, ip, serverport, clientport, pattern, sslctx ) -- this function wraps a client to a handler object
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	259
5331 ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	260 if socket:getfd() >= _maxfd then
ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	261 out_error("server.lua: Disallowed FD number: "..socket:getfd()) -- PROTIP: Switch to libevent
ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	262 socket:close( ) -- Should we send some kind of error here?
ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	263 server.pause( )
ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	264 return nil, nil, "fd-too-large"
ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	265 end
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	266 socket:settimeout( 0 )
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	267
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	268 --// local import of socket methods //--
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	269
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	270 local send
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	271 local receive
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	272 local shutdown
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	273
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	274 --// private closures of the object //--
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	275
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	276 local ssl
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	277
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	278 local dispatch = listeners.onincoming
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	279 local status = listeners.onstatus
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	280 local disconnect = listeners.ondisconnect
3002 d1f864ed44c0 net.server_select, net.server_event: Support for ondrain listener callback for when send buffer is empty Matthew Wild <mwild1@gmail.com> parents: 3001 diff changeset	281 local drain = listeners.ondrain
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	282
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	283 local bufferqueue = { } -- buffer array
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	284 local bufferqueuelen = 0 -- end of buffer array
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	285
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	286 local toclose
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	287 local fatalerror
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	288 local needtls
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	289
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	290 local bufferlen = 0
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	291
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	292 local noread = false
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	293 local nosend = false
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	294
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	295 local sendtraffic, readtraffic = 0, 0
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	296
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	297 local maxsendlen = _maxsendlen
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	298 local maxreadlen = _maxreadlen
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	299
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	300 --// public methods of the object //--
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	301
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	302 local handler = bufferqueue -- saves a table ^_^
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	303
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	304 handler.dispatch = function( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	305 return dispatch
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	306 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	307 handler.disconnect = function( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	308 return disconnect
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	309 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	310 handler.setlistener = function( self, listeners )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	311 dispatch = listeners.onincoming
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	312 disconnect = listeners.ondisconnect
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	313 status = listeners.onstatus
3002 d1f864ed44c0 net.server_select, net.server_event: Support for ondrain listener callback for when send buffer is empty Matthew Wild <mwild1@gmail.com> parents: 3001 diff changeset	314 drain = listeners.ondrain
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	315 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	316 handler.getstats = function( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	317 return readtraffic, sendtraffic
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	318 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	319 handler.ssl = function( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	320 return ssl
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	321 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	322 handler.sslctx = function ( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	323 return sslctx
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	324 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	325 handler.send = function( _, data, i, j )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	326 return send( socket, data, i, j )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	327 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	328 handler.receive = function( pattern, prefix )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	329 return receive( socket, pattern, prefix )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	330 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	331 handler.shutdown = function( pattern )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	332 return shutdown( socket, pattern )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	333 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	334 handler.setoption = function (self, option, value)
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	335 if socket.setoption then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	336 return socket:setoption(option, value);
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	337 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	338 return false, "setoption not implemented";
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	339 end
4980 2e35cfe11835 net.server_select: Never call ondisconnect() directly, go via handler:close() or handler:force_close() - fixes cases where ondisconnect() could be called multiple times for the same connection, leading to issues with s2sout retry logic. Matthew Wild <mwild1@gmail.com> parents: 4962 diff changeset	340 handler.force_close = function ( self, err )
4962 1d8fb12a9b2d net.server_select, net.server_event: Change semantics of conn:close() - always call ondisconnect (with err == nil), and never close with data in send buffer (i.e. 'now' parameter removed) Matthew Wild <mwild1@gmail.com> parents: 4904 diff changeset	341 if bufferqueuelen ~= 0 then
4980 2e35cfe11835 net.server_select: Never call ondisconnect() directly, go via handler:close() or handler:force_close() - fixes cases where ondisconnect() could be called multiple times for the same connection, leading to issues with s2sout retry logic. Matthew Wild <mwild1@gmail.com> parents: 4962 diff changeset	342 out_put("server.lua: discarding unwritten data for ", tostring(ip), ":", tostring(clientport))
4962 1d8fb12a9b2d net.server_select, net.server_event: Change semantics of conn:close() - always call ondisconnect (with err == nil), and never close with data in send buffer (i.e. 'now' parameter removed) Matthew Wild <mwild1@gmail.com> parents: 4904 diff changeset	343 bufferqueuelen = 0;
1d8fb12a9b2d net.server_select, net.server_event: Change semantics of conn:close() - always call ondisconnect (with err == nil), and never close with data in send buffer (i.e. 'now' parameter removed) Matthew Wild <mwild1@gmail.com> parents: 4904 diff changeset	344 end
4980 2e35cfe11835 net.server_select: Never call ondisconnect() directly, go via handler:close() or handler:force_close() - fixes cases where ondisconnect() could be called multiple times for the same connection, leading to issues with s2sout retry logic. Matthew Wild <mwild1@gmail.com> parents: 4962 diff changeset	345 return self:close(err);
4962 1d8fb12a9b2d net.server_select, net.server_event: Change semantics of conn:close() - always call ondisconnect (with err == nil), and never close with data in send buffer (i.e. 'now' parameter removed) Matthew Wild <mwild1@gmail.com> parents: 4904 diff changeset	346 end
4980 2e35cfe11835 net.server_select: Never call ondisconnect() directly, go via handler:close() or handler:force_close() - fixes cases where ondisconnect() could be called multiple times for the same connection, leading to issues with s2sout retry logic. Matthew Wild <mwild1@gmail.com> parents: 4962 diff changeset	347 handler.close = function( self, err )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	348 if not handler then return true; end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	349 _readlistlen = removesocket( _readlist, socket, _readlistlen )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	350 _readtimes[ handler ] = nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	351 if bufferqueuelen ~= 0 then
4962 1d8fb12a9b2d net.server_select, net.server_event: Change semantics of conn:close() - always call ondisconnect (with err == nil), and never close with data in send buffer (i.e. 'now' parameter removed) Matthew Wild <mwild1@gmail.com> parents: 4904 diff changeset	352 handler.sendbuffer() -- Try now to send any outstanding data
1d8fb12a9b2d net.server_select, net.server_event: Change semantics of conn:close() - always call ondisconnect (with err == nil), and never close with data in send buffer (i.e. 'now' parameter removed) Matthew Wild <mwild1@gmail.com> parents: 4904 diff changeset	353 if bufferqueuelen ~= 0 then -- Still not empty, so we'll try again later
1d8fb12a9b2d net.server_select, net.server_event: Change semantics of conn:close() - always call ondisconnect (with err == nil), and never close with data in send buffer (i.e. 'now' parameter removed) Matthew Wild <mwild1@gmail.com> parents: 4904 diff changeset	354 if handler then
1d8fb12a9b2d net.server_select, net.server_event: Change semantics of conn:close() - always call ondisconnect (with err == nil), and never close with data in send buffer (i.e. 'now' parameter removed) Matthew Wild <mwild1@gmail.com> parents: 4904 diff changeset	355 handler.write = nil -- ... but no further writing allowed
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	356 end
4962 1d8fb12a9b2d net.server_select, net.server_event: Change semantics of conn:close() - always call ondisconnect (with err == nil), and never close with data in send buffer (i.e. 'now' parameter removed) Matthew Wild <mwild1@gmail.com> parents: 4904 diff changeset	357 toclose = true
1d8fb12a9b2d net.server_select, net.server_event: Change semantics of conn:close() - always call ondisconnect (with err == nil), and never close with data in send buffer (i.e. 'now' parameter removed) Matthew Wild <mwild1@gmail.com> parents: 4904 diff changeset	358 return false
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	359 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	360 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	361 if socket then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	362 _ = shutdown and shutdown( socket )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	363 socket:close( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	364 _sendlistlen = removesocket( _sendlist, socket, _sendlistlen )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	365 _socketlist[ socket ] = nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	366 socket = nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	367 else
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	368 out_put "server.lua: socket already closed"
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	369 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	370 if handler then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	371 _writetimes[ handler ] = nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	372 _closelist[ handler ] = nil
4903 c576d2175fa4 net.server_select: Call ondisconnect handler, even if we closed the socket ourselves (fixes HTTP session hoarding, thanks Maranda) Matthew Wild <mwild1@gmail.com> parents: 4888 diff changeset	373 local _handler = handler;
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	374 handler = nil
4903 c576d2175fa4 net.server_select: Call ondisconnect handler, even if we closed the socket ourselves (fixes HTTP session hoarding, thanks Maranda) Matthew Wild <mwild1@gmail.com> parents: 4888 diff changeset	375 if disconnect then
4980 2e35cfe11835 net.server_select: Never call ondisconnect() directly, go via handler:close() or handler:force_close() - fixes cases where ondisconnect() could be called multiple times for the same connection, leading to issues with s2sout retry logic. Matthew Wild <mwild1@gmail.com> parents: 4962 diff changeset	376 disconnect(_handler, err or false);
2e35cfe11835 net.server_select: Never call ondisconnect() directly, go via handler:close() or handler:force_close() - fixes cases where ondisconnect() could be called multiple times for the same connection, leading to issues with s2sout retry logic. Matthew Wild <mwild1@gmail.com> parents: 4962 diff changeset	377 disconnect = nil
4903 c576d2175fa4 net.server_select: Call ondisconnect handler, even if we closed the socket ourselves (fixes HTTP session hoarding, thanks Maranda) Matthew Wild <mwild1@gmail.com> parents: 4888 diff changeset	378 end
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	379 end
3310 ddcab5be2a5e net.server_select: Fix a snippet of code that was living at the wrong indentation level Matthew Wild <mwild1@gmail.com> parents: 3008 diff changeset	380 if server then
ddcab5be2a5e net.server_select: Fix a snippet of code that was living at the wrong indentation level Matthew Wild <mwild1@gmail.com> parents: 3008 diff changeset	381 server.remove( )
ddcab5be2a5e net.server_select: Fix a snippet of code that was living at the wrong indentation level Matthew Wild <mwild1@gmail.com> parents: 3008 diff changeset	382 end
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	383 out_put "server.lua: closed client handler and removed socket from list"
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	384 return true
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	385 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	386 handler.ip = function( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	387 return ip
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	388 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	389 handler.serverport = function( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	390 return serverport
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	391 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	392 handler.clientport = function( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	393 return clientport
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	394 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	395 local write = function( self, data )
5337 c22dd451487f net.server_select: Use # operator instead of string.len Kim Alvefur <zash@zash.se> parents: 5331 diff changeset	396 bufferlen = bufferlen + #data
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	397 if bufferlen > maxsendlen then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	398 _closelist[ handler ] = "send buffer exceeded" -- cannot close the client at the moment, have to wait to the end of the cycle
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	399 handler.write = idfalse -- dont write anymore
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	400 return false
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	401 elseif socket and not _sendlist[ socket ] then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	402 _sendlistlen = addsocket(_sendlist, socket, _sendlistlen)
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	403 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	404 bufferqueuelen = bufferqueuelen + 1
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	405 bufferqueue[ bufferqueuelen ] = data
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	406 if handler then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	407 _writetimes[ handler ] = _writetimes[ handler ] or _currenttime
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	408 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	409 return true
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	410 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	411 handler.write = write
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	412 handler.bufferqueue = function( self )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	413 return bufferqueue
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	414 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	415 handler.socket = function( self )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	416 return socket
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	417 end
3001 fa00c6f563b4 net.server_select, net.server_event: Rename :pattern() method to :set_mode() for select backend, and add :set_mode() to event backend Matthew Wild <mwild1@gmail.com> parents: 3000 diff changeset	418 handler.set_mode = function( self, new )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	419 pattern = new or pattern
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	420 return pattern
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	421 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	422 handler.set_send = function ( self, newsend )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	423 send = newsend or send
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	424 return send
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	425 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	426 handler.bufferlen = function( self, readlen, sendlen )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	427 maxsendlen = sendlen or maxsendlen
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	428 maxreadlen = readlen or maxreadlen
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	429 return bufferlen, maxreadlen, maxsendlen
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	430 end
3000 43a4cb707fe4 net.server_select, net.server_event: Support for :pause() and :resume() on connections Matthew Wild <mwild1@gmail.com> parents: 2970 diff changeset	431 --TODO: Deprecate
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	432 handler.lock_read = function (self, switch)
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	433 if switch == true then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	434 local tmp = _readlistlen
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	435 _readlistlen = removesocket( _readlist, socket, _readlistlen )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	436 _readtimes[ handler ] = nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	437 if _readlistlen ~= tmp then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	438 noread = true
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	439 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	440 elseif switch == false then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	441 if noread then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	442 noread = false
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	443 _readlistlen = addsocket(_readlist, socket, _readlistlen)
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	444 _readtimes[ handler ] = _currenttime
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	445 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	446 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	447 return noread
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	448 end
3000 43a4cb707fe4 net.server_select, net.server_event: Support for :pause() and :resume() on connections Matthew Wild <mwild1@gmail.com> parents: 2970 diff changeset	449 handler.pause = function (self)
43a4cb707fe4 net.server_select, net.server_event: Support for :pause() and :resume() on connections Matthew Wild <mwild1@gmail.com> parents: 2970 diff changeset	450 return self:lock_read(true);
43a4cb707fe4 net.server_select, net.server_event: Support for :pause() and :resume() on connections Matthew Wild <mwild1@gmail.com> parents: 2970 diff changeset	451 end
43a4cb707fe4 net.server_select, net.server_event: Support for :pause() and :resume() on connections Matthew Wild <mwild1@gmail.com> parents: 2970 diff changeset	452 handler.resume = function (self)
43a4cb707fe4 net.server_select, net.server_event: Support for :pause() and :resume() on connections Matthew Wild <mwild1@gmail.com> parents: 2970 diff changeset	453 return self:lock_read(false);
43a4cb707fe4 net.server_select, net.server_event: Support for :pause() and :resume() on connections Matthew Wild <mwild1@gmail.com> parents: 2970 diff changeset	454 end
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	455 handler.lock = function( self, switch )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	456 handler.lock_read (switch)
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	457 if switch == true then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	458 handler.write = idfalse
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	459 local tmp = _sendlistlen
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	460 _sendlistlen = removesocket( _sendlist, socket, _sendlistlen )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	461 _writetimes[ handler ] = nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	462 if _sendlistlen ~= tmp then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	463 nosend = true
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	464 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	465 elseif switch == false then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	466 handler.write = write
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	467 if nosend then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	468 nosend = false
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	469 write( "" )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	470 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	471 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	472 return noread, nosend
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	473 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	474 local _readbuffer = function( ) -- this function reads data
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	475 local buffer, err, part = receive( socket, pattern ) -- receive buffer with "pattern"
3002 d1f864ed44c0 net.server_select, net.server_event: Support for ondrain listener callback for when send buffer is empty Matthew Wild <mwild1@gmail.com> parents: 3001 diff changeset	476 if not err or (err == "wantread" or err == "timeout") then -- received something
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	477 local buffer = buffer or part or ""
5337 c22dd451487f net.server_select: Use # operator instead of string.len Kim Alvefur <zash@zash.se> parents: 5331 diff changeset	478 local len = #buffer
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	479 if len > maxreadlen then
4980 2e35cfe11835 net.server_select: Never call ondisconnect() directly, go via handler:close() or handler:force_close() - fixes cases where ondisconnect() could be called multiple times for the same connection, leading to issues with s2sout retry logic. Matthew Wild <mwild1@gmail.com> parents: 4962 diff changeset	480 handler:close( "receive buffer exceeded" )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	481 return false
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	482 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	483 local count = len * STAT_UNIT
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	484 readtraffic = readtraffic + count
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	485 _readtraffic = _readtraffic + count
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	486 _readtimes[ handler ] = _currenttime
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	487 --out_put( "server.lua: read data '", buffer:gsub("[^%w%p ]", "."), "', error: ", err )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	488 return dispatch( handler, buffer, err )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	489 else -- connections was closed or fatal error
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	490 out_put( "server.lua: client ", tostring(ip), ":", tostring(clientport), " read error: ", tostring(err) )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	491 fatalerror = true
4980 2e35cfe11835 net.server_select: Never call ondisconnect() directly, go via handler:close() or handler:force_close() - fixes cases where ondisconnect() could be called multiple times for the same connection, leading to issues with s2sout retry logic. Matthew Wild <mwild1@gmail.com> parents: 4962 diff changeset	492 _ = handler and handler:force_close( err )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	493 return false
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	494 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	495 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	496 local _sendbuffer = function( ) -- this function sends data
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	497 local succ, err, byte, buffer, count;
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	498 if socket then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	499 buffer = table_concat( bufferqueue, "", 1, bufferqueuelen )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	500 succ, err, byte = send( socket, buffer, 1, bufferlen )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	501 count = ( succ or byte or 0 ) * STAT_UNIT
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	502 sendtraffic = sendtraffic + count
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	503 _sendtraffic = _sendtraffic + count
5338 3df649ec43ea net.server_select: Optimization, clean bufferqueue when it makes sense. Kim Alvefur <zash@zash.se> parents: 5337 diff changeset	504 for i = bufferqueuelen,1,-1 do
3df649ec43ea net.server_select: Optimization, clean bufferqueue when it makes sense. Kim Alvefur <zash@zash.se> parents: 5337 diff changeset	505 bufferqueue[ i ] = nil
3df649ec43ea net.server_select: Optimization, clean bufferqueue when it makes sense. Kim Alvefur <zash@zash.se> parents: 5337 diff changeset	506 end
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	507 --out_put( "server.lua: sended '", buffer, "', bytes: ", tostring(succ), ", error: ", tostring(err), ", part: ", tostring(byte), ", to: ", tostring(ip), ":", tostring(clientport) )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	508 else
4962 1d8fb12a9b2d net.server_select, net.server_event: Change semantics of conn:close() - always call ondisconnect (with err == nil), and never close with data in send buffer (i.e. 'now' parameter removed) Matthew Wild <mwild1@gmail.com> parents: 4904 diff changeset	509 succ, err, count = false, "unexpected close", 0;
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	510 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	511 if succ then -- sending succesful
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	512 bufferqueuelen = 0
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	513 bufferlen = 0
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	514 _sendlistlen = removesocket( _sendlist, socket, _sendlistlen ) -- delete socket from writelist
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	515 _writetimes[ handler ] = nil
3002 d1f864ed44c0 net.server_select, net.server_event: Support for ondrain listener callback for when send buffer is empty Matthew Wild <mwild1@gmail.com> parents: 3001 diff changeset	516 if drain then
d1f864ed44c0 net.server_select, net.server_event: Support for ondrain listener callback for when send buffer is empty Matthew Wild <mwild1@gmail.com> parents: 3001 diff changeset	517 drain(handler)
d1f864ed44c0 net.server_select, net.server_event: Support for ondrain listener callback for when send buffer is empty Matthew Wild <mwild1@gmail.com> parents: 3001 diff changeset	518 end
3398 abc4a52aef02 net.server_select: Remove 'now' parameter from starttls(), and base it on whether the sendbuffer is empty instead Matthew Wild <mwild1@gmail.com> parents: 3382 diff changeset	519 _ = needtls and handler:starttls(nil)
4962 1d8fb12a9b2d net.server_select, net.server_event: Change semantics of conn:close() - always call ondisconnect (with err == nil), and never close with data in send buffer (i.e. 'now' parameter removed) Matthew Wild <mwild1@gmail.com> parents: 4904 diff changeset	520 _ = toclose and handler:force_close( )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	521 return true
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	522 elseif byte and ( err == "timeout" or err == "wantwrite" ) then -- want write
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	523 buffer = string_sub( buffer, byte + 1, bufferlen ) -- new buffer
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	524 bufferqueue[ 1 ] = buffer -- insert new buffer in queue
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	525 bufferqueuelen = 1
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	526 bufferlen = bufferlen - byte
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	527 _writetimes[ handler ] = _currenttime
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	528 return true
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	529 else -- connection was closed during sending or fatal error
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	530 out_put( "server.lua: client ", tostring(ip), ":", tostring(clientport), " write error: ", tostring(err) )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	531 fatalerror = true
4980 2e35cfe11835 net.server_select: Never call ondisconnect() directly, go via handler:close() or handler:force_close() - fixes cases where ondisconnect() could be called multiple times for the same connection, leading to issues with s2sout retry logic. Matthew Wild <mwild1@gmail.com> parents: 4962 diff changeset	532 _ = handler and handler:force_close( err )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	533 return false
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	534 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	535 end
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	536
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	537 -- Set the sslctx
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	538 local handshake;
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	539 function handler.set_sslctx(self, new_sslctx)
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	540 sslctx = new_sslctx;
4347 4b5290dd4e58 net.server_select: Small fix to SSL handshake logic, read/wrote were being checked despite it not being possible for them to be other than nil Matthew Wild <mwild1@gmail.com> parents: 4346 diff changeset	541 local read, wrote
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	542 handshake = coroutine_wrap( function( client ) -- create handshake coroutine
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	543 local err
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	544 for i = 1, _maxsslhandshake do
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	545 _sendlistlen = ( wrote and removesocket( _sendlist, client, _sendlistlen ) ) or _sendlistlen
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	546 _readlistlen = ( read and removesocket( _readlist, client, _readlistlen ) ) or _readlistlen
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	547 read, wrote = nil, nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	548 _, err = client:dohandshake( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	549 if not err then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	550 out_put( "server.lua: ssl handshake done" )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	551 handler.readbuffer = _readbuffer -- when handshake is done, replace the handshake function with regular functions
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	552 handler.sendbuffer = _sendbuffer
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	553 _ = status and status( handler, "ssl-handshake-complete" )
5490 ab821566ede8 Backed out changeset f2631a14b953 Matthew Wild <mwild1@gmail.com> parents: 5482 diff changeset	554 if self.autostart_ssl and listeners.onconnect then
ab821566ede8 Backed out changeset f2631a14b953 Matthew Wild <mwild1@gmail.com> parents: 5482 diff changeset	555 listeners.onconnect(self);
ab821566ede8 Backed out changeset f2631a14b953 Matthew Wild <mwild1@gmail.com> parents: 5482 diff changeset	556 end
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	557 _readlistlen = addsocket(_readlist, client, _readlistlen)
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	558 return true
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	559 else
4347 4b5290dd4e58 net.server_select: Small fix to SSL handshake logic, read/wrote were being checked despite it not being possible for them to be other than nil Matthew Wild <mwild1@gmail.com> parents: 4346 diff changeset	560 if err == "wantwrite" then
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	561 _sendlistlen = addsocket(_sendlist, client, _sendlistlen)
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	562 wrote = true
4347 4b5290dd4e58 net.server_select: Small fix to SSL handshake logic, read/wrote were being checked despite it not being possible for them to be other than nil Matthew Wild <mwild1@gmail.com> parents: 4346 diff changeset	563 elseif err == "wantread" then
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	564 _readlistlen = addsocket(_readlist, client, _readlistlen)
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	565 read = true
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	566 else
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	567 break;
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	568 end
4348 5b240c6b5334 net.server_select: Update error logging logic during SSL handshake (now shows correctly when a handshake exceeded max. allowed iterations) Matthew Wild <mwild1@gmail.com> parents: 4347 diff changeset	569 err = nil;
5b240c6b5334 net.server_select: Update error logging logic during SSL handshake (now shows correctly when a handshake exceeded max. allowed iterations) Matthew Wild <mwild1@gmail.com> parents: 4347 diff changeset	570 coroutine_yield( ) -- handshake not finished
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	571 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	572 end
4348 5b240c6b5334 net.server_select: Update error logging logic during SSL handshake (now shows correctly when a handshake exceeded max. allowed iterations) Matthew Wild <mwild1@gmail.com> parents: 4347 diff changeset	573 out_put( "server.lua: ssl handshake error: ", tostring(err or "handshake too long") )
4980 2e35cfe11835 net.server_select: Never call ondisconnect() directly, go via handler:close() or handler:force_close() - fixes cases where ondisconnect() could be called multiple times for the same connection, leading to issues with s2sout retry logic. Matthew Wild <mwild1@gmail.com> parents: 4962 diff changeset	574 _ = handler and handler:force_close("ssl handshake failed")
5484 17ce73907c48 net.server_select: Normalize indentation Kim Alvefur <zash@zash.se> parents: 5482 diff changeset	575 return false, err -- handshake failed
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	576 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	577 )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	578 end
2582 d6afb6d919df net.server_select: Only allow starttls if luasec is available (thanks Nolan) Matthew Wild <mwild1@gmail.com> parents: 2581 diff changeset	579 if luasec then
4349 16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	580 handler.starttls = function( self, _sslctx)
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	581 if _sslctx then
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	582 handler:set_sslctx(_sslctx);
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	583 end
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	584 if bufferqueuelen > 0 then
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	585 out_put "server.lua: we need to do tls, but delaying until send buffer empty"
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	586 needtls = true
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	587 return
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	588 end
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	589 out_put( "server.lua: attempting to start tls on " .. tostring( socket ) )
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	590 local oldsocket, err = socket
2582 d6afb6d919df net.server_select: Only allow starttls if luasec is available (thanks Nolan) Matthew Wild <mwild1@gmail.com> parents: 2581 diff changeset	591 socket, err = ssl_wrap( socket, sslctx ) -- wrap socket
4349 16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	592 if not socket then
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	593 out_put( "server.lua: error while starting tls on client: ", tostring(err or "unknown error") )
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	594 return nil, err -- fatal error
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	595 end
4349 16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	596
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	597 socket:settimeout( 0 )
4349 16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	598
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	599 -- add the new socket to our system
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	600 send = socket.send
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	601 receive = socket.receive
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	602 shutdown = id
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	603 _socketlist[ socket ] = handler
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	604 _readlistlen = addsocket(_readlist, socket, _readlistlen)
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	605
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	606 -- remove traces of the old socket
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	607 _readlistlen = removesocket( _readlist, oldsocket, _readlistlen )
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	608 _sendlistlen = removesocket( _sendlist, oldsocket, _sendlistlen )
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	609 _socketlist[ oldsocket ] = nil
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	610
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	611 handler.starttls = nil
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	612 needtls = nil
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	613
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	614 -- Secure now (if handshake fails connection will close)
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	615 ssl = true
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	616
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	617 handler.readbuffer = handshake
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	618 handler.sendbuffer = handshake
5484 17ce73907c48 net.server_select: Normalize indentation Kim Alvefur <zash@zash.se> parents: 5482 diff changeset	619 return handshake( socket ) -- do handshake
4349 16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	620 end
4429 8e02ba75ef06 net.server_select: Autostart SSL negotiation as the last part of socket setup to avoid a traceback (#262) (thanks Maranda) Matthew Wild <mwild1@gmail.com> parents: 4399 diff changeset	621 end
2582 d6afb6d919df net.server_select: Only allow starttls if luasec is available (thanks Nolan) Matthew Wild <mwild1@gmail.com> parents: 2581 diff changeset	622
4429 8e02ba75ef06 net.server_select: Autostart SSL negotiation as the last part of socket setup to avoid a traceback (#262) (thanks Maranda) Matthew Wild <mwild1@gmail.com> parents: 4399 diff changeset	623 handler.readbuffer = _readbuffer
8e02ba75ef06 net.server_select: Autostart SSL negotiation as the last part of socket setup to avoid a traceback (#262) (thanks Maranda) Matthew Wild <mwild1@gmail.com> parents: 4399 diff changeset	624 handler.sendbuffer = _sendbuffer
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	625 send = socket.send
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	626 receive = socket.receive
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	627 shutdown = ( ssl and id ) or socket.shutdown
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	628
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	629 _socketlist[ socket ] = handler
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	630 _readlistlen = addsocket(_readlist, socket, _readlistlen)
4429 8e02ba75ef06 net.server_select: Autostart SSL negotiation as the last part of socket setup to avoid a traceback (#262) (thanks Maranda) Matthew Wild <mwild1@gmail.com> parents: 4399 diff changeset	631
8e02ba75ef06 net.server_select: Autostart SSL negotiation as the last part of socket setup to avoid a traceback (#262) (thanks Maranda) Matthew Wild <mwild1@gmail.com> parents: 4399 diff changeset	632 if sslctx and luasec then
8e02ba75ef06 net.server_select: Autostart SSL negotiation as the last part of socket setup to avoid a traceback (#262) (thanks Maranda) Matthew Wild <mwild1@gmail.com> parents: 4399 diff changeset	633 out_put "server.lua: auto-starting ssl negotiation..."
8e02ba75ef06 net.server_select: Autostart SSL negotiation as the last part of socket setup to avoid a traceback (#262) (thanks Maranda) Matthew Wild <mwild1@gmail.com> parents: 4399 diff changeset	634 handler.autostart_ssl = true;
5484 17ce73907c48 net.server_select: Normalize indentation Kim Alvefur <zash@zash.se> parents: 5482 diff changeset	635 local ok, err = handler:starttls(sslctx);
17ce73907c48 net.server_select: Normalize indentation Kim Alvefur <zash@zash.se> parents: 5482 diff changeset	636 if ok == false then
17ce73907c48 net.server_select: Normalize indentation Kim Alvefur <zash@zash.se> parents: 5482 diff changeset	637 return nil, nil, err
17ce73907c48 net.server_select: Normalize indentation Kim Alvefur <zash@zash.se> parents: 5482 diff changeset	638 end
4429 8e02ba75ef06 net.server_select: Autostart SSL negotiation as the last part of socket setup to avoid a traceback (#262) (thanks Maranda) Matthew Wild <mwild1@gmail.com> parents: 4399 diff changeset	639 end
8e02ba75ef06 net.server_select: Autostart SSL negotiation as the last part of socket setup to avoid a traceback (#262) (thanks Maranda) Matthew Wild <mwild1@gmail.com> parents: 4399 diff changeset	640
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	641 return handler, socket
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	642 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	643
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	644 id = function( )
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	645 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	646
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	647 idfalse = function( )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	648 return false
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	649 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	650
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	651 addsocket = function( list, socket, len )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	652 if not list[ socket ] then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	653 len = len + 1
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	654 list[ len ] = socket
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	655 list[ socket ] = len
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	656 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	657 return len;
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	658 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	659
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	660 removesocket = function( list, socket, len ) -- this function removes sockets from a list ( copied from copas )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	661 local pos = list[ socket ]
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	662 if pos then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	663 list[ socket ] = nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	664 local last = list[ len ]
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	665 list[ len ] = nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	666 if last ~= socket then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	667 list[ last ] = pos
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	668 list[ pos ] = last
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	669 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	670 return len - 1
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	671 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	672 return len
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	673 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	674
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	675 closesocket = function( socket )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	676 _sendlistlen = removesocket( _sendlist, socket, _sendlistlen )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	677 _readlistlen = removesocket( _readlist, socket, _readlistlen )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	678 _socketlist[ socket ] = nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	679 socket:close( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	680 --mem_free( )
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	681 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	682
3003 48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	683 local function link(sender, receiver, buffersize)
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	684 local sender_locked;
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	685 local _sendbuffer = receiver.sendbuffer;
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	686 function receiver.sendbuffer()
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	687 _sendbuffer();
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	688 if sender_locked and receiver.bufferlen() < buffersize then
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	689 sender:lock_read(false); -- Unlock now
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	690 sender_locked = nil;
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	691 end
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	692 end
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	693
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	694 local _readbuffer = sender.readbuffer;
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	695 function sender.readbuffer()
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	696 _readbuffer();
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	697 if not sender_locked and receiver.bufferlen() >= buffersize then
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	698 sender_locked = true;
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	699 sender:lock_read(true);
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	700 end
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	701 end
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	702 end
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	703
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	704 ----------------------------------// PUBLIC //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	705
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	706 addserver = function( addr, port, listeners, pattern, sslctx ) -- this function provides a way for other scripts to reg a server
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	707 local err
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	708 if type( listeners ) ~= "table" then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	709 err = "invalid listener table"
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	710 end
2968 fdd049a4e551 Merge 0.6->0.7 Matthew Wild <mwild1@gmail.com> parents: 2967 2925 diff changeset	711 if type( port ) ~= "number" or not ( port >= 0 and port <= 65535 ) then
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	712 err = "invalid port"
4322 aff627b1ce95 connlistener, server_select, prosody: Add support for binding to multiple addresses Florian Zeitz <florob@babelmonkeys.de> parents: 4145 diff changeset	713 elseif _server[ addr..":"..port ] then
aff627b1ce95 connlistener, server_select, prosody: Add support for binding to multiple addresses Florian Zeitz <florob@babelmonkeys.de> parents: 4145 diff changeset	714 err = "listeners on '[" .. addr .. "]:" .. port .. "' already exist"
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	715 elseif sslctx and not luasec then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	716 err = "luasec not found"
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	717 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	718 if err then
4322 aff627b1ce95 connlistener, server_select, prosody: Add support for binding to multiple addresses Florian Zeitz <florob@babelmonkeys.de> parents: 4145 diff changeset	719 out_error( "server.lua, [", addr, "]:", port, ": ", err )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	720 return nil, err
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	721 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	722 addr = addr or "*"
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	723 local server, err = socket_bind( addr, port )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	724 if err then
4322 aff627b1ce95 connlistener, server_select, prosody: Add support for binding to multiple addresses Florian Zeitz <florob@babelmonkeys.de> parents: 4145 diff changeset	725 out_error( "server.lua, [", addr, "]:", port, ": ", err )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	726 return nil, err
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	727 end
5331 ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	728 local handler, err = wrapserver( listeners, server, addr, port, pattern, sslctx ) -- wrap new server socket
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	729 if not handler then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	730 server:close( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	731 return nil, err
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	732 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	733 server:settimeout( 0 )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	734 _readlistlen = addsocket(_readlist, server, _readlistlen)
4322 aff627b1ce95 connlistener, server_select, prosody: Add support for binding to multiple addresses Florian Zeitz <florob@babelmonkeys.de> parents: 4145 diff changeset	735 _server[ addr..":"..port ] = handler
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	736 _socketlist[ server ] = handler
4322 aff627b1ce95 connlistener, server_select, prosody: Add support for binding to multiple addresses Florian Zeitz <florob@babelmonkeys.de> parents: 4145 diff changeset	737 out_put( "server.lua: new "..(sslctx and "ssl " or "").."server listener on '[", addr, "]:", port, "'" )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	738 return handler
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	739 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	740
4322 aff627b1ce95 connlistener, server_select, prosody: Add support for binding to multiple addresses Florian Zeitz <florob@babelmonkeys.de> parents: 4145 diff changeset	741 getserver = function ( addr, port )
aff627b1ce95 connlistener, server_select, prosody: Add support for binding to multiple addresses Florian Zeitz <florob@babelmonkeys.de> parents: 4145 diff changeset	742 return _server[ addr..":"..port ];
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	743 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	744
4322 aff627b1ce95 connlistener, server_select, prosody: Add support for binding to multiple addresses Florian Zeitz <florob@babelmonkeys.de> parents: 4145 diff changeset	745 removeserver = function( addr, port )
aff627b1ce95 connlistener, server_select, prosody: Add support for binding to multiple addresses Florian Zeitz <florob@babelmonkeys.de> parents: 4145 diff changeset	746 local handler = _server[ addr..":"..port ]
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	747 if not handler then
4322 aff627b1ce95 connlistener, server_select, prosody: Add support for binding to multiple addresses Florian Zeitz <florob@babelmonkeys.de> parents: 4145 diff changeset	748 return nil, "no server found on '[" .. addr .. "]:" .. tostring( port ) .. "'"
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	749 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	750 handler:close( )
4322 aff627b1ce95 connlistener, server_select, prosody: Add support for binding to multiple addresses Florian Zeitz <florob@babelmonkeys.de> parents: 4145 diff changeset	751 _server[ addr..":"..port ] = nil
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	752 return true
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	753 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	754
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	755 closeall = function( )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	756 for _, handler in pairs( _socketlist ) do
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	757 handler:close( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	758 _socketlist[ _ ] = nil
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	759 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	760 _readlistlen = 0
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	761 _sendlistlen = 0
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	762 _timerlistlen = 0
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	763 _server = { }
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	764 _readlist = { }
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	765 _sendlist = { }
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	766 _timerlist = { }
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	767 _socketlist = { }
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	768 --mem_free( )
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	769 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	770
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	771 getsettings = function( )
5385 d663483036db net.server, _select: Reorganise configuration of server_select to be more like server_event Kim Alvefur <zash@zash.se> parents: 5338 diff changeset	772 return {
d663483036db net.server, _select: Reorganise configuration of server_select to be more like server_event Kim Alvefur <zash@zash.se> parents: 5338 diff changeset	773 select_timeout = _selecttimeout;
d663483036db net.server, _select: Reorganise configuration of server_select to be more like server_event Kim Alvefur <zash@zash.se> parents: 5338 diff changeset	774 select_sleep_time = _sleeptime;
d663483036db net.server, _select: Reorganise configuration of server_select to be more like server_event Kim Alvefur <zash@zash.se> parents: 5338 diff changeset	775 max_send_buffer_size = _maxsendlen;
d663483036db net.server, _select: Reorganise configuration of server_select to be more like server_event Kim Alvefur <zash@zash.se> parents: 5338 diff changeset	776 max_receive_buffer_size = _maxreadlen;
d663483036db net.server, _select: Reorganise configuration of server_select to be more like server_event Kim Alvefur <zash@zash.se> parents: 5338 diff changeset	777 select_idle_check_interval = _checkinterval;
d663483036db net.server, _select: Reorganise configuration of server_select to be more like server_event Kim Alvefur <zash@zash.se> parents: 5338 diff changeset	778 send_timeout = _sendtimeout;
d663483036db net.server, _select: Reorganise configuration of server_select to be more like server_event Kim Alvefur <zash@zash.se> parents: 5338 diff changeset	779 read_timeout = _readtimeout;
d663483036db net.server, _select: Reorganise configuration of server_select to be more like server_event Kim Alvefur <zash@zash.se> parents: 5338 diff changeset	780 max_connections = _maxselectlen;
d663483036db net.server, _select: Reorganise configuration of server_select to be more like server_event Kim Alvefur <zash@zash.se> parents: 5338 diff changeset	781 max_ssl_handshake_roundtrips = _maxsslhandshake;
d663483036db net.server, _select: Reorganise configuration of server_select to be more like server_event Kim Alvefur <zash@zash.se> parents: 5338 diff changeset	782 highest_allowed_fd = _maxfd;
d663483036db net.server, _select: Reorganise configuration of server_select to be more like server_event Kim Alvefur <zash@zash.se> parents: 5338 diff changeset	783 }
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	784 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	785
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	786 changesettings = function( new )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	787 if type( new ) ~= "table" then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	788 return nil, "invalid settings table"
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	789 end
5197 beffe931b3b6 net.server_select: Better names for config options (thanks MattJ) Kim Alvefur <zash@zash.se> parents: 4983 diff changeset	790 _selecttimeout = tonumber( new.select_timeout ) or _selecttimeout
beffe931b3b6 net.server_select: Better names for config options (thanks MattJ) Kim Alvefur <zash@zash.se> parents: 4983 diff changeset	791 _sleeptime = tonumber( new.select_sleep_time ) or _sleeptime
beffe931b3b6 net.server_select: Better names for config options (thanks MattJ) Kim Alvefur <zash@zash.se> parents: 4983 diff changeset	792 _maxsendlen = tonumber( new.max_send_buffer_size ) or _maxsendlen
beffe931b3b6 net.server_select: Better names for config options (thanks MattJ) Kim Alvefur <zash@zash.se> parents: 4983 diff changeset	793 _maxreadlen = tonumber( new.max_receive_buffer_size ) or _maxreadlen
beffe931b3b6 net.server_select: Better names for config options (thanks MattJ) Kim Alvefur <zash@zash.se> parents: 4983 diff changeset	794 _checkinterval = tonumber( new.select_idle_check_interval ) or _checkinterval
beffe931b3b6 net.server_select: Better names for config options (thanks MattJ) Kim Alvefur <zash@zash.se> parents: 4983 diff changeset	795 _sendtimeout = tonumber( new.send_timeout ) or _sendtimeout
beffe931b3b6 net.server_select: Better names for config options (thanks MattJ) Kim Alvefur <zash@zash.se> parents: 4983 diff changeset	796 _readtimeout = tonumber( new.read_timeout ) or _readtimeout
5331 ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	797 _maxselectlen = new.max_connections or _maxselectlen
5197 beffe931b3b6 net.server_select: Better names for config options (thanks MattJ) Kim Alvefur <zash@zash.se> parents: 4983 diff changeset	798 _maxsslhandshake = new.max_ssl_handshake_roundtrips or _maxsslhandshake
5331 ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	799 _maxfd = new.highest_allowed_fd or _maxfd
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	800 return true
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	801 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	802
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	803 addtimer = function( listener )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	804 if type( listener ) ~= "function" then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	805 return nil, "invalid listener function"
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	806 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	807 _timerlistlen = _timerlistlen + 1
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	808 _timerlist[ _timerlistlen ] = listener
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	809 return true
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	810 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	811
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	812 stats = function( )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	813 return _readtraffic, _sendtraffic, _readlistlen, _sendlistlen, _timerlistlen
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	814 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	815
3491 e8c06d20a18b net.server_select: Add server.step() to run through a single iteration of the event loop Matthew Wild <mwild1@gmail.com> parents: 3458 diff changeset	816 local quitting;
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	817
4399 1b4161970842 net.server_event, net.server_select: Fixed some global pollution. Waqas Hussain <waqas20@gmail.com> parents: 4393 diff changeset	818 local function setquitting(quit)
3491 e8c06d20a18b net.server_select: Add server.step() to run through a single iteration of the event loop Matthew Wild <mwild1@gmail.com> parents: 3458 diff changeset	819 quitting = not not quit;
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	820 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	821
3491 e8c06d20a18b net.server_select: Add server.step() to run through a single iteration of the event loop Matthew Wild <mwild1@gmail.com> parents: 3458 diff changeset	822 loop = function(once) -- this is the main loop of the program
e8c06d20a18b net.server_select: Add server.step() to run through a single iteration of the event loop Matthew Wild <mwild1@gmail.com> parents: 3458 diff changeset	823 if quitting then return "quitting"; end
e8c06d20a18b net.server_select: Add server.step() to run through a single iteration of the event loop Matthew Wild <mwild1@gmail.com> parents: 3458 diff changeset	824 if once then quitting = "once"; end
3682 d12141cbbaa0 net.server_select: Make changes required for sub-second timer precision. Waqas Hussain <waqas20@gmail.com> parents: 3543 diff changeset	825 local next_timer_time = math_huge;
3491 e8c06d20a18b net.server_select: Add server.step() to run through a single iteration of the event loop Matthew Wild <mwild1@gmail.com> parents: 3458 diff changeset	826 repeat
3682 d12141cbbaa0 net.server_select: Make changes required for sub-second timer precision. Waqas Hussain <waqas20@gmail.com> parents: 3543 diff changeset	827 local read, write, err = socket_select( _readlist, _sendlist, math_min(_selecttimeout, next_timer_time) )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	828 for i, socket in ipairs( write ) do -- send data waiting in writequeues
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	829 local handler = _socketlist[ socket ]
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	830 if handler then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	831 handler.sendbuffer( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	832 else
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	833 closesocket( socket )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	834 out_put "server.lua: found no handler and closed socket (writelist)" -- this should not happen
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	835 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	836 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	837 for i, socket in ipairs( read ) do -- receive data
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	838 local handler = _socketlist[ socket ]
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	839 if handler then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	840 handler.readbuffer( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	841 else
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	842 closesocket( socket )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	843 out_put "server.lua: found no handler and closed socket (readlist)" -- this can happen
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	844 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	845 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	846 for handler, err in pairs( _closelist ) do
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	847 handler.disconnect( )( handler, err )
4962 1d8fb12a9b2d net.server_select, net.server_event: Change semantics of conn:close() - always call ondisconnect (with err == nil), and never close with data in send buffer (i.e. 'now' parameter removed) Matthew Wild <mwild1@gmail.com> parents: 4904 diff changeset	848 handler:force_close() -- forced disconnect
5338 3df649ec43ea net.server_select: Optimization, clean bufferqueue when it makes sense. Kim Alvefur <zash@zash.se> parents: 5337 diff changeset	849 _closelist[ handler ] = nil;
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	850 end
3682 d12141cbbaa0 net.server_select: Make changes required for sub-second timer precision. Waqas Hussain <waqas20@gmail.com> parents: 3543 diff changeset	851 _currenttime = luasocket_gettime( )
5485 5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	852
5486 bcf27dbef6c6 net.server_select: Add and improve some comments. Waqas Hussain <waqas20@gmail.com> parents: 5485 diff changeset	853 -- Check for socket timeouts
5485 5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	854 local difftime = os_difftime( _currenttime - _starttime )
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	855 if difftime > _checkinterval then
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	856 _starttime = _currenttime
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	857 for handler, timestamp in pairs( _writetimes ) do
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	858 if os_difftime( _currenttime - timestamp ) > _sendtimeout then
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	859 --_writetimes[ handler ] = nil
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	860 handler.disconnect( )( handler, "send timeout" )
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	861 handler:force_close() -- forced disconnect
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	862 end
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	863 end
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	864 for handler, timestamp in pairs( _readtimes ) do
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	865 if os_difftime( _currenttime - timestamp ) > _readtimeout then
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	866 --_readtimes[ handler ] = nil
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	867 handler.disconnect( )( handler, "read timeout" )
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	868 handler:close( ) -- forced disconnect?
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	869 end
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	870 end
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	871 end
5147eed0a42e net.server_select: Move socket timeout cleanup code out of a timer, into the select loop (which makes util.timer the only timer using server_select._addtimer). Waqas Hussain <waqas20@gmail.com> parents: 5484 diff changeset	872
5486 bcf27dbef6c6 net.server_select: Add and improve some comments. Waqas Hussain <waqas20@gmail.com> parents: 5485 diff changeset	873 -- Fire timers
3682 d12141cbbaa0 net.server_select: Make changes required for sub-second timer precision. Waqas Hussain <waqas20@gmail.com> parents: 3543 diff changeset	874 if _currenttime - _timer >= math_min(next_timer_time, 1) then
d12141cbbaa0 net.server_select: Make changes required for sub-second timer precision. Waqas Hussain <waqas20@gmail.com> parents: 3543 diff changeset	875 next_timer_time = math_huge;
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	876 for i = 1, _timerlistlen do
3682 d12141cbbaa0 net.server_select: Make changes required for sub-second timer precision. Waqas Hussain <waqas20@gmail.com> parents: 3543 diff changeset	877 local t = _timerlist[ i ]( _currenttime ) -- fire timers
d12141cbbaa0 net.server_select: Make changes required for sub-second timer precision. Waqas Hussain <waqas20@gmail.com> parents: 3543 diff changeset	878 if t then next_timer_time = math_min(next_timer_time, t); end
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	879 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	880 _timer = _currenttime
3682 d12141cbbaa0 net.server_select: Make changes required for sub-second timer precision. Waqas Hussain <waqas20@gmail.com> parents: 3543 diff changeset	881 else
d12141cbbaa0 net.server_select: Make changes required for sub-second timer precision. Waqas Hussain <waqas20@gmail.com> parents: 3543 diff changeset	882 next_timer_time = next_timer_time - (_currenttime - _timer);
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	883 end
5486 bcf27dbef6c6 net.server_select: Add and improve some comments. Waqas Hussain <waqas20@gmail.com> parents: 5485 diff changeset	884
bcf27dbef6c6 net.server_select: Add and improve some comments. Waqas Hussain <waqas20@gmail.com> parents: 5485 diff changeset	885 -- wait some time (0 by default)
bcf27dbef6c6 net.server_select: Add and improve some comments. Waqas Hussain <waqas20@gmail.com> parents: 5485 diff changeset	886 socket_sleep( _sleeptime )
3491 e8c06d20a18b net.server_select: Add server.step() to run through a single iteration of the event loop Matthew Wild <mwild1@gmail.com> parents: 3458 diff changeset	887 until quitting;
e8c06d20a18b net.server_select: Add server.step() to run through a single iteration of the event loop Matthew Wild <mwild1@gmail.com> parents: 3458 diff changeset	888 if once and quitting == "once" then quitting = nil; return; end
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	889 return "quitting"
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	890 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	891
4399 1b4161970842 net.server_event, net.server_select: Fixed some global pollution. Waqas Hussain <waqas20@gmail.com> parents: 4393 diff changeset	892 local function step()
3491 e8c06d20a18b net.server_select: Add server.step() to run through a single iteration of the event loop Matthew Wild <mwild1@gmail.com> parents: 3458 diff changeset	893 return loop(true);
e8c06d20a18b net.server_select: Add server.step() to run through a single iteration of the event loop Matthew Wild <mwild1@gmail.com> parents: 3458 diff changeset	894 end
e8c06d20a18b net.server_select: Add server.step() to run through a single iteration of the event loop Matthew Wild <mwild1@gmail.com> parents: 3458 diff changeset	895
2339 b04014f42fce net.server_select: Add get_backend() method Matthew Wild <mwild1@gmail.com> parents: 2318 diff changeset	896 local function get_backend()
b04014f42fce net.server_select: Add get_backend() method Matthew Wild <mwild1@gmail.com> parents: 2318 diff changeset	897 return "select";
b04014f42fce net.server_select: Add get_backend() method Matthew Wild <mwild1@gmail.com> parents: 2318 diff changeset	898 end
b04014f42fce net.server_select: Add get_backend() method Matthew Wild <mwild1@gmail.com> parents: 2318 diff changeset	899
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	900 --// EXPERIMENTAL //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	901
2541 2febd008214e net.server_select: Remove startssl parameter to the client/server creation functions - passing a sslctx now indicates you want to use SSL from the start Matthew Wild <mwild1@gmail.com> parents: 2478 diff changeset	902 local wrapclient = function( socket, ip, serverport, listeners, pattern, sslctx )
5331 ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	903 local handler, socket, err = wrapconnection( nil, listeners, socket, ip, serverport, "clientport", pattern, sslctx )
ffa740b4f08d net.server_select: Limit global number of sockets passed to select. Kim Alvefur <zash@zash.se> parents: 5330 diff changeset	904 if not handler then return nil, err end
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	905 _socketlist[ socket ] = handler
4349 16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	906 if not sslctx then
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	907 _sendlistlen = addsocket(_sendlist, socket, _sendlistlen)
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	908 if listeners.onconnect then
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	909 -- When socket is writeable, call onconnect
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	910 local _sendbuffer = handler.sendbuffer;
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	911 handler.sendbuffer = function ()
4353 f600591c87fa net.server_select: Remove socket from sendlist before calling listener callbacks (thanks darkrain) Matthew Wild <mwild1@gmail.com> parents: 4349 diff changeset	912 _sendlistlen = removesocket( _sendlist, socket, _sendlistlen );
4349 16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	913 handler.sendbuffer = _sendbuffer;
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	914 listeners.onconnect(handler);
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	915 -- If there was data with the incoming packet, handle it now.
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	916 if #handler:bufferqueue() > 0 then
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	917 return _sendbuffer();
16fd8061964e net.server_select: Merge straight-SSL and starttls code paths, also fixes onconnect being called before handshake completion for straight-SSL Matthew Wild <mwild1@gmail.com> parents: 4348 diff changeset	918 end
3382 fea0f8e19e4c net.server_select: Ensure that onconnect is called for server sockets as well as clients, rather than onincoming. Fixes mod_console traceback, issue #197 Matthew Wild <mwild1@gmail.com> parents: 3312 diff changeset	919 end
fea0f8e19e4c net.server_select: Ensure that onconnect is called for server sockets as well as clients, rather than onincoming. Fixes mod_console traceback, issue #197 Matthew Wild <mwild1@gmail.com> parents: 3312 diff changeset	920 end
fea0f8e19e4c net.server_select: Ensure that onconnect is called for server sockets as well as clients, rather than onincoming. Fixes mod_console traceback, issue #197 Matthew Wild <mwild1@gmail.com> parents: 3312 diff changeset	921 end
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	922 return handler, socket
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	923 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	924
2541 2febd008214e net.server_select: Remove startssl parameter to the client/server creation functions - passing a sslctx now indicates you want to use SSL from the start Matthew Wild <mwild1@gmail.com> parents: 2478 diff changeset	925 local addclient = function( address, port, listeners, pattern, sslctx )
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	926 local client, err = luasocket.tcp( )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	927 if err then
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	928 return nil, err
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	929 end
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	930 client:settimeout( 0 )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	931 _, err = client:connect( address, port )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	932 if err then -- try again
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	933 local handler = wrapclient( client, address, port, listeners )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	934 else
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	935 wrapconnection( nil, listeners, client, address, port, "clientport", pattern, sslctx )
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	936 end
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	937 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	938
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	939 --// EXPERIMENTAL //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	940
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	941 ----------------------------------// BEGIN //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	942
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	943 use "setmetatable" ( _socketlist, { __mode = "k" } )
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	944 use "setmetatable" ( _readtimes, { __mode = "k" } )
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	945 use "setmetatable" ( _writetimes, { __mode = "k" } )
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	946
3682 d12141cbbaa0 net.server_select: Make changes required for sub-second timer precision. Waqas Hussain <waqas20@gmail.com> parents: 3543 diff changeset	947 _timer = luasocket_gettime( )
d12141cbbaa0 net.server_select: Make changes required for sub-second timer precision. Waqas Hussain <waqas20@gmail.com> parents: 3543 diff changeset	948 _starttime = luasocket_gettime( )
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	949
2424 7d3b071e3b5b net.server_select: Fix oops in last commit Matthew Wild <mwild1@gmail.com> parents: 2423 diff changeset	950 local function setlogger(new_logger)
2423 0597c553acf2 net.server_select: Allow setting the logger using server.setlogger Matthew Wild <mwild1@gmail.com> parents: 2339 diff changeset	951 local old_logger = log;
0597c553acf2 net.server_select: Allow setting the logger using server.setlogger Matthew Wild <mwild1@gmail.com> parents: 2339 diff changeset	952 if new_logger then
0597c553acf2 net.server_select: Allow setting the logger using server.setlogger Matthew Wild <mwild1@gmail.com> parents: 2339 diff changeset	953 log = new_logger;
0597c553acf2 net.server_select: Allow setting the logger using server.setlogger Matthew Wild <mwild1@gmail.com> parents: 2339 diff changeset	954 end
0597c553acf2 net.server_select: Allow setting the logger using server.setlogger Matthew Wild <mwild1@gmail.com> parents: 2339 diff changeset	955 return old_logger;
0597c553acf2 net.server_select: Allow setting the logger using server.setlogger Matthew Wild <mwild1@gmail.com> parents: 2339 diff changeset	956 end
0597c553acf2 net.server_select: Allow setting the logger using server.setlogger Matthew Wild <mwild1@gmail.com> parents: 2339 diff changeset	957
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	958 ----------------------------------// PUBLIC INTERFACE //--
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	959
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	960 return {
4808 07d0a3a75c8a net.server, net.timer, net.server_select: Rearrange dependencies between these three modules. server.addtimer() is no longer a public function (renamed to _addtimer) and is not available at all from server_event (compat code removed to prevent traceback) (thanks Nulani) Matthew Wild <mwild1@gmail.com> parents: 4675 diff changeset	961 _addtimer = addtimer,
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	962
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	963 addclient = addclient,
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	964 wrapclient = wrapclient,
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	965
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	966 loop = loop,
3003 48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	967 link = link,
4145 e992650185c8 net.server_select: Expose step() function (thanks Łukasz) Matthew Wild <mwild1@gmail.com> parents: 4111 diff changeset	968 step = step,
2581 1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	969 stats = stats,
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	970 closeall = closeall,
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	971 addserver = addserver,
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	972 getserver = getserver,
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	973 setlogger = setlogger,
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	974 getsettings = getsettings,
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	975 setquitting = setquitting,
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	976 removeserver = removeserver,
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	977 get_backend = get_backend,
1b9f424e695e net.server_select: Major whitespace refactoring to meet Prosody's coding style guidelines, we now have tabs Matthew Wild <mwild1@gmail.com> parents: 2559 diff changeset	978 changesettings = changesettings,
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2312 diff changeset	979 }

prosody

e9090966c803

net/server_select.lua

Software / code / prosody

Annotate

net/server_select.lua @ 5524:e9090966c803

Software `/` code `/` prosody