Я бы не стал пытаться изменить код самостоятельно, если вы мало знаете о криптографии. Шансы получить что-то не так довольно высоки.
Если вы можете изменить используемую реализацию AES, почему бы не использовать предоставляемый PHP модуль mcypt (http://php.net/manual/en/ref.mcrypt.php)? Он поддерживает AES-128, -192, -256-bit (под названием MCRYPT_RIJNDAEL_xxx) и многое, многое другое. Кроме того, в качестве бесплатного бонуса вы можете найти множество примеров кода для него.
Вот пример: http://www.php.net/manual/en/function.mcrypt-encrypt.php#78531.