import zmq
from . import common
-from .gps303proto import HIBERNATION, LOGIN, parse_message, proto_of_message
+from .gps303proto import (
+ HIBERNATION,
+ LOGIN,
+ inline_response,
+ parse_message,
+ proto_of_message,
+)
log = getLogger("gps303/collector")
clntsock, clntaddr = tcpl.accept()
topoll.append((clntsock, clntaddr))
else:
- for imei, msg in clients.recv(sk):
- zpub.send(Bcast(imei, msg).as_bytes)
- if (
- msg is None
- or proto_of_message(msg) == HIBERNATION.PROTO
- ):
- log.debug(
- "HIBERNATION from fd %d (IMEI %s)", sk, imei
- )
- tostop.append(sk)
- elif proto_of_message(msg) == LOGIN.PROTO:
- clients.response(Resp(imei=imei, payload=LOGIN.response()))
+ received = clients.recv(sk)
+ if received is None:
+ log.debug(
+ "Terminal gone from fd %d (IMEI %s)", sk, imei
+ )
+ tostop.append(sk)
+ else:
+ for imei, msg in received:
+ zpub.send(Bcast(imei, msg).as_bytes)
+ if proto_of_message(msg) == HIBERNATION.PROTO:
+ log.debug(
+ "HIBERNATION from fd %d (IMEI %s)",
+ sk,
+ imei,
+ )
+ tostop.append(sk)
+ respmsg = inline_response(msg)
+ if respmsg is not None:
+ clients.response(
+ Resp(imei=imei, payload=respmsg)
+ )
# poll queue consumed, make changes now
for fd in tostop:
poller.unregister(fd)
__all__ = (
"handle_packet",
+ "inline_response",
"make_object",
"make_response",
"parse_message",
return pack("BB", self.length, self.PROTO) + self.payload
@classmethod
- def response(cls, *args):
- if len(args) == 0:
- return None
- assert len(args) == 1 and isinstance(args[0], bytes)
- payload = args[0]
+ def make_packet(cls, payload):
+ assert isinstance(payload, bytes)
length = len(payload) + 1
if length > 6:
length -= 6
return pack("BB", length, cls.PROTO) + payload
+ @classmethod
+ def inline_response(cls, packet):
+ return cls.make_packet(b"")
+
class UNKNOWN(GPS303Pkt):
PROTO = 256 # > 255 is impossible in real packets
+ @classmethod
+ def inline_response(cls, packet):
+ return None
+
class LOGIN(GPS303Pkt):
PROTO = 0x01
self.ver = unpack("B", payload[-1:])[0]
return self
- @classmethod
- def response(cls):
- return super().response(b"")
+ def response(self):
+ return self.make_packet(b"")
class SUPERVISION(GPS303Pkt): # Server sends supervision number status
PROTO = 0x05
- @classmethod
- def response(cls, supnum=0):
+ def response(self, supnum=0):
# 1: The device automatically answers Pickup effect
# 2: Automatically Answering Two-way Calls
# 3: Ring manually answer the two-way call
self.gps_nb_sat = payload[6] & 0x0F
lat, lon, speed, flags = unpack("!IIBH", payload[7:18])
self.gps_is_valid = bool(flags & 0b0001000000000000) # bit 3
- flip_lon = bool(flags & 0b0000100000000000) # bit 4
- flip_lat = not bool(flags & 0b0000010000000000) # bit 5
- self.heading = flags & 0b0000001111111111 # bits 6 - last
+ flip_lon = bool(flags & 0b0000100000000000) # bit 4
+ flip_lat = not bool(flags & 0b0000010000000000) # bit 5
+ self.heading = flags & 0b0000001111111111 # bits 6 - last
self.latitude = lat / (30000 * 60) * (-1 if flip_lat else 1)
self.longitude = lon / (30000 * 60) * (-2 if flip_lon else 1)
self.speed = speed
self.flags = flags
return self
+ @classmethod
+ def inline_response(cls, packet):
+ return cls.make_packet(packet[2:8])
+
def response(self):
- return super().response(self.dtime)
+ return self.make_packet(self.dtime)
class GPS_POSITIONING(_GPS_POSITIONING):
self.signal = None
return self
+ @classmethod
+ def inline_response(cls, packet):
+ return None
+
def response(self, upload_interval=25): # Set interval in minutes
return super().response(pack("B", upload_interval))
PROTO = 0x15
def response(self): # Server can send to initiate factory reset
- return super().response(b"")
+ return self.make_packet(b"")
class WHITELIST_TOTAL(GPS303Pkt): # Server sends to initiage sync (0x58)
class WIFI_OFFLINE_POSITIONING(_WIFI_POSITIONING):
PROTO = 0x17
+ @classmethod
+ def inline_response(cls, packet):
+ return cls.make_packet(packet[2:8])
+
def response(self):
return super().response(self.dtime)
class TIME(GPS303Pkt):
PROTO = 0x30
+ @classmethod
+ def inline_response(cls, packet):
+ return pack(
+ "!BBHBBBBB", 7, cls.PROTO, *datetime.utcnow().timetuple()[:6]
+ )
+
def response(self):
payload = pack("!HBBBBB", *datetime.utcnow().timetuple()[:6])
return super().response(payload)
class WIFI_POSITIONING(_WIFI_POSITIONING):
PROTO = 0x69
+ @classmethod
+ def inline_response(cls, packet):
+ return None
+
def response(self, lat=None, lon=None):
if lat is None or lon is None:
payload = b""
payload = "{:+#010.8g},{:+#010.8g}".format(lat, lon).encode(
"ascii"
)
- return super().response(payload)
+ return self.make_packet(payload)
class MANUAL_POSITIONING(GPS303Pkt):
self.interval = unpack("!H", payload[:2])
return self
+ @classmethod
+ def inline_response(cls, packet):
+ return cls.make_packet(packet[2:4])
+
def response(self):
- return super().response(pack("!H", self.interval))
+ return self.make_packet(pack("!H", self.interval))
class SOS_ALARM(GPS303Pkt):
return unpack("B", packet[1:2])[0]
+def inline_response(packet):
+ proto = proto_of_message(packet)
+ if proto in CLASSES:
+ return CLASSES[proto].inline_response(packet)
+ else:
+ return None
+
+
def make_object(length, proto, payload):
if proto in CLASSES:
return CLASSES[proto].from_packet(length, payload)
payload = packet[2:]
adjust = 2 if proto == STATUS.PROTO else 4 # Weird special case
if (
- proto
- not in (WIFI_POSITIONING.PROTO, WIFI_OFFLINE_POSITIONING.PROTO)
+ proto not in (WIFI_POSITIONING.PROTO, WIFI_OFFLINE_POSITIONING.PROTO)
and length > 1
and len(payload) + adjust != length
):
def make_response(msg, **kwargs): # DEPRECATED
inframe = msg.response(**kwargs)
return None if inframe is None else b"xx" + inframe + b"\r\n"
-
projects / loctrkd.git / commitdiff