txlyre 7 mesiacov pred
rodič
commit
30ce0480fc
1 zmenil súbory, kde vykonal 4 pridanie a 4 odobranie
  1. 4 4
      main.py

+ 4 - 4
main.py

@@ -19,6 +19,7 @@ except ImportError:
 
 tmp_dir = None
 
+
 class TemporaryFile:
     def __init__(self, name, io, delete):
         self.name = name
@@ -130,9 +131,9 @@ def extract_lsbs(data):
 
     for chunk in chunks(data, 2):
         tmp_byte = 0
-        for byte in chunk:
+        for i, byte in zip(range(len(chunk)), chunk):
             for n in range(4):
-                tmp_byte |= 1 << ((byte >> n) & 1)
+                tmp_byte |= 1 << ((byte >> (i + n)) & 1)
 
         buffer.append(tmp_byte & 255)
 
@@ -145,7 +146,7 @@ def whiten(data):
     buffer = b""
     for chunk in chunks(data, 256):
         if len(chunk) < 256:
-          break
+            break
 
         buffer += BLAKE2b.new(data=chunk, digest_bits=256).digest()
 
@@ -240,7 +241,6 @@ def sample(source, source_type, multiplier=1):
 def push(pool_url, data, secret):
     logger.info(f"Pushing {len(data)}b.")
 
-
     resp = requests.post(
         f"{pool_url}/api/pool", data=data, headers={"X-Secret": secret}
     )