Monotone-Parent: 1386ff575a581ce153a4d70c50dc0bed29d339ed

Monotone-Revision: 2f9e650d437ad6f3b4629d6df38c98b8151e4f0f

Monotone-Author: wsourdeau@inverse.ca
Monotone-Date: 2010-03-31T20:15:11
Monotone-Branch: ca.inverse.sogo

Tools/SOGoSockDOperation.m

@@ -0,0 +1,239 @@
+/* SOGoSockDOperation.m - this file is part of SOGo
+ *
+ * Copyright (C) 2010 Inverse inc.
+ *
+ * Author: Wolfgang Sourdeau <wsourdeau@inverse.ca>
+ *
+ * This file is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2, or (at your option)
+ * any later version.
+ *
+ * This file is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+#include <ldap.h>
+
+#import <Foundation/NSArray.h>
+#import <Foundation/NSString.h>
+
+#import <NGExtensions/NGBase64Coding.h>
+#import <NGStreams/NGActiveSocket.h>
+#import <Contacts/SOGoContactFolders.h>
+#import <Contacts/SOGoContactGCSFolder.h>
+#import <SOGo/SOGoProductLoader.h>
+#import <SOGo/NSDictionary+Utilities.h>
+#import <SOGo/NSString+Utilities.h>
+
+#import "SOGoSockDOperation.h"
+
+@interface _SOGoSockDOperationSearch : SOGoSockDOperation
+@end
+
+@interface _SOGoSockDOperationUnbind : SOGoSockDOperation
+@end
+
+@implementation SOGoSockDOperation
+
++ (void) initialize
+{
+  [[SOGoProductLoader productLoader]
+    loadProducts: [NSArray arrayWithObjects: @"Contacts.SOGo", nil]];
+}
+
++ (SOGoSockDOperation *) operationWithMethod: (NSString *) method
+                               andParameters: (NSDictionary *) parameters
+{
+  static NSArray *operations = nil;
+  NSString *className;
+  SOGoSockDOperation *newOperation;
+
+  if (!operations)
+    operations = [[NSArray alloc] initWithObjects:
+                                    @"SEARCH", @"UNBIND",
+                                  nil];
+  if ([operations containsObject: method])
+    {
+      className = [NSString stringWithFormat: @"_SOGoSockDOperation%@",
+                            [method capitalizedString]];
+      newOperation = [NSClassFromString (className) new];
+      [newOperation autorelease];
+      [newOperation setParameters: parameters];
+    }
+  else
+    newOperation = nil;
+
+  return newOperation;
+}
+
+- (void) dealloc
+{
+  [parameters release];
+  [super dealloc];
+}
+
+- (void) setParameters: (NSDictionary *) newParameters
+{
+  ASSIGN (parameters, newParameters);
+}
+
+- (void) _appendEntry: (NSDictionary *) entry
+             toResult: (NSMutableString *) result
+{
+  static NSArray *ldifFields = nil;
+  NSString *key, *value;
+  int count, max;
+
+  if (!ldifFields)
+    {
+      ldifFields = [NSArray arrayWithObjects: @"c_cn", @"c_givenname",
+                            @"c_mail", @"c_o", @"c_screenname", @"c_sn", nil];
+      [ldifFields retain];
+    }
+
+  [result appendFormat: @"dn: cn=%@,%@\n",
+   [entry objectForKey: @"c_name"], [parameters objectForKey: @"base"]];
+  [result appendString: @"objectClass: person\nobjectClass: inetOrgPerson\n"];
+
+  max = [ldifFields count];
+  for (count = 0; count < max; count++)
+    {
+      key = [ldifFields objectAtIndex: count];
+      value = [entry objectForKey: key];
+      if ([value isKindOfClass: [NSString class]] && [value length] > 0)
+        {
+          if ([value _isLDIFSafe])
+            [result appendFormat: @"%@: %@\n",
+                    [key substringFromIndex: 2], value];
+          else
+            [result appendFormat: @"%@:: %@\n",
+                    [key substringFromIndex: 2],
+                    [value stringByEncodingBase64]];
+        }
+    }
+  [result appendString: @"\n"];
+}
+
+- (void) respondOnSocket: (NGActiveSocket *) responseSocket
+{
+  int count, max;
+  NSMutableString *result;
+
+  result = [NSMutableString string];
+
+  max = [resultEntries count];
+  for (count = 0; count < max; count++)
+    [self _appendEntry: [resultEntries objectAtIndex: count]
+              toResult: result];
+
+  [result appendFormat: @"RESULT\ncode: %", resultCode];
+  [responseSocket
+    safeWriteData: [result dataUsingEncoding: NSASCIIStringEncoding]];
+}
+
+@end
+
+@implementation _SOGoSockDOperationSearch
+
+- (NSString *) _extractFirstValueFromDN: (NSString *) dn
+{
+  NSString *firstValue;
+  NSRange charRange, valueRange;
+
+  charRange = [dn rangeOfString: @"="];
+  valueRange.location = charRange.location + 1;
+  charRange = [dn rangeOfString: @","];
+  valueRange.length = charRange.location - valueRange.location;
+  firstValue = [dn substringFromRange: valueRange];
+
+  return firstValue;
+}
+
+- (NSString *) _convertLDAPFilter: (NSString *) ldapFilter
+{
+  /* Ultra hackish: we extract the value between the first "=" and the first
+     ")", thereby assuming it will be the same for all search fields */
+  NSString *filter;
+  NSRange charRange, valueRange;
+
+  charRange = [ldapFilter rangeOfString: @"="];
+  valueRange.location = charRange.location + 1;
+  charRange = [ldapFilter rangeOfString: @")"];
+  valueRange.length = charRange.location - valueRange.location;
+  filter = [ldapFilter substringFromRange: valueRange];
+  charRange = [filter rangeOfString: @"*"];
+  if (charRange.length == 1)
+    {
+      if (charRange.location == 0)
+        filter = [filter substringFromIndex: 1];
+      else
+        filter = [filter substringToIndex: [filter length] - 1];
+    }
+
+  return filter;
+}
+
+- (SOGoContactGCSFolder *) _getFolderWithId: (NSString *) folderId
+                                    forUser: (NSString *) uid
+{
+  SOGoUserFolder *userFolder;
+  SOGoContactFolders *parentFolder;
+  SOGoContactGCSFolder *folder;
+
+  userFolder = [SOGoUserFolder objectWithName: uid inContainer: nil];
+  parentFolder = [userFolder lookupName: @"Contacts"
+                              inContext: nil acquire: NO];
+  folder = [parentFolder lookupName: folderId inContext: nil acquire: NO];
+
+  return folder;
+}
+
+- (void) _performSearch
+{
+  NSString *bindDN, *baseDN, *uid, *folderId, *filter;
+  SOGoContactGCSFolder *folder;
+
+  bindDN = [parameters objectForKey: @"binddn"];
+  baseDN = [parameters objectForKey: @"base"];
+  if ([bindDN length] && [baseDN length])
+    {
+      uid = [self _extractFirstValueFromDN: bindDN];
+      folderId = [self _extractFirstValueFromDN: baseDN];
+      folder = [self _getFolderWithId: folderId forUser: uid];
+      if (folder)
+        {
+          filter
+            = [self _convertLDAPFilter: [parameters objectForKey: @"filter"]];
+          // if (![filter length]
+          //     && [folder isKindOfClass: [SOGoContactSourceFolder class]])
+          //   filter = @"*";
+          resultEntries
+            = [folder lookupContactsWithFilter: filter
+                                        sortBy: @"c_cn"
+                                      ordering: NSOrderedAscending];
+        }
+      else
+        resultCode = LDAP_NO_SUCH_OBJECT;
+    }
+  else
+    resultCode = LDAP_PROTOCOL_ERROR;
+}
+
+- (void) respondOnSocket: (NGActiveSocket *) responseSocket
+{
+  [self _performSearch];
+  [super respondOnSocket: responseSocket];
+}
+
+@end
+
+@implementation _SOGoSockDOperationUnbind
+@end