1 files changed, 372 insertions, 0 deletions
diff --git a/mailnews/addrbook/src/nsAbLDAPDirectoryModify.cpp b/mailnews/addrbook/src/nsAbLDAPDirectoryModify.cpp
new file mode 100644
index 000000000..95af79c04
--- /dev/null
+++ b/mailnews/addrbook/src/nsAbLDAPDirectoryModify.cpp
@@ -0,0 +1,372 @@
+/* -*- 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 "nsAbLDAPDirectoryModify.h"
+#include "nsILDAPMessage.h"
+#include "nsILDAPConnection.h"
+#include "nsILDAPErrors.h"
+#include "nsILDAPModification.h"
+#include "nsIServiceManager.h"
+#include "nsIAbLDAPDirectory.h"
+#include "nsIMutableArray.h"
+#include "nsComponentManagerUtils.h"
+#include "nsServiceManagerUtils.h"
+
+#include <stdio.h>
+
+using namespace mozilla;
+
+class nsAbModifyLDAPMessageListener : public nsAbLDAPListenerBase
+{
+public:
+  NS_DECL_THREADSAFE_ISUPPORTS
+
+  nsAbModifyLDAPMessageListener(const int32_t type,
+                                const nsACString &cardDN,
+                                nsIArray* modArray,
+                                const nsACString &newRDN,
+                                const nsACString &newBaseDN,
+                                nsILDAPURL* directoryUrl,
+                                nsILDAPConnection* connection,
+                                nsIMutableArray* serverSearchControls,
+                                nsIMutableArray* clientSearchControls,
+                                const nsACString &login,
+                                const int32_t timeOut = 0);
+  // nsILDAPMessageListener
+  NS_IMETHOD OnLDAPMessage(nsILDAPMessage *aMessage) override;
+
+protected:
+  virtual ~nsAbModifyLDAPMessageListener();
+
+  nsresult Cancel();
+  virtual void InitFailed(bool aCancelled = false) override;
+  virtual nsresult DoTask() override;
+  nsresult DoMainTask();
+  nsresult OnLDAPMessageModifyResult(nsILDAPMessage *aMessage);
+  nsresult OnLDAPMessageRenameResult(nsILDAPMessage *aMessage);
+
+  int32_t mType;
+  nsCString mCardDN;
+  nsCOMPtr<nsIArray> mModification;
+  nsCString mNewRDN;
+  nsCString mNewBaseDN;
+
+  bool mFinished;
+  bool mCanceled;
+  bool mFlagRename;
+
+  nsCOMPtr<nsILDAPOperation> mModifyOperation;
+  nsCOMPtr<nsIMutableArray> mServerSearchControls;
+  nsCOMPtr<nsIMutableArray> mClientSearchControls;
+};
+
+
+NS_IMPL_ISUPPORTS(nsAbModifyLDAPMessageListener, nsILDAPMessageListener)
+
+nsAbModifyLDAPMessageListener::nsAbModifyLDAPMessageListener(
+    const int32_t type,
+    const nsACString &cardDN,
+    nsIArray* modArray,
+    const nsACString &newRDN,
+    const nsACString &newBaseDN,
+    nsILDAPURL* directoryUrl,
+    nsILDAPConnection* connection,
+    nsIMutableArray* serverSearchControls,
+    nsIMutableArray* clientSearchControls,
+    const nsACString &login,
+    const int32_t timeOut) :
+    nsAbLDAPListenerBase(directoryUrl, connection, login, timeOut),
+    mType(type),
+    mCardDN(cardDN),
+    mModification(modArray),
+    mNewRDN(newRDN),
+    mNewBaseDN(newBaseDN),
+    mFinished(false),
+    mCanceled(false),
+    mFlagRename(false),
+    mServerSearchControls(serverSearchControls),
+    mClientSearchControls(clientSearchControls)
+{
+  if (mType == nsILDAPModification::MOD_REPLACE &&
+      !mNewRDN.IsEmpty() && !mNewBaseDN.IsEmpty())
+    mFlagRename = true;
+}
+
+nsAbModifyLDAPMessageListener::~nsAbModifyLDAPMessageListener ()
+{
+}
+
+nsresult nsAbModifyLDAPMessageListener::Cancel ()
+{
+  nsresult rv = Initiate();
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  MutexAutoLock lock(mLock);
+
+  if (mFinished || mCanceled)
+    return NS_OK;
+
+  mCanceled = true;
+
+  return NS_OK;
+}
+
+NS_IMETHODIMP nsAbModifyLDAPMessageListener::OnLDAPMessage(nsILDAPMessage *aMessage)
+{
+  nsresult rv = Initiate();
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  int32_t messageType;
+  rv = aMessage->GetType(&messageType);
+  NS_ENSURE_SUCCESS(rv, rv);
+  
+  bool cancelOperation = false;
+
+  // Enter lock
+  {
+    MutexAutoLock lock (mLock);
+
+    if (mFinished)
+      return NS_OK;
+
+    // for these messages, no matter the outcome, we're done
+    if ((messageType == nsILDAPMessage::RES_ADD) || 
+        (messageType == nsILDAPMessage::RES_DELETE) ||
+        (messageType == nsILDAPMessage::RES_MODIFY))
+      mFinished = true;
+    else if (mCanceled)
+    {
+      mFinished = true;
+      cancelOperation = true;
+    }
+  }
+  // Leave lock
+
+  //    nsCOMPtr<nsIAbDirectoryQueryResult> queryResult;
+  if (!cancelOperation)
+  {
+    switch (messageType)
+    {
+    case nsILDAPMessage::RES_BIND:
+      rv = OnLDAPMessageBind(aMessage);
+      if (NS_FAILED(rv)) 
+        // We know the bind failed and hence the message has an error, so we
+        // can just call ModifyResult with the message and that'll sort it out
+        // for us.
+        rv = OnLDAPMessageModifyResult(aMessage);
+      break;
+    case nsILDAPMessage::RES_ADD:
+    case nsILDAPMessage::RES_MODIFY:
+    case nsILDAPMessage::RES_DELETE:
+      rv = OnLDAPMessageModifyResult(aMessage);
+      break;
+    case nsILDAPMessage::RES_MODDN:
+      mFlagRename = false;
+      rv = OnLDAPMessageRenameResult(aMessage);
+      if (NS_FAILED(rv)) 
+        // Rename failed, so we stop here
+        mFinished = true;
+      break;
+    default:
+      break;
+    }
+  }
+  else
+  {
+    if (mModifyOperation)
+      rv = mModifyOperation->AbandonExt();
+
+    // reset because we might re-use this listener...except don't do this
+    // until the search is done, so we'll ignore results from a previous
+    // search.
+    mCanceled = mFinished = false;
+  }
+
+  return rv;
+}
+
+void nsAbModifyLDAPMessageListener::InitFailed(bool aCancelled)
+{
+  // XXX Just cancel the operation for now
+  // we'll need to review this when we've got the proper listeners in place.
+  Cancel();
+}
+
+nsresult nsAbModifyLDAPMessageListener::DoTask()
+{
+  nsresult rv;
+  mCanceled = mFinished = false;
+
+  mModifyOperation = do_CreateInstance(NS_LDAPOPERATION_CONTRACTID, &rv);
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  rv = mModifyOperation->Init (mConnection, this, nullptr);
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  // XXX do we need the search controls?
+  rv = mModifyOperation->SetServerControls(mServerSearchControls);
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  rv = mModifyOperation->SetClientControls(mClientSearchControls);
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  if (mFlagRename)
+    return mModifyOperation->Rename(mCardDN, mNewRDN, mNewBaseDN, true);
+
+  switch (mType)
+  {
+    case nsILDAPModification::MOD_ADD:
+      return mModifyOperation->AddExt(mCardDN, mModification);
+    case nsILDAPModification::MOD_DELETE:
+      return mModifyOperation->DeleteExt(mCardDN);
+    case nsILDAPModification::MOD_REPLACE:
+      return mModifyOperation->ModifyExt(mCardDN, mModification);
+    default:
+      NS_ERROR("Bad LDAP modification requested");
+      return NS_ERROR_UNEXPECTED;
+  }
+}
+
+nsresult nsAbModifyLDAPMessageListener::OnLDAPMessageModifyResult(nsILDAPMessage *aMessage)
+{
+  nsresult rv;
+  NS_ENSURE_ARG_POINTER(aMessage);
+  
+  int32_t errCode;
+  rv = aMessage->GetErrorCode(&errCode);
+  NS_ENSURE_SUCCESS(rv, rv);
+ 
+  if (errCode != nsILDAPErrors::SUCCESS)
+  {
+    nsAutoCString errMessage;
+    rv = aMessage->GetErrorMessage(errMessage);
+    NS_ENSURE_SUCCESS(rv, rv);
+
+    printf("LDAP modification failed (code: %i, message: %s)\n",
+           errCode, errMessage.get());
+    return NS_ERROR_FAILURE;
+  }
+  
+  printf("LDAP modification succeeded\n");
+  return NS_OK;
+}
+
+nsresult nsAbModifyLDAPMessageListener::OnLDAPMessageRenameResult(nsILDAPMessage *aMessage)
+{
+  nsresult rv;
+  NS_ENSURE_ARG_POINTER(aMessage);
+  
+  int32_t errCode;
+  rv = aMessage->GetErrorCode(&errCode);
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  if (errCode != nsILDAPErrors::SUCCESS)
+  {
+    nsAutoCString errMessage;
+    rv = aMessage->GetErrorMessage(errMessage);
+    NS_ENSURE_SUCCESS(rv, rv);
+
+    printf("LDAP rename failed (code: %i, message: %s)\n",
+           errCode, errMessage.get());
+    return NS_ERROR_FAILURE;
+  }
+  
+  // Rename succeeded, now update the card DN and
+  // process the main task
+  mCardDN.Assign(mNewRDN);
+  mCardDN.AppendLiteral(",");
+  mCardDN.Append(mNewBaseDN);
+     
+  printf("LDAP rename succeeded\n");
+  return DoTask();
+}
+ 
+nsAbLDAPDirectoryModify::nsAbLDAPDirectoryModify()
+{
+}
+
+nsAbLDAPDirectoryModify::~nsAbLDAPDirectoryModify()
+{
+}
+
+nsresult nsAbLDAPDirectoryModify::DoModify(nsIAbLDAPDirectory *directory,
+                                           const int32_t &updateType,
+                                           const nsACString &cardDN,
+                                           nsIArray* modArray,
+                                           const nsACString &newRDN,
+                                           const nsACString &newBaseDN)
+{
+  NS_ENSURE_ARG_POINTER(directory);
+  // modArray may be null in the delete operation case.
+  if (!modArray &&
+      (updateType == nsILDAPModification::MOD_ADD ||
+       updateType == nsILDAPModification::MOD_REPLACE))
+    return NS_ERROR_NULL_POINTER;
+
+  nsresult rv;
+
+  // it's an error if we don't have a dn
+  if (cardDN.IsEmpty())
+    return NS_ERROR_INVALID_ARG;
+ 
+  nsCOMPtr<nsILDAPURL> currentUrl;
+  rv = directory->GetLDAPURL(getter_AddRefs(currentUrl));
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  // Get the ldap connection
+  nsCOMPtr<nsILDAPConnection> ldapConnection =
+    do_CreateInstance(NS_LDAPCONNECTION_CONTRACTID, &rv);
+  
+  nsCOMPtr<nsIMutableArray> serverSearchControls;
+  rv = directory->GetSearchServerControls(getter_AddRefs(serverSearchControls));
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  nsCOMPtr<nsIMutableArray> clientSearchControls;
+  rv = directory->GetSearchClientControls(getter_AddRefs(clientSearchControls));
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  /*
+  // XXX we need to fix how this all works - specifically, see the first patch
+  // on bug 124553 for how the query equivalent did this
+  // too soon? Do we need a new listener?
+  if (alreadyInitialized)
+  {
+    nsAbQueryLDAPMessageListener *msgListener =
+      NS_STATIC_CAST(nsAbQueryLDAPMessageListener *, 
+                     NS_STATIC_CAST(nsILDAPMessageListener *, mListener.get()));
+    if (msgListener)
+      {
+        msgListener->mUrl = url;
+        return msgListener->DoSearch();
+      }
+  }*/
+
+  nsCString login;
+  rv = directory->GetAuthDn(login);
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  uint32_t protocolVersion;
+  rv = directory->GetProtocolVersion(&protocolVersion);
+  NS_ENSURE_SUCCESS(rv, rv);
+
+  // Initiate LDAP message listener
+  nsAbModifyLDAPMessageListener* _messageListener =
+    new nsAbModifyLDAPMessageListener(updateType, cardDN, modArray,
+                                      newRDN, newBaseDN,
+                                      currentUrl,
+                                      ldapConnection,
+                                      serverSearchControls,
+                                      clientSearchControls,
+                                      login,
+                                      0);
+  if (_messageListener == NULL)
+    return NS_ERROR_OUT_OF_MEMORY;
+  
+  // Now lets initialize the LDAP connection properly. We'll kick
+  // off the bind operation in the callback function, |OnLDAPInit()|.
+  return ldapConnection->Init(currentUrl, login,
+                              _messageListener, nullptr, protocolVersion);
+}
+