| // 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. |
| |
| #ifndef CHROME_SERVICES_FILE_UTIL_PUBLIC_CPP_SANDBOXED_ZIP_ANALYZER_H_ |
| #define CHROME_SERVICES_FILE_UTIL_PUBLIC_CPP_SANDBOXED_ZIP_ANALYZER_H_ |
| |
| #include "base/callback.h" |
| #include "base/files/file.h" |
| #include "base/files/file_path.h" |
| #include "base/macros.h" |
| #include "base/memory/ref_counted_delete_on_sequence.h" |
| #include "chrome/services/file_util/public/mojom/file_util_service.mojom.h" |
| #include "chrome/services/file_util/public/mojom/safe_archive_analyzer.mojom.h" |
| #include "mojo/public/cpp/bindings/pending_remote.h" |
| #include "mojo/public/cpp/bindings/remote.h" |
| |
| namespace safe_browsing { |
| struct ArchiveAnalyzerResults; |
| } |
| |
| // This class is used to analyze zip files in a sandboxed utility process for |
| // file download protection. This class lives on the UI thread, which is where |
| // the result callback will be invoked. |
| class SandboxedZipAnalyzer |
| : public base::RefCountedDeleteOnSequence<SandboxedZipAnalyzer> { |
| public: |
| using ResultCallback = |
| base::OnceCallback<void(const safe_browsing::ArchiveAnalyzerResults&)>; |
| |
| SandboxedZipAnalyzer( |
| const base::FilePath& zip_file, |
| ResultCallback callback, |
| mojo::PendingRemote<chrome::mojom::FileUtilService> service); |
| |
| // Starts the analysis. Must be called on the UI thread. |
| void Start(); |
| |
| private: |
| friend class base::RefCountedDeleteOnSequence<SandboxedZipAnalyzer>; |
| friend class base::DeleteHelper<SandboxedZipAnalyzer>; |
| |
| ~SandboxedZipAnalyzer(); |
| |
| // Prepare the file for analysis. |
| void PrepareFileToAnalyze(); |
| |
| // If file preparation failed, analysis has failed: report failure. |
| void ReportFileFailure(); |
| |
| // Starts the utility process and sends it a file analyze request. |
| void AnalyzeFile(base::File file, base::File temp); |
| |
| // The response containing the file analyze results. |
| void AnalyzeFileDone(const safe_browsing::ArchiveAnalyzerResults& results); |
| |
| // The file path of the file to analyze. |
| const base::FilePath file_path_; |
| |
| // Callback invoked on the UI thread with the file analyze results. |
| ResultCallback callback_; |
| |
| // Remote interfaces to the file util service. Only used from the UI thread. |
| mojo::Remote<chrome::mojom::FileUtilService> service_; |
| mojo::Remote<chrome::mojom::SafeArchiveAnalyzer> remote_analyzer_; |
| |
| DISALLOW_COPY_AND_ASSIGN(SandboxedZipAnalyzer); |
| }; |
| |
| #endif // CHROME_SERVICES_FILE_UTIL_PUBLIC_CPP_SANDBOXED_ZIP_ANALYZER_H_ |