View Javadoc
1   //
2   //  ========================================================================
3   //  Copyright (c) 1995-2016 Mort Bay Consulting Pty. Ltd.
4   //  ------------------------------------------------------------------------
5   //  All rights reserved. This program and the accompanying materials
6   //  are made available under the terms of the Eclipse Public License v1.0
7   //  and Apache License v2.0 which accompanies this distribution.
8   //
9   //      The Eclipse Public License is available at
10  //      http://www.eclipse.org/legal/epl-v10.html
11  //
12  //      The Apache License v2.0 is available at
13  //      http://www.opensource.org/licenses/apache2.0.php
14  //
15  //  You may elect to redistribute this code under either of these licenses.
16  //  ========================================================================
17  //
18  
19  package org.eclipse.jetty.jaas.spi;
20  
21  import java.sql.Connection;
22  import java.util.Map;
23  
24  import javax.naming.InitialContext;
25  import javax.naming.NamingException;
26  import javax.security.auth.Subject;
27  import javax.security.auth.callback.CallbackHandler;
28  import javax.sql.DataSource;
29  
30  /**
31   * DataSourceLoginModule
32   *
33   * A LoginModule that uses a DataSource to retrieve user authentication
34   * and authorisation information.
35   *
36   * @see JDBCLoginModule
37   */
38  public class DataSourceLoginModule extends AbstractDatabaseLoginModule
39  {
40  
41      private String dbJNDIName;
42      private DataSource dataSource;
43  
44      /* ------------------------------------------------ */
45      /** 
46       * Init LoginModule.
47       * <p>
48       * Called once by JAAS after new instance created.
49       * 
50       * @param subject the subject
51       * @param callbackHandler the callback handler
52       * @param sharedState the shared state map
53       * @param options the option map
54       */
55      public void initialize(Subject subject,
56                             CallbackHandler callbackHandler,
57                             Map<String,?> sharedState,
58                             Map<String,?> options)
59      {
60          try
61          {
62              super.initialize(subject, callbackHandler, sharedState, options);
63  
64              //get the datasource jndi name
65              dbJNDIName = (String)options.get("dbJNDIName");
66  
67              InitialContext ic = new InitialContext();
68              dataSource = (DataSource)ic.lookup("java:comp/env/"+dbJNDIName);
69          }
70          catch (NamingException e)
71          {
72              throw new IllegalStateException (e.toString());
73          }
74      }
75  
76      /**
77       * Get a connection from the DataSource
78       * @see AbstractDatabaseLoginModule#getConnection()
79       * @return the connection for the datasource
80       * @throws Exception if unable to get the connection
81       */
82      public Connection getConnection ()
83      throws Exception
84      {
85          return dataSource.getConnection();
86      }
87  }