#!/usr/bin/python
'test TCP server'
from socket import *
from time import ctime
import threading  #多線程模塊
import re   #正則表達式模塊
HOST = ''
PORT = 21567
BUFSIZ = 1024
ADDR = (HOST, PORT)
def Deal(sock, user):
  while True:
    data = sock.recv(BUFSIZ)  #接收用戶的數(shù)據(jù)
    if data == 'quit':  #用戶退出
      del clients[user]
      sock.send(data)
      sock.close()
      print '%s logout' %user
      break
    elif re.match('to:.+', data) is not None:  #選擇通信對象
      data = data[3:]
      if clients.has_key(data):
        chatwith[sock] = clients[data]
        chatwith[clients[data]] = sock
      else:
        sock.send('the user %s is not exist' %data)
    else:
      if chatwith.has_key(sock):  #進行通信
        chatwith[sock].send("[%s] %s: %s" %(ctime(), user, data))
      else:
        sock.send('Please input the user who you want to chat with')
tcpSerSock = socket(AF_INET, SOCK_STREAM)
tcpSerSock.bind(ADDR)
tcpSerSock.listen(5)
clients = {}  #提供 用戶名->socket 映射
chatwith = {}  #提供通信雙方映射
while True:
  print 'waiting for connection...'
  tcpCliSock, addr = tcpSerSock.accept()
  print '...connected from:',addr
  username = tcpCliSock.recv(BUFSIZ)  #接收用戶名
  print 'The username is:',username
  if clients.has_key(username):  #查找用戶名
    tcpCliSock.send("Reuse")  #用戶名已存在
    tcpCliSock.close()
  else:
    tcpCliSock.send("Welcome!")  #登入成功
    clients[username] = tcpCliSock
    chat = threading.Thread(target = Deal, args = (tcpCliSock,username))  #創(chuàng)建新線程進行處理
    chat.start()  #啟動線程
tcpSerSock.close()

#!/usr/bin/python
'test tcp client'
from socket import *
import threading
HOST = 'localhost'
PORT = 21567
BUFSIZ = 1024
ADDR = (HOST, PORT)
threads = []
def Send(sock, test):  #發(fā)送消息
  while True:
    data = raw_input('>')
    tcpCliSock.send(data)
    if data == 'quit':
      break
def Recv(sock, test):   #接收消息
  while True:
    data = tcpCliSock.recv(BUFSIZ)
    if data == 'quit':
      sock.close()   #退出時關(guān)閉socket
      break
    print data
tcpCliSock = socket(AF_INET, SOCK_STREAM)
tcpCliSock.connect(ADDR)
print 'Please input your username:',
username = raw_input()
tcpCliSock.send(username)
data = tcpCliSock.recv(BUFSIZ)
if data == 'Reuse':
  print 'The username has been used!'
else:
  print 'Welcome!'
  chat = threading.Thread(target = Send, args = (tcpCliSock,None))  #創(chuàng)建發(fā)送信息線程
  threads.append(chat)
  chat = threading.Thread(target = Recv, args = (tcpCliSock,None))  #創(chuàng)建接收信息線程
  threads.append(chat)
  for i in range(len(threads)):  #啟動線程
    threads[i].start()
  threads[0].join()  #在我們的設(shè)計中，send線程必然先于recv線程結(jié)束，所以此處只需要調(diào)用send的join，等待recv線程的結(jié)束。