diff options
Diffstat (limited to 'examples/python/example.py')
| -rw-r--r-- | examples/python/example.py | 159 |
1 files changed, 159 insertions, 0 deletions
diff --git a/examples/python/example.py b/examples/python/example.py new file mode 100644 index 0000000..5a4a759 --- /dev/null +++ b/examples/python/example.py @@ -0,0 +1,159 @@ +import time, sys + +import libzt +from prototype import * + +# Where identity files are stored +keyPath = "." + +# Network to join +networkId = 0 + +# Port used by ZeroTier to send encpryted UDP traffic +# NOTE: Should be different from other instances of ZeroTier +# running on the same machine +ztServicePort = 9997 + +remoteIP = None + +# A port your app logic may use +serverPort = 8080 + +# Flags to keep state +is_joined = False +is_online = False +mode = None + +def print_usage(): + print("\nUsage: <server|client> <id_path> <nwid> <ztServicePort> <remoteIP> <serverPort>\n") + print(" Ex: python3 example.py server . 0123456789abcdef 9994 8080") + print(" Ex: python3 example.py client . 0123456789abcdef 9994 192.168.22.1 8080\n") + if (len(sys.argv) < 6): + print('Too few arguments') + if (len(sys.argv) > 7): + print('Too many arguments') + exit(0) +# +if (len(sys.argv) < 6 or len(sys.argv) > 7): + print_usage() + +if (sys.argv[1] == 'server' and len(sys.argv) == 6): + mode = sys.argv[1] + keyPath = sys.argv[2] + networkId = int(sys.argv[3],16) + ztServicePort = int(sys.argv[4]) + serverPort = int(sys.argv[5]) + +if (sys.argv[1] == 'client' and len(sys.argv) == 7): + mode = sys.argv[1] + keyPath = sys.argv[2] + networkId = int(sys.argv[3],16) + ztServicePort = int(sys.argv[4]) + remoteIP = sys.argv[5] + serverPort = int(sys.argv[6]) + +if (mode is None): + print_usage() + +print('mode = ', mode) +print('path = ', keyPath) +print('networkId = ', networkId) +print('ztServicePort = ', ztServicePort) +print('remoteIP = ', remoteIP) +print('serverPort = ', serverPort) + + + +# +# Event handler +# +class MyEventCallbackClass(libzt.PythonDirectorCallbackClass): + def on_zerotier_event(self, msg): + global is_online + global is_joined + print("eventCode=", msg.eventCode) + if (msg.eventCode == libzt.ZTS_EVENT_NODE_ONLINE): + print("ZTS_EVENT_NODE_ONLINE") + print("nodeId="+hex(msg.node.address)) + # The node is now online, you can join/leave networks + is_online = True + if (msg.eventCode == libzt.ZTS_EVENT_NODE_OFFLINE): + print("ZTS_EVENT_NODE_OFFLINE") + if (msg.eventCode == libzt.ZTS_EVENT_NETWORK_READY_IP4): + print("ZTS_EVENT_NETWORK_READY_IP4") + is_joined = True + # The node has successfully joined a network and has an address + # you can perform network calls now + if (msg.eventCode == libzt.ZTS_EVENT_PEER_DIRECT): + print("ZTS_EVENT_PEER_DIRECT") + if (msg.eventCode == libzt.ZTS_EVENT_PEER_RELAY): + print("ZTS_EVENT_PEER_RELAY") + + + +# +# Example start and join logic +# +print("Starting ZeroTier..."); +eventCallback = MyEventCallbackClass() +libzt.zts_start(keyPath, eventCallback, ztServicePort) +print("Waiting for node to come online...") +while (not is_online): + time.sleep(1) +print("Joining network:", hex(networkId)); +libzt.zts_join(networkId) +while (not is_joined): + time.sleep(1) # You can ping this app at this point +print('Joined network') + + + +# +# Example server +# +if (mode == 'server'): + print("Starting server...") + try: + serv = zerotier.socket(libzt.ZTS_AF_INET6, libzt.ZTS_SOCK_STREAM, 0) + serv.bind(('::', serverPort)) + serv.listen(5) + while True: + conn, addr = serv.accept() + print('Accepted connection from: ', addr) + while True: + print('recv()...') + data = conn.recv(4096) + if data: + print('data = ', data) + #print(type(b'what')) + #exit(0) + if not data: break + print('send()...') + #bytes(data, 'ascii') + b'\x00' + n_bytes = conn.send(data) # echo back to the server + print('sent ' + str(n_bytes) + ' byte(s)') + conn.close() + print('client disconnected') + except Exception as e: + print(e) + + +# +# Example client +# +if (mode == 'client'): + print("Starting client...") + try: + client = zerotier.socket(libzt.ZTS_AF_INET6, libzt.ZTS_SOCK_STREAM, 0) + print("connecting...") + client.connect((remoteIP, serverPort)) + print("send...") + data = 'Hello, world!' + client.send(data) + print("rx...") + data = client.recv(1024) + + print('Received', repr(data)) + except Exception as e: + print(e) + |