gandc/mtprotoproxy
1
0
Fork 0
mirror of https://github.com/alexbers/mtprotoproxy.git synced 2026-03-22 10:35:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

add user data quotas

Browse Source
This commit is contained in:
Alexander Bersenev
2019-07-22 21:34:09 +05:00
parent 129f5cc981
commit 985e3eb546
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mtprotoproxy.py
View File

@@ -125,6 +125,9 @@ def init_config():
expiration = datetime.datetime.strptime(conf_dict["USER_EXPIRATIONS"][user], "%d/%m/%Y") expiration = datetime.datetime.strptime(conf_dict["USER_EXPIRATIONS"][user], "%d/%m/%Y")
conf_dict["USER_EXPIRATIONS"][user] = expiration conf_dict["USER_EXPIRATIONS"][user] = expiration
# the data quota for user
conf_dict.setdefault("USER_DATA_QUOTA", {})
# length of used handshake randoms for active fingerprinting protection # length of used handshake randoms for active fingerprinting protection
conf_dict.setdefault("REPLAY_CHECK_LEN", 32768) conf_dict.setdefault("REPLAY_CHECK_LEN", 32768)
@@ -1104,7 +1107,12 @@ async def handle_client(reader_clt, writer_clt):
datetime.datetime.now() > config.USER_EXPIRATIONS[user] datetime.datetime.now() > config.USER_EXPIRATIONS[user]
) )
if (not tcp_limit_hit) and (not user_expired): user_data_quota_hit = (
user in config.USER_DATA_QUOTA and
stats[user]["octets"] > config.USER_DATA_QUOTA[user]
)
if (not tcp_limit_hit) and (not user_expired) and (not user_data_quota_hit):
await asyncio.wait([task_tg_to_clt, task_clt_to_tg], return_when=asyncio.FIRST_COMPLETED) await asyncio.wait([task_tg_to_clt, task_clt_to_tg], return_when=asyncio.FIRST_COMPLETED)
update_stats(user, curr_connects=-1) update_stats(user, curr_connects=-1)