- for _ in range(1 << 8):
- kek = urandom(32)
- cek = urandom(32)
- ukm = urandom(8)
- wrapped = wrap_cryptopro(ukm, kek, cek)
- unwrapped = unwrap_cryptopro(kek, wrapped)
- self.assertSequenceEqual(unwrapped, cek)
+ for sbox in (DEFAULT_SBOX, "id-tc26-gost-28147-param-Z"):
+ for _ in range(1 << 8):
+ kek = urandom(32)
+ cek = urandom(32)
+ ukm = urandom(8)
+ wrapped = wrap_cryptopro(ukm, kek, cek, sbox=sbox)
+ unwrapped = unwrap_cryptopro(kek, wrapped, sbox=sbox)
+ self.assertSequenceEqual(unwrapped, cek)