套接字的应用

基于UDP实现ntp时间服务器:

服务端

from socket import *
import time

ip_port = ('127.0.0.2',8080)
buffer_size = 1204

udp_server = socket(AF_INET,SOCK_DGRAM)
udp_server.bind(ip_port)
while True:
    data, addr = udp_server.recvfrom(buffer_size)
    back_time = time.strftime('%Y-%m-%d-%X')
    udp_server.sendto(back_time.encode('utf-8'),addr)    
    print(data.decode('utf-8'))
udp_server.close()

客户端:

from socket import *

ip_port = ('127.0.0.2',8080)
buffer_size = 1204

udp_client = socket(AF_INET,SOCK_DGRAM)
while True:
    msg = input('任意键获取时间')
    udp_client.sendto(msg.encode('utf-8'),ip_port)
    data,addr = udp_client.recvfrom(buffer_size)
    print(data.decode('utf-8'))
udp_client.close()

输出:
任意键获取时间w3
2018-12-11-17:21:05
任意键获取时间
2018-12-11-17:21:06
任意键获取时间edf
2018-12-11-17:21:06

基于TCP实现远程控制CMD命令或Linux命令

服务端

from socket import *
import subprocess

ip_port = ('127.0.0.1',8080)
back_log = 5
buffer_size = 1024

tcp_server = socket(AF_INET,SOCK_STREAM)
tcp_server.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)  # 重新使用IP地址,防止出现Address被占用的错误
tcp_server.bind(ip_port)
tcp_server.listen(back_log)
while True:  #链接循环
        print('服务器运行了')
        conn,addr = tcp_server.accept()
        print('新的客户端链接',addr)
        while True:  #通讯循环
            try:
                cmd = conn.recv(buffer_size)
                if not cmd:     #当客户端正常断开时,cmd一直在收空信息
                    break
                print('收到的客户端命令是',cmd)
                res = subprocess.Popen(cmd.decode('utf8'),shell=True,stdout=subprocess.PIPE,
                                       stdin=subprocess.PIPE,stderr=subprocess.PIPE)
                err = res.stderr.read()
                if err:
                    cmd_res = err
                else:
                    cmd_res = res.stdout.read()
                  
                conn.send(cmd_res)
            except Exception as e:
                print(e)
                break      #异常退出时,进入下次循环
        conn.close()       #正常退出时,断开链接
tcp_server.close()

输出:
服务器运行了
新的客户端链接 (‘127.0.0.1’, 58242)
收到的客户端命令是 b’dd’
收到的客户端命令是 b’gg’
[WinError 10054] 远程主机强迫关闭了一个现有的连接。
服务器运行了
新的客户端链接 (‘127.0.0.1’, 58245)
收到的客户端命令是 b’dd’
收到的客户端命令是 b’ddd’
收到的客户端命令是 b’dir’
[WinError 10054] 远程主机强迫关闭了一个现有的连接。
服务器运行了
新的客户端链接 (‘127.0.0.1’, 58271)
收到的客户端命令是 b’dd’

客户端:

from socket import *

ip_port = ('127.0.0.1',8080)
buffer_size = 1024

tcp_client = socket(AF_INET,SOCK_STREAM)
tcp_client.connect(ip_port)
while True:
    cmd = input('>>:').strip()
    if not cmd:
        continue
    if cmd == 'quit':
        break
    tcp_client.send(cmd.encode('utf-8'))
    data = tcp_client.recv(buffer_size)
    print(data.decode('gbk'))
tcp_client.close()

输出:

: ddd
‘ddd’ 不是内部或外部命令，也不是可运行的程序
或批处理文件。

:dir
驱动器 C 中的卷没有标签。
卷的序列号是 EC05-9F96

C:\Users\Administrator\PycharmProjects\untitled1 的目录

2018/12/11 21:20

.
2018/12/11 21:20 …
2018/12/11 21:18 .idea
2018/12/11 21:07 399 客户端.py
2018/12/11 20:56 399 客户端2.py
2018/12/11 21:20 1,358 服务端.py
2018/11/30 20:28 542 马林
2018/11/30 22:05 542 马林2
13 个文件 5,251 字节
4 个目录 7,543,455,744 可用字节

:
Process finished with exit code -1