From 77778d24ae29a0f74b436dd1e5852cabdf43de05 Mon Sep 17 00:00:00 2001
From: Florent Kermarrec <florent@enjoy-digital.fr>
Date: Tue, 20 Jan 2015 13:17:59 +0100
Subject: [PATCH] bist: add decoding of capabilities

---
 test/bist.py | 41 ++++++++++++++++++++++++++---------------
 1 file changed, 26 insertions(+), 15 deletions(-)

diff --git a/test/bist.py b/test/bist.py
index 96603e484..f2a1e40a1 100644
--- a/test/bist.py
+++ b/test/bist.py
@@ -1,5 +1,6 @@
 import time
 import argparse
+from collections import OrderedDict
 from config import *
 
 KB = 1024
@@ -81,11 +82,19 @@ class LiteSATABISTIdentifyDriver:
 		self.total_sectors += (self.data[102] << 32)
 		self.total_sectors += (self.data[103] << 48)
 
+		self.capabilities = OrderedDict()
+		self.capabilities["SATA Gen1"] = (self.data[76] >> 1) & 0x1
+		self.capabilities["SATA Gen2"] = (self.data[76] >> 2) & 0x1
+		self.capabilities["SATA Gen3"] = (self.data[76] >> 3) & 0x1
+		self.capabilities["48 bits LBA supported"] = (self.data[83] >> 10) & 0x1
+
 	def hdd_info(self):
 		info = "Serial Number: " + self.serial_number + "\n"
 		info += "Firmware Revision: " + self.firmware_revision + "\n"
 		info += "Model Number: " + self.model_number + "\n"
 		info += "Capacity: %3.2f GB\n" %((self.total_sectors*logical_sector_size)/GB)
+		for k, v in self.capabilities.items():
+			info += k + ": " + str(v) + "\n"
 		print(info)
 
 def _get_args():
@@ -97,6 +106,7 @@ SATA BIST utility.
 	parser.add_argument("-l", "--total_length", default=256, help="total transfer length (in MB, up to HDD capacity)")
 	parser.add_argument("-r", "--random", action="store_true", help="use random data")
 	parser.add_argument("-c", "--continuous", action="store_true", help="continuous mode (Escape to exit)")
+	parser.add_argument("-i", "--identify", action="store_true", help="only run identify")
 	return parser.parse_args()
 
 if __name__ == "__main__":
@@ -110,23 +120,24 @@ if __name__ == "__main__":
 	identify.run()
 	identify.hdd_info()
 
-	sector = 0
-	count = int(args.transfer_size)*MB//logical_sector_size
-	length = int(args.total_length)*MB
-	random = int(args.random)
-	continuous = int(args.continuous)
-	try:
-		while ((sector*logical_sector_size < length) or continuous) and (sector < identify.total_sectors):
-			# generator (write data to HDD)
-			write_speed, write_errors = generator.run(sector, count, random)
+	if not int(args.identify):
+		sector = 0
+		count = int(args.transfer_size)*MB//logical_sector_size
+		length = int(args.total_length)*MB
+		random = int(args.random)
+		continuous = int(args.continuous)
+		try:
+			while ((sector*logical_sector_size < length) or continuous) and (sector < identify.total_sectors):
+				# generator (write data to HDD)
+				write_speed, write_errors = generator.run(sector, count, random)
 
-			# checker (read and check data from HDD)
-			read_speed, read_errors = checker.run(sector, count, random)
+				# checker (read and check data from HDD)
+				read_speed, read_errors = checker.run(sector, count, random)
 
-			print("sector=%d write_speed=%4.2fMB/sec read_speed=%4.2fMB/sec errors=%d" %(sector, write_speed/MB, read_speed/MB, write_errors + read_errors))
-			sector += count
+				print("sector=%d write_speed=%4.2fMB/sec read_speed=%4.2fMB/sec errors=%d" %(sector, write_speed/MB, read_speed/MB, write_errors + read_errors))
+				sector += count
 
-	except KeyboardInterrupt:
-		pass
+		except KeyboardInterrupt:
+			pass
 	###
 	wb.close()