setup_native/source/win32/customactions/reg4msdoc/register.cxx

   1 /*************************************************************************
   2  *
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * Copyright 2008 by Sun Microsystems, Inc.
   6  *
   7  * OpenOffice.org - a multi-platform office productivity suite
   8  *
   9  * $RCSfile: register.cxx,v $
  10  * $Revision: 1.7 $
  11  *
  12  * This file is part of OpenOffice.org.
  13  *
  14  * OpenOffice.org is free software: you can redistribute it and/or modify
  15  * it under the terms of the GNU Lesser General Public License version 3
  16  * only, as published by the Free Software Foundation.
  17  *
  18  * OpenOffice.org is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU Lesser General Public License version 3 for more details
  22  * (a copy is included in the LICENSE file that accompanied this code).
  23  *
  24  * You should have received a copy of the GNU Lesser General Public License
  25  * version 3 along with OpenOffice.org.  If not, see
  26  * <http://www.openoffice.org/license.html>
  27  * for a copy of the LGPLv3 License.
  28  *
  29  ************************************************************************/
  30
  31 #include "register.hxx"
  32 #include "registryexception.hxx"
  33 #include "registrationcontextinformation.hxx"
  34 #include "userregistrar.hxx"
  35 #include "windowsregistry.hxx"
  36 #include "stringconverter.hxx"
  37 #include "msihelper.hxx"
  38
  39 #ifdef _MSC_VER
  40 #pragma warning(push, 1) /* disable warnings within system headers */
  41 #pragma warning(disable: 4917)
  42 #endif
  43 #include <shlobj.h>
  44 #ifdef _MSC_VER
  45 #pragma warning(pop)
  46 #endif
  47
  48
  49 #include <assert.h>
  50 #ifdef _MSC_VER
  51 #pragma warning(disable: 4350)
  52 #endif
  53
  54 typedef std::auto_ptr<Registrar> RegistrarPtr;
  55
  56 namespace /* private */
  57 {
  58     RegistrarPtr CreateRegistrar(bool InstallForAllUser, const RegistrationContextInformation& RegCtx)
  59     {
  60         RegistrarPtr RegPtr;
  61
  62         if (InstallForAllUser)
  63             RegPtr = RegistrarPtr(new Registrar(RegCtx));
  64         else
  65             RegPtr = RegistrarPtr(new UserRegistrar(RegCtx));
  66
  67         return RegPtr;
  68     }
  69 } // namespace private
  70
  71 bool query_preselect_registration_for_ms_application(MSIHANDLE handle, int Register)
  72 {
  73     bool preselect = false;
  74
  75     try
  76     {
  77         RegistrationContextInformation RegContext(handle, GetOfficeExecutablePath(handle));
  78         RegistrarPtr CurrentRegistrar = CreateRegistrar(IsAllUserInstallation(handle), RegContext);
  79
  80         if (Register & MSWORD)
  81             preselect = CurrentRegistrar->QueryPreselectMsWordRegistration();
  82         else if (Register & MSEXCEL)
  83             preselect = CurrentRegistrar->QueryPreselectMsExcelRegistration();
  84         else if (Register & MSPOWERPOINT)
  85             preselect = CurrentRegistrar->QueryPreselectMsPowerPointRegistration();
  86     }
  87     catch(RegistryException&)
  88     {
  89         assert(false);
  90     }
  91     return preselect;
  92 }
  93
  94 //-----------------------------------------
  95 // registers StarOffice for MS document
  96 // types and as default HTML editor if
  97 // specified
  98 //-----------------------------------------
  99
 100 void Register4MsDoc(MSIHANDLE handle, int Register)
 101 {
 102     try
 103     {
 104         RegistrationContextInformation RegContext(handle, GetOfficeExecutablePath(handle));
 105         RegistrarPtr CurrentRegistrar = CreateRegistrar(IsAllUserInstallation(handle), RegContext);
 106
 107         if ((Register & MSWORD))
 108             CurrentRegistrar->RegisterForMsWord();
 109
 110         if ((Register & MSEXCEL))
 111             CurrentRegistrar->RegisterForMsExcel();
 112
 113         if ((Register & MSPOWERPOINT))
 114             CurrentRegistrar->RegisterForMsPowerPoint();
 115
 116         if ((Register & HTML_EDITOR))
 117             CurrentRegistrar->RegisterAsHtmlEditorForInternetExplorer();
 118
 119         if ((Register & DEFAULT_SHELL_HTML_EDITOR))
 120         {
 121             CurrentRegistrar->RegisterAsDefaultHtmlEditorForInternetExplorer();
 122             CurrentRegistrar->RegisterAsDefaultShellHtmlEditor();
 123         }
 124     }
 125     catch(RegistryException&)
 126     {
 127         assert(false);
 128     }
 129
 130     if (Register)
 131         SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, 0, 0);
 132 }
 133
 134 void Unregister4MsDoc(MSIHANDLE handle, int Unregister)
 135 {
 136     try
 137     {
 138         RegistrationContextInformation RegContext(handle, GetOfficeExecutablePath(handle));
 139         RegistrarPtr CurrentRegistrar = CreateRegistrar(IsAllUserInstallation(handle), RegContext);
 140
 141         if ((Unregister & MSWORD) && CurrentRegistrar->IsRegisteredFor(MSWORD))
 142             CurrentRegistrar->UnregisterForMsWord();
 143
 144         if ((Unregister & HTML_EDITOR) && CurrentRegistrar->IsRegisteredFor(HTML_EDITOR))
 145             CurrentRegistrar->UnregisterAsHtmlEditorForInternetExplorer();
 146
 147         if ((Unregister & MSEXCEL) && CurrentRegistrar->IsRegisteredFor(MSEXCEL))
 148             CurrentRegistrar->UnregisterForMsExcel();
 149
 150         if ((Unregister & MSPOWERPOINT) && CurrentRegistrar->IsRegisteredFor(MSPOWERPOINT))
 151             CurrentRegistrar->UnregisterForMsPowerPoint();
 152
 153         if ((Unregister & DEFAULT_HTML_EDITOR_FOR_IE) && CurrentRegistrar->IsRegisteredFor(DEFAULT_HTML_EDITOR_FOR_IE))
 154             CurrentRegistrar->UnregisterAsDefaultHtmlEditorForInternetExplorer();
 155
 156         if ((Unregister & DEFAULT_SHELL_HTML_EDITOR) && CurrentRegistrar->IsRegisteredFor(DEFAULT_SHELL_HTML_EDITOR))
 157             CurrentRegistrar->UnregisterAsDefaultShellHtmlEditor();
 158     }
 159     catch(RegistryException&)
 160     {
 161         assert(false);
 162     }
 163
 164     if (Unregister)
 165         SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, 0, 0);
 166 }
 167
 168 //-----------------------------------------
 169 // restores the entries for the selected
 170 // registry entries
 171 // Algorithm:
 172 //
 173 // 1.
 174 // Target key exist (e.g. '.doc')
 175 // Default value == soffice.?
 176 // Backup key != empty
 177 // Action: Replace Default value with backup
 178 // key
 179 //
 180 // 2.
 181 // Target key exist
 182 // Default value == soffice.?
 183 // Backup key == empty
 184 // Action: delete default value
 185 //
 186 // 3.
 187 // Target key exist
 188 // Default value != soffice.?
 189 // Action: nop
 190 //
 191 // 4.
 192 // Target key does not exist
 193 // Action: nop
 194 //-----------------------------------------
 195
 196 void Unregister4MsDocAll(MSIHANDLE handle)
 197 {
 198     try
 199     {
 200         RegistrationContextInformation RegContext(handle, GetOfficeExecutablePath(handle));
 201         RegistrarPtr CurrentRegistrar = CreateRegistrar(IsAllUserInstallation(handle), RegContext);
 202
 203         CurrentRegistrar->UnregisterAllAndCleanUpRegistry();
 204     }
 205     catch(RegistryException&)
 206     {
 207         assert(false);
 208     }
 209 }
 210
 211 //-----------------------------------------
 212 // restores lost settings formerly made
 213 // with Register4MsDoc
 214 //-----------------------------------------
 215
 216 void RepairRegister4MsDocSettings(MSIHANDLE handle)
 217 {
 218     try
 219     {
 220         RegistrationContextInformation RegContext(handle, GetOfficeExecutablePath(handle));
 221         RegistrarPtr CurrentRegistrar = CreateRegistrar(IsAllUserInstallation(handle), RegContext);
 222
 223         CurrentRegistrar->RepairRegistrationState();
 224     }
 225     catch(RegistryException&)
 226     {
 227         assert(false);
 228     }
 229 }
 230
 231 bool IsRegisteredFor(MSIHANDLE handle, int State)
 232 {
 233     bool Registered = false;
 234
 235     try
 236     {
 237         RegistrationContextInformation RegContext(handle, GetOfficeExecutablePath(handle));
 238         RegistrarPtr CurrentRegistrar = CreateRegistrar(IsAllUserInstallation(handle), RegContext);
 239
 240         Registered = CurrentRegistrar->IsRegisteredFor(State);
 241     }
 242     catch(RegistryException&)
 243     {
 244         assert(false);
 245     }
 246     return Registered;
 247 }
 248
 249 #define SO60_UNINSTALL_KEY L"Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\StarOffice 6.0"
 250 #define SO_BACKUP_KEY      L"soffice6.bak"
 251 #define REGMSDOCSTATE      L"Reg4MsDocState"
 252 #define SOFTWARE_CLASSES   L"Software\\Classes"
 253
 254 int FixReturnRegistrationState(MSIHANDLE handle)
 255 {
 256     int registration_state = 0;
 257
 258     try
 259     {
 260         WindowsRegistry registry;
 261
 262         RegistryValue rv_regmsdocstate = RegistryValue(
 263             new RegistryValueImpl(REGMSDOCSTATE, 0));
 264
 265         RegistryKey so_bak_key;
 266
 267         if (IsAllUserInstallation(handle))
 268         {
 269             RegistryKey hkcr_key = registry.GetClassesRootKey();
 270
 271             if (hkcr_key->HasSubKey(SO_BACKUP_KEY))
 272                 so_bak_key = hkcr_key->OpenSubKey(SO_BACKUP_KEY);
 273             else
 274                 so_bak_key = hkcr_key->CreateSubKey(SO_BACKUP_KEY);
 275
 276             if (!so_bak_key->HasValue(REGMSDOCSTATE))
 277             {
 278                 // set a defined value
 279                 so_bak_key->SetValue(rv_regmsdocstate);
 280
 281                 RegistryKey hklm_key = registry.GetLocalMachineKey();
 282
 283                 if (hklm_key->HasSubKey(SO60_UNINSTALL_KEY))
 284                 {
 285                     RegistryKey so_uninst_key =
 286                         hklm_key->OpenSubKey(SO60_UNINSTALL_KEY);
 287
 288                     if (so_uninst_key->HasValue(REGMSDOCSTATE))
 289                         so_bak_key->CopyValue(so_uninst_key, REGMSDOCSTATE);
 290                 }
 291             }
 292         }
 293         else
 294         {
 295             RegistryKey hkcu_classes_key =
 296                 registry.GetCurrentUserKey()->OpenSubKey(SOFTWARE_CLASSES);
 297
 298             so_bak_key = hkcu_classes_key->CreateSubKey(SO_BACKUP_KEY);
 299
 300             if (!so_bak_key->HasValue(REGMSDOCSTATE))
 301             {
 302                 // set a defined value
 303                 so_bak_key->SetValue(rv_regmsdocstate);
 304
 305                 RegistryKey hklm_sftw_classes =
 306                     registry.GetLocalMachineKey()->OpenSubKey(SOFTWARE_CLASSES, false);
 307
 308                 RegistryKey so_bak_key_old;
 309
 310                 if (hklm_sftw_classes->HasSubKey(SO_BACKUP_KEY))
 311                 {
 312                     so_bak_key_old = hklm_sftw_classes->OpenSubKey(SO_BACKUP_KEY, false);
 313
 314                     if (so_bak_key_old->HasValue(REGMSDOCSTATE))
 315                         so_bak_key->CopyValue(so_bak_key_old, REGMSDOCSTATE);
 316                 }
 317                 else // try the uninstall key
 318                 {
 319                     RegistryKey hklm_key = registry.GetLocalMachineKey();
 320
 321                     if (hklm_key->HasSubKey(SO60_UNINSTALL_KEY))
 322                     {
 323                         RegistryKey so_uninst_key =
 324                             hklm_key->OpenSubKey(SO60_UNINSTALL_KEY);
 325
 326                         if (so_uninst_key->HasValue(REGMSDOCSTATE))
 327                             so_bak_key->CopyValue(so_uninst_key, REGMSDOCSTATE);
 328                     }
 329                 }
 330             }
 331         }
 332
 333         rv_regmsdocstate = so_bak_key->GetValue(REGMSDOCSTATE);
 334         registration_state = rv_regmsdocstate->GetDataAsInt();
 335     }
 336     catch(RegistryException&)
 337     {
 338         registration_state = 0;
 339     }
 340
 341     return registration_state;
 342 }
 343