Throw AttestationClientException when protocol not valid

IABTechLab · Feb 5, 2025 · 1fdbb7d · 1fdbb7d
1 parent 9f806e6
commit 1fdbb7d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/src/main/java/com/uid2/shared/secure/Protocol.java b/src/main/java/com/uid2/shared/secure/Protocol.java
@@ -0,0 +1,11 @@
+package com.uid2.shared.secure;
+
+public enum Protocol {
+    GCP_OIDC,
+    AWS_NITRO,
+    GCP_VMID,
+    AZURE_CC_ACI,
+    AZURE_CC_AKS,
+
+
+}
diff --git a/src/main/java/com/uid2/shared/secure/azurecc/MaaTokenPayload.java b/src/main/java/com/uid2/shared/secure/azurecc/MaaTokenPayload.java
@@ -1,5 +1,7 @@
 package com.uid2.shared.secure.azurecc;
 
+import com.uid2.shared.secure.AttestationClientException;
+import com.uid2.shared.secure.AttestationException;
 import lombok.Builder;
 import lombok.Value;
 
@@ -26,11 +28,13 @@ public boolean isSevSnpVM(){
         return SEV_SNP_VM_TYPE.equalsIgnoreCase(attestationType);
     }
 
-    public boolean isUtilityVMCompliant(){
+    public boolean isUtilityVMCompliant() {
         if (azureProtocol == AZURE_CC_ACI_PROTOCOL) {
             return AZURE_COMPLIANT_UVM.equalsIgnoreCase(complianceStatus);
         } else if (azureProtocol == AZURE_CC_AKS_PROTOCOL) {
             return AZURE_COMPLIANT_UVM_AKS.equalsIgnoreCase(complianceStatus);
+        } else {
+            throw new AttestationClientException(String.format("Azure protocol: %s not supported", azureProtocol), INVALID_PROTOCOL)
         }
         return false;
     }