Elevated install of recovery component (component update part)

chrome/browser/component_updater/component_unpacker.h

 // 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_BROWSER_COMPONENT_UPDATER_COMPONENT_UNPACKER_H_
 #define CHROME_BROWSER_COMPONENT_UPDATER_COMPONENT_UNPACKER_H_
 
 #include <string>
 #include <vector>
 
@@ skipping 73 matching lines @@
     kDeltaUnsupportedCommand,
     kDeltaOperationFailure,
     kDeltaPatchProcessFailure,
     kDeltaMissingExistingFile,
     kFingerprintWriteFailed,
   };
 
   typedef base::Callback<void(Error, int)> Callback;
 
   // Constructs an unpacker for a specific component unpacking operation.
-  // |pk_hash| is the expected/ public key SHA256 hash. |path| is the current
+  // |pk_hash| is the expected public key SHA256 hash. |path| is the current
   // location of the CRX.
   ComponentUnpacker(const std::vector<uint8>& pk_hash,
                     const base::FilePath& path,
                     const std::string& fingerprint,
                     ComponentInstaller* installer,
                     bool in_process,
                     scoped_refptr<base::SequencedTaskRunner> task_runner);
 
   // Begins the actual unpacking of the files. May invoke a patcher if the
   // package is a differential update. Calls |callback| with the result.
   void Unpack(const Callback& callback);
 
+  // Set the base directory where the unpack directory will be created.
+  // If this is not set, files will be unzipped to temporary dir.
+  void set_unpack_base_dir(const base::FilePath& path);
+
+  // Creates a ComponentUnpacker by loading CRX and related install information
+  // from |backup_path| that was previously saved by SaveInstallSource(). This
+  // is to retry failed component installation. |pk_hash| is the expected public
+  // key SHA256 hash.
+  static scoped_refptr<ComponentUnpacker> CreateFromBackup(
+      const base::FilePath& backup_path,
+      const std::vector<uint8>& pk_hash,
+      ComponentInstaller* installer,
+      scoped_refptr<base::SequencedTaskRunner> task_runner);
+
+
  private:
   friend class base::RefCountedThreadSafe<ComponentUnpacker>;
+  friend class ComponentUnpackerTest;
 
   virtual ~ComponentUnpacker();
 
   bool UnpackInternal();
 
   // The first step of unpacking is to verify the file. Returns false if an
   // error is encountered, the file is malformed, or the file is incorrectly
   // signed.
   bool Verify();
 
@@ skipping 12 matching lines @@
 
   // The fourth step is to install the unpacked component.
   void Install();
 
   // The final step is to do clean-up for things that can't be tidied as we go.
   // If there is an error at any step, the remaining steps are skipped and
   // and Finish is called.
   // Finish is responsible for calling the callback provided in Start().
   void Finish();
 
+  // Saves the original CRX file, along with |fingerprint_|, |pk_hash_| and
+  // |in_process_|, to |backup_path|. This is to re-create ComponentUnpacker for
+  // retrying failed component patching.
+  bool SaveInstallSource(const base::FilePath& backup_path) const;
+
+  // Creates the directory for unzip. |new_dir| will be the full path to the
+  // directory.
+  bool CreateUnpackDirectory(base::FilePath* new_dir);
+
+  base::FilePath unpack_base_dir_;
   std::vector<uint8> pk_hash_;
   base::FilePath path_;
   base::FilePath unpack_path_;
   base::FilePath unpack_diff_path_;
   bool is_delta_;
   std::string fingerprint_;
   scoped_refptr<ComponentPatcher> patcher_;
   ComponentInstaller* installer_;
   Callback callback_;
   const bool in_process_;
   Error error_;
   int extended_error_;
   scoped_refptr<base::SequencedTaskRunner> task_runner_;
 
   DISALLOW_COPY_AND_ASSIGN(ComponentUnpacker);
 };
 
 }  // namespace component_updater
 
 #endif  // CHROME_BROWSER_COMPONENT_UPDATER_COMPONENT_UNPACKER_H_