Parse the download server response.

chrome/browser/safe_browsing/download_protection_service.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 "chrome/browser/safe_browsing/download_protection_service.h"
 
 #include "base/bind.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/metrics/histogram.h"
 #include "base/stl_util.h"
@@ skipping 168 matching lines @@
   }
 
   // From the content::URLFetcherDelegate interface.
   virtual void OnURLFetchComplete(const content::URLFetcher* source) OVERRIDE {
     DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
     DCHECK_EQ(source, fetcher_.get());
     VLOG(2) << "Received a response for URL: "
             << info_.download_url_chain.back() << ": success="
             << source->GetStatus().is_success() << " response_code="
             << source->GetResponseCode();
-    DownloadCheckResultReason reason = REASON_MAX;
-    reason = REASON_SERVER_PING_FAILED;
+    DownloadCheckResultReason reason = REASON_SERVER_PING_FAILED;
+    DownloadCheckResult result = SAFE;
     if (source->GetStatus().is_success() &&
         RC_REQUEST_OK == source->GetResponseCode()) {
+      ClientDownloadResponse response;
       std::string data;
-      source->GetResponseAsString(&data);
-      if (data.size() > 0) {
-        // For now no matter what we'll always say the download is safe.
-        // TODO(noelutz): Parse the response body to see exactly what's going
-        // on.
+      bool got_data = source->GetResponseAsString(&data);
+      DCHECK(got_data);
+      if (!response.ParseFromString(data)) {
         reason = REASON_INVALID_RESPONSE_PROTO;
+      } else if (response.verdict() == ClientDownloadResponse::DANGEROUS) {
+        reason = REASON_DOWNLOAD_DANGEROUS;
+        result = DANGEROUS;
+      } else {
+        reason = REASON_DOWNLOAD_SAFE;
       }
     }
-
-    if (reason != REASON_MAX) {
-      RecordStats(reason);
-    }
     // We don't need the fetcher anymore.
     fetcher_.reset();
-    FinishRequest(SAFE);
+    RecordStats(reason);
+    FinishRequest(result);
   }
 
  private:
   friend struct BrowserThread::DeleteOnThread<BrowserThread::UI>;
   friend class DeleteTask<CheckClientDownloadRequest>;
 
   virtual ~CheckClientDownloadRequest() {
     DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   }
 
@@ skipping 183 matching lines @@
 void DownloadProtectionService::RequestFinished(
     CheckClientDownloadRequest* request) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   std::set<scoped_refptr<CheckClientDownloadRequest> >::iterator it =
       download_requests_.find(request);
   DCHECK(it != download_requests_.end());
   download_requests_.erase(*it);
 }
 
 }  // namespace safe_browsing