Source code
Revision control
Copy as Markdown
Other Tools
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
#include "nsMapiAddressBook.h"
#include "mozilla/Logging.h"
#include "mozilla/DebugOnly.h"
#define PRINT_TO_CONSOLE 0
#if PRINT_TO_CONSOLE
# define PRINTF(args) printf args
#else
static mozilla::LazyLogModule gMapiAddressBookLog("MAPIAddressBook");
# define PRINTF(args) \
MOZ_LOG(gMapiAddressBookLog, mozilla::LogLevel::Debug, args)
#endif
using namespace mozilla;
HMODULE nsMapiAddressBook::mLibrary = NULL;
int32_t nsMapiAddressBook::mLibUsage = 0;
LPMAPIINITIALIZE nsMapiAddressBook::mMAPIInitialize = NULL;
LPMAPIUNINITIALIZE nsMapiAddressBook::mMAPIUninitialize = NULL;
LPMAPIALLOCATEBUFFER nsMapiAddressBook::mMAPIAllocateBuffer = NULL;
LPMAPIFREEBUFFER nsMapiAddressBook::mMAPIFreeBuffer = NULL;
LPMAPILOGONEX nsMapiAddressBook::mMAPILogonEx = NULL;
BOOL nsMapiAddressBook::mInitialized = FALSE;
BOOL nsMapiAddressBook::mLogonDone = FALSE;
LPMAPISESSION nsMapiAddressBook::mRootSession = NULL;
LPADRBOOK nsMapiAddressBook::mRootBook = NULL;
BOOL nsMapiAddressBook::LoadMapiLibrary(void) {
if (mLibrary) {
++mLibUsage;
return TRUE;
}
HMODULE libraryHandle = LoadLibraryW(L"MAPI32.DLL");
if (!libraryHandle) {
return FALSE;
}
FARPROC entryPoint = GetProcAddress(libraryHandle, "MAPIGetNetscapeVersion");
if (entryPoint) {
FreeLibrary(libraryHandle);
libraryHandle = LoadLibraryW(L"MAPI32BAK.DLL");
if (!libraryHandle) {
return FALSE;
}
}
mLibrary = libraryHandle;
++mLibUsage;
mMAPIInitialize = reinterpret_cast<LPMAPIINITIALIZE>(
GetProcAddress(mLibrary, "MAPIInitialize"));
if (!mMAPIInitialize) {
return FALSE;
}
mMAPIUninitialize = reinterpret_cast<LPMAPIUNINITIALIZE>(
GetProcAddress(mLibrary, "MAPIUninitialize"));
if (!mMAPIUninitialize) {
return FALSE;
}
mMAPIAllocateBuffer = reinterpret_cast<LPMAPIALLOCATEBUFFER>(
GetProcAddress(mLibrary, "MAPIAllocateBuffer"));
if (!mMAPIAllocateBuffer) {
return FALSE;
}
mMAPIFreeBuffer = reinterpret_cast<LPMAPIFREEBUFFER>(
GetProcAddress(mLibrary, "MAPIFreeBuffer"));
if (!mMAPIFreeBuffer) {
return FALSE;
}
mMAPILogonEx =
reinterpret_cast<LPMAPILOGONEX>(GetProcAddress(mLibrary, "MAPILogonEx"));
if (!mMAPILogonEx) {
return FALSE;
}
MAPIINIT_0 mapiInit = {MAPI_INIT_VERSION, MAPI_MULTITHREAD_NOTIFICATIONS};
HRESULT retCode = mMAPIInitialize(&mapiInit);
if (HR_FAILED(retCode)) {
PRINTF(("Cannot initialize MAPI %08lx.\n", retCode));
return FALSE;
}
mInitialized = TRUE;
retCode = mMAPILogonEx(
0, NULL, NULL,
MAPI_NO_MAIL | MAPI_USE_DEFAULT | MAPI_EXTENDED | MAPI_NEW_SESSION,
&mRootSession);
if (HR_FAILED(retCode)) {
PRINTF(("Cannot logon to MAPI %08lx.\n", retCode));
return FALSE;
}
mLogonDone = TRUE;
retCode = mRootSession->OpenAddressBook(0, NULL, 0, &mRootBook);
if (HR_FAILED(retCode)) {
PRINTF(("Cannot open MAPI address book %08lx.\n", retCode));
}
return HR_SUCCEEDED(retCode);
}
void nsMapiAddressBook::FreeMapiLibrary(void) {
if (mLibrary) {
if (--mLibUsage == 0) {
{
if (mRootBook) {
mRootBook->Release();
}
if (mRootSession) {
if (mLogonDone) {
mRootSession->Logoff(NULL, 0, 0);
mLogonDone = FALSE;
}
mRootSession->Release();
}
if (mInitialized) {
mMAPIUninitialize();
mInitialized = FALSE;
}
}
FreeLibrary(mLibrary);
mLibrary = NULL;
}
}
}
nsMapiAddressBook::nsMapiAddressBook(void) : nsAbWinHelper() {
mozilla::DebugOnly<BOOL> result = Initialize();
NS_ASSERTION(result == TRUE, "Couldn't initialize Mapi Helper");
MOZ_COUNT_CTOR(nsMapiAddressBook);
}
nsMapiAddressBook::~nsMapiAddressBook(void) {
StaticMutexAutoLock guard(sMutex);
FreeMapiLibrary();
MOZ_COUNT_DTOR(nsMapiAddressBook);
}
BOOL nsMapiAddressBook::Initialize(void) {
if (mAddressBook) {
return TRUE;
}
StaticMutexAutoLock guard(sMutex);
if (!LoadMapiLibrary()) {
PRINTF(("Cannot load library.\n"));
return FALSE;
}
mAddressBook = mRootBook;
mAddressSession = mRootSession;
mAddressFreeBuffer = mMAPIFreeBuffer;
return TRUE;
}
void nsMapiAddressBook::AllocateBuffer(ULONG aByteCount, LPVOID* aBuffer) {
mMAPIAllocateBuffer(aByteCount, aBuffer);
}
void nsMapiAddressBook::FreeBuffer(LPVOID aBuffer) { mMAPIFreeBuffer(aBuffer); }