Skip to content

Commit c0056d8

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

File tree

2 files changed

+17
-1
lines changed

2 files changed

+17
-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: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,8 @@
11
package com.uid2.shared.secure.azurecc;
22

3+
import com.uid2.shared.secure.AttestationClientException;
4+
import com.uid2.shared.secure.AttestationException;
5+
import com.uid2.shared.secure.AttestationFailure;
36
import lombok.Builder;
47
import lombok.Value;
58

@@ -26,11 +29,13 @@ public boolean isSevSnpVM(){
2629
return SEV_SNP_VM_TYPE.equalsIgnoreCase(attestationType);
2730
}
2831

29-
public boolean isUtilityVMCompliant(){
32+
public boolean isUtilityVMCompliant() throws AttestationClientException {
3033
if (azureProtocol == AZURE_CC_ACI_PROTOCOL) {
3134
return AZURE_COMPLIANT_UVM.equalsIgnoreCase(complianceStatus);
3235
} else if (azureProtocol == AZURE_CC_AKS_PROTOCOL) {
3336
return AZURE_COMPLIANT_UVM_AKS.equalsIgnoreCase(complianceStatus);
37+
} else {
38+
throw new AttestationClientException(String.format("Azure protocol: %s not supported", azureProtocol), AttestationFailure.INVALID_PROTOCOL);
3439
}
3540
return false;
3641
}

0 commit comments

Comments
 (0)