#coding:utf-8'''file:client.py.pydate:2017/9/11 11:01author:lockeyemail:lockey@123.complatform:win7.x86_64 pycharm python3desc:p2p communication clientside'''from socket import *import threading,sys,json,re#引入json模塊主要是為了數(shù)據(jù)的封裝傳輸，re的話是做一些合法性的驗(yàn)證HOST = '192.168.1.7'PORT=8022BUFSIZE = 1024 ##緩沖區(qū)大小 1KADDR = (HOST,PORT)myre = r"^[_a-zA-Z]/w{0,}"tcpCliSock = socket(AF_INET,SOCK_STREAM)#創(chuàng)建一個(gè)socket連接userAccount = None#用戶登錄標(biāo)志，也用來記錄登錄的用戶名稱def register():#用戶注冊(cè)函數(shù) print(""" Glad to have you a member of us! """) accout = input('Please input your account: ') if not re.findall(myre, accout):  print('Account illegal!')  return None password1 = input('Please input your password: ') password2 = input('Please confirm your password: ') if not (password1 and password1 == password2):  print('Password not illegal!')  return None global userAccount userAccount = accout regInfo = [accout,password1,'register'] datastr = json.dumps(regInfo) tcpCliSock.send(datastr.encode('utf-8')) data = tcpCliSock.recv(BUFSIZE) data = data.decode('utf-8') if data == '0':  print('Success to register!')  return True elif data == '1':  print('Failed to register, account existed!')  return False else:  print('Failed for exceptions!')  return Falsedef login():#用戶登錄函數(shù) print(""" Welcome to login in! """) accout = input('Account: ') if not re.findall(myre, accout):  print('Account illegal!')  return None password = input('Password: ') if not password:  print('Password illegal!')  return None global userAccount userAccount = accout loginInfo = [accout, password,'login'] datastr = json.dumps(loginInfo) tcpCliSock.send(datastr.encode('utf-8')) data = tcpCliSock.recv(BUFSIZE) if data == '0':  print('Success to login!')  return True else:  print('Failed to login in(user not exist or username not match the password)!')  return Falsedef addGroup():#群組添加 groupname = input('Please input group name: ') if not re.findall(myre, groupname):  print('group name illegal!')  return None return groupnamedef chat(target):#進(jìn)入聊天（群聊和點(diǎn)對(duì)點(diǎn)聊天可以選擇） while True:  print('{} -> {}: '.format(userAccount,target))  msg = input()  if len(msg) > 0 and not msg in 'qQ':   if 'group' in target:    optype = 'cg'   else:    optype = 'cp'   dataObj = {'type': optype, 'to': target, 'msg': msg, 'froms': userAccount}   datastr = json.dumps(dataObj)   tcpCliSock.send(datastr.encode('utf-8'))   continue  elif msg in 'qQ':   break  else:   print('Send data illegal!')class inputdata(threading.Thread):#用戶輸入選擇然后執(zhí)行不同的功能程序 def run(self):  menu = """      (CP): Chat with individual      (CG): Chat with group member      (AG): Add a group      (EG): Enter a group      (H): For help menu      (Q): Quit the system      """  print(menu)  while True:   operation = input('Please input your operation("h" for help): ')   if operation in 'cPCPCpcp':   #進(jìn)入個(gè)人聊天    target = input('Who would you like to chat with: ')    chat(target)    continue   if operation in 'cgCGCgcG':   #進(jìn)入群聊    target = input('Which group would you like to chat with: ')    chat('group'+target)    continue   if operation in 'agAGAgaG':   #添加群組    groupName = addGroup()    if groupName:     dataObj = {'type': 'ag', 'groupName': groupName}     dataObj = json.dumps(dataObj)     tcpCliSock.send(dataObj.encode('utf-8'))    continue   if operation in 'egEGEgeG':   #入群    groupname = input('Please input group name fro entering: ')    if not re.findall(myre, groupname):     print('group name illegal!')     return None    dataObj = {'type': 'eg', 'groupName': 'group'+groupname}    dataObj = json.dumps(dataObj)    tcpCliSock.send(dataObj.encode('utf-8'))    continue   if operation in 'hH':    print(menu)    continue   if operation in 'qQ':    sys.exit(1)   else:    print('No such operation!')class getdata(threading.Thread):#接收數(shù)據(jù)線程 def run(self):  while True:   data = tcpCliSock.recv(BUFSIZE).decode('utf-8')   if data == '-1':    print('can not connect to target!')    continue   if data == 'ag0':    print('Group added!')    continue   if data == 'eg0':    print('Entered group!')    continue   if data == 'eg1':    print('Failed to enter group!')    continue   dataObj = json.loads(data)   if dataObj['type'] == 'cg':   #群組消息的格式定義    print('{}(from {})-> : {}'.format(dataObj['froms'], dataObj['to'], dataObj['msg']))   else:   #個(gè)人消息的格式定義    print('{} ->{} : {}'.format(dataObj['froms'], userAccount, dataObj['msg']))def main():  try:   tcpCliSock.connect(ADDR)   print('Connected with server')   while True:    loginorReg = input('(l)ogin or (r)egister a new account: ')    if loginorReg in 'lL':     log = login()     if log:      break    if loginorReg in 'rR':     reg = register()     if reg:      break   myinputd = inputdata()   mygetdata = getdata()   myinputd.start()   mygetdata.start()   myinputd.join()   mygetdata.join()  except Exception:   print('error')   tcpCliSock.close()   sys.exit()if __name__ == '__main__': main()

#coding:utf-8'''file:server.pydate:2017/9/11 14:43author:lockeyemail:lockey@123.complatform:win7.x86_64 pycharm python3desc:p2p communication serverside'''import socketserver,json,timeimport subprocessconnLst = []groupLst = []## 代號(hào) 地址和端口 連接對(duì)象#optype = {'ag':'group adding','cp':'chat with individual','cg':'chat with group'}class Connector(object): ##連接對(duì)象類 def __init__(self,account,password,addrPort,conObj):  self.account = account  self.password = password  self.addrPort = addrPort  self.conObj = conObjclass Group(object):#群組類 def __init__(self,groupname,groupOwner):  self.groupId = 'group'+str(len(groupLst)+1)  self.groupName = 'group'+groupname  self.groupOwner = groupOwner  self.createTime = time.time()  self.members=[groupOwner]class MyServer(socketserver.BaseRequestHandler): def handle(self):  print("got connection from",self.client_address)  userIn = False  global connLst  global groupLst  while not userIn:   conn = self.request   data = conn.recv(1024)   if not data:    continue   dataobj = json.loads(data.decode('utf-8'))   #如果連接客戶端發(fā)送過來的信息格式是一個(gè)列表且注冊(cè)標(biāo)識(shí)為False時(shí)進(jìn)行用戶注冊(cè)或者登陸   ret = '0'   if type(dataobj) == list and not userIn:    account = dataobj[0]    password = dataobj[1]    optype = dataobj[2]    existuser = False    if len(connLst) > 0:     for obj in connLst:      if obj.account == account:       existuser = True       if obj.password == password:        userIn = True        print('{} has logged in system({})'.format(account,self.client_address))        break    if optype == 'login' and (not userIn or not existuser):     ret = '1'     print('{} failed to logged in system({})'.format(account, self.client_address))    else:     if existuser:      ret = '1'      print('{} failed to register({}),account existed!'.format(account, self.client_address))     else:      try:       conObj = Connector(account,password,self.client_address,self.request)       connLst.append(conObj)       print('{} has registered to system({})'.format(account,self.client_address))       userIn = True      except:       print('%s failed to register for exception!'%account)       ret = '99'   conn.sendall(ret.encode('utf-8'))   if ret == '0':    break  while True:  #除登陸注冊(cè)之外的請(qǐng)求的監(jiān)聽   conn = self.request   data = conn.recv(1024)   if not data:    continue   print(data)   dataobj = data.decode('utf-8')   dataobj = json.loads(dataobj)   if dataobj['type'] == 'ag' and userIn:   #如果判斷用戶操作請(qǐng)求類型為添加群組則進(jìn)行以下操作    groupName = dataobj['groupName']    groupObj = Group(groupName,self.request)    groupLst.append(groupObj)    conn.sendall('ag0'.encode('utf-8'))    print('%s added'%groupName)    continue   if dataobj['type'] == 'eg' and userIn:   #入群操作    groupName = dataobj['groupName']    ret = 'eg1'    for group in groupLst:     if groupName == group.groupName:      group.members.append(self.request)      print('{} added into {}'.format(self.client_address,groupName))      ret = 'eg0'      break    conn.sendall(ret.encode('utf-8'))    continue   #客戶端將數(shù)據(jù)發(fā)給服務(wù)器端然后由服務(wù)器轉(zhuǎn)發(fā)給目標(biāo)客戶端   print('connLst',connLst)   print('grouplst',groupLst)   if len(connLst) > 1:    sendok = False    if dataobj['type'] == 'cg':    #群內(nèi)廣播（除發(fā)消息的人）     print('group',data)     for obj in groupLst:      if obj.groupName == dataobj['to']:       for user in obj.members:        if user != self.request:         user.sendall(data)    else:    #個(gè)人信息發(fā)送     for obj in connLst:      if dataobj['to'] == obj.account:       obj.conObj.sendall(data)       sendok = True     if sendok == False:      print('no target valid!')   else:    conn.sendall('-1'.encode('utf-8'))    continueif __name__ == '__main__': server = socketserver.ThreadingTCPServer(('192.168.1.7',8022),MyServer) print('waiting for connection...') server.serve_forever()