Python Socket 编程

client:

 

import socket, sys
if __name__ == '__main__':
    #处理参数
    argv = sys.argv
    if (len(argv)!=3) or (len(argv)==2 and argv[1]=='/?'):
        print '>>>Useage:', argv[0], '<address> <port>'
        sys.exit(0)

    HOST = argv[1]          #Server IP地址
    PORT = int(argv[2])     #Server 端口
    print '>>>HOST:', HOST
    print '>>>PORT:', PORT

    #打开socket
    try:
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    except socket.error:
        print '>>>Fail to open socket!'
        sys.exit(1)

    print '>>>Socket opened...'

    #连接
    try:
        sock.connect((HOST, PORT))
    except socket.error:
        print '>>>Fail to connect!'
        sock.close()
        print '>>>Socket closed...'
        sys.exit(1)

    print '>>>Socket connected...'

    flag = 1            #循环标志
    iReturnValue = 0    #系统返回值

    #向Server发送信息并回显
    while(flag):
        msg = raw_input('>>>Input message: ')
        if msg == '':
            continue
        if msg == 'exit':
            flag = 0

        #发送信息
        try:
            sock.send(msg)
        except socket.error:
            print '>>>Fail to send message!'
            iReturnValue = 1
            break
        print '>>>Message sent    :', msg

        #接收信息
        try:    
            msg = sock.recv(1024)
        except socket.error:
            print '>>>Fail to receive message!'
            iReturnValue = 1
            break

        print '>>>Message received:', msg

    #关闭socket，退出程序
    sock.close()
    print '>>>Socket closed...'
    print 'Exit client program.'
    sys.exit(iReturnValue)

server:

 

import socket, threading
import sys, copy, time
iClientCount = 0 #客户端连接数

def ConnectThread(conn, addr):

    """

    Function called by thread to deal with client connection

    """

    global iClientCount         #引用全局变量
    sock = copy.copy(conn)      
    address = copy.copy(addr)
    flag = 1
    while(flag):
        try:
            msg = sock.recv(1024)
        except socket.error:
            print '>>>A 'receive' error happened! IP:', address[0],
'PORT:', address[1]
            break

        if str(msg) == 'exit': 
            msg = 'byebye'
            flag = 0
        try:
            sock.send(msg)
        except socket.error:
            print '>>>A 'send' error happened! IP:', address[0], 'PORT:',
address[1]
            break
    iClientCount -= 1
    print '>>>Disconnected with', address[0], 'at port', address[1] 
    print '>>>Total number of connections is', iClientCount
    sock.close()

def ListenThread(sock):

    """

    Function called by thread to listen a connection

    """

    global iClientCount
    while 1:
        if iClientCount >= MAX_CLIENTS:
            time.sleep(5)
            continue

        try:
            conn, addr = sock.accept()
        except socket.error:
            print '>>>An 'accept' error happened!'
            continue

        iClientCount += 1
        print '>>>Connecting from', addr[0], 'at port', addr[1]
        print '>>>Total number of connections is', iClientCount
        threading.Thread(target=ConnectThread, args=(conn, addr)).start()

if __name__ == '__main__':
    argv = sys.argv
    if (len(argv)>3) or (len(argv)==2 and argv[1]=='/?'):
        print '>>>Useage:', argv[0], '[port=<port>]
[max=<max_clients>]'
        sys.exit(0)

    HOST = ''
    PORT = 8006
    MAX_CLIENTS = 5

    for arg in argv[1:]:
        if arg.startwith('port='):
            PORT = int(arg[len('port='):])
        elif arg.startwith('max='):
            MAX_CLIENTS = int(arg[len('max='):])

    print '>>>PORT:', PORT
    print '>>>MAX_CLIENTS:', MAX_CLIENTS

    try:
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    except socket.error:
        print 'Fail to open socket!'
        sys.exit(1)

    print '>>>Socket opened...'

    try:
        sock.bind((HOST, PORT))
    except socket.error:
        print 'Fail to bind port!'
        sock.close()
        sys.exit(1)

    print '>>>Socket binding...'

    try:
        sock.listen(MAX_CLIENTS)
    except socket.error:
        print 'Fail to listen!'
        sock.close()
        sys.exit(1)

    print '>>>Socket listening...'
    threading.Thread(target=ListenThread, args=(sock,)).start()