Add support for the quick-enable-cf command to the installer. This encompase...

chrome/installer/util/installation_validator.h

 // Copyright (c) 2011 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_INSTALLER_UTIL_INSTALLATION_VALIDATOR_H_
 #define CHROME_INSTALLER_UTIL_INSTALLATION_VALIDATOR_H_
 #pragma once
 
+#include <map>
 #include <string>
 #include <utility>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/compiler_specific.h"
 #include "base/ref_counted.h"
 #include "base/scoped_ptr.h"
 #include "chrome/installer/util/browser_distribution.h"
 #include "chrome/installer/util/channel_info.h"
 
 class FilePath;
 class Version;
 
 namespace installer {
 
 class InstallationState;
+class ProductCommand;
 class ProductState;
 
 // A class that validates the state of an installation.  Violations are logged
 // via LOG(ERROR).
 class InstallationValidator {
  public:
   class ProductBits {
    public:
     // Bits that form the components of an installation type.
     enum {
@@ skipping 33 matching lines @@
       const InstallationState& machine_state,
       bool system_level,
       InstallationType* type);
 
   // Validates the machine's current installation at user or system level,
   // returning true and populating |type| if valid.
   static bool ValidateInstallationType(bool system_level,
                                        InstallationType* type);
 
  protected:
+  struct ProductContext;
   typedef std::vector<std::pair<std::string, bool> > SwitchExpectations;
+  typedef void (*CommandValidatorFn)(const ProductContext& ctx,
+                                     const ProductCommand& command,
+                                     bool* is_valid);
+  typedef std::map<std::wstring, CommandValidatorFn> CommandExpectations;
 
   // An interface to product-specific validation rules.
   class ProductRules {
    public:
     virtual ~ProductRules() { }
     virtual BrowserDistribution::Type distribution_type() const = 0;
     virtual void AddProductSwitchExpectations(
         const InstallationState& machine_state,
         bool system_install,
         const ProductState& product_state,
@@ skipping 15 matching lines @@
   class ChromeFrameRules : public ProductRules {
    public:
     virtual BrowserDistribution::Type distribution_type() const OVERRIDE;
     virtual void AddProductSwitchExpectations(
         const InstallationState& machine_state,
         bool system_install,
         const ProductState& product_state,
         SwitchExpectations* expectations) const OVERRIDE;
   };
 
+  // Validation rules for the multi-install Chrome binaries.
+  class ChromeBinariesRules : public ProductRules {
+   public:
+    virtual BrowserDistribution::Type distribution_type() const OVERRIDE;
+    virtual void AddProductSwitchExpectations(
+        const InstallationState& machine_state,
+        bool system_install,
+        const ProductState& product_state,
+        SwitchExpectations* expectations) const OVERRIDE;
+  };
+
   struct ProductContext {
     const InstallationState& machine_state;
     bool system_install;
     BrowserDistribution* dist;
     const ProductState& state;
     const ProductRules& rules;
   };
 
+  static void ValidateQuickEnableCfCommand(const ProductContext& ctx,
+                                           const ProductCommand& command,
+                                           bool* is_valid);
+  static void ValidateProductCommandExpectations(
+      const ProductContext& ctx,
+      const CommandExpectations& expectations,
+      bool* is_valid);
+  static void ValidateBinariesCommands(const ProductContext& ctx,
+                                       bool* is_valid);
   static void ValidateBinaries(const InstallationState& machine_state,
                                bool system_install,
                                const ProductState& binaries_state,
                                bool* is_valid);
   static void ValidateSetupPath(const ProductContext& ctx,
                                 const FilePath& setup_exe,
                                 const char* purpose,
                                 bool* is_valid);
   static void ValidateCommandExpectations(const ProductContext& ctx,
                                           const CommandLine& command,
                                           const SwitchExpectations& expected,
                                           const char* source,
                                           bool* is_valid);
   static void ValidateUninstallCommand(const ProductContext& ctx,
                                        const CommandLine& command,
                                        const char* source,
                                        bool* is_valid);
   static void ValidateRenameCommand(const ProductContext& ctx,
                                     bool* is_valid);
   static void ValidateOldVersionValues(const ProductContext& ctx,
                                        bool* is_valid);
   static void ValidateMultiInstallProduct(const ProductContext& ctx,
                                           bool* is_valid);
+  static void ValidateProductCommands(const ProductContext& ctx,
+                                      bool* is_valid);
   static void ValidateProduct(const InstallationState& machine_state,
                               bool system_install,
                               const ProductState& product_state,
                               const ProductRules& rules,
                               bool* is_valid);
 
   // A collection of all valid installation types.
   static const InstallationType kInstallationTypes[];
 
  private:
   DISALLOW_IMPLICIT_CONSTRUCTORS(InstallationValidator);
 };
 
 }  // namespace installer
 
 #endif  // CHROME_INSTALLER_UTIL_INSTALLATION_VALIDATOR_H_