KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > apache > servicemix > jbi > security > login > PropertiesLoginModule


1 /*
2  * Licensed to the Apache Software Foundation (ASF) under one or more
3  * contributor license agreements. See the NOTICE file distributed with
4  * this work for additional information regarding copyright ownership.
5  * The ASF licenses this file to You under the Apache License, Version 2.0
6  * (the "License"); you may not use this file except in compliance with
7  * the License. You may obtain a copy of the License at
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */

17 package org.apache.servicemix.jbi.security.login;
18
19 import java.io.File JavaDoc;
20 import java.io.IOException JavaDoc;
21 import java.util.Enumeration JavaDoc;
22 import java.util.HashSet JavaDoc;
23 import java.util.Map JavaDoc;
24 import java.util.Properties JavaDoc;
25 import java.util.Set JavaDoc;
26
27 import javax.security.auth.Subject JavaDoc;
28 import javax.security.auth.callback.Callback JavaDoc;
29 import javax.security.auth.callback.CallbackHandler JavaDoc;
30 import javax.security.auth.callback.NameCallback JavaDoc;
31 import javax.security.auth.callback.PasswordCallback JavaDoc;
32 import javax.security.auth.callback.UnsupportedCallbackException JavaDoc;
33 import javax.security.auth.login.LoginException JavaDoc;
34 import javax.security.auth.login.FailedLoginException JavaDoc;
35 import javax.security.auth.spi.LoginModule JavaDoc;
36
37 import org.apache.commons.logging.Log;
38 import org.apache.commons.logging.LogFactory;
39 import org.apache.servicemix.jbi.security.GroupPrincipal;
40 import org.apache.servicemix.jbi.security.UserPrincipal;
41
42 /**
43  * JAAS Login module for user / password, based on two properties files.
44  *
45  */

46 public class PropertiesLoginModule implements LoginModule JavaDoc {
47
48     private final String JavaDoc USER_FILE = "org.apache.servicemix.security.properties.user";
49     private final String JavaDoc GROUP_FILE = "org.apache.servicemix.security.properties.group";
50
51     private static final Log log = LogFactory.getLog(PropertiesLoginModule.class);
52
53     private Subject JavaDoc subject;
54     private CallbackHandler JavaDoc callbackHandler;
55
56     private boolean debug;
57     private String JavaDoc usersFile;
58     private String JavaDoc groupsFile;
59     private Properties JavaDoc users = new Properties JavaDoc();
60     private Properties JavaDoc groups = new Properties JavaDoc();
61     private String JavaDoc user;
62     private Set JavaDoc principals = new HashSet JavaDoc();
63     private File JavaDoc baseDir;
64
65     public void initialize(Subject JavaDoc subject, CallbackHandler JavaDoc callbackHandler, Map JavaDoc sharedState, Map JavaDoc options) {
66         this.subject = subject;
67         this.callbackHandler = callbackHandler;
68
69         if( System.getProperty("java.security.auth.login.config")!=null ) {
70             baseDir=new File JavaDoc(System.getProperty("java.security.auth.login.config")).getParentFile();
71         } else {
72             baseDir = new File JavaDoc(".");
73         }
74
75         debug = "true".equalsIgnoreCase((String JavaDoc) options.get("debug"));
76         usersFile = (String JavaDoc) options.get(USER_FILE)+"";
77         groupsFile = (String JavaDoc) options.get(GROUP_FILE)+"";
78
79         if (debug) {
80             log.debug("Initialized debug=" + debug + " usersFile=" + usersFile + " groupsFile=" + groupsFile+" basedir="+baseDir);
81         }
82     }
83
84     public boolean login() throws LoginException JavaDoc {
85         File JavaDoc f = new File JavaDoc(baseDir,usersFile);
86         try {
87             users.load(new java.io.FileInputStream JavaDoc(f));
88         } catch (IOException JavaDoc ioe) {
89             throw new LoginException JavaDoc("Unable to load user properties file " + f);
90         }
91         f = new File JavaDoc(baseDir, groupsFile);
92         try {
93             groups.load(new java.io.FileInputStream JavaDoc(f));
94         } catch (IOException JavaDoc ioe) {
95             throw new LoginException JavaDoc("Unable to load group properties file " + f);
96         }
97
98         Callback JavaDoc[] callbacks = new Callback JavaDoc[2];
99
100         callbacks[0] = new NameCallback JavaDoc("Username: ");
101         callbacks[1] = new PasswordCallback JavaDoc("Password: ", false);
102         try {
103             callbackHandler.handle(callbacks);
104         } catch (IOException JavaDoc ioe) {
105             throw new LoginException JavaDoc(ioe.getMessage());
106         } catch (UnsupportedCallbackException JavaDoc uce) {
107             throw new LoginException JavaDoc(uce.getMessage() + " not available to obtain information from user");
108         }
109         user = ((NameCallback JavaDoc) callbacks[0]).getName();
110         char[] tmpPassword = ((PasswordCallback JavaDoc) callbacks[1]).getPassword();
111         if (tmpPassword == null) tmpPassword = new char[0];
112
113         String JavaDoc password = users.getProperty(user);
114
115         if (password == null) throw new FailedLoginException JavaDoc("User does not exist");
116         if (!password.equals(new String JavaDoc(tmpPassword))) throw new FailedLoginException JavaDoc("Password does not match");
117
118         users.clear();
119
120         if (debug) {
121             log.debug("login " + user);
122         }
123         return true;
124     }
125
126     public boolean commit() throws LoginException JavaDoc {
127         principals.add(new UserPrincipal(user));
128
129         for (Enumeration JavaDoc enumeration = groups.keys(); enumeration.hasMoreElements();) {
130             String JavaDoc name = (String JavaDoc) enumeration.nextElement();
131             String JavaDoc[] userList = ((String JavaDoc) groups.getProperty(name) + "").split(",");
132             for (int i = 0; i < userList.length; i++) {
133                 if (user.equals(userList[i])) {
134                     principals.add(new GroupPrincipal(name));
135                     break;
136                 }
137             }
138         }
139
140         subject.getPrincipals().addAll(principals);
141
142         clear();
143
144         if (debug) {
145             log.debug("commit");
146         }
147         return true;
148     }
149
150     public boolean abort() throws LoginException JavaDoc {
151         clear();
152
153         if (debug) {
154             log.debug("abort");
155         }
156         return true;
157     }
158
159     public boolean logout() throws LoginException JavaDoc {
160         subject.getPrincipals().removeAll(principals);
161         principals.clear();
162
163         if (debug) {
164             log.debug("logout");
165         }
166         return true;
167     }
168
169     private void clear() {
170         groups.clear();
171         user = null;
172     }
173 }
174
Popular Tags