self_reg_work_item.h revision 5821806d5e7f356e8fa4b058a389a808ea183019 
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.
14class SelfRegWorkItem : public WorkItem {
15 public:
16  virtual ~SelfRegWorkItem();
17
18  virtual bool Do();
19  virtual void Rollback();
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__
54