Fixed focus for keystore password prompt

James Moger

2012-11-28 0bbdd9f9adf12ad9082a4c49ae1c9a0778b00bb4

 src/com/gitblit/authority/GitblitAuthority.java

@@ -29,10 +29,13 @@
import java.awt.event.WindowAdapter;

import java.awt.event.WindowEvent;

import java.io.BufferedInputStream;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileWriter;

import java.io.FilenameFilter;

import java.io.IOException;

import java.security.PrivateKey;

import java.security.cert.CertificateFactory;

import java.security.cert.X509Certificate;

import java.text.MessageFormat;

@@ -51,15 +54,20 @@
import javax.mail.internet.MimeBodyPart;

import javax.mail.internet.MimeMultipart;

import javax.swing.ImageIcon;

import javax.swing.InputVerifier;

import javax.swing.JButton;

import javax.swing.JComponent;

import javax.swing.JFrame;

import javax.swing.JLabel;

import javax.swing.JOptionPane;

import javax.swing.JPanel;

import javax.swing.JPasswordField;

import javax.swing.JScrollPane;

import javax.swing.JSplitPane;

import javax.swing.JTable;

import javax.swing.JTextField;

import javax.swing.RowFilter;

import javax.swing.SwingConstants;

import javax.swing.UIManager;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.ListSelectionListener;

@@ -69,6 +77,7 @@
import org.eclipse.jgit.lib.StoredConfig;

import org.eclipse.jgit.storage.file.FileBasedConfig;

import org.eclipse.jgit.util.FS;

import org.slf4j.LoggerFactory;



import com.gitblit.ConfigUserService;

import com.gitblit.Constants;

@@ -80,9 +89,12 @@
import com.gitblit.client.HeaderPanel;

import com.gitblit.client.Translation;

import com.gitblit.models.UserModel;

import com.gitblit.utils.ArrayUtils;

import com.gitblit.utils.StringUtils;

import com.gitblit.utils.TimeUtils;

import com.gitblit.utils.X509Utils;

import com.gitblit.utils.X509Utils.RevocationReason;

import com.gitblit.utils.X509Utils.X509Log;

import com.gitblit.utils.X509Utils.X509Metadata;



/**

@@ -91,7 +103,7 @@
 * @author James Moger

 *

 */

public class GitblitAuthority extends JFrame {

public class GitblitAuthority extends JFrame implements X509Log {



   private static final long serialVersionUID = 1L;

   

@@ -105,13 +117,19 @@
   

   private IUserService userService;

   

   private String caKeystorePassword = null;

   private String caKeystorePassword;



   private JTable table;

   

   private int defaultDuration;

   

   private TableRowSorter<UserCertificateTableModel> defaultSorter;

	
   private MailExecutor mail;



   private JButton certificateDefaultsButton;



   private JButton newSSLCertificate;



   public static void main(String... args) {

      EventQueue.invokeLater(new Runnable() {

@@ -136,7 +154,7 @@
   

   public void initialize() {

      setIconImage(new ImageIcon(getClass().getResource("/gitblt-favicon.png")).getImage());

      setTitle("Gitblit PKI Authority v" + Constants.VERSION + " (" + Constants.VERSION_DATE + ")");

      setTitle("Gitblit Certificate Authority v" + Constants.VERSION + " (" + Constants.VERSION_DATE + ")");

      setContentPane(getUI());

      setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);

      addWindowListener(new WindowAdapter() {

@@ -170,7 +188,7 @@


      // try to restore saved window size

      if (StringUtils.isEmpty(sz)) {

         setSize(850, 500);

         setSize(900, 600);

      } else {

         String[] chunks = sz.split("x");

         int width = Integer.parseInt(chunks[0]);

@@ -218,7 +236,7 @@
         return null;

      }

      gitblitSettings = new FileSettings(file.getAbsolutePath());

      caKeystorePassword = gitblitSettings.getString(Keys.server.storePassword, null);

      mail = new MailExecutor(gitblitSettings);

      String us = gitblitSettings.getString(Keys.realm.userService, "users.conf");

      String ext = us.substring(us.lastIndexOf(".") + 1).toLowerCase();

      IUserService service = null;

@@ -243,7 +261,9 @@
   private void load(File folder) {

      this.folder = folder;

      this.userService = loadUsers(folder);

      if (userService != null) {

      if (userService == null) {

         JOptionPane.showMessageDialog(this, MessageFormat.format("Sorry, {0} doesn't look like a Gitblit GO installation.", folder));

      } else {

         // build empty certificate model for all users

         Map<String, UserCertificateModel> map = new HashMap<String, UserCertificateModel>();

         for (String user : userService.getAllUsernames()) {

@@ -272,7 +292,39 @@
         tableModel.list = new ArrayList<UserCertificateModel>(map.values());

         Collections.sort(tableModel.list);

         tableModel.fireTableDataChanged();

         Utils.packColumns(table, Utils.MARGIN);

			
         File caKeystore = new File(folder, X509Utils.CA_KEY_STORE);

         if (!caKeystore.exists()) {

            // show certificate defaults dialog 
            certificateDefaultsButton.doClick();

				
            // create "localhost" ssl certificate

            prepareX509Infrastructure();

         }

      }

   }

	
   private boolean prepareX509Infrastructure() {

      if (caKeystorePassword == null) {

         JPasswordField pass = new JPasswordField(10);

         pass.setText(caKeystorePassword);

         pass.addAncestorListener(new RequestFocusListener());

         JPanel panel = new JPanel(new BorderLayout());

         panel.add(new JLabel(Translation.get("gb.enterKeystorePassword")), BorderLayout.NORTH);

         panel.add(pass, BorderLayout.CENTER);

         int result = JOptionPane.showConfirmDialog(GitblitAuthority.this, panel, Translation.get("gb.password"), JOptionPane.OK_CANCEL_OPTION);

         if (result == JOptionPane.OK_OPTION) {

            caKeystorePassword = new String(pass.getPassword());

         } else {

            return false;

         }

      }



      X509Metadata metadata = new X509Metadata("localhost", caKeystorePassword);

      setMetadataDefaults(metadata);

      X509Utils.prepareX509Infrastructure(metadata, folder, this);

      return true;

   }

   

   private List<X509Certificate> findCerts(File folder, String username) {

@@ -309,6 +361,11 @@
         public Insets getInsets() {

            return Utils.INSETS;

         }

			
         @Override

         public boolean isAllowEmail() {

            return mail.isReady();

         }



         @Override

         public Date getDefaultExpiration() {

@@ -322,33 +379,22 @@
         }

         

         @Override

         public void saveUser(String username, UserCertificateModel ucm) {

            userService.updateUserModel(username, ucm.user);

         public boolean saveUser(String username, UserCertificateModel ucm) {

            return userService.updateUserModel(username, ucm.user);

         }

         

         @Override

         public void newCertificate(UserCertificateModel ucm, X509Metadata metadata, boolean sendEmail) {

            Date notAfter = metadata.notAfter;

            metadata.serverHostname = gitblitSettings.getString(Keys.web.siteName, "localhost");

            UserModel user = ucm.user;				
				
            // set default values from config file

            File certificatesConfigFile = new File(folder, X509Utils.CA_CONFIG);

            FileBasedConfig config = new FileBasedConfig(certificatesConfigFile, FS.detect());

            if (certificatesConfigFile.exists()) {

               try {

                  config.load();

               } catch (Exception e) {

                  Utils.showException(GitblitAuthority.this, e);

               }

               NewCertificateConfig certificateConfig = NewCertificateConfig.KEY.parse(config);

               certificateConfig.update(metadata);

         public boolean newCertificate(UserCertificateModel ucm, X509Metadata metadata, boolean sendEmail) {

            if (!prepareX509Infrastructure()) {

               return false;

            }

				
            // restore expiration date



            Date notAfter = metadata.notAfter;

            setMetadataDefaults(metadata);

            metadata.notAfter = notAfter;

            

            // set user's specified OID values

            UserModel user = ucm.user;				
            if (!StringUtils.isEmpty(user.organizationalUnit)) {

               metadata.oids.put("OU", user.organizationalUnit);

            }

@@ -366,18 +412,14 @@
            }



            File caKeystoreFile = new File(folder, X509Utils.CA_KEY_STORE);

            File zip = X509Utils.newClientBundle(metadata, caKeystoreFile, caKeystorePassword);

				
            File zip = X509Utils.newClientBundle(metadata, caKeystoreFile, caKeystorePassword, GitblitAuthority.this);



            // save latest expiration date

            if (ucm.expires == null || metadata.notAfter.after(ucm.expires)) {

            if (ucm.expires == null || metadata.notAfter.before(ucm.expires)) {

               ucm.expires = metadata.notAfter;

            }

            ucm.update(config);

            try {

               config.save();

            } catch (Exception e) {

               Utils.showException(GitblitAuthority.this, e);

            }

				
            updateAuthorityConfig(ucm);

            

            // refresh user

            ucm.certs = null;

@@ -386,47 +428,20 @@
            table.getSelectionModel().setSelectionInterval(modelIndex, modelIndex);

            

            if (sendEmail) {

               // send email

               try {

                  MailExecutor mail = new MailExecutor(gitblitSettings);

                  if (mail.isReady()) {

                     Message message = mail.createMessage(user.emailAddress);

                     message.setSubject("Your Gitblit client certificate for " + metadata.serverHostname);



                     // body of email

                     String body = X509Utils.processTemplate(new File(caKeystoreFile.getParentFile(), "mail.tmpl"), metadata);

                     if (StringUtils.isEmpty(body)) {

                        body = MessageFormat.format("Hi {0}\n\nHere is your client certificate bundle.\nInside the zip file are installation instructions.", user.getDisplayName());

                     }

                     Multipart mp = new MimeMultipart();

                     MimeBodyPart messagePart = new MimeBodyPart();

                     messagePart.setText(body);

                     mp.addBodyPart(messagePart);



                     // attach zip

                     MimeBodyPart filePart = new MimeBodyPart();

                     FileDataSource fds = new FileDataSource(zip);

                     filePart.setDataHandler(new DataHandler(fds));

                     filePart.setFileName(fds.getName());

                     mp.addBodyPart(filePart);



                     message.setContent(mp);



                     mail.sendNow(message);

                  } else {

                     JOptionPane.showMessageDialog(GitblitAuthority.this, "Sorry, the mail server settings are not configured properly.\nCan not send email.", Translation.get("gb.error"), JOptionPane.ERROR_MESSAGE);

                  }

               } catch (Exception e) {

                  Utils.showException(GitblitAuthority.this, e);

               }

               sendEmail(user, metadata, zip);

            }

            return true;

         }

         

         @Override

         public void revoke(UserCertificateModel ucm, X509Certificate cert, RevocationReason reason) {

         public boolean revoke(UserCertificateModel ucm, X509Certificate cert, RevocationReason reason) {

            if (!prepareX509Infrastructure()) {

               return false;

            }



            File caRevocationList = new File(folder, X509Utils.CA_REVOCATION_LIST);

            File caKeystoreFile = new File(folder, X509Utils.CA_KEY_STORE);

            if (X509Utils.revoke(cert, reason, caRevocationList, caKeystoreFile, caKeystorePassword)) {

            if (X509Utils.revoke(cert, reason, caRevocationList, caKeystoreFile, caKeystorePassword, GitblitAuthority.this)) {

               File certificatesConfigFile = new File(folder, X509Utils.CA_CONFIG);

               FileBasedConfig config = new FileBasedConfig(certificatesConfigFile, FS.detect());

               if (certificatesConfigFile.exists()) {

@@ -450,7 +465,11 @@
               int modelIndex = table.convertRowIndexToModel(table.getSelectedRow());

               tableModel.fireTableDataChanged();

               table.getSelectionModel().setSelectionInterval(modelIndex, modelIndex);

					
               return true;

            }

				
            return false;

         }

      };

      

@@ -490,7 +509,159 @@
      usersPanel.add(new JScrollPane(table), BorderLayout.CENTER);

      usersPanel.setMinimumSize(new Dimension(400, 10));

      

      final JTextField filterTextfield = new JTextField(20);

      certificateDefaultsButton = new JButton(new ImageIcon(getClass().getResource("/settings_16x16.png")));

      certificateDefaultsButton.setFocusable(false);

      certificateDefaultsButton.setToolTipText(Translation.get("gb.newCertificateDefaults"));		
      certificateDefaultsButton.addActionListener(new ActionListener() {

         @Override

         public void actionPerformed(ActionEvent e) {

            X509Metadata metadata = new X509Metadata("whocares", "whocares");

            File certificatesConfigFile = new File(folder, X509Utils.CA_CONFIG);

            FileBasedConfig config = new FileBasedConfig(certificatesConfigFile, FS.detect());

            NewCertificateConfig certificateConfig = null;

            if (certificatesConfigFile.exists()) {

               try {

                  config.load();

               } catch (Exception x) {

                  Utils.showException(GitblitAuthority.this, x);

               }

               certificateConfig = NewCertificateConfig.KEY.parse(config);

               certificateConfig.update(metadata);

            }

            InputVerifier verifier = new InputVerifier() {

               public boolean verify(JComponent comp) {

                  boolean returnValue;

                  JTextField textField = (JTextField) comp;

                  try {

                     Integer.parseInt(textField.getText());

                     returnValue = true;

                  } catch (NumberFormatException e) {

                     returnValue = false;

                  }

                  return returnValue;

               }

            };



            JTextField durationTF = new JTextField(4);

            durationTF.setInputVerifier(verifier);

            durationTF.setVerifyInputWhenFocusTarget(true);

            durationTF.setText("" + certificateConfig.duration);

            JPanel durationPanel = Utils.newFieldPanel(Translation.get("gb.duration"), durationTF, Translation.get("gb.duration.days").replace("{0}",  "").trim());

            DefaultOidsPanel oids = new DefaultOidsPanel(metadata);



            JPanel panel = new JPanel(new BorderLayout());

            panel.add(durationPanel, BorderLayout.NORTH);

            panel.add(oids, BorderLayout.CENTER);



            int result = JOptionPane.showConfirmDialog(GitblitAuthority.this, 
                  panel, Translation.get("gb.newCertificateDefaults"), JOptionPane.OK_CANCEL_OPTION,

                  JOptionPane.QUESTION_MESSAGE, new ImageIcon(getClass().getResource("/settings_32x32.png")));

            if (result == JOptionPane.OK_OPTION) {

               try {

                  oids.update(metadata);

                  certificateConfig.duration = Integer.parseInt(durationTF.getText());

                  certificateConfig.store(config, metadata);

                  config.save();

               } catch (Exception e1) {

                  Utils.showException(GitblitAuthority.this, e1);

               }

            }

         }

      });

		
      newSSLCertificate = new JButton(new ImageIcon(getClass().getResource("/rosette_16x16.png")));

      newSSLCertificate.setFocusable(false);

      newSSLCertificate.setToolTipText(Translation.get("gb.newSSLCertificate"));		
      newSSLCertificate.addActionListener(new ActionListener() {

         @Override

         public void actionPerformed(ActionEvent e) {

            Date defaultExpiration = new Date(System.currentTimeMillis() + 10*TimeUtils.ONEYEAR);

            NewSSLCertificateDialog dialog = new NewSSLCertificateDialog(GitblitAuthority.this, defaultExpiration);

            dialog.setModal(true);

            dialog.setVisible(true);

            if (dialog.isCanceled()) {

               return;

            }

            final Date expires = dialog.getExpiration();

            final String hostname = dialog.getHostname();



            AuthorityWorker worker = new AuthorityWorker(GitblitAuthority.this) {



               @Override

               protected Boolean doRequest() throws IOException {

                  if (!prepareX509Infrastructure()) {

                     return false;

                  }

						
                  // read CA private key and certificate

                  File caKeystoreFile = new File(folder, X509Utils.CA_KEY_STORE);

                  PrivateKey caPrivateKey = X509Utils.getPrivateKey(X509Utils.CA_ALIAS, caKeystoreFile, caKeystorePassword);

                  X509Certificate caCert = X509Utils.getCertificate(X509Utils.CA_ALIAS, caKeystoreFile, caKeystorePassword);

						
                  // generate new SSL certificate

                  X509Metadata metadata = new X509Metadata(hostname, caKeystorePassword);

                  metadata.notAfter = expires;

                  File serverKeystoreFile = new File(folder, X509Utils.SERVER_KEY_STORE);

                  X509Certificate cert = X509Utils.newSSLCertificate(metadata, caPrivateKey, caCert, serverKeystoreFile, GitblitAuthority.this);

                  return cert != null;

               }



               @Override

               protected void onSuccess() {

                  JOptionPane.showMessageDialog(GitblitAuthority.this, 
                        MessageFormat.format(Translation.get("gb.sslCertificateGenerated"), hostname),

                        Translation.get("gb.newSSLCertificate"), JOptionPane.INFORMATION_MESSAGE);

               }

            };

				
            worker.execute();

         }

      });

		
      JButton emailBundle = new JButton(new ImageIcon(getClass().getResource("/mail_16x16.png")));

      emailBundle.setFocusable(false);

      emailBundle.setToolTipText(Translation.get("gb.emailCertificateBundle"));		
      emailBundle.addActionListener(new ActionListener() {

         @Override

         public void actionPerformed(ActionEvent e) {

            int row = table.getSelectedRow();

            if (row < 0) {

               return;

            }

            int modelIndex = table.convertRowIndexToModel(row);

            final UserCertificateModel ucm = tableModel.get(modelIndex);

            if (ArrayUtils.isEmpty(ucm.certs)) {

               JOptionPane.showMessageDialog(GitblitAuthority.this, MessageFormat.format(Translation.get("gb.pleaseGenerateClientCertificate"), ucm.user.getDisplayName()));

            }

            final File zip = new File(folder, X509Utils.CERTS + File.separator + ucm.user.username + File.separator + ucm.user.username + ".zip");

            if (!zip.exists()) {

               return;

            }

				
            AuthorityWorker worker = new AuthorityWorker(GitblitAuthority.this) {

               @Override

               protected Boolean doRequest() throws IOException {

                  X509Metadata metadata = new X509Metadata(ucm.user.username, "whocares");

                  metadata.serverHostname = gitblitSettings.getString(Keys.web.siteName, Constants.NAME);

                  if (StringUtils.isEmpty(metadata.serverHostname)) {

                     metadata.serverHostname = Constants.NAME;

                  }

                  metadata.userDisplayname = ucm.user.getDisplayName();

                  return sendEmail(ucm.user, metadata, zip);

               }



               @Override

               protected void onSuccess() {

                  JOptionPane.showMessageDialog(GitblitAuthority.this, MessageFormat.format(Translation.get("gb.clientCertificateBundleSent"),

                        ucm.user.getDisplayName()));

               }

					
            };

            worker.execute();				
         }

      });

		
      final JTextField filterTextfield = new JTextField(15);

      filterTextfield.addActionListener(new ActionListener() {

         public void actionPerformed(ActionEvent e) {

            filterUsers(filterTextfield.getText());

@@ -501,16 +672,33 @@
            filterUsers(filterTextfield.getText());

         }

      });

		
      JPanel buttonControls = new JPanel(new FlowLayout(FlowLayout.LEFT, Utils.MARGIN, Utils.MARGIN));

      buttonControls.add(certificateDefaultsButton);

      buttonControls.add(newSSLCertificate);

      buttonControls.add(emailBundle);



      JPanel userControls = new JPanel(new FlowLayout(FlowLayout.RIGHT, 5, 5));

      JPanel userControls = new JPanel(new FlowLayout(FlowLayout.RIGHT, Utils.MARGIN, Utils.MARGIN));

      userControls.add(new JLabel(Translation.get("gb.filter")));

      userControls.add(filterTextfield);

      

      JPanel topPanel = new JPanel(new BorderLayout(0, 0));

      topPanel.add(buttonControls, BorderLayout.WEST);

      topPanel.add(userControls, BorderLayout.EAST);

		
      JPanel leftPanel = new JPanel(new BorderLayout());

      leftPanel.add(userControls, BorderLayout.NORTH);

      leftPanel.add(topPanel, BorderLayout.NORTH);

      leftPanel.add(usersPanel, BorderLayout.CENTER);

      

      userCertificatePanel.setMinimumSize(new Dimension(375, 10));

		
      JLabel statusLabel = new JLabel();

      statusLabel.setHorizontalAlignment(SwingConstants.RIGHT);

      if (X509Utils.unlimitedStrength) {

         statusLabel.setText("JCE Unlimited Strength Jurisdiction Policy");

      } else {

         statusLabel.setText("JCE Standard Encryption Policy");

      }

      

      JPanel root = new JPanel(new BorderLayout()) {

         private static final long serialVersionUID = 1L;

@@ -520,7 +708,8 @@
      };

      JSplitPane splitPane = new JSplitPane(JSplitPane.HORIZONTAL_SPLIT, leftPanel, userCertificatePanel);

      splitPane.setDividerLocation(1d);

      root.add(splitPane);

      root.add(splitPane, BorderLayout.CENTER);

      root.add(statusLabel, BorderLayout.SOUTH);

      return root;

   }

   

@@ -544,4 +733,99 @@
      sorter.setRowFilter(containsFilter);

      table.setRowSorter(sorter);

   }

	
   @Override

   public void log(String message) {

      BufferedWriter writer = null;

      try {

         writer = new BufferedWriter(new FileWriter(new File(folder, X509Utils.CERTS + File.separator + "log.txt"), true));

         writer.write(MessageFormat.format("{0,date,yyyy-MM-dd HH:mm}: {1}", new Date(), message));

         writer.newLine();

         writer.flush();

      } catch (Exception e) {

         LoggerFactory.getLogger(GitblitAuthority.class).error("Failed to append log entry!", e);

      } finally {

         if (writer != null) {

            try {

               writer.close();

            } catch (IOException e) {

            }

         }

      }

   }

	
   private boolean sendEmail(UserModel user, X509Metadata metadata, File zip) {

      // send email

      try {

         if (mail.isReady()) {

            Message message = mail.createMessage(user.emailAddress);

            message.setSubject("Your Gitblit client certificate for " + metadata.serverHostname);



            // body of email

            String body = X509Utils.processTemplate(new File(folder, X509Utils.CERTS + File.separator + "mail.tmpl"), metadata);

            if (StringUtils.isEmpty(body)) {

               body = MessageFormat.format("Hi {0}\n\nHere is your client certificate bundle.\nInside the zip file are installation instructions.", user.getDisplayName());

            }

            Multipart mp = new MimeMultipart();

            MimeBodyPart messagePart = new MimeBodyPart();

            messagePart.setText(body);

            mp.addBodyPart(messagePart);



            // attach zip

            MimeBodyPart filePart = new MimeBodyPart();

            FileDataSource fds = new FileDataSource(zip);

            filePart.setDataHandler(new DataHandler(fds));

            filePart.setFileName(fds.getName());

            mp.addBodyPart(filePart);



            message.setContent(mp);



            mail.sendNow(message);

            return true;

         } else {

            JOptionPane.showMessageDialog(GitblitAuthority.this, "Sorry, the mail server settings are not configured properly.\nCan not send email.", Translation.get("gb.error"), JOptionPane.ERROR_MESSAGE);

         }

      } catch (Exception e) {

         Utils.showException(GitblitAuthority.this, e);

      }

      return false;

   }

	
   private void setMetadataDefaults(X509Metadata metadata) {

      metadata.serverHostname = gitblitSettings.getString(Keys.web.siteName, Constants.NAME);

      if (StringUtils.isEmpty(metadata.serverHostname)) {

         metadata.serverHostname = Constants.NAME;

      }

		
      // set default values from config file

      File certificatesConfigFile = new File(folder, X509Utils.CA_CONFIG);

      FileBasedConfig config = new FileBasedConfig(certificatesConfigFile, FS.detect());

      if (certificatesConfigFile.exists()) {

         try {

            config.load();

         } catch (Exception e) {

            Utils.showException(GitblitAuthority.this, e);

         }

         NewCertificateConfig certificateConfig = NewCertificateConfig.KEY.parse(config);

         certificateConfig.update(metadata);

      }

   }

	
   private void updateAuthorityConfig(UserCertificateModel ucm) {

      File certificatesConfigFile = new File(folder, X509Utils.CA_CONFIG);

      FileBasedConfig config = new FileBasedConfig(certificatesConfigFile, FS.detect());

      if (certificatesConfigFile.exists()) {

         try {

            config.load();

         } catch (Exception e) {

            Utils.showException(GitblitAuthority.this, e);

         }

      }

      ucm.update(config);

      try {

         config.save();

      } catch (Exception e) {

         Utils.showException(GitblitAuthority.this, e);

      }

   }

}