Skip to content

Commit 1fdbb7d

Browse files
committed
Throw AttestationClientException when protocol not valid
1 parent 9f806e6 commit 1fdbb7d

File tree

2 files changed

+16
-1
lines changed

2 files changed

+16
-1
lines changed
Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
package com.uid2.shared.secure;
2+
3+
public enum Protocol {
4+
GCP_OIDC,
5+
AWS_NITRO,
6+
GCP_VMID,
7+
AZURE_CC_ACI,
8+
AZURE_CC_AKS,
9+
10+
11+
}

src/main/java/com/uid2/shared/secure/azurecc/MaaTokenPayload.java

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
package com.uid2.shared.secure.azurecc;
22

3+
import com.uid2.shared.secure.AttestationClientException;
4+
import com.uid2.shared.secure.AttestationException;
35
import lombok.Builder;
46
import lombok.Value;
57

@@ -26,11 +28,13 @@ public boolean isSevSnpVM(){
2628
return SEV_SNP_VM_TYPE.equalsIgnoreCase(attestationType);
2729
}
2830

29-
public boolean isUtilityVMCompliant(){
31+
public boolean isUtilityVMCompliant() {
3032
if (azureProtocol == AZURE_CC_ACI_PROTOCOL) {
3133
return AZURE_COMPLIANT_UVM.equalsIgnoreCase(complianceStatus);
3234
} else if (azureProtocol == AZURE_CC_AKS_PROTOCOL) {
3335
return AZURE_COMPLIANT_UVM_AKS.equalsIgnoreCase(complianceStatus);
36+
} else {
37+
throw new AttestationClientException(String.format("Azure protocol: %s not supported", azureProtocol), INVALID_PROTOCOL)
3438
}
3539
return false;
3640
}

0 commit comments

Comments
 (0)