def run(self):
"""run() -> nothing.
Main loop for Tracker.
"""
print("Tracker.")
while True:
session_id, data = self.recv()
message_type, message = lightning.unpack(data)
print("Got message: {0}/{1}".format(message_type, message))
if message_type not in self._tr_table:
# TODO: Should be handled instead.
#self._sck_in.send(lightning.pack(lightning.OK)) # TEMPORARY
if not self.send(session_id, lightning.OK):
print("unable to send message.")
print("Unhandled message. Sent OK back.")
continue
try:
ret = self._tr_table[message_type](self, message)
if ret is None:
continue
if not self.send(ret):
print("unable to send message: {0}".format(str(ret)))
else:
print("sent in return: {0}".format(str(ret)))
except Exception as e:
print("Function {0} raised an exception:".format(message_type))
traceback.print_exc()
def main():
print "Client."
zmq_context = zmq.Context()
zmq_socket = zmq_context.socket(zmq.REQ)
zmq_socket_in = zmq_context.socket(zmq.REP)
session_id = 0
batchhash = hashlib.sha1("aoeu").hexdigest()
batchtype = tracker_pb2.NewBatch.RAYTRACE
tasks = 12000
available = 4
while True:
print "1: Connect to tracker\n" +\
"2: NewBatch\n" +\
"3: NodeRequest\n" +\
"4: GoodBye\n" +\
"9: PING\n" \
"0: Quit\n"
i = raw_input("Message to send: ")
try:
i = int(i)
except ValueError:
break
if i == 0: break
elif i == 1:
# INIT_HELLO
print "Connecting to tracker..."
zmq_socket.connect("tcp://" + lightning.tracker_address + ":" + lightning.tracker_port)
# Create and send hello message
message = lightning.pack(lightning.INIT_HELLO)
zmq_socket.send(message)
# Wait for HelloToClient()
print "..waiting for HelloToClient..."
hello_message = lightning.unpack(zmq_socket.recv())
assert(hello_message[0] == lightning.HELLO_TO_CLIENT)
session_id = hello_message[1].session_id
print "...got session_id {0}".format(session_id)
# Set socket identity to session_id
zmq_socket_in.setsockopt_unicode(zmq.IDENTITY, session_id)
# Find an open port to listen to, by testing.
listen_address = "tcp://*"
listen_port = zmq_socket_in.bind_to_random_port(listen_address, 23458, 24000, 500)
listen_address = "{0}:{1}".format("tcp://*", listen_port)
print("Listening to {0}".format(listen_address))
# HelloToTracker message
hello = tracker_pb2.HelloToTracker()
hello.address = listen_address
hello.available = available
hello.session_id = session_id
print("Sending new address information..")
zmq_socket.send(lightning.pack(lightning.HELLO_TO_TRACKER, hello))
ok = zmq_socket.recv()
if struct.unpack('<I', ok)[0] != lightning.OK:
print ok
break
zmq_socket.close()
print("Socket closed..")
elif i == 2:
print "Sending NewBatch..."
new = tracker_pb2.NewBatch()
new.session_id = session_id
new.batchhash = batchhash
new.tasks = tasks
new.batchtype = batchtype
print "protobuf data:\n" + str(new)
elif i == 3:
pass
elif i == 4:
print "Sending GoodBye..."
bye = tracker_pb2.GoodBye()
bye.session_id = session_id
elif i == 9:
print("Sending PING")
message = lightning.pack(lightning.PING)
