randogoth/quanttp2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Ignore invalid requests

Browse source
This commit is contained in:
Andika Wasisto 2020-07-14 00:36:05 +07:00
parent fd554c5d2d
commit a299141f9a
1 changed files with 4 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

12
quanttp/__main__.py
View file

@ -89,10 +89,8 @@ def handle_ws_message(message, websocket, subscribed):
websocket.send(str(qng_wrapper.randnormal())) websocket.send(str(qng_wrapper.randnormal()))
elif split_message[0] == 'RANDBYTES': elif split_message[0] == 'RANDBYTES':
length = int(split_message[1]) length = int(split_message[1])
if length < 1: if not 1 <= length <= 8192:
raise Exception('length must be greater than 0') raise ValueError()
elif length > 8192:
raise Exception('length must be less than or equal to 8192')
qng_wrapper.clear() qng_wrapper.clear()
websocket.send(qng_wrapper.randbytes(length)) websocket.send(qng_wrapper.randbytes(length))
elif split_message[0] == 'SUBSCRIBEINT32': elif split_message[0] == 'SUBSCRIBEINT32':
@ -115,10 +113,8 @@ def handle_ws_message(message, websocket, subscribed):
websocket.send(str(qng_wrapper.randnormal())) websocket.send(str(qng_wrapper.randnormal()))
elif split_message[0] == 'SUBSCRIBEBYTES': elif split_message[0] == 'SUBSCRIBEBYTES':
chunk = int(split_message[1]) chunk = int(split_message[1])
if chunk < 1: if not 1 <= chunk <= 8192:
raise Exception('chunk must be greater than 0') raise ValueError()
elif chunk > 8192:
raise Exception('chunk must be less than or equal to 8192')
qng_wrapper.clear() qng_wrapper.clear()
if not subscribed[0]: if not subscribed[0]:
subscribed[0] = True subscribed[0] = True