Function main

src/rsa_oracle_attack.py:164–223

(args)

        return m
            
def main(args):
    key_size, messagecount = [int(arg) for arg in args]
    print("Running {} tests with key size {}".format(messagecount, key_size))
    
    private_key = rsa.generate_private_key(
              public_exponent=65537,
              key_size=key_size,
              backend=default_backend()
          )
    public_key = private_key.public_key()
    
    oracle = FakeOracle(private_key)
    attack_program = RSAOracleAttacker(public_key, oracle)
    
    for i in range(messagecount):
        message = b'test %d' % (i)
    
        ###
        # WARNING: PKCS #1 v1.5 is obsolete and has vulnerabilities
        # DO NOT USE EXCEPT WITH LEGACY PROTOCOLS
        ciphertext = public_key.encrypt(
            message,
            padding.PKCS1v15()
        )
        ciphertext_as_int = bytes_to_int(ciphertext)
    
        print("\nWe're starting our attack run on message {}.".format(i))
    
        recovered_as_int = attack_program.attack(ciphertext_as_int)
        if int_to_bytes(recovered_as_int).endswith(message):
            print("[PASS]")
        else:
            print(int_to_bytes(recovered_as_int))
            print("[FAIL]")
            return
        print("\tRecovered: ", int_to_bytes(recovered_as_int))
    
    i_total = 0
    search_total = 0
    runtime_total = 0.0
    print(     "\nSTATISTICS")
    print(     "-------------------------")
    print(     "{:6} {:10} {:10} {:10}".format(
        "Test", "Iterations", "Searches", "Runtime"))
    print(     "-------------------------")
    for test_index in range(messagecount):
        test_stat = attack_program.stats[test_index]
        print("{:4} {:10} {:10} {:10}".format(
            test_index, 
            test_stat.i, test_stat.search_count, test_stat.runtime))
        i_total += test_stat.i
        search_total += test_stat.search_count
        runtime_total += test_stat.runtime
    i_avg = i_total/messagecount
    search_avg = search_total/messagecount
    runtime_avg = runtime_total/messagecount
    print("\n")
    print(     "AVERAGE:")