From 4be449207cd12831a1858860eba1c8bb0618aa80 Mon Sep 17 00:00:00 2001 From: Teddy 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') 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-70-g09d2