// vim: syntax=cpp
/*
 * Copyright (C) 2007-2012 Xagasoft, All rights reserved.
 *
 * This file is part of the libbu++ library and is released under the
 * terms of the license contained in the file LICENSE.
 */

#include <bu/blowfish.h>
#include <bu/file.h>
#include <bu/membuf.h>
#include <bu/hex.h>
#include <bu/strfilter.h>
#include <bu/sio.h>
#include <bu/ciphermodecfb.h>
#include <bu/ciphermodeecb.h>

using namespace Bu;

static const char *testdat[34][3] ={
{"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"}};


int main( int argc, char *argv[] )
{
    MemBuf mb;
    {
        BlowfishEcb bf( mb );
        bf.setPassword( "01234567" );
        for( int j = 0; j < 4; j++ )
        {
            bf.write("this is a test!!"+j, 1 );
        }
        bf.write("this is a test!!"+4, 12 );
    }
    mb.setPos( 0 );
    BlowfishEcb bf( mb );
    bf.setPassword( "01234567" );
    for( int j = 0; j < 3; j++ )
    {
        char c;
        bf.read( &c, 1 );
        sio << "char: '" << c << "'" << sio.nl;
    }

    char buf[100];
    int iR = bf.read( buf, 100 );
    buf[iR] = '\0';
    sio << "Got(" << iR << ") = '" << buf << "'" << sio.nl;

    return 0;

    for( int j = 0; j < 34; j++ )
    {
        MemBuf mb;
        BlowfishEcb bf( mb );
        bf.setPassword( decodeStr<Hex>( testdat[j][0] ) );
        bf.write( decodeStr<Hex>( testdat[j][1] ) );
        sio << "Test " << j << ": " << (mb.getString() == decodeStr<Hex>( testdat[j][2] )) << " (" << encodeStr<Hex>( mb.getString(), true ) << " == " << testdat[j][2] << ")" << sio.nl;

        mb.setPos( 0 );
        BlowfishEcb bf2( mb );
        bf2.setPassword( decodeStr<Hex>( testdat[j][0] ) );
        char buf[8];
        bf2.read( buf, 8 );
        
        sio << "  - Back: " << (Bu::String(testdat[j][1]) == encodeStr<Hex>(String(buf,8),true)) << sio.nl;
    }

    {
        File fIn("data.plain", File::Read );
        File fOut("data.crypt", File::WriteNew );

        BlowfishOfb bOut( fOut );
        bOut.setIv("01234567");
        bOut.setPassword("abcdefghijklmnop");
        bOut.write( fIn.readAll() );
    }
    {
        File fIn("data.java", File::Read );
        File fOut("data.stuff", File::WriteNew );

        BlowfishOfb bIn( fIn );
        bIn.setIv("01234567");
        bIn.setPassword("abcdefghijklmnop");
        char buf[64];
        bIn.read( buf, 64 );
        fOut.write( buf, 64 );
        sio << sio.nl << "All done." << sio.nl;
    }

    return 0;
}