Skip to content

Commit 8b3af91

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

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

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)