1 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved. 2 # Use of this source code is governed by a BSD-style license that can be 3 # found in the LICENSE file. 4 5 import os 6 import re 7 8 from autotest_lib.client.bin import test, utils 9 from autotest_lib.client.common_lib import error 10 11 class hardware_SsdDetection(test.test): 12 """Verify that a flash device is present. """ 13 14 version = 1 15 # Keep a list of boards that are expected to ship with hard drive. 16 boards_with_hdd = ['butterfly', 'kiev', 'parrot', 'stout'] 17 18 def setup(self): 19 """ 20 create a empty srcdir to prevent the error that checks 21 .version file 22 """ 23 if not os.path.exists(self.srcdir): 24 utils.system('mkdir %s' % self.srcdir) 25 26 27 def run_once(self, check_link_speed=()): 28 """ 29 Use rootdev to find the underlying block device even if the 30 system booted to /dev/dm-0. 31 """ 32 device = utils.get_root_device() 33 34 def is_fixed(dev): 35 """ Check the device is fixed. 36 37 @param dev: device to check, i.e. 'sda'. 38 """ 39 sysfs_path = '/sys/block/%s/removable' % dev 40 return (os.path.exists(sysfs_path) and 41 open(sysfs_path).read().strip() == '0') 42 43 # Catch device name like sda, mmcblk0, nvme0n1. 44 device_re = re.compile(r'^/dev/([a-zA-Z0-9]+)$') 45 dev = device_re.findall(device) 46 if len(dev) != 1 or not is_fixed(dev[0]): 47 raise error.TestFail('The main disk %s is not fixed' % dev) 48 49 # If it is an mmcblk or nvme device, then it is SSD. 50 # Else run hdparm to check for SSD. 51 if re.search("nvme", device): 52 return 53 54 if re.search("mmcblk", device): 55 return 56 57 hdparm = utils.run('/sbin/hdparm -I %s' % device) 58 59 # Check if device is a SSD 60 match = re.search(r'Nominal Media Rotation Rate: (.+)$', 61 hdparm.stdout, re.MULTILINE) 62 if match and match.group(1): 63 if match.group(1) != 'Solid State Device': 64 if utils.get_board() in self.boards_with_hdd: 65 return 66 raise error.TestFail('The main disk is not a SSD, ' 67 'Rotation Rate: %s' % match.group(1)) 68 else: 69 raise error.TestFail( 70 'Rotation Rate not reported from the device, ' 71 'unable to ensure it is a SSD') 72 73 # Check if SSD is > 8GB in size 74 match = re.search("device size with M = 1000\*1000: (.+) MBytes", 75 hdparm.stdout, re.MULTILINE) 76 if match and match.group(1): 77 size = int(match.group(1)) 78 self.write_perf_keyval({"mb_ssd_device_size" : size}) 79 else: 80 raise error.TestFail( 81 'Device size info missing from the device') 82 83 # Check supported link speed. 84 # 85 # check_link_speed is an empty tuple by default, which does not perform 86 # link speed checking. You can run the test while specifying 87 # check_link_speed=('1.5Gb/s', '3.0Gb/s') to check the 2 signaling 88 # speeds are both supported. 89 for link_speed in check_link_speed: 90 if not re.search(r'Gen. signaling speed \(%s\)' % link_speed, 91 hdparm.stdout, re.MULTILINE): 92 raise error.TestFail('Link speed %s not supported' % link_speed) 93
