1 files changed, 163 insertions, 0 deletions
diff --git a/comm/mailnews/addrbook/src/nsMapiAddressBook.cpp b/comm/mailnews/addrbook/src/nsMapiAddressBook.cpp
new file mode 100644
index 0000000000..bd6e080443
--- /dev/null
+++ b/comm/mailnews/addrbook/src/nsMapiAddressBook.cpp
@@ -0,0 +1,163 @@
+/* -*- 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
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+#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); }