Exemplo n.º 1
0
Arquivo: wxbc.py Projeto: lovee/Circus
 def __init__(self, name, ip='127.0.0.1', mac='0.0.0.0', bshost=("127.0.0.1", 11011)):
     self.socket = BCSocket(name)
     self.que_ack_pkt = Queue.Queue()
     self.name = name
     self.ip = ip
     self.mac = mac
     self.card = ""
     #BC模拟器状态0:保留, 1:保留, 2:锁屏, 3:解锁, 4:上机
     self.status = 2
     self.bshost = bshost
     self.bs_ip, self.bs_port = bshost
     self.conn_flag = threading.Event()
     self.conn_flag.set()
     self.thd_comm = None
     self.thd_listener = None
Exemplo n.º 2
0
Arquivo: wxbc.py Projeto: lovee/Circus
class WXBC():
    """
    # WXBC模拟器;
    # 1、实现启动、关闭、信令收发和状态监听等功能;
    """
    def __init__(self, name, ip='127.0.0.1', mac='0.0.0.0', bshost=("127.0.0.1", 11011)):
        self.socket = BCSocket(name)
        self.que_ack_pkt = Queue.Queue()
        self.name = name
        self.ip = ip
        self.mac = mac
        self.card = ""
        #BC模拟器状态0:保留, 1:保留, 2:锁屏, 3:解锁, 4:上机
        self.status = 2
        self.bshost = bshost
        self.bs_ip, self.bs_port = bshost
        self.conn_flag = threading.Event()
        self.conn_flag.set()
        self.thd_comm = None
        self.thd_listener = None

    def _connect(self):
        """
        登录BS,仅用于通信线程调用
        """
        if not self.socket.is_connected():
            try:
                self.socket.connect(self.bshost)
                self.conn_flag.set()
            except Exception as e:
                self.conn_flag.set()
                # self.status = 0
                return 1
        else:
            data = pkt_connect(self.name, self.bs_ip)
            login_pkt = get_pkt_by_data(2, data)
            self.socket.send_pkt(login_pkt)
            self.conn_flag.clear()
            # if self.card != "":
            #     self.status = 4
            # else:
            #     self.status = 2

    def _communicate(self):
        """
        维护BS连接,仅用于通信线程调用
        """
        data = pkt_heart(self.name, self.bs_ip)
        hrt_pkt = get_pkt_by_data(3, data)
        while True:
            if (self.socket.is_connected() and self.is_connected()):
                time.sleep(1)
            else:
                self._connect()

    def _heart_beat(self):
        """
        心跳,仅用于心跳线程调用;
        """
        data = pkt_heart(self.name, self.bs_ip)
        hrt_pkt = get_pkt_by_data(3, data)
        while True:
            if (self.socket.is_connected() and self.is_connected()):
                self.socket.send_pkt(hrt_pkt)
                time.sleep(10)
            else:
                time.sleep(1)

    def _listener(self):
        """
        监听并处理BS通知信令,仅用于监听线程调用;
        待完善!
        """
        while True:
            pkt = self.socket.recv_pkt(1)
            if pkt is None:
                continue
            opts = pkt.data_options()
            #上机通知信令
            if pkt.pkt_type == 0x4:
                if opts[4]=='':
                    self.status = 3
                else:
                    self.status = 4
                self.card = opts[4]
            #锁屏
            elif pkt.pkt_type == 0x5:
                self.status = 2
                self.card = ""
            #保存ACK信令
            # if (pkt.pkt_type & 0x80000000) > 0:
            #     self.que_ack_pkt.put(pkt)
            self.que_ack_pkt.put(pkt)

    def is_connected(self):
        """
        BC是否已经连接到BS
        """
        return not self.conn_flag.is_set()

    def startup(self):
        """
        启动BC
        """
        #通信维护线程
        self.thd_comm = threading.Thread(target=self._communicate, name='bc_communicate_thd')
        self.thd_comm.start()
        #信令监听线程
        self.thd_listener = threading.Thread(target=self._listener, name='bc_listener_thd')
        self.thd_listener.start()
        #心跳线程
        self.thd_heart_beat = threading.Thread(target=self._heart_beat, name='bc_heart_beat_thd')
        self.thd_heart_beat.start()

    def shutdown(self):
        """
        关闭BC
        待完善!
        """
        pass

    def get_status(self):
        """
        获取BC状态信息
        """
        s = (self.name, self.ip, self.mac, self.is_connected(), self.status, self.card, self.bshost)
        return s

    def chk_status_is(self, stat, timeout=5):
        """
        检查BC状态是
        0:保留, 1:保留, 2:锁屏, 3:解锁, 4:上机
        """
        res = False
        while timeout > 0 :
            if self.status == stat:
                res = True
                break
            else:
                time.sleep(1)
                timeout -= 1
        return res

    def chk_status_is_not(self, stat, timeout=2):
        """
        检查BC状态不是
        0:保留, 1:保留, 2:锁屏, 3:解锁, 4:上机
        """
        res = False
        while timeout > 0 :
            if self.status != stat:
                res = True
                break
            else:
                time.sleep(1)
                timeout -= 1
        return res

    def reset_status(self):
        """
        重置BC状态为锁屏状态
        """
        self.status = 2
        self.card = ""
        self.que_ack_pkt = Queue.Queue()

    def get_ack_pkt(self, ack_pkt_type, timeout=10):
        """
        获取请求信令的响应包
        """
        while timeout > 0:
            times = self.que_ack_pkt.qsize()
            for i in range(times):
                ack_pkt = self.que_ack_pkt.get_nowait()
                if ack_pkt.pkt_type == ack_pkt_type:
                    return ack_pkt
                else:
                    self.que_ack_pkt.put(ack_pkt)
            time.sleep(1)
            timeout -= 1
        return None

    def get_up(self, card, pwd='', veri_code=''):
        """
        会员上机
        """
        logging.info("[%s][%s][%d] Call method : get_up(card=%s, pwd=%s, veri_code=%s)" % (self.name, self.card, self.status, card, pwd, veri_code))
        res = False
        pkt_type = 7
        ack_pkt = None
        data = pkt_up(card, pwd, veri_code)
        #带验证码的上机信令
        if len(veri_code) != 0:
            pkt_type = 65
        pkt = get_pkt_by_data(pkt_type, data)
        self.socket.send_pkt(pkt)
        if self.chk_status_is(4) is True:
            res = True
        else:
            if pkt_type == 65:
                ack_pkt = self.get_ack_pkt(65)
            else:
                ack_pkt = self.get_ack_pkt(0x80000000 + pkt_type)
            if ack_pkt is not None:
                ack_pkt_opts = ack_pkt.data_options()
                print(ack_pkt_opts[3], ack_pkt_opts[4])
                res = False
            else:
                print(99, 'Failure up the machine, BS no response.')
                res = False
        return res

    def get_down(self):
        """
        会员下机
        """
        logging.info("[%s][%s][%d] Call method : get_down()" % (self.name, self.card, self.status))
        data = pkt_down()
        pkt = get_pkt_by_data(6, data)
        self.socket.send_pkt(pkt)
        return True