[pull] master from Azure:master

device/mellanox/x86_64-mlnx_msn2700-r0/plugins/sfplpmget.py

@@ -1,43 +1,51 @@
 #!/usr/bin/env python
+"""
+This utility get the power mode of a given module.
+"""
 
-import sys, errno
-import os
-from python_sdk_api.sxd_api import *
+import sys
+import errno
 from python_sdk_api.sx_api import *
 
+def mgmt_phy_mod_pwr_attr_get(handle, module_id, power_attr_type):
+    sx_mgmt_phy_mod_pwr_attr_p = new_sx_mgmt_phy_mod_pwr_attr_t_p()
+    sx_mgmt_phy_mod_pwr_attr = sx_mgmt_phy_mod_pwr_attr_t()
+    sx_mgmt_phy_mod_pwr_attr.power_attr_type = power_attr_type
+    sx_mgmt_phy_mod_pwr_attr_t_p_assign(sx_mgmt_phy_mod_pwr_attr_p, sx_mgmt_phy_mod_pwr_attr)
+    try:
+        rc = sx_mgmt_phy_mod_pwr_attr_get(handle, module_id, sx_mgmt_phy_mod_pwr_attr_p)
+        assert SX_STATUS_SUCCESS == rc, "sx_mgmt_phy_mod_pwr_attr_get failed"
+        sx_mgmt_phy_mod_pwr_attr = sx_mgmt_phy_mod_pwr_attr_t_p_value(sx_mgmt_phy_mod_pwr_attr_p)
+        pwr_mode_attr = sx_mgmt_phy_mod_pwr_attr.pwr_mode_attr
+        return pwr_mode_attr.admin_pwr_mode_e, pwr_mode_attr.oper_pwr_mode_e
+    finally:
+        delete_sx_mgmt_phy_mod_pwr_attr_t_p(sx_mgmt_phy_mod_pwr_attr_p)
+
 # Check if SFP port number is provided
 if len(sys.argv) < 2:
-    print "SFP module number is missed."
-    print "Usage: sfplpmget.py <SFP module>"
+    print("SFP module number is missed.")
+    print("Usage: sfplpmget.py <SFP module>")
     sys.exit(errno.EINVAL)
 
 # Init SDK API
 rc, handle = sx_api_open(None)
 if (rc != SX_STATUS_SUCCESS):
-    print "Failed to open api handle.\nPlease check that SDK is running."
-    sys.exit(errno.EACCES)
-
-pid = os.getpid()
-rc = sxd_access_reg_init(pid, None, 0)
-if (rc != 0):
-    print "Failed to initializing register access.\nPlease check that SDK is running."
+    print("Failed to open api handle.\nPlease check that SDK is running.")
     sys.exit(errno.EACCES)
 
 # Get SFP module number
 sfp_module = int(sys.argv[1]) - 1
 
-# Get MCION
-mcion = ku_mcion_reg()
-mcion.module = sfp_module
-meta = sxd_reg_meta_t()
-meta.dev_id = 1
-meta.swid = 0
-meta.access_cmd = SXD_ACCESS_CMD_GET
+admin_pwr_mode, oper_pwr_mode = mgmt_phy_mod_pwr_attr_get(handle, sfp_module, SX_MGMT_PHY_MOD_PWR_ATTR_PWR_MODE_E)
 
-rc = sxd_access_reg_mcion(mcion, meta, 1, None, None)
-assert rc == SXD_STATUS_SUCCESS, "sxd_access_reg_mcion failed, rc = %d" % rc
+lpm_status = None
+if oper_pwr_mode == SX_MGMT_PHY_MOD_PWR_MODE_HIGH_E:
+    lpm_status = False
+elif oper_pwr_mode == SX_MGMT_PHY_MOD_PWR_MODE_LOW_E:
+    lpm_status = True
+else:
+    print("LPM UNKNOWN")
 
-# Get low power mode status
-lpm_mask = 1 << 8
-lpm_status = (lpm_mask & mcion.module_status_bits) != 0
 print "LPM ON" if lpm_status else "LPM OFF"
+
+sx_api_close(handle)

device/mellanox/x86_64-mlnx_msn2700-r0/plugins/sfplpmset.py

@@ -1,23 +1,17 @@
 #!/usr/bin/env python
+"""
+This utility set the power mode of a given module.
+"""
 
-import sys, errno
-import os
-from python_sdk_api.sxd_api import *
+import sys
+import errno
 from python_sdk_api.sx_api import *
 
-REGISTER_NUM = 1
-SXD_LOG_VERBOSITY_LEVEL = 0
+
 DEVICE_ID = 1
 SWITCH_ID = 0
 SX_PORT_ATTR_ARR_SIZE = 64
 
-PMAOS_ASE = 1
-PMAOS_EE = 1
-PMAOS_E = 2
-PMAOS_RST = 0
-PMAOS_ENABLE = 1
-PMAOS_DISABLE = 2
-
 PORT_TYPE_CPU = 4
 PORT_TYPE_NVE = 8
 PORT_TYPE_OFFSET = 28
@@ -36,7 +30,7 @@ def is_port_admin_status_up(log_port):
     admin_state_p = new_sx_port_admin_state_t_p()
     module_state_p = new_sx_port_module_state_t_p()
     rc = sx_api_port_state_get(handle, log_port, oper_state_p, admin_state_p, module_state_p)
-    assert rc == SXD_STATUS_SUCCESS, "sx_api_port_state_get failed, rc = %d" % rc
+    assert rc == SX_STATUS_SUCCESS, "sx_api_port_state_get failed, rc = %d" % rc
 
     admin_state = sx_port_admin_state_t_p_value(admin_state_p)
     if admin_state == SX_PORT_ADMIN_STATUS_UP:
@@ -54,116 +48,109 @@ def get_log_ports(handle, sfp_module):
     port_cnt_p = new_uint32_t_p()
     uint32_t_p_assign(port_cnt_p, SX_PORT_ATTR_ARR_SIZE)
 
-    rc = sx_api_port_device_get(handle, DEVICE_ID , SWITCH_ID, port_attributes_list,  port_cnt_p)
+    rc = sx_api_port_device_get(handle, DEVICE_ID, SWITCH_ID, port_attributes_list,  port_cnt_p)
     assert rc == SX_STATUS_SUCCESS, "sx_api_port_device_get failed, rc = %d" % rc
 
     port_cnt = uint32_t_p_value(port_cnt_p)
     log_port_list = []
     for i in range(0, port_cnt):
         port_attributes = sx_port_attributes_t_arr_getitem(port_attributes_list, i)
-        if is_nve(int(port_attributes.log_port)) == False \
-           and is_cpu(int(port_attributes.log_port)) == False \
+        if not is_nve(int(port_attributes.log_port)) \
+           and not is_cpu(int(port_attributes.log_port)) \
            and port_attributes.port_mapping.module_port == sfp_module \
            and is_port_admin_status_up(port_attributes.log_port):
             log_port_list.append(port_attributes.log_port)
 
     return log_port_list
 
-def init_sx_meta_data():
-    meta = sxd_reg_meta_t()
-    meta.dev_id = DEVICE_ID
-    meta.swid = SWITCH_ID
-    return meta
-
-def set_sfp_admin_status(sfp_module, admin_status):
-    # Get PMAOS
-    pmaos = ku_pmaos_reg()
-    pmaos.module = sfp_module
-    meta = init_sx_meta_data()
-    meta.access_cmd = SXD_ACCESS_CMD_GET
-    rc = sxd_access_reg_pmaos(pmaos, meta, REGISTER_NUM, None, None)
-    assert rc == SXD_STATUS_SUCCESS, "sxd_access_reg_pmaos failed, rc = %d" % rc
-
-    # Set admin status to PMAOS
-    pmaos.ase = PMAOS_ASE
-    pmaos.ee = PMAOS_EE
-    pmaos.e = PMAOS_E
-    pmaos.rst = PMAOS_RST
-    if admin_status == SX_PORT_ADMIN_STATUS_DOWN:
-        pmaos.admin_status = PMAOS_DISABLE
+def mgmt_phy_mod_pwr_attr_set(handle, module_id, power_attr_type, admin_pwr_mode):
+    sx_mgmt_phy_mod_pwr_attr = sx_mgmt_phy_mod_pwr_attr_t()
+    sx_mgmt_phy_mod_pwr_mode_attr = sx_mgmt_phy_mod_pwr_mode_attr_t()
+    sx_mgmt_phy_mod_pwr_attr.power_attr_type = power_attr_type
+    sx_mgmt_phy_mod_pwr_mode_attr.admin_pwr_mode_e = admin_pwr_mode
+    sx_mgmt_phy_mod_pwr_attr.pwr_mode_attr = sx_mgmt_phy_mod_pwr_mode_attr
+    sx_mgmt_phy_mod_pwr_attr_p = new_sx_mgmt_phy_mod_pwr_attr_t_p()
+    sx_mgmt_phy_mod_pwr_attr_t_p_assign(sx_mgmt_phy_mod_pwr_attr_p, sx_mgmt_phy_mod_pwr_attr)
+    try:
+        rc = sx_mgmt_phy_mod_pwr_attr_set(handle, SX_ACCESS_CMD_SET, module_id, sx_mgmt_phy_mod_pwr_attr_p)
+        assert SX_STATUS_SUCCESS == rc, "sx_mgmt_phy_mod_pwr_attr_set failed"
+    finally:
+        delete_sx_mgmt_phy_mod_pwr_attr_t_p(sx_mgmt_phy_mod_pwr_attr_p)
+
+def mgmt_phy_mod_pwr_attr_get(handle, module_id, power_attr_type):
+    sx_mgmt_phy_mod_pwr_attr_p = new_sx_mgmt_phy_mod_pwr_attr_t_p()
+    sx_mgmt_phy_mod_pwr_attr = sx_mgmt_phy_mod_pwr_attr_t()
+    sx_mgmt_phy_mod_pwr_attr.power_attr_type = power_attr_type
+    sx_mgmt_phy_mod_pwr_attr_t_p_assign(sx_mgmt_phy_mod_pwr_attr_p, sx_mgmt_phy_mod_pwr_attr)
+    try:
+        rc = sx_mgmt_phy_mod_pwr_attr_get(handle, module_id, sx_mgmt_phy_mod_pwr_attr_p)
+        assert SX_STATUS_SUCCESS == rc, "sx_mgmt_phy_mod_pwr_attr_get failed"
+        sx_mgmt_phy_mod_pwr_attr = sx_mgmt_phy_mod_pwr_attr_t_p_value(sx_mgmt_phy_mod_pwr_attr_p)
+        pwr_mode_attr = sx_mgmt_phy_mod_pwr_attr.pwr_mode_attr
+        return pwr_mode_attr.admin_pwr_mode_e, pwr_mode_attr.oper_pwr_mode_e
+    finally:
+        delete_sx_mgmt_phy_mod_pwr_attr_t_p(sx_mgmt_phy_mod_pwr_attr_p)
+
+
+def pwr_attr_set(handle, module_id, ports, attr_type, power_mode):
+    # Check if the module already works in the same mode
+    admin_pwr_mode, oper_pwr_mode = mgmt_phy_mod_pwr_attr_get(handle, module_id, attr_type)
+    if (power_mode == SX_MGMT_PHY_MOD_PWR_MODE_LOW_E and oper_pwr_mode == SX_MGMT_PHY_MOD_PWR_MODE_LOW_E) \
+       or (power_mode == SX_MGMT_PHY_MOD_PWR_MODE_AUTO_E and admin_pwr_mode == SX_MGMT_PHY_MOD_PWR_MODE_AUTO_E):
+        return
+    try:
+        # Bring the port down
+        for port in ports:
+            set_port_admin_status_by_log_port(handle, port, SX_PORT_ADMIN_STATUS_DOWN)
+        # Set the desired power mode
+        mgmt_phy_mod_pwr_attr_set(handle, module_id, attr_type, power_mode)
+        # Bring the port up
+    finally:
+        for port in ports:
+            set_port_admin_status_by_log_port(handle, port, SX_PORT_ADMIN_STATUS_UP)
+
+def set_lpmode(handle, cmd, module_id):
+    # Construct the port module map.
+    log_port_list = get_log_ports(handle, module_id)
+
+    if cmd == "enable":
+        pwr_attr_set(handle, module_id, log_port_list, SX_MGMT_PHY_MOD_PWR_ATTR_PWR_MODE_E, SX_MGMT_PHY_MOD_PWR_MODE_LOW_E)
+        print("Enabled low power mode for module [%d]" % module_id)
+    elif cmd == "disable":
+        pwr_attr_set(handle, module_id, log_port_list, SX_MGMT_PHY_MOD_PWR_ATTR_PWR_MODE_E, SX_MGMT_PHY_MOD_PWR_MODE_AUTO_E)
+        print("Disabled low power mode for module [%d]" % module_id)
     else:
-        pmaos.admin_status = PMAOS_ENABLE
-
-    meta.access_cmd = SXD_ACCESS_CMD_SET
-    rc = sxd_access_reg_pmaos(pmaos, meta, REGISTER_NUM, None, None)
-    assert rc == SXD_STATUS_SUCCESS, "sxd_access_reg_pmaos failed, rc = %d" % rc
-
-def set_sfp_lpmode(sfp_module, lpm_enable):
-    # Get PMMP
-    pmmp = ku_pmmp_reg()
-    pmmp.module = sfp_module
-    meta = init_sx_meta_data()
-    meta.access_cmd = SXD_ACCESS_CMD_GET
-    rc = sxd_access_reg_pmmp(pmmp, meta, REGISTER_NUM, None, None)
-    assert rc == SXD_STATUS_SUCCESS, "sxd_access_reg_pmmp failed, rc = %d" % rc
-
-    # Set low power mode status
-    lpm_mask = 1 << 8
-    if lpm_enable:
-        pmmp.eeprom_override = pmmp.eeprom_override | lpm_mask
-    else:
-        pmmp.eeprom_override = pmmp.eeprom_override & (~lpm_mask)
-
-    meta.access_cmd = SXD_ACCESS_CMD_SET
-    rc = sxd_access_reg_pmmp(pmmp, meta, REGISTER_NUM, None, None)
-    assert rc == SXD_STATUS_SUCCESS, "sxd_access_reg_pmmp failed, rc = %d" % rc
+        print("Error: Invalid command")
+        sys.exit(0)
 
-# Check if SFP port number is provided
 if len(sys.argv) < 3:
-    print "SFP module number or LPM is missed."
-    print "Usage: sfplpmset.py <SFP module> <on|off>"
+    print("SFP module number or LPM is missed.")
+    print("Usage: sfplpmset.py <SFP module> <on|off>")
     sys.exit(errno.EINVAL)
 
+cmd = None
 lpm_enable = None
 if sys.argv[2] == 'on':
     lpm_enable = True
+    cmd = 'enable'
 elif sys.argv[2] == 'off':
     lpm_enable = False
+    cmd = 'disable'
 else:
-    print "Unrecognized LPM parameter. Please use <on> or <off> values"
+    print("Unrecognized LPM parameter. Please use <on> or <off> values")
     sys.exit(errno.EINVAL)
 
-# Init SDK API
-rc, handle = sx_api_open(None)
-if (rc != SX_STATUS_SUCCESS):
-    print "Failed to open api handle.\nPlease check that SDK is running."
-    sys.exit(errno.EACCES)
-
-pid = os.getpid()
-rc = sxd_access_reg_init(pid, None, SXD_LOG_VERBOSITY_LEVEL)
-if (rc != SXD_STATUS_SUCCESS):
-    print "Failed to initializing register access.\nPlease check that SDK is running."
-    sys.exit(errno.EACCES);
-
 # Get SFP module
 sfp_module = int(sys.argv[1]) - 1
 
-# Get all ports at admin up status that related to the SFP module
-log_port_list = get_log_ports(handle, sfp_module)
-
-# SET SFP related ports to admin down status
-for log_port in log_port_list:
-    set_port_admin_status_by_log_port(handle, log_port, SX_PORT_ADMIN_STATUS_DOWN)
-
-# Disable admin status before LPM settings
-set_sfp_admin_status(sfp_module, SX_PORT_ADMIN_STATUS_DOWN)
+print "[+] opening sdk"
+rc, handle = sx_api_open(None)
 
-# Set low power mode status
-set_sfp_lpmode(sfp_module, lpm_enable)
+if (rc != SX_STATUS_SUCCESS):
+    print("Failed to open api handle.\nPlease check that SDK is running.")
+    sys.exit(errno.EACCES)
 
-# Enable admin status after LPM settings
-set_sfp_admin_status(sfp_module, SX_PORT_ADMIN_STATUS_UP)
+# Set low power mode
+set_lpmode(handle, cmd, sfp_module)
 
-# SET SFP related ports to admin up status
-for log_port in log_port_list:
-    set_port_admin_status_by_log_port(handle, log_port, SX_PORT_ADMIN_STATUS_UP)
+sx_api_close(handle)

device/mellanox/x86_64-mlnx_msn2700-r0/plugins/sfpreset.py

@@ -1,45 +1,28 @@
-#!/usr/bin/env python
-
-import sys, errno
-import os
-from python_sdk_api.sxd_api import *
-from python_sdk_api.sx_api import *
-
-# Check if SFP port number is provided
-if len(sys.argv) < 2:
-    print "SFP module number or LPM is missed."
-    print "Usage: sfpreset.py <SFP module>"
-    sys.exit(errno.EINVAL)
-
-# Init SDK API
-rc, handle = sx_api_open(None)
-if (rc != SX_STATUS_SUCCESS):
-    print "Failed to open api handle.\nPlease check that SDK is running."
-    sys.exit(errno.EACCES)
-
-pid = os.getpid()
-rc = sxd_access_reg_init(pid, None, 0)
-if (rc != 0):
-    print "Failed to initializing register access.\nPlease check that SDK is running."
-    sys.exit(errno.EACCES)
-
-# Get SFP module number
-sfp_module = int(sys.argv[1]) - 1
-
-# Get PMAOS
-pmaos = ku_pmaos_reg()
-pmaos.module = sfp_module
-meta = sxd_reg_meta_t()
-meta.dev_id = 1
-meta.swid = 0
-meta.access_cmd = SXD_ACCESS_CMD_GET
-
-rc = sxd_access_reg_pmaos(pmaos, meta, 1, None, None)
-assert rc == SXD_STATUS_SUCCESS, "sxd_access_reg_pmaos failed, rc = %d" % rc
-
-# Reset SFP
-pmaos.rst = 1
-meta.access_cmd = SXD_ACCESS_CMD_SET
-rc = sxd_access_reg_pmaos(pmaos, meta, 1, None, None)
-assert rc == SXD_STATUS_SUCCESS, "sxd_access_reg_pmaos failed, rc = %d" % rc
-print "Reset flag is set"
+#!/usr/bin/env python
+"""
+This utility reset the given SFP module.
+"""
+
+import sys
+import errno
+from python_sdk_api.sx_api import *
+
+# Check if SFP port number is provided
+if len(sys.argv) < 2:
+    print("SFP module number or LPM is missed.")
+    print("Usage: sfpreset.py <SFP module>")
+    sys.exit(errno.EINVAL)
+
+# Init SDK API
+rc, handle = sx_api_open(None)
+if rc != SX_STATUS_SUCCESS:
+    print("Failed to open api handle.\nPlease check that SDK is running.")
+    sys.exit(errno.EACCES)
+
+# Get SFP module number
+sfp_module = int(sys.argv[1]) - 1
+
+rc = sx_mgmt_phy_mod_reset(handle, sfp_module)
+assert rc == SX_STATUS_SUCCESS, "sx_mgmt_phy_mod_reset failed, rc = %d" % rc
+
+sx_api_close(handle)