Make sure that a given app/extension requests only its own resources.

chrome/browser/extensions/api/socket/tcp_socket_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 "chrome/browser/extensions/api/socket/tcp_socket.h"
 
 #include "base/memory/scoped_ptr.h"
 #include "chrome/browser/extensions/api/api_resource_event_notifier.h"
 #include "net/base/address_list.h"
 #include "net/base/completion_callback.h"
@@ skipping 25 matching lines @@
   virtual bool IsConnected() const OVERRIDE {
     return true;
   }
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockTCPSocket);
 };
 
 class MockApiResourceEventNotifier : public ApiResourceEventNotifier {
  public:
-  MockApiResourceEventNotifier() : ApiResourceEventNotifier(NULL, NULL,
-                                                            std::string(),
-                                                            0, GURL()) {}
+  MockApiResourceEventNotifier(const std::string& owner_extension_id)
+      : ApiResourceEventNotifier(NULL, NULL,
+                                 owner_extension_id,
+                                 0, GURL()) {}
 
   MOCK_METHOD2(OnReadComplete, void(int result_code,
                                     const std::string& message));
   MOCK_METHOD1(OnWriteComplete, void(int result_code));
 
  protected:
   virtual ~MockApiResourceEventNotifier() {}
 };
 
 class CompleteHandler {
  public:
   CompleteHandler() {}
   MOCK_METHOD1(OnComplete, void(int result_code));
   MOCK_METHOD2(OnReadComplete, void(int result_code,
       scoped_refptr<net::IOBuffer> io_buffer));
  private:
   DISALLOW_COPY_AND_ASSIGN(CompleteHandler);
 };
 
+const std::string FAKE_ID = "abcdefghijklmnopqrst";
 
 TEST(SocketTest, TestTCPSocketRead) {
   net::AddressList address_list;
   MockTCPSocket* tcp_client_socket = new MockTCPSocket(address_list);
-  ApiResourceEventNotifier* notifier = new MockApiResourceEventNotifier();
+  ApiResourceEventNotifier* notifier = new MockApiResourceEventNotifier(
+      FAKE_ID);
   CompleteHandler handler;
 
   scoped_ptr<TCPSocket> socket(TCPSocket::CreateSocketForTesting(
-      tcp_client_socket, notifier));
+      tcp_client_socket, FAKE_ID, notifier));
 
   EXPECT_CALL(*tcp_client_socket, Read(_, _, _))
       .Times(1);
   EXPECT_CALL(handler, OnReadComplete(_, _))
       .Times(1);
 
   const int count = 512;
   socket->Read(count, base::Bind(&CompleteHandler::OnReadComplete,
         base::Unretained(&handler)));
 }
 
 TEST(SocketTest, TestTCPSocketWrite) {
   net::AddressList address_list;
   MockTCPSocket* tcp_client_socket = new MockTCPSocket(address_list);
-  ApiResourceEventNotifier* notifier = new MockApiResourceEventNotifier();
+  ApiResourceEventNotifier* notifier = new MockApiResourceEventNotifier(
+      FAKE_ID);
   CompleteHandler handler;
 
   scoped_ptr<TCPSocket> socket(TCPSocket::CreateSocketForTesting(
-      tcp_client_socket, notifier));
+      tcp_client_socket, FAKE_ID, notifier));
 
   net::CompletionCallback callback;
   EXPECT_CALL(*tcp_client_socket, Write(_, _, _))
       .Times(2)
       .WillRepeatedly(testing::DoAll(SaveArg<2>(&callback),
                                      Return(128)));
   EXPECT_CALL(handler, OnComplete(_))
       .Times(1);
 
   scoped_refptr<net::IOBufferWithSize> io_buffer(
       new net::IOBufferWithSize(256));
   socket->Write(io_buffer.get(), io_buffer->size(),
       base::Bind(&CompleteHandler::OnComplete, base::Unretained(&handler)));
 }
 
 TEST(SocketTest, TestTCPSocketBlockedWrite) {
   net::AddressList address_list;
   MockTCPSocket* tcp_client_socket = new MockTCPSocket(address_list);
-  MockApiResourceEventNotifier* notifier = new MockApiResourceEventNotifier();
+  MockApiResourceEventNotifier* notifier = new MockApiResourceEventNotifier(
+      FAKE_ID);
   CompleteHandler handler;
 
   scoped_ptr<TCPSocket> socket(TCPSocket::CreateSocketForTesting(
-      tcp_client_socket, notifier));
+      tcp_client_socket, FAKE_ID, notifier));
 
   net::CompletionCallback callback;
   EXPECT_CALL(*tcp_client_socket, Write(_, _, _))
       .Times(2)
       .WillRepeatedly(testing::DoAll(SaveArg<2>(&callback),
                                      Return(net::ERR_IO_PENDING)));
   scoped_refptr<net::IOBufferWithSize> io_buffer(new net::IOBufferWithSize(42));
   socket->Write(io_buffer.get(), io_buffer->size(),
       base::Bind(&CompleteHandler::OnComplete, base::Unretained(&handler)));
 
   // Good. Original call came back unable to complete. Now pretend the socket
   // finished, and confirm that we passed the error back.
   EXPECT_CALL(handler, OnComplete(42))
       .Times(1);
   callback.Run(40);
   callback.Run(2);
 }
 
 TEST(SocketTest, TestTCPSocketBlockedWriteReentry) {
   net::AddressList address_list;
   MockTCPSocket* tcp_client_socket = new MockTCPSocket(address_list);
-  MockApiResourceEventNotifier* notifier = new MockApiResourceEventNotifier();
+  MockApiResourceEventNotifier* notifier = new MockApiResourceEventNotifier(
+      FAKE_ID);
   CompleteHandler handlers[5];
 
   scoped_ptr<TCPSocket> socket(TCPSocket::CreateSocketForTesting(
-      tcp_client_socket, notifier));
+      tcp_client_socket, FAKE_ID, notifier));
 
   net::CompletionCallback callback;
   EXPECT_CALL(*tcp_client_socket, Write(_, _, _))
       .Times(5)
       .WillRepeatedly(testing::DoAll(SaveArg<2>(&callback),
                                      Return(net::ERR_IO_PENDING)));
   scoped_refptr<net::IOBufferWithSize> io_buffers[5];
   int i;
   for (i = 0; i < 5; i++) {
     io_buffers[i] = new net::IOBufferWithSize(128 + i * 50);
     scoped_refptr<net::IOBufferWithSize> io_buffer1(
         new net::IOBufferWithSize(42));
     socket->Write(io_buffers[i].get(), io_buffers[i]->size(),
         base::Bind(&CompleteHandler::OnComplete,
             base::Unretained(&handlers[i])));
 
     EXPECT_CALL(handlers[i], OnComplete(io_buffers[i]->size()))
         .Times(1);
   }
 
   for (i = 0; i < 5; i++) {
     callback.Run(128 + i * 50);
   }
 }
 
 TEST(SocketTest, TestTCPSocketSetNoDelay) {
   net::AddressList address_list;
   MockTCPSocket* tcp_client_socket = new MockTCPSocket(address_list);
-  MockApiResourceEventNotifier* notifier = new MockApiResourceEventNotifier();
+  MockApiResourceEventNotifier* notifier = new MockApiResourceEventNotifier(
+      FAKE_ID);
 
   scoped_ptr<TCPSocket> socket(TCPSocket::CreateSocketForTesting(
-      tcp_client_socket, notifier));
+      tcp_client_socket, FAKE_ID, notifier));
 
   bool no_delay = false;
   EXPECT_CALL(*tcp_client_socket, SetNoDelay(_))
       .WillOnce(testing::DoAll(SaveArg<0>(&no_delay), Return(true)));
   int result = socket->SetNoDelay(true);
   EXPECT_TRUE(result);
   EXPECT_TRUE(no_delay);
 
   EXPECT_CALL(*tcp_client_socket, SetNoDelay(_))
       .WillOnce(testing::DoAll(SaveArg<0>(&no_delay), Return(false)));
 
   result = socket->SetNoDelay(false);
   EXPECT_FALSE(result);
   EXPECT_FALSE(no_delay);
 }
 
 TEST(SocketTest, TestTCPSocketSetKeepAlive) {
   net::AddressList address_list;
   MockTCPSocket* tcp_client_socket = new MockTCPSocket(address_list);
-  MockApiResourceEventNotifier* notifier = new MockApiResourceEventNotifier();
+  MockApiResourceEventNotifier* notifier = new MockApiResourceEventNotifier(
+      FAKE_ID);
 
   scoped_ptr<TCPSocket> socket(TCPSocket::CreateSocketForTesting(
-      tcp_client_socket, notifier));
+      tcp_client_socket, FAKE_ID, notifier));
 
   bool enable = false;
   int delay = 0;
   EXPECT_CALL(*tcp_client_socket, SetKeepAlive(_, _))
       .WillOnce(testing::DoAll(SaveArg<0>(&enable),
                                SaveArg<1>(&delay),
                                Return(true)));
   int result = socket->SetKeepAlive(true, 4500);
   EXPECT_TRUE(result);
   EXPECT_TRUE(enable);
   EXPECT_EQ(4500, delay);
 
   EXPECT_CALL(*tcp_client_socket, SetKeepAlive(_, _))
       .WillOnce(testing::DoAll(SaveArg<0>(&enable),
                                SaveArg<1>(&delay),
                                Return(false)));
   result = socket->SetKeepAlive(false, 0);
   EXPECT_FALSE(result);
   EXPECT_FALSE(enable);
   EXPECT_EQ(0, delay);
 }
 
 }  // namespace extensions