#!/usr/bin/env python2
import os
import sys
import configmanager
import json
import socket
from gi.repository import Gtk, GObject
import pair
from multiprocessing import Process, Queue
import threading
import thread
class EntryWindow():
def __init__(self):
builder = Gtk.Builder()
builder.add_from_file(os.getcwd()+"/share/ui/pairingwindow.glade")
builder.connect_signals(self)
self.window = builder.get_object("pairingwindow")
self.notebook = builder.get_object("notebook1")
self.cancelbutton = builder.get_object("cancelbutton")
self.forwardbutton = builder.get_object("forwardbutton")
self.openportslabel = builder.get_object("openportslabel")
self.hostlabel = builder.get_object("hostlabel")
self.desktopfplabel = builder.get_object("desktopfplabel")
self.devicefplabel = builder.get_object("devicefplabel")
self.succlabel = builder.get_object("succlabel")
self.faillabel = builder.get_object("faillabel")
pairport = str(configmanager.port)
openports = pairport + ", " + str(configmanager.secure_port)
lanip = socket.gethostname()
hosttxt = "Host: "+lanip+" or IP"+"\nPort: "+pairport
self.stage = 0
self.openportslabel.set_text(openports)
self.hostlabel.set_text(hosttxt)
self.window.set_wmclass ("DeskCon", "Setup Device")
self.window.show_all()
def on_cancelbutton_clicked(self, widget):
if (self.stage == 0):
Gtk.main_quit()
elif (self.stage == 2):
self.pthread.response(0)
self.succlabel.hide()
self.notebook.set_current_page(3)
self.stage = 3
else:
self.pthread.stop()
Gtk.main_quit()
def on_forwardbutton_clicked(self, widget):
if (self.stage == 0):
self.pthread = pairingThread(self)
self.pthread.daemon = True
self.pthread.start()
self.dthread = discoveryThread()
self.dthread.daemon = True
self.dthread.start()
self.notebook.set_current_page(1)
self.stage = 1
elif (self.stage == 2):
self.notebook.set_current_page(4)
self.pthread.response(1)
def on_pairingwindow_destroy(self, *args):
if ("self.pthread" in locals()):
self.pthread.stop()
Gtk.main_quit(*args)
class discoveryThread (threading.Thread):
def __init__(self):
threading.Thread.__init__(self)
def run(self):
UDP_IP = "0.0.0.0"
UDP_PORT = 5108
sock = socket.socket(socket.AF_INET, # Internet
socket.SOCK_DGRAM) # UDP
sock.bind((UDP_IP, UDP_PORT))
while (True):
data, addr = sock.recvfrom(1024) # buffer size is 1024 bytes
spl = data.split("::")
if (spl[0] == "0"):
print "received discovery broadcast"
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.bind(('', 0))
s.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
uuid = str(configmanager.uuid)
pairport = str(configmanager.port)
hostname = socket.gethostname()
data = uuid+"::"+hostname+"::"+pairport
s.sendto(data, (addr[0], UDP_PORT))
class pairingThread (threading.Thread):
def __init__(self, window):
threading.Thread.__init__(self)
self.window = window
self.q = Queue()
self.pairprocess = Process(target=pair.pair, args=(self.q,))
self.pairprocess.daemon = True
def run(self):
self.pairprocess.start()
fingerprints = self.q.get(True)
self.window.desktopfplabel.set_text(fingerprints[0])
self.window.devicefplabel.set_text(fingerprints[1])
self.window.notebook.set_current_page(2)
self.window.stage = 2
def response(self, r):
thread.start_new_thread(self.sendresponse, (r,))
def sendresponse(self, r):
if (r == 1): #fingerprints match
self.q.put("yes", True)
result = self.q.get(True)
if (result == 1):
self.window.faillabel.hide()
self.window.notebook.set_current_page(3)
self.window.stage = 3
else:
self.window.succlabel.hide()
self.window.notebook.set_current_page(3)
self.window.stage = 3
else: #fingerprints do not match
self.q.put("no", True)
def stop(self):
self.pairprocess.terminate()
def main(args):
GObject.threads_init()
win = EntryWindow()
Gtk.main()
if __name__ == '__main__':
main(sys.argv)
