From 048a78be2e1cb898077c46b82e847dcbe0bbc1e4 Mon Sep 17 00:00:00 2001 From: mwiegand Date: Sun, 20 Jun 2021 18:09:48 +0200 Subject: [PATCH] wip --- libs/derive_string.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 libs/derive_string.py diff --git a/libs/derive_string.py b/libs/derive_string.py new file mode 100644 index 0000000..d4693f1 --- /dev/null +++ b/libs/derive_string.py @@ -0,0 +1,32 @@ +#!/usr/bin/env python + +from hashlib import sha3_256 +from itertools import count +from Crypto.Cipher import ChaCha20 +from math import ceil + +def derive_string(input, length, choices): + result = "" + cipher = ChaCha20.new(key=sha3_256(input).digest()) + + print(len(choices)) + + for pow in count(): + if 2**pow > len(choices): + break + + print(pow) + + while len(result) < length: + seek = int.from_bytes(cipher.encrypt(b'0'*ceil(pow/8)), byteorder='little') + print(seek, len(choices)) + if seek < len(choices): + result += choices[seek] + else: + continue + + return result + +print( + derive_string(b'12345', length=100, choices='abcde12345') +)