SafeBrowsing Opt-in Malware Reports: Collect remote ip:port pairs for requests that were not fetc...

chrome/browser/safe_browsing/malware_details_unittest.cc

 // Copyright (c) 2011 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 <algorithm>
 
 #include "base/pickle.h"
 #include "base/time.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/safe_browsing/malware_details.h"
@@ skipping 38 matching lines @@
 static const char* kLandingData = "<iframe src='http://www.malware.com'>";
 
 using safe_browsing::ClientMalwareReportRequest;
 
 namespace {
 
 void WriteHeaders(disk_cache::Entry* entry, const std::string headers) {
   net::HttpResponseInfo responseinfo;
   std::string raw_headers = net::HttpUtil::AssembleRawHeaders(
       headers.c_str(), headers.size());
+  responseinfo.socket_address = net::HostPortPair("1.2.3.4", 80);
   responseinfo.headers = new net::HttpResponseHeaders(raw_headers);
 
   Pickle pickle;
   responseinfo.Persist(&pickle, false, false);
 
   scoped_refptr<net::WrappedIOBuffer> buf(new net::WrappedIOBuffer(
       reinterpret_cast<const char*>(pickle.data())));
   int len = static_cast<int>(pickle.size());
 
   TestCompletionCallback cb;
@@ skipping 20 matching lines @@
   disk_cache::Entry* entry;
   int rv = cache->CreateEntry(key, &entry, &cb);
   rv = cb.GetResult(rv);
   if (rv != net::OK) {
     rv = cache->OpenEntry(key, &entry, &cb);
     ASSERT_EQ(net::OK, cb.GetResult(rv));
   }
 
   WriteHeaders(entry, headers);
   WriteData(entry, data);
-
   entry->Close();
 }
 
 void FillCache(net::URLRequestContext* context) {
   TestCompletionCallback cb;
   disk_cache::Backend* cache;
   int rv =
       context->http_transaction_factory()->GetCache()->GetBackend(&cache, &cb);
   ASSERT_EQ(net::OK, cb.GetResult(rv));
 
@@ skipping 159 matching lines @@
         EXPECT_EQ(expected->response().headers(i).value(),
                   resource->response().headers(i).value());
       }
 
       EXPECT_EQ(expected->response().body(), resource->response().body());
       EXPECT_EQ(expected->response().bodylength(),
                 resource->response().bodylength());
       EXPECT_EQ(expected->response().bodydigest(),
                 resource->response().bodydigest());
     }
+
+    // Verify IP:port pair
+    EXPECT_EQ(expected->response().remote_ip(),
+              resource->response().remote_ip());
   }
 
   BrowserThread ui_thread_;
   BrowserThread io_thread_;
   scoped_refptr<MockSafeBrowsingService> sb_service_;
 };
 
 // Tests creating a simple malware report.
 TEST_F(MalwareDetailsTest, MalwareSubResource) {
   // Start a load.
@@ skipping 238 matching lines @@
   pb_header->set_value("text/html");
   pb_header = pb_response->add_headers();
   pb_header->set_name("Content-Length");
   pb_header->set_value("1024");
   pb_header = pb_response->add_headers();
   pb_header->set_name("Set-Cookie");
   pb_header->set_value("");  // The cookie is dropped.
   pb_response->set_body(kLandingData);
   pb_response->set_bodylength(37);
   pb_response->set_bodydigest("9ca97475598a79bc1e8fc9bd6c72cd35");
+  pb_response->set_remote_ip("1.2.3.4:80");
 
   pb_resource = expected.add_resources();
   pb_resource->set_id(1);
   pb_resource->set_url(kMalwareURL);
   pb_response = pb_resource->mutable_response();
   pb_response->mutable_firstline()->set_code(200);
   pb_header = pb_response->add_headers();
   pb_header->set_name("Content-Type");
   pb_header->set_value("image/jpeg");
   pb_response->set_body(kMalwareData);
   pb_response->set_bodylength(10);
   pb_response->set_bodydigest("581373551c43d4cf33bfb3b26838ff95");
+  pb_response->set_remote_ip("1.2.3.4:80");
   expected.set_complete(true);
 
   VerifyResults(actual, expected);
 }
 
 // Tests the interaction with the HTTP cache (where the cache is empty).
 TEST_F(MalwareDetailsTest, HTTPCacheNoEntries) {
   controller().LoadURL(GURL(kLandingURL), GURL(), PageTransition::TYPED);
 
   SafeBrowsingService::UnsafeResource resource;
@@ skipping 26 matching lines @@
   ClientMalwareReportRequest::Resource* pb_resource = expected.add_resources();
   pb_resource->set_id(0);
   pb_resource->set_url(kLandingURL);
   pb_resource = expected.add_resources();
   pb_resource->set_id(1);
   pb_resource->set_url(kMalwareURL);
   expected.set_complete(true);
 
   VerifyResults(actual, expected);
 }