Socket通信サンプル

PythonのSocket通信サンプル。
データの送信自体は最小2行で書ける。

with socket.socket(socket.AF_INET, args.p) as nsocket:
	# TCPの場合のみconnect
	if args.p is socket.SOCK_STREAM:
		nsocket.connect((host, port))
	# データ送信
	nsocket.send(data)

簡易データ投げ込みツール

パラメータを加えたncコマンド風、簡易データ投げ込みツール。
拡張予定。

import argparse
import socket

# パラメータ処理
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()

# 実際の通信
with socket.socket(socket.AF_INET, args.p) as nsocket:
	# TCPの場合のみconnect
	if args.p is socket.SOCK_STREAM:
		nsocket.connect((args.host, int(args.port)))
	# データ送信
	nsocket.send(send_data)
"""
	# 応答受信
	nsocket.settimeout(3)
	data = nsocket.recv(1024)
	
	print(repr(data))
"""