1 | PACK_IS_LITTLE_ENDIAN = "\x01\00".unpack('S')[0] == 0x01
|
---|
2 |
|
---|
3 | def assert_pack tmpl, packed, unpacked
|
---|
4 | t = tmpl.inspect
|
---|
5 | assert "assert_pack" do
|
---|
6 | assert_equal packed, unpacked.pack(tmpl), "#{unpacked.inspect}.pack(#{t})"
|
---|
7 | assert_equal unpacked, packed.unpack(tmpl), "#{packed.inspect}.unpack(#{t})"
|
---|
8 | end
|
---|
9 | end
|
---|
10 |
|
---|
11 | # pack & unpack 'm' (base64)
|
---|
12 | assert('[""].pack("m")') do
|
---|
13 | assert_pack "m", "", [""]
|
---|
14 | end
|
---|
15 |
|
---|
16 | assert('["\0"].pack("m")') do
|
---|
17 | assert_pack "m", "AA==\n", ["\0"]
|
---|
18 | end
|
---|
19 |
|
---|
20 | assert('["\0\0"].pack("m")') do
|
---|
21 | assert_pack "m", "AAA=\n", ["\0\0"]
|
---|
22 | end
|
---|
23 |
|
---|
24 | assert('["\0\0\0"].pack("m")') do
|
---|
25 | assert_pack "m", "AAAA\n", ["\0\0\0"]
|
---|
26 | end
|
---|
27 |
|
---|
28 | assert('["abc..xyzABC..XYZ"].pack("m")') do
|
---|
29 | assert_pack "m", "YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNERUZHSElKS0xNTk9QUVJT\nVFVWV1hZWg==\n", ["abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"]
|
---|
30 | end
|
---|
31 |
|
---|
32 | assert('"YWJ...".unpack("m") should "abc..xyzABC..XYZ"') do
|
---|
33 | ary = ["abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"]
|
---|
34 | assert_equal ary, "YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNERUZHSElKS0xNTk9QUVJT\nVFVWV1hZWg==\n".unpack("m")
|
---|
35 | assert_equal ary, "YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNERUZHSElKS0xNTk9QUVJTVFVWV1hZWg==\n".unpack("m")
|
---|
36 | end
|
---|
37 |
|
---|
38 | assert('["A", "B"].pack') do
|
---|
39 | assert_equal "QQ==\n", ["A", "B"].pack("m50")
|
---|
40 | assert_equal ["A"], "QQ==\n".unpack("m50")
|
---|
41 | assert_equal "QQ==Qg==", ["A", "B"].pack("m0 m0")
|
---|
42 | assert_equal ["A", "B"], "QQ==Qg==".unpack("m10 m10")
|
---|
43 | end
|
---|
44 |
|
---|
45 | assert('["abc..xyzABC..XYZ"].pack("m0")') do
|
---|
46 | assert_pack "m0", "YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNERUZHSElKS0xNTk9QUVJTVFVWV1hZWg==", ["abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"]
|
---|
47 | end
|
---|
48 |
|
---|
49 | # pack & unpack 'H'
|
---|
50 | assert('["3031"].pack("H*")') do
|
---|
51 | assert_pack "H*", "01", ["3031"]
|
---|
52 | end
|
---|
53 |
|
---|
54 | assert('["10"].pack("H*")') do
|
---|
55 | assert_pack "H*", "\020", ["10"]
|
---|
56 | end
|
---|
57 |
|
---|
58 | assert('[0,1,127,128,255].pack("C*")') do
|
---|
59 | assert_pack "C*", "\x00\x01\x7F\x80\xFF", [0, 1, 127, 128, 255]
|
---|
60 | end
|
---|
61 |
|
---|
62 | # pack "a"
|
---|
63 | assert('["abc"].pack("a")') do
|
---|
64 | assert_equal "a", ["abc"].pack("a")
|
---|
65 | assert_equal "abc", ["abc"].pack("a*")
|
---|
66 | assert_equal "abc\0", ["abc"].pack("a4")
|
---|
67 | end
|
---|
68 |
|
---|
69 | # upack "a"
|
---|
70 | assert('["abc"].pack("a")') do
|
---|
71 | assert_equal ["abc\0"], "abc\0".unpack("a4")
|
---|
72 | assert_equal ["abc "], "abc ".unpack("a4")
|
---|
73 | end
|
---|
74 |
|
---|
75 | # pack "A"
|
---|
76 | assert('["abc"].pack("A")') do
|
---|
77 | assert_equal "a", ["abc"].pack("A")
|
---|
78 | assert_equal "abc", ["abc"].pack("A*")
|
---|
79 | assert_equal "abc ", ["abc"].pack("A4")
|
---|
80 | end
|
---|
81 |
|
---|
82 | # upack "A"
|
---|
83 | assert('["abc"].pack("A")') do
|
---|
84 | assert_equal ["abc"], "abc\0".unpack("A4")
|
---|
85 | assert_equal ["abc"], "abc ".unpack("A4")
|
---|
86 | end
|
---|
87 |
|
---|
88 | # regression tests
|
---|
89 | assert('issue #1') do
|
---|
90 | assert_equal "\000\001\000\002", [1, 2].pack("nn")
|
---|
91 | end
|
---|
92 |
|
---|
93 | assert 'pack float' do
|
---|
94 | assert_pack 'e', "\x00\x00@@", [3.0]
|
---|
95 | assert_pack 'g', "@@\x00\x00", [3.0]
|
---|
96 |
|
---|
97 | if PACK_IS_LITTLE_ENDIAN
|
---|
98 | assert_pack 'f', "\x00\x00@@", [3.0]
|
---|
99 | assert_pack 'F', "\x00\x00@@", [3.0]
|
---|
100 | else
|
---|
101 | assert_pack 'f', "@@\x00\x00", [3.0]
|
---|
102 | assert_pack 'F', "@@\x00\x00", [3.0]
|
---|
103 | end
|
---|
104 | end
|
---|
105 |
|
---|
106 | assert 'pack double' do
|
---|
107 | assert_pack 'E', "\x00\x00\x00\x00\x00\x00\b@", [3.0]
|
---|
108 | assert_pack 'G', "@\b\x00\x00\x00\x00\x00\x00", [3.0]
|
---|
109 |
|
---|
110 | if PACK_IS_LITTLE_ENDIAN
|
---|
111 | assert_pack 'd', "\x00\x00\x00\x00\x00\x00\b@", [3.0]
|
---|
112 | assert_pack 'D', "\x00\x00\x00\x00\x00\x00\b@", [3.0]
|
---|
113 | else
|
---|
114 | assert_pack 'd', "@\b\x00\x00\x00\x00\x00\x00", [3.0]
|
---|
115 | assert_pack 'D', "@\b\x00\x00\x00\x00\x00\x00", [3.0]
|
---|
116 | end
|
---|
117 | end
|
---|
118 |
|
---|
119 | assert 'pack/unpack "i"' do
|
---|
120 | int_size = [0].pack('i').size
|
---|
121 | raise "pack('i').size is too small (#{int_size})" if int_size < 2
|
---|
122 |
|
---|
123 | if PACK_IS_LITTLE_ENDIAN
|
---|
124 | str = "\xC7\xCF" + "\xFF" * (int_size-2)
|
---|
125 | else
|
---|
126 | str = "\xFF" * (int_size-2) + "\xCF\xC7"
|
---|
127 | end
|
---|
128 | assert_pack 'i', str, [-12345]
|
---|
129 | end
|
---|
130 |
|
---|
131 | assert 'pack/unpack "I"' do
|
---|
132 | uint_size = [0].pack('I').size
|
---|
133 | raise "pack('I').size is too small (#{uint_size})" if uint_size < 2
|
---|
134 |
|
---|
135 | if PACK_IS_LITTLE_ENDIAN
|
---|
136 | str = "\x39\x30" + "\0" * (uint_size-2)
|
---|
137 | else
|
---|
138 | str = "\0" * (uint_size-2) + "\x30\x39"
|
---|
139 | end
|
---|
140 | assert_pack 'I', str, [12345]
|
---|
141 | end
|
---|
142 |
|
---|
143 | assert 'pack/unpack "U"' do
|
---|
144 | assert_equal [], "".unpack("U")
|
---|
145 | assert_equal [], "".unpack("U*")
|
---|
146 | assert_equal [65, 66], "ABC".unpack("U2")
|
---|
147 | assert_equal [12371, 12435, 12395, 12385, 12399, 19990, 30028], "こんにちは世界".unpack("U*")
|
---|
148 |
|
---|
149 | assert_equal "", [].pack("U")
|
---|
150 | assert_equal "", [].pack("U*")
|
---|
151 | assert_equal "AB", [65, 66, 67].pack("U2")
|
---|
152 | assert_equal "こんにちは世界", [12371, 12435, 12395, 12385, 12399, 19990, 30028].pack("U*")
|
---|
153 |
|
---|
154 | assert_equal "\000", [0].pack("U")
|
---|
155 |
|
---|
156 | assert_raise(RangeError) { [-0x40000000].pack("U") }
|
---|
157 | assert_raise(RangeError) { [-1].pack("U") }
|
---|
158 | assert_raise(RangeError) { [0x40000000].pack("U") }
|
---|
159 | end
|
---|