You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

191 lines
5.2 KiB

11 years ago
#!/usr/bin/python
import socket, os, time, select, urllib, sys, threading
11 years ago
11 years ago
class Meshenger:
devices = {} #the dictionary of all the nodes this this node has seen
11 years ago
serve_port = 13338
announce_port = 13337
11 years ago
#own_ip = "0.0.0.0"
msg_dir = os.path.relpath('msg/')
exitapp = False #to kill all threads on
index_last_update = ''
11 years ago
11 years ago
def __init__(self):
11 years ago
#os.system("echo 1 >> /proc/sys/net/ipv6/conf/br-lan/disable_ipv6")
11 years ago
self.own_ip = self.get_ip_adress()
print 'Building own index for the first time\n'
self.build_index()
11 years ago
try:
d = threading.Thread(target=self.discover)
d.daemon = True
d.start()
a = threading.Thread(target=self.announce)
a.daemon = True
a.start()
except (KeyboardInterrupt, SystemExit):
print 'exiting discovery thread'
d.join()
a.join()
sys.exit()
while True:
11 years ago
11 years ago
if len(self.devices) > 0:
print 'found', len(self.devices),'device(s) retreiving indices'
11 years ago
for device in self.devices:
nodepath = self.ip_to_hash(device) #make a folder for the node (nodes/'hash'/)
nodeupdatepath = self.node_timestamp(device, nodepath) #contains the path to the update timestamp of the node (nodes/'hash'/lastupdate)
print 'Checking age of foreign node index'
if self.devices[device] > nodeupdatepath:
print 'Foreign node"s index is newer, proceed to download index'
self.get_index(device, nodepath)
print 'downloading messages'
self.get_messages(device, nodepath)
print 'updating own index'
self.build_index()
time.sleep(5) #free process or ctrl+c
def node_timestamp(self, ip, path):
updatepath = os.path.join(path, 'lastupdate')
with open(updatepath, 'wb') as lastupdate:
lastupdate.write(self.devices[ip])
return updatepath
11 years ago
11 years ago
def announce(self):
"""
Announce the node's existance to other nodes
"""
print 'Announcing'
while not self.exitapp:
sock = socket.socket(socket.AF_INET6, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
sock.sendto(self.index_last_update, ("ff02::1", self.announce_port))
sock.close()
time.sleep(5)
11 years ago
11 years ago
def discover(self):
"""
Discover other devices by listening to the Meshenger announce port
"""
11 years ago
print 'Discovering'
11 years ago
bufferSize = 1024 # whatever you need
11 years ago
11 years ago
s = socket.socket(socket.AF_INET6, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
s.bind(('::', self.announce_port))
s.setblocking(0)
11 years ago
while not self.exitapp:
11 years ago
result = select.select([s],[],[])[0][0].recvfrom(bufferSize)
if result[1][0] not in self.devices and result[1][0] != self.own_ip:
self.devices[result[1][0]] = result[0][0]
#self.devices.append(result[1][0])
11 years ago
time.sleep(1)
11 years ago
11 years ago
def serve(self):
11 years ago
# try:
# t = threading.Thread(target=BorderCheckWebserver, args=(self, ))
# t.daemon = True
# t.start()
# time.sleep(2)
# except (KeyboardInterrupt, SystemExit):
# t.join()
# sys.exit()
11 years ago
a = ''
# serves both the index and the messages on the node over http
# plus manages the client-side web interface
11 years ago
11 years ago
def build_index(self):
"""
Make an index file of all the messages present on the node.
Save the time of the last update.
"""
11 years ago
previous_index = []
11 years ago
current_index = os.listdir(self.msg_dir)
11 years ago
11 years ago
if current_index != previous_index:
with open('index', 'wb') as index:
for message in os.listdir(self.msg_dir):
11 years ago
index.write(message)
index.write('\n')
self.index_last_update = str(int(time.time()))
with open('index_last_update', 'wb') as indexupdate: ### misschien is dit overbodig
11 years ago
indexupdate.write(str(int(time.time())))
11 years ago
current_index = previous_index
11 years ago
11 years ago
def get_index(self,ip, path):
"""
Download the indices from other nodes.
"""
11 years ago
11 years ago
os.system('wget http://['+ip+'%adhoc0]:13338/index -O '+os.path.join(path,'index'))
11 years ago
11 years ago
def get_messages(self, ip, path):
"""
Get new messages from other node based on it's index file
"""
11 years ago
11 years ago
with open(os.path.join(path,'index')) as index:
index = index.read().split('\n')
for message in index:
messagepath = os.path.join(os.path.abspath(self.msg_dir), message)
11 years ago
if not os.path.exists(messagepath):
print 'downloading', message, 'to', messagepath
os.system('wget http://['+ip+'%adhoc0]:13338/msg/'+message+' -O '+messagepath)
11 years ago
11 years ago
def ip_to_hash(self, ip):
"""
Convert a node's ip into a hash and make a directory to store it's files
"""
11 years ago
import hashlib
hasj = hashlib.md5(ip).hexdigest()
nodepath = os.path.join(os.path.abspath('nodes/'), hasj)
if not os.path.exists(nodepath):
os.mkdir(nodepath)
11 years ago
11 years ago
return nodepath
11 years ago
11 years ago
def clientsite(self):
a = ''
11 years ago
11 years ago
#tools
11 years ago
11 years ago
def get_ip_adress(self):
"""
Hack to adhoc0's inet6 adress
"""
11 years ago
if not os.path.isfile('interfaceip6adress'):
os.system('ifconfig -a adhoc0 | grep inet6 > interfaceip6adress')
with open('interfaceip6adress', 'r') as a:
return a.read().split()[2].split('/')[0]
11 years ago
11 years ago
if __name__ == "__main__":
print "test"
try:
meshenger = Meshenger()
except (KeyboardInterrupt, SystemExit):
exitapp = True
raise