本文假设你学过Python方面的知识
文章目录
前言
Socket
1.基本结构
2.高级应用
前言
如何加密自己的信息?如何基本安全的发送自己的信息?
Python的Socket库搭建的服务器可以完全胜任这种任务!
一、Socket简介
Socket其实就是Python其中关于网络套接字运用方面的库。
二、Socket
1.基本结构
服务端代码如下:
#coding:utf-8
def TCP_he(ip="",port=9999):
import socket #导入库
import threading
server=socket.socket(socket.AF_INET,socket.SOCK_STREAM) #实例化Socket对象
server.bind((ip,port)) #绑定Ip和端口
server.listen(5) #开始监听
print("[*] Listen on %s:%d"%(ip,port))
def chuli(a):
requst=a.recv(1024) #获取信息
print("[*] Received: "+str(requst))
a.send("I'm already") #往回发送字符串"I'm already"
a.close() #关闭与客户端的链接
while True:
client,addr=server.accept() #如果有客户端链接
print("[*]成功建立连接于 %s:%d" % (addr[0],addr[1]))
handler=threading.Thread(target=chuli,args=(client,)) #开启线程处理连接
handler.start()
try:
TCP_he()
except:
print("[*]Error!")
客户端代码如下(来源:堡塔插件:堡塔应用管理器 使用说明 - Linux面板 - 宝塔面板论坛 (bt.cn)https://www.bt.cn/bbs/thread-54473-1-1.html
#!/usr/bin/python
# coding: utf-8
import socket
socCli = socket.socket() #实例化Socket
try:
socCli.connect(('IP', port)) #连接指定服务器 (需更改IP和port)
while True:
data = input("# 输入内容:") #正常输入
socCli.send(data.encode("utf-8")) #向服务器发送转码后的信息
reply = socCli.recv(1024) #接收回传信息
if reply:
print("接收到信息:")
print(reply.decode("utf-8")) #解码后输出
except Exception as e:
print(e)
finally:
socCli.close() #关闭Socket连接
2.高级应用(来源:堡塔插件:堡塔应用管理器 使用说明 - Linux面板 - 宝塔面板论坛 (bt.cn)https://www.bt.cn/bbs/thread-54473-1-1.html
服务端代码(非堵塞服务器):
#!/usr/bin/python
# coding: utf-8
from __future__ import print_function, absolute_import
import socket
import queue
import logging
from select import select
import sys
SERVER_IP = ('', port ) #设定监听的端口
# 保存客户端发送过来的消息,将消息放入队列中
message_queue = {}
input_list = []
output_list = []
if __name__ == "__main__":
server = socket.socket()
server.bind(SERVER_IP)
server.listen(10)
# 设置为非阻塞
server.setblocking(False)
# 初始化将服务端加入监听列表
input_list.append(server)
print("sys stdout:")
print(sys.stdout.fileno())
print(sys.stdout)
while True:
# 开始 select 监听,对input_list中的服务端server进行监听
stdinput, stdoutput, stderr = select(input_list, output_list,
input_list)
# 循环判断是否有客户端连接进来,当有客户端连接进来时select将触发
for obj in stdinput:
# 判断当前触发的是不是服务端对象, 当触发的对象是服务端对象时,说明有新客户端连接进来了
if obj == server:
# 接收客户端的连接, 获取客户端对象和客户端地址信息
conn, addr = server.accept()
message = "Client {0} connected! ".format(addr)
print(message)
logging.warning(message)
# 将客户端对象也加入到监听的列表中, 当客户端发送消息时 select 将触发
input_list.append(conn)
# 为连接的客户端单独创建一个消息队列,用来保存客户端发送的消息
message_queue[conn] = queue.Queue()
else:
# 由于客户端连接进来时服务端接收客户端连接请求,将客户端加入到了监听列表中(input_list),客户端发送消息将触发
# 所以判断是否是客户端对象触发
try:
recv_data = obj.recv(1024)
# 客户端未断开
if recv_data:
message = "received {0} from client {1}".format(
recv_data.decode(), addr)
logging.warning(message)
print(message)
# 将收到的消息放入到各客户端的消息队列中
message_queue[obj].put(recv_data)
# 将回复操作放到output列表中,让select监听
if obj not in output_list:
output_list.append(obj)
except ConnectionResetError:
# 客户端断开连接了,将客户端的监听从input列表中移除
input_list.remove(obj)
# 移除客户端对象的消息队列
del message_queue[obj]
error_message = "\n[input] Client {0} disconnected".format(addr)
logging.error(error_message)
print(error_message)
# 如果现在没有客户端请求,也没有客户端发送消息时,开始对发送消息列表进行处理,是否需要发送消息
for sendobj in output_list:
try:
# 如果消息队列中有消息,从消息队列中获取要发送的消息
if not message_queue[sendobj].empty():
# 从该客户端对象的消息队列中获取要发送的消息
send_data = message_queue[sendobj].get()
sendobj.sendall(send_data)
else:
# 将监听移除等待下一次客户端发送消息
output_list.remove(sendobj)
except ConnectionResetError:
# 客户端连接断开了
del message_queue[sendobj]
output_list.remove(sendobj)
print("\n[output] Client {0} disconnected".format(addr))