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)
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()