check artifact path when creating entry

Signed-off-by: Samsondeen Dare <samsondeen.dare@hashicorp.com>
sigstore · Aug 11, 2022 · 8fa6ed6 · 8fa6ed6
1 parent 6882d35
commit 8fa6ed6
Show file tree

Hide file tree

Showing 7 changed files with 18 additions and 7 deletions.
diff --git a/cmd/rekor-cli/app/pflag_groups.go b/cmd/rekor-cli/app/pflag_groups.go
@@ -122,10 +122,6 @@ func validateArtifactPFlags(uuidValid, indexValid bool) error {
 		return errors.New("either 'entry' or 'artifact' or 'artifact-hash' must be specified")
 	}
 
-	if viper.GetString("artifact-hash") != "" && viper.GetString("artifact") == "" {
-		return errors.New("'artifact-hash' can only be used with 'artifact'")
-	}
-
 	return nil
 }
 
@@ -142,9 +138,6 @@ func CreatePropsFromPflags() (*types.ArtifactProperties, error) {
 	}
 
 	props.ArtifactHash = viper.GetString("artifact-hash")
-	if props.ArtifactHash != "" && props.ArtifactPath == nil {
-		return nil, errors.New("'artifact-hash' can only be used with 'artifact'")
-	}
 
 	signatureString := viper.GetString("signature")
 	if signatureString != "" {

diff --git a/pkg/types/alpine/v0.0.1/entry.go b/pkg/types/alpine/v0.0.1/entry.go
@@ -299,6 +299,9 @@ func (v V001Entry) CreateFromArtifactProperties(ctx context.Context, props types
 	artifactBytes := props.ArtifactBytes
 	if artifactBytes == nil {
 		var artifactReader io.ReadCloser
+		if props.ArtifactPath == nil {
+			return nil, errors.New("path to artifact file must be specified")
+		}
 		if props.ArtifactPath.IsAbs() {
 			artifactReader, err = util.FileOrURLReadCloser(ctx, props.ArtifactPath.String(), nil)
 			if err != nil {

diff --git a/pkg/types/helm/v0.0.1/entry.go b/pkg/types/helm/v0.0.1/entry.go
@@ -294,6 +294,9 @@ func (v V001Entry) CreateFromArtifactProperties(ctx context.Context, props types
 	artifactBytes := props.ArtifactBytes
 	if artifactBytes == nil {
 		var artifactReader io.ReadCloser
+		if props.ArtifactPath == nil {
+			return nil, errors.New("path to artifact file must be specified")
+		}
 		if props.ArtifactPath.IsAbs() {
 			artifactReader, err = util.FileOrURLReadCloser(ctx, props.ArtifactPath.String(), nil)
 			if err != nil {

diff --git a/pkg/types/jar/v0.0.1/entry.go b/pkg/types/jar/v0.0.1/entry.go
@@ -283,6 +283,9 @@ func (v *V001Entry) CreateFromArtifactProperties(ctx context.Context, props type
 	artifactBytes := props.ArtifactBytes
 	if artifactBytes == nil {
 		var artifactReader io.ReadCloser
+		if props.ArtifactPath == nil {
+			return nil, errors.New("path to artifact file must be specified")
+		}
 		if props.ArtifactPath.IsAbs() {
 			artifactReader, err = util.FileOrURLReadCloser(ctx, props.ArtifactPath.String(), nil)
 			if err != nil {

diff --git a/pkg/types/rekord/v0.0.1/entry.go b/pkg/types/rekord/v0.0.1/entry.go
@@ -343,6 +343,9 @@ func (v V001Entry) CreateFromArtifactProperties(ctx context.Context, props types
 	artifactBytes := props.ArtifactBytes
 	if artifactBytes == nil {
 		var artifactReader io.ReadCloser
+		if props.ArtifactPath == nil {
+			return nil, errors.New("path to artifact file must be specified")
+		}
 		if props.ArtifactPath.IsAbs() {
 			artifactReader, err = util.FileOrURLReadCloser(ctx, props.ArtifactPath.String(), nil)
 			if err != nil {

diff --git a/pkg/types/rpm/v0.0.1/entry.go b/pkg/types/rpm/v0.0.1/entry.go
@@ -319,6 +319,9 @@ func (v V001Entry) CreateFromArtifactProperties(ctx context.Context, props types
 	artifactBytes := props.ArtifactBytes
 	if artifactBytes == nil {
 		var artifactReader io.ReadCloser
+		if props.ArtifactPath == nil {
+			return nil, errors.New("path to artifact file must be specified")
+		}
 		if props.ArtifactPath.IsAbs() {
 			artifactReader, err = util.FileOrURLReadCloser(ctx, props.ArtifactPath.String(), nil)
 			if err != nil {

diff --git a/pkg/types/tuf/v0.0.1/entry.go b/pkg/types/tuf/v0.0.1/entry.go
@@ -306,6 +306,9 @@ func (v V001Entry) CreateFromArtifactProperties(ctx context.Context, props types
 	re.TufObj.Metadata = &models.TUFV001SchemaMetadata{}
 	if artifactBytes == nil {
 		var artifactReader io.ReadCloser
+		if props.ArtifactPath == nil {
+			return nil, errors.New("path to artifact file must be specified")
+		}
 		if props.ArtifactPath.IsAbs() {
 			artifactReader, err = util.FileOrURLReadCloser(ctx, props.ArtifactPath.String(), nil)
 			if err != nil {