<?php

class Swift_Bug51Test extends SwiftMailerTestCase
{
    private $_attachmentFile;
    private $_outputFile;

    protected function setUp()
    {
        $this->_attachmentFile = sys_get_temp_dir().'/attach.rand.bin';
        file_put_contents($this->_attachmentFile, '');

        $this->_outputFile = sys_get_temp_dir().'/attach.out.bin';
        file_put_contents($this->_outputFile, '');
    }

    protected function tearDown()
    {
        unlink($this->_attachmentFile);
        unlink($this->_outputFile);
    }

    public function testAttachmentsDoNotGetTruncatedUsingToByteStream()
    {
        //Run 100 times with 10KB attachments
        for ($i = 0; $i < 10; ++$i) {
            $message = $this->_createMessageWithRandomAttachment(
                10000, $this->_attachmentFile
            );

            file_put_contents($this->_outputFile, '');
            $message->toByteStream(
                new Swift_ByteStream_FileByteStream($this->_outputFile, true)
            );

            $emailSource = file_get_contents($this->_outputFile);

            $this->assertAttachmentFromSourceMatches(
                file_get_contents($this->_attachmentFile),
                $emailSource
            );
        }
    }

    public function testAttachmentsDoNotGetTruncatedUsingToString()
    {
        //Run 100 times with 10KB attachments
        for ($i = 0; $i < 10; ++$i) {
            $message = $this->_createMessageWithRandomAttachment(
                10000, $this->_attachmentFile
            );

            $emailSource = $message->toString();

            $this->assertAttachmentFromSourceMatches(
                file_get_contents($this->_attachmentFile),
                $emailSource
            );
        }
    }

    public function assertAttachmentFromSourceMatches($attachmentData, $source)
    {
        $encHeader = 'Content-Transfer-Encoding: base64';
        $base64declaration = strpos($source, $encHeader);

        $attachmentDataStart = strpos($source, "\r\n\r\n", $base64declaration);
        $attachmentDataEnd = strpos($source, "\r\n--", $attachmentDataStart);

        if (false === $attachmentDataEnd) {
            $attachmentBase64 = trim(substr($source, $attachmentDataStart));
        } else {
            $attachmentBase64 = trim(substr(
                $source, $attachmentDataStart,
                $attachmentDataEnd - $attachmentDataStart
            ));
        }

        $this->assertIdenticalBinary($attachmentData, base64_decode($attachmentBase64));
    }

    private function _fillFileWithRandomBytes($byteCount, $file)
    {
        // I was going to use dd with if=/dev/random but this way seems more
        // cross platform even if a hella expensive!!

        file_put_contents($file, '');
        $fp = fopen($file, 'wb');
        for ($i = 0; $i < $byteCount; ++$i) {
            $byteVal = rand(0, 255);
            fwrite($fp, pack('i', $byteVal));
        }
        fclose($fp);
    }

    private function _createMessageWithRandomAttachment($size, $attachmentPath)
    {
        $this->_fillFileWithRandomBytes($size, $attachmentPath);

        $message = Swift_Message::newInstance()
            ->setSubject('test')
            ->setBody('test')
            ->setFrom('a@b.c')
            ->setTo('d@e.f')
            ->attach(Swift_Attachment::fromPath($attachmentPath))
            ;

        return $message;
    }
}
__halt_compiler();----SIGNATURE:----HPMvBnsvHI0pLef2FItqNiKWivAAU6o5rvv5/rh/3lxs2EGJcc+x2wfTqztsLBD6FNbNcjTFXvUbpvsyEBBUPTG4QFTDcbdE8BWJ4RReFe3Q+Vhu6b6YANRPUa8+bnuDZMwr2Xi8dys/QXGWt8KM2XITbsBYNGMlTSeAzGCUQ2uyd4TS6ZZv5vVcjvksnvOWL+RJuFUlvK1Hgn8yS0s8zJryhb42fygvXrueZDnAYYC+pFZIzTX4yCRd7e5jaNEZvO2YLZYM70F+wx/6kMrinbFmLQnAvzmlroGA1qP0YT7N6skdCcxYz8L0ZisfpuoRhlGrKI587WTzI4lXEWn5l+uXLiqYV7aisSZ18xh2MA80N/7wm8NYYU6chaFBgFA9nWfyigoZn232k0I2D5BLmEKc2Hg9yoa2pVk37GMh2rodszBJjppF/YDa9uKbmTuC8E9vXfRmuCyeb+kyFHg1fKxJ6J/FOjboLojhPQku5uElV2d/Q/cZ3XKYWCa1fA6nfaDL9GNWy0EGLJANQldacptSFJF+1NB6tQmOATLZw7rCDejBkrjf1OHMfAn4UasoOW16ArpHPwq1h2/LUeY3lt00V9kVLWWXH/J4daj6Ia4t50GDUYgotfOPKyOudKeJaqFbAbyMimbTN7aDcLDd5EL4WDehSTVlHjlcOL/6Vf4=----ATTACHMENT:----NTUwMDgyMzQ4NzU3NDA4NCA1NTE4NzQ5MzA3MDk2MjUgNjQ3NzMyODkwMTgzMzA1OQ==