def send(self, data, session_id=None):
"""send(str/int/protobuf data, int session_id=None) -> bool success.
Translates the data into protobuf-data with an enum at the beginning
(if it's not already in that format), and sends it to the client
with session id session_id. If no session_id is specified, the last
client that sent data will be the receiver.
Takes either an enum (int), a string (already in final format),
or a tuple with enum and protobuf.
Returns True on success, otherwise False.
"""
if session_id is None:
session_id = self._last_session_id
# Find the client.
client = None
for sck, info in self._scks.items():
if info['session_id'] == session_id:
client = sck
if client is None:
# Couldn't find the client.
print("unable to find client.")
return False
if type(data) == type(str()):
msg = data
elif type(data) == type(tuple()) and len(data) == 2:
msg = lightning.pack(data[0], data[1])
elif type(data) == type(int()):
msg = lightning.pack(data)
else:
# Unknown format.
print("unknown format.")
return False
# Add the length of the packet.
msg = "{0}{1}".format(struct.pack('<I', len(msg)), msg)
l = len(msg)
tries = 20
while msg != "":
msg = msg[client.send(msg):]
# To not end up in an infinite loop.
tries -= 1
if tries == 0:
raise LightningException("Tried too many times.")
print("successfully sent {1} bytes to socket {0}.".format(client, l))
return True
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)