1 package club.wpia.gigi.pages;
3 import static club.wpia.gigi.Gigi.*;
5 import java.io.IOException;
6 import java.io.PrintWriter;
7 import java.math.BigInteger;
8 import java.security.cert.X509Certificate;
11 import javax.servlet.http.HttpServletRequest;
12 import javax.servlet.http.HttpServletResponse;
13 import javax.servlet.http.HttpSession;
15 import club.wpia.gigi.GigiApiException;
16 import club.wpia.gigi.database.GigiPreparedStatement;
17 import club.wpia.gigi.database.GigiResultSet;
18 import club.wpia.gigi.dbObjects.CertificateOwner;
19 import club.wpia.gigi.dbObjects.Group;
20 import club.wpia.gigi.dbObjects.User;
21 import club.wpia.gigi.localisation.Language;
22 import club.wpia.gigi.output.template.Form;
23 import club.wpia.gigi.output.template.TranslateCommand;
24 import club.wpia.gigi.pages.main.RegisterPage;
25 import club.wpia.gigi.util.AuthorizationContext;
26 import club.wpia.gigi.util.PasswordHash;
27 import club.wpia.gigi.util.RateLimit;
28 import club.wpia.gigi.util.RateLimit.RateLimitException;
29 import club.wpia.gigi.util.ServerConstants;
30 import club.wpia.gigi.util.ServerConstants.Host;
32 public class LoginPage extends Page {
34 public static final RateLimit RATE_LIMIT = new RateLimit(10, 5 * 60 * 1000);
36 public class LoginForm extends Form {
38 public LoginForm(HttpServletRequest hsr) {
43 public RedirectResult submit(HttpServletRequest req) throws GigiApiException {
44 if (RegisterPage.RATE_LIMIT.isLimitExceeded(req.getRemoteAddr())) {
45 throw new RateLimitException();
48 return new RedirectResult(redirectPath(req));
52 protected void outputContent(PrintWriter out, Language l, Map<String, Object> vars) {
53 getDefaultTemplate().output(out, l, vars);
58 public static final String LOGIN_RETURNPATH = "login-returnpath";
61 super("Password Login");
65 public void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
66 if (req.getHeader("Host").equals(ServerConstants.getHostNamePortSecure(Host.SECURE))) {
67 resp.getWriter().println(getLanguage(req).getTranslation("Authentication with certificate failed. Try another certificate or use a password."));
69 new LoginForm(req).output(resp.getWriter(), getLanguage(req), getDefaultVars(req));
74 public void doPost(HttpServletRequest req, HttpServletResponse resp) throws IOException {
75 if (Form.printFormErrors(req, resp.getWriter())) {
76 Form.getForm(req, LoginForm.class).output(resp.getWriter(), getLanguage(req), getDefaultVars(req));
81 public boolean beforeTemplate(HttpServletRequest req, HttpServletResponse resp) throws IOException {
82 if (req.getSession().getAttribute("loggedin") == null) {
83 X509Certificate cert = getCertificateFromRequest(req);
85 tryAuthWithCertificate(req, cert);
87 if (req.getMethod().equals("POST")) {
88 return Form.getForm(req, LoginForm.class).submitExceptionProtected(req, resp);
92 if (req.getSession().getAttribute("loggedin") != null) {
93 resp.sendRedirect(redirectPath(req));
99 private static String redirectPath(HttpServletRequest req) {
100 String redir = (String) req.getAttribute(LOGIN_RETURNPATH);
103 if ( !s.startsWith("/")) {
113 public boolean needsLogin() {
117 private void tryAuthWithUnpw(HttpServletRequest req) throws GigiApiException {
118 String un = req.getParameter("username");
119 String pw = req.getParameter("password");
120 try (GigiPreparedStatement ps = new GigiPreparedStatement("SELECT `password`, `id` FROM `users` WHERE `email`=? AND verified='1'")) {
122 GigiResultSet rs = ps.executeQuery();
124 String dbHash = rs.getString(1);
125 String hash = PasswordHash.verifyHash(pw, dbHash);
127 if ( !hash.equals(dbHash)) {
128 try (GigiPreparedStatement gps = new GigiPreparedStatement("UPDATE `users` SET `password`=? WHERE `email`=?")) {
129 gps.setString(1, hash);
130 gps.setString(2, un);
134 loginSession(req, User.getById(rs.getInt(2)));
135 req.getSession().setAttribute(LOGIN_METHOD, new TranslateCommand("Password"));
140 throw new GigiApiException("Username and password didn't match.");
143 public static User getUser(HttpServletRequest req) {
144 AuthorizationContext ac = getAuthorizationContext(req);
148 return ac.getActor();
151 public static AuthorizationContext getAuthorizationContext(HttpServletRequest req) {
152 return ((AuthorizationContext) req.getSession().getAttribute(AUTH_CONTEXT));
155 private void tryAuthWithCertificate(HttpServletRequest req, X509Certificate x509Certificate) {
156 BigInteger serial = extractSerialFormCert(x509Certificate);
157 User user = fetchUserBySerial(serial);
161 loginSession(req, user);
162 req.getSession().setAttribute(CERT_SERIAL, serial);
163 req.getSession().setAttribute(CERT_ISSUER, x509Certificate.getIssuerDN());
164 req.getSession().setAttribute(LOGIN_METHOD, new TranslateCommand("Certificate"));
167 public static BigInteger extractSerialFormCert(X509Certificate x509Certificate) {
168 return x509Certificate.getSerialNumber();
171 public static User fetchUserBySerial(BigInteger serial) {
172 CertificateOwner o = CertificateOwner.getByEnabledSerial(serial);
173 if (o == null || !(o instanceof User)) {
179 public static X509Certificate getCertificateFromRequest(HttpServletRequest req) {
180 X509Certificate[] cert = (X509Certificate[]) req.getAttribute("javax.servlet.request.X509Certificate");
181 X509Certificate uc = null;
182 if (cert != null && cert[0] != null) {
188 private static final Group LOGIN_BLOCKED = Group.BLOCKED_LOGIN;
190 private void loginSession(HttpServletRequest req, User user) {
191 if (user.isInGroup(LOGIN_BLOCKED)) {
194 req.setAttribute(LOGIN_RETURNPATH, req.getSession().getAttribute(LOGIN_RETURNPATH));
195 req.getSession().invalidate();
196 HttpSession hs = req.getSession();
197 hs.setAttribute(LOGGEDIN, true);
198 hs.setAttribute(Language.SESSION_ATTRIB_NAME, user.getPreferredLocale());
199 hs.setAttribute(AUTH_CONTEXT, new AuthorizationContext(user, user));
203 public boolean isPermitted(AuthorizationContext ac) {