src/club/wpia/gigi/Gigi.java

   1 package club.wpia.gigi;
   2
   3 import java.io.IOException;
   4 import java.io.PrintWriter;
   5 import java.io.UnsupportedEncodingException;
   6 import java.security.KeyStore;
   7 import java.security.cert.X509Certificate;
   8 import java.util.Calendar;
   9 import java.util.Collections;
  10 import java.util.HashMap;
  11 import java.util.LinkedList;
  12 import java.util.Locale;
  13 import java.util.Map;
  14 import java.util.Properties;
  15 import java.util.regex.Pattern;
  16
  17 import javax.servlet.ServletException;
  18 import javax.servlet.http.HttpServlet;
  19 import javax.servlet.http.HttpServletRequest;
  20 import javax.servlet.http.HttpServletResponse;
  21 import javax.servlet.http.HttpSession;
  22
  23 import club.wpia.gigi.database.DatabaseConnection;
  24 import club.wpia.gigi.database.DatabaseConnection.Link;
  25 import club.wpia.gigi.dbObjects.CACertificate;
  26 import club.wpia.gigi.dbObjects.CATS.CATSType;
  27 import club.wpia.gigi.dbObjects.CertificateProfile;
  28 import club.wpia.gigi.dbObjects.DomainPingConfiguration;
  29 import club.wpia.gigi.localisation.Language;
  30 import club.wpia.gigi.output.Menu;
  31 import club.wpia.gigi.output.MenuCollector;
  32 import club.wpia.gigi.output.PageMenuItem;
  33 import club.wpia.gigi.output.SimpleMenuItem;
  34 import club.wpia.gigi.output.SimpleUntranslatedMenuItem;
  35 import club.wpia.gigi.output.template.Form.CSRFException;
  36 import club.wpia.gigi.output.template.Outputable;
  37 import club.wpia.gigi.output.template.Template;
  38 import club.wpia.gigi.pages.AboutPage;
  39 import club.wpia.gigi.pages.HandlesMixedRequest;
  40 import club.wpia.gigi.pages.LoginPage;
  41 import club.wpia.gigi.pages.LogoutPage;
  42 import club.wpia.gigi.pages.MainPage;
  43 import club.wpia.gigi.pages.OneFormPage;
  44 import club.wpia.gigi.pages.Page;
  45 import club.wpia.gigi.pages.PasswordResetPage;
  46 import club.wpia.gigi.pages.RootCertPage;
  47 import club.wpia.gigi.pages.StaticPage;
  48 import club.wpia.gigi.pages.TestSecure;
  49 import club.wpia.gigi.pages.Verify;
  50 import club.wpia.gigi.pages.account.ChangePasswordPage;
  51 import club.wpia.gigi.pages.account.FindAgentAccess;
  52 import club.wpia.gigi.pages.account.History;
  53 import club.wpia.gigi.pages.account.MyDetails;
  54 import club.wpia.gigi.pages.account.UserTrainings;
  55 import club.wpia.gigi.pages.account.certs.CertificateAdd;
  56 import club.wpia.gigi.pages.account.certs.Certificates;
  57 import club.wpia.gigi.pages.account.domain.DomainOverview;
  58 import club.wpia.gigi.pages.account.domain.EditDomain;
  59 import club.wpia.gigi.pages.account.mail.MailOverview;
  60 import club.wpia.gigi.pages.admin.TTPAdminPage;
  61 import club.wpia.gigi.pages.admin.support.FindCertPage;
  62 import club.wpia.gigi.pages.admin.support.FindUserByDomainPage;
  63 import club.wpia.gigi.pages.admin.support.FindUserByEmailPage;
  64 import club.wpia.gigi.pages.admin.support.SupportEnterTicketPage;
  65 import club.wpia.gigi.pages.admin.support.SupportUserDetailsPage;
  66 import club.wpia.gigi.pages.error.AccessDenied;
  67 import club.wpia.gigi.pages.error.PageNotFound;
  68 import club.wpia.gigi.pages.main.RegisterPage;
  69 import club.wpia.gigi.pages.orga.CreateOrgPage;
  70 import club.wpia.gigi.pages.orga.ViewOrgPage;
  71 import club.wpia.gigi.pages.statistics.StatisticsRoles;
  72 import club.wpia.gigi.pages.wot.AssurePage;
  73 import club.wpia.gigi.pages.wot.Points;
  74 import club.wpia.gigi.pages.wot.RequestTTPPage;
  75 import club.wpia.gigi.ping.PingerDaemon;
  76 import club.wpia.gigi.util.AuthorizationContext;
  77 import club.wpia.gigi.util.DomainAssessment;
  78 import club.wpia.gigi.util.PasswordHash;
  79 import club.wpia.gigi.util.ServerConstants;
  80 import club.wpia.gigi.util.TimeConditions;
  81
  82 public final class Gigi extends HttpServlet {
  83
  84     public static final String LINK_HOST = "linkHost";
  85
  86     private class MenuBuilder {
  87
  88         private LinkedList<Menu> categories = new LinkedList<Menu>();
  89
  90         private HashMap<String, Page> pages = new HashMap<String, Page>();
  91
  92         private MenuCollector rootMenu;
  93
  94         public MenuBuilder() {}
  95
  96         private void putPage(String path, Page p, String category) {
  97             pages.put(path, p);
  98             if (category == null) {
  99                 return;
 100             }
 101             Menu m = getMenu(category);
 102             m.addItem(new PageMenuItem(p, path.replaceFirst("/?\\*$", "")));
 103
 104         }
 105
 106         private Menu getMenu(String category) {
 107             Menu m = null;
 108             for (Menu menu : categories) {
 109                 if (menu.getMenuName().equals(category)) {
 110                     m = menu;
 111                     break;
 112                 }
 113             }
 114             if (m == null) {
 115                 m = new Menu(category);
 116                 categories.add(m);
 117             }
 118             return m;
 119         }
 120
 121         public MenuCollector generateMenu() throws ServletException {
 122             putPage("/denied", new AccessDenied(), null);
 123             putPage("/error", new PageNotFound(), null);
 124             putPage("/login", new LoginPage(), null);
 125             getMenu("SomeCA.org").addItem(new SimpleMenuItem("https://" + ServerConstants.getWwwHostNamePort() + "/login", "Password Login") {
 126
 127                 @Override
 128                 public boolean isPermitted(AuthorizationContext ac) {
 129                     return ac == null;
 130                 }
 131             });
 132             getMenu("SomeCA.org").addItem(new SimpleMenuItem("https://" + ServerConstants.getSecureHostNamePortSecure() + "/login", "Certificate Login") {
 133
 134                 @Override
 135                 public boolean isPermitted(AuthorizationContext ac) {
 136                     return ac == null;
 137                 }
 138             });
 139             putPage("/", new MainPage(), null);
 140             putPage("/roots", new RootCertPage(truststore), "SomeCA.org");
 141             putPage(StatisticsRoles.PATH, new StatisticsRoles(), "SomeCA.org");
 142             putPage("/about", new AboutPage(), "SomeCA.org");
 143
 144             putPage("/secure", new TestSecure(), null);
 145             putPage(Verify.PATH, new Verify(), null);
 146             putPage(Certificates.PATH + "/*", new Certificates(false), "Certificates");
 147             putPage(RegisterPage.PATH, new RegisterPage(), "SomeCA.org");
 148             putPage(CertificateAdd.PATH, new CertificateAdd(), "Certificates");
 149             putPage(MailOverview.DEFAULT_PATH, new MailOverview(), "Certificates");
 150             putPage(DomainOverview.PATH, new DomainOverview(), "Certificates");
 151             putPage(EditDomain.PATH + "*", new EditDomain(), null);
 152
 153             putPage(AssurePage.PATH + "/*", new AssurePage(), "Web of Trust");
 154             putPage(Points.PATH, new Points(false), "Web of Trust");
 155             putPage(RequestTTPPage.PATH, new RequestTTPPage(), "Web of Trust");
 156
 157             putPage(TTPAdminPage.PATH + "/*", new TTPAdminPage(), "Admin");
 158             putPage(CreateOrgPage.DEFAULT_PATH, new CreateOrgPage(), "Organisation Admin");
 159             putPage(ViewOrgPage.DEFAULT_PATH + "/*", new ViewOrgPage(), "Organisation Admin");
 160
 161             putPage(SupportEnterTicketPage.PATH, new SupportEnterTicketPage(), "Support Console");
 162             putPage(FindUserByEmailPage.PATH, new FindUserByEmailPage(), "Support Console");
 163             putPage(FindUserByDomainPage.PATH, new FindUserByDomainPage(), "Support Console");
 164             putPage(FindCertPage.PATH, new FindCertPage(), "Support Console");
 165
 166             putPage(SupportUserDetailsPage.PATH + "*", new SupportUserDetailsPage(), null);
 167             putPage(ChangePasswordPage.PATH, new ChangePasswordPage(), "My Account");
 168             putPage(History.PATH, new History(false), "My Account");
 169             putPage(FindAgentAccess.PATH, new OneFormPage("Access to Find Agent", FindAgentAccess.class), "My Account");
 170             putPage(History.SUPPORT_PATH, new History(true), null);
 171             putPage(UserTrainings.PATH, new UserTrainings(false), "My Account");
 172             putPage(MyDetails.PATH, new MyDetails(), "My Account");
 173             putPage(UserTrainings.SUPPORT_PATH, new UserTrainings(true), null);
 174             putPage(Points.SUPPORT_PATH, new Points(true), null);
 175             putPage(Certificates.SUPPORT_PATH + "/*", new Certificates(true), null);
 176
 177             putPage(PasswordResetPage.PATH, new PasswordResetPage(), null);
 178             putPage(LogoutPage.PATH, new LogoutPage(), null);
 179
 180             if (testing) {
 181                 try {
 182                     Class<?> manager = Class.forName("club.wpia.gigi.pages.Manager");
 183                     Page p = (Page) manager.getMethod("getInstance").invoke(null);
 184                     String pa = (String) manager.getField("PATH").get(null);
 185                     putPage(pa + "/*", p, "Gigi test server");
 186                 } catch (ReflectiveOperationException e) {
 187                     e.printStackTrace();
 188                 }
 189             }
 190
 191             try {
 192                 putPage("/wot/rules", new StaticPage("Web of Trust Rules", AssurePage.class.getResourceAsStream("Rules.templ")), "Web of Trust");
 193             } catch (UnsupportedEncodingException e) {
 194                 throw new ServletException(e);
 195             }
 196             rootMenu = new MenuCollector();
 197
 198             Menu languages = new Menu("Language");
 199             addLanguages(languages);
 200             categories.add(languages);
 201             for (Menu menu : categories) {
 202                 menu.prepare();
 203                 rootMenu.put(menu);
 204             }
 205
 206             // rootMenu.prepare();
 207             return rootMenu;
 208         }
 209
 210         private void addLanguages(Menu languages) {
 211             for (Locale l : Language.getSupportedLocales()) {
 212                 languages.addItem(new SimpleUntranslatedMenuItem("?lang=" + l.toString(), l.getDisplayName(l)));
 213             }
 214         }
 215
 216         public Map<String, Page> getPages() {
 217             return Collections.unmodifiableMap(pages);
 218         }
 219     }
 220
 221     public static final String LOGGEDIN = "loggedin";
 222
 223     public static final String CERT_SERIAL = "club.wpia.gigi.serial";
 224
 225     public static final String CERT_ISSUER = "club.wpia.gigi.issuer";
 226
 227     public static final String AUTH_CONTEXT = "auth";
 228
 229     public static final String LOGIN_METHOD = "club.wpia.gigi.loginMethod";
 230
 231     private static final long serialVersionUID = -6386785421902852904L;
 232
 233     private static Gigi instance;
 234
 235     private static final Template baseTemplate = new Template(Gigi.class.getResource("Gigi.templ"));
 236
 237     private PingerDaemon pinger;
 238
 239     private KeyStore truststore;
 240
 241     private boolean testing;
 242
 243     private MenuCollector rootMenu;
 244
 245     private Map<String, Page> pages;
 246
 247     private boolean firstInstanceInited = false;
 248
 249     public Gigi(Properties conf, KeyStore truststore) {
 250         synchronized (Gigi.class) {
 251             if (instance != null) {
 252                 throw new IllegalStateException("Multiple Gigi instances!");
 253             }
 254             testing = conf.getProperty("testing") != null;
 255             instance = this;
 256             DomainAssessment.init(conf);
 257             DatabaseConnection.init(conf);
 258             TimeConditions.init(conf);
 259             PasswordHash.init(conf);
 260             this.truststore = truststore;
 261             pinger = new PingerDaemon(truststore);
 262             pinger.start();
 263         }
 264     }
 265
 266     @Override
 267     public synchronized void init() throws ServletException {
 268         if (firstInstanceInited) {
 269             super.init();
 270             return;
 271         }
 272         // ensure those static initializers are finished
 273         try (Link l = DatabaseConnection.newLink(false)) {
 274             CACertificate.getById(1);
 275             CertificateProfile.getById(1);
 276             CATSType.ASSURER_CHALLENGE.getDisplayName();
 277         } catch (InterruptedException e) {
 278             throw new Error(e);
 279         }
 280
 281         MenuBuilder mb = new MenuBuilder();
 282         rootMenu = mb.generateMenu();
 283         pages = mb.getPages();
 284
 285         firstInstanceInited = true;
 286         super.init();
 287     }
 288
 289     private Page getPage(String pathInfo) {
 290         if (pathInfo.endsWith("/") && !pathInfo.equals("/")) {
 291             pathInfo = pathInfo.substring(0, pathInfo.length() - 1);
 292         }
 293         Page page = pages.get(pathInfo);
 294         if (page != null) {
 295             return page;
 296         }
 297         page = pages.get(pathInfo + "/*");
 298         if (page != null) {
 299             return page;
 300         }
 301         int idx = pathInfo.lastIndexOf('/');
 302         if (idx == -1 || idx == 0) {
 303             return null;
 304         }
 305
 306         page = pages.get(pathInfo.substring(0, idx) + "/*");
 307         if (page != null) {
 308             return page;
 309         }
 310         int lIdx = pathInfo.lastIndexOf('/', idx - 1);
 311         if (lIdx == -1) {
 312             return null;
 313         }
 314         String lastResort = pathInfo.substring(0, lIdx) + "/*" + pathInfo.substring(idx);
 315         page = pages.get(lastResort);
 316         return page;
 317
 318     }
 319
 320     private static String staticTemplateVar = "//" + ServerConstants.getStaticHostNamePort();
 321
 322     private static String staticTemplateVarSecure = "//" + ServerConstants.getStaticHostNamePortSecure();
 323
 324     @Override
 325     protected void service(final HttpServletRequest req, final HttpServletResponse resp) throws ServletException, IOException {
 326         if ("/error".equals(req.getPathInfo()) || "/denied".equals(req.getPathInfo())) {
 327             if (DatabaseConnection.hasInstance()) {
 328                 serviceWithConnection(req, resp);
 329                 return;
 330             }
 331         }
 332         try (DatabaseConnection.Link l = DatabaseConnection.newLink( !req.getMethod().equals("POST"))) {
 333             serviceWithConnection(req, resp);
 334         } catch (InterruptedException e) {
 335             e.printStackTrace();
 336         }
 337     }
 338
 339     protected void serviceWithConnection(final HttpServletRequest req, final HttpServletResponse resp) throws ServletException, IOException {
 340         boolean isSecure = req.isSecure();
 341         addXSSHeaders(resp, isSecure);
 342         // Firefox only sends this, if it's a cross domain access; safari sends
 343         // it always
 344         String originHeader = req.getHeader("Origin");
 345         if (originHeader != null //
 346                 && !(originHeader.matches("^" + Pattern.quote("https://" + ServerConstants.getWwwHostNamePortSecure()) + "(/.*|)") || //
 347                         originHeader.matches("^" + Pattern.quote("http://" + ServerConstants.getWwwHostNamePort()) + "(/.*|)") || //
 348                         originHeader.matches("^" + Pattern.quote("https://" + ServerConstants.getSecureHostNamePortSecure()) + "(/.*|)"))) {
 349             resp.setContentType("text/html; charset=utf-8");
 350             resp.getWriter().println("<html><head><title>Alert</title></head><body>No cross domain access allowed.<br/><b>If you don't know why you're seeing this you may have been fished! Please change your password immediately!</b></body></html>");
 351             return;
 352         }
 353         HttpSession hs = req.getSession();
 354         String clientSerial = (String) hs.getAttribute(CERT_SERIAL);
 355         if (clientSerial != null) {
 356             X509Certificate[] cert = (X509Certificate[]) req.getAttribute("javax.servlet.request.X509Certificate");
 357             if (cert == null || cert[0] == null//
 358                     || !cert[0].getSerialNumber().toString(16).toLowerCase().equals(clientSerial) //
 359                     || !cert[0].getIssuerDN().equals(hs.getAttribute(CERT_ISSUER))) {
 360                 hs.invalidate();
 361                 resp.sendError(403, "Certificate mismatch.");
 362                 return;
 363             }
 364
 365         }
 366         if (req.getParameter("lang") != null) {
 367             Locale l = Language.getLocaleFromString(req.getParameter("lang"));
 368             Language lu = Language.getInstance(l);
 369             req.getSession().setAttribute(Language.SESSION_ATTRIB_NAME, lu != null ? lu.getLocale() : Locale.ENGLISH);
 370         }
 371         final Page p = getPage(req.getPathInfo());
 372
 373         if (p != null) {
 374             if ( !isSecure && (p.needsLogin() || p instanceof LoginPage || p instanceof RegisterPage)) {
 375                 resp.sendRedirect("https://" + ServerConstants.getWwwHostNamePortSecure() + req.getPathInfo());
 376                 return;
 377             }
 378             AuthorizationContext currentAuthContext = LoginPage.getAuthorizationContext(req);
 379             if ( !p.isPermitted(currentAuthContext)) {
 380                 if (hs.getAttribute("loggedin") == null) {
 381                     String request = req.getPathInfo();
 382                     request = request.split("\\?")[0];
 383                     hs.setAttribute(LoginPage.LOGIN_RETURNPATH, request);
 384                     resp.sendRedirect("/login");
 385                     return;
 386                 }
 387                 resp.sendError(403);
 388                 return;
 389             }
 390             if (p.beforeTemplate(req, resp)) {
 391                 return;
 392             }
 393             HashMap<String, Object> vars = new HashMap<String, Object>();
 394             // System.out.println(req.getMethod() + ": " + req.getPathInfo() +
 395             // " -> " + p);
 396             Outputable content = new Outputable() {
 397
 398                 @Override
 399                 public void output(PrintWriter out, Language l, Map<String, Object> vars) {
 400                     try {
 401                         if (req.getMethod().equals("POST")) {
 402                             if (req.getQueryString() != null && !(p instanceof HandlesMixedRequest)) {
 403                                 return;
 404                             }
 405                             p.doPost(req, resp);
 406                         } else {
 407                             p.doGet(req, resp);
 408                         }
 409                     } catch (CSRFException err) {
 410                         try {
 411                             resp.sendError(500, "CSRF invalid");
 412                         } catch (IOException e) {
 413                             e.printStackTrace();
 414                         }
 415                     } catch (IOException e) {
 416                         e.printStackTrace();
 417                     }
 418
 419                 }
 420             };
 421             Language lang = Page.getLanguage(req);
 422
 423             vars.put(Menu.AUTH_VALUE, currentAuthContext);
 424             vars.put("menu", rootMenu);
 425             vars.put("title", lang.getTranslation(p.getTitle()));
 426             vars.put("static", isSecure ? staticTemplateVarSecure : staticTemplateVar);
 427             vars.put("year", Calendar.getInstance().get(Calendar.YEAR));
 428             vars.put("content", content);
 429             if (isSecure) {
 430                 req.setAttribute(LINK_HOST, ServerConstants.getLinkHostNamePortSecure());
 431             } else {
 432                 req.setAttribute(LINK_HOST, ServerConstants.getLinkHostNamePort());
 433             }
 434             if (currentAuthContext != null) {
 435                 // TODO maybe move this information into the AuthContext object
 436                 vars.put("loginMethod", req.getSession().getAttribute(LOGIN_METHOD));
 437                 vars.put("authContext", currentAuthContext);
 438
 439             }
 440             resp.setContentType("text/html; charset=utf-8");
 441             baseTemplate.output(resp.getWriter(), lang, vars);
 442         } else {
 443             resp.sendError(404, "Page not found.");
 444         }
 445
 446     }
 447
 448     public static void addXSSHeaders(HttpServletResponse hsr, boolean doHttps) {
 449         hsr.addHeader("Access-Control-Allow-Origin", "https://" + ServerConstants.getWwwHostNamePortSecure() + " https://" + ServerConstants.getSecureHostNamePortSecure());
 450         hsr.addHeader("Access-Control-Max-Age", "60");
 451         if (doHttps) {
 452             hsr.addHeader("Content-Security-Policy", httpsCSP);
 453         } else {
 454             hsr.addHeader("Content-Security-Policy", httpCSP);
 455         }
 456         hsr.addHeader("Strict-Transport-Security", "max-age=31536000");
 457
 458     }
 459
 460     private static String httpsCSP = genHttpsCSP();
 461
 462     private static String httpCSP = genHttpCSP();
 463
 464     private static String genHttpsCSP() {
 465         StringBuffer csp = new StringBuffer();
 466         csp.append("default-src 'none'");
 467         csp.append(";font-src https://" + ServerConstants.getStaticHostNamePortSecure());
 468         csp.append(";img-src https://" + ServerConstants.getStaticHostNamePortSecure());
 469         csp.append(";media-src 'none'; object-src 'none'");
 470         csp.append(";script-src https://" + ServerConstants.getStaticHostNamePortSecure());
 471         csp.append(";style-src https://" + ServerConstants.getStaticHostNamePortSecure());
 472         csp.append(";form-action https://" + ServerConstants.getSecureHostNamePortSecure() + " https://" + ServerConstants.getWwwHostNamePortSecure());
 473         // csp.append(";report-url https://api.wpia.club/security/csp/report");
 474         return csp.toString();
 475     }
 476
 477     private static String genHttpCSP() {
 478         StringBuffer csp = new StringBuffer();
 479         csp.append("default-src 'none'");
 480         csp.append(";font-src http://" + ServerConstants.getStaticHostNamePort());
 481         csp.append(";img-src http://" + ServerConstants.getStaticHostNamePort());
 482         csp.append(";media-src 'none'; object-src 'none'");
 483         csp.append(";script-src http://" + ServerConstants.getStaticHostNamePort());
 484         csp.append(";style-src http://" + ServerConstants.getStaticHostNamePort());
 485         csp.append(";form-action https://" + ServerConstants.getSecureHostNamePortSecure() + " https://" + ServerConstants.getWwwHostNamePort());
 486         // csp.append(";report-url http://api.wpia.club/security/csp/report");
 487         return csp.toString();
 488     }
 489
 490     /**
 491      * Requests Pinging of domains.
 492      *
 493      * @param toReping
 494      *            if not null, the {@link DomainPingConfiguration} to test, if
 495      *            null, just re-check if there is something to do.
 496      */
 497     public static void notifyPinger(DomainPingConfiguration toReping) {
 498         if (toReping != null) {
 499             instance.pinger.queue(toReping);
 500         }
 501         instance.pinger.interrupt();
 502     }
 503
 504 }