Return to comsetup.cpp CVS log

Up to [truecrypt] / truecrypt / setup

Annotation of truecrypt/setup/comsetup.cpp, revision 1.1

1.1     ! root        1: /*
        !             2:  Copyright (c) TrueCrypt Foundation. All rights reserved.
        !             3: 
        !             4:  Covered by the TrueCrypt License 2.2 the full text of which is contained
        !             5:  in the file License.txt included in TrueCrypt binary and source code
        !             6:  distribution packages.
        !             7: */
        !             8: 
        !             9: #include <atlbase.h>
        !            10: #include <statreg.h>
        !            11: #include <windows.h>
        !            12: #include "ComSetup.h"
        !            13: #include "Resource.h"
        !            14: #include "../Mount/MainCom_i.c"
        !            15: #include "../Format/FormatCom_i.c"
        !            16: 
        !            17: 
        !            18: extern "C" BOOL RegisterComServers (char *modulePath)
        !            19: {
        !            20:        BOOL ret = FALSE;
        !            21:        wchar_t mainModule[1024], formatModule[1024];
        !            22:        CComPtr<ITypeLib> tl, tl2;
        !            23: 
        !            24:        wsprintfW (mainModule, L"%hsTrueCrypt.exe", modulePath);
        !            25:        wsprintfW (formatModule, L"%hsTrueCrypt Format.exe", modulePath);
        !            26: 
        !            27:        UnRegisterTypeLib (LIBID_TrueCrypt, 1, 0, 0, SYS_WIN32);
        !            28:        UnRegisterTypeLib (LIBID_TrueCryptFormat, 1, 0, 0, SYS_WIN32);
        !            29: 
        !            30:        CRegObject ro;
        !            31:        ro.FinalConstruct ();
        !            32: 
        !            33:        ro.AddReplacement (L"MAIN_MODULE", mainModule);
        !            34:        ro.AddReplacement (L"FORMAT_MODULE", formatModule);
        !            35: 
        !            36:        wchar_t setupModule[MAX_PATH];
        !            37:        GetModuleFileNameW (NULL, setupModule, sizeof (setupModule));
        !            38:        if (ro.ResourceRegister (setupModule, IDR_COMREG, L"REGISTRY") != S_OK)
        !            39:                goto error;
        !            40: 
        !            41:        if (LoadTypeLib (mainModule, &tl) != S_OK
        !            42:                || RegisterTypeLib (tl, mainModule, 0) != S_OK)
        !            43:                goto error;
        !            44: 
        !            45:        if (LoadTypeLib (formatModule, &tl2) != S_OK
        !            46:                || RegisterTypeLib (tl2, formatModule, 0) != S_OK)
        !            47:                goto error;
        !            48: 
        !            49:        ret = TRUE;
        !            50: error:
        !            51:        ro.FinalRelease ();
        !            52:        return ret;
        !            53: }
        !            54: 
        !            55: 
        !            56: extern "C" BOOL UnregisterComServers (char *modulePath)
        !            57: {
        !            58:        BOOL ret;
        !            59: 
        !            60:        if (UnRegisterTypeLib (LIBID_TrueCrypt, 1, 0, 0, SYS_WIN32) != S_OK)
        !            61:                return FALSE;
        !            62:        if (UnRegisterTypeLib (LIBID_TrueCryptFormat, 1, 0, 0, SYS_WIN32) != S_OK)
        !            63:                return FALSE;
        !            64: 
        !            65:        wchar_t module[1024];
        !            66:        CRegObject ro;
        !            67:        ro.FinalConstruct ();
        !            68: 
        !            69:        wsprintfW (module, L"%hs\\TrueCrypt.exe", modulePath);
        !            70:        ro.AddReplacement (L"MAIN_MODULE", module);
        !            71: 
        !            72:        wsprintfW (module, L"%hs\\TrueCrypt Format.exe", modulePath);
        !            73:        ro.AddReplacement (L"FORMAT_MODULE", module);
        !            74: 
        !            75:        wchar_t setupModule[MAX_PATH];
        !            76:        GetModuleFileNameW (NULL, setupModule, sizeof (setupModule));
        !            77: 
        !            78:        ret = ro.ResourceUnregister (setupModule, IDR_COMREG, L"REGISTRY") == S_OK;
        !            79: 
        !            80:        ro.FinalRelease ();
        !            81:        return ret;
        !            82: }

This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.