Exemple de communication Python Socket / outil de lancement de données simple

Exemple de communication de socket

Exemple de communication Python Socket. La transmission de données elle-même peut être écrite en un minimum de 2 lignes.

with socket.socket(socket.AF_INET, args.p) as nsocket:
	#Se connecter uniquement pour TCP
	if args.p is socket.SOCK_STREAM:
		nsocket.connect((host, port))
	#Transmission de données
	nsocket.send(data)

Outil de lancement de données simple

Un outil de génération de données simple de type commande nc avec des paramètres ajoutés. Prévu pour être étendu.

import argparse
import socket

#Traitement des paramètres
parser = argparse.ArgumentParser()
parser.add_argument("-p", default=socket.SOCK_STREAM, help="Potocol TCP or UDP")
parser.add_argument("host", help="Hostname or IP Address")
parser.add_argument("port", help="Port number")
parser.add_argument("-m", default='0123456789abcdef' , help="Send data")
parser.add_argument("-f", type=argparse.FileType('rb'), help="Send data from a file")
args = parser.parse_args()

send_data = '0123456789abcdef'
if args.m:
	send_data = args.m.encode()

if args.f:
	send_data = args.f.read()

#Communication réelle
with socket.socket(socket.AF_INET, args.p) as nsocket:
	#Se connecter uniquement pour TCP
	if args.p is socket.SOCK_STREAM:
		nsocket.connect((args.host, int(args.port)))
	#Transmission de données
	nsocket.send(send_data)
"""
	#Recevoir une réponse
	nsocket.settimeout(3)
	data = nsocket.recv(1024)
	
	print(repr(data))
"""