iframe-proxy | Sunbelt Computer Software

180 lines (160 loc) · 6.41 KB
#!/usr/bin python
import select
import socket
import signal
import cPickle
import struct
import argparse
SERVER_HOST = 'localhost'
CHAT_SERVER_NAME = 'server'
def send(channel, *args):
    buffer = cPickle.dumps(args)
    value = socket.htonl(len(buffer))
    size = struct.pack("L", value)
    channel.send(size)
    channel.send(buffer)
def receive(channel):
    size = struct.calcsize("L")
    size = channel.recv(size)
        size = socket.ntohl(struct.unpack("L", size)[0])
    except struct.error, e:
        return ' '
    buf = ""
    while len(buf) < size:
        buf = channel.recv(size - len(buf))
    return cPickle.loads(buf)[0]
class ChatServer(object):
    """An example chat server using select"""
    def __init__(self, port, backlog=5):
        self.clients = 0
        self.clientmap = {}
        self.outputs = []
        self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
        self.server.bind((SERVER_HOST, port))
        print "Server listening to port: %s ..." %port
        self.server.listen(backlog)
        signal.signal(signal.SIGINT, self.sighandler)
    def sighandler(self, signumber, frame):
        print "Shutting down server..."
        for output in self.outputs:
            output.close()
        self.server.close()
    def get_client_name(self, client):
        info = self.clientmap[client]
        host, name = info[0][0], info[1]
        return '@'.join((name, host))
    def run(self):
        inputs = [self.server, sys.stdin]
        self.outputs = []
        running = True
        while running:
            try:
                readable, writeable, exceptional = select.select(inputs, self.outputs, [])
            except select.error, e:
                break
            for sock in readable:
                if sock == self.server:
                        client, address = self.server.accept()
                    except socket.error, e:
                        print "socket error: %s" %e
                        continue
                    print "Chat server: got connection %d from %s" %(client.fileno(), address)
                    cname = receive(client).split('NAME: ')[1]
#                    print "received from client %s" %str(address)
                    self.clients += 1
                    send(client, 'CLIENT: ' + str(address[0]))
                    inputs.append(client)
                    self.clientmap[client] = (address, cname)
                    msg = "\n(Connected: New client (%d) from %s)" %(self.clients, self.get_client_name(client))
                    for output in self.outputs:
                        send(output, msg)
                    self.outputs.append(client)
                elif sock == sys.stdin:
                    junk = sys.stdin.readline()
                    running = False
                else:
                        data = receive(sock)
                        if data:
                            msg = "\n#[" + self.get_client_name(sock) + "]>>" + data
                            for output in self.outputs:
                                if output != sock:
                                    send(output, msg)
                            print "Chat server: %d hung up" %sock.fileno()
                            self.clients -= 1
                            sock.close()
                            inputs.remove(sock)
                            self.outputs.remove(sock)
                            msg = "\n(Now hung up: Client from %s)" %self.get_client_name(sock)
                            for output in self.outputs:
                                send(output, msg)
                    except socket.error, e:
                        inputs.remove(sock)
                        self.outputs.remove(sock)
                    except KeyboardInterrupt:
                        print "Server interrupted ."
                        self.server.close()
        self.server.close()
class ChatClient(object):
    def __init__(self, name, port, host = SERVER_HOST):
        self.name = name
        self.connected = False
        self.host = host
        self.port = port
        self.prompt = "[" + "@".join((name, socket.gethostname().split(".")[0])) + "]> "
        try:
            self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            self.sock.connect((host, self.port))
            print "Now connected to chat server @ port %d" %self.port
            self.connected = True
            send(self.sock, 'NAME: ' +self.name)
            data=receive(self.sock)
            print "%s" %data
            addr = data.split('CLIENT: ')[1]
            self.prompt = "[" + "@".join((self.name, addr)) + "]> "
        except socket.error, e:
            print "Failed to connect to chat Server @ port %d" %self.port
            sys.exit(1)
    def run(self):
        while self.connected:
            try:
                sys.stdout.write(self.prompt)
                sys.stdout.flush()
                readable, writeable, exceptional = select.select([0, self.sock], [], [])
                for sock in readable:
                    if sock == 0:
                        data = sys.stdin.readline().strip()
                        if data:
                            send(self.sock, data)
                    elif sock == self.sock:
                        data = receive(self.sock)
                        if not data:
                            print "Client shutting down."
                            self.connected = False
                            sys.stdout.write(data + '\n')
                            sys.stdout.flush()
            except KeyboardInterrupt:
                print " Client interrupted. """
                self.sock.close()
                break
if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Socket Server example with select')
    parser.add_argument('--name', action='store', dest='name', required=True)
    parser.add_argument('--port', action='store', dest='port', type=int, required=True)
    given_args = parser.parse_args()
    port = given_args.port
    name = given_args.name
    if name == CHAT_SERVER_NAME:
        server = ChatServer(port)
        server.run()
        Client = ChatClient(name=name, port=port)
        Client.run()
Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

2_3_chat_server_select.py

Latest commit

History

2_3_chat_server_select.py

File metadata and controls
