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