1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
<?php
class Swift_Bug51Test extends \SwiftMailerTestCase
{
private $_attachmentFile;
private $_outputFile;
public function setUp()
{
if (!defined('SWIFT_TMP_DIR') || !is_writable(SWIFT_TMP_DIR)) {
$this->markTestSkipped(
'Cannot run test without a writable directory to use ('.
'define SWIFT_TMP_DIR in tests/config.php if you wish to run this test)'
);
}
$this->_attachmentFile = SWIFT_TMP_DIR.'/attach.rand.bin';
file_put_contents($this->_attachmentFile, '');
$this->_outputFile = SWIFT_TMP_DIR.'/attach.out.bin';
file_put_contents($this->_outputFile, '');
}
public 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
);
}
}
// -- Custom Assertions
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));
}
// -- Creation Methods
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;
}
}
|
