54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
# This example code is in the Public Domain (or CC0 licensed, at your option.)
|
|
|
|
# Unless required by applicable law or agreed to in writing, this
|
|
# software is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
|
# CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from builtins import input
|
|
import socket
|
|
import sys
|
|
|
|
# ----------- Config ----------
|
|
PORT = 3333
|
|
IP_VERSION = 'IPv6'
|
|
IPV4 = '192.168.0.42'
|
|
IPV6 = 'fd00:0000:0000:0000:260a:c4ff:fe09:885c'
|
|
# -------------------------------
|
|
|
|
if IP_VERSION == 'IPv4':
|
|
family_addr = socket.AF_INET
|
|
addr = (IPV4, PORT)
|
|
elif IP_VERSION == 'IPv6':
|
|
family_addr = socket.AF_INET6
|
|
for res in socket.getaddrinfo(IPV6, PORT, socket.AF_INET6,
|
|
socket.SOCK_STREAM, socket.SOL_TCP):
|
|
af, socktype, proto, canonname, addr = res
|
|
else:
|
|
print('IP_VERSION must be IPv4 or IPv6')
|
|
sys.exit(1)
|
|
|
|
try:
|
|
sock = socket.socket(family_addr, socket.SOCK_STREAM)
|
|
except socket.error as msg:
|
|
print('Could not create socket: ' + str(msg[0]) + ': ' + msg[1])
|
|
sys.exit(1)
|
|
|
|
try:
|
|
sock.connect((IPV6, PORT))
|
|
except socket.error as msg:
|
|
print('Could not open socket: ', msg)
|
|
sock.close()
|
|
sys.exit(1)
|
|
|
|
while True:
|
|
msg = input('Enter message to send: ')
|
|
assert isinstance(msg, str)
|
|
msg = msg.encode()
|
|
sock.sendall(msg)
|
|
data = sock.recv(1024)
|
|
if not data:
|
|
break
|
|
print('Reply: ' + data.decode())
|
|
sock.close()
|