1 // Copyright 2017 The Chromium Authors
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "net/ntlm/ntlm_buffer_writer.h"
6
7 #include "base/strings/utf_string_conversions.h"
8 #include "testing/gtest/include/gtest/gtest.h"
9
10 namespace net::ntlm {
11
12 namespace {
13
14 // Helper method to get a raw pointer to the buffer.
GetBufferPtr(const NtlmBufferWriter & writer)15 const uint8_t* GetBufferPtr(const NtlmBufferWriter& writer) {
16 return writer.GetBuffer().data();
17 }
18
19 // Helper method to get a byte at a specific index in the buffer.
GetByteFromBuffer(const NtlmBufferWriter & writer,size_t index)20 uint8_t GetByteFromBuffer(const NtlmBufferWriter& writer, size_t index) {
21 EXPECT_TRUE(index < writer.GetLength());
22 return writer.GetBuffer()[index];
23 }
24
25 } // namespace
26
TEST(NtlmBufferWriterTest,Initialization)27 TEST(NtlmBufferWriterTest, Initialization) {
28 NtlmBufferWriter writer(1);
29
30 ASSERT_EQ(1u, writer.GetLength());
31 ASSERT_EQ(1u, writer.GetBuffer().size());
32 ASSERT_EQ(0u, writer.GetCursor());
33 ASSERT_FALSE(writer.IsEndOfBuffer());
34 ASSERT_TRUE(writer.CanWrite(1));
35 ASSERT_FALSE(writer.CanWrite(2));
36 }
37
TEST(NtlmBufferWriterTest,EmptyWrite)38 TEST(NtlmBufferWriterTest, EmptyWrite) {
39 NtlmBufferWriter writer(0);
40
41 ASSERT_EQ(0u, writer.GetLength());
42 ASSERT_EQ(0u, writer.GetBuffer().size());
43 ASSERT_EQ(0u, writer.GetCursor());
44 ASSERT_EQ(nullptr, GetBufferPtr(writer));
45
46 // An empty (zero-byte) write into a zero-byte writer should succeed as a
47 // no-op.
48 std::vector<uint8_t> b;
49 ASSERT_TRUE(writer.CanWrite(0));
50 ASSERT_TRUE(writer.WriteBytes(b));
51
52 ASSERT_EQ(0u, writer.GetLength());
53 ASSERT_EQ(0u, writer.GetBuffer().size());
54 ASSERT_EQ(0u, writer.GetCursor());
55 ASSERT_EQ(nullptr, GetBufferPtr(writer));
56
57 // An empty (zero-byte) write into a non-zero-byte writer should succeed as
58 // a no-op.
59 NtlmBufferWriter writer2(1);
60 ASSERT_EQ(1u, writer2.GetLength());
61 ASSERT_EQ(1u, writer2.GetBuffer().size());
62 ASSERT_EQ(0u, writer2.GetCursor());
63 ASSERT_NE(nullptr, GetBufferPtr(writer2));
64
65 ASSERT_TRUE(writer2.CanWrite(0));
66 ASSERT_TRUE(writer2.WriteBytes(b));
67
68 ASSERT_EQ(1u, writer2.GetLength());
69 ASSERT_EQ(1u, writer2.GetBuffer().size());
70 ASSERT_EQ(0u, writer2.GetCursor());
71 ASSERT_NE(nullptr, GetBufferPtr(writer2));
72 }
73
TEST(NtlmBufferWriterTest,Write16)74 TEST(NtlmBufferWriterTest, Write16) {
75 uint8_t expected[2] = {0x22, 0x11};
76 const uint16_t value = 0x1122;
77
78 NtlmBufferWriter writer(sizeof(uint16_t));
79
80 ASSERT_TRUE(writer.WriteUInt16(value));
81 ASSERT_TRUE(writer.IsEndOfBuffer());
82 ASSERT_EQ(std::size(expected), writer.GetLength());
83 ASSERT_FALSE(writer.WriteUInt16(value));
84
85 ASSERT_EQ(0,
86 memcmp(expected, writer.GetBuffer().data(), std::size(expected)));
87 }
88
TEST(NtlmBufferWriterTest,Write16PastEob)89 TEST(NtlmBufferWriterTest, Write16PastEob) {
90 NtlmBufferWriter writer(sizeof(uint16_t) - 1);
91
92 ASSERT_FALSE(writer.WriteUInt16(0));
93 ASSERT_EQ(0u, writer.GetCursor());
94 }
95
TEST(NtlmBufferWriterTest,Write32)96 TEST(NtlmBufferWriterTest, Write32) {
97 uint8_t expected[4] = {0x44, 0x33, 0x22, 0x11};
98 const uint32_t value = 0x11223344;
99
100 NtlmBufferWriter writer(sizeof(uint32_t));
101
102 ASSERT_TRUE(writer.WriteUInt32(value));
103 ASSERT_TRUE(writer.IsEndOfBuffer());
104 ASSERT_FALSE(writer.WriteUInt32(value));
105
106 ASSERT_EQ(0, memcmp(expected, GetBufferPtr(writer), std::size(expected)));
107 }
108
TEST(NtlmBufferWriterTest,Write32PastEob)109 TEST(NtlmBufferWriterTest, Write32PastEob) {
110 NtlmBufferWriter writer(sizeof(uint32_t) - 1);
111
112 ASSERT_FALSE(writer.WriteUInt32(0));
113 ASSERT_EQ(0u, writer.GetCursor());
114 }
115
TEST(NtlmBufferWriterTest,Write64)116 TEST(NtlmBufferWriterTest, Write64) {
117 uint8_t expected[8] = {0x88, 0x77, 0x66, 0x55, 0x44, 0x33, 0x22, 0x11};
118 const uint64_t value = 0x1122334455667788;
119
120 NtlmBufferWriter writer(sizeof(uint64_t));
121
122 ASSERT_TRUE(writer.WriteUInt64(value));
123 ASSERT_TRUE(writer.IsEndOfBuffer());
124 ASSERT_FALSE(writer.WriteUInt64(value));
125
126 ASSERT_EQ(0, memcmp(expected, GetBufferPtr(writer), std::size(expected)));
127 }
128
TEST(NtlmBufferWriterTest,Write64PastEob)129 TEST(NtlmBufferWriterTest, Write64PastEob) {
130 NtlmBufferWriter writer(sizeof(uint64_t) - 1);
131
132 ASSERT_FALSE(writer.WriteUInt64(0));
133 ASSERT_EQ(0u, writer.GetCursor());
134 }
135
TEST(NtlmBufferWriterTest,WriteBytes)136 TEST(NtlmBufferWriterTest, WriteBytes) {
137 uint8_t expected[8] = {0x88, 0x77, 0x66, 0x55, 0x44, 0x33, 0x22, 0x11};
138
139 NtlmBufferWriter writer(std::size(expected));
140
141 ASSERT_TRUE(writer.WriteBytes(expected));
142 ASSERT_EQ(0, memcmp(GetBufferPtr(writer), expected, std::size(expected)));
143 ASSERT_TRUE(writer.IsEndOfBuffer());
144 ASSERT_FALSE(writer.WriteBytes(base::make_span(expected, 1u)));
145
146 ASSERT_EQ(0, memcmp(expected, GetBufferPtr(writer), std::size(expected)));
147 }
148
TEST(NtlmBufferWriterTest,WriteBytesPastEob)149 TEST(NtlmBufferWriterTest, WriteBytesPastEob) {
150 uint8_t buffer[8];
151
152 NtlmBufferWriter writer(std::size(buffer) - 1);
153
154 ASSERT_FALSE(writer.WriteBytes(buffer));
155 }
156
TEST(NtlmBufferWriterTest,WriteSecurityBuffer)157 TEST(NtlmBufferWriterTest, WriteSecurityBuffer) {
158 uint8_t expected[8] = {0x22, 0x11, 0x22, 0x11, 0x88, 0x77, 0x66, 0x55};
159 uint16_t length = 0x1122;
160 uint32_t offset = 0x55667788;
161
162 NtlmBufferWriter writer(kSecurityBufferLen);
163
164 ASSERT_TRUE(writer.WriteSecurityBuffer(SecurityBuffer(offset, length)));
165 ASSERT_TRUE(writer.IsEndOfBuffer());
166 ASSERT_FALSE(writer.WriteSecurityBuffer(SecurityBuffer(offset, length)));
167
168 ASSERT_EQ(0, memcmp(expected, GetBufferPtr(writer), std::size(expected)));
169 }
170
TEST(NtlmBufferWriterTest,WriteSecurityBufferPastEob)171 TEST(NtlmBufferWriterTest, WriteSecurityBufferPastEob) {
172 SecurityBuffer sec_buf;
173 NtlmBufferWriter writer(kSecurityBufferLen - 1);
174
175 ASSERT_FALSE(writer.WriteSecurityBuffer(sec_buf));
176 }
177
TEST(NtlmBufferWriterTest,WriteNarrowString)178 TEST(NtlmBufferWriterTest, WriteNarrowString) {
179 uint8_t expected[8] = {'1', '2', '3', '4', '5', '6', '7', '8'};
180 std::string value("12345678");
181
182 NtlmBufferWriter writer(value.size());
183
184 ASSERT_TRUE(writer.WriteUtf8String(value));
185 ASSERT_TRUE(writer.IsEndOfBuffer());
186 ASSERT_FALSE(writer.WriteUtf8String(value));
187
188 ASSERT_EQ(0, memcmp(expected, GetBufferPtr(writer), std::size(expected)));
189 }
190
TEST(NtlmBufferWriterTest,WriteAsciiStringPastEob)191 TEST(NtlmBufferWriterTest, WriteAsciiStringPastEob) {
192 std::string str("12345678");
193 NtlmBufferWriter writer(str.length() - 1);
194
195 ASSERT_FALSE(writer.WriteUtf8String(str));
196 }
197
TEST(NtlmBufferWriterTest,WriteUtf16String)198 TEST(NtlmBufferWriterTest, WriteUtf16String) {
199 uint8_t expected[16] = {'1', 0, '2', 0, '3', 0, '4', 0,
200 '5', 0, '6', 0, '7', 0, '8', 0};
201 std::u16string value = u"12345678";
202
203 NtlmBufferWriter writer(value.size() * 2);
204
205 ASSERT_TRUE(writer.WriteUtf16String(value));
206 ASSERT_TRUE(writer.IsEndOfBuffer());
207 ASSERT_FALSE(writer.WriteUtf16String(value));
208
209 ASSERT_EQ(0, memcmp(expected, GetBufferPtr(writer), std::size(expected)));
210 }
211
TEST(NtlmBufferWriterTest,WriteUtf16StringPastEob)212 TEST(NtlmBufferWriterTest, WriteUtf16StringPastEob) {
213 std::u16string str = u"12345678";
214 NtlmBufferWriter writer((str.length() * 2) - 1);
215
216 ASSERT_FALSE(writer.WriteUtf16String(str));
217 }
218
TEST(NtlmBufferWriterTest,WriteUtf8AsUtf16String)219 TEST(NtlmBufferWriterTest, WriteUtf8AsUtf16String) {
220 uint8_t expected[16] = {'1', 0, '2', 0, '3', 0, '4', 0,
221 '5', 0, '6', 0, '7', 0, '8', 0};
222 std::string input = "12345678";
223
224 NtlmBufferWriter writer(input.size() * 2);
225
226 ASSERT_TRUE(writer.WriteUtf8AsUtf16String(input));
227 ASSERT_TRUE(writer.IsEndOfBuffer());
228 ASSERT_FALSE(writer.WriteUtf8AsUtf16String(input));
229
230 ASSERT_EQ(0, memcmp(expected, GetBufferPtr(writer), std::size(expected)));
231 }
232
TEST(NtlmBufferWriterTest,WriteSignature)233 TEST(NtlmBufferWriterTest, WriteSignature) {
234 uint8_t expected[8] = {'N', 'T', 'L', 'M', 'S', 'S', 'P', 0};
235 NtlmBufferWriter writer(kSignatureLen);
236
237 ASSERT_TRUE(writer.WriteSignature());
238 ASSERT_TRUE(writer.IsEndOfBuffer());
239
240 ASSERT_EQ(0, memcmp(expected, GetBufferPtr(writer), std::size(expected)));
241 }
242
TEST(NtlmBufferWriterTest,WriteSignaturePastEob)243 TEST(NtlmBufferWriterTest, WriteSignaturePastEob) {
244 NtlmBufferWriter writer(1);
245
246 ASSERT_FALSE(writer.WriteSignature());
247 }
248
TEST(NtlmBufferWriterTest,WriteMessageType)249 TEST(NtlmBufferWriterTest, WriteMessageType) {
250 NtlmBufferWriter writer(4);
251
252 ASSERT_TRUE(writer.WriteMessageType(MessageType::kNegotiate));
253 ASSERT_TRUE(writer.IsEndOfBuffer());
254 ASSERT_EQ(static_cast<uint32_t>(MessageType::kNegotiate),
255 GetByteFromBuffer(writer, 0));
256 ASSERT_EQ(0, GetByteFromBuffer(writer, 1));
257 ASSERT_EQ(0, GetByteFromBuffer(writer, 2));
258 ASSERT_EQ(0, GetByteFromBuffer(writer, 3));
259 }
260
TEST(NtlmBufferWriterTest,WriteMessageTypePastEob)261 TEST(NtlmBufferWriterTest, WriteMessageTypePastEob) {
262 NtlmBufferWriter writer(sizeof(uint32_t) - 1);
263
264 ASSERT_FALSE(writer.WriteMessageType(MessageType::kNegotiate));
265 }
266
TEST(NtlmBufferWriterTest,WriteAvPairHeader)267 TEST(NtlmBufferWriterTest, WriteAvPairHeader) {
268 const uint8_t expected[4] = {0x06, 0x00, 0x11, 0x22};
269 NtlmBufferWriter writer(std::size(expected));
270
271 ASSERT_TRUE(writer.WriteAvPairHeader(TargetInfoAvId::kFlags, 0x2211));
272 ASSERT_TRUE(writer.IsEndOfBuffer());
273
274 ASSERT_EQ(0, memcmp(expected, GetBufferPtr(writer), std::size(expected)));
275 }
276
TEST(NtlmBufferWriterTest,WriteAvPairHeaderPastEob)277 TEST(NtlmBufferWriterTest, WriteAvPairHeaderPastEob) {
278 NtlmBufferWriter writer(kAvPairHeaderLen - 1);
279
280 ASSERT_FALSE(writer.WriteAvPairHeader(TargetInfoAvId::kFlags, 0x2211));
281 ASSERT_EQ(0u, writer.GetCursor());
282 }
283
284 } // namespace net::ntlm
285