Return to maincom.cpp CVS log

Up to [truecrypt] / truecrypt / mount

Annotation of truecrypt/mount/maincom.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 <atlcomcli.h>
        !            10: #include <atlconv.h>
        !            11: #include <strsafe.h>
        !            12: #include <windows.h>
        !            13: #include "BaseCom.h"
        !            14: #include "Dlgcode.h"
        !            15: #include "MainCom.h"
        !            16: #include "MainCom_h.h"
        !            17: #include "MainCom_i.c"
        !            18: #include "Password.h"
        !            19: 
        !            20: static volatile LONG ObjectCount = 0;
        !            21: 
        !            22: class TrueCrypt : public ITrueCrypt
        !            23: {
        !            24: 
        !            25: public:
        !            26:        TrueCrypt (DWORD messageThreadId) : RefCount (0), MessageThreadId (messageThreadId)
        !            27:        {
        !            28:                InterlockedIncrement (&ObjectCount);
        !            29:        }
        !            30: 
        !            31:        ~TrueCrypt ()
        !            32:        {
        !            33:                if (InterlockedDecrement (&ObjectCount) == 0)
        !            34:                        PostThreadMessage (MessageThreadId, WM_APP, 0, 0);
        !            35:        }
        !            36: 
        !            37:        virtual ULONG STDMETHODCALLTYPE AddRef ()
        !            38:        {
        !            39:                return InterlockedIncrement (&RefCount);
        !            40:        }
        !            41: 
        !            42:        virtual ULONG STDMETHODCALLTYPE Release ()
        !            43:        {
        !            44:                if (!InterlockedDecrement (&RefCount))
        !            45:                {
        !            46:                        delete this;
        !            47:                        return 0;
        !            48:                }
        !            49: 
        !            50:                return RefCount;
        !            51:        }
        !            52: 
        !            53:        virtual HRESULT STDMETHODCALLTYPE QueryInterface (REFIID riid, void **ppvObject)
        !            54:        {
        !            55:                if (riid == IID_IUnknown || riid == IID_ITrueCrypt)
        !            56:                        *ppvObject = this;
        !            57:                else
        !            58:                {
        !            59:                        *ppvObject = NULL;
        !            60:                        return E_NOINTERFACE;
        !            61:                }
        !            62: 
        !            63:                AddRef ();
        !            64:                return S_OK;
        !            65:        }
        !            66:        
        !            67:        virtual int STDMETHODCALLTYPE BackupVolumeHeader (LONG_PTR hwndDlg, BOOL bRequireConfirmation, BSTR lpszVolume)
        !            68:        {
        !            69:                USES_CONVERSION;
        !            70:                return ::BackupVolumeHeader ((HWND) hwndDlg, bRequireConfirmation, CW2A (lpszVolume));
        !            71:        }
        !            72: 
        !            73:        virtual int STDMETHODCALLTYPE RestoreVolumeHeader (LONG_PTR hwndDlg, BSTR lpszVolume)
        !            74:        {
        !            75:                USES_CONVERSION;
        !            76:                return ::RestoreVolumeHeader ((HWND) hwndDlg, CW2A (lpszVolume));
        !            77:        }
        !            78: 
        !            79:        virtual int STDMETHODCALLTYPE ChangePassword (BSTR volumePath, Password *oldPassword, Password *newPassword, int pkcs5, LONG_PTR hWnd)
        !            80:        {
        !            81:                USES_CONVERSION;
        !            82:                return ::ChangePwd (CW2A (volumePath), oldPassword, newPassword, pkcs5, (HWND) hWnd);
        !            83:        }
        !            84: 
        !            85: protected:
        !            86:        DWORD MessageThreadId;
        !            87:        LONG RefCount;
        !            88: };
        !            89: 
        !            90: 
        !            91: extern "C" BOOL ComServerMain ()
        !            92: {
        !            93:        TrueCryptFactory<TrueCrypt> factory (GetCurrentThreadId ());
        !            94:        DWORD cookie;
        !            95: 
        !            96:        if (IsUacSupported ())
        !            97:                UacElevated = TRUE;
        !            98: 
        !            99:        if (CoRegisterClassObject (CLSID_TrueCrypt, (LPUNKNOWN) &factory,
        !           100:                CLSCTX_LOCAL_SERVER, REGCLS_SINGLEUSE, &cookie) != S_OK)
        !           101:                return FALSE;
        !           102: 
        !           103:        MSG msg;
        !           104:        while (GetMessage (&msg, NULL, 0, 0))
        !           105:        {
        !           106:                TranslateMessage (&msg);
        !           107:                DispatchMessage (&msg);
        !           108: 
        !           109:                if (msg.message == WM_APP
        !           110:                        && ObjectCount < 1
        !           111:                        && !factory.IsServerLocked ())
        !           112:                        break;
        !           113:        }
        !           114:        CoRevokeClassObject (cookie);
        !           115: 
        !           116:        return TRUE;
        !           117: }
        !           118: 
        !           119: 
        !           120: static BOOL ComGetInstance (HWND hWnd, ITrueCrypt **tcServer)
        !           121: {
        !           122:        return ComGetInstanceBase (hWnd, CLSID_TrueCrypt, IID_ITrueCrypt, (void **) tcServer);
        !           123: }
        !           124: 
        !           125: 
        !           126: extern "C" int UacBackupVolumeHeader (HWND hwndDlg, BOOL bRequireConfirmation, char *lpszVolume)
        !           127: {
        !           128:        CComPtr<ITrueCrypt> tc;
        !           129:        int r;
        !           130: 
        !           131:        CoInitialize (NULL);
        !           132: 
        !           133:        if (ComGetInstance (hwndDlg, &tc))
        !           134:                r = tc->BackupVolumeHeader ((LONG_PTR) hwndDlg, bRequireConfirmation, CComBSTR (lpszVolume));
        !           135:        else
        !           136:                r = -1;
        !           137: 
        !           138:        CoUninitialize ();
        !           139: 
        !           140:        return r;
        !           141: }
        !           142: 
        !           143: 
        !           144: extern "C" int UacRestoreVolumeHeader (HWND hwndDlg, char *lpszVolume)
        !           145: {
        !           146:        CComPtr<ITrueCrypt> tc;
        !           147:        int r;
        !           148: 
        !           149:        CoInitialize (NULL);
        !           150: 
        !           151:        if (ComGetInstance (hwndDlg, &tc))
        !           152:                r = tc->RestoreVolumeHeader ((LONG_PTR) hwndDlg, CComBSTR (lpszVolume));
        !           153:        else
        !           154:                r = -1;
        !           155: 
        !           156:        CoUninitialize ();
        !           157: 
        !           158:        return r;
        !           159: }
        !           160: 
        !           161: 
        !           162: extern "C" int UacChangePwd (char *lpszVolume, Password *oldPassword, Password *newPassword, int pkcs5, HWND hwndDlg)
        !           163: {
        !           164:        CComPtr<ITrueCrypt> tc;
        !           165:        int r;
        !           166: 
        !           167:        if (ComGetInstance (hwndDlg, &tc))
        !           168:        {
        !           169:                WaitCursor ();
        !           170:                r = tc->ChangePassword (CComBSTR (lpszVolume), oldPassword, newPassword, 0, (LONG_PTR) hwndDlg);
        !           171:                NormalCursor ();
        !           172:        }
        !           173:        else
        !           174:                r = -1;
        !           175: 
        !           176:        return r;
        !           177: }

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