]> WPIA git - gigi.git/blobdiff - util/org/cacert/gigi/util/SimpleSigner.java
Update Certificate-DN-API (for escape-safe-strings)
[gigi.git] / util / org / cacert / gigi / util / SimpleSigner.java
index 812705c964a12db25e8e2672dfa12df71b84d84f..04602d82c7b66db2ba0c8799a0366903219f68e4 100644 (file)
 package org.cacert.gigi.util;
 
+import java.io.BufferedReader;
 import java.io.File;
+import java.io.FileInputStream;
 import java.io.FileReader;
 import java.io.IOException;
-import java.sql.PreparedStatement;
-import java.sql.ResultSet;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.io.PrintWriter;
+import java.math.BigInteger;
+import java.security.GeneralSecurityException;
+import java.security.cert.CertificateFactory;
+import java.security.cert.X509Certificate;
 import java.sql.SQLException;
+import java.sql.Timestamp;
+import java.text.ParseException;
+import java.text.SimpleDateFormat;
+import java.util.Calendar;
+import java.util.Date;
+import java.util.HashMap;
 import java.util.Properties;
+import java.util.TimeZone;
 
 import org.cacert.gigi.database.DatabaseConnection;
+import org.cacert.gigi.database.GigiPreparedStatement;
+import org.cacert.gigi.database.GigiResultSet;
+import org.cacert.gigi.dbObjects.Certificate;
+import org.cacert.gigi.dbObjects.Certificate.CSRType;
+import org.cacert.gigi.output.DateSelector;
 
 public class SimpleSigner {
-       private static PreparedStatement warnMail;
-       private static PreparedStatement updateMail;
-       private static PreparedStatement readyMail;
-
-       public static void main(String[] args) throws IOException, SQLException,
-                       InterruptedException {
-               Properties p = new Properties();
-               p.load(new FileReader("config/gigi.properties"));
-               DatabaseConnection.init(p);
-
-               readyMail = DatabaseConnection.getInstance().prepare(
-                               "SELECT id, csr_name FROM emailcerts" + " WHERE csr_name!=null"//
-                                               + " AND created=0"//
-                                               + " AND crt_name=''"//
-                                               + " AND warning<3");
-
-               updateMail = DatabaseConnection.getInstance().prepare(
-                               "UPDATE emailcerts SET crt_name=?,"
-                                               + " created=NOW() WHERE id=?");
-               warnMail = DatabaseConnection.getInstance().prepare(
-                               "UPDATE emailcerts SET warning=warning+1 WHERE id=?");
-               while (true) {
-                       System.out.println("ping");
-                       executeOutstanders();
-                       Thread.sleep(5000);
-               }
-       }
-
-       private static void executeOutstanders() throws SQLException, IOException,
-                       InterruptedException {
-               ResultSet rs = readyMail.executeQuery();
-               while (rs.next()) {
-                       String csrname = rs.getString(2);
-                       System.out.println("sign: " + csrname);
-                       int id = rs.getInt(1);
-                       File crt = KeyStorage.locateCrt(id);
-                       String[] call = new String[]{"openssl", "ca",//
-                                       "-cert", "testca.crt",//
-                                       "-keyfile", "testca.key",//
-                                       "-in", "../" + csrname,//
-                                       "-out", "../" + crt.getPath(),//
-                                       "-days", "356",//
-                                       "-batch",//
-                                       "-config", "selfsign.config"
-
-                       };
-                       Process p1 = Runtime.getRuntime()
-                                       .exec(call, null, new File("keys"));
-
-                       int waitFor = p1.waitFor();
-                       if (waitFor == 0) {
-                               updateMail.setString(1, crt.getPath());
-                               updateMail.setInt(2, id);
-                               updateMail.execute();
-                               System.out.println("sign: " + id);
-                       } else {
-                               System.out.println("ERROR: " + id);
-                               warnMail.setInt(1, id);
-                               warnMail.execute();
-                       }
-
-               }
-               rs.close();
-       }
+
+    private static GigiPreparedStatement warnMail;
+
+    private static GigiPreparedStatement updateMail;
+
+    private static GigiPreparedStatement readyCerts;
+
+    private static GigiPreparedStatement getSANSs;
+
+    private static GigiPreparedStatement revoke;
+
+    private static GigiPreparedStatement revokeCompleted;
+
+    private static GigiPreparedStatement finishJob;
+
+    private static boolean running = true;
+
+    private static Thread runner;
+
+    private static SimpleDateFormat sdf = new SimpleDateFormat("yyMMddHHmmss'Z'");
+
+    static {
+        TimeZone.setDefault(TimeZone.getTimeZone("UTC"));
+        sdf.setTimeZone(TimeZone.getTimeZone("UTC"));
+    }
+
+    public static void main(String[] args) throws IOException, SQLException, InterruptedException {
+        Properties p = new Properties();
+        p.load(new FileReader("config/gigi.properties"));
+        DatabaseConnection.init(p);
+
+        runSigner();
+    }
+
+    public synchronized static void stopSigner() throws InterruptedException {
+        if (runner == null) {
+            throw new IllegalStateException("already stopped");
+        }
+        running = false;
+        runner.interrupt();
+        runner.join();
+        runner = null;
+    }
+
+    public synchronized static void runSigner() throws SQLException, IOException, InterruptedException {
+        if (runner != null) {
+            throw new IllegalStateException("already running");
+        }
+        running = true;
+        readyCerts = DatabaseConnection.getInstance().prepare("SELECT certs.id AS id, certs.csr_name, jobs.id AS jobid, csr_type, md, keyUsage, extendedKeyUsage, executeFrom, executeTo, rootcert FROM jobs " + //
+                "INNER JOIN certs ON certs.id=jobs.targetId " + //
+                "INNER JOIN profiles ON profiles.id=certs.profile " + //
+                "WHERE jobs.state='open' "//
+                + "AND task='sign'");
+
+        getSANSs = DatabaseConnection.getInstance().prepare("SELECT contents, type FROM subjectAlternativeNames " + //
+                "WHERE certId=?");
+
+        updateMail = DatabaseConnection.getInstance().prepare("UPDATE certs SET crt_name=?," + " created=NOW(), serial=? WHERE id=?");
+        warnMail = DatabaseConnection.getInstance().prepare("UPDATE jobs SET warning=warning+1, state=IF(warning<3, 'open','error') WHERE id=?");
+
+        revoke = DatabaseConnection.getInstance().prepare("SELECT certs.id, certs.csr_name,jobs.id FROM jobs INNER JOIN certs ON jobs.targetId=certs.id" + " WHERE jobs.state='open' AND task='revoke'");
+        revokeCompleted = DatabaseConnection.getInstance().prepare("UPDATE certs SET revoked=NOW() WHERE id=?");
+
+        finishJob = DatabaseConnection.getInstance().prepare("UPDATE jobs SET state='done' WHERE id=?");
+
+        runner = new Thread() {
+
+            @Override
+            public void run() {
+                work();
+            }
+
+        };
+        runner.start();
+    }
+
+    private static void work() {
+        try {
+            gencrl();
+        } catch (IOException e2) {
+            e2.printStackTrace();
+        } catch (InterruptedException e2) {
+            e2.printStackTrace();
+        }
+        while (running) {
+            try {
+                signCertificates();
+                revokeCertificates();
+                Thread.sleep(5000);
+            } catch (IOException e) {
+                e.printStackTrace();
+            } catch (SQLException e) {
+                e.printStackTrace();
+            } catch (InterruptedException e1) {
+            }
+        }
+    }
+
+    private static void revokeCertificates() throws SQLException, IOException, InterruptedException {
+        GigiResultSet rs = revoke.executeQuery();
+        boolean worked = false;
+        while (rs.next()) {
+            int id = rs.getInt(1);
+            File crt = KeyStorage.locateCrt(id);
+            String[] call = new String[] {
+                    "openssl", "ca",//
+                    "-cert",
+                    "../unassured.crt",//
+                    "-keyfile",
+                    "../unassured.key",//
+                    "-revoke",
+                    "../../" + crt.getPath(),//
+                    "-batch",//
+                    "-config",
+                    "../selfsign.config"
+
+            };
+            Process p1 = Runtime.getRuntime().exec(call, null, new File("keys/unassured.ca"));
+            System.out.println("revoking: " + crt.getPath());
+            if (p1.waitFor() == 0) {
+                worked = true;
+                revokeCompleted.setInt(1, id);
+                revokeCompleted.execute();
+                finishJob.setInt(1, rs.getInt(3));
+                finishJob.execute();
+            } else {
+                System.out.println("Failed");
+            }
+        }
+        if (worked) {
+            gencrl();
+        }
+    }
+
+    private static void gencrl() throws IOException, InterruptedException {
+        String[] call = new String[] {
+                "openssl", "ca",//
+                "-cert",
+                "../unassured.crt",//
+                "-keyfile",
+                "../unassured.key",//
+                "-gencrl",//
+                "-crlhours",//
+                "12",//
+                "-out",
+                "../unassured.crl",//
+                "-config",
+                "../selfsign.config"
+
+        };
+        Process p1 = Runtime.getRuntime().exec(call, null, new File("keys/unassured.ca"));
+        if (p1.waitFor() != 0) {
+            System.out.println("Error while generating crl.");
+        }
+    }
+
+    private static int counter = 0;
+
+    private static void signCertificates() throws SQLException {
+        GigiResultSet rs = readyCerts.executeQuery();
+
+        Calendar c = Calendar.getInstance();
+        c.setTimeZone(TimeZone.getTimeZone("UTC"));
+
+        while (rs.next()) {
+            String csrname = rs.getString("csr_name");
+            int id = rs.getInt("id");
+            System.out.println("sign: " + csrname);
+            try {
+                String csrType = rs.getString("csr_type");
+                CSRType ct = CSRType.valueOf(csrType);
+                File crt = KeyStorage.locateCrt(id);
+
+                String keyUsage = rs.getString("keyUsage");
+                String ekeyUsage = rs.getString("extendedKeyUsage");
+
+                Timestamp from = rs.getTimestamp("executeFrom");
+                String length = rs.getString("executeTo");
+                Date fromDate;
+                Date toDate;
+                if (from == null) {
+                    fromDate = new Date(System.currentTimeMillis());
+                } else {
+                    fromDate = new Date(from.getTime());
+                }
+                if (length.endsWith("m") || length.endsWith("y")) {
+                    String num = length.substring(0, length.length() - 1);
+                    int inter = Integer.parseInt(num);
+                    c.setTime(fromDate);
+                    if (length.endsWith("m")) {
+                        c.add(Calendar.MONTH, inter);
+                    } else {
+                        c.add(Calendar.YEAR, inter);
+                    }
+                    toDate = c.getTime();
+                } else {
+                    toDate = DateSelector.getDateFormat().parse(length);
+                }
+
+                getSANSs.setInt(1, id);
+                GigiResultSet san = getSANSs.executeQuery();
+
+                File f = new File("keys", "SANFile" + System.currentTimeMillis() + (counter++) + ".cfg");
+                PrintWriter cfg = new PrintWriter(f);
+                boolean first = true;
+                while (san.next()) {
+                    if ( !first) {
+                        cfg.print(", ");
+                    } else {
+                        cfg.print("subjectAltName=");
+                    }
+                    first = false;
+                    cfg.print(san.getString("type"));
+                    cfg.print(":");
+                    cfg.print(san.getString("contents"));
+                }
+                cfg.println();
+                cfg.println("keyUsage=critical," + keyUsage);
+                cfg.println("extendedKeyUsage=critical," + ekeyUsage);
+                cfg.close();
+
+                int rootcert = rs.getInt("rootcert");
+                String ca = "unassured";
+                if (rootcert == 0) {
+                    ca = "unassured";
+                } else if (rootcert == 1) {
+                    ca = "assured";
+                }
+                HashMap<String, String> subj = new HashMap<>();
+                GigiPreparedStatement ps = DatabaseConnection.getInstance().prepare("SELECT name, value FROM certAvas WHERE certId=?");
+                ps.setInt(1, rs.getInt("id"));
+                GigiResultSet rs2 = ps.executeQuery();
+                while (rs2.next()) {
+                    subj.put(rs2.getString("name"), rs2.getString("value"));
+                }
+                if (subj.size() == 0) {
+                    subj.put("CN", "<empty>");
+                    System.out.println("WARNING: DN was empty");
+                }
+                String[] call = new String[] {
+                        "openssl", "ca",//
+                        "-in",
+                        "../../" + csrname,//
+                        "-cert",
+                        "../" + ca + ".crt",//
+                        "-keyfile",
+                        "../" + ca + ".key",//
+                        "-out",
+                        "../../" + crt.getPath(),//
+                        "-utf8",
+                        "-startdate",
+                        sdf.format(fromDate),//
+                        "-enddate",
+                        sdf.format(toDate),//
+                        "-batch",//
+                        "-md",
+                        rs.getString("md"),//
+                        "-extfile",
+                        "../" + f.getName(),//
+
+                        "-subj",
+                        Certificate.stringifyDN(subj),//
+                        "-config",
+                        "../selfsign.config"//
+
+                };
+                if (ct == CSRType.SPKAC) {
+                    call[2] = "-spkac";
+                }
+                Process p1 = Runtime.getRuntime().exec(call, null, new File("keys/unassured.ca"));
+
+                int waitFor = p1.waitFor();
+                f.delete();
+                if (waitFor == 0) {
+                    try (InputStream is = new FileInputStream(crt)) {
+                        CertificateFactory cf = CertificateFactory.getInstance("X.509");
+                        X509Certificate crtp = (X509Certificate) cf.generateCertificate(is);
+                        BigInteger serial = crtp.getSerialNumber();
+                        updateMail.setString(1, crt.getPath());
+                        updateMail.setString(2, serial.toString(16));
+                        updateMail.setInt(3, id);
+                        updateMail.execute();
+
+                        finishJob.setInt(1, rs.getInt("jobid"));
+                        finishJob.execute();
+                        System.out.println("signed: " + id);
+                        continue;
+                    }
+                } else {
+                    BufferedReader br = new BufferedReader(new InputStreamReader(p1.getErrorStream()));
+                    String s;
+                    while ((s = br.readLine()) != null) {
+                        System.out.println(s);
+                    }
+                }
+            } catch (GeneralSecurityException e) {
+                e.printStackTrace();
+            } catch (IOException e) {
+                e.printStackTrace();
+            } catch (ParseException e) {
+                e.printStackTrace();
+            } catch (InterruptedException e1) {
+                e1.printStackTrace();
+            }
+            System.out.println("Error with: " + id);
+            warnMail.setInt(1, rs.getInt("jobid"));
+            warnMail.execute();
+
+        }
+        rs.close();
+    }
 }