Example #1
0
class Client:
    def __init__(self, sock):
        self._sock = sock
        self._in_buffer = ByteBuffer()
        self._out_buffer = ByteBuffer()
        self._read_thread = Thread(target=self._recv)
        self._write_thread = Thread(target=self._send)
        self._open = True
        self._read_thread.start()
        self._write_thread.start()
    def _recv(self):
        '''
        reads constantly into the buffer
        '''
        while self._open:
            self._in_buffer.write_bytes(self._sock.recv(4096))
    def _send(self):
        '''
        writes the out buffer to the socket
        '''
        while self._open:
            self._sock.send(self._out_buffer.flush())
    def get_buffers(self):
        return self._in_buffer, self._out_buffer
    def close(self):
        self._open = False
        self._sock.close()
        self._in_buffer.close()
        self._out_buffer.close()
Example #2
0
class Client:
    def __init__(self, sock, main_interrupt=False):
        self._inter_main = main_interrupt
        self._sock = sock
        self._sock.settimeout(1)
        self._in_buffer = ByteBuffer()
        self._out_buffer = ByteBuffer()
        self._read_thread = Thread(target=self._recv)
        self._write_thread = Thread(target=self._send)
        self._open = True
        self._read_thread.start()
        self._write_thread.start()
    def _recv(self):
        '''
        reads constantly into the buffer
        '''
        try:
            while self._open:
                try:
                    bts = self._sock.recv(4096)
                    if not len(bts):
                        raise OSError()
                    self._in_buffer.write_bytes(bts)
                except timeout:
                    pass
        except OSError:
            print("Socket closed")
            self.close()
    def _send(self):
        '''
        writes the out buffer to the socket
        '''
        try:
            while self._open:
                bts = self._out_buffer.flush()
                while len(bts):
                    sent = self._sock.send(bts)
                    bts = bts[sent:]
        except OSError:
            self._close()
    def get_buffers(self):
        return self._in_buffer, self._out_buffer
    def close(self):
        if self._open:
            self._open = False
            self._out_buffer.close()
            try:
                self._sock.shutdown(SHUT_WR)
                self._sock.close()
            except OSError:
                pass
            print("Closing input buffer")
            self._in_buffer.close()
            print("Closed input buffer")
            try:
                self._read_thread.join()
                self._write_thread.join()
            except RuntimeError:
                if self._inter_main:
                    print("Interrupting main thread")
                    kill(getpid(), SIGINT)