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,
}
})
