Merge pull request #59 from TheManticoreProject/Improved-error-handling-of-CustomKeyInformationVolumeType

p0dalirius · web-flow · commit 08589d4ded12 · 2025-11-12T10:41:15.000+01:00
[enhancement] Improved error handling of CustomKeyInformationVolumeType
diff --git a/windows/keycredentiallink/key/customkeyinformation/CustomKeyInformationVolumeType.go b/windows/keycredentiallink/key/customkeyinformation/CustomKeyInformationVolumeType.go
@@ -44,6 +44,10 @@ func (vt CustomKeyInformationVolumeType) Unmarshal(data []byte) (int, error) {
 
 	vt = CustomKeyInformationVolumeType(data[0])
 
+	if vt != CustomKeyInformationVolumeType_None && vt != CustomKeyInformationVolumeType_OSV && vt != CustomKeyInformationVolumeType_FDV && vt != CustomKeyInformationVolumeType_RDV {
+		return 0, fmt.Errorf("invalid CustomKeyInformationVolumeType: %d", vt)
+	}
+
 	return 1, nil
 }
 

Original file line number	Diff line number	Diff line change
`@@ -44,6 +44,10 @@ func (vt CustomKeyInformationVolumeType) Unmarshal(data []byte) (int, error) {`
`44`	`44`
`45`	`45`	`vt = CustomKeyInformationVolumeType(data[0])`
`46`	`46`
	`47`	`+ if vt != CustomKeyInformationVolumeType_None && vt != CustomKeyInformationVolumeType_OSV && vt != CustomKeyInformationVolumeType_FDV && vt != CustomKeyInformationVolumeType_RDV {`
	`48`	`+ return 0, fmt.Errorf("invalid CustomKeyInformationVolumeType: %d", vt)`
	`49`	`+ }`
	`50`	`+`
`47`	`51`	`return 1, nil`
`48`	`52`	`}`
`49`	`53`