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

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