#include <boost/test/unit_test.hpp>

#include <fc/crypto/blowfish.hpp>
#include <fc/crypto/hex.hpp>

// Test vectors from https://www.schneier.com/code/vectors.txt

struct ecb_testdata {
    char key[17], plain[17], cipher[17];
} ecb_tests[] = {
  { "0000000000000000", "0000000000000000", "4EF997456198DD78" },
  { "FFFFFFFFFFFFFFFF", "FFFFFFFFFFFFFFFF", "51866FD5B85ECB8A" },
  { "3000000000000000", "1000000000000001", "7D856F9A613063F2" },
  { "1111111111111111", "1111111111111111", "2466DD878B963C9D" },
  { "0123456789ABCDEF", "1111111111111111", "61F9C3802281B096" },
  { "1111111111111111", "0123456789ABCDEF", "7D0CC630AFDA1EC7" },
  { "0000000000000000", "0000000000000000", "4EF997456198DD78" },
  { "FEDCBA9876543210", "0123456789ABCDEF", "0ACEAB0FC6A0A28D" },
  { "7CA110454A1A6E57", "01A1D6D039776742", "59C68245EB05282B" },
  { "0131D9619DC1376E", "5CD54CA83DEF57DA", "B1B8CC0B250F09A0" },
  { "07A1133E4A0B2686", "0248D43806F67172", "1730E5778BEA1DA4" },
  { "3849674C2602319E", "51454B582DDF440A", "A25E7856CF2651EB" },
  { "04B915BA43FEB5B6", "42FD443059577FA2", "353882B109CE8F1A" },
  { "0113B970FD34F2CE", "059B5E0851CF143A", "48F4D0884C379918" },
  { "0170F175468FB5E6", "0756D8E0774761D2", "432193B78951FC98" },
  { "43297FAD38E373FE", "762514B829BF486A", "13F04154D69D1AE5" },
  { "07A7137045DA2A16", "3BDD119049372802", "2EEDDA93FFD39C79" },
  { "04689104C2FD3B2F", "26955F6835AF609A", "D887E0393C2DA6E3" },
  { "37D06BB516CB7546", "164D5E404F275232", "5F99D04F5B163969" },
  { "1F08260D1AC2465E", "6B056E18759F5CCA", "4A057A3B24D3977B" },
  { "584023641ABA6176", "004BD6EF09176062", "452031C1E4FADA8E" },
  { "025816164629B007", "480D39006EE762F2", "7555AE39F59B87BD" },
  { "49793EBC79B3258F", "437540C8698F3CFA", "53C55F9CB49FC019" },
  { "4FB05E1515AB73A7", "072D43A077075292", "7A8E7BFA937E89A3" },
  { "49E95D6D4CA229BF", "02FE55778117F12A", "CF9C5D7A4986ADB5" },
  { "018310DC409B26D6", "1D9D5C5018F728C2", "D1ABB290658BC778" },
  { "1C587F1C13924FEF", "305532286D6F295A", "55CB3774D13EF201" },
  { "0101010101010101", "0123456789ABCDEF", "FA34EC4847B268B2" },
  { "1F1F1F1F0E0E0E0E", "0123456789ABCDEF", "A790795108EA3CAE" },
  { "E0FEE0FEF1FEF1FE", "0123456789ABCDEF", "C39E072D9FAC631D" },
  { "0000000000000000", "FFFFFFFFFFFFFFFF", "014933E0CDAFF6E4" },
  { "FFFFFFFFFFFFFFFF", "0000000000000000", "F21E9A77B71C49BC" },
  { "0123456789ABCDEF", "0000000000000000", "245946885754369A" },
  { "FEDCBA9876543210", "FFFFFFFFFFFFFFFF", "6B5C5A9C5D9E0A5A" }
};

const std::string key_test_key = "F0E1D2C3B4A5968778695A4B3C2D1E0F0011223344556677";
const std::string key_test_plain = "FEDCBA9876543210";
const char key_test_ciphers[][17] = {
  "F9AD597C49DB005E",
  "E91D21C1D961A6D6",
  "E9C2B70A1BC65CF3",
  "BE1E639408640F05",
  "B39E44481BDB1E6E",
  "9457AA83B1928C0D",
  "8BB77032F960629D",
  "E87A244E2CC85E82",
  "15750E7A4F4EC577",
  "122BA70B3AB64AE0",
  "3A833C9AFFC537F6",
  "9409DA87A90F6BF2",
  "884F80625060B8B4",
  "1F85031C19E11968",
  "79D9373A714CA34F",
  "93142887EE3BE15C",
  "03429E838CE2D14B",
  "A4299E27469FF67B",
  "AFD5AED1C1BC96A8",
  "10851C0E3858DA9F",
  "E6F51ED79B9DB21F",
  "64A6E14AFD36B46F",
  "80C7D7D45A5479AD",
  "05044B62FA52D080",
};

const std::string chain_test_key = "0123456789ABCDEFF0E1D2C3B4A59687";
const std::string chain_test_iv = "FEDCBA9876543210";
const std::string chain_test_plain = "7654321 Now is the time for \0\0\0\0";
const std::string chain_test_cbc = "6B77B4D63006DEE605B156E27403979358DEB9E7154616D959F1652BD5FF92CC";
const std::string chain_test_cfb = "E73214A2822139CAF26ECF6D2EB9E76E3DA3DE04D1517200519D57A6C3";

BOOST_AUTO_TEST_SUITE(fc_crypto)

BOOST_AUTO_TEST_CASE(blowfish_ecb_test)
{
    for ( int i = 0; i < 34; i++ ) {
        unsigned char key[8], plain[8], cipher[8], out[8];
        BOOST_CHECK_EQUAL( 8, fc::from_hex( ecb_tests[i].key, (char*) key, sizeof(key) ) );
        BOOST_CHECK_EQUAL( 8, fc::from_hex( ecb_tests[i].plain, (char*) plain, sizeof(plain) ) );
        BOOST_CHECK_EQUAL( 8, fc::from_hex( ecb_tests[i].cipher, (char*) cipher, sizeof(cipher) ) );

        fc::blowfish fish;
        fish.start( key, 8 );
        fish.encrypt( plain, out, 8, fc::blowfish::ECB );
        BOOST_CHECK( !memcmp( cipher, out, 8) );
        fish.decrypt( out, 8, fc::blowfish::ECB );
        BOOST_CHECK( !memcmp( plain, out, 8) );
        fish.decrypt( cipher, out, 8, fc::blowfish::ECB );
        BOOST_CHECK( !memcmp( plain, out, 8) );
        fish.encrypt( out, 8, fc::blowfish::ECB );
        BOOST_CHECK( !memcmp( cipher, out, 8) );
    }
}

BOOST_AUTO_TEST_CASE(blowfish_key_test)
{
    unsigned char key[24], plain[8], cipher[8], out[8];
    BOOST_CHECK_EQUAL( 24, fc::from_hex( key_test_key.c_str(), (char*) key, sizeof(key) ) );
    BOOST_CHECK_EQUAL( 8, fc::from_hex( key_test_plain.c_str(), (char*) plain, sizeof(plain) ) );

    for ( unsigned int i = 0; i < sizeof(key); i++ ) {
        BOOST_CHECK_EQUAL( 8, fc::from_hex( key_test_ciphers[i], (char*) cipher, sizeof(cipher) ) );
        fc::blowfish fish;
        fish.start( key, i + 1 );
        fish.encrypt( plain, out, 8, fc::blowfish::ECB );
        BOOST_CHECK( !memcmp( cipher, out, 8) );
        fish.decrypt( out, 8, fc::blowfish::ECB );
        BOOST_CHECK( !memcmp( plain, out, 8) );
        fish.decrypt( cipher, out, 8, fc::blowfish::ECB );
        BOOST_CHECK( !memcmp( plain, out, 8) );
        fish.encrypt( out, 8, fc::blowfish::ECB );
        BOOST_CHECK( !memcmp( cipher, out, 8) );
    }
}

static unsigned int from_bytes( const unsigned char* p ) {
    return   (((unsigned int) p[0]) << 24)
           | (((unsigned int) p[1]) << 16)
           | (((unsigned int) p[2]) << 8)
           |  ((unsigned int) p[3]);
}

BOOST_AUTO_TEST_CASE(blowfish_chain_test)
{
    unsigned char key[16], iv[8], cipher[32], out[32];
    BOOST_CHECK_EQUAL( 16, fc::from_hex( chain_test_key.c_str(), (char*) key, sizeof(key) ) );
    BOOST_CHECK_EQUAL( 8, fc::from_hex( chain_test_iv.c_str(), (char*) iv, sizeof(iv) ) );

    BOOST_CHECK_EQUAL( 32, fc::from_hex( chain_test_cbc.c_str(), (char*) cipher, sizeof(cipher) ) );
    fc::blowfish fish;
    fish.start( key, sizeof(key), fc::sblock( from_bytes( iv ), from_bytes( iv + 4 ) ) );

    char buffer[32];
    memset(buffer, 0, sizeof(buffer));
    memcpy(buffer, chain_test_plain.c_str(), std::min(sizeof(buffer), chain_test_plain.size()));

    fish.encrypt( (unsigned char*) buffer, out, sizeof(out), fc::blowfish::CBC );
    BOOST_CHECK( !memcmp( cipher, out, sizeof(cipher) ) );
    fish.reset_chain();
    fish.decrypt( out, sizeof(out), fc::blowfish::CBC );
    BOOST_CHECK( !memcmp( buffer, out, 29 ) );
    fish.reset_chain();
    fish.encrypt( out, sizeof(out), fc::blowfish::CBC );
    BOOST_CHECK( !memcmp( cipher, out, sizeof(cipher) ) );
    fish.reset_chain();
    fish.decrypt( cipher, out, sizeof(cipher), fc::blowfish::CBC );
    BOOST_CHECK( !memcmp( buffer, out, 29 ) );

    BOOST_CHECK_EQUAL( 29, fc::from_hex( chain_test_cfb.c_str(), (char*) cipher, sizeof(cipher) ) );
    fish.reset_chain();
    fish.encrypt( (unsigned char*) buffer, out, sizeof(out), fc::blowfish::CFB );
    BOOST_CHECK( !memcmp( cipher, out, 29 ) );
    fish.reset_chain(); memset( out + 29, 0, 3 );
    fish.decrypt( out, sizeof(out), fc::blowfish::CFB );
    BOOST_CHECK( !memcmp( buffer, out, 29 ) );
    fish.reset_chain(); memset( out + 29, 0, 3 );
    fish.encrypt( out, sizeof(out), fc::blowfish::CFB );
    BOOST_CHECK( !memcmp( cipher, out, 29 ) );
    fish.reset_chain(); memset( out + 29, 0, 3 );
    fish.decrypt( cipher, out, sizeof(cipher), fc::blowfish::CFB );
    BOOST_CHECK( !memcmp( buffer, out, 29 ) );
}

BOOST_AUTO_TEST_SUITE_END()