Dogcows Code - chaz/p5-File-KDBX/blob - t/crypt.t

   1 #!/usr/bin/env perl
   2
   3 use warnings;
   4 use strict;
   5
   6 use lib 't/lib';
   7 use TestCommon;
   8
   9 use Crypt::Misc 0.029 qw(decode_b64 encode_b64);
  10 use File::KDBX::Cipher;
  11 use File::KDBX::Constants qw(CIPHER_UUID_AES256);
  12 use File::KDBX::IO::Crypt;
  13 use IO::Handle;
  14 use Test::More;
  15
  16 subtest 'Round-trip block stream' => sub {
  17     plan tests => 3;
  18     my $block_cipher = File::KDBX::Cipher->new(uuid => CIPHER_UUID_AES256, key => 0x01 x 32, iv => 0x01 x 16);
  19     test_roundtrip($block_cipher,
  20         'Smell the pretty flowers.',
  21         decode_b64('pB10mV+mhTuh7bKg0KEUl5H1ajFMaP4uPnTZNcDgq6s='),
  22     );
  23 };
  24
  25 subtest 'Round-trip cipher stream' => sub {
  26     plan tests => 3;
  27     my $cipher_stream = File::KDBX::Cipher->new(stream_id => 2, key => 0x01 x 16);
  28     test_roundtrip($cipher_stream,
  29         'Smell the pretty flowers.',
  30         decode_b64('gNj2Ud9tWtFDy+xDN/U01RxmCoI6MAlTKQ=='),
  31     );
  32 };
  33
  34 subtest 'Error handling' => sub {
  35     plan tests => 4;
  36
  37     my $block_cipher = File::KDBX::Cipher->new(uuid => CIPHER_UUID_AES256, key => 0x01 x 32, iv => 0x01 x 16);
  38     pipe(my $read, my $write) or die "pipe failed: $!";
  39     $read = File::KDBX::IO::Crypt->new($read, cipher => $block_cipher);
  40
  41     print $write "blah blah blah!\1";
  42     close($write) or die "close failed: $!";
  43
  44     is $read->error, '', 'Read handle starts out fine';
  45     my $plaintext = do { local $/; <$read> };
  46     is $plaintext, '', 'Read can fail';
  47     is $read->error, 1, 'Read handle can enter an error state';
  48
  49     like $File::KDBX::IO::Crypt::ERROR, qr/fatal/i,
  50         'Error object is available';
  51 };
  52
  53 done_testing;
  54 exit;
  55
  56 sub test_roundtrip {
  57     my $cipher = shift;
  58     my $expected_plaintext = shift;
  59     my $expected_ciphertext = shift;
  60
  61     pipe(my $read, my $write) or die "pipe failed: $!";
  62     $write = File::KDBX::IO::Crypt->new($write, cipher => $cipher);
  63
  64     print $write $expected_plaintext;
  65     close($write) or die "close failed: $!";
  66
  67     my $ciphertext = do { local $/; <$read> };
  68     close($read);
  69     is $ciphertext, $expected_ciphertext, 'Encrypted a string'
  70         or diag encode_b64($ciphertext);
  71
  72     my $ciphertext2 = $cipher->encrypt_finish($expected_plaintext);
  73     is $ciphertext, $ciphertext2, 'Same result';
  74
  75     open(my $fh, '<', \$ciphertext) or die "open failed: $!\n";
  76     $fh = File::KDBX::IO::Crypt->new($fh, cipher => $cipher);
  77
  78     my $plaintext = do { local $/; <$fh> };
  79     close($fh);
  80     is $plaintext, $expected_plaintext, 'Decrypted a string'
  81         or diag encode_b64($plaintext);
  82 }