chrome/installer/util/self_reg_work_item.h

   1 // Copyright (c) 2006-2008 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_INSTALLER_UTIL_SELF_REG_WORK_ITEM_H__
   6 #define CHROME_INSTALLER_UTIL_SELF_REG_WORK_ITEM_H__
   7
   8 #include <windows.h>
   9 #include <string>
  10
  11 #include "chrome/installer/util/work_item.h"
  12
  13 // Registers or unregisters the DLL at the given path.
  14 class SelfRegWorkItem : public WorkItem {
  15  public:
  16   ~SelfRegWorkItem() override;
  17
  18   bool Do() override;
  19   void Rollback() override;
  20
  21  private:
  22   friend class WorkItem;
  23
  24   // Constructs a work item that will call upon a self-registering DLL to
  25   // register itself.
  26   // dll_path: The path to the DLL.
  27   // do_register: Whether this action is to register or unregister the DLL.
  28   // user_level_registration: If true, then the exports called
  29   //    "DllRegisterUserServer" and "DllUnregisterUserServer" will be called to
  30   //    register and unregister the DLL. If false, the default exports named
  31   //    "DllRegisterServer" and "DllUnregisterUserServer" will be used.
  32   SelfRegWorkItem(const std::wstring& dll_path, bool do_register,
  33                   bool user_level_registration);
  34
  35   // Examines the DLL at dll_path looking for either DllRegisterServer (if
  36   // do_register is true) or DllUnregisterServer (if do_register is false).
  37   // Returns true if the DLL exports the function and it a call to it
  38   // succeeds, false otherwise.
  39   bool RegisterDll(bool do_register);
  40
  41   // The path to the dll to be registered.
  42   std::wstring dll_path_;
  43
  44   // Whether this work item will register or unregister the dll. The rollback
  45   // action just inverts this parameter.
  46   bool do_register_;
  47
  48   // Whether to use alternate export names on the DLL that will perform
  49   // user level registration.
  50   bool user_level_registration_;
  51 };
  52
  53 #endif  // CHROME_INSTALLER_UTIL_SELF_REG_WORK_ITEM_H__