def __init__(self, addr, pubkey):

self.cmds = {

"delete":delete.delete,

"checkin":checkin.checkin,

"message":message.message,

"get_messages":get_messages.get_messages,

"get_nodes":get_nodes.get_nodes,

}

self.broker = ("198.147.20.190", 4321)

self.addr = addr

self.pubkey = pubkey

self.port = 5333

self.host = "0.0.0.0"

self.open_port = False

self.config = {

"relay":False

}

def main(self):

if not db.nodes.find("nodes", "all"):

check = self.config['relay']

if check:

self.open_port = True

print "You are running as a relay node."

db.data.insert("data", {"port":self.port})

print "Downloading Nodes..."

self.get_nodes()

print "Checking in to nodes..."

self.send_checkin()

print "Done!"

check = self.config['relay']

if check:

self.open_port = True

print "You are running as a relay node."

else:

print "You are not running as a relay node."

if self.open_port:

sock = ssl.socket()

sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

sock.bind((self.host, self.port))

sock.listen(5)

while True:

obj, conn = sock.accept()

threading.Thread(target=self.handle, args=(obj, conn[0])).start()

else:

while True:

check = db.nodes.find("nodes", "all")

random.shuffle(check)

node = None

for x in check:

s = ssl.socket()

try:

s.connect((x['ip'], x['port']))

except:

s.close()

continue

else:

node = (x['ip'], x['port'])

s.close()

break

sock = ssl.socket()

try:

sock.connect(node)

except:

continue

else:

while True:

try:

get_messages.send_get_messages(x['ip'], x['port'])

get_nodes.send_get_nodes(x['ip'], x['port'])

time.sleep(10)

except Exception, error:

sock.close()

break

def handle(self, obj, ip):

try:

data = obj.recv(102400)

except:

obj.close()

return

if data:

try:

data = json.loads(data)

except ValueError:

obj.close()

else:

try:

self.cmds[data['cmd']](obj, ip, data)

except KeyError:

pass

def send_checkin(self):

nodes = db.nodes.find("nodes", "all")

for x in nodes:

s = ssl.socket()

try:

s.settimeout(1)

s.connect((x['ip'], x['port']))

s.send(json.dumps({"cmd":"checkin", "addr":self.addr, "port":self.port, "publickey":self.pubkey}))

s.close()

except Exception, error:

s.close()

db.unsent.insert("unsent", {"to":[x['ip'], x['port']], "message":{"cmd":"checkin", "addr":self.addr, "port":self.port, "publickey":self.pubkey}})

def get_nodes(self):

send = {"addr":self.addr, "port":self.port, "publickey":self.pubkey}

s = ssl.socket()

s.connect(self.broker)

s.send(json.dumps(send))

with open("nodes.db", 'wb') as file:

while True:

data = s.recv(1024)

if data:

file.write(data)

else:

break

prompt = "ByteMail$ "

def do_send(self, line):

self.lastcmd = ""

addr = db.data.find("data", "all")[0]['addr']

to = raw_input("To: ")

if to == "":

return "You must fill out the field."

if len(to) != 32:

if len(str(addressbook.check_entry(to))) != 32:

print "Address Invalid."

return

else:

to = str(addressbook.check_entry(to))

title = raw_input("Title: ")

msg = raw_input("Message: ")

if not to or not title or not msg:

print "You need to fill out all the fields."

else:

print message.send_msg(msg, title, to, addr)

def do_version(self, line):

print __version__

self.lastcmd = ""

def help_send(self):

print "Sends a message."

self.lastcmd = ""

def do_check(self, line):

addr = db.data.find("data", "all")[0]['addr']

check_ = check.check(addr)

if not check_:

print "You have no messages."

else:

for x in check_:

print x

self.lastcmd = ""

def help_check(self):

print "Displays a list of messages"

self.lastcmd = ""

def do_read(self, id):

addr = db.data.find("data", "all")[0]['addr']

print read.read(id, addr).decode("utf-8")

self.lastcmd = ""

def help_read(self):

print "Usage: read <id>"

print "Reads the specified message ID"

self.lastcmd = ""

def do_addr(self, line):

addr = db.data.find("data", "all")[0]['addr']

print "Your address is:", addr

self.lastcmd = ""

def help_addr(self):

print "Displays your ByteMail address"

self.lastcmd = ""

def do_add_address(self, line):

self.lastcmd = ""

name = raw_input("Name: ")

address = raw_input("Address: ")

addressbook.add_entry(name,address)

def help_addresses(self, line):

self.lastcmd = ""

print "Shows all addresses in your address book"

def do_addresses(self, line):

self.lastcmd = ""

print addressbook.addresses()

def help_add_address(self):

print "Adds an address to your address book"

def do_delete(self, line):

self.lastcmd = ""

addr = db.data.find("data", "all")[0]['addr']

print delete.send_delete(line, addr)

def help_delete(self):

self.lastcmd = ""

print "Usage: delete <id>"

print "Deletes the message with the specified ID"

def do_purge(self,line):

print "Purging..."

addr = db.data.find("data", "all")[0]['addr']

print delete.send_delete_all(addr)

def help_purge(self):

print "Deletes all messages sent to this node."

def do_exit(self, line):

print "Bye!"

exit()

def help_exit(self):

self.lastcmd = ""

print "Closes ByteMail."

def help_sent(self):

self.lastcmd = ""

print "Shows all sent messages."

def do_sent(self, line):

self.lastcmd = ""

print sent.sent()

def do_sent_delete(self, line):

self.lastcmd = ""

print sent.sent_delete(line)

def help_remove_address(self):

self.lastcmd = ""

print "Delete an address from your addressbook. Usage: remove_address <entryname>"

def do_remove_address(self, line):

self.lastcmd = ""

print addressbook.remove_address(line)

def do_sent_purge(self, line):

self.lastcmd = ""

sent_ = db.sent.find("sent", "all")

for x in sent_:

sent.sent_delete(x['id'])