1 package org.cacert.gigi;
3 import java.io.BufferedReader;
5 import java.io.FileInputStream;
6 import java.io.IOException;
7 import java.io.InputStreamReader;
8 import java.security.cert.X509Certificate;
9 import java.sql.PreparedStatement;
10 import java.sql.ResultSet;
11 import java.sql.SQLException;
12 import java.util.Calendar;
13 import java.util.HashMap;
15 import javax.servlet.ServletException;
16 import javax.servlet.http.HttpServlet;
17 import javax.servlet.http.HttpServletRequest;
18 import javax.servlet.http.HttpServletResponse;
19 import javax.servlet.http.HttpSession;
21 import org.cacert.gigi.database.DatabaseConnection;
22 import org.cacert.gigi.pages.LoginPage;
23 import org.cacert.gigi.pages.MainPage;
24 import org.cacert.gigi.pages.Page;
25 import org.cacert.gigi.util.PasswordHash;
26 import org.eclipse.jetty.util.log.Log;
28 public class Gigi extends HttpServlet {
29 public static final String LOGGEDIN = "loggedin";
30 public static final String USER = "user";
31 private static final long serialVersionUID = -6386785421902852904L;
32 private String[] baseTemplate;
33 private HashMap<String, Page> pages = new HashMap<String, Page>();
36 public void init() throws ServletException {
37 pages.put("/login", new LoginPage("CACert - Login"));
38 pages.put("/", new MainPage("CACert - Home"));
41 BufferedReader reader = new BufferedReader(new InputStreamReader(
42 new FileInputStream(new File("templates/base.html"))));
44 while ((tmp = reader.readLine()) != null) {
47 baseTemplate = templ.split("\\$content\\$");
48 } catch (Exception e) {
49 Log.getLogger(Gigi.class).warn("Error loading template!", e);
56 protected void service(HttpServletRequest req, HttpServletResponse resp)
57 throws ServletException, IOException {
58 X509Certificate[] cert = (X509Certificate[]) req
59 .getAttribute("javax.servlet.request.X509Certificate");
60 HttpSession hs = req.getSession(false);
61 if (hs == null || !((Boolean) hs.getAttribute(LOGGEDIN))) {
63 tryAuthWithCertificate(req, cert[0]);
64 hs = req.getSession(false);
67 if (hs != null && ((Boolean) hs.getAttribute("loggedin"))
68 && req.getPathInfo().equals("/login")) {
69 resp.sendRedirect("/");
72 if (req.getMethod().equals("POST") && req.getPathInfo() != null
73 && req.getPathInfo().equals("/login")) {
75 resp.sendRedirect("/");
78 if (req.getPathInfo() != null && req.getPathInfo().equals("/logout")) {
80 hs.setAttribute(LOGGEDIN, false);
83 resp.sendRedirect("/");
87 if ((hs == null || !((Boolean) hs.getAttribute("loggedin")))
88 && !"/login".equals(req.getPathInfo())) {
89 System.out.println(req.getPathInfo());
90 resp.sendRedirect("/login");
93 if (pages.containsKey(req.getPathInfo())) {
94 String b0 = baseTemplate[0];
95 Page p = pages.get(req.getPathInfo());
96 b0 = makeDynTempl(b0, p);
97 resp.setContentType("text/html");
98 resp.getWriter().print(b0);
100 String b1 = baseTemplate[1];
101 b1 = makeDynTempl(b1, p);
102 resp.getWriter().print(b1);
104 resp.sendError(404, "Page not found.");
109 private String makeDynTempl(String in, Page p) {
110 int year = Calendar.getInstance().get(Calendar.YEAR);
111 in = in.replaceAll("\\$title\\$", p.getTitle());
112 in = in.replaceAll("\\$year\\$", year + "");
115 private void authWithUnpw(HttpServletRequest req) {
116 String un = req.getParameter("username");
117 String pw = req.getParameter("password");
119 PreparedStatement ps = DatabaseConnection.getInstance().prepare(
120 "SELECT `password`, `id` FROM `users` WHERE `email`=?");
122 ResultSet rs = ps.executeQuery();
124 if (PasswordHash.verifyHash(pw, rs.getString(1))) {
125 HttpSession hs = req.getSession();
126 hs.setAttribute(LOGGEDIN, true);
127 hs.setAttribute(USER, new User(rs.getInt(2)));
131 } catch (SQLException e) {
135 private void tryAuthWithCertificate(HttpServletRequest req,
136 X509Certificate x509Certificate) {
137 // TODO ckeck if certificate is valid
138 HttpSession hs = req.getSession();
139 hs.setAttribute(LOGGEDIN, true);