removed is_mellanox_device() check in save/load pci configuration

Signed-off-by: Roei Yitzhak <[email protected]>

Author: roei-mlnx

small_utils/mstfwreset.py

@@ -487,11 +487,10 @@ def __init__(self):
 
     def savePCIConfigurationSpace(self, devAddr):
         confSpaceList = []
-        for addr in range(0, 0xfff, 4):
-            if self.isMellanoxDevice(devAddr):
-                if (addr == 0x58 or addr == 0x5c):
-                    confSpaceList.append(0)
-                    continue
+        for addr in range(0, 0xff, 4):
+            if (addr == 0x58 or addr == 0x5c):
+                confSpaceList.append(0)
+                continue
             val = self.read(devAddr, addr)
             confSpaceList.append(val)
         self.pciConfSpaceMap[devAddr] = confSpaceList
@@ -503,13 +502,12 @@ def loadPCIConfigurationSpace(self, devAddr):
             pciConfSpaceList = self.pciConfSpaceMap[devAddr]
         except KeyError as ke:
             raise RuntimeError("PCI configuration space for device %s was not saved please reboot server" % devAddr)
-        if len(pciConfSpaceList) != 0x400:
+        if len(pciConfSpaceList) != 64:
             raise RuntimeError("PCI configuration space for device %s was not saved properly please reboot server" % devAddr)
-        for i in range(0, 0x400):
+        for i in range(0, 64):
             addr = i * 4
-            if self.isMellanoxDevice(devAddr):
-                if (addr == 0x58 or addr == 0x5c):
-                    continue
+            if (addr == 0x58 or addr == 0x5c):
+                continue
             self.write(devAddr, addr, pciConfSpaceList[i])
         return
 
@@ -593,10 +591,9 @@ def getMFDeviceList(self, mstDev,devAddr):
     def savePCIConfigurationSpace(self, devAddr):
         confSpaceList = []
         for addr in range(0, 0xff, 4):
-            if self.isMellanoxDevice(devAddr):
-                if (addr == 0x58 or addr == 0x5c):
-                    confSpaceList.append(0)
-                    continue
+            if (addr == 0x58 or addr == 0x5c):
+                confSpaceList.append(0)
+                continue
             val = self.read(devAddr, addr)
             confSpaceList.append(val)
         self.pciConfSpaceMap[devAddr] = confSpaceList
@@ -612,9 +609,8 @@ def loadPCIConfigurationSpace(self, devAddr):
             raise RuntimeError("PCI configuration space for device %s was not saved properly please reboot server" % devAddr)
         for i in range(0, 64):
             addr = i * 4
-            if self.isMellanoxDevice(devAddr):
-                if (addr == 0x58 or addr == 0x5c):
-                    continue
+            if (addr == 0x58 or addr == 0x5c):
+                continue
             self.write(devAddr, addr, pciConfSpaceList[i])
         return