qmsk.net: examples/nc.py@ecd89f9605ac (annotated)

48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	1	#!/usr/bin/env python
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	2	"""
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	3	Simple (limited) netcat-like example, implemented using the socket/lib.event2 layer.
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	4	"""
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	5
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	6	from qmsk.net.socket import socket, address
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	7	from qmsk.net.socket.constants import *
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	8
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	9	from qmsk.net.lib.event2 import base, event
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	10	from qmsk.net.lib.event2.constants import *
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	11	from qmsk.net.lib.event2.event import CallbackEvent as cb_event
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	12
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	13	import sys, os, fcntl, errno
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	14	import optparse, logging
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	15
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	16	# global options
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	17	options = None
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	18
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	19	# global event_base
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	20	ev_base = base.event_base()
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	21
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	22	# buffer size to use, 4k - perhaps slightly smaller
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	23	BUFSIZE = 4 * 1024
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	24
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	25	# root logger
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	26	log = logging.getLogger()
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	27
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	28
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	29	def parse_argv (argv) :
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	30	global options
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	31
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	32	prog = argv.pop(0)
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	33
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	34	parser = optparse.OptionParser(prog=prog)
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	35
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	36	parser.add_option('-4', "--ipv4", help="Force AF_INET", action='store_true')
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	37	parser.add_option('-6', "--ipv6", help="Force AF_INET6", action='store_true')
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	38	parser.add_option('-v', "--verbose", help="Display status output", action='store_true')
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	39	parser.add_option('-d', "--debug", help="Display extra output", action='store_true')
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	40	parser.add_option('-c', "--connect", help="Act in client mode (default)", action='store_true')
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	41	parser.add_option('-l', "--listen", help="Act in server mode", action='store_true')
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	42	parser.add_option('-w', "--timeout", help="Timeout for connect()", type='float')
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	43
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	44	options, args = parser.parse_args(argv)
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	45
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	46	if options.ipv4 and options.ipv6 :
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	47	raise Exception("-4 and -6 are mutually exclusive")
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	48
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	49	if options.connect and options.listen :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	50	raise Exception("-c and -l are mutually exclusive")
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	51
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	52	return args
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	53
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	54
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	55	def setnonblocking (file) :
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	56	"""
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	57	Set the non-blocking IO flag on the given file object
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	58	"""
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	59
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	60	fcntl.fcntl(file.fileno(), fcntl.F_SETFL, os.O_NONBLOCK)
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	61
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	62	def sock_connect (host, port, family, socktype) :
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	63	"""
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	64	Try and perform a series of non-blocking connects to the given host:port using the given family and socktype,
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	65	yielding a series of socket objects.
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	66	"""
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	67
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	68	for ai in address.getaddrinfo(host, port, family, socktype) :
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	69	log.debug("Attempting connect() to: %s", ai)
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	70
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	71	# build socket
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	72	try :
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	73	# construct
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	74	log.debug("socket(%d, %d, %d)", ai.family, ai.socktype, ai.protocol)
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	75	sock = socket.socket(ai.family, ai.socktype, ai.protocol)
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	76
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	77	# set nonblock mode
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	78	log.debug("sock=%s: setnonblocking()", sock)
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	79	setnonblocking(sock)
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	80
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	81	# start connect
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	82	try :
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	83	log.debug("sock=%s: connect(%s)", sock, ai.addr)
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	84	sock.connect(ai.addr)
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	85
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	86	# XXX: socket.error
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	87	except OSError, e :
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	88	# this should return EINPROGRESS
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	89	if e.errno != errno.EINPROGRESS :
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	90	raise
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	91
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	92	else :
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	93	# wut???
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	94	log.warning("Connect was blocking")
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	95
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	96	except OSError, e :
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	97	# fsck
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	98	log.warning("Failed to connect to %s: %s", ai.addr, e)
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	99
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	100	else :
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	101	# yay
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	102	log.info("Connecting to %s...", ai.addr)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	103
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	104	yield sock
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	105
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	106	def sock_listen (port, family, socktype) :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	107	"""
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	108	Try and bind onto the given local service port, yielding a series of listen()'ing socket objects
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	109	"""
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	110
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	111	for ai in address.getaddrinfo(None, port, family, socktype) :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	112	log.debug("Binding on: %s", ai)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	113
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	114	# build socket
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	115	try :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	116	sock = socket.socket(ai.family, ai.socktype, ai.protocol)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	117
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	118	setnonblocking(sock)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	119
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	120	# bind local
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	121	sock.bind(ai.addr)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	122
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	123	# listen
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	124	sock.listen(5)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	125
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	126	except OSError, e :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	127	# fsck, next
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	128	log.warning("Failed to listen on %s: %s", ai.addr, e)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	129
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	130	else :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	131	# yay
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	132	log.debug("Listening on %s", ai.addr)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	133
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	134	yield sock
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	135
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	136	def client_connect_next (sock_iter) :
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	137	"""
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	138	Given an interable of sockets in the non-blocking connect() state, will attempt to wait for each, proceeding to
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	139	client_connected when one succeeds.
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	140	"""
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	141
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	142	for sock in sock_iter :
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	143	# pend for writing
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	144	log.debug("cb_event(%d, EV_WRITE, on_connect, %r)", sock.fd, sock)
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	145	ev = cb_event(ev_base, sock.fd, EV_WRITE, on_connect, sock, sock_iter)
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	146
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	147	# wait specified timeout
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	148	log.debug("ev=%r: add(timeout=%s)", ev, options.timeout)
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	149	ev.add(options.timeout)
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	150
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	151	# ok
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	152	break
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	153
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	154	else :
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	155	# fail, ran out of addresses to try
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	156	log.error("Unable to connect")
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	157
54 e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	158	def on_connect (ev, events, sock, sock_iter) :
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	159	"""
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	160	Outbound connect EV_WRITE callback, i.e. connection failed or was established
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	161	"""
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	162
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	163	log.debug("ev=%r: events=%#04x", ev, events)
54 e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	164
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	165	# test for timeout
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	166	if events & EV_TIMEOUT :
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	167	log.warning("Connection to %s timed out", sock.getpeername())
54 e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	168
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	169	# keep trying
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	170	return client_connect_next(sock_iter)
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	171
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	172
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	173	# test for errno
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	174	err = sock.getsockopt_int(SOL_SOCKET, SO_ERROR)
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	175
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	176	if err :
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	177	# fail
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	178	log.warning("Connection to %s failed: %d", sock.getpeername(), err)
54 e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	179
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	180	# keep trying
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	181	return client_connect_next(sock_iter)
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	182
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	183
e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	184	# ok, connected
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	185	client_connected(sock)
54 e7c136812b0d flesh out nc.py on_connect Tero Marttila <terom@fixme.fi> parents: 48 diff changeset	186
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	187
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	188	def client_connected (sock) :
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	189	"""
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	190	Connection established, start shovelling data
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	191	"""
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	192
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	193	log.info("Connected to %s (from %s)", sock.getpeername(), sock.getsockname())
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	194
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	195	# go!
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	196	start_sock(sock)
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	197
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	198
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	199	def server_start_accept (sock) :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	200	"""
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	201	Start waiting for connections on the given socket
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	202	"""
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	203
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	204	log.info("Listening on %s", sock.getsockname())
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	205
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	206	ev = cb_event(ev_base, sock.fd, EV_READ \| EV_PERSIST, on_server_accept, sock)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	207	ev.add()
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	208
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	209	def on_server_accept (ev, what, sock) :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	210	"""
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	211	Server socket is ready for accept()
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	212	"""
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	213
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	214	# no timeout support yet
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	215	assert what == EV_READ
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	216
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	217	log.debug("Incoming connection on %s", sock.getsockname())
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	218
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	219	# accept new client socket
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	220	client, addr = sock.accept()
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	221
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	222	# say so
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	223	log.info("Client connected from %s (to %s)", client.getpeername(), client.getsockname())
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	224
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	225	# handle it
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	226	start_sock(client)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	227
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	228
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	229	def start_sock (sock, fd_in=0, fd_out=1) :
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	230	"""
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	231	Start shipping data between sock and fin/fout.
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	232	"""
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	233
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	234	log.debug("Starting pipeline: %s -> %s -> %s", fd_in, sock, fd_out)
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	235
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	236	# event for each
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	237	ev_in = cb_event(ev_base, fd_in, EV_READ \| EV_PERSIST, on_read_in, sock, fd_in)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	238	ev_out = cb_event(ev_base, sock.fd, EV_READ \| EV_PERSIST, on_read_sock, sock, fd_out)
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	239
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	240	ev_in.add()
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	241	ev_out.add()
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	242
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	243
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	244	def on_read_in (ev, events, sock, fd_in) :
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	245	"""
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	246	Data readable on ev.fd, ship to sock
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	247	"""
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	248
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	249	# no timeouts yet
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	250	assert events & EV_READ
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	251
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	252	# try and read
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	253	buf = os.read(fd_in, BUFSIZE)
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	254
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	255	if buf :
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	256	# pass on
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	257	assert sock.write(buf) == len(buf)
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	258
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	259	else :
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	260	ev.delete()
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	261
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	262	# EOF
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	263	sock.shutdown(SHUT_WR)
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	264
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	265	log.info("EOF on input")
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	266
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	267	def on_read_sock (ev, events, sock, fd_out) :
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	268	"""
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	269	Data readable on sock, ship to fd_out
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	270	"""
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	271
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	272	# no timeouts yet
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	273	assert events & EV_READ
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	274
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	275	# try and read
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	276	buf = sock.read(BUFSIZE)
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	277
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	278	if buf :
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	279	# pass on
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	280	assert os.write(fd_out, buf) == len(buf)
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	281
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	282	else :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	283	ev.delete()
55 99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	284
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	285	# EOF
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	286	os.close(fd_out)
99c4344a35ce nc.py is now functional enough for basic client use Tero Marttila <terom@fixme.fi> parents: 54 diff changeset	287
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	288	log.info("EOF from %s", sock.getpeername())
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	289
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	290	def make_sockargs () :
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	291	"""
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	292	Returns (family, socktype) pair to use
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	293	"""
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	294	# figure out AF to use
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	295	if options.ipv4 :
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	296	family = AF_INET
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	297
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	298	elif options.ipv6 :
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	299	family = AF_INET6
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	300
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	301	else :
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	302	family = AF_UNSPEC
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	303
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	304	# fixed socktype
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	305	socktype = SOCK_STREAM
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	306
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	307	return family, socktype
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	308
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	309	def run_client (host, port) :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	310	"""
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	311	Execute in client-mode
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	312	"""
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	313
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	314	family, socktype = make_sockargs()
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	315
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	316	# look up the address and start a non-blocking connect
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	317	sock_iter = sock_connect(host, port, family, socktype)
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	318
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	319	# start waiting
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	320	client_connect_next(sock_iter)
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	321
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	322	def run_server (port) :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	323	"""
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	324	Execute in server-mode
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	325	"""
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	326
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	327	family, socktype = make_sockargs()
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	328
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	329	# get the first valid socket
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	330	for sock in sock_listen(port, family, socktype) :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	331	break
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	332
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	333	else :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	334	return log.error("Unable to listen on local service: %s", port)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	335
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	336	# wait
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	337	server_start_accept(sock)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	338
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	339	def main (argv) :
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	340	# setup logging
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	341	logging.basicConfig(
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	342	format = "[%(levelname)7s] %(funcName)15s: %(message)s",
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	343	level = logging.WARNING,
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	344	)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	345
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	346	# parse args
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	347	args = parse_argv(argv)
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	348
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	349	# pre-options
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	350	if options.debug :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	351	log.setLevel(logging.DEBUG)
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	352
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	353	elif options.verbose :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	354	log.setLevel(logging.INFO)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	355
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	356	# setup
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	357	if options.listen :
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	358	port, = args
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	359
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	360	run_server(port)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	361
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	362	else : # options.connect
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	363	host, port = args
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	364
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	365	run_client(host, port)
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	366
ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	367
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	368	# run mainloop
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	369	log.debug("Entering event loop")
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	370
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	371	if ev_base.loop() :
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	372	log.debug("Event loop done")
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	373
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	374	else :
58 ecd89f9605ac use logging for nc.py Tero Marttila <terom@fixme.fi> parents: 55 diff changeset	375	log.warning("Nothing to do!")
48 ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	376
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	377	if __name__ == '__main__' :
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	378	main(sys.argv)
ee7ade660c0b start writing socket+lib.event2 example (netcat clone), still broken due to lack of refcount magic in event Tero Marttila <terom@fixme.fi> parents: diff changeset	379

author	Tero Marttila <terom@fixme.fi>
	Mon, 05 Oct 2009 17:12:29 +0300
changeset 58	ecd89f9605ac
parent 55	99c4344a35ce
permissions	-rw-r--r--