码迷,mamicode.com
首页 > 其他好文 > 详细

socket模块

时间:2018-06-01 21:33:25      阅读:221      评论:0      收藏:0      [点我收藏+]

标签:数据   block   字符   serve   comm   结果   for   host   sam   

1、TCP和UDP

TCP可靠性的实现:

  • 校验码
  • 接收方反馈
  • 信息包附带序号

UDP:

  • 快 不需要花费时间建立和关闭连接
  • 快 偶尔丢失一两个消息包无所谓,但是TCP会严格检查
  • 快 UDP的限制是一个信息包不超过64KB的数据

TCP和UDP区别就是UDP不建立连接,只保证数据的完整性,数据传输快,但是不保证数据是否真的被收到,也不保证数据是够只接收一次,也不保证次序。

服务端是用来给一个或者多个客户端提供服务的,当客户端发起请求,开始等待服务端的返回结果,服务端接受完请求以后,根据自己的逻辑进行处理请求,并返回给客户端,客户端接收到返回结果以后,关闭和服务端的连接。

最常用个客户端和服务端有两种模式C/S(mysql)模式和B/S模式(网站)

实现图:
技术分享图片

2、socket常用函数

①、创建套接字
s = socket.socket(address_family, socket_type)
address family:
socket.AF_INET           默认ipv4
Socket.AF_INET             ipv6
Socket.AF_UNIX       只能用于单一的unix系统进行间通信
socket type:
socket.SOCK_STREAM    流式socket,TCP
socket.SOCK_DGRAM    数据报是socket   UDP

TCP:
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

 绑定地址:   bind(address)
address = (‘0.0.0.0’, 8009)
s.bind(address) 或者s.bind((‘0.0.0.0’, 8009))
address 必须是一个元组,容易错误,address = (host,port)
host:服务端ip,字符串类型, 如果为0.0.0.0  代表本机的任意一个ip
port:服务端提供的端口, 整型, 0-1024为系统保留

监听消息:  
s.listen(badklog)
backlog代表可以同时接受多少个socket连接

接受连接:
conn, addr = s.accept()
接受连接并返回元组(conn, addr), 其中conn是新的套接字对象,每一个新的连接就创建一个新的对象。可以用来接受和发送数据,addr是客户端的地址。包含host和port

发送数据:
s.send(string)             发送字符串到连接的套接字,可能未将指定内容全部发送
s.sendall(string)      内部递归调用send,将所有内容发送出去,建议使用。

接收数据:
data = s.recv(bufsize)
接收套接字数据,数据以字符串形式返回,bufsize指定最多接收的数据量,可以使用1024, 2048
如果不知道接收的数量有多少,可能几个字节,可能几M,一般通过循环接收

UDP:
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.sendto(string)
data, address = s.recvfrom(bufsize)

客户端:
客户端首先也要创建socket套接字
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
客户端连接服务端函数:
s.connect(address)                连接到address的套接字
result = connect_ex(address)        成功返回0,失败返回错误码(推荐使用)

通用:
s.close()                关闭socket套接字
s.getsocketname()                   获取套接字的名字
s.settimeout(timeout)        设置套接字超时时间,timeout为float类型,单位为秒。
s.gettimeout()             获得套接字超时时间

通用:
s.setblocking(flag)
flage为bool值
setblocking(True) is equivalent to settimeout(None); 相当于不设置超时时间,一直阻塞在那里
setblocking(False) is equivalent to settimeout(0.0). 相当于设置超时时间为0, 如果设置False,那么accept和recv时一旦无数据,则报错。

s.fileno()
返回套接字的文件描述符(一个小整数)。这对于select.select()是有用的。

3、实现ssh链接功能

server端01:

import socket
import paramiko
from threading import Thread
import sys


class SSH(Thread):
def __init__(self, ip, port, user, password):
# super().__init__()
Thread.__init__(self)
self.ip = ip
self.prot = port
self.user = user
self.password = password

def ssh(self, cmd):
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try:
ssh.connect(hostname=self.ip, username=self.user, password=self.password, timeout=1)
except:
print("Timeout or not parmission")
return 1
stdin, stdout, stderr = ssh.exec_command(cmd) # 输出会分别保存到三个文件里
stdout = stdout.read() # [:-1]指的是去掉最后一个字符,因为这个最后输出会有个换行符
stderr = stderr.read()[:-1]
ssh.close()
if stdout:
return stdout.decode(‘utf-8‘) or "ok"
else:
return stderr

def run(self):
HOST = ‘‘
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind((HOST, self.prot))
s.listen(1)
conn, addr = s.accept()
print("Connected by: {0}".format(addr))
while True:
try:
data = conn.recv(1024)
if not data:
break
conn.sendall(self.ssh(data))
except:
print("close connection")
break
s.close()

if __name__ == ‘__main__‘:
ip = ‘127.0.0.1‘
prot = 1234
user = ‘root‘
password = ‘123456‘
ssh = SSH(ip, prot, user, password)
ssh.start()

server端02:

import socket
import commands

HOST = ‘‘ # Symbolic name meaning all available interfaces
PORT = 1234 # Arbitrary non-privileged port
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind((HOST, PORT))
s.listen(1)
conn, addr = s.accept()
print(‘Connected by‘, addr)
while True:
data = conn.recv(1024)
if not data:
break
result = commands.getoutput(data)
conn.sendall(result)
s.close()

client端:

import socket
import sys

HOST = ‘192.168.137.130‘ # The remote host
PORT = 1234 # The same port as used by the server
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
s.connect((HOST, PORT))
while 1:
world = input("[root@jiang socket]#")
if world:
s.sendall(world.encode(‘utf-8‘))
data = s.recv(1024)
if data:
# print(‘Received‘, repr(data.decode(‘utf-8‘)).upper())
print("{0}".format(data.decode(‘utf-8‘)))

else:
print("input nothing!")
sys.exit(1)

 

 

 

 

 

 

 

 

 

 

 

socket模块

标签:数据   block   字符   serve   comm   结果   for   host   sam   

原文地址:https://www.cnblogs.com/Jweiqing/p/9123468.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!