content/renderer/npapi/webplugin_impl_unittest.cc

   1 // Copyright (c) 2010 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 "base/strings/string_util.h"
   6 #include "content/renderer/npapi/webplugin_impl.h"
   7 #include "testing/gtest/include/gtest/gtest.h"
   8 #include "third_party/WebKit/public/platform/WebCString.h"
   9 #include "third_party/WebKit/public/platform/WebString.h"
  10 #include "third_party/WebKit/public/platform/WebURLRequest.h"
  11
  12 using blink::WebHTTPBody;
  13 using blink::WebString;
  14 using blink::WebURLRequest;
  15
  16 namespace content {
  17
  18 namespace {
  19
  20 std::string GetHeader(const WebURLRequest& request, const char* name) {
  21   std::string result;
  22   base::TrimWhitespace(
  23       request.httpHeaderField(WebString::fromUTF8(name)).utf8(),
  24       base::TRIM_ALL,
  25       &result);
  26   return result;
  27 }
  28
  29 std::string GetBodyText(const WebURLRequest& request) {
  30   const WebHTTPBody& body = request.httpBody();
  31   if (body.isNull())
  32     return std::string();
  33
  34   std::string result;
  35   size_t i = 0;
  36   WebHTTPBody::Element element;
  37   while (body.elementAt(i++, element)) {
  38     if (element.type == WebHTTPBody::Element::TypeData) {
  39       result.append(element.data.data(), element.data.size());
  40     } else {
  41       NOTREACHED() << "unexpected element type encountered!";
  42     }
  43   }
  44   return result;
  45 }
  46
  47 }  // namespace
  48
  49 // The Host functions for NPN_PostURL and NPN_PostURLNotify
  50 // need to parse out some HTTP headers.  Make sure it works
  51 // with the following tests
  52
  53 TEST(WebPluginImplTest, PostParserSimple) {
  54   // Test a simple case with headers & data
  55   const char *ex1 = "foo: bar\nContent-length: 10\n\nabcdefghij";
  56   WebURLRequest request;
  57   request.initialize();
  58   bool rv = WebPluginImpl::SetPostData(&request, ex1,
  59                                        static_cast<uint32>(strlen(ex1)));
  60   EXPECT_TRUE(rv);
  61   EXPECT_EQ("bar", GetHeader(request, "foo"));
  62   EXPECT_EQ(0U, GetHeader(request, "bar").length());
  63   EXPECT_EQ(0U, GetHeader(request, "Content-length").length());
  64   EXPECT_EQ("abcdefghij", GetBodyText(request));
  65 }
  66
  67 TEST(WebPluginImplTest, PostParserLongHeader) {
  68   // Test a simple case with long headers
  69   const char *ex1 = "foo: 0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789\n\nabcdefghij";
  70   WebURLRequest request;
  71   request.initialize();
  72   bool rv = WebPluginImpl::SetPostData(&request, ex1,
  73                                        static_cast<uint32>(strlen(ex1)));
  74   EXPECT_TRUE(rv);
  75   EXPECT_EQ(100U, GetHeader(request, "foo").length());
  76 }
  77
  78 TEST(WebPluginImplTest, PostParserManyHeaders) {
  79   // Test a simple case with long headers
  80   const char *ex1 = "h1:h1\nh2:h2\nh3:h3\nh4:h4\nh5:h5\nh6:h6\nh7:h7\nh8:h8\nh9:h9\nh10:h10\n\nbody";
  81   WebURLRequest request;
  82   request.initialize();
  83   bool rv = WebPluginImpl::SetPostData(&request, ex1,
  84                                        static_cast<uint32>(strlen(ex1)));
  85   EXPECT_TRUE(rv);
  86   EXPECT_EQ("h1", GetHeader(request, "h1"));
  87   EXPECT_EQ("h2", GetHeader(request, "h2"));
  88   EXPECT_EQ("h3", GetHeader(request, "h3"));
  89   EXPECT_EQ("h4", GetHeader(request, "h4"));
  90   EXPECT_EQ("h5", GetHeader(request, "h5"));
  91   EXPECT_EQ("h6", GetHeader(request, "h6"));
  92   EXPECT_EQ("h7", GetHeader(request, "h7"));
  93   EXPECT_EQ("h8", GetHeader(request, "h8"));
  94   EXPECT_EQ("h9", GetHeader(request, "h9"));
  95   EXPECT_EQ("h10", GetHeader(request, "h10"));
  96   EXPECT_EQ("body", GetBodyText(request));
  97 }
  98
  99 TEST(WebPluginImplTest, PostParserDuplicateHeaders) {
 100   // Test a simple case with long headers
 101   // What value gets returned doesn't really matter.  It shouldn't error
 102   // out.
 103   const char *ex1 = "h1:h1\nh1:h2\n\nbody";
 104   WebURLRequest request;
 105   request.initialize();
 106   bool rv = WebPluginImpl::SetPostData(&request, ex1,
 107                                        static_cast<uint32>(strlen(ex1)));
 108   EXPECT_TRUE(rv);
 109 }
 110
 111 TEST(WebPluginImplTest, PostParserNoHeaders) {
 112   // Test a simple case with no headers but with data
 113   const char *ex1 = "\nabcdefghij";
 114   WebURLRequest request;
 115   request.initialize();
 116   bool rv = WebPluginImpl::SetPostData(&request, ex1,
 117                                        static_cast<uint32>(strlen(ex1)));
 118   EXPECT_TRUE(rv);
 119   EXPECT_EQ(0U, GetHeader(request, "foo").length());
 120   EXPECT_EQ(0U, GetHeader(request, "bar").length());
 121   EXPECT_EQ(0U, GetHeader(request, "Content-length").length());
 122   EXPECT_EQ("abcdefghij", GetBodyText(request));
 123 }
 124
 125 TEST(WebPluginImplTest, PostParserNoBody) {
 126   // Test a simple case with headers and no body
 127   const char *ex1 = "Foo:bar\n\n";
 128   WebURLRequest request;
 129   request.initialize();
 130   bool rv = WebPluginImpl::SetPostData(&request, ex1,
 131                                        static_cast<uint32>(strlen(ex1)));
 132   EXPECT_TRUE(rv);
 133   EXPECT_EQ("bar", GetHeader(request, "foo"));
 134   EXPECT_EQ(0U, GetHeader(request, "bar").length());
 135   EXPECT_EQ(0U, GetHeader(request, "Content-length").length());
 136   EXPECT_EQ(0U, GetBodyText(request).length());
 137 }
 138
 139 TEST(WebPluginImplTest, PostParserBodyWithNewLines) {
 140   // Test a simple case with headers and no body
 141   const char *ex1 = "Foo:bar\n\n\n\nabcdefg\n\nabcdefg";
 142   WebURLRequest request;
 143   request.initialize();
 144   bool rv = WebPluginImpl::SetPostData(&request, ex1,
 145                                        static_cast<uint32>(strlen(ex1)));
 146   EXPECT_TRUE(rv);
 147   EXPECT_EQ(GetBodyText(request), "\n\nabcdefg\n\nabcdefg");
 148 }
 149
 150 TEST(WebPluginImplTest, PostParserErrorNoBody) {
 151   // Test with headers and no body
 152   const char *ex1 = "Foo:bar\n";
 153   WebURLRequest request;
 154   request.initialize();
 155   bool rv = WebPluginImpl::SetPostData(&request, ex1,
 156                                        static_cast<uint32>(strlen(ex1)));
 157   EXPECT_TRUE(rv);
 158 }
 159
 160 TEST(WebPluginImplTest, PostParserErrorEmpty) {
 161   // Test with an empty string
 162   const char *ex1 = "";
 163   WebURLRequest request;
 164   request.initialize();
 165   bool rv = WebPluginImpl::SetPostData(&request, ex1,
 166                                        static_cast<uint32>(strlen(ex1)));
 167   EXPECT_TRUE(rv);
 168 }
 169
 170 TEST(WebPluginImplTest, PostParserEmptyName) {
 171   // Test an error case with an empty header name field
 172   const char *ex1 = "foo:bar\n:blat\n\nbody";
 173   WebURLRequest request;
 174   request.initialize();
 175   bool rv = WebPluginImpl::SetPostData(&request, ex1,
 176                                        static_cast<uint32>(strlen(ex1)));
 177   EXPECT_TRUE(rv);
 178   EXPECT_EQ("bar", GetHeader(request, "foo"));
 179   EXPECT_EQ("body", GetBodyText(request));
 180 }
 181
 182 TEST(WebPluginImplTest, PostParserEmptyValue) {
 183   // Test an error case with an empty value field
 184   const char *ex1 = "foo:bar\nbar:\n\nbody";
 185   WebURLRequest request;
 186   request.initialize();
 187   bool rv = WebPluginImpl::SetPostData(&request, ex1,
 188                                        static_cast<uint32>(strlen(ex1)));
 189   EXPECT_TRUE(rv);
 190   EXPECT_EQ("bar", GetHeader(request, "foo"));
 191   EXPECT_EQ(0U, GetHeader(request, "bar").length());
 192   EXPECT_EQ("body", GetBodyText(request));
 193 }
 194
 195 TEST(WebPluginImplTest, PostParserCRLF) {
 196   // Test an error case with an empty value field
 197   const char *ex1 = "foo: bar\r\nbar:\r\n\r\nbody\r\n\r\nbody2";
 198   WebURLRequest request;
 199   request.initialize();
 200   bool rv = WebPluginImpl::SetPostData(&request, ex1,
 201                                        static_cast<uint32>(strlen(ex1)));
 202   EXPECT_TRUE(rv);
 203   EXPECT_EQ("bar", GetHeader(request, "foo"));
 204   EXPECT_EQ(0U, GetHeader(request, "bar").length());
 205   EXPECT_EQ("body\r\n\r\nbody2", GetBodyText(request));
 206 }
 207
 208 TEST(WebPluginImplTest, PostParserBodyWithBinaryData) {
 209   // Test a simple case with headers and binary data.
 210   char ex1[33] = "foo: bar\nContent-length: 10\n\n";
 211   unsigned int binary_data = 0xFFFFFFF0;
 212   memcpy(ex1 + strlen("foo: bar\nContent-length: 10\n\n"), &binary_data,
 213         sizeof(binary_data));
 214
 215   WebURLRequest request;
 216   request.initialize();
 217   bool rv = WebPluginImpl::SetPostData(&request, ex1,
 218                                       sizeof(ex1)/sizeof(ex1[0]));
 219   EXPECT_TRUE(rv);
 220   EXPECT_EQ("bar", GetHeader(request, "foo"));
 221   EXPECT_EQ(0U, GetHeader(request, "bar").length());
 222   EXPECT_EQ(0U, GetHeader(request, "Content-length").length());
 223
 224   std::string body = GetBodyText(request);
 225
 226   EXPECT_EQ(0xF0, (unsigned char)body[0]);
 227   EXPECT_EQ(0xFF, (unsigned char)body[1]);
 228   EXPECT_EQ(0xFF, (unsigned char)body[2]);
 229   EXPECT_EQ(0xFF, (unsigned char)body[3]);
 230 }
 231
 232 }  // namespace content