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

Handle invalid requests

Browse source
This commit is contained in:
Andika Wasisto 2020-07-11 18:22:44 +07:00
parent 3f3caa446e
commit fd554c5d2d
1 changed files with 12 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
quanttp/__main__.py
View file

@ -58,6 +58,10 @@ def api_randnormal():
def api_randbytes(): def api_randbytes():
try: try:
length = int(request.args.get('length')) length = int(request.args.get('length'))
if length < 1:
return Response('length must be greater than 0', status=400, content_type='text/plain')
elif length > 8192:
return Response('length must be less than or equal to 8192', status=400, content_type='text/plain')
qng_wrapper.clear() qng_wrapper.clear()
return Response(qng_wrapper.randbytes(length), content_type='application/octet-stream') return Response(qng_wrapper.randbytes(length), content_type='application/octet-stream')
except ValueError as e: except ValueError as e:
@ -85,6 +89,10 @@ 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:
raise Exception('length must be greater than 0')
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':
@ -107,6 +115,10 @@ 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:
raise Exception('chunk must be greater than 0')
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