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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
require "spec"
require "mime/multipart"
private def parse(delim, data, *, gsub = true)
data_io = IO::Memory.new(gsub ? data.gsub('\n', "\r\n") : data)
parser = MIME::Multipart::Parser.new(data_io, delim)
parsed = [] of {headers: HTTP::Headers, body: String}
while parser.has_next?
parser.next { |h, io| parsed << {headers: h, body: io.gets_to_end} }
end
parsed
end
describe MIME::Multipart::Parser do
it "parses basic multipart messages" do
data = parse "AaB03x", <<-MULTIPART
--AaB03x
Content-Disposition: form-data; name="submit-name"
Larry
--AaB03x
Content-Disposition: form-data; name="files"; filename="file1.txt"
Content-Type: text/plain
... contents of file1.txt ...
--AaB03x--
MULTIPART
data[0][:body].should eq("Larry")
data[0][:headers]["Content-Disposition"].should eq("form-data; name=\"submit-name\"")
data[1][:body].should eq("... contents of file1.txt ...")
data[1][:headers]["Content-Disposition"].should eq("form-data; name=\"files\"; filename=\"file1.txt\"")
data[1][:headers]["Content-Type"].should eq("text/plain")
end
it "parses messages with preambles and epilogues" do
data = parse "AaB03x", <<-MULTIPART
preamble
AaB03x
--AaB03x
Content-Disposition: form-data; name="foo"
foo
--AaB03x
Content-Disposition: form-data; name="bar"
bar
--AaB03x--
AaB03x
epilogue
MULTIPART
data[0][:body].should eq("foo")
data[0][:headers]["Content-Disposition"].should eq("form-data; name=\"foo\"")
data[1][:body].should eq("bar")
data[1][:headers]["Content-Disposition"].should eq("form-data; name=\"bar\"")
end
it "handles invalid multipart data" do
expect_raises(MIME::Multipart::Error, "EOF reading delimiter") do
parse "AaB03x", "--AaB03x", gsub: false
end
expect_raises(MIME::Multipart::Error, "EOF reading delimiter") do
parse "AaB03x", "--AaB03x\r\n\r\n--AaB03x", gsub: false
end
expect_raises(MIME::Multipart::Error, "EOF reading delimiter padding") do
parse "AaB03x", "--AaB03x ", gsub: false
end
expect_raises(MIME::Multipart::Error, "padding contained non-whitespace character") do
parse "AaB03x", "--AaB03x foo \r\n\r\n--AaB03x--", gsub: false
end
end
it "handles padding" do
data = parse "AaB03x", "--AaB03x \t\t \r\n\r\n--AaB03x--", gsub: false
data[0][:body].should eq("")
end
it "raises calling #next after finished" do
input = <<-MULTIPART
--AaB03x
Foo
--AaB03x--
MULTIPART
parser = MIME::Multipart::Parser.new(IO::Memory.new(input.gsub('\n', "\r\n")), "AaB03x")
parser.next { }
parser.has_next?.should eq(false)
expect_raises(MIME::Multipart::Error, "Multipart parser already finished parsing") do
parser.next { }
end
end
it "raises calling #next after errored" do
parser = MIME::Multipart::Parser.new(IO::Memory.new("--AaB03x--"), "AaB03x")
expect_raises(MIME::Multipart::Error, "no parts") do
parser.next { }
end
parser.has_next?.should eq(false)
expect_raises(MIME::Multipart::Error, "Multipart parser is in an errored state") do
parser.next { }
end
end
it "handles break/next in blocks" do
input = <<-MULTIPART
--b
--b
--b
--b
--b--
MULTIPART
parser = MIME::Multipart::Parser.new(IO::Memory.new(input.gsub('\n', "\r\n")), "b")
ios = [] of IO
2.times do
parser.next do |headers, io|
ios << io
next
end
end
2.times do
parser.next do |headers, io|
ios << io
break
end
end
parser.@state.finished?.should be_true
ios.each(&.closed?.should(eq(true)))
end
end
|
