Home | History | Annotate | Download | only in utility 
      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   virtual ~ChromeContentUtilityClient();
     28 
     29   virtual void UtilityThreadStarted() OVERRIDE;
     30   virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE;
     31 
     32   static void PreSandboxStartup();
     33 
     34   // Shared with extensions::ExtensionsHandler.
     35   static SkBitmap DecodeImage(const std::vector<unsigned char>& encoded_data,
     36                               bool shrink_to_fit);
     37   static void DecodeImageAndSend(const std::vector<unsigned char>& encoded_data,
     38                                  bool shrink_to_fit);
     39 
     40   static void set_max_ipc_message_size_for_test(int64_t max_message_size) {
     41     max_ipc_message_size_ = max_message_size;
     42   }
     43 
     44  private:
     45   // IPC message handlers.
     46   void OnUnpackWebResource(const std::string& resource_data);
     47   void OnDecodeImage(const std::vector<unsigned char>& encoded_data,
     48                      bool shrink_to_fit);
     49   void OnRobustJPEGDecodeImage(
     50       const std::vector<unsigned char>& encoded_data);
     51 
     52 #if defined(OS_CHROMEOS)
     53   void OnCreateZipFile(const base::FilePath& src_dir,
     54                        const std::vector<base::FilePath>& src_relative_paths,
     55                        const base::FileDescriptor& dest_fd);
     56 #endif  // defined(OS_CHROMEOS)
     57 
     58   void OnPatchFileBsdiff(const base::FilePath& input_file,
     59                          const base::FilePath& patch_file,
     60                          const base::FilePath& output_file);
     61   void OnPatchFileCourgette(const base::FilePath& input_file,
     62                             const base::FilePath& patch_file,
     63                             const base::FilePath& output_file);
     64   void OnStartupPing();
     65 #if defined(FULL_SAFE_BROWSING)
     66   void OnAnalyzeZipFileForDownloadProtection(
     67       const IPC::PlatformFileForTransit& zip_file);
     68 #endif
     69 #if defined(ENABLE_EXTENSIONS)
     70   void OnParseMediaMetadata(const std::string& mime_type,
     71                             int64 total_size,
     72                             bool get_attached_images);
     73 #endif
     74 
     75   typedef ScopedVector<UtilityMessageHandler> Handlers;
     76   Handlers handlers_;
     77 
     78   // Flag to enable whitelisting.
     79   bool filter_messages_;
     80   // A list of message_ids to filter.
     81   std::set<int> message_id_whitelist_;
     82   // Maximum IPC msg size (default to kMaximumMessageSize; override for testing)
     83   static int64_t max_ipc_message_size_;
     84 
     85   DISALLOW_COPY_AND_ASSIGN(ChromeContentUtilityClient);
     86 };
     87 
     88 #endif  // CHROME_UTILITY_CHROME_CONTENT_UTILITY_CLIENT_H_
     89