import crypto.aes
import crypto.cipher
import crypto.des

struct StreamCipher {
    cipher cipher.Stream
}

fn test_ctr_stream_cipher() ! {
    key := '123456789012345678901234'.bytes()
    iv := 'abcdegfh'.bytes()

    block := des.new_cipher(key[..8])
    c := cipher.new_ctr(block, iv)

    s := StreamCipher{
        cipher: c
    }
}

fn test_ctr_byte_by_byte() {
    key := []u8{len: 16, init: index}
    iv := []u8{len: 16, init: index}
    txt := []u8{len: 32, init: index}
    mut out := []u8{len: 32}

    mut ofb := cipher.new_ctr(aes.new_cipher(key), iv)
    for i in 0 .. 32 {
        ofb.xor_key_stream(mut out[i..i + 1], txt[i..i + 1])
    }
    assert out == [u8(10), 149, 9, 182, 69, 107, 246, 66, 249, 202, 158, 83, 202, 94, 228, 85,
        18, 114, 254, 135, 114, 13, 100, 129, 130, 195, 231, 20, 87, 185, 17, 195]
}