net/spdy/spdy_frame_reader_test.cc

   1 // Copyright (c) 2012 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 <algorithm>
   6 #include <iostream>
   7
   8 #include "base/memory/scoped_ptr.h"
   9 #include "base/sys_byteorder.h"
  10 #include "net/spdy/spdy_frame_reader.h"
  11 #include "testing/platform_test.h"
  12
  13 namespace net {
  14
  15 TEST(SpdyFrameReaderTest, ReadUInt16) {
  16   // Frame data in network byte order.
  17   const uint16 kFrameData[] = {
  18     htons(1), htons(1<<15),
  19   };
  20
  21   SpdyFrameReader frame_reader(reinterpret_cast<const char *>(kFrameData),
  22                                arraysize(kFrameData) * sizeof(uint16));
  23   EXPECT_FALSE(frame_reader.IsDoneReading());
  24
  25   uint16 uint16_val;
  26   EXPECT_TRUE(frame_reader.ReadUInt16(&uint16_val));
  27   EXPECT_FALSE(frame_reader.IsDoneReading());
  28   EXPECT_EQ(1, uint16_val);
  29
  30   EXPECT_TRUE(frame_reader.ReadUInt16(&uint16_val));
  31   EXPECT_TRUE(frame_reader.IsDoneReading());
  32   EXPECT_EQ(1<<15, uint16_val);
  33 }
  34
  35 TEST(SpdyFrameReaderTest, ReadUInt32) {
  36   // Frame data in network byte order.
  37   const uint32 kFrameData[] = {
  38     htonl(1), htonl(1<<31),
  39   };
  40
  41   SpdyFrameReader frame_reader(reinterpret_cast<const char *>(kFrameData),
  42                                arraysize(kFrameData) * sizeof(uint32));
  43   EXPECT_FALSE(frame_reader.IsDoneReading());
  44
  45   uint32 uint32_val;
  46   EXPECT_TRUE(frame_reader.ReadUInt32(&uint32_val));
  47   EXPECT_FALSE(frame_reader.IsDoneReading());
  48   EXPECT_EQ(1u, uint32_val);
  49
  50   EXPECT_TRUE(frame_reader.ReadUInt32(&uint32_val));
  51   EXPECT_TRUE(frame_reader.IsDoneReading());
  52   EXPECT_EQ(1u<<31, uint32_val);
  53 }
  54
  55 TEST(SpdyFrameReaderTest, ReadStringPiece16) {
  56   // Frame data in network byte order.
  57   const char kFrameData[] = {
  58     0x00, 0x02,  // uint16(2)
  59     0x48, 0x69,  // "Hi"
  60     0x00, 0x10,  // uint16(16)
  61     0x54, 0x65, 0x73, 0x74,
  62     0x69, 0x6e, 0x67, 0x2c,
  63     0x20, 0x31, 0x2c, 0x20,
  64     0x32, 0x2c, 0x20, 0x33,  // "Testing, 1, 2, 3"
  65   };
  66
  67   SpdyFrameReader frame_reader(kFrameData, arraysize(kFrameData));
  68   EXPECT_FALSE(frame_reader.IsDoneReading());
  69
  70   base::StringPiece stringpiece_val;
  71   EXPECT_TRUE(frame_reader.ReadStringPiece16(&stringpiece_val));
  72   EXPECT_FALSE(frame_reader.IsDoneReading());
  73   EXPECT_EQ(0, stringpiece_val.compare("Hi"));
  74
  75   EXPECT_TRUE(frame_reader.ReadStringPiece16(&stringpiece_val));
  76   EXPECT_TRUE(frame_reader.IsDoneReading());
  77   EXPECT_EQ(0, stringpiece_val.compare("Testing, 1, 2, 3"));
  78 }
  79
  80 TEST(SpdyFrameReaderTest, ReadStringPiece32) {
  81   // Frame data in network byte order.
  82   const char kFrameData[] = {
  83     0x00, 0x00, 0x00, 0x03,  // uint32(3)
  84     0x66, 0x6f, 0x6f,  // "foo"
  85     0x00, 0x00, 0x00, 0x10,  // uint32(16)
  86     0x54, 0x65, 0x73, 0x74,
  87     0x69, 0x6e, 0x67, 0x2c,
  88     0x20, 0x34, 0x2c, 0x20,
  89     0x35, 0x2c, 0x20, 0x36,  // "Testing, 4, 5, 6"
  90   };
  91
  92   SpdyFrameReader frame_reader(kFrameData, arraysize(kFrameData));
  93   EXPECT_FALSE(frame_reader.IsDoneReading());
  94
  95   base::StringPiece stringpiece_val;
  96   EXPECT_TRUE(frame_reader.ReadStringPiece32(&stringpiece_val));
  97   EXPECT_FALSE(frame_reader.IsDoneReading());
  98   EXPECT_EQ(0, stringpiece_val.compare("foo"));
  99
 100   EXPECT_TRUE(frame_reader.ReadStringPiece32(&stringpiece_val));
 101   EXPECT_TRUE(frame_reader.IsDoneReading());
 102   EXPECT_EQ(0, stringpiece_val.compare("Testing, 4, 5, 6"));
 103 }
 104
 105 TEST(SpdyFrameReaderTest, ReadUInt16WithBufferTooSmall) {
 106   // Frame data in network byte order.
 107   const char kFrameData[] = {
 108     0x00,  // part of a uint16
 109   };
 110
 111   SpdyFrameReader frame_reader(kFrameData, arraysize(kFrameData));
 112   EXPECT_FALSE(frame_reader.IsDoneReading());
 113
 114   uint16 uint16_val;
 115   EXPECT_FALSE(frame_reader.ReadUInt16(&uint16_val));
 116 }
 117
 118 TEST(SpdyFrameReaderTest, ReadUInt32WithBufferTooSmall) {
 119   // Frame data in network byte order.
 120   const char kFrameData[] = {
 121     0x00, 0x00, 0x00,  // part of a uint32
 122   };
 123
 124   SpdyFrameReader frame_reader(kFrameData, arraysize(kFrameData));
 125   EXPECT_FALSE(frame_reader.IsDoneReading());
 126
 127   uint32 uint32_val;
 128   EXPECT_FALSE(frame_reader.ReadUInt32(&uint32_val));
 129
 130   // Also make sure that trying to read a uint16, which technically could work,
 131   // fails immediately due to previously encountered failed read.
 132   uint16 uint16_val;
 133   EXPECT_FALSE(frame_reader.ReadUInt16(&uint16_val));
 134 }
 135
 136 // Tests ReadStringPiece16() with a buffer too small to fit the entire string.
 137 TEST(SpdyFrameReaderTest, ReadStringPiece16WithBufferTooSmall) {
 138   // Frame data in network byte order.
 139   const char kFrameData[] = {
 140     0x00, 0x03,  // uint16(3)
 141     0x48, 0x69,  // "Hi"
 142   };
 143
 144   SpdyFrameReader frame_reader(kFrameData, arraysize(kFrameData));
 145   EXPECT_FALSE(frame_reader.IsDoneReading());
 146
 147   base::StringPiece stringpiece_val;
 148   EXPECT_FALSE(frame_reader.ReadStringPiece16(&stringpiece_val));
 149
 150   // Also make sure that trying to read a uint16, which technically could work,
 151   // fails immediately due to previously encountered failed read.
 152   uint16 uint16_val;
 153   EXPECT_FALSE(frame_reader.ReadUInt16(&uint16_val));
 154 }
 155
 156 // Tests ReadStringPiece16() with a buffer too small even to fit the length.
 157 TEST(SpdyFrameReaderTest, ReadStringPiece16WithBufferWayTooSmall) {
 158   // Frame data in network byte order.
 159   const char kFrameData[] = {
 160     0x00,  // part of a uint16
 161   };
 162
 163   SpdyFrameReader frame_reader(kFrameData, arraysize(kFrameData));
 164   EXPECT_FALSE(frame_reader.IsDoneReading());
 165
 166   base::StringPiece stringpiece_val;
 167   EXPECT_FALSE(frame_reader.ReadStringPiece16(&stringpiece_val));
 168
 169   // Also make sure that trying to read a uint16, which technically could work,
 170   // fails immediately due to previously encountered failed read.
 171   uint16 uint16_val;
 172   EXPECT_FALSE(frame_reader.ReadUInt16(&uint16_val));
 173 }
 174
 175 // Tests ReadStringPiece32() with a buffer too small to fit the entire string.
 176 TEST(SpdyFrameReaderTest, ReadStringPiece32WithBufferTooSmall) {
 177   // Frame data in network byte order.
 178   const char kFrameData[] = {
 179     0x00, 0x00, 0x00, 0x03,  // uint32(3)
 180     0x48, 0x69,  // "Hi"
 181   };
 182
 183   SpdyFrameReader frame_reader(kFrameData, arraysize(kFrameData));
 184   EXPECT_FALSE(frame_reader.IsDoneReading());
 185
 186   base::StringPiece stringpiece_val;
 187   EXPECT_FALSE(frame_reader.ReadStringPiece32(&stringpiece_val));
 188
 189   // Also make sure that trying to read a uint16, which technically could work,
 190   // fails immediately due to previously encountered failed read.
 191   uint16 uint16_val;
 192   EXPECT_FALSE(frame_reader.ReadUInt16(&uint16_val));
 193 }
 194
 195 // Tests ReadStringPiece32() with a buffer too small even to fit the length.
 196 TEST(SpdyFrameReaderTest, ReadStringPiece32WithBufferWayTooSmall) {
 197   // Frame data in network byte order.
 198   const char kFrameData[] = {
 199     0x00, 0x00, 0x00,  // part of a uint32
 200   };
 201
 202   SpdyFrameReader frame_reader(kFrameData, arraysize(kFrameData));
 203   EXPECT_FALSE(frame_reader.IsDoneReading());
 204
 205   base::StringPiece stringpiece_val;
 206   EXPECT_FALSE(frame_reader.ReadStringPiece32(&stringpiece_val));
 207
 208   // Also make sure that trying to read a uint16, which technically could work,
 209   // fails immediately due to previously encountered failed read.
 210   uint16 uint16_val;
 211   EXPECT_FALSE(frame_reader.ReadUInt16(&uint16_val));
 212 }
 213
 214 TEST(SpdyFrameReaderTest, ReadBytes) {
 215   // Frame data in network byte order.
 216   const char kFrameData[] = {
 217     0x66, 0x6f, 0x6f,  // "foo"
 218     0x48, 0x69,  // "Hi"
 219   };
 220
 221   SpdyFrameReader frame_reader(kFrameData, arraysize(kFrameData));
 222   EXPECT_FALSE(frame_reader.IsDoneReading());
 223
 224   char dest1[3] = {};
 225   EXPECT_TRUE(frame_reader.ReadBytes(&dest1, arraysize(dest1)));
 226   EXPECT_FALSE(frame_reader.IsDoneReading());
 227   EXPECT_EQ("foo", base::StringPiece(dest1, arraysize(dest1)));
 228
 229   char dest2[2] = {};
 230   EXPECT_TRUE(frame_reader.ReadBytes(&dest2, arraysize(dest2)));
 231   EXPECT_TRUE(frame_reader.IsDoneReading());
 232   EXPECT_EQ("Hi", base::StringPiece(dest2, arraysize(dest2)));
 233 }
 234
 235 TEST(SpdyFrameReaderTest, ReadBytesWithBufferTooSmall) {
 236   // Frame data in network byte order.
 237   const char kFrameData[] = {
 238     0x01,
 239   };
 240
 241   SpdyFrameReader frame_reader(kFrameData, arraysize(kFrameData));
 242   EXPECT_FALSE(frame_reader.IsDoneReading());
 243
 244   char dest[arraysize(kFrameData) + 2] = {};
 245   EXPECT_FALSE(frame_reader.ReadBytes(&dest, arraysize(kFrameData) + 1));
 246   EXPECT_STREQ("", dest);
 247 }
 248
 249 }  // namespace net