phpseclib/tests/Crypt/AES/ContinuousBufferTest.php

<?php
/**
 * @author    Andreas Fischer <bantu@phpbb.com>
 * @copyright MMXIII Andreas Fischer
 * @license   http://www.opensource.org/licenses/mit-license.html  MIT License
 */

class Crypt_AES_ContinuousBufferTest extends Crypt_AES_TestCase
{
    // String intented
    protected $modes = array(
        'CRYPT_AES_MODE_CTR',
        'CRYPT_AES_MODE_OFB',
        'CRYPT_AES_MODE_CFB',
    );

    protected $plaintexts = array(
        '',
        '12345678901234567', // https://github.com/phpseclib/phpseclib/issues/39
        "\xDE\xAD\xBE\xAF",
        ':-):-):-):-):-):-)', // https://github.com/phpseclib/phpseclib/pull/43
    );

    protected $ivs = array(
        '',
        'test123',
    );

    protected $keys = array(
        '',
        ':-8', // https://github.com/phpseclib/phpseclib/pull/43
        'FOOBARZ',
    );

    /**
    * Produces all combinations of test values.
    *
    * @return array
    */
    public function allCombinations()
    {
        $result = array();

        // @codingStandardsIgnoreStart
        foreach ($this->modes as $mode)
        foreach ($this->plaintexts as $plaintext)
        foreach ($this->ivs as $iv)
        foreach ($this->keys as $key)
            $result[] = array($mode, $plaintext, $iv, $key);
        // @codingStandardsIgnoreEnd

        return $result;
    }

    /**
    * @dataProvider allCombinations
    */
    public function testEncryptDecrypt($mode, $plaintext, $iv, $key)
    {
        $aes = new Crypt_AES(constant($mode));
        $aes->enableContinuousBuffer();
        $aes->setIV($iv);
        $aes->setKey($key);

        $actual = '';
        for ($i = 0, $strlen = strlen($plaintext); $i < $strlen; ++$i) {
            $actual .= $aes->decrypt($aes->encrypt($plaintext[$i]));
        }

        $this->assertEquals($plaintext, $actual);
    }
}