RanchiMall/electrumx
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Revert "Add feature to rate limit connections"

Browse Source 
This reverts commit a82ab0b075.
This commit is contained in:
Neil Booth 2016-12-05 23:27:36 +09:00
parent bc9027094b
commit 643c990684
3 changed files with 11 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
docs/ENV-NOTES
View File

@ -83,9 +83,6 @@ BANDWIDTH_LIMIT - per-session periodic bandwith usage limit in bytes.
end of each period. Currently the period is
hard-coded to be one hour. The default limit value
is 2 million bytes.
CONN_LIMIT - the number of new incoming connections is limited to
CONN_LIMIT_SECS CONN_LIMIT connections every CONN_LIMIT_SECS.
The default is 10 every 10 seconds.
SESSION_TIMEOUT - an integer number of seconds defaulting to 600.
Sessions with no activity for longer than this are
disconnected.

2
server/env.py
View File

@ -50,8 +50,6 @@ class Env(LoggedClass):
self.max_subs = self.integer('MAX_SUBS', 250000)
self.max_session_subs = self.integer('MAX_SESSION_SUBS', 50000)
self.bandwidth_limit = self.integer('BANDWIDTH_LIMIT', 2000000)
self.conn_limit = self.integer('CONN_LIMIT', 10)
self.conn_limit_secs = self.integer('CONN_LIMIT_SECS', 10)
self.session_timeout = self.integer('SESSION_TIMEOUT', 600)
# IRC
self.irc = self.default('IRC', False)

37
server/protocol.py
View File

@ -233,15 +233,11 @@ class ServerManager(util.LoggedClass):
self.next_stale_check = 0
self.history_cache = pylru.lrucache(512)
self.futures = []
self.cl_kizami = 0
self.cl_session_id = 0
env.max_send = max(350000, env.max_send)
self.logger.info('session timeout: {:,d} seconds'
.format(env.session_timeout))
self.logger.info('session bandwidth limit {:,d} bytes'
.format(env.bandwidth_limit))
self.logger.info('new sessions limited to {:,d} every {:,d}s'
.format(env.conn_limit, env.conn_limit_secs))
self.logger.info('max response size {:,d} bytes'.format(env.max_send))
self.logger.info('max subscriptions across all sessions: {:,d}'
.format(self.max_subs))
@ -383,30 +379,18 @@ class ServerManager(util.LoggedClass):
.format(len(self.sessions)))
def add_session(self, session):
now = time.time()
self.clear_stale_sessions(now)
if now > self.cl_kizami:
self.cl_kizami = now + self.env.conn_limit_secs
self.cl_session_id = session.id_
count = session.id_ - self.cl_session_id
if count > self.env.conn_limit:
session.log_info('closing connection from {}: {:,d} in last {:,d}s'
.format(session.peername(), count,
self.env.conn_limit_secs))
session.transport.close()
else:
coro = session.serve_requests()
future = asyncio.ensure_future(coro)
self.sessions[session] = future
session.log_info('connection from {}, {:,d} total'
.format(session.peername(), len(self.sessions)))
# Some connections are acknowledged after the servers are closed
if not self.servers:
self.close_session(session)
self.clear_stale_sessions()
coro = session.serve_requests()
future = asyncio.ensure_future(coro)
self.sessions[session] = future
session.log_info('connection from {}, {:,d} total'
.format(session.peername(), len(self.sessions)))
# Some connections are acknowledged after the servers are closed
if not self.servers:
self.close_session(session)
def remove_session(self, session):
# It might have been forcefully removed earlier by close_session()
# or never added because of connection rate limiting
if session in self.sessions:
self.subscription_count -= session.sub_count()
future = self.sessions.pop(session)
@ -423,8 +407,9 @@ class ServerManager(util.LoggedClass):
session.log_me = not session.log_me
return 'log {:d}: {}'.format(session.id_, session.log_me)
def clear_stale_sessions(self, now):
def clear_stale_sessions(self):
'''Cut off sessions that haven't done anything for 10 minutes.'''
now = time.time()
if now > self.next_stale_check:
self.next_stale_check = now + 60
cutoff = now - self.env.session_timeout