]> WPIA git - gigi.git/blob - src/org/cacert/gigi/pages/LoginPage.java
fix: SQL change database call pattern
[gigi.git] / src / org / cacert / gigi / pages / LoginPage.java
1 package org.cacert.gigi.pages;
2
3 import static org.cacert.gigi.Gigi.*;
4
5 import java.io.IOException;
6 import java.io.PrintWriter;
7 import java.security.cert.X509Certificate;
8 import java.util.HashMap;
9 import java.util.Map;
10
11 import javax.servlet.http.HttpServletRequest;
12 import javax.servlet.http.HttpServletResponse;
13 import javax.servlet.http.HttpSession;
14
15 import org.cacert.gigi.GigiApiException;
16 import org.cacert.gigi.database.GigiPreparedStatement;
17 import org.cacert.gigi.database.GigiResultSet;
18 import org.cacert.gigi.dbObjects.CertificateOwner;
19 import org.cacert.gigi.dbObjects.Group;
20 import org.cacert.gigi.dbObjects.User;
21 import org.cacert.gigi.localisation.Language;
22 import org.cacert.gigi.output.template.Form;
23 import org.cacert.gigi.util.AuthorizationContext;
24 import org.cacert.gigi.util.PasswordHash;
25 import org.cacert.gigi.util.ServerConstants;
26
27 public class LoginPage extends Page {
28
29     public class LoginForm extends Form {
30
31         public LoginForm(HttpServletRequest hsr) {
32             super(hsr);
33         }
34
35         @Override
36         public boolean submit(PrintWriter out, HttpServletRequest req) throws GigiApiException {
37             tryAuthWithUnpw(req);
38             return false;
39         }
40
41         @Override
42         protected void outputContent(PrintWriter out, Language l, Map<String, Object> vars) {
43             getDefaultTemplate().output(out, l, vars);
44         }
45
46     }
47
48     public static final String LOGIN_RETURNPATH = "login-returnpath";
49
50     public LoginPage(String title) {
51         super(title);
52     }
53
54     @Override
55     public void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
56         if (req.getHeader("Host").equals(ServerConstants.getSecureHostNamePort())) {
57             resp.getWriter().println(getLanguage(req).getTranslation("Authentication with certificate failed. Try another certificate or use a password."));
58         } else {
59             new LoginForm(req).output(resp.getWriter(), getLanguage(req), new HashMap<String, Object>());
60         }
61     }
62
63     @Override
64     public boolean beforeTemplate(HttpServletRequest req, HttpServletResponse resp) throws IOException {
65         String redir = (String) req.getSession().getAttribute(LOGIN_RETURNPATH);
66         if (req.getSession().getAttribute("loggedin") == null) {
67             X509Certificate cert = getCertificateFromRequest(req);
68             if (cert != null) {
69                 tryAuthWithCertificate(req, cert);
70             }
71             if (req.getMethod().equals("POST")) {
72                 try {
73                     Form.getForm(req, LoginForm.class).submit(resp.getWriter(), req);
74                 } catch (GigiApiException e) {
75                 }
76             }
77         }
78
79         if (req.getSession().getAttribute("loggedin") != null) {
80             String s = redir;
81             if (s != null) {
82                 if ( !s.startsWith("/")) {
83                     s = "/" + s;
84                 }
85                 resp.sendRedirect(s);
86             } else {
87                 resp.sendRedirect("/");
88             }
89             return true;
90         }
91         return false;
92     }
93
94     @Override
95     public boolean needsLogin() {
96         return false;
97     }
98
99     private void tryAuthWithUnpw(HttpServletRequest req) {
100         String un = req.getParameter("username");
101         String pw = req.getParameter("password");
102         try (GigiPreparedStatement ps = new GigiPreparedStatement("SELECT `password`, `id` FROM `users` WHERE `email`=? AND verified='1'")) {
103             ps.setString(1, un);
104             GigiResultSet rs = ps.executeQuery();
105             if (rs.next()) {
106                 String dbHash = rs.getString(1);
107                 String hash = PasswordHash.verifyHash(pw, dbHash);
108                 if (hash != null) {
109                     if ( !hash.equals(dbHash)) {
110                         try (GigiPreparedStatement gps = new GigiPreparedStatement("UPDATE `users` SET `password`=? WHERE `email`=?")) {
111                             gps.setString(1, hash);
112                             gps.setString(2, un);
113                             gps.executeUpdate();
114                         }
115                     }
116                     loginSession(req, User.getById(rs.getInt(2)));
117                     req.getSession().setAttribute(LOGIN_METHOD, "Password");
118                 }
119             }
120         }
121     }
122
123     public static User getUser(HttpServletRequest req) {
124         AuthorizationContext ac = getAuthorizationContext(req);
125         if (ac == null) {
126             return null;
127         }
128         return ac.getActor();
129     }
130
131     public static AuthorizationContext getAuthorizationContext(HttpServletRequest req) {
132         return ((AuthorizationContext) req.getSession().getAttribute(AUTH_CONTEXT));
133     }
134
135     private void tryAuthWithCertificate(HttpServletRequest req, X509Certificate x509Certificate) {
136         String serial = extractSerialFormCert(x509Certificate);
137         User user = fetchUserBySerial(serial);
138         if (user == null) {
139             return;
140         }
141         loginSession(req, user);
142         req.getSession().setAttribute(CERT_SERIAL, serial);
143         req.getSession().setAttribute(CERT_ISSUER, x509Certificate.getIssuerDN());
144         req.getSession().setAttribute(LOGIN_METHOD, "Certificate");
145     }
146
147     public static String extractSerialFormCert(X509Certificate x509Certificate) {
148         return x509Certificate.getSerialNumber().toString(16).toUpperCase();
149     }
150
151     public static User fetchUserBySerial(String serial) {
152         if ( !serial.matches("[A-Fa-f0-9]+")) {
153             throw new Error("serial malformed.");
154         }
155
156         CertificateOwner o = CertificateOwner.getByEnabledSerial(serial);
157         if (o == null || !(o instanceof User)) {
158             return null;
159         }
160         return (User) o;
161     }
162
163     public static X509Certificate getCertificateFromRequest(HttpServletRequest req) {
164         X509Certificate[] cert = (X509Certificate[]) req.getAttribute("javax.servlet.request.X509Certificate");
165         X509Certificate uc = null;
166         if (cert != null && cert[0] != null) {
167             uc = cert[0];
168         }
169         return uc;
170     }
171
172     private static final Group LOGIN_BLOCKED = Group.getByString("blockedlogin");
173
174     private void loginSession(HttpServletRequest req, User user) {
175         if (user.isInGroup(LOGIN_BLOCKED)) {
176             return;
177         }
178         req.getSession().invalidate();
179         HttpSession hs = req.getSession();
180         hs.setAttribute(LOGGEDIN, true);
181         hs.setAttribute(Language.SESSION_ATTRIB_NAME, user.getPreferredLocale());
182         hs.setAttribute(AUTH_CONTEXT, new AuthorizationContext(user, user));
183     }
184
185     @Override
186     public boolean isPermitted(AuthorizationContext ac) {
187         return ac == null;
188     }
189 }