chrome/utility/chrome_content_utility_client.h

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_UTILITY_CHROME_CONTENT_UTILITY_CLIENT_H_
   6 #define CHROME_UTILITY_CHROME_CONTENT_UTILITY_CLIENT_H_
   7
   8 #include <set>
   9 #include <string>
  10 #include <vector>
  11
  12 #include "base/compiler_specific.h"
  13 #include "base/memory/scoped_vector.h"
  14 #include "content/public/utility/content_utility_client.h"
  15 #include "ipc/ipc_platform_file.h"
  16
  17 namespace base {
  18 class FilePath;
  19 struct FileDescriptor;
  20 }
  21
  22 class UtilityMessageHandler;
  23
  24 class ChromeContentUtilityClient : public content::ContentUtilityClient {
  25  public:
  26   ChromeContentUtilityClient();
  27   ~ChromeContentUtilityClient() override;
  28
  29   void UtilityThreadStarted() override;
  30   bool OnMessageReceived(const IPC::Message& message) override;
  31   void RegisterMojoServices(content::ServiceRegistry* registry) override;
  32
  33   static void PreSandboxStartup();
  34
  35   // Shared with extensions::ExtensionsHandler.
  36   static SkBitmap DecodeImage(const std::vector<unsigned char>& encoded_data,
  37                               bool shrink_to_fit);
  38   static void DecodeImageAndSend(const std::vector<unsigned char>& encoded_data,
  39                                  bool shrink_to_fit,
  40                                  int request_id);
  41
  42   static void set_max_ipc_message_size_for_test(int64_t max_message_size) {
  43     max_ipc_message_size_ = max_message_size;
  44   }
  45
  46  private:
  47   // IPC message handlers.
  48   void OnUnpackWebResource(const std::string& resource_data);
  49   void OnDecodeImage(const std::vector<unsigned char>& encoded_data,
  50                      bool shrink_to_fit,
  51                      int request_id);
  52 #if defined(OS_CHROMEOS)
  53   void OnRobustJPEGDecodeImage(const std::vector<unsigned char>& encoded_data,
  54                                int request_id);
  55
  56   void OnCreateZipFile(const base::FilePath& src_dir,
  57                        const std::vector<base::FilePath>& src_relative_paths,
  58                        const base::FileDescriptor& dest_fd);
  59 #endif  // defined(OS_CHROMEOS)
  60
  61   void OnPatchFileBsdiff(const base::FilePath& input_file,
  62                          const base::FilePath& patch_file,
  63                          const base::FilePath& output_file);
  64   void OnPatchFileCourgette(const base::FilePath& input_file,
  65                             const base::FilePath& patch_file,
  66                             const base::FilePath& output_file);
  67   void OnStartupPing();
  68 #if defined(FULL_SAFE_BROWSING)
  69   void OnAnalyzeZipFileForDownloadProtection(
  70       const IPC::PlatformFileForTransit& zip_file,
  71       const IPC::PlatformFileForTransit& temp_file);
  72 #if defined(OS_MACOSX)
  73   void OnAnalyzeDmgFileForDownloadProtection(
  74       const IPC::PlatformFileForTransit& dmg_file);
  75 #endif
  76 #endif
  77 #if defined(ENABLE_EXTENSIONS)
  78   void OnParseMediaMetadata(const std::string& mime_type,
  79                             int64 total_size,
  80                             bool get_attached_images);
  81 #endif
  82
  83   typedef ScopedVector<UtilityMessageHandler> Handlers;
  84   Handlers handlers_;
  85
  86   // Flag to enable whitelisting.
  87   bool filter_messages_;
  88   // A list of message_ids to filter.
  89   std::set<int> message_id_whitelist_;
  90   // Maximum IPC msg size (default to kMaximumMessageSize; override for testing)
  91   static int64_t max_ipc_message_size_;
  92
  93   DISALLOW_COPY_AND_ASSIGN(ChromeContentUtilityClient);
  94 };
  95
  96 #endif  // CHROME_UTILITY_CHROME_CONTENT_UTILITY_CLIENT_H_