基于数据包(UDP)的简易网络聊天室

话不多说，直接上代码.

服务器端：

server.py

from socket import *
import os,sys
\’\’\’
聊天室文件
服务端
by:dyh
\’\’\’
###用户退出并且告知所有的用户###
def do_logout(s,user,name):
   #删除列表中的用户
   del user[name]
   msg=\’\\n\’+name+\’离开了聊天室\’
   for i in user:
       s.sendto(msg.encode(),addr)
   return
###用户登陆代码###
def do_login(s,user,name,addr):
   #用户注册时对用户name进行检查
   if (name in user) or name==\’管理员\’:
       s.sendto(\’该用户已存在\’.encode(),addr)
       return
   #用户name符合要求时发送OK给客户端提示其可以正常登陆
   #用户接收到信息以后对信息判断进行登陆操作
   s.sendto(b\’OK\’,addr)
   msg=\’\\n欢迎%s进入聊天室\’%name
   #通知所有人
   for i in user:
       s.sendto(msg.encode(),user[i])
   #将用户插进字典
   user[name]=addr
   return

def do_chat(s,user,cmd):
   msg=\’\\n%-4s:%s\’%(cmd[1],\’ \’.join(cmd[2:]))
   #将用户name发送给除了自己之外的所有人
   for i in user:
       if i!=cmd[1]:
           s.sendto(msg.encode(),user[i])
   return

#处理客户端请求
def do_child(s):
   #字典存储用户信息
   user={}
   #循环接收请求
   while True:
       msg,addr=s.recvfrom(1024)
       msg=msg.decode()
       cmd=msg.split(\’.\’)
       #根据不同请求做不同的事情
       if cmd[0]==\’l\’:
           do_login(s,user,cmd[1],addr)
           s.sendto(b\’OK\’,addr)
       elif cmd[0]==\’m\’:
           do_chat(s,user,cmd)
       elif cmd[0]==\’q\’:
           do_logout()
       else:
           s.sendto(\’请求错误\’.encode(),addr)

#发送管理员消息
def do_parent(s,ADDR):
   while True:
       msg=input(\’管理员发消息：\’)
       msg=\’m.管理员\’+msg
       s.sendto(msg.encode(),ADDR)
   s.close()
   sys.exit(0)

def main():
   if len(sys.argv)<3:
       print(\’argv error,\’)
       return
   HOST=sys.argv[1]
   PORT=int(sys.argv[2])
   ADDR=(HOST,PORT)

   #使用数据b报套接字
   s=socket(AF_INET,SOCK_DGRAM)
   #设置端口重用代码
   s.setsockopt(SOL_SOCKET,SO_REUSEADDR,1)
   s.bind(ADDR)
   ###设置二级进程是为了将收消息和发消息分离###
   #创建一级子进程
   pid1=os.fork()

   if pid1<0:
       print()
   elif pid1==0:
       print(\’child process\’)
       pid2=os.fork()
       if pid2<0:
           print(\”\”)
       elif pid2==0:
           do_child(s)
       else:
           #一级子进程退出，使二级子进程成为孤儿
           os._exit(0)
   else:
       #等待一级子进程的退出
       os.wait()
       do_parent(s,ADDR)

if __name__==\’__main__\’:
main()

客户端:

client.py

from socket import *
import os,sys,signal
\’\’\’
聊天室文件
客户端
by:dyh
\’\’\’
#发送消息
def do_child(s,name,addr):
   while True:
       text=input(\”say:\”)
       if text==\’quit\’:
           msg=\”q.\”+name
           s.sendto(msg.encode(),addr)
           #从子进程种杀掉父进程
           os.kill(os.getppid(),signal.SIGKILL)
           sys.exit(\’已退出！\’)
       else:
           msg=\’m.\’+\’%s.%s\’%(name,text)
           s.sendto(msg.encode(),addr)
#收消息
def do_parent(s):
   while True:
       msg,addr=s.recvfrom(1024)
       msg=msg.decode()
       msg=msg[:-1]
       print(msg+\”\\nsay:\”,end=\’\’)

def main():
   if len(sys.argv)<3:
       print(\’argv error,\’)
       return
   HOST=sys.argv[1]
   PORT=int(sys.argv[2])
   ADDR=(HOST,PORT)

   #使用数据报套接字
   s=socket(AF_INET,SOCK_DGRAM)
   ####登陆###
   while True:
       name=input(\”请输入姓名:\”)
       msg=\’l.\’+name
       s.sendto(msg.encode(),ADDR)
       data,addr=s.recvfrom(1024)
       if data.decode()==\’OK\’:
           print(\’###进入聊天室###\’)
           break
       else:
           print(data.decode())
   ###注意这里不要设置二级进程###
   #如果设置可能会在关闭客户端的时候杀死系统进程导致关机
   pid=os.fork()

   if pid<0:
       print(\’ \’)
   elif pid==0:
       do_child(s,name,ADDR)
   else:
       #等待一级子进程的退出
       do_parent(s)

if __name__==\’__main__\’:
main()