sshkeygenerationdialog.java

一个非常好的ssh客户端实现

/******************************************************************************
 *
 * Copyright (c) 1999-2003 AppGate Network Security AB. All Rights Reserved.
 * 
 * This file contains Original Code and/or Modifications of Original Code as
 * defined in and that are subject to the MindTerm Public Source License,
 * Version 2.0, (the 'License'). You may not use this file except in compliance
 * with the License.
 * 
 * You should have received a copy of the MindTerm Public Source License
 * along with this software; see the file LICENSE.  If not, write to
 * AppGate Network Security AB, Otterhallegatan 2, SE-41118 Goteborg, SWEDEN
 *
 *****************************************************************************/

package com.mindbright.ssh;

import java.io.File;
import java.io.IOException;
import java.io.FileOutputStream;
import java.io.OutputStream;

import java.awt.*;
import java.awt.event.*;

import com.mindbright.jca.security.KeyPair;
import com.mindbright.jca.security.KeyPairGenerator;
import com.mindbright.jca.security.NoSuchAlgorithmException;
import com.mindbright.jca.security.interfaces.RSAPrivateCrtKey;

import com.mindbright.util.RandomSeed;

import com.mindbright.gui.AWTConvenience;
import com.mindbright.gui.AWTGridBagContainer;
import com.mindbright.gui.ProgressBar;

import com.mindbright.terminal.Terminal;
import com.mindbright.terminal.TerminalWin;

import com.mindbright.ssh2.SSH2PublicKeyFile;
import com.mindbright.ssh2.SSH2KeyPairFile;
import com.mindbright.ssh2.SSH2Exception;
import com.mindbright.ssh2.SSH2AccessDeniedException;

// !!! TODO Cleanout !!!

public final class SSHKeyGenerationDialog {

    private final static String keyGenerationHelp =
	"The key is generated using a random number generator, which " +
	"is seeded by mouse movement in the field containing this text. " +
	"Please move the mouse around in here until the progress bar below " +
	"registers 100%.\n" +
	"\n" +
	"This will create a publickey identity which can be used with the " +
	"publickey authentication method. Your identity will consist of two " +
	"parts: public and private keys. Your private key will be saved " +
	"in the location which you specify; the corresponding public key " +
	"is saved in a file with an identical name but with an extension of " +
	"'.pub' added to it.\n" +
	"\n" +
	"Your private key is protected by encryption, if you entered a " +
	"password. If you left the password field blank, the key will " +
	"not be encrypted. This should only be used in protected " +
	"environments where unattended logins are desired. The contents " +
	"of the 'comment' field are stored with your key, and displayed " +
	"each time you are prompted for the key's password.";

    private final static String keyGenerationComplete =
	"Key Generation Complete\n\n" +
	"To use the key, you must transfer the '.pub' public key " +
	"file to an SSH server and add it to the set of authorized keys. " +
	"See your server documentation for details on this.\n\n" +
	"For convenience, your public key has been copied to the clipboard.\n\n" +
	"Examples:\n" +
	"In ssh2 the '.pub' file should be pointed out in the file " +
	"'authorization' in the config directory (e.g. ~/.ssh2)\n\n" +
	"In OpenSSH's ssh2 the contents of the '.pub' file should be added " +
	"to the file 'authorized_keys2' in your config directory (e.g. ~/.ssh) " +
	"on the server.\n\n" +
	"In ssh1 the contents of the '.pub' file should be added to the " +
	"file 'authorized_keys' in your ssh directory (e.g. ~/.ssh).\n\n" +
	"Press 'Back' to generate a new keypair.";

    private static Dialog      keyGenerationDialog;
    private static FileDialog  keyGenFD;
    private static Choice      choiceBits, choiceType;
    private static TextField   fileText;
    private static TextField   pwdText;
    private static TextField   pwdText2;
    private static TextField   commText;
    private static TextArea    descText;
    private static ProgressBar progBar;
    private static Button      okBut;
    private static Checkbox    cbOpenSSH;
    private static Panel       cardPanel;
    private static CardLayout  cardLayout;
    private static Frame       parent;
    private static boolean     generatedAndSaved;
    private static SSHInteractiveClient client;

    public static void show(Frame par, SSHInteractiveClient cli) {
	client = cli;
	parent = par;

        if (com.mindbright.util.Util.isNetscapeJava()) {
            try {
                netscape.security.PrivilegeManager.enablePrivilege("UniversalFileAccess");
            } catch (netscape.security.ForbiddenTargetException e) {
            }
        }
        
	if(keyGenerationDialog == null) {
	    keyGenerationDialog = new Dialog(parent, "MindTerm - Publickey Keypair Generation", true);

	    Button b;
	    AWTGridBagContainer grid = new AWTGridBagContainer(keyGenerationDialog);

	    keyGenFD = new FileDialog(parent, "MindTerm - Select file to save identity to", FileDialog.SAVE);
	    keyGenFD.setDirectory(client.propsHandler.getSSHHomeDir());

	    createCardPanel();

	    grid.add(cardPanel, 0, GridBagConstraints.REMAINDER);

	    grid.getConstraints().anchor = GridBagConstraints.CENTER;

	    Panel bp = new Panel(new FlowLayout());
	    bp.add(okBut = new Button("Generate"));
	    okBut.addActionListener(new ActionListener() {
		    public void actionPerformed(ActionEvent e) {
			if(generatedAndSaved) {
			    resetValues();
			} else {
			    if(checkValues(pwdText.getText(),
					   pwdText2.getText(),
					   fileText.getText())) {
				cardLayout.show(cardPanel, "second");
				okBut.setEnabled(false);
			    }
			}
		    }
		});
	    bp.add(b = new Button("Close"));
	    b.addActionListener(new ActionListener() {
		    public void actionPerformed(ActionEvent e) {
			keyGenerationDialog.setVisible(false);
			resetValues();
		    }
		});

	    grid.add(bp, 2, GridBagConstraints.REMAINDER);

	    keyGenerationDialog.addWindowListener(new AWTConvenience.CloseAdapter(b));

	    AWTConvenience.setBackgroundOfChildren(keyGenerationDialog);

	    keyGenerationDialog.setResizable(true);
	    keyGenerationDialog.pack();
	}

	resetValues();

	RandomSeed seed = client.randomSeed();
	seed.addEntropyGenerator(descText);

	AWTConvenience.placeDialog(keyGenerationDialog);
	choiceBits.requestFocus();
	keyGenerationDialog.setVisible(true);
    }

    private static void alert(String msg) {
	SSHMiscDialogs.alert("MindTerm - Alert", msg, parent);
    }

    private static void setDefaultFileName(SSHInteractiveClient client) {
	try {
	    String fn = client.propsHandler.getSSHHomeDir() + SSHPropertyHandler.DEF_IDFILE;
	    File   f  = new File(fn);
	    int    fi = 0;
	    while(f.exists()) {
		fn = client.propsHandler.getSSHHomeDir() + SSHPropertyHandler.DEF_IDFILE + fi;
		f  = new File(fn);
		fi++;
	    }
	    fi--;
	    fileText.setText(SSHPropertyHandler.DEF_IDFILE + (fi >= 0 ? String.valueOf(fi) : ""));
	} catch (Throwable t) {
	    // !!!
	    // Don't care...
	}
    }

    private static void createCardPanel() {
	cardPanel  = new Panel();
	cardLayout = new CardLayout();
	cardPanel.setLayout(cardLayout);

	Button b;
	Label  label;
	Panel  p;

	p = new Panel();
	AWTGridBagContainer grid = new AWTGridBagContainer(p);

	grid.getConstraints().fill = GridBagConstraints.BOTH;

	descText = new TextArea(keyGenerationHelp, 12, 34, TextArea.SCROLLBARS_VERTICAL_ONLY);
	descText.setEditable(false);
	grid.add(descText, 2, 8);
	descText.addMouseMotionListener(new MouseMotionAdapter() {
		public void mouseMoved(MouseEvent e) {
		    if(progBar.isFinished()) {
			try {
			    client.randomSeed().removeProgress();
			    progBar.setValue(0);
			    int     bits = Integer.valueOf(choiceBits.getSelectedItem()).intValue();
			    String  alg  = choiceType.getSelectedItem().substring(0, 3);

			    descText.setText("Generating keypair, please wait...");
			    Thread.yield();

			    KeyPair kp = generateKeyPair(alg, bits);
			    saveKeyPair(kp);

			    okBut.setEnabled(true);
			    okBut.setLabel("Back");
			    descText.setText(keyGenerationComplete);
			    generatedAndSaved = true;

			} catch (Throwable t) {
			    alert("Error while generating/saving key pair: " +
				  t.getMessage());
			    cardLayout.show(cardPanel, "first");
			}
		    }
		}
	    });

	grid.getConstraints().fill   = GridBagConstraints.NONE;
	grid.getConstraints().anchor = GridBagConstraints.CENTER;

	progBar = new ProgressBar(512, 150, 20);
	grid.add(progBar, 3, 8);

	cardPanel.add(p, "second");

	p = new Panel();
	grid = new AWTGridBagContainer(p);

	label = new Label("Key type/format:");
	grid.add(label, 0, 2);

	choiceType = AWTConvenience.newChoice(new String[] { "DSA (ssh2)",
							     "RSA (ssh2)",
							     "RSA (ssh1)" });
	grid.add(choiceType, 0, 2);
	label = new Label("Key length (bits):");
	grid.add(label, 1, 2);
	choiceBits = AWTConvenience.newChoice(new String[] { "768",
							     "1024",
							     "1536" });
	grid.add(choiceBits, 1, 2);

	label = new Label("Identity file:");
	grid.add(label, 2, 2);

	fileText = new TextField("", 18);
	grid.add(fileText, 2, 2);

	b = new Button("...");
	b.addActionListener(new ActionListener() {
		public void actionPerformed(ActionEvent e) {
		    keyGenFD.setVisible(true);
		    if(keyGenFD.getFile() != null && keyGenFD.getFile().length() > 0)
			fileText.setText(keyGenFD.getDirectory() + keyGenFD.getFile());
		}
	    });
	grid.getConstraints().fill = GridBagConstraints.NONE;
	grid.add(b, 2, 1);

	grid.getConstraints().fill = GridBagConstraints.HORIZONTAL;

	label = new Label("Password:");
	grid.add(label, 3, 2);

	pwdText = new TextField("", 18);
	pwdText.setEchoChar('*');

	grid.add(pwdText, 3, 2);

	label = new Label("Password again:");
	grid.add(label, 4, 2);

	pwdText2 = new TextField("", 18);
	pwdText2.setEchoChar('*');
	grid.add(pwdText2, 4, 2);

	label = new Label("Comment:");
	grid.add(label, 5, 2);

	commText = new TextField("", 18);
	grid.add(commText, 5, 2);

	cbOpenSSH = new Checkbox("OpenSSH .pub format");
	grid.add(cbOpenSSH, 6, 4);

	cardPanel.add(p, "first");