James Moger
2012-11-27 73f1adb2e84b8b9cd4045bcdd7d9afa72d3875b5
src/com/gitblit/authority/GitblitAuthority.java
@@ -61,6 +61,7 @@
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;
@@ -116,7 +117,7 @@
   
   private IUserService userService;
   
   private String caKeystorePassword = null;
   private String caKeystorePassword;
   private JTable table;
   
@@ -127,6 +128,8 @@
   private MailExecutor mail;
   private JButton certificateDefaultsButton;
   private JButton newSSLCertificate;
   public static void main(String... args) {
      EventQueue.invokeLater(new Runnable() {
@@ -234,7 +237,6 @@
      }
      gitblitSettings = new FileSettings(file.getAbsolutePath());
      mail = new MailExecutor(gitblitSettings);
      caKeystorePassword = gitblitSettings.getString(Keys.server.storePassword, null);
      String us = gitblitSettings.getString(Keys.realm.userService, "users.conf");
      String ext = us.substring(us.lastIndexOf(".") + 1).toLowerCase();
      IUserService service = null;
@@ -296,13 +298,39 @@
         if (!caKeystore.exists()) {
            // show certificate defaults dialog 
            certificateDefaultsButton.doClick();
            // create "localhost" ssl certificate
            prepareX509Infrastructure();
         }
      }
   }
   
   private void prepareX509Infrastructure() {
   private boolean prepareX509Infrastructure() {
      if (caKeystorePassword == null) {
         JPasswordField pass = new JPasswordField(10){
            private static final long serialVersionUID = 1L;
            public void addNotify()
             {
                 super.addNotify();
                 requestFocusInWindow();
             }
         };
         pass.setText(caKeystorePassword);
         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);
      X509Utils.prepareX509Infrastructure(metadata, folder, this);
      return true;
   }
   
   private List<X509Certificate> findCerts(File folder, String username) {
@@ -357,13 +385,16 @@
         }
         
         @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) {
            prepareX509Infrastructure();
         public boolean newCertificate(UserCertificateModel ucm, X509Metadata metadata, boolean sendEmail) {
            if (!prepareX509Infrastructure()) {
               return false;
            }
            Date notAfter = metadata.notAfter;
            metadata.serverHostname = gitblitSettings.getString(Keys.web.siteName, Constants.NAME);
            if (StringUtils.isEmpty(metadata.serverHostname)) {
@@ -408,7 +439,7 @@
            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);
@@ -427,10 +458,15 @@
            if (sendEmail) {
               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, GitblitAuthority.this)) {
@@ -458,7 +494,10 @@
               tableModel.fireTableDataChanged();
               table.getSelectionModel().setSelectionInterval(modelIndex, modelIndex);
               
               return true;
            }
            return false;
         }
      };
      
@@ -551,8 +590,6 @@
                  certificateConfig.duration = Integer.parseInt(durationTF.getText());
                  certificateConfig.store(config, metadata);
                  config.save();
                  prepareX509Infrastructure();
               } catch (Exception e1) {
                  Utils.showException(GitblitAuthority.this, e1);
               }
@@ -560,7 +597,7 @@
         }
      });
      
      JButton newSSLCertificate = new JButton(new ImageIcon(getClass().getResource("/rosette_16x16.png")));
      newSSLCertificate = new JButton(new ImageIcon(getClass().getResource("/rosette_16x16.png")));
      newSSLCertificate.setFocusable(false);
      newSSLCertificate.setToolTipText(Translation.get("gb.newSSLCertificate"));      
      newSSLCertificate.addActionListener(new ActionListener() {
@@ -580,7 +617,9 @@
               @Override
               protected Boolean doRequest() throws IOException {
                  prepareX509Infrastructure();
                  if (!prepareX509Infrastructure()) {
                     return false;
                  }
                  
                  // read CA private key and certificate
                  File caKeystoreFile = new File(folder, X509Utils.CA_KEY_STORE);
@@ -636,8 +675,7 @@
                     metadata.serverHostname = Constants.NAME;
                  }
                  metadata.userDisplayname = ucm.user.getDisplayName();
                  sendEmail(ucm.user, metadata, zip);
                  return true;
                  return sendEmail(ucm.user, metadata, zip);
               }
               @Override
@@ -744,7 +782,7 @@
      }
   }
   
   private void sendEmail(UserModel user, X509Metadata metadata, File zip) {
   private boolean sendEmail(UserModel user, X509Metadata metadata, File zip) {
      // send email
      try {
         if (mail.isReady()) {
@@ -771,11 +809,13 @@
            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;
   }
}