v2 / vlib / crypto / cipher / aes_ofb_test.v
31 lines · 26 sloc · 931 bytes · fb192d949bd2e8f814c5826411ff61c92a96f796
Raw 


1import crypto.aes


2import crypto.cipher


3


4fn test_aes_ofb() {


5    key := '6368616e676520746869732070617373'.bytes()


6    iv := '1234567890123456'.bytes()


7    str := '73c86d43a9d700a253a96c85b0f6b03ac9792e0e757f869cca306bd3cba1c62b'


8


9    mut src := str.bytes()


10


11    aes_ofb_en(mut src, key, iv)


12    assert src.hex() == '04380c1470ab2d8a0b3f9b4c1949b8aca120db226fa0a848236f9904f786dfc39bc433b5d3870d1cfdef36debdee5e6f39b0e8e9462b0151d61e822944181b51'


13    dump(src.hex())


14    mut plaintext := src.clone()


15    aes_ofb_de(mut plaintext, key, iv)


16    assert plaintext.bytestr() == str


17    dump(plaintext.bytestr())


18    println('test_aes_ofb ok')


19}


20


21fn aes_ofb_en(mut src []u8, key []u8, iv []u8) {


22    block := aes.new_cipher(key)


23    mut mode := cipher.new_ofb(block, iv)


24    mode.xor_key_stream(mut src, src.clone())


25}


26


27fn aes_ofb_de(mut src []u8, key []u8, iv []u8) {


28    block := aes.new_cipher(key)


29    mut mode := cipher.new_ofb(block, iv)


30    mode.xor_key_stream(mut src, src.clone())


31}


32