TCP/UDP服务器交互

@[toc]TCP/UDP服务器交互

一.TCP服务器交互检查

import socket
import time

def TCP_Server_Interaction(host,port):
    MaxBytes=1024*1024
    try:
        client = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
        client.settimeout(30)
        client.connect((host,port))
        check_time = time.strftime('%Y-%m-%d_%H_%M_%S', time.localtime(time.time()))

        inputData= 'CheckDataCommunication-{}'.format(check_time)   
        sendBytes = client.send(inputData.encode())
        print('sendBytes : {}'.format(sendBytes))
        recvData = client.recv(MaxBytes)
        client.close()
        if recvData.decode()==inputData:
            print('TCP Communication normal')
            return 'PASS'
        else:
            return 'TCP server connect fail'

    except:
        return 'TCP server connect fail'


if __name__=="__main__":
     host = '0.0.0.0'
     port =xxxx
   
    TCP_Server_Interaction(host, port)

二.UDP服务器交互检查

import socket
import time

def UDP_Server_Interaction(host,port):
    addr = (host, port)
    server_conn = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    print('server_conn res:{}'.format(server_conn))
    check_time = time.strftime('%Y-%m-%d_%H_%M_%S', time.localtime(time.time()))
    send_data = 'CheckDataCommunication-{}'.format(check_time)
    UDP_send = server_conn.sendto(send_data.encode(), addr)
    print('UDP_send_res : {}'.format(UDP_send))

    UDP_response, addr = server_conn.recvfrom(2048)
    print('UDP_response : {}'.format(UDP_response))
    server_conn.close()

    if UDP_response.decode() == send_data:
        print('UDP Communication normal')
        return 'PASS'
    else:
        return 'UDP server connect fail'
    

if __name__=="__main__":
    host = '0.0.0.0'
    port = xxxx
    print(UDP_Server_Interaction(host, port))