src/org/cacert/gigi/DevelLauncher.java

   1 package org.cacert.gigi;
   2
   3 import java.io.BufferedReader;
   4 import java.io.ByteArrayInputStream;
   5 import java.io.ByteArrayOutputStream;
   6 import java.io.DataOutputStream;
   7 import java.io.File;
   8 import java.io.FileInputStream;
   9 import java.io.IOException;
  10 import java.io.InputStream;
  11 import java.io.InputStreamReader;
  12 import java.io.OutputStream;
  13 import java.lang.reflect.Field;
  14 import java.net.URL;
  15 import java.nio.file.Files;
  16 import java.nio.file.Paths;
  17 import java.util.Collections;
  18 import java.util.HashMap;
  19 import java.util.Map;
  20 import java.util.Properties;
  21
  22 import javax.servlet.http.HttpServletRequest;
  23 import javax.servlet.http.HttpServletResponse;
  24
  25 import org.cacert.gigi.dbObjects.ObjectCache;
  26 import org.cacert.gigi.pages.Page;
  27 import org.kamranzafar.jtar.TarEntry;
  28 import org.kamranzafar.jtar.TarHeader;
  29 import org.kamranzafar.jtar.TarOutputStream;
  30
  31 public class DevelLauncher {
  32
  33     public static final boolean DEVEL = true;
  34
  35     public static void main(String[] args) throws Exception {
  36         Properties mainProps = new Properties();
  37         try (FileInputStream inStream = new FileInputStream("config/gigi.properties")) {
  38             mainProps.load(inStream);
  39         }
  40         for (int i = 0; i < args.length; i++) {
  41             if (args[i].equals("--port")) {
  42                 mainProps.setProperty("port", args[i + 1]);
  43             }
  44             i++;
  45         }
  46         try {
  47             String targetPort = mainProps.getProperty("http.port");
  48             String targetHost = mainProps.getProperty("name.www");
  49             URL u = new URL("http://" + targetHost + ":" + targetPort + "/kill");
  50             u.openStream();
  51         } catch (IOException e) {
  52         }
  53
  54         ByteArrayOutputStream chunkConfig = new ByteArrayOutputStream();
  55         DataOutputStream dos = new DataOutputStream(chunkConfig);
  56         byte[] cacerts = Files.readAllBytes(Paths.get("config/cacerts.jks"));
  57         byte[] keystore = Files.readAllBytes(Paths.get("config/keystore.pkcs12"));
  58
  59         DevelLauncher.writeGigiConfig(dos, "changeit".getBytes("UTF-8"), "changeit".getBytes("UTF-8"), mainProps, cacerts, keystore);
  60         dos.flush();
  61         InputStream oldin = System.in;
  62         System.setIn(new ByteArrayInputStream(chunkConfig.toByteArray()));
  63         Launcher.boot();
  64         addDevelPage();
  65         System.setIn(oldin);
  66         BufferedReader br = new BufferedReader(new InputStreamReader(System.in, "UTF-8"));
  67         System.out.println("Cacert-gigi system sucessfully started.");
  68         System.out.println("Press enter to shutdown.");
  69         br.readLine();
  70         System.exit(0);
  71     }
  72
  73     public static void addDevelPage() {
  74         try {
  75             Field instF = Gigi.class.getDeclaredField("instance");
  76             Field pageF = Gigi.class.getDeclaredField("pages");
  77             instF.setAccessible(true);
  78             pageF.setAccessible(true);
  79             Object gigi = instF.get(null);
  80
  81             // Check if we got a proper map (as much as we can tell)
  82             Object pagesObj = pageF.get(gigi);
  83             @SuppressWarnings("unchecked")
  84             HashMap<String, Page> pages = pagesObj instanceof Map ? new HashMap<>((Map<String, Page>) pagesObj) : null;
  85
  86             pages.put("/manage", new Page("Page-manager") {
  87
  88                 @Override
  89                 public void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
  90                     ObjectCache.clearAllCaches();
  91                     resp.getWriter().println("All caches cleared.");
  92                     System.out.println("Caches cleared.");
  93
  94                 }
  95
  96                 @Override
  97                 public boolean needsLogin() {
  98                     return false;
  99                 }
 100
 101             });
 102
 103             pages.put("/kill", new Page("Kill") {
 104
 105                 @Override
 106                 public void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
 107                     System.exit(0);
 108                 }
 109
 110                 @Override
 111                 public boolean needsLogin() {
 112                     return false;
 113                 }
 114             });
 115
 116             pageF.set(gigi, Collections.unmodifiableMap(pages));
 117         } catch (ReflectiveOperationException e) {
 118             e.printStackTrace();
 119         }
 120     }
 121
 122     public static void writeGigiConfig(OutputStream target, byte[] keystorepw, byte[] truststorepw, Properties mainprop, byte[] cacerts, byte[] keystore) throws IOException {
 123         TarOutputStream tos = new TarOutputStream(target);
 124         ByteArrayOutputStream baos = new ByteArrayOutputStream();
 125         mainprop.store(baos, "");
 126
 127         putTarEntry(baos.toByteArray(), tos, "gigi.properties");
 128         putTarEntry(keystorepw, tos, "keystorepw");
 129         putTarEntry(truststorepw, tos, "truststorepw");
 130         putTarEntry(keystore, tos, "keystore.pkcs12");
 131         putTarEntry(cacerts, tos, "cacerts.jks");
 132         tos.close();
 133
 134     }
 135
 136     private static void putTarEntry(byte[] data, TarOutputStream tos, String name) throws IOException {
 137         TarHeader th = new TarHeader();
 138         th.name = new StringBuffer(name);
 139         th.size = data.length;
 140         tos.putNextEntry(new TarEntry(th));
 141         tos.write(data);
 142     }
 143
 144     public static void writeChunk(DataOutputStream dos, byte[] chunk) throws IOException {
 145         dos.writeInt(chunk.length);
 146         dos.write(chunk);
 147     }
 148
 149     public static void launch(Properties props, File cacerts, File keystore) throws IOException {
 150         ByteArrayOutputStream config = new ByteArrayOutputStream();
 151         props.store(config, "");
 152     }
 153 }