Example #1
0
def test_digest(protocol_sample):
    s = create_read_packet(generate_data(64))

    rw = WritePacket()
    protocol_sample.generate_digest(s, rw)
    rr = create_read_packet(rw.get_data())
    protocol_sample.validate_digest(s, rr)
Example #2
0
def load_card_file(card_file_path, create=False):
    if not validation.validate_card_file(card_file_path, exists=not create):
        raise Exception('Invalid card file')
    if create:
        p = WritePacket()
        p.write_number(random.getrandbits(512), 64)
        card = p.get_data()
        with open(card_file_path, 'wb') as f:
            failure_hooks.append(lambda: os.remove(card_file_path))
            f.write(card)
        return card
    else:
        with open(card_file_path, 'rb') as f:
            return f.read()