<?php
/**
 * @author    Jim Wigginton <terrafrost@php.net>
 * @copyright 2014 Jim Wigginton
 * @license   http://www.opensource.org/licenses/mit-license.html  MIT License
 */

use phpseclib\Crypt\Base;
use phpseclib\Crypt\RC4;

class Unit_Crypt_RC4Test extends PhpseclibTestCase
{
    public function engineVectors()
    {
        $engines = array(
            Base::ENGINE_INTERNAL => 'internal',
            Base::ENGINE_MCRYPT => 'mcrypt',
            Base::ENGINE_OPENSSL => 'OpenSSL',
        );
        // tests from https://tools.ietf.org/html/rfc6229
        $tests = array(
            array(
                'key' => pack('H*', '0102030405'), // 40-bit key
                'output' => array(
                    array('offset' =>    0, 'result' => 'b2396305f03dc027ccc3524a0a1118a8'),
                    array('offset' =>   16, 'result' => '6982944f18fc82d589c403a47a0d0919'),
                    array('offset' =>  240, 'result' => '28cb1132c96ce286421dcaadb8b69eae'),
                    array('offset' =>  256, 'result' => '1cfcf62b03eddb641d77dfcf7f8d8c93'),
                    array('offset' =>  496, 'result' => '42b7d0cdd918a8a33dd51781c81f4041'),
                    array('offset' =>  512, 'result' => '6459844432a7da923cfb3eb4980661f6'),
                    array('offset' =>  752, 'result' => 'ec10327bde2beefd18f9277680457e22'),
                    array('offset' =>  768, 'result' => 'eb62638d4f0ba1fe9fca20e05bf8ff2b'),
                    array('offset' => 1008, 'result' => '45129048e6a0ed0b56b490338f078da5'),
                    array('offset' => 1024, 'result' => '30abbcc7c20b01609f23ee2d5f6bb7df'),
                    array('offset' => 1520, 'result' => '3294f744d8f9790507e70f62e5bbceea'),
                    array('offset' => 1536, 'result' => 'd8729db41882259bee4f825325f5a130'),
                    array('offset' => 2032, 'result' => '1eb14a0c13b3bf47fa2a0ba93ad45b8b'),
                    array('offset' => 2048, 'result' => 'cc582f8ba9f265e2b1be9112e975d2d7'),
                    array('offset' => 3056, 'result' => 'f2e30f9bd102ecbf75aaade9bc35c43c'),
                    array('offset' => 3072, 'result' => 'ec0e11c479dc329dc8da7968fe965681'),
                    array('offset' => 4080, 'result' => '068326a2118416d21f9d04b2cd1ca050'),
                    array('offset' => 4096, 'result' => 'ff25b58995996707e51fbdf08b34d875')
                )
            ),
            array(
                'key' => pack('H*', '01020304050607'), // 56-bit key
                'output' => array(
                    array('offset' =>    0, 'result' => '293f02d47f37c9b633f2af5285feb46b'),
                    array('offset' =>   16, 'result' => 'e620f1390d19bd84e2e0fd752031afc1'),
                    array('offset' =>  240, 'result' => '914f02531c9218810df60f67e338154c'),
                    array('offset' =>  256, 'result' => 'd0fdb583073ce85ab83917740ec011d5'),
                    array('offset' =>  496, 'result' => '75f81411e871cffa70b90c74c592e454'),
                    array('offset' =>  512, 'result' => '0bb87202938dad609e87a5a1b079e5e4'),
                    array('offset' =>  752, 'result' => 'c2911246b612e7e7b903dfeda1dad866'),
                    array('offset' =>  768, 'result' => '32828f91502b6291368de8081de36fc2'),
                    array('offset' => 1008, 'result' => 'f3b9a7e3b297bf9ad804512f9063eff1'),
                    array('offset' => 1024, 'result' => '8ecb67a9ba1f55a5a067e2b026a3676f'),
                    array('offset' => 1520, 'result' => 'd2aa902bd42d0d7cfd340cd45810529f'),
                    array('offset' => 1536, 'result' => '78b272c96e42eab4c60bd914e39d06e3'),
                    array('offset' => 2032, 'result' => 'f4332fd31a079396ee3cee3f2a4ff049'),
                    array('offset' => 2048, 'result' => '05459781d41fda7f30c1be7e1246c623'),
                    array('offset' => 3056, 'result' => 'adfd3868b8e51485d5e610017e3dd609'),
                    array('offset' => 3072, 'result' => 'ad26581c0c5be45f4cea01db2f3805d5'),
                    array('offset' => 4080, 'result' => 'f3172ceffc3b3d997c85ccd5af1a950c'),
                    array('offset' => 4096, 'result' => 'e74b0b9731227fd37c0ec08a47ddd8b8')
                )
            ),
            array(
                'key' => pack('H*', '0102030405060708'), // 64-bit key
                'output' => array(
                    array('offset' =>    0, 'result' => '97ab8a1bf0afb96132f2f67258da15a8'),
                    array('offset' =>   16, 'result' => '8263efdb45c4a18684ef87e6b19e5b09'),
                    array('offset' =>  240, 'result' => '9636ebc9841926f4f7d1f362bddf6e18'),
                    array('offset' =>  256, 'result' => 'd0a990ff2c05fef5b90373c9ff4b870a'),
                    array('offset' =>  496, 'result' => '73239f1db7f41d80b643c0c52518ec63'),
                    array('offset' =>  512, 'result' => '163b319923a6bdb4527c626126703c0f'),
                    array('offset' =>  752, 'result' => '49d6c8af0f97144a87df21d91472f966'),
                    array('offset' =>  768, 'result' => '44173a103b6616c5d5ad1cee40c863d0'),
                    array('offset' => 1008, 'result' => '273c9c4b27f322e4e716ef53a47de7a4'),
                    array('offset' => 1024, 'result' => 'c6d0e7b226259fa9023490b26167ad1d'),
                    array('offset' => 1520, 'result' => '1fe8986713f07c3d9ae1c163ff8cf9d3'),
                    array('offset' => 1536, 'result' => '8369e1a965610be887fbd0c79162aafb'),
                    array('offset' => 2032, 'result' => '0a0127abb44484b9fbef5abcae1b579f'),
                    array('offset' => 2048, 'result' => 'c2cdadc6402e8ee866e1f37bdb47e42c'),
                    array('offset' => 3056, 'result' => '26b51ea37df8e1d6f76fc3b66a7429b3'),
                    array('offset' => 3072, 'result' => 'bc7683205d4f443dc1f29dda3315c87b'),
                    array('offset' => 4080, 'result' => 'd5fa5a3469d29aaaf83d23589db8c85b'),
                    array('offset' => 4096, 'result' => '3fb46e2c8f0f068edce8cdcd7dfc5862')
                )
            ),
            array(
                'key' => pack('H*', '0102030405060708090a'), // 80-bit key
                'output' => array(
                    array('offset' =>    0, 'result' => 'ede3b04643e586cc907dc21851709902'),
                    array('offset' =>   16, 'result' => '03516ba78f413beb223aa5d4d2df6711'),
                    array('offset' =>  240, 'result' => '3cfd6cb58ee0fdde640176ad0000044d'),
                    array('offset' =>  256, 'result' => '48532b21fb6079c9114c0ffd9c04a1ad'),
                    array('offset' =>  496, 'result' => '3e8cea98017109979084b1ef92f99d86'),
                    array('offset' =>  512, 'result' => 'e20fb49bdb337ee48b8d8dc0f4afeffe'),
                    array('offset' =>  752, 'result' => '5c2521eacd7966f15e056544bea0d315'),
                    array('offset' =>  768, 'result' => 'e067a7031931a246a6c3875d2f678acb'),
                    array('offset' => 1008, 'result' => 'a64f70af88ae56b6f87581c0e23e6b08'),
                    array('offset' => 1024, 'result' => 'f449031de312814ec6f319291f4a0516'),
                    array('offset' => 1520, 'result' => 'bdae85924b3cb1d0a2e33a30c6d79599'),
                    array('offset' => 1536, 'result' => '8a0feddbac865a09bcd127fb562ed60a'),
                    array('offset' => 2032, 'result' => 'b55a0a5b51a12a8be34899c3e047511a'),
                    array('offset' => 2048, 'result' => 'd9a09cea3ce75fe39698070317a71339'),
                    array('offset' => 3056, 'result' => '552225ed1177f44584ac8cfa6c4eb5fc'),
                    array('offset' => 3072, 'result' => '7e82cbabfc95381b080998442129c2f8'),
                    array('offset' => 4080, 'result' => '1f135ed14ce60a91369d2322bef25e3c'),
                    array('offset' => 4096, 'result' => '08b6be45124a43e2eb77953f84dc8553')
                )
            ),
            array(
                'key' => pack('H*', '0102030405060708090a0b0c0d0e0f10'), // 128-bit key
                'output' => array(
                    array('offset' =>    0, 'result' => '9ac7cc9a609d1ef7b2932899cde41b97'),
                    array('offset' =>   16, 'result' => '5248c4959014126a6e8a84f11d1a9e1c'),
                    array('offset' =>  240, 'result' => '065902e4b620f6cc36c8589f66432f2b'),
                    array('offset' =>  256, 'result' => 'd39d566bc6bce3010768151549f3873f'),
                    array('offset' =>  496, 'result' => 'b6d1e6c4a5e4771cad79538df295fb11'),
                    array('offset' =>  512, 'result' => 'c68c1d5c559a974123df1dbc52a43b89'),
                    array('offset' =>  752, 'result' => 'c5ecf88de897fd57fed301701b82a259'),
                    array('offset' =>  768, 'result' => 'eccbe13de1fcc91c11a0b26c0bc8fa4d'),
                    array('offset' => 1008, 'result' => 'e7a72574f8782ae26aabcf9ebcd66065'),
                    array('offset' => 1024, 'result' => 'bdf0324e6083dcc6d3cedd3ca8c53c16'),
                    array('offset' => 1520, 'result' => 'b40110c4190b5622a96116b0017ed297'),
                    array('offset' => 1536, 'result' => 'ffa0b514647ec04f6306b892ae661181'),
                    array('offset' => 2032, 'result' => 'd03d1bc03cd33d70dff9fa5d71963ebd'),
                    array('offset' => 2048, 'result' => '8a44126411eaa78bd51e8d87a8879bf5'),
                    array('offset' => 3056, 'result' => 'fabeb76028ade2d0e48722e46c4615a3'),
                    array('offset' => 3072, 'result' => 'c05d88abd50357f935a63c59ee537623'),
                    array('offset' => 4080, 'result' => 'ff38265c1642c1abe8d3c2fe5e572bf8'),
                    array('offset' => 4096, 'result' => 'a36a4c301ae8ac13610ccbc12256cacc')
                )
            ),
            array(
                'key' => pack('H*', '0102030405060708090a0b0c0d0e0f101112131415161718'), // 192-bit key
                'output' => array(
                    array('offset' =>    0, 'result' => '0595e57fe5f0bb3c706edac8a4b2db11'),
                    array('offset' =>   16, 'result' => 'dfde31344a1af769c74f070aee9e2326'),
                    array('offset' =>  240, 'result' => 'b06b9b1e195d13d8f4a7995c4553ac05'),
                    array('offset' =>  256, 'result' => '6bd2378ec341c9a42f37ba79f88a32ff'),
                    array('offset' =>  496, 'result' => 'e70bce1df7645adb5d2c4130215c3522'),
                    array('offset' =>  512, 'result' => '9a5730c7fcb4c9af51ffda89c7f1ad22'),
                    array('offset' =>  752, 'result' => '0485055fd4f6f0d963ef5ab9a5476982'),
                    array('offset' =>  768, 'result' => '591fc66bcda10e452b03d4551f6b62ac'),
                    array('offset' => 1008, 'result' => '2753cc83988afa3e1688a1d3b42c9a02'),
                    array('offset' => 1024, 'result' => '93610d523d1d3f0062b3c2a3bbc7c7f0'),
                    array('offset' => 1520, 'result' => '96c248610aadedfeaf8978c03de8205a'),
                    array('offset' => 1536, 'result' => '0e317b3d1c73b9e9a4688f296d133a19'),
                    array('offset' => 2032, 'result' => 'bdf0e6c3cca5b5b9d533b69c56ada120'),
                    array('offset' => 2048, 'result' => '88a218b6e2ece1e6246d44c759d19b10'),
                    array('offset' => 3056, 'result' => '6866397e95c140534f94263421006e40'),
                    array('offset' => 3072, 'result' => '32cb0a1e9542c6b3b8b398abc3b0f1d5'),
                    array('offset' => 4080, 'result' => '29a0b8aed54a132324c62e423f54b4c8'),
                    array('offset' => 4096, 'result' => '3cb0f3b5020a98b82af9fe154484a168')
                )
            ),
            array(
                'key' => pack('H*', '0102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f20'), // 256-bit key
                'output' => array(
                    array('offset' =>    0, 'result' => 'eaa6bd25880bf93d3f5d1e4ca2611d91'),
                    array('offset' =>   16, 'result' => 'cfa45c9f7e714b54bdfa80027cb14380'),
                    array('offset' =>  240, 'result' => '114ae344ded71b35f2e60febad727fd8'),
                    array('offset' =>  256, 'result' => '02e1e7056b0f623900496422943e97b6'),
                    array('offset' =>  496, 'result' => '91cb93c787964e10d9527d999c6f936b'),
                    array('offset' =>  512, 'result' => '49b18b42f8e8367cbeb5ef104ba1c7cd'),
                    array('offset' =>  752, 'result' => '87084b3ba700bade955610672745b374'),
                    array('offset' =>  768, 'result' => 'e7a7b9e9ec540d5ff43bdb12792d1b35'),
                    array('offset' => 1008, 'result' => 'c799b596738f6b018c76c74b1759bd90'),
                    array('offset' => 1024, 'result' => '7fec5bfd9f9b89ce6548309092d7e958'),
                    array('offset' => 1520, 'result' => '40f250b26d1f096a4afd4c340a588815'),
                    array('offset' => 1536, 'result' => '3e34135c79db010200767651cf263073'),
                    array('offset' => 2032, 'result' => 'f656abccf88dd827027b2ce917d464ec'),
                    array('offset' => 2048, 'result' => '18b62503bfbc077fbabb98f20d98ab34'),
                    array('offset' => 3056, 'result' => '8aed95ee5b0dcbfbef4eb21d3a3f52f9'),
                    array('offset' => 3072, 'result' => '625a1ab00ee39a5327346bddb01a9c18'),
                    array('offset' => 4080, 'result' => 'a13a7c79c7e119b5ab0296ab28c300b9'),
                    array('offset' => 4096, 'result' => 'f3e4c0a2e02d1d01f7f0a74618af2b48')
                )
            )
        );

        $result = array();

        foreach ($engines as $engine => $engineName) {
            foreach ($tests as $test) {
                foreach ($test['output'] as $output) {
                    $result[] = array($engine, $engineName, $test['key'], $output['offset'], $output['result']);
                }
            }
        }

        return $result;
    }

    /**
     * @dataProvider engineVectors
     */
    public function testVectors($engine, $engineName, $key, $offset, $expected)
    {
        $rc4 = new RC4();
        $rc4->setPreferredEngine($engine);
        $rc4->setKey($key);
        if ($rc4->getEngine() != $engine) {
            self::markTestSkipped('Unable to initialize ' . $engineName . ' engine for ' . (strlen($key) * 8) . '-bit key');
        }
        $result = $rc4->encrypt(str_repeat("\0", $offset + 16));
        $this->assertEquals(bin2hex(substr($result, -16)), $expected, "Failed asserting that key $key yielded expected output at offset $offset in $engineName engine");
    }
}