Include attributes of zipped binaries in ClientDownloadRequests.

chrome/browser/safe_browsing/sandboxed_zip_analyzer.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/safe_browsing/sandboxed_zip_analyzer.h"
 
 #include "base/bind.h"
 #include "base/command_line.h"
+#include "base/files/file_util.h"
 #include "base/location.h"
 #include "base/logging.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "chrome/common/chrome_utility_messages.h"
-#include "chrome/common/safe_browsing/zip_analyzer.h"
+#include "chrome/common/safe_browsing/zip_analyzer_results.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/child_process_data.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/common/content_switches.h"
 #include "ipc/ipc_message_macros.h"
 #include "ipc/ipc_platform_file.h"
 
 using content::BrowserThread;
 
 namespace safe_browsing {
@@ skipping 14 matching lines @@
           FROM_HERE,
           base::Bind(&SandboxedZipAnalyzer::AnalyzeInSandbox, this),
           base::SequencedWorkerPool::CONTINUE_ON_SHUTDOWN)) {
     NOTREACHED();
   }
 }
 
 SandboxedZipAnalyzer::~SandboxedZipAnalyzer() {
   // If we're using UtilityProcessHost, we may not be destroyed on
   // the UI or IO thread.
+  CloseTemporaryFile();
 }
 
 void SandboxedZipAnalyzer::AnalyzeInSandbox() {
+  // This zip file will be closed on the IO thread once it has been handed
+  // off to the child process.
   zip_file_.Initialize(zip_file_name_,
                        base::File::FLAG_OPEN | base::File::FLAG_READ);
   if (!zip_file_.IsValid()) {
     DVLOG(1) << "Could not open zip file: " << zip_file_name_.value();
     if (!BrowserThread::PostTask(
             BrowserThread::IO, FROM_HERE,
             base::Bind(&SandboxedZipAnalyzer::OnAnalyzeZipFileFinished, this,
                        zip_analyzer::Results()))) {
       NOTREACHED();
     }
     return;
   }
 
+  // This temp file will be closed in the blocking pool when results from the
+  // analyzer return.
+  base::FilePath temp_path;
+  if (base::CreateTemporaryFile(&temp_path)) {
+    temp_file_.Initialize(temp_path, (base::File::FLAG_CREATE_ALWAYS |
+                                      base::File::FLAG_READ |
+                                      base::File::FLAG_WRITE |
+                                      base::File::FLAG_TEMPORARY |
+                                      base::File::FLAG_DELETE_ON_CLOSE));
+  }
+  DVLOG_IF(1, !temp_file_.IsValid())
+      << "Could not open temporary output file: " << temp_path.value();
+
   BrowserThread::PostTask(
       BrowserThread::IO, FROM_HERE,
       base::Bind(&SandboxedZipAnalyzer::StartProcessOnIOThread, this));
-  // The file will be closed on the IO thread once it has been handed
-  // off to the child process.
 }
 
 bool SandboxedZipAnalyzer::OnMessageReceived(const IPC::Message& message) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(SandboxedZipAnalyzer, message)
     IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_ProcessStarted,
                         OnUtilityProcessStarted)
     IPC_MESSAGE_HANDLER(
         ChromeUtilityHostMsg_AnalyzeZipFileForDownloadProtection_Finished,
         OnAnalyzeZipFileFinished)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
   return handled;
 }
 
 void SandboxedZipAnalyzer::OnAnalyzeZipFileFinished(
     const zip_analyzer::Results& results) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   if (callback_called_)
     return;
   BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
                           base::Bind(callback_, results));
   callback_called_ = true;
+  CloseTemporaryFile();
 }
 
 void SandboxedZipAnalyzer::StartProcessOnIOThread() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   utility_process_host_ = content::UtilityProcessHost::Create(
       this,
       BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO).get())
       ->AsWeakPtr();
   utility_process_host_->Send(new ChromeUtilityMsg_StartupPing);
   // Wait for the startup notification before sending the main IPC to the
   // utility process, so that we can dup the file handle.
 }
 
+void SandboxedZipAnalyzer::CloseTemporaryFile() {

[mattm, 2015/03/23 01:53:40]

nit: I'd move this so that StartProcessOnIOThread and OnUtilityProcessStarted
stay next to each other

[grt (UTC plus 2), 2015/03/23 13:29:16]

Done. I also made the order in the .h match that here in the .cc.

+  if (!temp_file_.IsValid())
+    return;
+  // Close the temporary file in the blocking pool since doing so will delete
+  // the file.
+  if (!BrowserThread::GetBlockingPool()->PostWorkerTaskWithShutdownBehavior(
+          FROM_HERE, base::Bind(&base::File::Close,
+                                base::Owned(new base::File(temp_file_.Pass()))),
+          base::SequencedWorkerPool::CONTINUE_ON_SHUTDOWN)) {
+    NOTREACHED();
+  }
+}
+
 void SandboxedZipAnalyzer::OnUtilityProcessStarted() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   base::ProcessHandle utility_process =
       content::RenderProcessHost::run_renderer_in_process() ?
           base::GetCurrentProcessHandle() :
           utility_process_host_->GetData().handle;
 
   if (utility_process == base::kNullProcessHandle) {
     DLOG(ERROR) << "Child process handle is null";
   }
   utility_process_host_->Send(
       new ChromeUtilityMsg_AnalyzeZipFileForDownloadProtection(
-          IPC::TakeFileHandleForProcess(zip_file_.Pass(), utility_process)));
+          IPC::TakeFileHandleForProcess(zip_file_.Pass(), utility_process),
+          IPC::GetFileHandleForProcess(temp_file_.GetPlatformFile(),
+                                       utility_process,
+                                       false /* !close_source_handle */)));
 }
 
 }  // namespace safe_browsing