util-testing/org/cacert/gigi/pages/Manager.java

   1 package org.cacert.gigi.pages;
   2
   3 import java.io.IOException;
   4 import java.io.PrintWriter;
   5 import java.lang.reflect.Field;
   6 import java.security.GeneralSecurityException;
   7 import java.security.KeyPair;
   8 import java.security.KeyPairGenerator;
   9 import java.security.Signature;
  10 import java.sql.Date;
  11 import java.util.Base64;
  12 import java.util.Calendar;
  13 import java.util.GregorianCalendar;
  14 import java.util.HashMap;
  15 import java.util.Iterator;
  16 import java.util.LinkedList;
  17 import java.util.Locale;
  18 import java.util.Map;
  19 import java.util.Properties;
  20
  21 import javax.servlet.http.HttpServletRequest;
  22 import javax.servlet.http.HttpServletResponse;
  23
  24 import org.cacert.gigi.Gigi;
  25 import org.cacert.gigi.GigiApiException;
  26 import org.cacert.gigi.crypto.SPKAC;
  27 import org.cacert.gigi.database.DatabaseConnection;
  28 import org.cacert.gigi.database.GigiPreparedStatement;
  29 import org.cacert.gigi.dbObjects.Certificate;
  30 import org.cacert.gigi.dbObjects.Certificate.CertificateStatus;
  31 import org.cacert.gigi.dbObjects.CertificateOwner;
  32 import org.cacert.gigi.dbObjects.Digest;
  33 import org.cacert.gigi.dbObjects.Domain;
  34 import org.cacert.gigi.dbObjects.DomainPingType;
  35 import org.cacert.gigi.dbObjects.EmailAddress;
  36 import org.cacert.gigi.dbObjects.Group;
  37 import org.cacert.gigi.dbObjects.Name;
  38 import org.cacert.gigi.dbObjects.User;
  39 import org.cacert.gigi.email.EmailProvider;
  40 import org.cacert.gigi.localisation.Language;
  41 import org.cacert.gigi.output.template.IterableDataset;
  42 import org.cacert.gigi.output.template.Template;
  43 import org.cacert.gigi.pages.account.certs.CertificateRequest;
  44 import org.cacert.gigi.ping.DomainPinger;
  45 import org.cacert.gigi.ping.PingerDaemon;
  46 import org.cacert.gigi.util.AuthorizationContext;
  47 import org.cacert.gigi.util.Notary;
  48
  49 import sun.security.x509.X509Key;
  50
  51 public class Manager extends Page {
  52
  53     public static final String PATH = "/manager";
  54
  55     Field f;
  56
  57     private static HashMap<DomainPingType, DomainPinger> dps;
  58
  59     private Manager() {
  60         super("Test Manager");
  61         try {
  62             f = EmailAddress.class.getDeclaredField("hash");
  63             f.setAccessible(true);
  64         } catch (ReflectiveOperationException e) {
  65             // TODO
  66             System.out.println("I don't have 'hash', we are working probably in layered mode. Test Manager may not work.");
  67             // throw new Error(e);
  68         }
  69
  70         try {
  71             Field gigiInstance = Gigi.class.getDeclaredField("instance");
  72             gigiInstance.setAccessible(true);
  73             Gigi g = (Gigi) gigiInstance.get(null);
  74
  75             Field gigiPinger = Gigi.class.getDeclaredField("pinger");
  76             gigiPinger.setAccessible(true);
  77             PingerDaemon pd = (PingerDaemon) gigiPinger.get(g);
  78
  79             Field f = PingerDaemon.class.getDeclaredField("pingers");
  80             f.setAccessible(true);
  81             dps = (HashMap<DomainPingType, DomainPinger>) f.get(pd);
  82             HashMap<DomainPingType, DomainPinger> pingers = new HashMap<>();
  83             for (DomainPingType dpt : DomainPingType.values()) {
  84                 pingers.put(dpt, new PingerFetcher(dpt));
  85             }
  86             f.set(pd, pingers);
  87         } catch (ReflectiveOperationException e) {
  88             e.printStackTrace();
  89         }
  90     }
  91
  92     public User[] getAssurers() {
  93         if (assurers != null) {
  94             return assurers;
  95         }
  96         assurers = new User[10];
  97         try {
  98             GigiPreparedStatement ps = DatabaseConnection.getInstance().prepare("INSERT INTO `notary` SET `from`=?, `to`=?, `points`=?, `location`=?, `date`=?");
  99             for (int i = 0; i < assurers.length; i++) {
 100                 String mail = "test-assurer" + i + "@example.com";
 101                 User u = User.getByEmail(mail);
 102                 if (u == null) {
 103                     System.out.println("Creating assurer");
 104                     createUser(mail);
 105                     u = User.getByEmail(mail);
 106                     passCATS(u);
 107                     ps.setInt(1, u.getId());
 108                     ps.setInt(2, u.getId());
 109                     ps.setInt(3, 100);
 110                     ps.setString(4, "Manager init code");
 111                     ps.setString(5, "1990-01-01");
 112                     ps.execute();
 113                 }
 114                 assurers[i] = u;
 115
 116             }
 117         } catch (ReflectiveOperationException | GigiApiException e) {
 118             e.printStackTrace();
 119         }
 120         return assurers;
 121     }
 122
 123     private void passCATS(User u) {
 124         GigiPreparedStatement ps = DatabaseConnection.getInstance().prepare("INSERT INTO cats_passed SET user_id=?, variant_id=1");
 125         ps.setInt(1, u.getId());
 126         ps.execute();
 127     }
 128
 129     private static Manager instance;
 130
 131     Template t = new Template(Manager.class.getResource("ManagerMails.templ"));
 132
 133     HashMap<String, LinkedList<String>> emails = new HashMap<>();
 134
 135     public static Manager getInstance() {
 136         if (instance == null) {
 137             instance = new Manager();
 138         }
 139         return instance;
 140     }
 141
 142     public static class MailFetcher extends EmailProvider {
 143
 144         public MailFetcher(Properties p) {}
 145
 146         @Override
 147         public String checkEmailServer(int forUid, String address) throws IOException {
 148             return OK;
 149         }
 150
 151         @Override
 152         public synchronized void sendmail(String to, String subject, String message, String from, String replyto, String toname, String fromname, String errorsto, boolean extra) throws IOException {
 153             HashMap<String, LinkedList<String>> mails = Manager.getInstance().emails;
 154             LinkedList<String> hismails = mails.get(to);
 155             if (hismails == null) {
 156                 mails.put(to, hismails = new LinkedList<>());
 157             }
 158             hismails.addFirst(subject + "\n" + message);
 159         }
 160
 161     }
 162
 163     public static class PingerFetcher extends DomainPinger {
 164
 165         private DomainPingType dpt;
 166
 167         public PingerFetcher(DomainPingType dpt) {
 168             this.dpt = dpt;
 169         }
 170
 171         @Override
 172         public void ping(Domain domain, String configuration, CertificateOwner target, int confId) {
 173             System.out.println("Test: " + domain);
 174             dps.get(dpt).ping(domain, configuration, target, confId);
 175         }
 176
 177     }
 178
 179     public void batchCreateUsers(String mailPrefix, String domain, int amount, PrintWriter out) {
 180
 181         try {
 182             if (amount > 100) {
 183                 out.print("100 at most, please.");
 184                 return;
 185             }
 186             for (int i = 0; i < amount; i++) {
 187                 String email = mailPrefix + i + "@" + domain;
 188                 createUser(email);
 189             }
 190         } catch (ReflectiveOperationException e) {
 191             out.println("failed");
 192             e.printStackTrace();
 193         } catch (GigiApiException e) {
 194             out.println("failed: " + e.getMessage());
 195             e.printStackTrace();
 196         }
 197     }
 198
 199     private void createUser(String email) throws GigiApiException, IllegalAccessException {
 200         Calendar gc = GregorianCalendar.getInstance();
 201         gc.set(1990, 0, 1);
 202         User u = new User(email, "xvXV12°§", new Name("Först", "Läst", "Müddle", "Süffix"), new Date(gc.getTime().getTime()), Locale.ENGLISH);
 203         EmailAddress ea = u.getEmails()[0];
 204         if (f == null) {
 205             System.out.println("verification failed");
 206             return;
 207         }
 208         String hash = (String) f.get(ea);
 209
 210         ea.verify(hash);
 211     }
 212
 213     User[] assurers;
 214
 215     @Override
 216     public void doPost(HttpServletRequest req, HttpServletResponse resp) throws IOException {
 217         if (req.getParameter("create") != null) {
 218             batchCreateUsers(req.getParameter("prefix"), req.getParameter("suffix"), Integer.parseInt(req.getParameter("amount")), resp.getWriter());
 219             resp.getWriter().println("User batch created.");
 220         } else if (req.getParameter("addpriv") != null || req.getParameter("delpriv") != null) {
 221             User u = User.getByEmail(req.getParameter("email"));
 222             if (u == null) {
 223                 resp.getWriter().println("User not found.");
 224                 return;
 225             }
 226             if (req.getParameter("addpriv") != null) {
 227                 u.grantGroup(u, Group.getByString(req.getParameter("priv")));
 228                 resp.getWriter().println("Privilege granted");
 229             } else {
 230                 u.revokeGroup(u, Group.getByString(req.getParameter("priv")));
 231                 resp.getWriter().println("Privilege revoked");
 232             }
 233         } else if (req.getParameter("fetch") != null) {
 234             String mail = req.getParameter("femail");
 235             fetchMails(req, resp, mail);
 236         } else if (req.getParameter("cats") != null) {
 237             String mail = req.getParameter("catsEmail");
 238             User byEmail = User.getByEmail(mail);
 239             if (byEmail == null) {
 240                 resp.getWriter().println("User not found.");
 241                 return;
 242             }
 243             passCATS(byEmail);
 244             resp.getWriter().println("User has been passed CATS");
 245         } else if (req.getParameter("assure") != null) {
 246             String mail = req.getParameter("assureEmail");
 247             User byEmail = User.getByEmail(mail);
 248             if (byEmail == null) {
 249                 resp.getWriter().println("User not found.");
 250                 return;
 251             }
 252             try {
 253                 for (int i = 0; i < getAssurers().length; i++) {
 254                     Notary.assure(getAssurers()[i], byEmail, byEmail.getName(), byEmail.getDoB(), 10, "Testmanager Assure up code", "2014-11-06");
 255                 }
 256             } catch (GigiApiException e) {
 257                 throw new Error(e);
 258             }
 259             resp.getWriter().println("User has been assured.");
 260         } else if (req.getParameter("addEmail") != null) {
 261             User u = User.getByEmail(req.getParameter("addEmailEmail"));
 262             try {
 263                 EmailAddress ea = new EmailAddress(u, req.getParameter("addEmailNew"), Locale.ENGLISH);
 264                 if (f != null) {
 265                     String hash = (String) f.get(ea);
 266                     ea.verify(hash);
 267                     resp.getWriter().println("Email added and verified");
 268                 } else {
 269                     resp.getWriter().println("Email added but verificatio failed.");
 270                 }
 271             } catch (IllegalArgumentException e) {
 272                 e.printStackTrace();
 273                 resp.getWriter().println("An internal error occured.");
 274             } catch (IllegalAccessException e) {
 275                 e.printStackTrace();
 276                 resp.getWriter().println("An internal error occured.");
 277             } catch (GigiApiException e) {
 278                 e.format(resp.getWriter(), Language.getInstance(Locale.ENGLISH));
 279             }
 280         } else if (req.getParameter("addCert") != null) {
 281             User u = User.getByEmail(req.getParameter("addCertEmail"));
 282             try {
 283                 KeyPairGenerator kpg = KeyPairGenerator.getInstance("RSA");
 284                 kpg.initialize(4096);
 285                 KeyPair kp = kpg.generateKeyPair();
 286                 SPKAC s = new SPKAC((X509Key) kp.getPublic(), "challange");
 287                 Signature sign = Signature.getInstance("SHA512withRSA");
 288                 sign.initSign(kp.getPrivate());
 289
 290                 byte[] res = s.getEncoded(sign);
 291
 292                 CertificateRequest cr = new CertificateRequest(new AuthorizationContext(u, u), Base64.getEncoder().encodeToString(res), "challange");
 293                 cr.update(CertificateRequest.DEFAULT_CN, Digest.SHA512.toString(), "client", null, "", "email:" + u.getEmail(), resp.getWriter(), req);
 294                 Certificate draft = cr.draft();
 295                 draft.issue(null, "2y", u).waitFor(10000);
 296                 if (draft.getStatus() == CertificateStatus.ISSUED) {
 297                     resp.getWriter().println("added certificate");
 298                 } else {
 299                     resp.getWriter().println("signer failed");
 300                 }
 301             } catch (GeneralSecurityException e1) {
 302                 e1.printStackTrace();
 303                 resp.getWriter().println("error");
 304             } catch (GigiApiException e) {
 305                 e.format(resp.getWriter(), Language.getInstance(Locale.ENGLISH));
 306             } catch (InterruptedException e) {
 307                 e.printStackTrace();
 308                 resp.getWriter().println("interrupted");
 309             }
 310
 311         }
 312     }
 313
 314     private void fetchMails(HttpServletRequest req, HttpServletResponse resp, String mail) throws IOException {
 315         final LinkedList<String> mails = emails.get(mail);
 316         HashMap<String, Object> vars = new HashMap<>();
 317         vars.put("mail", mail);
 318         if (mails != null) {
 319             vars.put("mails", new IterableDataset() {
 320
 321                 Iterator<String> s = mails.iterator();
 322
 323                 @Override
 324                 public boolean next(Language l, Map<String, Object> vars) {
 325                     if ( !s.hasNext()) {
 326                         return false;
 327                     }
 328                     vars.put("body", s.next().replaceAll("(https?://\\S+)", "<a href=\"$1\">$1</a>"));
 329                     return true;
 330                 }
 331             });
 332         }
 333         t.output(resp.getWriter(), getLanguage(req), vars);
 334         if (mails == null) {
 335             resp.getWriter().println("No mails");
 336
 337         }
 338     }
 339
 340     private Template form = new Template(Manager.class.getResource("Manager.templ"));
 341
 342     @Override
 343     public boolean needsLogin() {
 344         return false;
 345     }
 346
 347     @Override
 348     public void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
 349         getAssurers();
 350         String pi = req.getPathInfo().substring(PATH.length());
 351         if (pi.length() > 1 && pi.startsWith("/fetch-")) {
 352             String mail = pi.substring(pi.indexOf('-', 2) + 1);
 353             fetchMails(req, resp, mail);
 354             return;
 355         }
 356
 357         form.output(resp.getWriter(), getLanguage(req), new HashMap<String, Object>());
 358     }
 359 }