KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > columba > ristretto > auth > mechanism > DigestMD5Mechanism


1 /* ***** BEGIN LICENSE BLOCK *****
2  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
3  *
4  * The contents of this file are subject to the Mozilla Public License Version
5  * 1.1 (the "License"); you may not use this file except in compliance with
6  * the License. You may obtain a copy of the License at
7  * http://www.mozilla.org/MPL/
8  *
9  * Software distributed under the License is distributed on an "AS IS" basis,
10  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
11  * for the specific language governing rights and limitations under the
12  * License.
13  *
14  * The Original Code is Ristretto Mail API.
15  *
16  * The Initial Developers of the Original Code are
17  * Timo Stich and Frederik Dietz.
18  * Portions created by the Initial Developers are Copyright (C) 2004
19  * All Rights Reserved.
20  *
21  * Contributor(s):
22  *
23  * Alternatively, the contents of this file may be used under the terms of
24  * either the GNU General Public License Version 2 or later (the "GPL"), or
25  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
26  * in which case the provisions of the GPL or the LGPL are applicable instead
27  * of those above. If you wish to allow use of your version of this file only
28  * under the terms of either the GPL or the LGPL, and not to allow others to
29  * use your version of this file under the terms of the MPL, indicate your
30  * decision by deleting the provisions above and replace them with the notice
31  * and other provisions required by the GPL or the LGPL. If you do not delete
32  * the provisions above, a recipient may use your version of this file under
33  * the terms of any one of the MPL, the GPL or the LGPL.
34  *
35  * ***** END LICENSE BLOCK ***** */
36 package org.columba.ristretto.auth.mechanism;
37
38 import java.io.IOException JavaDoc;
39 import java.util.Hashtable JavaDoc;
40 import java.util.logging.ConsoleHandler JavaDoc;
41 import java.util.logging.Level JavaDoc;
42 import java.util.logging.Logger JavaDoc;
43
44 import javax.security.auth.callback.Callback JavaDoc;
45 import javax.security.auth.callback.CallbackHandler JavaDoc;
46 import javax.security.auth.callback.NameCallback JavaDoc;
47 import javax.security.auth.callback.PasswordCallback JavaDoc;
48 import javax.security.auth.callback.TextOutputCallback JavaDoc;
49 import javax.security.auth.callback.UnsupportedCallbackException JavaDoc;
50
51 import org.columba.ristretto.auth.AuthenticationException;
52 import org.columba.ristretto.auth.AuthenticationMechanism;
53 import org.columba.ristretto.auth.AuthenticationServer;
54
55 /**
56  * Implementation of the Digest-MD5 SASL AuthenticationMechanism.
57  * The implementation is based on the jsse.jar delivered
58  * starting with JRE 1.4.2.
59  *
60  * @author Timo Stich <tstich@users.sourceforge.net>
61  */
62 public class DigestMD5Mechanism implements
63             AuthenticationMechanism,
64             CallbackHandler JavaDoc {
65
66     /** JDK 1.4+ logging framework logger, used for logging. */
67     private static final Logger JavaDoc LOG = Logger.getLogger("org.columba.ristretto.auth.mechanism");
68
69     private String JavaDoc username;
70     private char[] password;
71     
72     SaslWrapper sasl;
73     
74     
75     /**
76      * @see org.columba.ristretto.auth.AuthenticationMechanism#authenticate(org.columba.ristretto.auth.AuthenticationServer, java.lang.String, char[])
77      */
78     public void authenticate(AuthenticationServer server, String JavaDoc user,
79             char[] password) throws IOException JavaDoc, AuthenticationException {
80         
81         this.username = user;
82         this.password = password;
83         try {
84             sasl = new SaslWrapper();
85             sasl.createClient(new String JavaDoc[]{"DIGEST-MD5"},
86                     user, server.getService(), server.getHostName(),
87                     new Hashtable JavaDoc(), this);
88             byte[] serverChallenge = server.authReceive();
89             LOG.fine(new String JavaDoc(serverChallenge));
90             
91             byte[] response = sasl.evaluateChallenge(serverChallenge);
92             LOG.fine(new String JavaDoc(response));
93             server.authSend(response);
94             
95             serverChallenge = server.authReceive();
96             LOG.fine(new String JavaDoc(serverChallenge));
97             response = sasl.evaluateChallenge(serverChallenge);
98             if (!sasl.isComplete()) {
99                 throw new AuthenticationException("Authentication failed");
100             }
101             server.authSend(new byte[0]);
102         } catch (Exception JavaDoc e) {
103             if( e.getCause() instanceof IOException JavaDoc ) throw (IOException JavaDoc)e.getCause();
104             else throw new AuthenticationException(e.getCause());
105         }
106     }
107     
108     /**
109      * @see javax.security.auth.callback.CallbackHandler#handle(javax.security.auth.callback.Callback[])
110      */
111     public void handle(Callback JavaDoc[] callbacks) throws IOException JavaDoc, UnsupportedCallbackException JavaDoc {
112         for (int i = 0; i < callbacks.length; i++) {
113             if (callbacks[i] instanceof TextOutputCallback JavaDoc) {
114                 // display the message according to the specified type
115 TextOutputCallback JavaDoc toc = (TextOutputCallback JavaDoc) callbacks[i];
116                 switch (toc.getMessageType()) {
117                     case TextOutputCallback.INFORMATION :
118                         LOG.info(toc.getMessage());
119                         break;
120                     case TextOutputCallback.ERROR :
121                         LOG.severe(toc.getMessage());
122                         break;
123                     case TextOutputCallback.WARNING :
124                         LOG.warning(toc.getMessage());
125                         break;
126                     default :
127                         LOG.fine(toc.getMessage());
128                 }
129             } else if (callbacks[i] instanceof NameCallback JavaDoc) {
130                 NameCallback JavaDoc nc = (NameCallback JavaDoc) callbacks[i];
131                 nc.setName(username);
132             } else if (callbacks[i] instanceof PasswordCallback JavaDoc) {
133                 PasswordCallback JavaDoc pc = (PasswordCallback JavaDoc) callbacks[i];
134                 pc.setPassword(password);
135             } else if (sasl.handle(callbacks[i]) ) {
136             } else {
137                 throw new UnsupportedCallbackException JavaDoc(callbacks[i],
138                         "Unrecognized Callback");
139             }
140         }
141     }
142 }
143
Popular Tags