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;
12 import javax.servlet.http.HttpServletRequest;
13 import javax.servlet.http.HttpServletResponse;
14 import javax.servlet.http.HttpSession;
16 import club.wpia.gigi.GigiApiException;
17 import club.wpia.gigi.database.GigiPreparedStatement;
18 import club.wpia.gigi.database.GigiResultSet;
19 import club.wpia.gigi.dbObjects.Certificate;
20 import club.wpia.gigi.dbObjects.CertificateOwner;
21 import club.wpia.gigi.dbObjects.Group;
22 import club.wpia.gigi.dbObjects.User;
23 import club.wpia.gigi.localisation.Language;
24 import club.wpia.gigi.output.template.Form;
25 import club.wpia.gigi.output.template.TranslateCommand;
26 import club.wpia.gigi.pages.main.RegisterPage;
27 import club.wpia.gigi.util.AuthorizationContext;
28 import club.wpia.gigi.util.PasswordHash;
29 import club.wpia.gigi.util.RateLimit;
30 import club.wpia.gigi.util.RateLimit.RateLimitException;
31 import club.wpia.gigi.util.ServerConstants;
32 import club.wpia.gigi.util.ServerConstants.Host;
34 public class LoginPage extends Page {
36 public static final RateLimit RATE_LIMIT = new RateLimit(10, 5 * 60 * 1000);
38 public class LoginForm extends Form {
40 public LoginForm(HttpServletRequest hsr) {
45 public RedirectResult submit(HttpServletRequest req) throws GigiApiException {
46 if (RegisterPage.RATE_LIMIT.isLimitExceeded(req.getRemoteAddr())) {
47 throw new RateLimitException();
50 return new RedirectResult(redirectPath(req));
54 protected void outputContent(PrintWriter out, Language l, Map<String, Object> vars) {
55 getDefaultTemplate().output(out, l, vars);
60 public static final String LOGIN_RETURNPATH = "login-returnpath";
63 super("Password Login");
67 public void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
68 if (req.getHeader("Host").equals(ServerConstants.getHostNamePortSecure(Host.SECURE))) {
69 resp.getWriter().println(getLanguage(req).getTranslation("Authentication with certificate failed. Try another certificate or use a password."));
71 new LoginForm(req).output(resp.getWriter(), getLanguage(req), getDefaultVars(req));
76 public void doPost(HttpServletRequest req, HttpServletResponse resp) throws IOException {
77 if (Form.printFormErrors(req, resp.getWriter())) {
78 Form.getForm(req, LoginForm.class).output(resp.getWriter(), getLanguage(req), getDefaultVars(req));
83 public boolean beforeTemplate(HttpServletRequest req, HttpServletResponse resp) throws IOException {
84 if (req.getSession().getAttribute("loggedin") == null) {
85 X509Certificate cert = getCertificateFromRequest(req);
87 tryAuthWithCertificate(req, cert);
89 if (req.getMethod().equals("POST")) {
90 return Form.getForm(req, LoginForm.class).submitExceptionProtected(req, resp);
94 if (req.getSession().getAttribute("loggedin") != null) {
95 resp.sendRedirect(redirectPath(req));
101 private static String redirectPath(HttpServletRequest req) {
102 String redir = (String) req.getAttribute(LOGIN_RETURNPATH);
105 if ( !s.startsWith("/")) {
115 public boolean needsLogin() {
119 private void tryAuthWithUnpw(HttpServletRequest req) throws GigiApiException {
120 String un = req.getParameter("username");
121 String pw = req.getParameter("password");
122 try (GigiPreparedStatement ps = new GigiPreparedStatement("SELECT `password`, `id` FROM `users` WHERE `email`=? AND verified='1'")) {
124 GigiResultSet rs = ps.executeQuery();
126 throw new GigiApiException("Username and password didn't match.");
129 User user = User.getById(rs.getInt(2));
131 throw new GigiApiException("Username and password didn't match.");
134 String dbHash = rs.getString(1);
135 String hash = PasswordHash.verifyHash(pw, dbHash);
137 if ( !hash.equals(dbHash)) {
138 try (GigiPreparedStatement gps = new GigiPreparedStatement("UPDATE `users` SET `password`=? WHERE `email`=?")) {
139 gps.setString(1, hash);
140 gps.setString(2, un);
145 loginSession(req, user, false);
146 req.getSession().setAttribute(LOGIN_METHOD, new TranslateCommand("Password"));
152 public static User getUser(HttpServletRequest req) {
153 AuthorizationContext ac = getAuthorizationContext(req);
157 return ac.getActor();
160 public static AuthorizationContext getAuthorizationContext(HttpServletRequest req) {
161 return ((AuthorizationContext) req.getSession().getAttribute(AUTH_CONTEXT));
164 private void tryAuthWithCertificate(HttpServletRequest req, X509Certificate x509Certificate) {
165 BigInteger serial = extractSerialFormCert(x509Certificate);
166 Certificate c = Certificate.getBySerial(serial);
167 User user = fetchUserBySerial(serial);
171 if (c.getExpiryDate().before(new Date()) || c.getRevocationDate() != null || c.isLoginEnabled() == false) {
174 loginSession(req, user, true);
175 req.getSession().setAttribute(CERT_SERIAL, serial);
176 req.getSession().setAttribute(CERT_ISSUER, x509Certificate.getIssuerDN());
177 req.getSession().setAttribute(LOGIN_METHOD, new TranslateCommand("Certificate"));
180 public static BigInteger extractSerialFormCert(X509Certificate x509Certificate) {
181 return x509Certificate.getSerialNumber();
184 public static User fetchUserBySerial(BigInteger serial) {
185 CertificateOwner o = CertificateOwner.getByEnabledSerial(serial);
186 if (o == null || !(o instanceof User)) {
192 public static X509Certificate getCertificateFromRequest(HttpServletRequest req) {
193 X509Certificate[] cert = (X509Certificate[]) req.getAttribute("javax.servlet.request.X509Certificate");
194 X509Certificate uc = null;
195 if (cert != null && cert[0] != null) {
201 private static final Group LOGIN_BLOCKED = Group.BLOCKED_LOGIN;
203 private void loginSession(HttpServletRequest req, User user, boolean isStronglyAuthenticated) {
204 if (user.isInGroup(LOGIN_BLOCKED)) {
207 req.setAttribute(LOGIN_RETURNPATH, req.getSession().getAttribute(LOGIN_RETURNPATH));
208 req.getSession().invalidate();
209 HttpSession hs = req.getSession();
210 hs.setAttribute(LOGGEDIN, true);
211 hs.setAttribute(Language.SESSION_ATTRIB_NAME, user.getPreferredLocale());
212 hs.setAttribute(AUTH_CONTEXT, new AuthorizationContext(user, user, isStronglyAuthenticated));
216 public boolean isPermitted(AuthorizationContext ac) {