control.py

import numpy as np
import qiskit
import my_secrets

BLOCK_BITS = 4
BLOCK_SIZE = 2**BLOCK_BITS
BLOCK_EL = BLOCK_SIZE**2
NBITS = 20
STATE_SIZE = 2**NBITS

encoding_dict = { c:'{:06b}'.format(v) for v,c in enumerate('ABCDEFGHIJKLMNOPQRSTUVXWYZabcdefghijklmnopqrstuvxwyz01234567{}-_')}


def encode_string(s):
    return ''.join(encoding_dict[c] for c in s )

def gen_keys(enc_s):
    if len(enc_s)<BLOCK_EL: enc_s +='0'*(BLOCK_EL-len(enc_s))
    return [enc_s[i:i+BLOCK_SIZE] for i in range(0,BLOCK_EL,BLOCK_SIZE)]

def get_states(keys,otp):
    return [[otp[int(i)][j][kn] for j,i in enumerate(k)] for kn,k in enumerate(keys)]

def initial_state(s):
    res = np.array(s)
    return res/np.linalg.norm(res)

def setup_state(vec):
    out = ( 'initialize('
            + ','.join('{:.9f}'.format(v) for v in vec)
            + ')'
            + ','.join('b[{}]'.format(j) for j in range(NBITS))
            + ';\n'
            )
    return out

def get_otp():
    otp0 = list( list(v) for v in np.loadtxt('./otp/otp0.csv',delimiter=','))
    otp1 = list( list(v) for v in np.loadtxt('./otp/otp1.csv',delimiter=','))
    return [otp0,otp1]

def main():
    pwd = input("Password: ")

    otp = get_otp()

    count=0
    keys = gen_keys(encode_string(pwd))
    states = get_states(gen_keys(encode_string(pwd)), otp)

    for kn,padding,state in zip(range(len(states)),my_secrets.padding, states):
        with open("converted_input.inc","w") as f:
            print(setup_state(initial_state(state + padding)))
            f.write(setup_state(initial_state(state + padding)))

        circ = qiskit.QuantumCircuit.from_qasm_file('chall.qasm')
        backend = qiskit.Aer.get_backend('harpa_secret_quantum_computer')

        target = []
        while len(target)==0:
            results = qiskit.execute(circ, backend, shots=30).results()
            counts = results.get_counts(circ)
            target = list(filter(lambda s: s[-1]=='1', counts.keys()))

        if len(target)==1 and int(target[0][-NBITS-2:-1], 2) == int(kn):
            count+=1

    if count==BLOCK_SIZE:
        print("Login successful!")

if __name__=='__main__':
    main()