From f4a45264cb8f0cfeacdb89ce9bf4c60474923f80 Mon Sep 17 00:00:00 2001
From: Ugnich Anton
Date: Sat, 14 Dec 2013 00:43:10 +0700
Subject: getVisitorUser sent X-Username header

---
 src/java/com/juick/http/www/Login.java | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'src/java/com/juick/http/www/Login.java')

diff --git a/src/java/com/juick/http/www/Login.java b/src/java/com/juick/http/www/Login.java
index 329cb767..954d21c9 100644
--- a/src/java/com/juick/http/www/Login.java
+++ b/src/java/com/juick/http/www/Login.java
@@ -36,7 +36,7 @@ import javax.servlet.http.HttpServletResponse;
 public class Login {
 
     protected void doGetLoginForm(Connection sql, HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
-        com.juick.User visitor = Utils.getVisitorUser(sql, request);
+        com.juick.User visitor = Utils.getVisitorUser(sql, request, response);
         if (visitor != null) {
             Utils.sendTemporaryRedirect(response, "/");
             return;
@@ -244,12 +244,12 @@ public class Login {
     }
 
     protected void doGetLogout(Connection sql, HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
-        int uid = Utils.getVisitorUID(sql, request);
-        if (uid > 0) {
+        com.juick.User visitor = Utils.getVisitorUser(sql, request, response);
+        if (visitor != null) {
             PreparedStatement stmt = null;
             try {
                 stmt = sql.prepareStatement("DELETE FROM logins WHERE user_id=?");
-                stmt.setInt(1, uid);
+                stmt.setInt(1, visitor.UID);
                 stmt.executeUpdate();
             } catch (SQLException e) {
                 System.err.println(e);
-- 
cgit v1.2.3