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