de.rub.nds.x509attacker.x509.preparator.GeneralNamePreparator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of x509-attacker Show documentation

X.509-Attacker is a tool based on ASN.1 Tool for creating arbitrary certificates; including especially invalid and malformed certificates. Since X.509 certificates encode their contents in ASN.1, this tool extends the features of ASN.1 Tool in terms of certificate signing. Also, X.509-Attacker introduces a feature of referencing XML elements in order to avoid redundancies when defining certificates in XML.

The newest version!

/*
 * X.509-Attacker - A Library for Arbitrary X.509 Certificates
 *
 * Copyright 2014-2023 Ruhr University Bochum, Paderborn University, Technology Innovation Institute, and Hackmanit GmbH
 *
 * Licensed under Apache License, Version 2.0
 * http://www.apache.org/licenses/LICENSE-2.0.txt
 */
package de.rub.nds.x509attacker.x509.preparator;

import de.rub.nds.asn1.model.Asn1Field;
import de.rub.nds.asn1.model.Asn1Ia5String;
import de.rub.nds.asn1.model.Asn1ObjectIdentifier;
import de.rub.nds.asn1.model.Asn1OctetString;
import de.rub.nds.asn1.oid.ObjectIdentifier;
import de.rub.nds.asn1.preparator.Asn1PreparatorHelper;
import de.rub.nds.x509attacker.chooser.X509Chooser;
import de.rub.nds.x509attacker.x509.model.GeneralName;
import de.rub.nds.x509attacker.x509.model.X509Component;

public class GeneralNamePreparator implements X509Preparator {

    private final GeneralName generalName;
    private final X509Chooser chooser;

    public GeneralNamePreparator(X509Chooser chooser, GeneralName generalName) {
        this.generalName = generalName;
        this.chooser = chooser;
    }

    @Override
    public void prepare() {
        switch (generalName.getGeneralNameChoiceTypeConfig()) {
            case DIRECTORY_NAME:
                generalName.makeSelection(generalName.getDirectoryName());
                break;
            case DNS_NAME:
                generalName.makeSelection(generalName.getDnsName());
                break;
            case EDI_PARTY_NAME:
                generalName.makeSelection(generalName.getEdiPartyName());
                break;
            case IP_ADDRESS:
                generalName.makeSelection(generalName.getIpAddress());
                break;
            case OTHER_NAME:
                generalName.makeSelection(generalName.getOtherName());
                break;
            case REGISTERED_ID:
                generalName.makeSelection(generalName.getRegisteredId());
                break;
            case RFC822_NAME:
                generalName.makeSelection(generalName.getRfc822Name());
                break;
            case UNIFORM_RESOURCE_IDENTIFIER:
                generalName.makeSelection(generalName.getUniformResourceIdentifier());
                break;
            case X400_ADDRESS:
                generalName.makeSelection(generalName.getX400Address());
                break;
            default:
                throw new UnsupportedOperationException(
                        "GeneralNameChoiceType "
                                + generalName.getGeneralNameChoiceTypeConfig().name()
                                + " not yet implemented.");
        }
        if (generalName instanceof X509Component) {
            ((X509Component) generalName).getPreparator(chooser).prepare();
        } else if (generalName.getSelectedChoice() instanceof Asn1Field) {
            HelperPreparator preparator =
                    new HelperPreparator<>(
                            chooser,
                            (Asn1Field) generalName.getSelectedChoice(),
                            generalName.getGeneralNameConfigValue());
            preparator.prepare();
            // TODO we are not adjusting the context here
        } else {
            throw new UnsupportedOperationException(
                    "GeneralName only supports Asn1Field and X509 Components at the time");
        }
    }

    /** Small hack to get access to helper function */
    private class HelperPreparator extends X509Asn1FieldPreparator {

        private Object value;

        public HelperPreparator(X509Chooser chooser, T type, Object value) {
            super(chooser, type);
            this.value = value;
        }

        @Override
        protected byte[] encodeContent() {
            if (field instanceof Asn1Ia5String) {
                Asn1PreparatorHelper.prepareField(((Asn1Ia5String) field), (String) value);
            } else if (field instanceof Asn1ObjectIdentifier) {
                Asn1PreparatorHelper.prepareField(
                        ((Asn1ObjectIdentifier) field), new ObjectIdentifier((String) value));
            } else if (field instanceof Asn1OctetString) {
                Asn1PreparatorHelper.prepareField(((Asn1OctetString) field), (byte[]) value);
            } else {
                throw new UnsupportedOperationException(
                        "Unimplemented Asn1Field: " + field.getClass().getName());
            }
            return field.getContent().getOriginalValue();
        }
    }
}