Support socket endpoint permissions for AppsV2 Socket API.

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 extensions::SocketPermissionData;
+
+namespace {
+
+std::string Parse(const std::string& permission) {
+  SocketPermissionData data;
+  CHECK(data.Parse(permission)) << "Parse permission \"" << permission
+    << "\" failed.";
+  return data.GetAsString();
+}
+
+}  // namespace
+
+namespace extensions {
+
+class SocketPermissionTest : public testing::Test {
+};
+
+TEST(SocketPermissionTest, General) {
+  SocketPermissionData data1, data2;
+
+  CHECK(data1.Parse("tcp-connect"));
+  CHECK(data2.Parse("tcp-connect"));
+
+  EXPECT_TRUE(data1 == data2);
+  EXPECT_FALSE(data1 < data2);
+
+  CHECK(data1.Parse("tcp-connect"));
+  CHECK(data2.Parse("tcp-connect:www.example.com"));
+
+  EXPECT_FALSE(data1 == data2);
+  EXPECT_TRUE(data1 < data2);
+}
+
+TEST(SocketPermissionTest, Parse) {
+  SocketPermissionData data;
+
+  EXPECT_FALSE(data.Parse(""));
+  EXPECT_FALSE(data.Parse("*"));
+  EXPECT_FALSE(data.Parse("\00\00*"));
+  EXPECT_FALSE(data.Parse("\01*"));
+  EXPECT_FALSE(data.Parse("tcp-connect:www.example.com:-1"));
+  EXPECT_FALSE(data.Parse("tcp-connect:www.example.com:65536"));
+  EXPECT_FALSE(data.Parse("tcp-connect:::"));
+
+  EXPECT_EQ(Parse("tcp-connect"), "tcp-connect:*:*");
+  EXPECT_EQ(Parse("tcp-listen"), "tcp-listen:*:*");
+  EXPECT_EQ(Parse("udp-bind"), "udp-bind:*:*");
+  EXPECT_EQ(Parse("udp-send-to"), "udp-send-to:*:*");
+
+  EXPECT_EQ(Parse("  tcp-connect  "), "tcp-connect:*:*");
+  EXPECT_EQ(Parse("  tcp-listen  "), "tcp-listen:*:*");
+  EXPECT_EQ(Parse("  udp-bind  "), "udp-bind:*:*");
+  EXPECT_EQ(Parse("  udp-send-to  "), "udp-send-to:*:*");
+
+  EXPECT_EQ(Parse("tcp-connect:"), "tcp-connect:*:*");
+  EXPECT_EQ(Parse("tcp-listen:"), "tcp-listen:*:*");
+  EXPECT_EQ(Parse("udp-bind:"), "udp-bind:*:*");
+  EXPECT_EQ(Parse("udp-send-to:"), "udp-send-to:*:*");
+
+  EXPECT_EQ(Parse("tcp-connect::"), "tcp-connect:*:*");
+  EXPECT_EQ(Parse("tcp-listen::"), "tcp-listen:*:*");
+  EXPECT_EQ(Parse("udp-bind::"), "udp-bind:*:*");
+  EXPECT_EQ(Parse("udp-send-to::"), "udp-send-to:*:*");
+
+  EXPECT_EQ(Parse("tcp-connect:*"), "tcp-connect:*:*");
+  EXPECT_EQ(Parse("tcp-listen:*"), "tcp-listen:*:*");
+  EXPECT_EQ(Parse("udp-bind:*"), "udp-bind:*:*");
+  EXPECT_EQ(Parse("udp-send-to:*"), "udp-send-to:*:*");
+
+  EXPECT_EQ(Parse("tcp-connect:*:"), "tcp-connect:*:*");
+  EXPECT_EQ(Parse("tcp-listen:*:"), "tcp-listen:*:*");
+  EXPECT_EQ(Parse("udp-bind:*:"), "udp-bind:*:*");
+  EXPECT_EQ(Parse("udp-send-to:*:"), "udp-send-to:*:*");
+
+  EXPECT_EQ(Parse("tcp-connect::*"), "tcp-connect:*:*");
+  EXPECT_EQ(Parse("tcp-listen::*"), "tcp-listen:*:*");
+  EXPECT_EQ(Parse("udp-bind::*"), "udp-bind:*:*");
+  EXPECT_EQ(Parse("udp-send-to::*"), "udp-send-to:*:*");
+
+  EXPECT_EQ(Parse("tcp-connect:www.example.com"),
+      "tcp-connect:www.example.com:*");
+  EXPECT_EQ(Parse("tcp-listen:www.example.com"),
+      "tcp-listen:www.example.com:*");
+  EXPECT_EQ(Parse("udp-bind:www.example.com"),
+      "udp-bind:www.example.com:*");
+  EXPECT_EQ(Parse("udp-send-to:www.example.com"),
+      "udp-send-to:www.example.com:*");
+  EXPECT_EQ(Parse("udp-send-to:wWW.ExAmPlE.cOm"),
+      "udp-send-to:www.example.com:*");
+
+  EXPECT_EQ(Parse("tcp-connect : www.example.com "),
+      "tcp-connect:www.example.com:*");
+  EXPECT_EQ(Parse("tcp-listen : www.example.com "),
+      "tcp-listen:www.example.com:*");
+  EXPECT_EQ(Parse("udp-bind : www.example.com "),
+      "udp-bind:www.example.com:*");
+  EXPECT_EQ(Parse("udp-send-to : www.example.com "),
+      "udp-send-to:www.example.com:*");
+  EXPECT_EQ(Parse("udp-send-to : wWW.ExAmPlE.cOm"),
+      "udp-send-to:www.example.com:*");
+
+  EXPECT_EQ(Parse("tcp-connect:.example.com"),
+      "tcp-connect:*.example.com:*");
+  EXPECT_EQ(Parse("tcp-listen:.example.com"),
+      "tcp-listen:*.example.com:*");
+  EXPECT_EQ(Parse("udp-bind:.example.com"),
+      "udp-bind:*.example.com:*");
+  EXPECT_EQ(Parse("udp-send-to:.example.com"),
+      "udp-send-to:*.example.com:*");
+
+  EXPECT_EQ(Parse("tcp-connect:*.example.com"),
+      "tcp-connect:*.example.com:*");
+  EXPECT_EQ(Parse("tcp-listen:*.example.com"),
+      "tcp-listen:*.example.com:*");
+  EXPECT_EQ(Parse("udp-bind:*.example.com"),
+      "udp-bind:*.example.com:*");
+  EXPECT_EQ(Parse("udp-send-to:*.example.com"),
+      "udp-send-to:*.example.com:*");
+
+  EXPECT_EQ(Parse("tcp-connect::0"), "tcp-connect:*:*");
+  EXPECT_EQ(Parse("tcp-listen::0"), "tcp-listen:*:*");
+  EXPECT_EQ(Parse("udp-bind::0"), "udp-bind:*:*");
+  EXPECT_EQ(Parse("udp-send-to::0"), "udp-send-to:*:*");
+
+  EXPECT_EQ(Parse("tcp-connect::99"), "tcp-connect:*:99");
+  EXPECT_EQ(Parse("tcp-listen::99"), "tcp-listen:*:99");
+  EXPECT_EQ(Parse("udp-bind::99"), "udp-bind:*:99");
+  EXPECT_EQ(Parse("udp-send-to::99"), "udp-send-to:*:99");
+
+  EXPECT_EQ(Parse("tcp-connect : :  99 "), "tcp-connect:*:99");

[miket_OOO, 2012/08/07 23:02:47]

Consider that if you accept these kinds of weird strings, you're making a de
facto promise to _always_ support them, and you might be prevented in the future
from doing a refactoring or redesign that rejects them. If you want to be
extremely strict and reject constructs including arbitrary whitespace, or to
make the explicit wildcard mandatory rather than implied, then now is the time
to enforce that.

There's this expression from Larry Wall: "Be strict in what you emit and liberal
in what you accept." I agree entirely with that expression except for the
liberal part. Be strict in what you accept. Then everyone feeding inputs to your
code will be forced to get it right. You will have a smaller, simpler grammar,
making future testing easier, reducing the cost of static analysis, and
simplifying code paths.

[Peng, 2012/08/08 15:40:07]

A good advice. Done. 

+  EXPECT_EQ(Parse("tcp-listen : :  99 "), "tcp-listen:*:99");
+  EXPECT_EQ(Parse("udp-bind :  :  99  "), "udp-bind:*:99");
+  EXPECT_EQ(Parse("udp-send-to : :  99"), "udp-send-to:*:99");
+
+  EXPECT_EQ(Parse("tcp-connect:www.example.com"),
+      "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));
+
+  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));
+
+  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));
+
+  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));
+
+  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));
+}
+
+TEST(SocketPermissionTest, IPC) {
+  scoped_refptr<APIPermissionDetail> detail1;
+  scoped_refptr<APIPermissionDetail> detail2;
+
+  const APIPermission* permission =
+    PermissionsInfo::GetInstance()->GetByID(APIPermission::kSocket);
+
+  {
+    IPC::Message m;
+
+    detail1 = permission->CreateDetail();
+    detail2 = permission->CreateDetail();
+
+    detail1->Write(&m);
+    PickleIterator iter(m);
+    detail2->Read(&m, &iter);
+
+    EXPECT_TRUE(detail1->Equal(detail2));
+  }
+
+
+  {
+    IPC::Message m;
+
+    detail1 = permission->CreateDetail();
+    detail2 = permission->CreateDetail();
+
+    scoped_ptr<ListValue> value(new ListValue());
+    value->Append(Value::CreateStringValue("tcp-connect:*.example.com:80"));
+    value->Append(Value::CreateStringValue("udp-bind::8080"));
+    value->Append(Value::CreateStringValue("udp-send-to::8888"));
+    CHECK(detail1->FromValue(value.get()));
+
+    EXPECT_FALSE(detail1->Equal(detail2));
+
+    detail1->Write(&m);
+    PickleIterator iter(m);
+    detail2->Read(&m, &iter);
+    EXPECT_TRUE(detail1->Equal(detail2));
+  }
+}
+
+TEST(SocketPermissionTest, Value) {
+  scoped_refptr<APIPermissionDetail> detail1;
+  scoped_refptr<APIPermissionDetail> detail2;
+
+  const APIPermission* permission =
+    PermissionsInfo::GetInstance()->GetByID(APIPermission::kSocket);
+
+  detail1 = permission->CreateDetail();
+  detail2 = permission->CreateDetail();
+
+  scoped_ptr<ListValue> value(new ListValue());
+  value->Append(Value::CreateStringValue("tcp-connect:*.example.com:80"));
+  value->Append(Value::CreateStringValue("udp-bind::8080"));
+  value->Append(Value::CreateStringValue("udp-send-to::8888"));
+  CHECK(detail1->FromValue(value.get()));
+
+  EXPECT_FALSE(detail1->Equal(detail2));
+
+  base::Value* vtmp = NULL;
+  detail1->ToValue(&vtmp);
+  CHECK(vtmp);
+  CHECK(detail2->FromValue(vtmp));
+  EXPECT_TRUE(detail1->Equal(detail2));
+
+  delete vtmp;
+}

[miket_OOO, 2012/08/07 23:02:47]

Great! This is nice because it's a programmatic expression of how this class is
supposed to be used.

+
+}  // namespace extensions