1// Copyright (c) 2013 The Chromium Authors. All rights reserved.
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/spdy/spdy_frame_builder.h"
6
7#include "net/spdy/spdy_framer.h"
8#include "net/spdy/spdy_protocol.h"
9#include "testing/platform_test.h"
10
11namespace net {
12
13class SpdyFrameBuilderTest : public ::testing::TestWithParam<SpdyMajorVersion> {
14 protected:
15  virtual void SetUp() {
16    spdy_version_ = GetParam();
17  }
18
19  // Major version of SPDY protocol to be used.
20  SpdyMajorVersion spdy_version_;
21};
22
23// All tests are run with two different SPDY versions: SPDY/2 and SPDY/3.
24INSTANTIATE_TEST_CASE_P(SpdyFrameBuilderTests,
25                        SpdyFrameBuilderTest,
26                        ::testing::Values(SPDY2, SPDY3, SPDY4));
27
28TEST_P(SpdyFrameBuilderTest, GetWritableBuffer) {
29  const size_t builder_size = 10;
30  SpdyFrameBuilder builder(builder_size, spdy_version_);
31  char* writable_buffer = builder.GetWritableBuffer(builder_size);
32  memset(writable_buffer, ~1, builder_size);
33  EXPECT_TRUE(builder.Seek(builder_size));
34  scoped_ptr<SpdyFrame> frame(builder.take());
35  char expected[builder_size];
36  memset(expected, ~1, builder_size);
37  EXPECT_EQ(base::StringPiece(expected, builder_size),
38            base::StringPiece(frame->data(), builder_size));
39}
40
41TEST_P(SpdyFrameBuilderTest, RewriteLength) {
42  // Create an empty SETTINGS frame both via framer and manually via builder.
43  // The one created via builder is initially given the incorrect length, but
44  // then is corrected via RewriteLength().
45  SpdyFramer framer(spdy_version_);
46  SpdySettingsIR settings_ir;
47  scoped_ptr<SpdyFrame> expected(framer.SerializeSettings(settings_ir));
48  SpdyFrameBuilder builder(expected->size() + 1, spdy_version_);
49  if (spdy_version_ <= SPDY3) {
50    builder.WriteControlFrameHeader(framer, SETTINGS, 0);
51    builder.WriteUInt32(0);  // Write the number of settings.
52  } else {
53    builder.BeginNewFrame(framer, SETTINGS, 0, 0);
54  }
55  EXPECT_TRUE(builder.GetWritableBuffer(1) != NULL);
56  builder.RewriteLength(framer);
57  scoped_ptr<SpdyFrame> built(builder.take());
58  EXPECT_EQ(base::StringPiece(expected->data(), expected->size()),
59            base::StringPiece(built->data(), expected->size()));
60}
61
62TEST_P(SpdyFrameBuilderTest, OverwriteFlags) {
63  // Create a HEADERS frame both via framer and manually via builder with
64  // different flags set, then make them match using OverwriteFlags().
65  SpdyFramer framer(spdy_version_);
66  if (spdy_version_ <= SPDY3) {
67    return;
68  }
69  SpdyHeadersIR headers_ir(1);
70  scoped_ptr<SpdyFrame> expected(framer.SerializeHeaders(headers_ir));
71  SpdyFrameBuilder builder(expected->size(), spdy_version_);
72  builder.BeginNewFrame(framer, HEADERS, 0, 1);
73  builder.OverwriteFlags(framer, HEADERS_FLAG_END_HEADERS);
74  scoped_ptr<SpdyFrame> built(builder.take());
75  EXPECT_EQ(base::StringPiece(expected->data(), expected->size()),
76            base::StringPiece(built->data(), built->size()));
77}
78
79}  // namespace net
80