python socket 学习

Python在网络通讯方面功能强大，今天学习一下Socket通讯的基本方式，分别是UDP通讯和TCP通讯。

UDP通讯

upd 服务端

 #!/usr/bin/env python
 # -*- coding:utf-8 -*-

 import socket

 ADDR,PORT = 'localhost',7878
 sock = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
 sock.bind((ADDR,PORT))

 print 'waiting for connection...'

 while True:
     data, addr = sock.recvfrom(1024)
     print('Received data:', data, 'from', addr)

upd客户端

 #!/usr/bin/env python
 # -*- coding:utf-8 -*-

 import socket

 ADDR,PORT = 'localhost',7878
 sock = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
 sock.sendto(b'hello,this is a test info !',(ADDR,PORT))

先开启server端，等待client端的接入，每请求一次client会打印如下内容

waiting for connection...
('Received data:', 'hello,this is a test info !', 'from', ('127.0.0.1', 57331))
('Received data:', 'hello,this is a test info !', 'from', ('127.0.0.1', 61396))
('Received data:', 'hello,this is a test info !', 'from', ('127.0.0.1', 61261))
('Received data:', 'hello,this is a test info !', 'from', ('127.0.0.1', 54875))

TCP通讯

TCP服务端

 #!/usr/bin/env python
 # -*- coding:utf-8 -*-

 from socket import *
 import os

 ADDR,PORT = 'localhost',7878
 sock = socket(AF_INET,SOCK_STREAM)
 sock.bind((ADDR,PORT))
 sock.listen(5)

 while True:
     conn,addr = sock.accept()
     print "new conn:",addr
     while True:
         print 'waiting for connection'
         data = conn.recv(1024)
         if not data:
             print '客户端已经断开'
             break
         print '执行指令',data
         cmd_res = os.popen(data).read() #为执行传回的指令
         if len(cmd_res) == 0:
             print 'cmd has no output...'

         conn.send(str(len(cmd_res)).encode('utf-8')) #发送大小
         #client_chk = conn.recv(1024) 解决粘包问题 #wait client to confirm
         conn.send(cmd_res)
         print 'send done'
     conn.close()
 sock.close()

TCP客户端

 #!/usr/bin/env python
 # -*- coding:utf-8 -*-
 from socket import *

 ADDR,PORT = 'localhost',7878
 sock = socket(AF_INET,SOCK_STREAM)
 sock.connect((ADDR,PORT))
 while True:
     data = raw_input('>>')
     sock.send(data)
     print('发送信息到%s：%s' % (host, data))
     cmd_size = sock.recv(1024)
     print '命令结果大小 size',cmd_size
     sock.send('准备好接收了，可以发了')
     received_size = 0
     received_data = b''
     while received_size < int(cmd_size):
         data = sock.recv(1024)
         received_size += len(data)
         received_data += data
         print received_size
     else:
         print '=================\r\n'
         print 'cmd receive done',received_size
         print 'receive data:\r\n',received_data

 sock.close()