Use extensions socket permission for TCP/UDP socket APIs in Pepper

chrome/common/extensions/permissions/socket_permission_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <string>
 
 #include "base/pickle.h"
 #include "base/values.h"
 #include "chrome/common/extensions/permissions/permissions_info.h"
 #include "chrome/common/extensions/permissions/socket_permission.h"
 #include "chrome/common/extensions/permissions/socket_permission_data.h"
 #include "ipc/ipc_message.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
+using content::SocketPermissionRequest;
 using extensions::SocketPermissionData;
 
 namespace {
 
 std::string Parse(const std::string& permission) {
   SocketPermissionData data;
   CHECK(data.Parse(permission)) << "Parse permission \"" << permission
     << "\" failed.";
   return data.GetAsString();
 }
@@ skipping 107 matching lines @@
       "tcp-connect:www.example.com:*");
 
   EXPECT_EQ(Parse("tcp-connect:*.example.com:99"),
       "tcp-connect:*.example.com:99");
 }
 
 TEST(SocketPermissionTest, Match) {
   SocketPermissionData data;
 
   CHECK(data.Parse("tcp-connect"));
-  EXPECT_TRUE(data.Match(
-        SocketPermissionData::TCP_CONNECT, "www.example.com", 80));
-  EXPECT_FALSE(data.Match(
-        SocketPermissionData::UDP_SEND_TO, "www.example.com", 80));
+  EXPECT_TRUE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::TCP_CONNECT, "www.example.com", 80)));
+  EXPECT_FALSE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_SEND_TO, "www.example.com", 80)));
 
   CHECK(data.Parse("udp-send-to::8800"));
-  EXPECT_TRUE(data.Match(
-        SocketPermissionData::UDP_SEND_TO, "www.example.com", 8800));
-  EXPECT_TRUE(data.Match(
-        SocketPermissionData::UDP_SEND_TO, "smtp.example.com", 8800));
-  EXPECT_FALSE(data.Match(
-        SocketPermissionData::TCP_CONNECT, "www.example.com", 80));
+  EXPECT_TRUE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_SEND_TO, "www.example.com", 8800)));
+  EXPECT_TRUE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_SEND_TO, "smtp.example.com", 8800)));
+  EXPECT_FALSE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::TCP_CONNECT, "www.example.com", 80)));
 
   CHECK(data.Parse("udp-send-to:*.example.com:8800"));
-  EXPECT_TRUE(data.Match(
-        SocketPermissionData::UDP_SEND_TO, "www.example.com", 8800));
-  EXPECT_TRUE(data.Match(
-        SocketPermissionData::UDP_SEND_TO, "smtp.example.com", 8800));
-  EXPECT_TRUE(data.Match(
-        SocketPermissionData::UDP_SEND_TO, "SMTP.example.com", 8800));
-  EXPECT_FALSE(data.Match(
-        SocketPermissionData::TCP_CONNECT, "www.example.com", 80));
-  EXPECT_FALSE(data.Match(
-        SocketPermissionData::UDP_SEND_TO, "www.google.com", 8800));
-  EXPECT_FALSE(data.Match(
-        SocketPermissionData::UDP_SEND_TO, "wwwexample.com", 8800));
+  EXPECT_TRUE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_SEND_TO, "www.example.com", 8800)));
+  EXPECT_TRUE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_SEND_TO, "smtp.example.com", 8800)));
+  EXPECT_TRUE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_SEND_TO, "SMTP.example.com", 8800)));
+  EXPECT_FALSE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::TCP_CONNECT, "www.example.com", 80)));
+  EXPECT_FALSE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_SEND_TO, "www.google.com", 8800)));
+  EXPECT_FALSE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_SEND_TO, "wwwexample.com", 8800)));
 
   CHECK(data.Parse("udp-send-to:*.ExAmPlE.cOm:8800"));
-  EXPECT_TRUE(data.Match(
-        SocketPermissionData::UDP_SEND_TO, "www.example.com", 8800));
-  EXPECT_TRUE(data.Match(
-        SocketPermissionData::UDP_SEND_TO, "smtp.example.com", 8800));
-  EXPECT_TRUE(data.Match(
-        SocketPermissionData::UDP_SEND_TO, "SMTP.example.com", 8800));
-  EXPECT_FALSE(data.Match(
-        SocketPermissionData::TCP_CONNECT, "www.example.com", 80));
-  EXPECT_FALSE(data.Match(
-        SocketPermissionData::UDP_SEND_TO, "www.google.com", 8800));
+  EXPECT_TRUE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_SEND_TO, "www.example.com", 8800)));
+  EXPECT_TRUE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_SEND_TO, "smtp.example.com", 8800)));
+  EXPECT_TRUE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_SEND_TO, "SMTP.example.com", 8800)));
+  EXPECT_FALSE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::TCP_CONNECT, "www.example.com", 80)));
+  EXPECT_FALSE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_SEND_TO, "www.google.com", 8800)));
 
   CHECK(data.Parse("udp-bind::8800"));
-  EXPECT_TRUE(data.Match(
-        SocketPermissionData::UDP_BIND, "127.0.0.1", 8800));
-  EXPECT_FALSE(data.Match(
-        SocketPermissionData::UDP_BIND, "127.0.0.1", 8888));
-  EXPECT_FALSE(data.Match(
-        SocketPermissionData::TCP_CONNECT, "www.example.com", 80));
-  EXPECT_FALSE(data.Match(
-        SocketPermissionData::UDP_SEND_TO, "www.google.com", 8800));
+  EXPECT_TRUE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_BIND, "127.0.0.1", 8800)));
+  EXPECT_FALSE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_BIND, "127.0.0.1", 8888)));
+  EXPECT_FALSE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::TCP_CONNECT, "www.example.com", 80)));
+  EXPECT_FALSE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::UDP_SEND_TO, "www.google.com", 8800)));
 
   // Do not wildcard part of ip address.
   CHECK(data.Parse("tcp-connect:*.168.0.1:8800"));
-  EXPECT_FALSE(data.Match(
-        SocketPermissionData::TCP_CONNECT, "192.168.0.1", 8800));
+  EXPECT_FALSE(data.Match(SocketPermissionRequest(
+        SocketPermissionRequest::TCP_CONNECT, "192.168.0.1", 8800)));
 }
 
 TEST(SocketPermissionTest, IPC) {
   const APIPermissionInfo* permission_info =
     PermissionsInfo::GetInstance()->GetByID(APIPermission::kSocket);
 
   {
     IPC::Message m;
 
     scoped_ptr<APIPermission> permission1(
@@ skipping 52 matching lines @@
   base::Value* vtmp = NULL;
   permission1->ToValue(&vtmp);
   CHECK(vtmp);
   CHECK(permission2->FromValue(vtmp));
   EXPECT_TRUE(permission1->Equal(permission2.get()));
 
   delete vtmp;
 }
 
 }  // namespace extensions