def test_sys_tools(self): SysTools.touch('/tmp/test_utils') SysTools.REBOOT_SKIP_FILES = ('/tmp/test_utils') self.assertFalse(SysTools.is_system_openwrt()) self.assertEqual(SysTools.get_mac_address('if_test'), "00:00:00:00:00:00") self.assertEqual(SysTools.get_sys_mac_address(), "00:00:00:00:00:00") with open('/tmp/test_mac', 'w') as f: f.write('01:02:03:04:05:06') f.close() SysTools.SYS_MAC_FILE = '/tmp/test_mac' self.assertEqual(SysTools.get_sys_mac_address(), "01:02:03:04:05:06") SysTools.SYS_MAC_FILE = '/etc/gshadow' self.assertEqual(SysTools.get_sys_mac_address(), "00:00:00:00:00:00") print(SysTools.get_host_name()) print(SysTools.get_ip_address('if_test')) if not SysTools.reboot(): SysTools.sys_failure_reboot('skip files exist') SysTools.external_reboot(('test', 'test')) if os.path.exists('/tmp/test_utils'): os.remove('/tmp/test_utils') # d = Dispatcher() # timestamp = time.time() + 1000 # SysTools.set_system_time(d, timestamp) # SysTools.touch('/etc/openwrt_release') # SysTools.set_system_time(d, timestamp) # time.sleep(2) # self.assertTrue(timestamp > time.time()) #should no permission to change system time # del d # if os.path.exists('/etc/openwrt_release'): # os.remove('/etc/openwrt_release') self.assertIsNotNone(SysTools.sys_up_time()) ret = SysTools.if_indextoname(1) print "ifname of index 1:", ret self.assertIsNotNone(ret) ret = SysTools.if_indextoname("test") self.assertIsNone(ret) ret = SysTools.if_indextoname(600) self.assertIsNone(ret) ret = SysTools.if_nametoindex("lo") print "ifindex of lo:", ret self.assertIsNotNone(ret) ret = SysTools.if_nametoindex(5) self.assertIsNone(ret) ret = SysTools.if_nametoindex("dafsd") self.assertIsNone(ret) ret = SysTools.is_if_oper_up('eth0') self.assertTrue(ret)
def gen_dhcp_mapping_info(self, interface): """Generate dhcp mapping info if not exist. :param interface: :return: """ rpd_ident = ['cfg', 'RpdCapabilities', 'RpdIdentification'] self.dhcp_args_mapping = { '0x02': rpd_ident + ['DeviceDescription'], '0x04': rpd_ident + ['SerialNumber'], '0x05': self.hw_version_path, '0x06': rpd_ident + ['CurrentSwVersion'], '0x07': rpd_ident + ['BootRomVersion'], '0x08': "".join(SysTools.get_mac_address(interface).split(':') [0:3]), # vendor ID '0x09': rpd_ident + ['ModelNumber'], '0x0A': rpd_ident + ['VendorName'] } # Fill device information to DB, if not loaded mac_addr_str = rpd_ident + ['DeviceMacAddress'] if self.db_adapter.get_leaf(mac_addr_str) is None: self.db_adapter.set_leaf(mac_addr_str, SysTools.get_mac_address(interface), True) hostname_str = rpd_ident + ['DeviceAlias'] if self.db_adapter.get_leaf(hostname_str) is None: self.db_adapter.set_leaf(hostname_str, SysTools.get_host_name(), True) if self.db_adapter.get_leaf(self.hw_version_path) is None: self.db_adapter.set_leaf(self.hw_version_path, self.hw_version.encode('hex'), True)
def __init__(self): self.dhcpv4_process = None self.dhcpv6_process = None self.sfd_stream = None self.db = RPD_DB() self.db_adapter = CfgDbAdapter(self.db) self.dhcp_data = t_DhcpData() self.disp = Dispatcher() self.signal_mask = self.create_signal_mask() self.dhcp_sock = None self.dhcp_timer = None self.processes = {} ProcessInfo.dispatcher = self.disp self.processes['dhcpv6'] = ProcessInfo( [ "odhcp6c", # Args - start "-s", "/lib/netifd/dhcpv6.script", "-P", "0", # Request IPv6 Prefix = auto "-t", "256", # Random backoff <1, 256> "-v", # Verbose "-I", self.DHCP_SOCK_ADDRESS ], # Args - end self.DHCP_SOCK_ADDRESS, # IPC address self.BACKOFF_TIMEOUT, # Timeout in seconds self._dhcp_no_lease # Timeout callback ) self.processes['dhcpv4'] = ProcessInfo( [ "udhcpc", # Args - start "-p", # Create PID file "/var/run/udhcpc-" + self.INTF + ".pid", "-f", "-t", "8", # Random backoff <1, 256> "-i", self.INTF, "-C", # Don't send MAC address as client-id "-B", # Enable broadcast "-S", # Enable logging to syslog "-n", # Exit if lease is not obtained "-I", self.DHCP_SOCK_ADDRESS ], # Args - end self.DHCP_SOCK_ADDRESS, # IPC address self.BACKOFF_TIMEOUT, # Timeout in seconds self._dhcp_no_lease # Timeout callback ) self.processes['rcp'] = ProcessInfo( [ "python", # Args - start "-m", "rpd.rcp.rcp_process", "--ipc-address", self.RCP_SOCK_ADDRESS ], # Args - end self.RCP_SOCK_ADDRESS # IPC address ) confFile = '/etc/config/hal.conf' self.processes['hal'] = ProcessInfo( ("python -m rpd.hal.src.HalMain --conf=" + confFile).split(" "), "") self.processes['tpc'] = ProcessInfo( [ "python", # Args - start "-m", "rpd.tpc", "--ipc-address", self.TPC_SOCK_ADDRESS ], # Args - end self.TPC_SOCK_ADDRESS, # IPC address self.BACKOFF_TIMEOUT, # Timeout in seconds self.reboot # Timeout callback ) self.processes['example'] = ProcessInfo( [ "python", # Args - start "-m", "rpd.example", "--ipc-address", self.EXAMPLE_PROC_SOCK_ADDRESS ], # Args - end self.EXAMPLE_PROC_SOCK_ADDRESS, # IPC address 60, # Timeout in seconds self.reboot # Timeout callback ) self.processes['example'].start(self.example_msg_cb) # Source of subTLV codes - section 6.4.1 of # http://www.cablelabs.com/wp-content/uploads/specdocs/CM-SP-R-PHY-I01_150615.pdf rpd_ident = ['cfg', 'RpdCapabilities', 'RpdIdentification'] self.dhcp_args_mapping = { '0x02': rpd_ident + ['DeviceDescription'], '0x04': rpd_ident + ['SerialNumber'], '0x05': Manager.hw_version_path, '0x06': rpd_ident + ['CurrentSwVersion'], '0x07': rpd_ident + ['BootRomVersion'], '0x08': "".join(SysTools.get_mac_address( self.INTF).split(':')[0:3]), # vendor ID '0x09': rpd_ident + ['ModelNumber'], '0x0A': rpd_ident + ['VendorName'] } # Fill device information to DB, if not loaded mac_addr_str = rpd_ident + ['DeviceMacAddress'] if self.db_adapter.get_leaf(mac_addr_str) is None: # TODO negative case handling self.db_adapter.set_leaf(mac_addr_str, SysTools.get_mac_address(self.INTF), True) hostname_str = rpd_ident + ['DeviceAlias'] if self.db_adapter.get_leaf(hostname_str) is None: # TODO negative case handling self.db_adapter.set_leaf(hostname_str, SysTools.get_host_name(), True) # TODO get from HW if self.db_adapter.get_leaf(Manager.hw_version_path) is None: # TODO negative case handling self.db_adapter.set_leaf(Manager.hw_version_path, Manager.hw_version, True) self.fsm = Fysom({ 'initial': { 'state': 'init' }, 'events': [{ 'name': 'init_done', 'src': 'init', 'dst': 'if_up_waiting' }, { 'name': 'if_is_up', 'src': 'if_up_waiting', 'dst': 'dhcpv6_waiting' }, { 'name': 'dhcpv6_failed', 'src': 'dhcpv6_waiting', 'dst': 'dhcpv4_waiting' }, { 'name': 'dhcp_ack', 'src': [ 'dhcpv6_waiting', 'dhcpv4_waiting', 'time_waiting', 'log_waiting', 'gcp_started' ], 'dst': 'time_waiting' }, { 'name': 'time_cfged', 'src': 'time_waiting', 'dst': 'log_waiting' }, { 'name': 'log_done', 'src': 'log_waiting', 'dst': 'gcp_started' }, { 'name': 'fatal_failure', 'src': [ 'if_up_waiting', 'dhcpv6_waiting', 'dhcpv4_waiting', 'time_waiting', 'gcp_started' ], 'dst': 'reboot' }], 'callbacks': { 'onchangestate': self._onchangestate, 'onif_is_up': self._on_iface_is_up, 'ondhcp_ack': self._ondhcp_ack, 'ontime_cfged': self._ontime_cfged, 'onlog_done': self._onlog_done, 'onfatal_failure': self.reboot, } })