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