File: ConsoleCallbackHandler.java

package info (click to toggle)
gridengine 8.1.9%2Bdfsg-9
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 56,756 kB
  • sloc: ansic: 432,689; java: 87,068; cpp: 31,958; sh: 29,429; jsp: 7,757; perl: 6,336; xml: 5,828; makefile: 4,701; csh: 3,934; ruby: 2,221; tcl: 1,676; lisp: 669; yacc: 519; python: 503; lex: 361
file content (127 lines) | stat: -rw-r--r-- 5,142 bytes parent folder | download | duplicates (10)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
/*___INFO__MARK_BEGIN__*/
/*************************************************************************
 *
 *  The Contents of this file are made available subject to the terms of
 *  the Sun Industry Standards Source License Version 1.2
 *
 *  Sun Microsystems Inc., March, 2001
 *
 *
 *  Sun Industry Standards Source License Version 1.2
 *  =================================================
 *  The contents of this file are subject to the Sun Industry Standards
 *  Source License Version 1.2 (the "License"); You may not use this file
 *  except in compliance with the License. You may obtain a copy of the
 *  License at http://gridengine.sunsource.net/Gridengine_SISSL_license.html
 *
 *  Software provided under this License is provided on an "AS IS" basis,
 *  WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING,
 *  WITHOUT LIMITATION, WARRANTIES THAT THE SOFTWARE IS FREE OF DEFECTS,
 *  MERCHANTABLE, FIT FOR A PARTICULAR PURPOSE, OR NON-INFRINGING.
 *  See the License for the specific provisions governing your rights and
 *  obligations concerning the Software.
 *
 *   The Initial Developer of the Original Code is: Sun Microsystems, Inc.
 *
 *   Copyright: 2001 by Sun Microsystems, Inc.
 *
 *   All Rights Reserved.
 *
 ************************************************************************/
/*___INFO__MARK_END__*/
package com.sun.grid.security.login;

import com.sun.grid.util.SGEUtil;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import javax.security.auth.callback.Callback;
import javax.security.auth.callback.CallbackHandler;
import javax.security.auth.callback.NameCallback;
import javax.security.auth.callback.PasswordCallback;
import javax.security.auth.callback.TextInputCallback;
import javax.security.auth.callback.TextOutputCallback;
import javax.security.auth.callback.UnsupportedCallbackException;

/**
 * This <code>CallbackHandler</code> reads input from the console.
 *
 * The following callback can be handled:
 *
 * <ul>
 *   <li>{@link javax.security.auth.callback.NameCallback}</li>
 *   <li>{@link javax.security.auth.callback.PasswordCallback}</li>
 *   <li>{@link javax.security.auth.callback.TextInputCallback}</li>
 *   <li>{@link javax.security.auth.callback.TextOutputCallback}</li>
 * </uL>
 *
 * It uses the native library juti to turn of echoing.
 *
 * @see com.sun.grid.util.SGEUtil#getPassword
 */
public class ConsoleCallbackHandler implements CallbackHandler {
    
    private BufferedReader reader;
    
    private String readLine() throws IOException {
       if(reader == null) {
           reader = new BufferedReader(new InputStreamReader(System.in));
       } 
       return reader.readLine();
    }
    
    /**
     * Handle callbacks
     * @param callbacks  array of callbacks 
     * @throws java.io.IOException  on an IO error
     * @throws javax.security.auth.callback.UnsupportedCallbackException if a callback
     *              is not supported
     */
    public void handle(Callback[] callbacks) throws IOException, UnsupportedCallbackException {
        
        for(int i = 0; i < callbacks.length; i++) {
            if(callbacks[i] instanceof NameCallback) {
                NameCallback cb = (NameCallback)callbacks[i];
                System.out.print(cb.getPrompt());
                if(cb.getDefaultName() != null) {
                    System.out.print(" [" + cb.getDefaultName() + "] ");
                }
                String name = readLine();
                if(name.length() == 0) {
                    name = cb.getDefaultName();
                }
                cb.setName(name);
            } else if(callbacks[i] instanceof TextOutputCallback) {
                TextOutputCallback cb = (TextOutputCallback)callbacks[i];
                switch(cb.getMessageType()) {
                    case TextOutputCallback.ERROR:
                        System.out.print(RB.rb().getString("ccb.error"));
                        break;
                    case TextOutputCallback.WARNING:
                        System.out.print(RB.rb().getString("ccb.warning"));
                        break;
                    default:
                        // do nothing
                }
                System.out.println(cb.getMessage());
            } else if(callbacks[i] instanceof TextInputCallback) {
                TextInputCallback cb = (TextInputCallback)callbacks[i];
                System.out.print(cb.getPrompt());
                if(cb.getDefaultText() != null) {
                    System.out.print(" [" + cb.getDefaultText() + "] ");
                }
                String text = readLine();
                if(text.length() == 0) {
                    text = cb.getDefaultText();
                }
                cb.setText(text);
            } else if(callbacks[i] instanceof PasswordCallback) {
                PasswordCallback cb = (PasswordCallback)callbacks[i];
                cb.setPassword(SGEUtil.getInstance().getPassword(cb.getPrompt()));
            } else {
                throw new UnsupportedCallbackException(callbacks[i]);
            }
        }
    }
    
}