web代码

代码

day 27 socket

代码

1 socket

import socket

sk = socket.socket()
sk.bind(('127.0.0.1',9002))
sk.listen()

conn,addr = sk.accept()  # 阻塞
conn.send(b'hello')
msg = conn.recv(1024)
print(msg.decode('utf-8'),addr)

conn.close()
sk.close()

client

import socket
sk = socket.socket()
sk.connect(('127.0.0.1',9002))

msg = sk.recv(1024)
print(msg)
sk.send(b'wahaha')

sk.close()

day28 udp

tcp(两个client)

import socket

sk = socket.socket()
sk.bind(('127.0.0.1',9001))
sk.listen()
while True:
    conn,addr = sk.accept()   # 等待用户来连接我
    while  True:
        msg = input('>>>')
        conn.send(msg.encode('utf-8'))
        if msg.upper() == 'Q':
            break
        content = conn.recv(1024).decode('utf-8')  # 等待 客户端给我φ消息
        if content.upper() == 'Q': break
        print(content)
    conn.close()
sk.close()

import socket

sk = socket.socket()



sk.connect(('127.0.0.1',9001))
while True:
    ret = sk.recv(1024).decode('utf-8')
    if ret.upper() == 'Q':break
    print(ret)
    msg = input('>>>')
    sk.send(msg.encode('utf-8'))
    if msg.upper() == 'Q':
        break

sk.close()

udp(两个client)

import socket
sk = socket.socket(type=socket.SOCK_DGRAM)
sk.bind(('127.0.0.1',9001))


while True:
    msg,client_addr = sk.recvfrom(1024)
    print(msg.decode('utf-8'))
    content = input('>>>')
    sk.sendto(content.encode('utf-8'),client_addr)
sk.close()

import socket
sk = socket.socket(type=socket.SOCK_DGRAM)

server_addr = ('127.0.0.1',9001)
while True:
    content = input('>>>')
    if content.upper() == 'Q':break
    sk.sendto(content.encode('utf-8'),server_addr)
    msg = sk.recv(1024).decode('utf-8')
    if msg.upper() == 'Q':break
    print(msg)
sk.close()

day29

一对多聊天（两个client）

import json
import socket

sk = socket.socket(type=socket.SOCK_DGRAM)
sk.bind(('127.0.0.1',9001))
user_info = {
    1234:('alex','33[1;32m'),
    5678:('宝元','33[1;31m'),
}
while True:
    msg,addr = sk.recvfrom(1024)
    str_msg = msg.decode('utf-8')
    dic_msg = json.loads(str_msg)
    code = dic_msg['code']
    content = dic_msg['msg']
    print('%s%s : %s33[0m'%(user_info[code][1],user_info[code][0],content))
    ret = input('>>>').encode('utf-8')
    sk.sendto(ret,addr)
sk.close()

import json
import socket

sk = socket.socket(type=socket.SOCK_DGRAM)
addr = ('127.0.0.1',9001)
code = 5678
while True:
    msg = input('>>>')
    if msg.upper() == 'Q': break
    send_msg = {'code':code,'msg':msg}
    str_msg = json.dumps(send_msg)
    sk.sendto(str_msg.encode('utf-8'),addr)
    ret = sk.recv(1024).decode('utf-8')
    if ret.upper() == 'Q': break
    print(ret)
sk.close()

登录

import json
import socket

sk = socket.socket(type=socket.SOCK_DGRAM)
addr = ('127.0.0.1',9001)
code = 5678
while True:
    msg = input('>>>')
    if msg.upper() == 'Q': break
    send_msg = {'code':code,'msg':msg}
    str_msg = json.dumps(send_msg)
    sk.sendto(str_msg.encode('utf-8'),addr)
    ret = sk.recv(1024).decode('utf-8')
    if ret.upper() == 'Q': break
    print(ret)
sk.close()

import json
import socket

username = input('用户名 :')
password = input('密  码 :')
sk = socket.socket()
sk.connect(('127.0.0.1',9001))
msg = {'operate':'login',
       'user':username,
       'pwd':password}
str_msg = json.dumps(msg)
sk.send(str_msg.encode('utf-8'))
content = sk.recv(1024)
str_content = content.decode('utf-8')
dic_content = json.loads(str_content)
if dic_content['result']:
    print('登录成功')
else:
    print('登录失败')
sk.close()

alex|ee838c58e5bb3c9e687065edd0ec454f

登录2

import sys
import json
import socket
import hashlib

def get_md5(username,password):
    md5 = hashlib.md5(username.encode('utf-8'))
    md5.update(password.encode('utf-8'))
    return md5.hexdigest()

def login(dic_msg):
    print(dic_msg['user'], dic_msg['pwd'])
    with open('userinfo', encoding='utf-8') as f:
        for line in f:
            user, pwd = line.strip().split('|')
            print(pwd,get_md5(dic_msg['user'], dic_msg['pwd']))
            if user == dic_msg['user'] and pwd == get_md5(dic_msg['user'], dic_msg['pwd']):
                return {'opt':'login', 'result':True}
        else:return {'opt':'login', 'result':False}

sk= socket.socket()
sk.bind(('127.0.0.1',9001))
sk.listen()
while True:
    conn,addr = sk.accept()
    msg = conn.recv(1024).decode('utf-8')
    dic_msg = json.loads(msg)
    if hasattr(sys.modules[__name__],dic_msg['operate']):
        ret = getattr(sys.modules[__name__],dic_msg['operate'])(dic_msg)
        content = json.dumps(ret).encode('utf-8')
        conn.send(content)
    conn.close()
sk.close()

import json
import socket
import hashlib

def get_md5(username,password):
    md5 = hashlib.md5(username[::2].encode('utf-8'))
    md5.update(password.encode('utf-8'))
    return md5.hexdigest()
username = input('用户名 :')
password = input('密  码 :')
sk = socket.socket()
sk.connect(('127.0.0.1',9001))
msg = {'operate':'login',
       'user':username,
       'pwd':get_md5(username,password)}
str_msg = json.dumps(msg)
sk.send(str_msg.encode('utf-8'))
content = sk.recv(1024)
str_content = content.decode('utf-8')
dic_content = json.loads(str_content)
if dic_content['result']:
    print('登录成功')
else:
    print('登录失败')
sk.close()

黏包sever/client

import json
import struct
import socket

sk = socket.socket()
sk.bind(('127.0.0.1',9001))
sk.listen()

conn,addr = sk.accept()
num = conn.recv(4)
num = struct.unpack('i', num)[0]
str_dic = conn.recv(num).decode('utf-8')
dic = json.loads(str_dic)
with open(dic['filename'],'wb') as f:
    content = conn.recv(dic['filesize'])
    f.write(content)

conn.close()
sk.close()

import os
import json
import struct
import socket

sk = socket.socket()
sk.connect(('127.0.0.1',9001))

filepath  = input('请输入文件路径 :')
filename = os.path.basename(filepath)
filesize = os.path.getsize(filepath)
dic = {'filename':filename,'filesize':filesize}   #包含文件名和文件字节
str_dic = json.dumps(dic)
bytes_dic = str_dic.encode('utf-8')
len_dic = len(bytes_dic)
bytes_len = struct.pack('i',len_dic)
sk.send(bytes_len)
sk.send(bytes_dic)
with open(filepath,'rb') as f:
    content = f.read()
    sk.send(content)
sk.close()
# D:programfeiqRecv Filesday29day295.总结.py
# 先发送字典的长度
# 再发字典 {'filename':xxxx,'filesize':xxxxx}
# 再发文件内容

黏包现象

import struct
import socket

sk = socket.socket()
sk.bind(('127.0.0.1',9001))
sk.listen()

conn,addr = sk.accept()
str_msg = 'hello,你好么'
byets_msg = str_msg.encode('utf-8')
num = len(byets_msg)
len_bytes = struct.pack('i',num)
conn.send(len_bytes)
conn.send(byets_msg)
conn.send(b'world')
conn.close()

sk.close()

import time
import struct
import socket

sk = socket.socket()
sk.connect(('127.0.0.1',9001))

time.sleep(0.1)
num= sk.recv(4)
num = struct.unpack('i',num)[0]
msg2 = sk.recv(num)
print(msg2.decode('utf-8'))
print(sk.recv(1024))

sk.close()

day30

并发的socketserver（两个client）

import time
import socketserver

class Myserver(socketserver.BaseRequestHandler):
    def handle(self):
        conn = self.request
        for i in range(200):
            conn.send(('hello%s'%i).encode('utf-8'))
            print(conn.recv(1024))
            time.sleep(0.5)


server = socketserver.ThreadingTCPServer(('127.0.0.1',9001),Myserver)
server.serve_forever()

import socket

sk = socket.socket()
sk.connect(('127.0.0.1',9001))
while True:
    msg = sk.recv(1024)
    print(msg)
    sk.send(b'byebye')
sk.close()

验证客户端的合法性

hmac模块

import os
import hmac

hmac = hmac.new(b'alex sb',os.urandom(32))
print(hmac.digest())

验证

import os
import hashlib
import socket

secret_key = b'alex sb'
#os.urandom(32) 给每一客户端发送一个随机的字符串,来保证即使数据被拦截你也不能使用这个消息
sk = socket.socket()
sk.bind(('127.0.0.1',9001))
sk.listen()

conn,addr = sk.accept()
rand = os.urandom(32)
conn.send(rand)

sha = hashlib.sha1(secret_key)
sha.update(rand)
res = sha.hexdigest()

ret = conn.recv(1024).decode('utf-8')
if ret == res:
    print('是合法的客户端')
else:
    print('不是合法的客户端')
    conn.close()

import socket
import hashlib

secret_key = b'alexsb'
sk = socket.socket()
sk.connect(('127.0.0.1',9001))

rand = sk.recv(32)

sha = hashlib.sha1(secret_key)
sha.update(rand)
res = sha.hexdigest()

sk.send(res.encode('utf-8'))

sk.close()