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