FTP项目
查看列表
客户端
from socket import *
import pickle
def list_file():
# 3.创建对象
tcp_client = socket()
# 4.连接服务器
tcp_client.connect(("172.16.13.212", 8989))
# 5.发送消息
tcp_client.send("L".encode("utf-8"))
# 6.接收消息
raw_data = tcp_client.recv(1024)
# 7.转化消息类型
data = pickle.loads(raw_data)
# 8.展示消息内容
print(data)
# 关闭连接
# tcp_client.close()
def down_file():
pass
def upload_file():
pass
def main():
while True:
# 1.提示输入操作命令
print("L：查看列表\nD：下载\nU：上传\nexit：退出")
data = input(">>>请选择命令：")
# 2.判断用户的功能
if data == "exit":
# 2.0退出
break
elif data[0].upper().strip() == "L":
# 2.1查看列表
list_file()
elif data[0].upper().strip() == "D":
# 2.2下载
down_file()
elif data[0].upper().strip() == "U":
# 2.3上传
upload_file()
else:
print("命令错误，请重新输入")
if __name__ == '__main__':
main()
客户端
from socket import *
import os
import pickle
def list_file(sock):
print("查看列表")
# a.查看当前目录，生成列表
data = os.listdir()
# b.对列表进行序列化
data = pickle.dumps(data)
# c.返回消息
res = sock.send(data)
print(f"成功发送{res}字节")
# d.断开连接
sock.close()
def down_file(sock):
pass
def upload_file(sock):
pass
def main():
# 1.创建对象
tcp_server = socket()
# 2.保证端口不被占用
tcp_server.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
# 3.绑定端口
tcp_server.bind(("", 8989))
# 4.改主动为被动 注：只有在unix下生效
tcp_server.listen(5)
# 5.等待连接
while True:
# 6.解包
new_socket, client_info = tcp_server.accept()
# 7.接收消息
raw_data = new_socket.recv(1024)
# 8.判断客户有没有断开
if raw_data:
# 9.判断客户的命令
if raw_data.decode("utf-8")[0] == "L":
# 9.1查看文件列表
list_file(new_socket)
if raw_data.decode("utf-8")[0] == "L":
# 9.2下载
down_file(new_socket)
if raw_data.decode("utf-8")[0] == "L":
# 9.3上传
upload_file(new_socket)
else:
# 10.关闭
new_socket.close()
break
if __name__ == '__main__':
main()
下载
遇坑：linux下为服务端，在windows连接不上虚拟机的服务端
解决办法：在linux下运行命令 service iptables stop 关闭了linux防火墙
客户端
from socket import *
import pickle
def list_file():
# 3.创建对象
tcp_client = socket()
# 4.连接服务器
tcp_client.connect(("192.168.2.107", 8989))
# 5.发送消息
tcp_client.send("L".encode("utf-8"))
# 6.接收消息
raw_data = tcp_client.recv(1024)
# 7.反序列化
data = pickle.loads(raw_data)
# 8.展示消息内容
print(data)
# 关闭连接
# tcp_client.close()
def down_file(data):
# 3.创建对象
tcp_client = socket()
# 4.连接服务器
tcp_client.connect(("192.168.2.107", 8989))
# 5.获取到文件名
file_name = data[2:]
# 6.发送消息
tcp_client.send(f"D:{file_name}".encode("utf-8"))
# 7.接收消息
raw_data = tcp_client.recv(1024)
# 8.打开新文件保存消息
with open(file_name, "ab") as fw:
# 9判断是否接收到消息
while raw_data:
# 10.保存到文件中
fw.write(raw_data)
# 11.继续接收消息
raw_data = tcp_client.recv(1024)
# 12.关闭连接
# tcp_client.close()
def upload_file():
pass
def main():
while True:
# 1.提示输入操作命令
print("=" * 30, "\nL：查看列表\nD：下载\nU：上传\nexit：退出\n", "=" * 30)
data = input(">>>请选择命令：")
# 2.判断用户的功能
if data == "exit":
# 2.0退出
break
elif data[0].upper().strip() == "L":
# 2.1查看列表
list_file()
elif data[0].upper().strip() == "D":
# 2.2下载
down_file(data)
elif data[0].upper().strip() == "U":
# 2.3上传
upload_file()
else:
print("命令错误，请重新输入")
if __name__ == '__main__':
main()
服务端
from socket import *
import os
import pickle
def list_file(sock):
print("查看列表")
# 11.查看当前目录，生成列表
data = os.listdir()
# 12.对列表进行序列化
data = pickle.dumps(data)
# 13.返回消息
res = sock.send(data)
print(f"成功发送{res}字节")
# 14.断开连接
sock.close()
def down_file(sock, data):
print("下载")
# 11.获取文件名
file_name = data.decode("utf-8")[2:]
# 12.发送文件内容
# 13.打开文件
with open(file_name, "rb") as fr:
# 14.一次获取1024
buf = fr.read(1024)
# 15.判断是否取到消息
while buf:
# 16.把取到的写入到文件中
sock.send(buf)
# 17.继续从缓存区取
buf = fr.read(1024)
# 18.关闭文件
fr.close()
# 19.关闭连接
sock.close()
def upload_file(sock):
pass
def main():
# 1.创建对象
tcp_server = socket()
# 2.保证端口不被占用
tcp_server.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
# 3.绑定端口
tcp_server.bind(("", 8989))
# 4.改主动为被动 注：只有在unix下生效
tcp_server.listen(5)
# 5.等待连接
while True:
# 6.解包
new_socket, client_info = tcp_server.accept()
# 7.接收消息
raw_data = new_socket.recv(1024)
# 8.判断客户有没有断开
if raw_data:
# 9.判断客户的命令
if raw_data.decode("utf-8")[0] == "L":
# 9.1查看文件列表
list_file(new_socket)
if raw_data.decode("utf-8")[0] == "D":
# 9.2下载
down_file(new_socket, raw_data)
if raw_data.decode("utf-8")[0] == "U":
# 9.3上传
upload_file(new_socket)
else:
# 10.关闭
new_socket.close()
break
if __name__ == '__main__':
main()
上传
遇坑：粘包
解决办法：在文件名前传递了文件名长度
注意：文件名最大长度255
客户端
from socket import *
import pickle
def list_file():
# 3.创建对象
tcp_client = socket()
# 4.连接服务器
tcp_client.connect(("192.168.2.107", 8989))
# 5.发送消息
tcp_client.send("L".encode("utf-8"))
# 6.接收消息
raw_data = tcp_client.recv(1024)
# 7.反序列化
data = pickle.loads(raw_data)
# 8.展示消息内容
print(data)
# 9.关闭连接
tcp_client.close()
def down_file(data):
# 3.创建对象
tcp_client = socket()
# 4.连接服务器
tcp_client.connect(("192.168.2.107", 8989))
# 5.获取到文件名
file_name = data[2:]
# 6.发送消息
tcp_client.send(f"D:{file_name}".encode("utf-8"))
# 7.接收消息
raw_data = tcp_client.recv(1024)
# 8.打开新文件保存消息
with open(file_name, "ab") as fw:
# 9判断是否接收到消息
while raw_data:
# 10.保存到文件中
fw.write(raw_data)
# 11.继续接收消息
raw_data = tcp_client.recv(1024)
# 12.关闭连接
tcp_client.close()
def upload_file(data):
# 3.创建对象
tcp_client = socket()
# 4.连接服务器
# tcp_client.connect(("192.168.2.107", 8989))
# tcp_client.connect(("192.168.2.104", 8989))
tcp_client.connect(("172.16.13.216", 8988))
# 5.获取到文件名
file_name = data[2:].strip()
# 6.获取到文件名的长度,并转化为3位数的字符串,左边补0
length = "%05d" % len(file_name.encode("utf-8"))
# 6.发送消息
tcp_client.send(f"U:{length}{file_name}".encode("utf-8"))
# 7.打开文件
with open(file_name, "rb") as fr:
# 8.一次取1024
buf = fr.read(1024)
# 9.如果取出内容
while buf:
# 10.把取出的文件发给服务器
tcp_client.send(buf)
# 11.继续获取
buf = fr.read(1024)
# 12.关闭连接
tcp_client.close()
def main():
while True:
# 1.提示输入操作命令
print("=" * 30, "\nL：查看列表\nD：下载\nU：上传\nexit：退出\n", "=" * 30)
data = input(">>>请选择命令：")
# 2.判断用户的功能
if data == "exit":
# 2.0退出
break
elif data[0].upper().strip() == "L":
# 2.1查看列表
list_file()
elif data[0].upper().strip() == "D":
# 2.2下载
down_file(data)
elif data[0].upper().strip() == "U":
# 2.3上传
upload_file(data)
else:
print("命令错误，请重新输入")
if __name__ == '__main__':
main()
服务端
from socket import *
import os
import pickle
def list_file(sock):
print("查看列表")
# 11.查看当前目录，生成列表
data = os.listdir()
# 12.对列表进行序列化
data = pickle.dumps(data)
# 13.返回消息
res = sock.send(data)
print(f"成功发送{res}字节")
# 14.断开连接
sock.close()
def down_file(sock, data):
print("下载")
# 11.获取文件名
file_name = data.decode("utf-8")[2:]
# 12.发送文件内容
# 13.打开文件
with open(file_name, "rb") as fr:
# 14.一次获取1024
buf = fr.read(1024)
# 15.判断是否取到消息
while buf:
# 16.把取到的写入到文件中
sock.send(buf)
# 17.继续从缓存区取
buf = fr.read(1024)
# 18.关闭文件
fr.close()
# 19.关闭连接
sock.close()
def upload_file(sock, data):
# 11.输出操作
print("上传")
# print(data)
# 12.获取文件名字
file_name = data[7:7 + int(data[2:7])].decode("utf-8")
# print(file_name)
# print("我是粘包", data)
# 13.获取粘包内容
file_last = data[7 + int(data[2:7]):]
# 14.打开文件
with open(file_name, "ab") as fw:
# 15.如果存在粘包，就把粘包也写进去
if file_last:
# 16.写入到文件
fw.write(file_last)
# 17.继续获取
raw_data = sock.recv(1024)
# 18.如果获取到消息
while raw_data:
# 19.写入到文件
fw.write(raw_data)
# 20.继续获取
raw_data = sock.recv(1024)
# 21.关闭文件
fw.close()
# 22.关闭连接
sock.close()
def main():
try:
# 1.创建对象
tcp_server = socket()
# 2.保证端口不被占用
tcp_server.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
# 3.绑定端口
tcp_server.bind(("", 8988))
# 4.改主动为被动 注：只有在unix下生效
tcp_server.listen(5)
# 5.等待连接
while True:
# 6.解包
new_socket, client_info = tcp_server.accept()
# 7.接收消息
raw_data = new_socket.recv(1024)
# 8.判断客户有没有断开
if raw_data:
# 9.判断客户的命令
if raw_data[0:1] == b"L":
# 9.1查看文件列表
list_file(new_socket)
if raw_data[0:1] == b"D":
# 9.2下载
down_file(new_socket, raw_data)
if raw_data[0:1] == b"U":
# 9.3上传
upload_file(new_socket, raw_data)
else:
# 10.关闭
new_socket.close()
break
except:
new_socket.close()
if __name__ == '__main__':
main()
网友评论