Python socket模塊方法實(shí)現(xiàn)詳解

更新時(shí)間：2019年11月05日 11:04:17 作者：星輝笑

這篇文章主要介紹了Python socket模塊方法實(shí)現(xiàn)詳解,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下

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)) #綁定被監(jiān)聽端口
server.listen(5)  #監(jiān)聽端口
while True:
  print("我要開始等電話了")
  conn, addr = server.accept() # 就是等待的意思
  #conn就是客戶端連過(guò)來(lái)的時(shí)候，在服務(wù)器端為其生成的一個(gè)連接實(shí)例
  print("電話來(lái)了%s"% [conn, addr])
  while True:
    data = conn.recv(1024)
    if not data:
      print('client is lost.')
      break
    # res = os.popen(data).read() #popen就是打開命令執(zhí)行，read就是獲取結(jié)果
    # with open('filename', 'r') as ret: #這兩行就 可以用過(guò)來(lái)傳輸文件了。
    #   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類型，同時(shí)生成socket鏈接對(duì)象
client.connect(('localhost',6969))  #localhost就是本機(jī)地址

while True:
  msg = input('input msg >>:').strip()
  if len(msg) == 0: continue #檢查msg的信息，防止無(wú)輸入信息
  #client.send(b"Hello, world!") #發(fā)送信息
  client.send(msg.encode('utf-8'))

  data = client.recv(1024) #默認(rèn)接受1024字節(jié)，就是1k
  # with open('filename', 'w') as ret: # 這兩行就 可以用過(guò)來(lái)傳輸文件了。
  #   ret = data.write()
  print(data.decode())
client.close() #關(guān)閉端口

防止粘包的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)) #綁定被監(jiān)聽端口
server.listen(5)  #監(jiān)聽端口
while True:
  print("我要開始等電話了")
  conn, addr = server.accept() # 就是等待的意思
  #conn就是客戶端連過(guò)來(lái)的時(shí)候，在服務(wù)器端為其生成的一個(gè)連接實(shí)例

  while True:
    data = conn.recv(1024).decode()
    print("電話來(lái)了%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就是打開命令執(zhí)行，read就是獲取結(jié)果
    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()) #通過(guò)接收數(shù)據(jù)的形式來(lái)強(qiáng)制發(fā)送緩沖區(qū)的數(shù)據(jù)，防止粘包。
    # with open('filename', 'r') as ret: #這兩行就 可以用過(guò)來(lái)傳輸文件了。
    #   data = ret.read()
    #print('receive:',data)
    print('receive:', data)
    conn.send(cmd_res.encode("utf-8"))
    # conn.send(bytes(cmd_res)) #不可行。傳輸?shù)臅r(shí)候是需要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類型，同時(shí)生成socket鏈接對(duì)象
client.connect(('localhost',6969))  #localhost就是本機(jī)地址

while True:
  msg = input('input msg >>:').strip()
  if len(msg) == 0: continue #檢查msg的信息，防止無(wú)輸入信息
  #client.send(b"Hello, world!") #發(fā)送信息
  client.send(msg.encode('utf-8'))
  received_size = client.recv(1024).decode() #用來(lái)記錄接受的數(shù)據(jù)大小
  print("接收的數(shù)據(jù)大小", received_size)
  received_cont = b''
  received_cont_size = 0 # 用來(lái)判斷接受數(shù)據(jù)的大小
  if received_size != "there is no output." :
    client.send("準(zhǔn)備好了，可以發(fā)送。".encode()) #發(fā)送確認(rèn)信息，以防止粘包
    received_size = int(received_size) #數(shù)據(jù)需要變成int才能進(jìn)行判斷
    while received_size != received_cont_size: #判斷encode后的長(zhǎng)度是否一致。
      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("當(dāng)前結(jié)束后的數(shù)據(jù)大小為：", received_cont_size)
      # print(received_cont_size)
    else:
      print("數(shù)據(jù)接收完成，接收的數(shù)據(jù)大小為：", received_cont_size)
      print("接收的內(nèi)容為：\n",received_cont.decode(),"\n")
  else:
    print("output:\n", received_size)
    # data = client.recv(1024) #默認(rèn)接受1024字節(jié)，就是1k
    # with open('filename', 'w') as ret: # 這兩行就 可以用過(guò)來(lái)傳輸文件了。
    #   ret = data.write()
    # print(data.decode())
    # print(str(data))
client.close() #關(guān)閉端口

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: