Python学习——socket 模块

http://www.cnblogs.com/alex3714/articles/5830365.html 可以看的内容

socket ssh (不带防止粘包的方法）

#! /usr/bin/env python
# -*- coding:utf-8 -*-
# Author Ian Ying
# mail: kongqing.ying@yitu-inc.com

import socket
import os

server = socket.socket()
server.bind(('localhost', 6969))  #绑定被监听端口
server.listen(5)   #监听端口
while True:
    print("我要开始等电话了")
    conn, addr = server.accept()  # 就是等待的意思
    #conn就是客户端连过来的时候，在服务器端为其生成的一个连接实例
    print("电话来了%s"% [conn, addr])
    while True:
        data = conn.recv(1024)
        if not data:
            print('client is lost.')
            break
        # res = os.popen(data).read() #popen就是打开命令执行，read就是获取结果
        # with open('filename', 'r') as ret:  #这两行就 可以用过来传输文件了。
        #     data = ret.read()
        print('receive:',data)
        conn.send(data.upper())


server.close()

socket client 模块

#! /usr/bin/env python
# -*- coding:utf-8 -*-
# Author Ian Ying
# mail: kongqing.ying@yitu-inc.com

import socket


client = socket.socket() #声明socket类型，同时生成socket链接对象
client.connect(('localhost',6969))   #localhost就是本机地址

while True:
    msg = input('input msg >>:').strip()
    if len(msg) == 0: continue  #检查msg的信息，防止无输入信息
    #client.send(b"Hello, world!") #发送信息
    client.send(msg.encode('utf-8'))

    data = client.recv(1024) #默认接受1024字节，就是1k
    # with open('filename', 'w') as ret:  # 这两行就 可以用过来传输文件了。
    #     ret = data.write()

    print(data.decode())

client.close() #关闭端口

防止粘包的socket_ssh.py

#! /usr/bin/env python
# -*- coding:utf-8 -*-
# Author Ian Ying
# mail: kongqing.ying@yitu-inc.com

import socket
import os

server = socket.socket()
server.bind(('localhost', 6969))  #绑定被监听端口
server.listen(5)   #监听端口
while True:
    print("我要开始等电话了")
    conn, addr = server.accept()  # 就是等待的意思
    #conn就是客户端连过来的时候，在服务器端为其生成的一个连接实例

    while True:
        data = conn.recv(1024).decode()
        print("电话来了%s" % type(data))
        # if type(data) is str:
        #     data = data.strip()
        if not data:
            print('client is lost.')
            break
        cmd_res = os.popen(data).read() #popen就是打开命令执行，read就是获取结果
        cmd_res_size = str(len(cmd_res.encode("utf-8")))
        print("before send",len(cmd_res),"size after encode", cmd_res_size)
        if len(cmd_res) == 0:
            print("there is no output.")
            res_warning = "there is no output."
            conn.send(res_warning.encode("utf-8"))
            continue
        else:
            conn.send(cmd_res_size.encode("utf8"))
            print(conn.recv(1024).decode())  #通过接收数据的形式来强制发送缓冲区的数据，防止粘包。
        # with open('filename', 'r') as ret:  #这两行就 可以用过来传输文件了。
        #     data = ret.read()
        #print('receive:',data)
        print('receive:', data)
        conn.send(cmd_res.encode("utf-8"))
        # conn.send(bytes(cmd_res)) #不可行。传输的时候是需要encoding


server.close()

socket_client.py

#! /usr/bin/env python
# -*- coding:utf-8 -*-
# Author Ian Ying
# mail: kongqing.ying@yitu-inc.com

import socket


client = socket.socket() #声明socket类型，同时生成socket链接对象
client.connect(('localhost',6969))   #localhost就是本机地址

while True:
    msg = input('input msg >>:').strip()
    if len(msg) == 0: continue  #检查msg的信息，防止无输入信息
    #client.send(b"Hello, world!") #发送信息
    client.send(msg.encode('utf-8'))
    received_size = client.recv(1024).decode() #用来记录接受的数据大小
    print("接收的数据大小", received_size)
    received_cont = b''
    received_cont_size = 0  # 用来判断接受数据的大小
    if received_size != "there is no output." :
        client.send("准备好了，可以发送。".encode())  #发送确认信息，以防止粘包
        received_size = int(received_size) #数据需要变成int才能进行判断
        while received_size != received_cont_size:  #判断encode后的长度是否一致。
            received_cont_for_test = client.recv(1024)
            received_cont_size += int(len(received_cont_for_test))
            received_cont = received_cont + received_cont_for_test
            print("当前结束后的数据大小为：", received_cont_size)
            # print(received_cont_size)
        else:
            print("数据接收完成，接收的数据大小为：", received_cont_size)
            print("接收的内容为：
",received_cont.decode(),"
")
    else:
        print("output:
", received_size)


        # data = client.recv(1024) #默认接受1024字节，就是1k
        # with open('filename', 'w') as ret:  # 这两行就 可以用过来传输文件了。
        #     ret = data.write()
        # print(data.decode())
        # print(str(data))


client.close() #关闭端口