Change FMS fields to a single CPUID_1_EAX formatted field.

abi/abi.go

@@ -121,10 +121,11 @@ const (
 	extendedFamilyShift = 20
 	extendedModelShift  = 16
 	familyShift         = 8
-	sevExtendedFamily   = 0xA
-	sevFamily           = 0xF
-	milanExtendedModel  = 0
-	genoaExtendedModel  = 1
+	modelShift          = 4
+	// Combined extended values
+	zen3zen4Family = 0x19
+	milanModel     = 0
+	genoaModel     = 1 << 4
 
 	// ReportVersion2 is set by the SNP API specification
 	// https://web.archive.org/web/20231222054111if_/http://www.amd.com/content/dam/amd/en/documents/epyc-technical-docs/specifications/56860.pdf
@@ -487,9 +488,7 @@ func ReportToProto(data []uint8) (*pb.Report, error) {
 	mbzLo := 0x188
 	if r.Version == ReportVersion3 {
 		mbzLo = 0x18B
-		r.CpuidFamId = []byte{data[0x188]}
-		r.CpuidModId = []byte{data[0x189]}
-		r.CpuidStep = []byte{data[0x18A]}
+		r.Cpuid1EaxFms = FmsToCpuid1Eax(data[0x188], data[0x189], data[0x18A])
 	}
 
 	if err := mbz(data, mbzLo, 0x1A0); err != nil {
@@ -635,9 +634,10 @@ func ReportToAbiBytes(r *pb.Report) ([]byte, error) {
 
 	// Add CPUID information if this is a version 3 report.
 	if r.Version == ReportVersion3 {
-		data[0x188] = r.CpuidFamId[0]
-		data[0x189] = r.CpuidModId[0]
-		data[0x18A] = r.CpuidStep[0]
+		family, model, stepping := FmsFromCpuid1Eax(r.Cpuid1EaxFms)
+		data[0x188] = family
+		data[0x189] = model
+		data[0x18A] = stepping
 	}
 
 	copy(data[0x1A0:0x1E0], r.ChipId[:])
@@ -903,27 +903,57 @@ func (c *CertTable) Proto() *pb.CertificateChain {
 // See assembly implementations in cpuid_*.s
 var cpuid func(op uint32) (eax, ebx, ecx, edx uint32)
 
-// SevProductFromCpuid1Eax returns the SevProduct that is represented by cpuid(1).eax.
-func SevProductFromCpuid1Eax(eax uint32) *pb.SevProduct {
+// FmsToCpuid1Eax returns the masked CPUID_1_EAX value that represents the given
+// family, model, stepping (FMS) values.
+func FmsToCpuid1Eax(family, model, stepping byte) uint32 {
+	var extendedFamily byte
+
+	familyID := family
+	if family >= 0xf {
+		extendedFamily = family - 0xf
+		familyID = 0xf
+	}
+	extendedModel := model >> 4
+	modelID := model & 0xf
+	return (uint32(extendedFamily) << extendedFamilyShift) |
+		(uint32(extendedModel) << extendedModelShift) |
+		(uint32(familyID) << familyShift) |
+		(uint32(modelID) << modelShift) |
+		(uint32(stepping & 0xf))
+}
+
+// FmsFromCpuid1Eax returns the family, model, stepping (FMS) values extracted from a
+// CPUID_1_EAX value.
+func FmsFromCpuid1Eax(eax uint32) (byte, byte, byte) {
 	// 31:28 reserved
 	// 27:20 Extended Family ID
-	extendedFamily := (eax >> extendedFamilyShift) & 0xff
+	extendedFamily := byte((eax >> extendedFamilyShift) & 0xff)
 	// 19:16 Extended Model ID
-	extendedModel := (eax >> extendedModelShift) & 0xf
+	extendedModel := byte((eax >> extendedModelShift) & 0xf)
 	// 15:14 reserved
 	// 11:8 Family ID
-	family := (eax >> familyShift) & 0xf
+	familyID := byte((eax >> familyShift) & 0xf)
+	// 7:4 Model
+	modelID := byte((eax >> modelShift) & 0xf)
 	// 3:0 Stepping
-	stepping := eax & 0xf
+	family := extendedFamily + familyID
+	model := (extendedModel << 4) | modelID
+	stepping := byte(eax & 0xf)
+	return family, model, stepping
+}
+
+// SevProductFromCpuid1Eax returns the SevProduct that is represented by cpuid(1).eax.
+func SevProductFromCpuid1Eax(eax uint32) *pb.SevProduct {
+	family, model, stepping := FmsFromCpuid1Eax(eax)
 	// Ah, Fh, {0h,1h} values from the KDS specification,
 	// section "Determining the Product Name".
 	var productName pb.SevProduct_SevProductName
 	// Product information specified by processor programming reference publications.
-	if extendedFamily == sevExtendedFamily && family == sevFamily {
-		switch extendedModel {
-		case milanExtendedModel:
+	if family == zen3zen4Family {
+		switch model {
+		case milanModel:
 			productName = pb.SevProduct_SEV_PRODUCT_MILAN
-		case genoaExtendedModel:
+		case genoaModel:
 			productName = pb.SevProduct_SEV_PRODUCT_GENOA
 		default:
 			productName = pb.SevProduct_SEV_PRODUCT_UNKNOWN
@@ -932,7 +962,7 @@ func SevProductFromCpuid1Eax(eax uint32) *pb.SevProduct {
 	}
 	return &pb.SevProduct{
 		Name:            productName,
-		MachineStepping: &wrapperspb.UInt32Value{Value: stepping},
+		MachineStepping: &wrapperspb.UInt32Value{Value: uint32(stepping)},
 	}
 }
 
@@ -943,15 +973,15 @@ func MaskedCpuid1EaxFromSevProduct(product *pb.SevProduct) uint32 {
 	if product.MachineStepping != nil {
 		stepping = product.MachineStepping.Value & 0xf
 	}
-	extendedFamily := uint32(sevExtendedFamily) << extendedFamilyShift
-	family := uint32(sevFamily) << familyShift
+	extendedFamily := uint32(zen3zen4Family-0xf) << extendedFamilyShift
+	family := uint32(0xf) << familyShift
 
 	var extendedModel uint32
 	switch product.Name {
 	case pb.SevProduct_SEV_PRODUCT_MILAN:
-		extendedModel = milanExtendedModel
+		extendedModel = milanModel
 	case pb.SevProduct_SEV_PRODUCT_GENOA:
-		extendedModel = genoaExtendedModel
+		extendedModel = genoaModel
 	default:
 		return 0
 	}

abi/abi_test.go

@@ -61,9 +61,7 @@ var (
 	author_key_digest: '\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
 	report_id: '\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
 	report_id_ma: '\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
-	cpuid_fam_id: '\x00\x00'
-	cpuid_mod_id: '\x00\x00'
-	cpuid_step: '\x00\x00'
+	cpuid1eax_fms: 0
 	chip_id: '\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
 	signature: '\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
 	`
@@ -362,28 +360,28 @@ func TestSevProduct(t *testing.T) {
 		want *spb.SevProduct
 	}{
 		{
-			eax: 0x00a00f10,
+			eax: 0x00a00f00,
 			want: &spb.SevProduct{
 				Name:            spb.SevProduct_SEV_PRODUCT_MILAN,
 				MachineStepping: &wrapperspb.UInt32Value{Value: 0},
 			},
 		},
 		{
-			eax: 0x00a00f11,
+			eax: 0x00a00f01,
 			want: &spb.SevProduct{
 				Name:            spb.SevProduct_SEV_PRODUCT_MILAN,
 				MachineStepping: &wrapperspb.UInt32Value{Value: 1},
 			},
 		},
 		{
-			eax: 0x00a10f10,
+			eax: 0x00a10f00,
 			want: &spb.SevProduct{
 				Name:            spb.SevProduct_SEV_PRODUCT_GENOA,
 				MachineStepping: &wrapperspb.UInt32Value{Value: 0},
 			},
 		},
 		{
-			eax: 0x00a10f12,
+			eax: 0x00a10f02,
 			want: &spb.SevProduct{
 				Name:            spb.SevProduct_SEV_PRODUCT_GENOA,
 				MachineStepping: &wrapperspb.UInt32Value{Value: 2},
@@ -446,8 +444,8 @@ func TestExtendedPlatformCertTable(t *testing.T) {
 		eax      uint32
 		stepping uint32
 	}{
-		{name: "Genoa-B2 cruft", pname: spb.SevProduct_SEV_PRODUCT_GENOA, eax: 0x00a10f12, stepping: 2},
-		{name: "Milan-B1 cruft", pname: spb.SevProduct_SEV_PRODUCT_MILAN, eax: 0x00a00f11, stepping: 1},
+		{name: "Genoa-B2", pname: spb.SevProduct_SEV_PRODUCT_GENOA, eax: 0x00a10f02, stepping: 2},
+		{name: "Milan-B1", pname: spb.SevProduct_SEV_PRODUCT_MILAN, eax: 0x00a00f01, stepping: 1},
 		{name: "Milan-B0", pname: spb.SevProduct_SEV_PRODUCT_MILAN, eax: 0x00a00f00, stepping: 0},
 	}
 	for _, tc := range tcs {

go.mod

@@ -3,6 +3,7 @@ module github.com/google/go-sev-guest
 go 1.19
 
 require (
+	github.com/golang/protobuf v1.5.0
 	github.com/google/go-cmp v0.5.7
 	github.com/google/go-configfs-tsm v0.2.2
 	github.com/google/logger v1.1.1

go.sum

@@ -1,4 +1,7 @@
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/golang/protobuf v1.5.0 h1:LUVKkCeviFUMKqHa4tXIIij/lbhnMbP7Fn5wKdKkRh4=
+github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
+github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
 github.com/google/go-cmp v0.5.7 h1:81/ik6ipDQS2aGcBfIN5dHDB36BwrStyeAQquSYCV4o=
 github.com/google/go-cmp v0.5.7/go.mod h1:n+brtR0CgQNWTVd5ZUFpTBC8YFBDLK/h/bpaJ8/DtOE=
 github.com/google/go-configfs-tsm v0.2.2 h1:YnJ9rXIOj5BYD7/0DNnzs8AOp7UcvjfTvt215EWcs98=
@@ -18,6 +21,7 @@ golang.org/x/sys v0.15.0 h1:h48lPFYpsTvQJZF4EKyI4aLHaev3CxivZmv7yZig9pc=
 golang.org/x/sys v0.15.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
 golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
 google.golang.org/protobuf v1.33.0 h1:uNO2rsAINq/JlFpSdYEKIZ0uKD/R9cpdv0T+yoGwGmI=
 google.golang.org/protobuf v1.33.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=

proto/sevsnp.proto

@@ -56,9 +56,7 @@ message Report {
   uint64 launch_tcb = 27;
   bytes signature = 28;  // Should be 512 bytes long
 
-  bytes cpuid_fam_id = 29;  // 1 byte combined Extended Family ID and Family ID
-  bytes cpuid_mod_id = 30;  // 1 byte combined Extended Model and Model fields
-  bytes cpuid_step = 31;    // 1 byte Stepping
+  uint32 cpuid1eax_fms = 29;  // The cpuid(1).eax & 0x0fff0fff representation of family/model/stepping
 }
 
 message CertificateChain {