cli: exit set loop early in fatal errors

edgelesssys · Feb 12, 2024 · 6e0fb83 · 6e0fb83
1 parent a8c0e0b
commit 6e0fb83
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/cli/set.go b/cli/set.go
@@ -18,6 +18,8 @@ import (
 	"github.com/edgelesssys/nunki/internal/manifest"
 	"github.com/edgelesssys/nunki/internal/spinner"
 	"github.com/spf13/cobra"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
 )
 
 func newSetCmd() *cobra.Command {
@@ -178,6 +180,14 @@ func setLoop(
 		if rpcErr == nil {
 			return resp, nil
 		}
+		grpcSt, ok := status.FromError(rpcErr)
+		if ok {
+			switch grpcSt.Code() {
+			case codes.PermissionDenied, codes.InvalidArgument:
+				// These errors are not retryable
+				return nil, rpcErr
+			}
+		}
 		timer := time.NewTimer(1 * time.Second)
 		select {
 		case <-ctx.Done():