Advertisement
Guest User

Untitled

a guest
May 23rd, 2017
93
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Java 3.11 KB | None | 0 0
  1. import javax.servlet.ServletException;
  2. import javax.servlet.http.HttpServlet;
  3. import javax.servlet.http.HttpServletRequest;
  4. import javax.servlet.http.HttpServletResponse;
  5. import javax.servlet.http.HttpSession;
  6.  
  7. public class LoginServlet extends HttpServlet {
  8.   public static String USER_KEY = "ServletLogin.user";
  9.  
  10.   public static String FIELD_USER = "username";
  11.  
  12.   public static String FIELD_PASSWORD = "password";
  13.  
  14.   public void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException,
  15.       java.io.IOException {
  16.     resp.setContentType("text/html");
  17.     java.io.PrintWriter out = resp.getWriter();
  18.     resp.setHeader("Expires", "Tues, 01 Jan 1980 00:00:00 GMT");
  19.     String uri = req.getRequestURI();
  20.  
  21.     HttpSession session = req.getSession();
  22.     String user = (String) session.getAttribute(USER_KEY);
  23.  
  24.     if (user == null) {
  25.       login(out, uri);
  26.       return;
  27.     }
  28.     out.println("<html>");
  29.     out.println("<head>");
  30.     out.println("<title>Welcome</title>");
  31.     out.println("</head>");
  32.     out.println("<body>");
  33.     out.println("<center><h2>Welcome to our site!</h2>");
  34.     out.println("</center><br><br>");
  35.     out.println("</body>");
  36.     out.println("</html>");
  37.     out.flush();
  38.   }
  39.  
  40.   public void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException,
  41.       java.io.IOException {
  42.     resp.setContentType("text/html");
  43.     java.io.PrintWriter out = resp.getWriter();
  44.     HttpSession session = req.getSession(true);
  45.     String user = (String) session.getAttribute(USER_KEY);
  46.  
  47.     if (user == null) {
  48.       String username = req.getParameter(FIELD_USER);
  49.       String password = req.getParameter(FIELD_PASSWORD);
  50.  
  51.       if (!validUser(username, password)) {
  52.         out.println("<html>");
  53.         out.println("<title>Invalid User</title>");
  54.         out.println("<body><center><h2>" + "Invalid User!</h2><br>");
  55.         out.println("Press the 'Back' button to try again");
  56.         out.println("</center></body></html>");
  57.         out.flush();
  58.         return;
  59.       }
  60.       session.setAttribute(USER_KEY, username);
  61.     }
  62.     resp.sendRedirect(req.getRequestURI());
  63.   }
  64.  
  65.   protected void login(java.io.PrintWriter out, String uri) throws java.io.IOException {
  66.     out.println("<html>");
  67.     out.println("<head>");
  68.     out.println("<title>Login</title>");
  69.     out.println("<center><h2>Welcome! Please login</h2>");
  70.     out.println("<br><form method=POST action=\"" + uri + "\">");
  71.     out.println("<table>");
  72.     out.println("<tr><td>User ID:</td>");
  73.     out.println("<td><input type=text name=" + FIELD_USER + " size=30></td></tr>");
  74.     out.println("<tr><td>Password:</td>");
  75.     out.println("<td><input type=password name=" + FIELD_PASSWORD + " size=10></td></tr>");
  76.     out.println("</table><br>");
  77.     out.println("<input type=submit value=\"Login\">");
  78.     out.println("</form></center></body></html>");
  79.   }
  80.  
  81.   protected boolean validUser(String username, String password) {
  82.     boolean valid = false;
  83.     if ((username != null) && (username.length() > 0)) {
  84.       valid = username.equals(password);
  85.     }
  86.  
  87.     return valid;
  88.   }
  89. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement