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

Support arbitrary length for the randbytes method

Browse source
This commit is contained in:
Andika Wasisto 2020-08-12 03:42:19 +07:00
parent a299141f9a
commit 9cf1a177cc
2 changed files with 15 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
quanttp/__main__.py
View file

@ -60,8 +60,6 @@ def api_randbytes():
length = int(request.args.get('length')) length = int(request.args.get('length'))
if length < 1: if length < 1:
return Response('length must be greater than 0', status=400, content_type='text/plain') 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:
@ -89,7 +87,7 @@ 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 not 1 <= length <= 8192: if length < 1:
raise ValueError() raise ValueError()
qng_wrapper.clear() qng_wrapper.clear()
websocket.send(qng_wrapper.randbytes(length)) websocket.send(qng_wrapper.randbytes(length))
@ -113,7 +111,7 @@ 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 not 1 <= chunk <= 8192: if chunk < 1:
raise ValueError() raise ValueError()
qng_wrapper.clear() qng_wrapper.clear()
if not subscribed[0]: if not subscribed[0]:

14
quanttp/data/qng_wrapper.py
View file

@ -49,10 +49,22 @@ class QngWrapper:
def randbytes(self, length): def randbytes(self, length):
try: try:
return bytes(self.qng.RandBytes(length)) return self._randbytes_arbitrary_length(length)
except: except:
self.qng.Reset() self.qng.Reset()
return self._randbytes_arbitrary_length(length)
def _randbytes_arbitrary_length(self, length):
if length <= 8192:
return bytes(self.qng.RandBytes(length)) return bytes(self.qng.RandBytes(length))
else:
data = bytearray()
for x in range(length // 8192):
data.extend(bytearray(self.qng.RandBytes(8192)))
bytes_needed = length % 8192
if bytes_needed != 0:
data.extend(bytearray(self.qng.RandBytes(bytes_needed)))
return bytes(data)
def clear(self): def clear(self):
self.qng.Clear() self.qng.Clear()