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.security.cert.X509Certificate;
   7 import java.sql.PreparedStatement;
   8 import java.sql.ResultSet;
   9 import java.sql.SQLException;
  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.User;
  16 import org.cacert.gigi.database.DatabaseConnection;
  17 import org.cacert.gigi.util.PasswordHash;
  18
  19 public class LoginPage extends Page {
  20
  21     public static final String LOGIN_RETURNPATH = "login-returnpath";
  22
  23     public LoginPage(String title) {
  24         super(title);
  25     }
  26
  27     @Override
  28     public void doGet(HttpServletRequest req, HttpServletResponse resp) throws IOException {
  29         resp.getWriter().println("<form method='POST' action='/login'>" + "<input type='text' name='username'>" + "<input type='password' name='password'> <input type='submit' value='login'></form>");
  30     }
  31
  32     @Override
  33     public boolean beforeTemplate(HttpServletRequest req, HttpServletResponse resp) throws IOException {
  34         String redir = (String) req.getSession().getAttribute(LOGIN_RETURNPATH);
  35         if (req.getSession().getAttribute("loggedin") == null) {
  36             X509Certificate[] cert = (X509Certificate[]) req.getAttribute("javax.servlet.request.X509Certificate");
  37             if (cert != null && cert[0] != null) {
  38                 tryAuthWithCertificate(req, cert[0]);
  39             }
  40             if (req.getMethod().equals("POST")) {
  41                 tryAuthWithUnpw(req);
  42             }
  43         }
  44
  45         if (req.getSession().getAttribute("loggedin") != null) {
  46             String s = redir;
  47             if (s != null) {
  48                 if ( !s.startsWith("/")) {
  49                     s = "/" + s;
  50                 }
  51                 resp.sendRedirect(s);
  52             } else {
  53                 resp.sendRedirect("/");
  54             }
  55             return true;
  56         }
  57         return false;
  58     }
  59
  60     @Override
  61     public boolean needsLogin() {
  62         return false;
  63     }
  64
  65     private void tryAuthWithUnpw(HttpServletRequest req) {
  66         String un = req.getParameter("username");
  67         String pw = req.getParameter("password");
  68         try {
  69             PreparedStatement ps = DatabaseConnection.getInstance().prepare("SELECT `password`, `id` FROM `users` WHERE `email`=? AND locked='0' AND verified='1'");
  70             ps.setString(1, un);
  71             ResultSet rs = ps.executeQuery();
  72             if (rs.next()) {
  73                 if (PasswordHash.verifyHash(pw, rs.getString(1))) {
  74                     req.getSession().invalidate();
  75                     HttpSession hs = req.getSession();
  76                     hs.setAttribute(LOGGEDIN, true);
  77                     hs.setAttribute(USER, new User(rs.getInt(2)));
  78                 }
  79             }
  80             rs.close();
  81         } catch (SQLException e) {
  82             e.printStackTrace();
  83         }
  84     }
  85
  86     public static User getUser(HttpServletRequest req) {
  87         return (User) req.getSession().getAttribute(USER);
  88     }
  89
  90     private void tryAuthWithCertificate(HttpServletRequest req, X509Certificate x509Certificate) {
  91         String serial = x509Certificate.getSerialNumber().toString(16).toUpperCase();
  92         try {
  93             PreparedStatement ps = DatabaseConnection.getInstance().prepare("SELECT `memid` FROM `certs` WHERE `serial`=? AND `disablelogin`='0' AND `revoked` = " + "'0000-00-00 00:00:00'");
  94             ps.setString(1, serial);
  95             ResultSet rs = ps.executeQuery();
  96             if (rs.next()) {
  97                 req.getSession().invalidate();
  98                 HttpSession hs = req.getSession();
  99                 hs.setAttribute(LOGGEDIN, true);
 100                 hs.setAttribute(USER, new User(rs.getInt(1)));
 101             }
 102             rs.close();
 103         } catch (SQLException e) {
 104             e.printStackTrace();
 105         }
 106     }
 107
 108     @Override
 109     public boolean isPermitted(User u) {
 110         return u == null;
 111     }
 112 }