-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
chore(examples): Lets isvalid be useful (#1856)
### Proposed Changes * Replaces isvalid example with a utility for checking isvalid on a file ### Checklist - [ ] I have added or updated unit tests - [ ] I have added or updated integration tests (if appropriate) - [ ] I have added or updated documentation ### Testing Instructions
- Loading branch information
1 parent
c3cda1b
commit bc3808f
Showing
2 changed files
with
57 additions
and
32 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,56 +1,81 @@ | ||
package cmd | ||
|
||
import ( | ||
"bytes" | ||
"encoding/base64" | ||
"fmt" | ||
"io" | ||
"os" | ||
|
||
"github.com/opentdf/platform/sdk" | ||
"github.com/spf13/cobra" | ||
) | ||
|
||
func init() { | ||
encryptCmd := cobra.Command{ | ||
Use: "isvalid", | ||
var isValidCmd = &cobra.Command{ | ||
Use: "isvalid [files...]", | ||
Short: "Check validity of a TDF", | ||
RunE: isValid, | ||
RunE: func(cmd *cobra.Command, args []string) error { | ||
return areValid(cmd, args) | ||
}, | ||
} | ||
|
||
ExamplesCmd.AddCommand(&encryptCmd) | ||
ExamplesCmd.AddCommand(isValidCmd) | ||
} | ||
|
||
func isValid(cmd *cobra.Command, args []string) error { | ||
goodNanoTdfStr := "TDFMABJsb2NhbGhvc3Q6ODA4MC9rYXOAAQIA2qvjMRfg7b27lT2kf9SwHRkDIg8ZXtfRoiIvdMUHq/gL5AUMfmv4Di8sKCyLkmUm/WITVj5hDeV/z4JmQ0JL7ZxqSmgZoK6TAHvkKhUly4zMEWMRXH8IktKhFKy1+fD+3qwDopqWAO5Nm2nYQqi75atEFckstulpNKg3N+Ul22OHr/ZuR127oPObBDYNRfktBdzoZbEQcPlr8q1B57q6y5SPZFjEzL9weK+uS5bUJWkF3nsHASo2bZw7IPhTZxoFVmCDjwvj6MbxNa7zG6aClHJ162zKxLLnD9TtIHuZ59R7LgiSieipXeExj+ky9OgIw5DfwyUuxsQLtKpMIAFPmLY9Hy2naUJxke0MT1EUBgastCq+YtFGslV9LJo/A8FtrRqludwtM0O+Z9FlAkZ1oNL7M7uOkLrh7eRrv+C1AAAX6FaBQoOtqnmyu6Jp+VzkxDddEeLRUyI=" | ||
badNanoTdfStr := "TDFMABfg7b27lT2kf9SwHRkDIg8ZXtfRoiIvdMUHq/gL5AUMfmv4Di8sKCyLkmUm/WITVj5hDeV/z4JmQ0JL7ZxqSmgZoK6TAHvkKhUly4zMEWMRXH8IktKhFKy1+fD+3qwDopqWAO5Nm2nYQqi75atEFckstulpNKg3N+Ul22OHr/ZuR127oPObBDYNRfktBdzoZbEQcPlr8q1B57q6y5SPZFjEzL9weK+uS5bUJWkF3nsHASo2bZw7IPhTZxoFVmCDjwvj6MbxNa7zG6aClHJ162zKxLLnD9TtIHuZ59R7LgiSieipXeExj+ky9OgIw5DfwyUuxsQLtKpMIAFPmLY9Hy2naUJxke0MT1EUBgastCq+YtFGslV9LJo/A8FtrRqludwtM0O+Z9FlAkZ1oNL7M7uOkLrh7eRrv+C1AAAX6FaBQoOtqnmyu6Jp+VzkxDddEeLRUyI=" | ||
type typeInfo struct { | ||
Valid bool | ||
Type string | ||
Error error | ||
} | ||
|
||
func (t typeInfo) String() string { | ||
if t.Valid { | ||
return fmt.Sprintf("[✅ %s]", t.Type) | ||
} | ||
if t.Error != nil { | ||
return fmt.Sprintf("[🚮🔥 %s %v]", t.Type, t.Error) | ||
} | ||
return fmt.Sprintf("[📛 %s]", t.Type) | ||
} | ||
|
||
goodStandardTdf := "UEsDBC0ACAAAAJ2TFTEAAAAAAAAAAAAAAAAJAAAAMC5wYXlsb2Fktu4m+vdwl0mtjhY3U5e7TG2o1s8ifK+RAhFNjRjGTLJ7V3w5UEsHCGiY7skkAAAAJAAAAFBLAwQtAAgAAACdkxUxAAAAAAAAAAAAAAAADwAAADAubWFuaWZlc3QuanNvbnsiZW5jcnlwdGlvbkluZm9ybWF0aW9uIjp7InR5cGUiOiJzcGxpdCIsInBvbGljeSI6ImV5SjFkV2xrSWpvaU1HTTFORGsyWlRZdE5EYzRaaTB4TVdWbUxXSXlOakV0WWpJMVl6UmhORE14TjJFM0lpd2lZbTlrZVNJNmV5SmtZWFJoUVhSMGNtbGlkWFJsY3lJNlczc2lZWFIwY21saWRYUmxJam9pYUhSMGNITTZMeTlsZUdGdGNHeGxMbU52YlM5aGRIUnlMMkYwZEhJeEwzWmhiSFZsTDNaaGJIVmxNU0lzSW1ScGMzQnNZWGxPWVcxbElqb2lJaXdpYVhORVpXWmhkV3gwSWpwbVlXeHpaU3dpY0hWaVMyVjVJam9pSWl3aWEyRnpWVkpNSWpvaUluMWRMQ0prYVhOelpXMGlPbHRkZlgwPSIsImtleUFjY2VzcyI6W3sidHlwZSI6IndyYXBwZWQiLCJ1cmwiOiJodHRwOi8vbG9jYWxob3N0OjgwODAiLCJwcm90b2NvbCI6ImthcyIsIndyYXBwZWRLZXkiOiJ0VVMvUE9TaVBtOGV6OGhyL2dMVGN6Y1lOT0trcUNEclZiQTBWdHZna29QbHB0M1BDZVpTdDNndnlQNVZKZXBNMmNqdVBhUWJJUGlyMjlWdVJ2T1RXZmQzRUh1KzgyVCtFNEVZbEpBM25VbDdGQTRMUGZhUEtXWk1zTExHUkJJVUxZT0VhMWJma1MvUm9Xb0EwK283WlFFVkNhYmdJN2JFRDJKV2Q2aG1yam1iUnM2d0lwOVFXNUs4Q3dJWjZVZjlGMXEwRDViTmlrbGxHaCtiaVJsV1NucEwxbHBPaFdva1gxdUJsU0VRSDNvM2JtVXFTNVVaUjRmYUxuTW5xOGR0bS8wYnJjTjUwaFNiK0xTTlZkd2daTEszTTRHTmxEeGdzcDkxY0VuYjZoZktLemdSY0VCS0tMQTF1b3BXNHdCRG9BamFuWWplQlZVT3ZBZEI5ek45T3c9PSIsInBvbGljeUJpbmRpbmciOnsiYWxnIjoiSFMyNTYiLCJoYXNoIjoiWmpBek1HWXlZekl4WlRCbU16Tm1NamhoTWpGalpqSTJaRE5oWlRrMk5ERTNaREJoWlRrM05ESTJNREExTnpVMU1UVTFNV0ZpTTJSak9EUTFabU0yWWc9PSJ9LCJraWQiOiJyMSJ9XSwibWV0aG9kIjp7ImFsZ29yaXRobSI6IkFFUy0yNTYtR0NNIiwiaXYiOiIiLCJpc1N0cmVhbWFibGUiOnRydWV9LCJpbnRlZ3JpdHlJbmZvcm1hdGlvbiI6eyJyb290U2lnbmF0dXJlIjp7ImFsZyI6IkhTMjU2Iiwic2lnIjoiWkdWaFltRmtNRGhsTURCbU1UVm1ZekJtTVdFME0ySmhOamhrTmpBMVpUazFNVGRtWmpoa1pETmtNekk0Tldaa01XUXhOVFZsWXpjME1EVXhPRE13Tmc9PSJ9LCJzZWdtZW50SGFzaEFsZyI6IkdNQUMiLCJzZWdtZW50U2l6ZURlZmF1bHQiOjIwOTcxNTIsImVuY3J5cHRlZFNlZ21lbnRTaXplRGVmYXVsdCI6MjA5NzE4MCwic2VnbWVudHMiOlt7Imhhc2giOiJNakkzWTJGbU9URXdNakV4TkdRNFpERTRZelkwWTJJeU4ySTFOemRqTXprPSIsInNlZ21lbnRTaXplIjo4LCJlbmNyeXB0ZWRTZWdtZW50U2l6ZSI6MzZ9XX19LCJwYXlsb2FkIjp7InR5cGUiOiJyZWZlcmVuY2UiLCJ1cmwiOiIwLnBheWxvYWQiLCJwcm90b2NvbCI6InppcCIsIm1pbWVUeXBlIjoiYXBwbGljYXRpb24vb2N0ZXQtc3RyZWFtIiwiaXNFbmNyeXB0ZWQiOnRydWV9fVBLBwgwpFOlrwUAAK8FAABQSwECLQAtAAgAAACdkxUxaJjuySQAAAAkAAAACQAAAAAAAAAAAAAAAAAAAAAAMC5wYXlsb2FkUEsBAi0ALQAIAAAAnZMVMTCkU6WvBQAArwUAAA8AAAAAAAAAAAAAAAAAWwAAADAubWFuaWZlc3QuanNvblBLBQYAAAAAAgACAHQAAABHBgAAAAA=" | ||
func areValid(cmd *cobra.Command, files []string) error { | ||
if len(files) == 0 { | ||
// TK Add support for handling stdin | ||
return cmd.Usage() | ||
} | ||
for _, file := range files { | ||
in, err := os.Open(file) | ||
if err != nil { | ||
cmd.PrintErrf("Error opening file %s: %v\n", file, err) | ||
return err | ||
} | ||
defer in.Close() | ||
|
||
// Decode the base64 string | ||
goodDecodedNanoTdf, _ := base64.StdEncoding.DecodeString(goodNanoTdfStr) | ||
badDecodedNanoTdf, _ := base64.StdEncoding.DecodeString(badNanoTdfStr) | ||
cmd.Printf("File: [%s], TypeInfo: %v\n", file, isValid(cmd, in)) | ||
} | ||
return nil | ||
} | ||
|
||
goodDecodedStandardTdf, _ := base64.StdEncoding.DecodeString(goodStandardTdf) | ||
func isValid(cmd *cobra.Command, in io.ReadSeeker) []typeInfo { | ||
var typeInfos []typeInfo | ||
|
||
inGoodStandard := bytes.NewReader(goodDecodedStandardTdf) | ||
isValidTdf, _ := sdk.IsValidTdf(inGoodStandard) | ||
cmd.Println("Valid TDF: ") | ||
cmd.Println(isValidTdf) | ||
isValidTdf, err := sdk.IsValidTdf(in) | ||
typeInfos = append(typeInfos, typeInfo{Valid: isValidTdf, Error: err, Type: "TDF3"}) | ||
|
||
inGood := bytes.NewReader(goodDecodedNanoTdf) | ||
inBad := bytes.NewReader(badDecodedNanoTdf) | ||
isValidNanoTdfGood, _ := sdk.IsValidNanoTdf(inGood) | ||
isValidNanoTdfBad, _ := sdk.IsValidNanoTdf(inBad) | ||
if _, err := in.Seek(0, io.SeekStart); err != nil { | ||
cmd.PrintErrf("Error seeking to start of file: %v\n", err) | ||
return typeInfos | ||
} | ||
|
||
cmd.Println("Valid NanoTDF (Good): ") | ||
cmd.Println(isValidNanoTdfGood) | ||
isValidNano, err := sdk.IsValidNanoTdf(in) | ||
typeInfos = append(typeInfos, typeInfo{Valid: isValidNano, Error: err, Type: "NanoTDF"}) | ||
|
||
cmd.Println("Valid NanoTDF (Bad): ") | ||
cmd.Println(isValidNanoTdfBad) | ||
if _, err := in.Seek(0, io.SeekStart); err != nil { | ||
cmd.PrintErrf("Error seeking to start of file: %v\n", err) | ||
return typeInfos | ||
} | ||
|
||
_, _ = inGood.Seek(0, io.SeekStart) | ||
tdfType := sdk.GetTdfType(inGood) | ||
cmd.Println("Type: ") | ||
cmd.Println(tdfType.String()) | ||
tdfType := sdk.GetTdfType(in) | ||
typeInfos = append(typeInfos, typeInfo{Valid: tdfType != sdk.Invalid, Error: nil, Type: tdfType.String()}) | ||
|
||
return nil | ||
return typeInfos | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters