From 4be449207cd12831a1858860eba1c8bb0618aa80 Mon Sep 17 00:00:00 2001
From: Teddy <ted.sybil@gmail.com>
Date: Sun, 1 Sep 2013 23:49:12 +0800
Subject: ...

---
 server/piztor/easy_import.py |  2 +-
 server/piztor/model.py       |  1 +
 server/piztor/server.py      | 55 +++++++++++++++++++++++++++++++++-----------
 3 files changed, 44 insertions(+), 14 deletions(-)

(limited to 'server/piztor')

diff --git a/server/piztor/easy_import.py b/server/piztor/easy_import.py
index e103de7..261d44d 100644
--- a/server/piztor/easy_import.py
+++ b/server/piztor/easy_import.py
@@ -48,7 +48,7 @@ def import_user_data(data):
             g = find_or_create_group(int(cn), int(sn), session)
             um.sub.append(g)
         um.auth = UserAuth(user.password)
-        um.location = LocationInfo(lat = 0, lng = 0)
+        um.location = LocationInfo(lat = NOT_A_LAT, lng = NOT_A_LNG)
         session.add(um)
         session.commit()
 
diff --git a/server/piztor/model.py b/server/piztor/model.py
index dd44fa2..cade3ea 100644
--- a/server/piztor/model.py
+++ b/server/piztor/model.py
@@ -11,6 +11,7 @@ _SALT_LEN = 16
 _TOKEN_LEN = 16
 MAX_USERNAME_SIZE = 20
 MAX_PASSWORD_SIZE = 20
+NOT_A_LAT = NOT_A_LNG = 300
 
 _table_typical_settings = {
         'mysql_engine' : 'InnoDB',
diff --git a/server/piztor/server.py b/server/piztor/server.py
index 38b74c9..980ae03 100644
--- a/server/piztor/server.py
+++ b/server/piztor/server.py
@@ -102,14 +102,16 @@ class PushTunnel(object):
         self.blocked = False
 
     def close(self):
+        print "closing TUNNEL"
         if self.conn:
+            print "loseConn called"
             self.conn.transport.loseConnection()
 
     def add(self, pdata):
-        logger.info("-- Push data enqued --")
+        #logger.info("-- Push data enqued --")
         self.pending.append(pdata)
         if not self.blocked and len(self.pending) > _MAX_PENDING_PUSH:
-            logger.info("-- Push queue is full, discarded an obsolete push --")
+            #logger.info("-- Push queue is full, discarded an obsolete push --")
             self.pending.popleft()  # discard old push
 
     def on_receive(self, data):
@@ -117,22 +119,22 @@ class PushTunnel(object):
         length, optcode, fingerprint = struct.unpack("!LB32s", data)
         if front.finger_print != fingerprint:
             raise PiztorError
-        logger.info("-- Push data confirmed by client %s --", str(self.uid))
+        #logger.info("-- Push data confirmed by client %s --", str(self.uid))
         self.blocked = False
         self.push()
 
     def push(self):
         if self.blocked:
             return
-        print "Pushing via " + str(self.uid)
-        print "Pending size: " + str(len(self.pending))
-        logger.info("Pushing...")
+        #print "Pushing via " + str(self.uid)
+        #print "Pending size: " + str(len(self.pending))
+        #logger.info("Pushing...")
         if (self.conn is None) or len(self.pending) == 0:
             return
         front = self.pending.popleft()
         self.pending.appendleft(front)
         self.conn.transport.write(front.data)
-        logger.info("-- Wrote push: %s --", get_hex(front.data))
+        #logger.info("-- Wrote push: %s --", get_hex(front.data))
         self.blocked = True
 
     def clear(self):
@@ -146,6 +148,7 @@ class PushTunnel(object):
         self.conn = conn
 
     def on_connection_lost(self, conn):
+        print "TUNNEL closed"
         if conn == self.conn:
             self.conn = None
 
@@ -458,9 +461,6 @@ class UpdateSubscription(RequestHandler):
         return entry
 
     def handle(self, tr_data, conn):
-        print _MAX_AUTH_HEAD_SIZE
-        print _SectionSize.GROUP_ID * _MAX_SUB_LIST_SIZE
-        print _SectionSize.PADDING
 
         self.check_size(tr_data)
         logger.info("Reading update subscription data...")
@@ -517,6 +517,35 @@ class UserLogoutHandler(RequestHandler):
         if uauth is None:
             logger.warning("Authentication failure")
             return self.pack(struct.pack("!B", _StatusCode.auth_fail))
+
+        pt = RequestHandler.push_tunnels
+        u = uauth.user
+        loc = u.location
+        loc.lat = NOT_A_LAT
+        loc.lng = NOT_A_LNG
+
+        comp = self.session.query(GroupInfo) \
+                .filter(GroupInfo.id == u.comp_id).one()
+        sec = self.session.query(GroupInfo) \
+                .filter(GroupInfo.id == u.sec_id).one()
+
+        pdata = PushLocationData(u.id, loc.lat, loc.lng)
+        for user in comp.subscribers:
+            uid = user.id
+            if uid == uauth.uid: continue
+            if pt.has_key(uid):
+                tunnel = pt[uid]
+                tunnel.add(pdata)
+                tunnel.push()
+
+        for user in sec.subscribers:
+            uid = user.id
+            if uid == uauth.uid: continue
+            if pt.has_key(uid):
+                tunnel = pt[uid]
+                tunnel.add(pdata)
+                tunnel.push()
+
         pt = RequestHandler.push_tunnels
         uid = uauth.uid
         pt[uid].close()
@@ -675,9 +704,9 @@ class ChangePasswordHandler(RequestHandler):
         except struct.error:
             raise BadReqError("User logout: Malformed request body")
 
-        logger.info("Trying to change password with "
-                    "(token = {0}, username = {1}, old_pass = {2}, new_pass = {3})"\
-                .format(get_hex(token), username, old_pass, new_pass))
+#        logger.info("Trying to change password with "
+#                    "(token = {0}, username = {1}, old_pass = {2}, new_pass = {3})"\
+#                .format(get_hex(token), username, old_pass, new_pass))
 
         uauth = RequestHandler.get_uauth(token, username, self.session)
         # Authentication failure
-- 
cgit v1.2.3