jdk/test/javax/management/security/TestSampleLoginModule.java
author olagneau
Fri, 18 Dec 2015 17:42:06 +0100
changeset 34827 de746e7eb1e3
permissions -rw-r--r--
8058865: JMX Test Refactoring Reviewed-by: jbachorik
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
34827
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
     1
/*
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
     2
 * Copyright (c) 2006, 2015, Oracle and/or its affiliates. All rights reserved.
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
     4
 *
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
     7
 * published by the Free Software Foundation.
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
     8
 *
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
     9
 * This code is distributed in the hope that it will be useful, but WITHOUT
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    10
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    11
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    12
 * version 2 for more details (a copy is included in the LICENSE file that
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    13
 * accompanied this code).
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    14
 *
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    15
 * You should have received a copy of the GNU General Public License version
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    16
 * 2 along with this work; if not, write to the Free Software Foundation,
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    17
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    18
 *
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    19
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    20
 * or visit www.oracle.com if you need additional information or have any
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    21
 * questions.
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    22
 */
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    23
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    24
import java.util.Map;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    25
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    26
import javax.security.auth.Subject;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    27
import javax.security.auth.callback.Callback;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    28
import javax.security.auth.callback.CallbackHandler;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    29
import javax.security.auth.callback.NameCallback;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    30
import javax.security.auth.callback.PasswordCallback;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    31
import javax.security.auth.login.LoginException;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    32
import javax.security.auth.spi.LoginModule;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    33
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    34
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    35
public final class TestSampleLoginModule implements LoginModule {
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    36
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    37
    private Subject subject;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    38
    private CallbackHandler callbackHandler;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    39
    private Map<String, ?> sharedState;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    40
    private Map<String, ?> options;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    41
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    42
    public TestSampleLoginModule() {
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    43
    }
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    44
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    45
    public void initialize(Subject subject,
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    46
            CallbackHandler callbackHandler,
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    47
            Map<String,?> sharedState,
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    48
            Map<String,?> options) {
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    49
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    50
        this.subject = subject;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    51
        this.callbackHandler = callbackHandler;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    52
        this.sharedState = sharedState;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    53
        this.options = options;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    54
    }
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    55
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    56
  /*
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    57
   * Authenticate the user by comparing the values of the java properties
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    58
   * (username and password) against the values of the credentials.
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    59
   * */
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    60
    public boolean login() throws LoginException {
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    61
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    62
        String credentials_username = null;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    63
        String credentials_password = null;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    64
        String authenticated_username = System.getProperty("susername");
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    65
        String authenticated_password = System.getProperty("spassword");
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    66
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    67
        System.out.println("TestSampleLoginModule::login: Start");
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    68
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    69
        // First retreive the credentials {username, password} from
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    70
        // the callback handler
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    71
        Callback[] callbacks = new Callback[2];
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    72
        callbacks[0] = new NameCallback("username");
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    73
        callbacks[1] = new PasswordCallback("password", false);
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    74
        try {
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    75
            callbackHandler.handle(callbacks);
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    76
            credentials_username = ((NameCallback)callbacks[0]).getName();
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    77
            credentials_password = new String(((PasswordCallback)callbacks[1]).
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    78
                    getPassword());
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    79
        } catch (Exception e) {
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    80
            throw new LoginException(e.toString());
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    81
        }
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    82
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    83
        System.out.println("TestSampleLoginModule::login: credentials username = " +
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    84
                credentials_username);
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    85
        System.out.println("TestSampleLoginModule::login: credentials password = " +
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    86
                credentials_password);
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    87
        System.out.println("TestSampleLoginModule::login: authenticated username = " +
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    88
                authenticated_username);
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    89
        System.out.println("TestSampleLoginModule::login: authenticated password = " +
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    90
                authenticated_password);
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    91
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    92
        if (credentials_username.equals(authenticated_username) &&
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    93
                credentials_password.equals(authenticated_password)) {
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    94
            System.out.println("TestSampleLoginModule::login: " +
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    95
                    "Authentication should succeed");
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    96
            return true;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    97
        } else {
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    98
            System.out.println("TestSampleLoginModule::login: " +
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
    99
                    "Authentication should reject");
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   100
            throw new LoginException("TestSampleLoginModule throws EXCEPTION");
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   101
        }
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   102
    }
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   103
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   104
    public boolean commit() throws LoginException {
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   105
        return true;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   106
    }
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   107
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   108
    public boolean abort() throws LoginException {
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   109
        return true;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   110
    }
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   111
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   112
    public boolean logout() throws LoginException {
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   113
        return true;
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   114
    }
de746e7eb1e3 8058865: JMX Test Refactoring
olagneau
parents:
diff changeset
   115
}