def __init__(self, name, regtype, port=None):
self.name = name
self.regtype = regtype
self.port = port
self.browserQueried = []
self.browserResolved = []
self.browserLock = threading.Lock()
self.timeout = 5
self.registerStopEvent = threading.Event()
self.browserStopEvent = threading.Event()
self.clients = dict()
self.currentClient = Client()
self.clientEventHandler = EventHook()
def __init__(self,name,regtype,port=None):
self.name=name
self.regtype=regtype
self.port=port
self.browserQueried = []
self.browserResolved = []
self.browserLock=threading.Lock()
self.timeout = 5
self.registerStopEvent = threading.Event()
self.browserStopEvent = threading.Event()
self.clients = dict()
self.currentClient=Client()
self.clientEventHandler=EventHook()
class Bonjour():
def __init__(self, name, regtype, port=None):
self.name = name
self.regtype = regtype
self.port = port
self.browserQueried = []
self.browserResolved = []
self.browserLock = threading.Lock()
self.timeout = 5
self.registerStopEvent = threading.Event()
self.browserStopEvent = threading.Event()
self.clients = dict()
self.currentClient = Client()
self.clientEventHandler = EventHook()
def addClientEventHandler(self, fnc):
self.clientEventHandler += fnc
def removeClientEventHandler(self, fnc):
self.clientEventHandler -= fnc
def runRegister(self):
self.registerStopEvent.clear()
self.registerThread = threading.Thread(target=self.register)
self.registerThread.start()
def stopRegister(self):
self.registerStopEvent.set()
self.registerThread.join()
def runBrowser(self):
self.browserStopEvent.clear()
self.browserThread = threading.Thread(target=self.browser)
self.browserThread.start()
def stopBrowser(self):
self.browserStopEvent.set()
self.browserThread.join()
def register(self):
def register_callback(sdRef, flags, errorCode, name, regtype, domain):
if errorCode == pybonjour.kDNSServiceErr_NoError:
print 'Registered service:'
print ' name =', name
print ' regtype =', regtype
print ' domain =', domain
sdRef = pybonjour.DNSServiceRegister(name=self.name,
regtype=self.regtype,
port=self.port,
callBack=register_callback)
try:
while not self.registerStopEvent.is_set():
ready = select.select([sdRef], [], [], self.timeout * 2)
if sdRef in ready[0]:
pybonjour.DNSServiceProcessResult(sdRef)
# self.regStopEvent.wait(0.01)
finally:
sdRef.close()
print("exiting register thread")
def browser(self):
def query_record_callback(sdRef, flags, interfaceIndex, errorCode,
fullname, rrtype, rrclass, rdata, ttl):
if errorCode == pybonjour.kDNSServiceErr_NoError:
with self.browserLock:
self.currentClient.ip = socket.inet_ntoa(rdata)
self.currentClient.resolved = True
self.currentClient.fullName = fullname
self.browserQueried.append(True)
def resolve_callback(sdRef, flags, interfaceIndex, errorCode, fullname,
hosttarget, port, txtRecord):
if errorCode != pybonjour.kDNSServiceErr_NoError:
return
with self.browserLock:
self.currentClient.fullname = fullname
self.currentClient.port = port
self.currentClient.hostname = hosttarget.decode('utf-8')
query_sdRef = \
pybonjour.DNSServiceQueryRecord(interfaceIndex = interfaceIndex,
fullname = hosttarget,
rrtype = pybonjour.kDNSServiceType_A,
callBack = query_record_callback)
try:
while not self.browserQueried:
ready = select.select([query_sdRef], [], [], self.timeout)
if query_sdRef not in ready[0]:
print 'Query record timed out'
break
pybonjour.DNSServiceProcessResult(query_sdRef)
else:
self.browserQueried.pop()
finally:
query_sdRef.close()
self.browserResolved.append(True)
def browse_callback(sdRef, flags, interfaceIndex, errorCode,
serviceName, regtype, replyDomain):
if errorCode != pybonjour.kDNSServiceErr_NoError:
return
if not (flags & pybonjour.kDNSServiceFlagsAdd):
with self.browserLock:
if self.clients.has_key(serviceName):
print("client exists to be removed= " +
str(serviceName))
client = self.clients.get(serviceName)
self.clientEventHandler.fire(client.ip, client.port)
self.clients.pop(serviceName)
return
with self.browserLock:
self.currentClient = Client()
self.currentClient.serviceName = serviceName
resolve_sdRef = pybonjour.DNSServiceResolve(
0, interfaceIndex, serviceName, regtype, replyDomain,
resolve_callback)
try:
while not self.browserResolved:
ready = select.select([resolve_sdRef], [], [],
self.timeout * 2)
if resolve_sdRef not in ready[0]:
print 'Resolve timed out'
break
pybonjour.DNSServiceProcessResult(resolve_sdRef)
else:
with self.browserLock:
if not self.clients.has_key(
serviceName) and self.currentClient.resolved:
print("ading client=" + str(serviceName))
self.currentClient.regType = regtype
#print(self.currentClient)
self.clients[serviceName] = self.currentClient
self.clientEventHandler.fire(
self.currentClient.ip, self.currentClient.port)
self.browserResolved.pop()
finally:
resolve_sdRef.close()
browse_sdRef = pybonjour.DNSServiceBrowse(regtype=self.regtype,
callBack=browse_callback)
try:
while not self.browserStopEvent.is_set():
ready = select.select([browse_sdRef], [], [], self.timeout)
if browse_sdRef in ready[0]:
pybonjour.DNSServiceProcessResult(browse_sdRef)
finally:
browse_sdRef.close()
print("exiting browser thread")
def printClients(self):
with self.browserLock:
for client in self.clients.itervalues():
print(client)
def getFirstClient(self):
if len(self.clients) > 0:
return self.clients.get(self.clients.keys()[0])
return None
class Bonjour():
def __init__(self,name,regtype,port=None):
self.name=name
self.regtype=regtype
self.port=port
self.browserQueried = []
self.browserResolved = []
self.browserLock=threading.Lock()
self.timeout = 5
self.registerStopEvent = threading.Event()
self.browserStopEvent = threading.Event()
self.clients = dict()
self.currentClient=Client()
self.clientEventHandler=EventHook()
def addClientEventHandler(self,fnc):
self.clientEventHandler += fnc
def removeClientEventHandler(self,fnc):
self.clientEventHandler -= fnc
def runRegister(self):
self.registerStopEvent.clear()
self.registerThread=threading.Thread(target=self.register)
self.registerThread.start()
def stopRegister(self):
self.registerStopEvent.set()
self.registerThread.join()
def runBrowser(self):
self.browserStopEvent.clear()
self.browserThread=threading.Thread(target=self.browser)
self.browserThread.start()
def stopBrowser(self):
self.browserStopEvent.set()
self.browserThread.join()
def register(self):
def register_callback(sdRef, flags, errorCode, name, regtype, domain):
if errorCode == pybonjour.kDNSServiceErr_NoError:
print 'Registered service:'
print ' name =', name
print ' regtype =', regtype
print ' domain =', domain
sdRef = pybonjour.DNSServiceRegister(name = self.name,
regtype = self.regtype,
port = self.port,
callBack = register_callback)
try:
while not self.registerStopEvent.is_set():
ready = select.select([sdRef], [], [],self.timeout*2)
if sdRef in ready[0]:
pybonjour.DNSServiceProcessResult(sdRef)
# self.regStopEvent.wait(0.01)
finally:
sdRef.close()
print("exiting register thread")
def browser(self):
def query_record_callback(sdRef, flags, interfaceIndex, errorCode, fullname,
rrtype, rrclass, rdata, ttl):
if errorCode == pybonjour.kDNSServiceErr_NoError:
with self.browserLock:
self.currentClient.ip=socket.inet_ntoa(rdata)
self.currentClient.resolved=True
self.currentClient.fullName=fullname
self.browserQueried.append(True)
def resolve_callback(sdRef, flags, interfaceIndex, errorCode, fullname,
hosttarget, port, txtRecord):
if errorCode != pybonjour.kDNSServiceErr_NoError:
return
with self.browserLock:
self.currentClient.fullname=fullname
self.currentClient.port=port
self.currentClient.hostname=hosttarget.decode('utf-8')
query_sdRef = \
pybonjour.DNSServiceQueryRecord(interfaceIndex = interfaceIndex,
fullname = hosttarget,
rrtype = pybonjour.kDNSServiceType_A,
callBack = query_record_callback)
try:
while not self.browserQueried:
ready = select.select([query_sdRef], [], [], self.timeout)
if query_sdRef not in ready[0]:
print 'Query record timed out'
break
pybonjour.DNSServiceProcessResult(query_sdRef)
else:
self.browserQueried.pop()
finally:
query_sdRef.close()
self.browserResolved.append(True)
def browse_callback(sdRef, flags, interfaceIndex, errorCode, serviceName,
regtype, replyDomain):
if errorCode != pybonjour.kDNSServiceErr_NoError:
return
if not (flags & pybonjour.kDNSServiceFlagsAdd):
with self.browserLock:
if self.clients.has_key(serviceName):
print("client exists to be removed= "+str(serviceName))
client=self.clients.get(serviceName)
self.clientEventHandler.fire(client.ip, client.port)
self.clients.pop(serviceName)
return
with self.browserLock:
self.currentClient=Client()
self.currentClient.serviceName=serviceName
resolve_sdRef = pybonjour.DNSServiceResolve(0,
interfaceIndex,
serviceName,
regtype,
replyDomain,
resolve_callback)
try:
while not self.browserResolved:
ready = select.select([resolve_sdRef], [], [], self.timeout*2)
if resolve_sdRef not in ready[0]:
print 'Resolve timed out'
break
pybonjour.DNSServiceProcessResult(resolve_sdRef)
else:
with self.browserLock:
if not self.clients.has_key(serviceName) and self.currentClient.resolved:
print("ading client="+str(serviceName))
self.currentClient.regType=regtype
#print(self.currentClient)
self.clients[serviceName] = self.currentClient
self.clientEventHandler.fire(self.currentClient.ip,self.currentClient.port)
self.browserResolved.pop()
finally:
resolve_sdRef.close()
browse_sdRef = pybonjour.DNSServiceBrowse(regtype = self.regtype,
callBack = browse_callback)
try:
while not self.browserStopEvent.is_set():
ready = select.select([browse_sdRef], [], [],self.timeout)
if browse_sdRef in ready[0]:
pybonjour.DNSServiceProcessResult(browse_sdRef)
finally:
browse_sdRef.close()
print("exiting browser thread")
def printClients(self):
with self.browserLock:
for client in self.clients.itervalues():
print(client)
def getFirstClient(self):
if len(self.clients)>0:
return self.clients.get(self.clients.keys()[0])
return None
