Project

General

Profile

1
/**
2
 *  '$RCSfile$'
3
 *    Purpose: An implementation of the AuthInterface interface that
4
 *             allows Metacat to use the LDAP protocol for
5
 *             directory services
6
 *  Copyright: 2000 Regents of the University of California and the
7
 *             National Center for Ecological Analysis and Synthesis
8
 *    Authors: Matt Jones
9
 *    Release: @release@
10
 *
11
 *   '$Author: tao $'
12
 *     '$Date: 2002-04-11 08:35:53 -0700 (Thu, 11 Apr 2002) $'
13
 * '$Revision: 1006 $'
14
 *
15
 * This program is free software; you can redistribute it and/or modify
16
 * it under the terms of the GNU General Public License as published by
17
 * the Free Software Foundation; either version 2 of the License, or
18
 * (at your option) any later version.
19
 *
20
 * This program is distributed in the hope that it will be useful,
21
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
22
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
23
 * GNU General Public License for more details.
24
 *
25
 * You should have received a copy of the GNU General Public License
26
 * along with this program; if not, write to the Free Software
27
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
28
 */
29

    
30
package edu.ucsb.nceas.metacat;
31

    
32
import java.net.ConnectException;
33
import javax.naming.AuthenticationException;
34
import javax.naming.Context;
35
import javax.naming.NamingEnumeration;
36
import javax.naming.NamingException;
37
import javax.naming.SizeLimitExceededException;
38
import javax.naming.InitialContext;
39
import javax.naming.directory.InvalidSearchFilterException;
40
import javax.naming.directory.Attribute;
41
import javax.naming.directory.Attributes;
42
import javax.naming.directory.BasicAttribute;
43
import javax.naming.directory.BasicAttributes;
44
import javax.naming.directory.DirContext;
45
import javax.naming.directory.InitialDirContext;
46
import javax.naming.directory.SearchResult;
47
import javax.naming.directory.SearchControls;
48
import javax.naming.ReferralException;
49
import javax.naming.ldap.*;
50
import java.util.Iterator;
51
import java.util.HashMap;
52
import java.util.Hashtable;
53
import java.util.Enumeration;
54
import java.util.Set;
55
import java.util.Vector;
56

    
57
/**
58
 * An implementation of the AuthInterface interface that
59
 * allows Metacat to use the LDAP protocol for directory services.
60
 * The LDAP authentication service is used to determine if a user 
61
 * is authenticated, and whether they are a member of a particular group.
62
 */
63
public class AuthLdap implements AuthInterface, Runnable {
64
  private MetaCatUtil util = new MetaCatUtil();
65
  private String ldapUrl;
66
  private String ldapsUrl;
67
  private String ldapBase;
68
  private String referral;
69
  private Context referralContext;
70
  Hashtable env = new Hashtable(11);
71
  private Context rContext;
72
  private String userName;
73
  private String userPassword;
74
  ReferralException refExc;
75
  String uid=null;
76

    
77
  /** 
78
   * Construct an AuthLdap
79
   */
80
  public AuthLdap() {
81
    // Read LDAP URI for directory service information
82
    this.ldapUrl = MetaCatUtil.getOption("ldapurl");
83
    this.ldapsUrl = MetaCatUtil.getOption("ldapsurl");
84
    this.ldapBase = MetaCatUtil.getOption("ldapbase");
85
    this.referral = MetaCatUtil.getOption("referral");
86
  }
87

    
88
  /**
89
   * Determine if a user/password are valid according to the authentication
90
   * service.
91
   *
92
   * @param user the name of the principal to authenticate
93
   * @param password the password to use for authentication
94
   * @returns boolean true if authentication successful, false otherwise
95
   */
96
  public boolean authenticate(String user, String password)
97
                    throws ConnectException
98
  {
99
    String ldapUrl = this.ldapUrl;
100
    String ldapsUrl = this.ldapsUrl;
101
    String ldapBase = this.ldapBase;
102
    boolean authenticated = false;
103
    String identifier = user;
104
    //get uid here.
105
    uid=user.substring(0, user.indexOf(","));
106
  
107
    try {
108
        // Check the usename as passed in
109
        authenticated = ldapAuthenticate(identifier, password);
110
        // if not found, try looking up a valid DN then auth again
111
        if (!authenticated) {
112
            MetaCatUtil.debugMessage("Looking up DN for: " + identifier);
113
            identifier = getIdentifyingName(identifier,ldapUrl,ldapBase);
114
            //System.out.println("identifier: "+identifier);
115
            String refUrl = "";
116
            String refBase = "";
117
            if (identifier.startsWith("ldap")) {
118
                refUrl = identifier.substring(0,
119
                               identifier.lastIndexOf("/")+1);
120
                MetaCatUtil.debugMessage("Ref ldapUrl: " + refUrl);
121
                int position = identifier.indexOf(",");
122
                int position2 = identifier.indexOf(",", position+1);
123
                refBase = identifier.substring(position2+1);
124
                MetaCatUtil.debugMessage("Ref ldapBase: " + refBase);
125
                identifier = identifier.substring(
126
                             identifier.lastIndexOf("/")+1);
127
                MetaCatUtil.debugMessage("Trying: " + identifier);
128
                authenticated = ldapAuthenticate(identifier, password,
129
                                                 refUrl, refBase);
130
            } else {
131
                identifier = identifier+","+ldapBase;
132
                MetaCatUtil.debugMessage("Trying: " + identifier);
133
                authenticated = ldapAuthenticate(identifier, password);
134
            }
135
            //authenticated = ldapAuthenticate(identifier, password);
136
        }
137
     
138
    } catch (NullPointerException e) {
139
      util.debugMessage("NullPointerException b' password is null");
140
      util.debugMessage("NullPointerException while authenticating in " + 
141
                        "AuthLdap.authenticate: " + e);
142
      throw new ConnectException(
143
      "NullPointerException while authenticating in " + 
144
                        "AuthLdap.authenticate: " + e);
145
    } catch (NamingException e) {
146
      util.debugMessage("Naming exception while authenticating in " + 
147
                        "AuthLdap.authenticate: " + e);
148
      e.printStackTrace();
149
    } catch (Exception e) {
150
      util.debugMessage(e.getMessage());
151
    }
152
    return authenticated;
153
  }
154

    
155
  /**
156
   * Connect to the LDAP directory and do the authentication using the
157
   * username and password as passed into the routine.
158
   *
159
   * @param identifier the distinguished name to check against LDAP
160
   * @param password the password for authentication
161
   */
162
  private boolean ldapAuthenticate(String identifier, String password)
163
            throws ConnectException, NamingException, NullPointerException
164
  {
165
      return ldapAuthenticate(identifier, password, 
166
                              this.ldapsUrl, this.ldapBase);
167
  }
168

    
169
  /**
170
   * Connect to the LDAP directory and do the authentication using the
171
   * username and password as passed into the routine.
172
   *
173
   * @param identifier the distinguished name to check against LDAP
174
   * @param password the password for authentication
175
   */
176
  private boolean ldapAuthenticate(String identifier, String password,
177
            String directoryUrl, String searchBase)
178
            throws ConnectException, NamingException, NullPointerException
179
  {
180
    double totStartTime = System.currentTimeMillis();
181
    boolean authenticated = false;
182
    if (identifier != null && !password.equals("")) 
183
    {
184
    
185
        //Pass the username and password to run() method
186
        userName=identifier;
187
        userPassword=password;
188
        // Identify service provider to use
189
        Hashtable env = new Hashtable(11);
190
        env.put(Context.INITIAL_CONTEXT_FACTORY, 
191
              "com.sun.jndi.ldap.LdapCtxFactory");
192
        env.put(Context.REFERRAL, "throw");
193
        env.put(Context.PROVIDER_URL, directoryUrl + searchBase);
194
        if ( !ldapsUrl.equals(ldapUrl) ) 
195
        {
196
          // ldap is set on default port 389
197
          // ldaps is set on second port - 636 by default
198
          //env.put(Context.SECURITY_PROTOCOL, "ssl");
199
        }
200
        env.put(Context.SECURITY_AUTHENTICATION, "simple");
201
        env.put(Context.SECURITY_PRINCIPAL, identifier);
202
        env.put(Context.SECURITY_CREDENTIALS, password);
203
        // If our auth credentials are invalid, an exception will be thrown
204
        DirContext ctx = null;
205
        try 
206
        {
207
          double startTime = System.currentTimeMillis();
208
          //Here to check the autheration
209
          ctx = new InitialDirContext(env);
210
          double stopTime = System.currentTimeMillis();
211
          util.debugMessage("Connection time thru " + ldapsUrl + " was: " +
212
                             (stopTime-startTime)/1000 + " seconds.");
213
          authenticated = true;
214
          //tls.close();
215
          ctx.close();
216
          this.ldapUrl = ldapUrl;
217
          this.ldapBase = ldapBase;
218
          //break;
219
        } 
220
        catch (AuthenticationException ae) 
221
        {
222
          authenticated = false;
223
          if ( ctx != null ) 
224
          {
225
            ctx.close();
226
          }
227
        } 
228
        catch (javax.naming.InvalidNameException ine) 
229
        {
230
            util.debugMessage("An invalid DN was provided!");
231
        } 
232
        catch (javax.naming.ReferralException re) 
233
        {
234
          util.debugMessage("referral during authentication");
235
          util.debugMessage("Referral information: "+re.getReferralInfo());
236
          try
237
          {
238
            refExc = re;
239

    
240
            Thread t = new Thread(this);
241
            t.start();
242
            Thread.sleep(5000); //this is a manual override of ldap's 
243
                                //hideously long time out period.
244
            util.debugMessage("Awake after 5 seconds.");
245
            if (referralContext == null)
246
            {
247
              t.interrupt();
248
              authenticated = false;
249
            }
250
            else
251
            {
252
              authenticated = true;
253

    
254
            }
255
          }
256
          catch (Exception e)
257
          {
258
            authenticated = false;
259
          }
260
        }
261
    } 
262
    else 
263
    { 
264
        util.debugMessage("User not found");
265
    }
266
    double totStopTime = System.currentTimeMillis();
267
    util.debugMessage("total ldap authentication time: " + 
268
                      (totStopTime - totStartTime)/1000 + " seconds");
269
    return authenticated;
270
  }
271
  
272
  
273

    
274
  /**
275
   * Get the identifying name for a given userid or name.  This is the name
276
   * that is used in conjunction withthe LDAP BaseDN to create a
277
   * distinguished name (dn) for the record
278
   *
279
   * @param user the user for which the identifying name is requested
280
   * @returns String the identifying name for the user, 
281
   *          or null if not found
282
   */
283
  private String getIdentifyingName(String user, String ldapUrl,
284
                                    String ldapBase) throws NamingException
285
  {
286
    String identifier = null;
287
    // Identify service provider to use
288
    Hashtable env = new Hashtable(11);
289
    env.put(Context.INITIAL_CONTEXT_FACTORY,
290
            "com.sun.jndi.ldap.LdapCtxFactory");
291
    util.debugMessage("setting referrals to: " + referral);
292
    env.put(Context.REFERRAL, referral);
293
    env.put(Context.PROVIDER_URL, ldapUrl + ldapBase);
294
    //    non-secure LDAP context; dn are publicly readable
295
    try {
296
      
297
      // Bind to the LDAP server, in order to search for the right
298
      // distinguished name (dn) based on userid (uid) or common name (cn)
299
      DirContext ctx = new InitialDirContext(env);
300
      SearchControls ctls = new SearchControls();
301
      ctls.setSearchScope(SearchControls.SUBTREE_SCOPE);
302
      // Search for the user id or name using the uid, then cn and sn 
303
      //attributes
304
      // If we find a record, determine the dn for the record
305
      // The following blocks need to be refactored into a subroutine
306
      // they have a ridiculous amount of redundancy
307

    
308
      // Parse out the uid and org components and look up the real DN
309
      // This assumes a dn like "uid=x,o=y,dc=someinst,dc=org"
310
      int position = user.indexOf(",");
311
      String comp1 = user.substring(0, position);
312
      MetaCatUtil.debugMessage("First comp is: " + comp1);
313
      String comp2 = user.substring(position+1, 
314
                                    user.indexOf(",", position+1));
315
      MetaCatUtil.debugMessage("Second comp is: " + comp2);
316

    
317
      String filter = "(&(" + comp1 + ")(" + comp2 + "))";
318
      MetaCatUtil.debugMessage("Filter is: " + filter);
319
      MetaCatUtil.debugMessage("Provider URL is: " + ldapUrl + ldapBase);
320
      NamingEnumeration answer;
321
      try {
322
        answer = ctx.search("", filter, ctls);
323
        if (answer.hasMore()) {
324
          SearchResult sr = (SearchResult)answer.next();
325
          identifier = sr.getName();
326
          util.debugMessage("Originally Found: " + identifier);
327
          return identifier;
328
        }
329
      } catch (InvalidSearchFilterException e) {}
330

    
331
      // That failed, so check if it is just a username
332
      filter = "(" + user + ")";
333
      try {
334
        answer = ctx.search("", filter, ctls);
335
        if (answer.hasMore()) {
336
          SearchResult sr = (SearchResult)answer.next();
337
          identifier = sr.getName();
338
          if ( !sr.isRelative() ) { 
339
            this.ldapUrl = identifier.substring(0,
340
                                                identifier.lastIndexOf("/")+1);
341
            this.ldapBase = identifier.substring(identifier.indexOf(",")+1);
342
            identifier = identifier.substring(identifier.lastIndexOf("/")+1,
343
                                              identifier.indexOf(","));
344
          }
345
          util.debugMessage("Found: " + identifier);
346
          return identifier;
347
        }
348
      } catch (InvalidSearchFilterException e) {}
349

    
350
      // Maybe its a user id (uid)
351
      filter = "(uid=" + user + ")";
352
      answer = ctx.search("", filter, ctls);
353
      if (answer.hasMore()) {
354
        SearchResult sr = (SearchResult)answer.next();
355
        identifier = sr.getName();
356
        if ( !sr.isRelative() ) { 
357
          this.ldapUrl = identifier.substring(0,identifier.lastIndexOf("/")+1);
358
          this.ldapBase = identifier.substring(identifier.indexOf(",")+1);
359
          identifier = identifier.substring(identifier.lastIndexOf("/")+1,
360
                                            identifier.indexOf(","));
361
        }
362
        util.debugMessage("Found: " + identifier);
363
      } else {
364

    
365
        // maybe its just a common name
366
        filter = "(cn=" + user + ")";
367
        NamingEnumeration answer2 = ctx.search("", filter, ctls);
368
        if (answer2.hasMore()) {
369
          SearchResult sr = (SearchResult)answer2.next();
370
          identifier = sr.getName();
371
          if ( !sr.isRelative() ) { 
372
            this.ldapUrl = identifier.substring(0,
373
                                                identifier.lastIndexOf("/")+1);
374
            this.ldapBase = identifier.substring(identifier.indexOf(",")+1);
375
            identifier = identifier.substring(identifier.lastIndexOf("/")+1,
376
                                              identifier.indexOf(","));
377
          }
378
          util.debugMessage("Found: " + identifier);
379
        } else {
380

    
381
          // ok, last resort, is it a surname?
382
          filter = "(sn=" + user + ")";
383
          NamingEnumeration answer3 = ctx.search("", filter, ctls);
384
          if (answer3.hasMore()) {
385
            SearchResult sr = (SearchResult)answer3.next();
386
            identifier = sr.getName();
387
            if ( !sr.isRelative() ) { 
388
              this.ldapUrl = identifier.substring(0,
389
                                                identifier.lastIndexOf("/")+1);
390
              this.ldapBase = identifier.substring(identifier.indexOf(",")+1);
391
              identifier = identifier.substring(identifier.lastIndexOf("/")+1,
392
                                                identifier.indexOf(","));
393
            }
394
            util.debugMessage("Found: " + identifier);
395
          }
396
        }
397
      }
398
      // Close the context when we're done the initial search
399
      ctx.close();
400
    } catch (NamingException e) {
401
      util.debugMessage("Naming exception while getting dn: " + e);
402
      throw new NamingException(
403
      "Naming exception in AuthLdap.getIdentifyingName: " + e);
404
    }
405
    return identifier;
406
  }
407

    
408
  /**
409
   * Get all users from the authentication service
410
   *
411
   * @param user the user for authenticating against the service
412
   * @param password the password for authenticating against the service
413
   * @returns string array of all of the user names
414
   */
415
  public String[] getUsers(String user, String password) 
416
         throws ConnectException
417
  {
418
    String[] users = null;
419

    
420
    // Identify service provider to use
421
    Hashtable env = new Hashtable(11);
422
    env.put(Context.INITIAL_CONTEXT_FACTORY, 
423
            "com.sun.jndi.ldap.LdapCtxFactory");
424
    env.put(Context.REFERRAL, referral);
425
    env.put(Context.PROVIDER_URL, ldapUrl);
426
 
427
    try {
428

    
429
        // Create the initial directory context
430
        DirContext ctx = new InitialDirContext(env);
431

    
432
        // Specify the attributes to match.
433
        // Users are objects that have the attribute objectclass=InetOrgPerson.
434
        SearchControls ctls = new SearchControls();
435
        String[] attrIDs = {"dn"};
436
        ctls.setReturningAttributes(attrIDs);
437
        ctls.setSearchScope(SearchControls.SUBTREE_SCOPE);
438
        //ctls.setCountLimit(1000);
439
        String filter = "(objectClass=inetOrgPerson)";
440
        NamingEnumeration enum = ctx.search(ldapBase, filter, ctls);
441
        
442
        // Store the users in a vector
443
        Vector uvec = new Vector();
444
        try {
445
            while (enum.hasMore()) {
446
                SearchResult sr = (SearchResult)enum.next();
447
                uvec.add(sr.getName()+","+ldapBase);
448
            }
449
        } catch (SizeLimitExceededException slee) {
450
            util.debugMessage("LDAP Server size limit exceeded. " +
451
                    "Returning incomplete record set.");
452
        }
453

    
454
        // initialize users[]; fill users[]
455
        users = new String[uvec.size()];
456
        for (int i=0; i < uvec.size(); i++) {
457
          users[i] = (String)uvec.elementAt(i); 
458
        }
459

    
460
        // Close the context when we're done
461
        ctx.close();
462

    
463
    } catch (NamingException e) {
464
      util.debugMessage("Problem getting users in AuthLdap.getUsers:" + e);
465
      e.printStackTrace(System.err);
466
      throw new ConnectException(
467
      "Problem getting users in AuthLdap.getUsers:" + e);
468
    }
469

    
470
    return users;
471
  }
472

    
473
  /**
474
   * Get the users for a particular group from the authentication service
475
   *
476
   * @param user the user for authenticating against the service
477
   * @param password the password for authenticating against the service
478
   * @param group the group whose user list should be returned
479
   * @returns string array of the user names belonging to the group
480
   */
481
  public String[] getUsers(String user, String password, String group) 
482
         throws ConnectException
483
  {
484
    String[] users = null;
485

    
486
    // Identify service provider to use
487
    Hashtable env = new Hashtable(11);
488
    env.put(Context.INITIAL_CONTEXT_FACTORY, 
489
            "com.sun.jndi.ldap.LdapCtxFactory");
490
    env.put(Context.REFERRAL, referral);
491
    env.put(Context.PROVIDER_URL, ldapUrl);
492

    
493
    try {
494

    
495
        // Create the initial directory context
496
        DirContext ctx = new InitialDirContext(env);
497

    
498
        // Specify the ids of the attributes to return
499
        String[] attrIDs = {"uniqueMember"};
500

    
501
        Attributes answer = ctx.getAttributes(group, attrIDs);
502

    
503
        Vector uvec = new Vector();
504
        try {
505
            for (NamingEnumeration ae = answer.getAll(); ae.hasMore();) {
506
                Attribute attr = (Attribute)ae.next();
507
                for (NamingEnumeration e = attr.getAll(); 
508
                     e.hasMore();
509
                     uvec.add(e.next()) 
510
                    );
511
            }
512
        } catch (SizeLimitExceededException slee) {
513
            util.debugMessage("LDAP Server size limit exceeded. " +
514
                    "Returning incomplete record set.");
515
        }
516

    
517
        // initialize users[]; fill users[]
518
        users = new String[uvec.size()];
519
        for (int i=0; i < uvec.size(); i++) {
520
          users[i] = (String)uvec.elementAt(i); 
521
        }
522

    
523
        // Close the context when we're done
524
        ctx.close();
525

    
526
    } catch (NamingException e) {
527
      util.debugMessage("Problem getting users for a group in " +
528
              "AuthLdap.getUsers:" + e);
529
      throw new ConnectException(
530
      "Problem getting users for a group in AuthLdap.getUsers:" + e);
531
    }
532

    
533
    return users;
534
  }
535

    
536
  /**
537
   * Get all groups from the authentication service
538
   *
539
   * @param user the user for authenticating against the service
540
   * @param password the password for authenticating against the service
541
   * @returns string array of the group names
542
   */
543
  public String[] getGroups(String user, String password) 
544
         throws ConnectException
545
  {
546
      return getGroups(user, password, null);
547
  }
548

    
549
  /**
550
   * Get the groups for a particular user from the authentication service
551
   *
552
   * @param user the user for authenticating against the service
553
   * @param password the password for authenticating against the service
554
   * @param foruser the user whose group list should be returned
555
   * @returns string array of the group names
556
   */
557
  public String[] getGroups(String user, String password, String foruser) 
558
         throws ConnectException
559
  {
560
    Vector uvec = new Vector();
561
    //Pass the username and password to run() method
562
    userName=user;
563
    userPassword=password;
564
    // Identify service provider to use
565
    env.put(Context.INITIAL_CONTEXT_FACTORY, 
566
            "com.sun.jndi.ldap.LdapCtxFactory");
567
    env.put(Context.REFERRAL, "throw");
568
    env.put(Context.PROVIDER_URL, ldapUrl);
569
    try {
570
        // Create the initial directory context
571
        DirContext ctx = new InitialDirContext(env);
572
        // Specify the ids of the attributes to return
573
        String[] attrIDs = {"cn"};
574
        // Specify the attributes to match.
575
        // Groups are objects with attribute objectclass=groupofuniquenames.
576
        // and have attribute uniquemember: uid=foruser,ldapbase.
577
        SearchControls ctls = new SearchControls();
578
        ctls.setReturningAttributes(attrIDs);
579
        ctls.setSearchScope(SearchControls.SUBTREE_SCOPE);
580
        
581
        String filter = null;
582
        String gfilter = "(objectClass=groupOfUniqueNames)";
583
        if (null == foruser) {
584
            filter = gfilter;
585
        } else {
586
            filter = "(& " + gfilter + "(uniqueMember=" + foruser + "))";
587
        }
588
        MetaCatUtil.debug("searching for groups: " + filter);
589
        NamingEnumeration enum = ctx.search(ldapBase, filter, ctls);
590

    
591
        // Print the groups
592
        MetaCatUtil.debug("getting group results.");
593
        while (enum.hasMore()) {
594
          SearchResult sr = (SearchResult)enum.next();          
595
          uvec.add(sr.getName()+","+ldapBase);
596
          MetaCatUtil.debug("group " + sr.getName() + " added to Group vector");
597
        }
598
        // Close the context when we're done
599
        ctx.close();
600

    
601
    } 
602
    catch (ReferralException re) 
603
    {
604
      refExc = re;
605
      Thread t = new Thread(new GetGroup());
606
      util.debugMessage("Starting thread...");
607
      t.start();
608
      util.debugMessage("sleeping for 5 seconds.");
609
      try
610
      {
611
        Thread.sleep(5000);
612
      }
613
      catch(InterruptedException ie)
614
      {
615
        MetaCatUtil.debug("main thread interrupted: " + ie.getMessage());
616
      }
617
      //this is a manual override of jndi's hideously long time
618
      //out period.
619
      util.debugMessage("Awake after 5 seconds.");
620
      if (referralContext == null)
621
      {
622
        util.debug("thread timed out...returning groups: " + uvec.toString());
623
        String groups[] = new String[uvec.size()];
624
        for(int i=0; i<uvec.size(); i++)
625
        {
626
          groups[i] = (String)uvec.elementAt(i);
627
        }
628
        t.interrupt();
629
        return groups;
630
      }
631
      DirContext dc = (DirContext)referralContext;
632
      String[] attrIDs = {"cn"};
633
      // Specify the attributes to match.
634
      // Groups are objects with attribute objectclass=groupofuniquenames.
635
      // and have attribute uniquemember: uid=foruser,ldapbase.
636
      SearchControls ctls = new SearchControls();
637
      ctls.setReturningAttributes(attrIDs);
638
      ctls.setSearchScope(SearchControls.SUBTREE_SCOPE);
639
      
640
      String filter = null;
641
      String gfilter = "(objectClass=groupOfUniqueNames)";
642
      if (null == foruser) {
643
          filter = gfilter;
644
      } else {
645
          filter = "(& " + gfilter + "(uniqueMember=" + foruser + "))";
646
      }
647
      
648
      try
649
      {
650
        NamingEnumeration enum = dc.search(ldapBase, filter, ctls);
651
        // Print the groups
652
        while (enum.hasMore()) {
653
          SearchResult sr = (SearchResult)enum.next();
654
          uvec.add(sr.getName()+","+ldapBase);
655
        }
656
        
657
        referralContext.close();
658
        dc.close();
659
      }
660
      catch(NamingException ne)
661
      {
662
        MetaCatUtil.debug("Naming Exception in AuthLdap.getGroups");
663
      }
664
    } catch (NamingException e) {
665
      e.printStackTrace(System.err);
666
      String groups[] = new String[uvec.size()];
667
      for(int i=0; i<uvec.size(); i++)
668
      {
669
        groups[i] = (String)uvec.elementAt(i);
670
      }
671
      return groups;
672
       /*throw new ConnectException(
673
      "Problem getting groups for a user in AuthLdap.getGroups:" + e);*/
674
    } 
675
    
676
    MetaCatUtil.debug("The user is in the following groups: " + uvec.toString());
677
    String groups[] = new String[uvec.size()];
678
    for(int i=0; i<uvec.size(); i++)
679
    {
680
      groups[i] = (String)uvec.elementAt(i);
681
    }
682
    return groups;
683
  }
684

    
685
  /**
686
   * Get attributes describing a user or group
687
   *
688
   * @param foruser the user for which the attribute list is requested
689
   * @returns HashMap a map of attribute name to a Vector of values
690
   */
691
  public HashMap getAttributes(String foruser) 
692
         throws ConnectException
693
  {
694
    return getAttributes(null, null, foruser);
695
  }
696

    
697
  /**
698
   * Get attributes describing a user or group
699
   *
700
   * @param user the user for authenticating against the service
701
   * @param password the password for authenticating against the service
702
   * @param foruser the user whose attributes should be returned
703
   * @returns HashMap a map of attribute name to a Vector of values
704
   */
705
  public HashMap getAttributes(String user, String password, String foruser) 
706
         throws ConnectException
707
  {
708
    HashMap attributes = new HashMap();
709
    String ldapUrl = this.ldapUrl;
710
    String ldapBase = this.ldapBase;
711
    String userident = foruser;
712
   
713
    // Identify service provider to use
714
    Hashtable env = new Hashtable(11);
715
    env.put(Context.INITIAL_CONTEXT_FACTORY, 
716
        "com.sun.jndi.ldap.LdapCtxFactory");
717
    env.put(Context.REFERRAL, referral);
718
    env.put(Context.PROVIDER_URL, ldapUrl);
719

    
720
    try {
721
      
722
      // Create the initial directory context
723
      DirContext ctx = new InitialDirContext(env);
724
        
725
      // Ask for all attributes of the user 
726
      //Attributes attrs = ctx.getAttributes(userident);
727
      Attributes attrs = ctx.getAttributes(foruser);
728
 
729
      // Print all of the attributes
730
      NamingEnumeration en = attrs.getAll();
731
      while (en.hasMore()) {
732
        Attribute att = (Attribute)en.next();
733
        Vector values = new Vector();
734
        String attName = att.getID();
735
        NamingEnumeration attvalues = att.getAll();
736
        while (attvalues.hasMore()) {
737
          String value = (String)attvalues.next();
738
          values.add(value);
739
        }
740
        attributes.put(attName, values);
741
      }
742
  
743
      // Close the context when we're done
744
      ctx.close();
745
    } catch (NamingException e) {
746
      util.debugMessage("Problem getting attributes in " + 
747
              "AuthLdap.getAttributes:" + e);
748
      throw new ConnectException(
749
      "Problem getting attributes in AuthLdap.getAttributes:" + e);
750
    }
751

    
752
    return attributes;
753
  }
754

    
755
  /**
756
   * Get list of all subtrees holding Metacat's groups and users
757
   * starting from the Metacat LDAP root, 
758
   * i.e. ldap://dev.nceas.ucsb.edu/dc=ecoinformatics,dc=org
759
   */
760
  private Hashtable getSubtrees(String user, String password,
761
                                String ldapUrl, String ldapBase)
762
                throws ConnectException
763
  {
764
    Hashtable trees = new Hashtable();
765

    
766
    // Identify service provider to use
767
    Hashtable env = new Hashtable(11);
768
    env.put(Context.INITIAL_CONTEXT_FACTORY, 
769
            "com.sun.jndi.ldap.LdapCtxFactory");
770
    env.put(Context.REFERRAL, referral);
771
    env.put(Context.PROVIDER_URL, ldapUrl + ldapBase);
772

    
773
    try {
774

    
775
        // Create the initial directory context
776
        DirContext ctx = new InitialDirContext(env);
777

    
778
        // Specify the ids of the attributes to return
779
        String[] attrIDs = {"o","ref"};
780
        SearchControls ctls = new SearchControls();
781
        ctls.setReturningAttributes(attrIDs);
782
        ctls.setSearchScope(SearchControls.SUBTREE_SCOPE);
783
          
784
        // Specify the attributes to match.
785
        // Subtrees from the main server are found as objects with attribute
786
        // objectclass=organization or objectclass=referral to the subtree
787
        // resided on other server.
788
        String filter = "(|(objectclass=organization)(objectclass=referral))";
789

    
790
        // Search for objects in the current context
791
        NamingEnumeration enum = ctx.search("", filter, ctls);
792

    
793
        // Print the subtrees' <ldapURL, baseDN>
794
        while (enum.hasMore()) {
795
          SearchResult sr = (SearchResult)enum.next();
796
          Attributes attrs = sr.getAttributes();
797
          NamingEnumeration enum1 = attrs.getAll(); // "dc" and "ref" attrs
798
          if (enum1.hasMore()) {
799
            Attribute attr = (Attribute)enum1.next();
800
            String attrValue = (String)attr.get();
801
            String attrName = (String)attr.getID();
802
 
803
            if ( enum1.hasMore() ) {
804
              attr = (Attribute)enum1.next();
805
              String refValue = (String)attr.get();
806
              String refName = (String)attr.getID();
807
               if ( ldapBase.startsWith(refName + "=" + refValue) ) {
808
                trees.put(ldapBase,
809
                         attrValue.substring(0,attrValue.lastIndexOf("/")+1) );
810
              } else {
811
                trees.put(refName + "=" + refValue + "," + ldapBase,
812
                         attrValue.substring(0,attrValue.lastIndexOf("/")+1) );
813
              }
814
 
815
            } else if ( ldapBase.startsWith(attrName + "=" + attrValue) ) {
816
                trees.put(ldapBase, ldapUrl);
817
            } else {              
818
                trees.put(attrName + "=" + attrValue + "," + ldapBase, ldapUrl);               
819
            }
820
          }
821
        }
822

    
823
        // Close the context when we're done
824
        ctx.close();
825

    
826
    } catch (NamingException e) {
827
      util.debugMessage("Problem getting subtrees in AuthLdap.getSubtrees:" 
828
                        + e);
829
      throw new ConnectException(
830
      "Problem getting subtrees in AuthLdap.getSubtrees:" + e);
831
    }
832

    
833
    return trees;
834
  }
835

    
836
  /**
837
   * Get all groups and users from authentication scheme.
838
   * The output is formatted in XML.
839
   * @param user the user which requests the information
840
   * @param password the user's password
841
   */
842
  public String getPrincipals(String user, String password)
843
                throws ConnectException
844
  {
845
    StringBuffer out = new StringBuffer();
846
    Vector usersIn = new Vector();
847
    
848
    out.append("<?xml version=\"1.0\"?>\n");
849
    out.append("<principals>\n");
850
    
851
    /*
852
     * get all subtrees first in the current dir context 
853
     * and then the Metacat users under them
854
     */
855
    Hashtable subtrees = getSubtrees(user,password,this.ldapUrl,this.ldapBase);
856
    
857
    Enumeration enum = subtrees.keys();
858
    while ( enum.hasMoreElements() ) {
859
      this.ldapBase = (String)enum.nextElement();
860
      this.ldapUrl = (String)subtrees.get(ldapBase);
861
      
862
      out.append("  <authSystem URI=\"" + 
863
                 this.ldapUrl + this.ldapBase + "\">\n");
864

    
865
      // get all groups for directory context
866
      String[] groups = getGroups(user, password);
867

    
868
      // for the groups and users that belong to them
869
      if ( groups!=null && groups.length > 0 ) {
870
        for (int i=0; i < groups.length; i++ ) {
871
          out.append("    <group>\n");
872
          out.append("      <groupname>" + groups[i] + "</groupname>\n");
873
          String[] usersForGroup = getUsers(user,password,groups[i]);
874
          for (int j=0; j < usersForGroup.length; j++ ) {
875
            usersIn.addElement(usersForGroup[j]);
876
            out.append("      <user>\n");
877
            out.append("        <username>" + usersForGroup[j] + 
878
                        "</username>\n");
879
            out.append("      </user>\n");
880
          }
881
          out.append("    </group>\n");
882
        }
883
      }
884
      // for the users not belonging to any group
885
      String[] users = getUsers(user, password);
886
      for (int j=0; j < users.length; j++ ) {
887
        if ( !usersIn.contains(users[j]) ) {
888
          out.append("    <user>\n");
889
          out.append("      <username>" + users[j] + "</username>\n");
890
          out.append("    </user>\n");
891
        }
892
      }
893
    
894
      out.append("  </authSystem>\n");
895
      if ( !usersIn.isEmpty() ) {
896
        usersIn.removeAllElements();
897
        usersIn.trimToSize();
898
      }
899
    
900
    }
901
    out.append("</principals>");
902
    return out.toString();
903
  }
904

    
905
  /**
906
   * Test method for the class
907
   */
908
  public static void main(String[] args) {
909

    
910
    // Provide a user, such as: "Matt Jones", or "jones"
911
    String user = args[0];
912
    String password = args[1];
913

    
914
    MetaCatUtil.debugMessage("Creating session...");
915
    AuthLdap authservice = new AuthLdap();
916
    MetaCatUtil.debugMessage("Session exists...");
917
                
918
    boolean isValid = false;
919
    try {
920
      MetaCatUtil.debugMessage("Authenticating...");
921
      isValid = authservice.authenticate(user, password);
922
      if (isValid) {
923
        MetaCatUtil.debugMessage("Authentication successful for: " + user );
924
      } else {
925
        MetaCatUtil.debugMessage("Authentication failed for: " + user);
926
      }
927

    
928
      // Get attributes for the user
929
      if (isValid) {
930
        MetaCatUtil.debugMessage("\nGetting attributes for user....");
931
        HashMap userInfo = authservice.getAttributes(user, password, user);
932
        // Print all of the attributes
933
        Iterator attList = (Iterator)(((Set)userInfo.keySet()).iterator());
934
        while (attList.hasNext()) {
935
          String att = (String)attList.next();
936
          Vector values = (Vector)userInfo.get(att);
937
          Iterator attvalues = values.iterator();
938
          while (attvalues.hasNext()) {
939
            String value = (String)attvalues.next();
940
            MetaCatUtil.debugMessage(att + ": " + value);
941
          }
942
        }
943
      }
944

    
945
      // get the groups
946
      if (isValid) {
947
        MetaCatUtil.debugMessage("\nGetting all groups....");
948
        String[] groups = authservice.getGroups(user, password);
949
        MetaCatUtil.debugMessage("Groups found: " + groups.length);
950
        for (int i=0; i < groups.length; i++) {
951
            MetaCatUtil.debugMessage("Group " + i + ": " + groups[i]);
952
        }
953
      }
954

    
955
      // get the groups for the user
956
      String savedGroup = null;
957
      if (isValid) {
958
        MetaCatUtil.debugMessage("\nGetting groups for user....");
959
        String[] groups = authservice.getGroups(user, password, user);
960
        MetaCatUtil.debugMessage("Groups found: " + groups.length);
961
        for (int i=0; i < groups.length; i++) {
962
            MetaCatUtil.debugMessage("Group " + i + ": " + groups[i]);
963
            savedGroup = groups[i];
964
        }
965
      }
966

    
967
      // get the users for a group
968
      if (isValid) {
969
        MetaCatUtil.debugMessage("\nGetting users for group....");
970
        MetaCatUtil.debugMessage("Group: " + savedGroup);
971
        String[] users = authservice.getUsers(user, password, savedGroup);
972
        MetaCatUtil.debugMessage("Users found: " + users.length);
973
        for (int i=0; i < users.length; i++) {
974
            MetaCatUtil.debugMessage("User " + i + ": " + users[i]);
975
        }
976
      }
977

    
978
      // get all users
979
      if (isValid) {
980
        MetaCatUtil.debugMessage("\nGetting all users ....");
981
        String[] users = authservice.getUsers(user, password);
982
        MetaCatUtil.debugMessage("Users found: " + users.length);
983
        
984
      }
985

    
986
      // get the whole list groups and users in XML format
987
      if (isValid) {
988
        MetaCatUtil.debugMessage("\nTrying principals....");
989
        authservice = new AuthLdap();
990
        String out = authservice.getPrincipals(user, password);
991
        java.io.File f = new java.io.File("principals.xml");
992
        java.io.FileWriter fw = new java.io.FileWriter(f);
993
        java.io.BufferedWriter buff = new java.io.BufferedWriter(fw);
994
        buff.write(out);
995
        buff.flush();
996
        buff.close();
997
        fw.close();
998
        MetaCatUtil.debugMessage("\nFinished getting principals.");
999
      }
1000

    
1001
    } catch (ConnectException ce) {
1002
      MetaCatUtil.debugMessage(ce.getMessage());
1003
    } catch (java.io.IOException ioe) {
1004
      MetaCatUtil.debugMessage("I/O Error writing to file principals.txt");
1005
    }
1006
  }
1007
  
1008
  /**
1009
   * This method will be called by start a thread.
1010
   * It can handle if a referral exception happend.
1011
   */ 
1012
  public void run()
1013
  {
1014
    referralContext = null;
1015
    DirContext refDirContext=null;
1016
    boolean moreReferrals=true;
1017
    String referralInfo=null;
1018
    //set a while loop is because we don't know if a referral excption contains
1019
    //another referral exception
1020
    while (moreReferrals)
1021
    {
1022
      try
1023
      {  
1024
        //revise environment variable
1025
        referralInfo=(String)refExc.getReferralInfo();
1026
        env.put(Context.PROVIDER_URL,refExc.getReferralInfo());
1027
        //System.out.println("referral info: "+referralInfo);
1028
        //if (referralInfo.indexOf("piscoweb")!=-1)
1029
        //{
1030
          //userName=uid+",o=PISCO,dc=piscoweb,dc=org";
1031
        //}
1032
        env.put(Context.INITIAL_CONTEXT_FACTORY, 
1033
                "com.sun.jndi.ldap.LdapCtxFactory");
1034
        //System.out.println("principal: "+userName);
1035
        env.put(Context.SECURITY_PRINCIPAL, userName);
1036
        env.put(Context.SECURITY_CREDENTIALS, userPassword);
1037
        env.put(Context.REFERRAL, "throw");
1038
        //get a context object for referral in the new envriment
1039
        rContext = refExc.getReferralContext(env);
1040
        //casting the context to dircontext and it will create a
1041
        //autherntication or naming exception if DN and password is incorrect
1042
        referralContext=rContext;
1043
        refDirContext=(DirContext)rContext;
1044
        refDirContext.close();
1045
        //get context and jump out the while loop
1046
        moreReferrals=false;
1047
      }//try
1048
      //if referral have another referral excption
1049
      catch (ReferralException re)
1050
      {
1051
        //keep running in while loop
1052
        moreReferrals=true;
1053
        //assign refExc to new referral exception re
1054
        refExc=re;
1055
      }
1056
      //catch a authentication exception
1057
      catch (AuthenticationException ae)
1058
      {
1059
        util.debugMessage("Error running referral handler thread: " + 
1060
                          ae.getMessage());
1061
        //check if has another referral
1062
        moreReferrals=refExc.skipReferral();
1063
        //don't get the context
1064
        referralContext = null;
1065
      }
1066
      //catch a naming exception
1067
      catch (NamingException ne)
1068
      {
1069
        util.debugMessage("Error running referral handler thread: " + 
1070
                          ne.getMessage());
1071
        //check if has another referral
1072
        moreReferrals=refExc.skipReferral();
1073
        //don't get context
1074
        referralContext = null;		
1075
      }
1076
    }//while
1077
  }//run()
1078
  
1079
  private class GetGroup implements Runnable
1080
  {
1081
    public void run()
1082
    {
1083
      referralContext = null;
1084
      MetaCatUtil.debug("getting groups context");
1085
      DirContext refDirContext=null;
1086
      boolean moreReferrals=true;
1087
      //set a while loop is because we don't know if a referral excption 
1088
      //contains another referral exception
1089
      while (moreReferrals)
1090
      {
1091
        try
1092
        {
1093
          //revise environment variable
1094
          String refInfo = null;
1095
          refInfo = (String)refExc.getReferralInfo();
1096
          if(refInfo != null)
1097
          {
1098
            MetaCatUtil.debug("Referral in thread to: " + 
1099
                              refInfo.toString());
1100
          }
1101
          else
1102
          {
1103
            MetaCatUtil.debug("getting refInfo Manually");
1104
            refInfo = (String)refExc.getReferralContext().getEnvironment().
1105
                                                  get(Context.PROVIDER_URL);
1106
          }
1107
          MetaCatUtil.debug("refInfo: " + refInfo);
1108
        
1109
          env.put(Context.INITIAL_CONTEXT_FACTORY, 
1110
              "com.sun.jndi.ldap.LdapCtxFactory");
1111
          env.put(Context.REFERRAL, "throw");
1112
          env.put(Context.PROVIDER_URL, refInfo);
1113
        
1114
          MetaCatUtil.debug("creating referralContext");
1115
          referralContext = new InitialDirContext(env);
1116
          MetaCatUtil.debug("referralContext created");
1117
          //get context and jump out the while loop
1118
          moreReferrals=false;
1119
        }//try
1120
        catch (ReferralException re)
1121
        {
1122
          //keep running in while loop
1123
          moreReferrals=true;
1124
          //assign refExc to new referral exception re
1125
          refExc=re;
1126
        }
1127
        catch (AuthenticationException ae)
1128
        {
1129
          util.debugMessage("Error running referral handler thread: " + 
1130
                          ae.getMessage());
1131
          //check if has another referral
1132
          moreReferrals=refExc.skipReferral();
1133
          //don't get the context
1134
          referralContext = null;
1135
        }
1136
        catch (NamingException ne)
1137
        {
1138
          util.debugMessage("Error running referral handler thread: " + 
1139
                          ne.getMessage());
1140
          //check if has another referral
1141
          moreReferrals=refExc.skipReferral();
1142
          //don't get context
1143
          referralContext = null;		
1144
        }
1145
      }//while
1146
    }//run()
1147
 }
1148
}
(7-7/41)