src/chromiumos/tast/local/bundles/cros/platform/crossystem.go

// Copyright 2019 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package platform

import (
	"bytes"
	"context"
	"regexp"
	"strings"

	"chromiumos/tast/common/testexec"
	"chromiumos/tast/shutil"
	"chromiumos/tast/testing"
)

func init() {
	testing.AddTest(&testing.Test{
		Func: Crossystem,
		Desc: "Checks the crossystem command's basic functionality",
		Contacts: []string{
			"mka@chromium.org",
			"kasaiah.bogineni@intel.com", // Port author
			"tast-users@chromium.org",
		},
		SoftwareDeps: []string{"crossystem"},
		Attr:         []string{"group:mainline", "group:labqual"},
	})
}

// Crossystem checks that the "crossystem" command's basic functionality.
// This includes commands that rely on the presence and correct
// initialization of the chromeos driver (drivers/platform/chromeos.c)
// in the kernel
func Crossystem(ctx context.Context, s *testing.State) {
	const (
		crossystemCmd = "crossystem"
		alphaNum      = `^[\d\w]+$`
		num           = `^[\d]+$`
		hexNum        = `^0x[\da-fA-F]+$`
		bit           = `^[01]$`
		anything      = "" // match everything that isn't an error or empty
	)

	cmdRegexMap := map[string]string{
		"cros_debug":  bit,
		"debug_build": bit,
		"devsw_boot":  bit,
		"devsw_cur":   bit,
		"fwid":        anything,
		"hwid":        anything,
		"loc_idx":     num,
		"mainfw_act":  alphaNum,
		"mainfw_type": alphaNum,
		"ro_fwid":     anything,
		"tpm_fwver":   hexNum,
		"tpm_kernver": hexNum,
		"wpsw_cur":    bit,
	}
	checkOutput := func(pattern string, out []byte) bool {
		if pattern == anything {
			return strings.TrimSpace(string(out)) != "" && strings.TrimSpace(string(out)) != "(error)"
		}
		return regexp.MustCompile(pattern).Match(bytes.TrimSpace(out))
	}

	for subCommand, regExp := range cmdRegexMap {
		cmd := testexec.CommandContext(ctx, crossystemCmd, subCommand)
		output, err := cmd.Output(testexec.DumpLogOnError)
		if err != nil {
			s.Errorf("%q failed: %v", shutil.EscapeSlice(cmd.Args), err)
		} else if !checkOutput(regExp, output) {
			s.Errorf("%q printed %q, which isn't matched by %q", shutil.EscapeSlice(cmd.Args), output, regExp)
		}
	}
}