-
Notifications
You must be signed in to change notification settings - Fork 58
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #580 from nsacyber/v3_issue-579
[#579] MISC ACA issues fixed
- Loading branch information
Showing
7 changed files
with
211 additions
and
142 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
165 changes: 165 additions & 0 deletions
165
HIRS_AttestationCA/src/main/java/hirs/attestationca/persist/util/CredentialHelper.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,165 @@ | ||
| package hirs.attestationca.persist.util; | ||
|
|
||
| import hirs.attestationca.persist.entity.userdefined.certificate.CertificateVariables; | ||
| import lombok.AccessLevel; | ||
| import lombok.NoArgsConstructor; | ||
| import lombok.extern.log4j.Log4j2; | ||
| import org.bouncycastle.util.encoders.Base64; | ||
|
|
||
| import java.nio.ByteBuffer; | ||
| import java.util.Arrays; | ||
| import java.util.Collections; | ||
| import java.util.List; | ||
| import java.util.ListIterator; | ||
|
|
||
| @Log4j2 | ||
| @NoArgsConstructor(access = AccessLevel.PRIVATE) | ||
| public final class CredentialHelper { | ||
|
|
||
| public static boolean isPEM(final String possiblePEM) { | ||
| return possiblePEM.contains(CertificateVariables.PEM_HEADER) | ||
| || possiblePEM.contains(CertificateVariables.PEM_ATTRIBUTE_HEADER); | ||
| } | ||
|
|
||
| public static boolean isMultiPEM(final String possiblePEM) { | ||
| boolean multiPem = false; | ||
| int iniIndex = possiblePEM.indexOf(CertificateVariables.PEM_HEADER); | ||
| if (iniIndex >= 0) { | ||
| iniIndex = possiblePEM.indexOf(CertificateVariables.PEM_HEADER, | ||
| iniIndex + CertificateVariables.PEM_HEADER.length()); | ||
| if (iniIndex > 1) { | ||
| multiPem = true; | ||
| } | ||
| } | ||
| return multiPem; | ||
| } | ||
|
|
||
| public static byte[] stripPemHeaderFooter(final String pemFile) { | ||
| String strippedFile; | ||
| strippedFile = pemFile.replace(CertificateVariables.PEM_HEADER, ""); | ||
| strippedFile = strippedFile.replace(CertificateVariables.PEM_FOOTER, ""); | ||
| strippedFile = strippedFile.replace(CertificateVariables.PEM_ATTRIBUTE_HEADER, ""); | ||
| strippedFile = strippedFile.replace(CertificateVariables.PEM_ATTRIBUTE_FOOTER, ""); | ||
| return Base64.decode(strippedFile); | ||
| } | ||
|
|
||
| @SuppressWarnings("magicnumber") | ||
| public static byte[] trimCertificate(final byte[] certificateBytes) { | ||
| int certificateStart = 0; | ||
| int certificateLength = 0; | ||
| ByteBuffer certificateByteBuffer = ByteBuffer.wrap(certificateBytes); | ||
|
|
||
| StringBuilder malformedCertStringBuilder = new StringBuilder(CertificateVariables.MALFORMED_CERT_MESSAGE); | ||
| while (certificateByteBuffer.hasRemaining()) { | ||
| // Check if there isn't an ASN.1 structure in the provided bytes | ||
| if (certificateByteBuffer.remaining() <= 2) { | ||
| throw new IllegalArgumentException(malformedCertStringBuilder | ||
| .append(" No certificate length field could be found.").toString()); | ||
| } | ||
|
|
||
| // Look for first ASN.1 Sequence marked by the two bytes (0x30) and (0x82) | ||
| // The check advances our position in the ByteBuffer by one byte | ||
| int currentPosition = certificateByteBuffer.position(); | ||
| if (certificateByteBuffer.get() == (byte) 0x30 | ||
| && certificateByteBuffer.get(currentPosition + 1) == (byte) 0x82) { | ||
| // Check if we have anything more in the buffer than an ASN.1 Sequence header | ||
| if (certificateByteBuffer.remaining() <= 3) { | ||
| throw new IllegalArgumentException(malformedCertStringBuilder | ||
| .append(" Certificate is nothing more than ASN.1 Sequence.") | ||
| .toString()); | ||
| } | ||
| // Mark the start of the first ASN.1 Sequence / Certificate Body | ||
| certificateStart = currentPosition; | ||
|
|
||
| // Parse the length as the 2-bytes following the start of the ASN.1 Sequence | ||
| certificateLength = Short.toUnsignedInt( | ||
| certificateByteBuffer.getShort(currentPosition + 2)); | ||
| // Add the 4 bytes that comprise the start of the ASN.1 Sequence and the length | ||
| certificateLength += 4; | ||
| break; | ||
| } | ||
| } | ||
|
|
||
| if (certificateStart + certificateLength > certificateBytes.length) { | ||
| throw new IllegalArgumentException(malformedCertStringBuilder | ||
| .append(" Value of certificate length field extends beyond length") | ||
| .append(" of provided certificate.").toString()); | ||
| } | ||
| // Return bytes representing the main certificate body | ||
| return Arrays.copyOfRange(certificateBytes, certificateStart, | ||
| certificateStart + certificateLength); | ||
| } | ||
|
|
||
| /** | ||
| * Return the string associated with the boolean slot. | ||
| * @param bit associated with the location in the array. | ||
| * @return string value of the bit set. | ||
| */ | ||
| public static String getKeyUsageString(final int bit) { | ||
| String tempStr = ""; | ||
|
|
||
| switch (bit) { | ||
| case CertificateVariables.KEY_USAGE_BIT0: | ||
| tempStr = String.format("%s%n", CertificateVariables.KEY_USAGE_DS); | ||
| break; | ||
| case CertificateVariables.KEY_USAGE_BIT1: | ||
| tempStr = String.format("%s%n", CertificateVariables.KEY_USAGE_NR); | ||
| break; | ||
| case CertificateVariables.KEY_USAGE_BIT2: | ||
| tempStr = String.format("%s%n", CertificateVariables.KEY_USAGE_KE); | ||
| break; | ||
| case CertificateVariables.KEY_USAGE_BIT3: | ||
| tempStr = String.format("%s%n", CertificateVariables.KEY_USAGE_DE); | ||
| break; | ||
| case CertificateVariables.KEY_USAGE_BIT4: | ||
| tempStr = String.format("%s%n", CertificateVariables.KEY_USAGE_KA); | ||
| break; | ||
| case CertificateVariables.KEY_USAGE_BIT5: | ||
| tempStr = String.format("%s%n", CertificateVariables.KEY_USAGE_KC); | ||
| break; | ||
| case CertificateVariables.KEY_USAGE_BIT6: | ||
| tempStr = String.format("%s%n", CertificateVariables.KEY_USAGE_CS); | ||
| break; | ||
| case CertificateVariables.KEY_USAGE_BIT7: | ||
| tempStr = String.format("%s%n", CertificateVariables.KEY_USAGE_EO); | ||
| break; | ||
| case CertificateVariables.KEY_USAGE_BIT8: | ||
| tempStr = String.format("%s%n", CertificateVariables.KEY_USAGE_DO); | ||
| break; | ||
| default: | ||
| break; | ||
| } | ||
|
|
||
| return tempStr; | ||
| } | ||
|
|
||
| /** | ||
| * This method is to take the DNs from certificates and sort them in an order | ||
| * that will be used to lookup issuer certificates. This will not be stored in | ||
| * the certificate, just the DB for lookup. | ||
| * @param distinguishedName the original DN string. | ||
| * @return a modified string of sorted DNs | ||
| */ | ||
| public static String parseSortDNs(final String distinguishedName) { | ||
| StringBuilder sb = new StringBuilder(); | ||
| String dnsString; | ||
|
|
||
| if (distinguishedName == null || distinguishedName.isEmpty()) { | ||
| sb.append("BLANK"); | ||
| } else { | ||
| dnsString = distinguishedName.trim(); | ||
| dnsString = dnsString.toLowerCase(); | ||
| List<String> dnValArray = Arrays.asList(dnsString.split(",")); | ||
| Collections.sort(dnValArray); | ||
| ListIterator<String> dnListIter = dnValArray.listIterator(); | ||
| while (dnListIter.hasNext()) { | ||
| sb.append(dnListIter.next()); | ||
| if (dnListIter.hasNext()) { | ||
| sb.append(","); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| return sb.toString(); | ||
| } | ||
| } |
Oops, something went wrong.