import avahi import sys, os sys.path.insert(0, os.path.abspath("../../")) from sugar import util import dbus, dbus.service import random import logging import gobject def compose_service_name(name, activity_id): if isinstance(name, str): name = unicode(name) if not name: raise ValueError("name must be a valid string.") if not activity_id: return name if not isinstance(name, unicode): raise ValueError("name must be in unicode.") composed = "%s [%s]" % (name, activity_id) return composed def decompose_service_name(name): """Break a service name into the name and activity ID, if we can.""" if not isinstance(name, unicode): raise ValueError("name must be a valid unicode string.") name_len = len(name) if name_len < util.ACTIVITY_ID_LEN + 5: return (None, name) # check for activity id end marker if name[name_len - 1] != "]": return (None, name) start = name_len - 1 - util.ACTIVITY_ID_LEN end = name_len - 1 # check for activity id start marker if name[start - 1] != "[" or name[start - 2] != " ": return (None, name) activity_id = name[start:end] if not util.validate_activity_id(activity_id): return (None, name) return (activity_id, name[:start - 2]) def _one_dict_differs(dict1, dict2): diff_keys = [] for key, value in dict1.items(): if not dict2.has_key(key) or dict2[key] != value: diff_keys.append(key) return diff_keys def _dicts_differ(dict1, dict2): diff_keys = [] diff1 = _one_dict_differs(dict1, dict2) diff2 = _one_dict_differs(dict2, dict1) for key in diff2: if key not in diff1: diff_keys.append(key) diff_keys += diff1 return diff_keys def _convert_properties_to_dbus_byte_array(props): # Ensure properties are converted to ByteArray types # because python sometimes can't figure that out info = dbus.Array([], signature="aay") for k, v in props.items(): info.append(dbus.types.ByteArray("%s=%s" % (k, v))) return info _ACTIVITY_ID_TAG = "ActivityID" SERVICE_DBUS_INTERFACE = "org.laptop.Presence.Service" SERVICE_DBUS_OBJECT_PATH = "/org/laptop/Presence/Services/" class ServiceDBusHelper(dbus.service.Object): """Handle dbus requests and signals for Service objects""" def __init__(self, parent, bus_name, object_path): self._parent = parent self._bus_name = bus_name self._object_path = object_path dbus.service.Object.__init__(self, bus_name, self._object_path) @dbus.service.signal(SERVICE_DBUS_INTERFACE, signature="as") def PublishedValueChanged(self, keylist): pass @dbus.service.method(SERVICE_DBUS_INTERFACE, in_signature="", out_signature="a{sv}") def getProperties(self): """Return service properties.""" pary = {} pary['name'] = self._parent.get_name() pary['type'] = self._parent.get_type() pary['domain'] = self._parent.get_domain() actid = self._parent.get_activity_id() if actid: pary['activityId'] = actid port = self._parent.get_port() if port: pary['port'] = self._parent.get_port() addr = self._parent.get_address() if addr: pary['address'] = addr source_addr = self._parent.get_source_address() if source_addr: pary['sourceAddress'] = source_addr return pary @dbus.service.method(SERVICE_DBUS_INTERFACE, in_signature="s") def getPublishedValue(self, key): """Return the value belonging to the requested key from the service's TXT records.""" val = self._parent.get_one_property(key) if not val: raise KeyError("Value was not found.") return val @dbus.service.method(SERVICE_DBUS_INTERFACE, in_signature="", out_signature="a{sv}") def getPublishedValues(self): pary = {} props = self._parent.get_properties() for key, value in props.items(): pary[key] = str(value) return dbus.Dictionary(pary) @dbus.service.method(SERVICE_DBUS_INTERFACE, sender_keyword="sender") def setPublishedValue(self, key, value, sender): self._parent.set_property(key, value, sender) @dbus.service.method(SERVICE_DBUS_INTERFACE, in_signature="a{sv}", sender_keyword="sender") def setPublishedValues(self, values, sender): if not self._parent.is_local(): raise ValueError("Service was not not registered by requesting process!") self._parent.set_properties(values, sender) class Service(gobject.GObject): """Encapsulates information about a specific ZeroConf/mDNS service as advertised on the network.""" __gsignals__ = { 'property-changed': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])) } def __init__(self, bus_name, object_id, name, stype, domain=u"local", address=None, port=-1, properties=None, source_address=None, local_publisher=None): gobject.GObject.__init__(self) if not bus_name: raise ValueError("DBus bus name must be valid") if not object_id or not isinstance(object_id, int): raise ValueError("object id must be a valid number") # Validate immutable options if name and not isinstance(name, unicode): raise ValueError("name must be unicode.") if not name or not len(name): raise ValueError("must specify a valid service name.") if stype and not isinstance(stype, unicode): raise ValueError("service type must be in unicode.") if not stype or not len(stype): raise ValueError("must specify a valid service type.") if not stype.endswith("._tcp") and not stype.endswith("._udp"): raise ValueError("must specify a TCP or UDP service type.") if not isinstance(domain, unicode): raise ValueError("domain must be in unicode.") if domain and domain != "local": raise ValueError("must use the 'local' domain (for now).") # ID of the D-Bus connection that published this service, if any. # We only let the local publisher modify the service. self._local_publisher = local_publisher self._avahi_entry_group = None (actid, real_name) = decompose_service_name(name) self._name = real_name self._full_name = name self._stype = stype self._domain = domain self._port = -1 self.set_port(port) self._properties = {} self._dbus_helper = None self._internal_set_properties(properties) # Source address is the unicast source IP self._source_address = None if source_address is not None: self.set_source_address(source_address) # Address is the published address, could be multicast or unicast self._address = None if self._properties.has_key('address'): self.set_address(self._properties['address']) elif address is not None: self.set_address(address) self._properties['address'] = address # Ensure that an ActivityID tag, if given, matches # what we expect from the service type if self._properties.has_key(_ACTIVITY_ID_TAG): prop_actid = self._properties[_ACTIVITY_ID_TAG] if (prop_actid and not actid) or (prop_actid != actid): raise ValueError("ActivityID property specified, but the " \ "service names's activity ID didn't match it: %s," \ " %s" % (prop_actid, actid)) self._activity_id = actid if actid and not self._properties.has_key(_ACTIVITY_ID_TAG): self._properties[_ACTIVITY_ID_TAG] = actid self._owner = None # register ourselves with dbus self._object_id = object_id self._object_path = SERVICE_DBUS_OBJECT_PATH + str(self._object_id) self._dbus_helper = ServiceDBusHelper(self, bus_name, self._object_path) def object_path(self): return dbus.ObjectPath(self._object_path) def get_owner(self): return self._owner def set_owner(self, owner): if self._owner is not None: raise RuntimeError("Can only set a service's owner once") self._owner = owner def is_local(self): if self._local_publisher is not None: return True return False def get_name(self): """Return the service's name, usually that of the buddy who provides it.""" return self._name def get_full_name(self): return self._full_name def get_one_property(self, key): """Return one property of the service, or None if the property was not found. Cannot distinguish between lack of a property, and a property value that actually is None.""" if key in self._properties.keys(): return self._properties[key] return None def get_properties(self): """Return a python dictionary of all the service's properties.""" return self._properties def __emit_properties_changed_signal(self, keys): if self._dbus_helper: self._dbus_helper.PublishedValueChanged(keys) self.emit('property-changed', keys) def set_property(self, key, value, sender=None): """Set one service property""" if not self._local_publisher: raise ValueError("Service was not not registered by requesting process!") if sender is not None and self._local_publisher != sender: raise ValueError("Service was not not registered by requesting process!") if not isinstance(key, unicode): raise ValueError("Key must be a unicode string.") if not isinstance(value, unicode) and not isinstance(value, bool): raise ValueError("Key must be a unicode string or a boolean.") # Ignore setting the key to it's current value if self._properties.has_key(key): if self._properties[key] == value: return # Blank value means remove key remove = False if isinstance(value, unicode) and len(value) == 0: remove = True if isinstance(value, bool) and value == False: remove = True if remove: # If the key wasn't present, return without error if self._properties.has_key(key): del self._properties[key] else: # Otherwise set it if isinstance(value, bool): value = "" self._properties[key] = value # if the service is locally published already, update the TXT records if self._local_publisher and self._avahi_entry_group: self.__internal_update_avahi_properties() self.__emit_properties_changed_signal([key]) def set_properties(self, properties, sender=None, from_network=False): """Set all service properties in one call""" if sender is not None and self._local_publisher != sender: raise ValueError("Service was not not registered by requesting process!") self._internal_set_properties(properties, from_network) def _internal_set_properties(self, properties, from_network=False): """Set the service's properties from either an Avahi TXT record (a list of lists of integers), or a python dictionary.""" if not isinstance (properties, dict): raise ValueError("Properties must be a dictionary.") # Make sure the properties are actually different diff_keys = _dicts_differ(self._properties, properties) if len(diff_keys) == 0: return self._properties = {} # Set key/value pairs on internal property list for key, value in properties.items(): if len(key) == 0: continue tmp_key = key tmp_val = value if not isinstance(tmp_key, unicode): tmp_key = unicode(tmp_key) if not isinstance(tmp_val, unicode): tmp_val = unicode(tmp_val) self._properties[tmp_key] = tmp_val # if the service is locally published already, update the TXT records if self._local_publisher and self._avahi_entry_group and not from_network: self.__internal_update_avahi_properties() self.__emit_properties_changed_signal(diff_keys) def __internal_update_avahi_properties(self): info = _convert_properties_to_dbus_byte_array(self._properties) self._avahi_entry_group.UpdateServiceTxt(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC, 0, dbus.String(self._full_name), dbus.String(self._stype), dbus.String(self._domain), info) def get_type(self): """Return the service's service type.""" return self._stype def get_activity_id(self): """Return the activity ID this service is associated with, if any.""" return self._activity_id def get_port(self): return self._port def set_port(self, port): if not isinstance(port, int) or (port <= 1024 and port > 65536): raise ValueError("must specify a valid port number between 1024 and 65536.") self._port = port def get_source_address(self): return self._source_address def set_source_address(self, address): if not address or not isinstance(address, unicode): raise ValueError("address must be unicode") self._source_address = address def get_address(self): return self._address def set_address(self, address): if not address or not isinstance(address, unicode): raise ValueError("address must be a unicode string") self._address = address self._properties['address'] = address def get_domain(self): """Return the ZeroConf/mDNS domain the service was found in.""" return self._domain def register(self, system_bus, avahi_service): if self._avahi_entry_group is not None: raise RuntimeError("Service already registered!") obj = system_bus.get_object(avahi.DBUS_NAME, avahi_service.EntryGroupNew()) self._avahi_entry_group = dbus.Interface(obj, avahi.DBUS_INTERFACE_ENTRY_GROUP) info = _convert_properties_to_dbus_byte_array(self._properties) logging.debug("Will register service with name='%s', stype='%s'," \ " domain='%s', address='%s', port=%d, info='%s'" % (self._full_name, self._stype, self._domain, self._address, self._port, info)) self._avahi_entry_group.AddService(avahi.IF_UNSPEC, avahi.PROTO_UNSPEC, 0, dbus.String(self._full_name), dbus.String(self._stype), dbus.String(self._domain), dbus.String(""), # let Avahi figure the 'host' out dbus.UInt16(self._port), info) self._avahi_entry_group.connect_to_signal('StateChanged', self.__entry_group_changed_cb) self._avahi_entry_group.Commit() def __entry_group_changed_cb(self, state, error): pass # logging.debug("** %s.%s Entry group changed: state %s, error %s" % (self._full_name, self._stype, state, error)) def unregister(self, sender): # Refuse to unregister if we can't get the dbus connection this request # came from for some reason if not sender: raise RuntimeError("Service registration request must have a sender.") if not self._local_publisher: raise ValueError("Service was not a local service provided by this laptop!") if sender is not None and self._local_publisher != sender: raise ValueError("Service was not registered by requesting process!") if not self._avahi_entry_group: raise ValueError("Service was not registered by requesting process!") self._avahi_entry_group.Free() del self._avahi_entry_group self._avahi_entry_group = None ################################################################# # Tests ################################################################# import unittest __objid_seq = 0 def _next_objid(): global __objid_seq __objid_seq = __objid_seq + 1 return __objid_seq class ServiceTestCase(unittest.TestCase): _DEF_NAME = u"foobar" _DEF_STYPE = u"_foo._bar._tcp" _DEF_DOMAIN = u"local" _DEF_ADDRESS = u"1.1.1.1" _DEF_PORT = 1234 _DEF_PROPS = {'foobar': 'baz'} _STR_TEST_ARGS = [None, 0, [], {}] def __init__(self, name): self._bus = dbus.SessionBus() self._bus_name = dbus.service.BusName('org.laptop.Presence', bus=self._bus) unittest.TestCase.__init__(self, name) def __del__(self): del self._bus_name del self._bus def _test_init_fail(self, name, stype, domain, address, port, properties, fail_msg): """Test something we expect to fail.""" try: objid = _next_objid() service = Service(self._bus_name, objid, name, stype, domain, address, port, properties) except ValueError, exc: pass else: self.fail("expected a ValueError for %s." % fail_msg) def testName(self): for item in self._STR_TEST_ARGS: self._test_init_fail(item, self._DEF_STYPE, self._DEF_DOMAIN, self._DEF_ADDRESS, self._DEF_PORT, self._DEF_PROPS, "invalid name") def testType(self): for item in self._STR_TEST_ARGS: self._test_init_fail(self._DEF_NAME, item, self._DEF_DOMAIN, self._DEF_ADDRESS, self._DEF_PORT, self._DEF_PROPS, "invalid service type") self._test_init_fail(self._DEF_NAME, u"_bork._foobar", self._DEF_DOMAIN, self._DEF_ADDRESS, self._DEF_PORT, self._DEF_PROPS, "invalid service type") def testDomain(self): for item in self._STR_TEST_ARGS: self._test_init_fail(self._DEF_NAME, self._DEF_STYPE, item, self._DEF_ADDRESS, self._DEF_PORT, self._DEF_PROPS, "invalid domain") # Only accept local for now self._test_init_fail(self._DEF_NAME, self._DEF_STYPE, u"foobar", self._DEF_ADDRESS, self._DEF_PORT, self._DEF_PROPS, "invalid domain") # Make sure "" works objid = _next_objid() service = Service(self._bus_name, objid, self._DEF_NAME, self._DEF_STYPE, u"", self._DEF_ADDRESS, self._DEF_PORT, self._DEF_PROPS) assert service, "Empty domain was not accepted!" def testAddress(self): self._test_init_fail(self._DEF_NAME, self._DEF_STYPE, self._DEF_DOMAIN, [], self._DEF_PORT, self._DEF_PROPS, "invalid address") self._test_init_fail(self._DEF_NAME, self._DEF_STYPE, self._DEF_DOMAIN, {}, self._DEF_PORT, self._DEF_PROPS, "invalid address") self._test_init_fail(self._DEF_NAME, self._DEF_STYPE, self._DEF_DOMAIN, 1234, self._DEF_PORT, self._DEF_PROPS, "invalid address") def testPort(self): self._test_init_fail(self._DEF_NAME, self._DEF_STYPE, self._DEF_DOMAIN, self._DEF_ADDRESS, [], self._DEF_PROPS, "invalid port") self._test_init_fail(self._DEF_NAME, self._DEF_STYPE, self._DEF_DOMAIN, self._DEF_ADDRESS, {}, self._DEF_PROPS, "invalid port") self._test_init_fail(self._DEF_NAME, self._DEF_STYPE, self._DEF_DOMAIN, self._DEF_ADDRESS, "adf", self._DEF_PROPS, "invalid port") def testGoodInit(self): objid = _next_objid() service = Service(self._bus_name, objid, self._DEF_NAME, self._DEF_STYPE, self._DEF_DOMAIN, self._DEF_ADDRESS, self._DEF_PORT, self._DEF_PROPS) assert service.get_name() == self._DEF_NAME, "service name wasn't correct after init." assert service.get_type() == self._DEF_STYPE, "service type wasn't correct after init." assert service.get_domain() == "local", "service domain wasn't correct after init." assert service.get_address() == self._DEF_ADDRESS, "service address wasn't correct after init." assert service.get_port() == self._DEF_PORT, "service port wasn't correct after init." assert service.object_path() == SERVICE_DBUS_OBJECT_PATH + str(objid) value = service.get_one_property('foobar') assert value and value == 'baz', "service property wasn't correct after init." def testAvahiProperties(self): props = [[111, 114, 103, 46, 102, 114, 101, 101, 100, 101, 115, 107, 116, 111, 112, 46, 65, 118, 97, 104, 105, 46, 99, 111, 111, 107, 105, 101, 61, 50, 54, 48, 49, 53, 52, 51, 57, 53, 50]] key = "org.freedesktop.Avahi.cookie" expected_value = "2601543952" objid = _next_objid() service = Service(self._bus_name, objid, self._DEF_NAME, self._DEF_STYPE, self._DEF_DOMAIN, self._DEF_ADDRESS, self._DEF_PORT, props) value = service.get_one_property(key) assert value and value == expected_value, "service properties weren't correct after init." value = service.get_one_property('bork') assert not value, "service properties weren't correct after init." def testBoolProperty(self): props = [[111, 114, 103, 46, 102, 114, 101, 101, 100, 101, 115, 107, 116, 111, 112, 46, 65, 118, 97, 104, 105, 46, 99, 111, 111, 107, 105, 101]] key = "org.freedesktop.Avahi.cookie" expected_value = True objid = _next_objid() service = Service(self._bus_name, objid, self._DEF_NAME, self._DEF_STYPE, self._DEF_DOMAIN, self._DEF_ADDRESS, self._DEF_PORT, props) value = service.get_one_property(key) assert value is not None and value == expected_value, "service properties weren't correct after init." def testActivityService(self): # Valid group service type, non-multicast address actid = "4569a71b80805aa96a847f7ac1c407327b3ec2b4" name = compose_service_name("Tommy", actid) # Valid activity service name, None address objid = _next_objid() service = Service(self._bus_name, objid, name, self._DEF_STYPE, self._DEF_DOMAIN, None, self._DEF_PORT, self._DEF_PROPS) assert service.get_address() == None, "address was not None as expected!" assert service.get_activity_id() == actid, "activity id was different than expected!" # Valid activity service name and multicast address, ensure it works mc_addr = u"224.0.0.34" objid = _next_objid() service = Service(self._bus_name, objid, name, self._DEF_STYPE, self._DEF_DOMAIN, mc_addr, self._DEF_PORT, self._DEF_PROPS) assert service.get_address() == mc_addr, "address was not expected address!" def addToSuite(suite): suite.addTest(ServiceTestCase("testName")) suite.addTest(ServiceTestCase("testType")) suite.addTest(ServiceTestCase("testDomain")) suite.addTest(ServiceTestCase("testAddress")) suite.addTest(ServiceTestCase("testPort")) suite.addTest(ServiceTestCase("testGoodInit")) suite.addTest(ServiceTestCase("testAvahiProperties")) suite.addTest(ServiceTestCase("testBoolProperty")) suite.addTest(ServiceTestCase("testActivityService")) addToSuite = staticmethod(addToSuite) def main(): suite = unittest.TestSuite() ServiceTestCase.addToSuite(suite) runner = unittest.TextTestRunner() runner.run(suite) if __name__ == "__main__": main()