fix make update-rootcerts
[org.ibex.crypto.git] / tools / GenCompactCAList.java
1
2 import java.io.*;
3 import org.bouncycastle.asn1.*;
4 import org.bouncycastle.asn1.x509.*;
5
6 public class GenCompactCAList {
7     public static void main(String[] args) throws Exception {
8         if(args.length < 2) throw new Exception("Usage: GenCAList format file(s)");
9         String format = args[0];
10         DEREncodableVector vec = new DEREncodableVector();
11         for(int i=1;i<args.length;i++) {
12             X509CertificateStructure x509 = new X509CertificateStructure((ASN1Sequence) new ASN1InputStream(new FileInputStream(args[i])).readObject());
13             X509Name subject = x509.getSubject();
14             SubjectPublicKeyInfo pki = x509.getSubjectPublicKeyInfo();
15             RSAPublicKeyStructure rsa = new RSAPublicKeyStructure((ASN1Sequence) pki.getPublicKey());
16             DEREncodableVector vec2 = new DEREncodableVector();
17             vec2.add(subject);
18             vec2.add(rsa);
19             vec.add(new DERSequence(vec2));
20         }
21         if(format.equals("binary")) {
22             DEROutputStream dos = new DEROutputStream(System.out);
23             dos.writeObject(new DERSequence(vec));
24             dos.close();
25         } else if(format.equals("class")){
26             ByteArrayOutputStream baos = new ByteArrayOutputStream();
27             DEROutputStream dos = new DEROutputStream(baos);
28             dos.writeObject(new DERSequence(vec));
29             dos.close();
30             baos.close();            
31             byte[] buf = baos.toByteArray();
32             StringBuffer sb = new StringBuffer();
33             for(int i=0;i<buf.length;i+=7) {
34                 long l = 0;
35                 for(int j=0;j<7;j++) {
36                     l <<= 8;
37                     byte b = (i+j < buf.length) ? buf[i+j] : -1;
38                     l |= (b & 0xffL);
39                 }
40                 for(int j=0;j<8;j++) {
41                     char c = (char) ((l>>>(7*(7-j)))&0x7f);
42                     if(c=='\n') sb.append("\\n"); 
43                     else if(c=='\r') sb.append("\\r");
44                     else if(c=='\\') sb.append("\\\\");
45                     else if(c=='"') sb.append("\\\"");
46                     else if(c >= 32 && c <= 126) sb.append(c);
47                     else sb.append("\\" +  toOctal3(c));
48                 }
49             }
50             System.out.println("package org.ibex.net.ssl;");
51             System.out.println("public final class RootCerts {");
52             System.out.println("    private final static String DATA = \"" + sb.toString() + "\";");
53             System.out.print(
54                     "    static {\n" +
55                     "        try {\n" + 
56                     "            org.ibex.net.SSL.addCompactCAKeys(new java.io.ByteArrayInputStream(unpack(DATA)));\n" + 
57                     "        } catch(Exception e) {\n" + 
58                     "            System.err.println(\"Error loading root CA keys: \" + e.getMessage());\n" + 
59                     "        }\n" +
60                     "    }\n");
61             System.out.println("    public static void load() {   }");  // force clinit
62             System.out.print(
63                     "    private static byte[] unpack(String s) {\n" + 
64                     "        int len = s.length();\n" + 
65                     "        if(len % 8 != 0) throw new IllegalArgumentException(\"not a multiple of 8\");\n" + 
66                     "        byte[] ret = new byte[(len / 8) * 7];\n" + 
67                     "        for(int i=0; i<len; i += 8) {\n" + 
68                     "            long l = 0;\n" + 
69                     "            for(int j=0;j<8;j++) {\n" + 
70                     "                l <<= 7;\n" + 
71                     "                l |= (s.charAt(i + j) & 0x7fL);\n" + 
72                     "            }\n" + 
73                     "            int base = (i / 8) * 7;\n" + 
74                     "            for(int j=6; j>=0; j--) {\n" + 
75                     "                ret[base + j] = (byte)(l & 0xff);\n" + 
76                     "                l >>>= 8;\n" + 
77                     "            }\n" + 
78                     "        }\n" + 
79                     "        return ret;\n" + 
80                     "    }");
81             System.out.println("}");
82         } else {
83             throw new Error("unknown format");
84         }
85     }
86     
87     private final static String toOctal3(int n) {
88         char[] buf = new char[3];
89         for(int i=2;i>=0;i--) {
90             buf[i] = (char) ('0' + (n & 7));
91             n >>= 3;
92         }
93         return new String(buf);
94     }
95 }