Python网络化编程

#server

import socket

s=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)#调用 socket 函数初始化

#print(s)

s.bind(("",8001)) #指定 IP 和端口

while True:

    data,addr=s.recvfrom(1024)#不断的接收数据

    print(data.decode())

    if data.decode=='exit':

        break

s.close()

#client

import socket

import uuid

import sys

s=socket.socket(socket.AF_INET,socket.SOCK_DGRAM)

def get_mac_adress():

    mac=uuid.UUID(int=uuid.getnode()).hex[-12:]

    return ":".join([mac[e:e+2] for e in range(0,11,2)])

ip=socket.gethostbyname(socket.gethostname())

#print ("ip 的值：",ip)

mac=get_mac_adress()

#print ("mac 的值",mac)

info="ip addr:"+ip+"\n"+"mac addr:"+mac

#print(info)

s.sendto(info.encode(),("127.0.0.1",8001))

s.sendto(sys.argv[1].encode(),("127.0.0.1",8001))

s.close()

#Server

import socket

s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

s.bind(("",8001))

s.listen(1)

print("Listen at port: 8001")

conn,addr = s.accept()

print(conn)

print(addr)

while True:

    data = conn.recv(1024)

    data = data.decode()

    print('Recv: ',data)

    c = input ('Input:')

    conn.sendall(c.encode())

    if c.lower()=='bye':

        break

conn.close()

s.close()

#Client

import socket

s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

try:

    s.connect(('127.0.0.1',8001))

except Exception as q:

    print("Error!!!")

while True:

    c = input("input: ")

    s.sendall(c.encode())

    data = s.recv(1024).decode()

    print('Recv:',data)

    if c.lower() == 'bye':

        break

s.close()

#Server

import socketserver

import struct

import os

import re

import json

import struct

from optparse import OptionParser

def sendFile(conn, head_info,head_info_len,filename):

    try:

        conn.send(head_info_len)

        conn.send(head_info.encode('utf-8'))

        with open(filename, 'rb') as f:

            conn.sendall(f.read())

        print('[+]send success! '+filename)

    except:

        print('[-]send fail! ' + filename)

def operafile(filename):

    filesize_bytes = os.path.getsize(filename)

    pattern = re.compile(r'([^<>/\\\|:""\*\?]+\.\w+$)')

    data = pattern.findall(filename)

    head_dir = {

        'filename': data,

        'filesize_bytes': filesize_bytes,

    }

    head_info = json.dumps(head_dir)

    head_info_len = struct.pack('i', len(head_info))

    return head_info_len, head_info

class MyServer(socketserver.BaseRequestHandler):

    buffsize = 1024

    def handle(self):

        print('[+]远程客户端ip地址：', self.client_address[0],'\n')

        while True:

            filename = input('请输入要发送的文件名>>>').strip() #移除字符串头尾指定的字符（默认为空格或换行符）或字符序列

            if(filename == "exit"):

                break

            head_info_len, head_info = operafile(filename)

            sendFile(self.request,head_info,head_info_len,filename)

        self.request.close()

def main():

    parser = OptionParser("Usage:%prog -p <port> ")   # 输出帮助信息

    parser.add_option('-p',type='string',dest='port',help='specify targer port')   # 获取 ip 地址参数

    options,args = parser.parse_args()

    port = int(options.port)

    print("[+]listening at " + str(port))

    s = socketserver.ThreadingTCPServer(('0.0.0.0', port), MyServer)  #

    s.serve_forever()

if __name__ == '__main__':

    try:

        main()

    except KeyboardInterrupt:

        print("interrupted by user, killing all threads...")

#client

from socket import *

import os

import sys

import json

import struct

from optparse import OptionParser

def recv_file(head_dir, tcp_client):

    filename = head_dir['filename']

    filesize = head_dir['filesize_bytes']

    print("[+]filename: "+filename[0])

    print("[+]filesize: "+ str(filesize))

    recv_len = 0

    f = open(filename[0], 'wb')

    while recv_len < filesize:

        if(filesize > 1024):

            recv_mesg = tcp_client.recv(1024)

            recv_len += len(recv_mesg)

            f.write(recv_mesg)

        else:

            recv_mesg = tcp_client.recv(filesize)

            recv_len += len(recv_mesg)

            f.write(recv_mesg)

    f.close()

    print('[+]文件传输完成!')

def main():

    parser = OptionParser("Usage:%prog -u <target address> -p <port> ")   # 输出帮助信息

    parser.add_option('-u', type='string', dest='ip', help='specify targer ip')  # 获取 ip 地址参数

    parser.add_option('-p',type='string',dest='port',help='specify targer port')   # 获取 ip 地址参数

    options,args = parser.parse_args()

    target_port = int(options.port)

    target_ip = options.ip

    tcp_client = socket(AF_INET, SOCK_STREAM)  # socket 初始化

    ip_port = ((target_ip, target_port))

    tcp_client.connect_ex(ip_port)

    print('[+]等待服务端应答数据....')

    struct_len = tcp_client.recv(4)  # 接收报头长度

    struct_info_len = struct.unpack('i', struct_len)[0]  # 解析得到报头信息的长度

    print("[+]接收头信息长度：" + str(struct_info_len))

    head_info = tcp_client.recv(struct_info_len)

    head_dir = json.loads(head_info.decode('utf-8'))  # 将报头的内容反序列化

    print("[+]输出头部信息内容：" + str(head_dir))

    recv_file(head_dir, tcp_client)

if __name__ == '__main__':

    try:

        main()

    except KeyboardInterrupt:

        print("interrupted by user, killing all threads...")