eu.europa.esig.dss.pades.signature.PAdESLevelBTest.java Source code

Java tutorial

Introduction

Here is the source code for eu.europa.esig.dss.pades.signature.PAdESLevelBTest.java

Source

/**
 * DSS - Digital Signature Services
 * Copyright (C) 2015 European Commission, provided under the CEF programme
 *
 * This file is part of the "DSS - Digital Signature Services" project.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */
package eu.europa.esig.dss.pades.signature;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.fail;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.InputStream;
import java.math.BigInteger;
import java.security.cert.X509Certificate;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.List;

import javax.crypto.Cipher;

import org.apache.commons.io.IOUtils;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.pdmodel.interactive.digitalsignature.PDSignature;
import org.apache.xml.security.utils.Base64;
import org.bouncycastle.asn1.ASN1Encodable;
import org.bouncycastle.asn1.ASN1InputStream;
import org.bouncycastle.asn1.ASN1Integer;
import org.bouncycastle.asn1.ASN1ObjectIdentifier;
import org.bouncycastle.asn1.ASN1OctetString;
import org.bouncycastle.asn1.ASN1Sequence;
import org.bouncycastle.asn1.ASN1Set;
import org.bouncycastle.asn1.DERTaggedObject;
import org.bouncycastle.asn1.cms.ContentInfo;
import org.bouncycastle.asn1.cms.IssuerAndSerialNumber;
import org.bouncycastle.asn1.cms.SignedData;
import org.bouncycastle.asn1.cms.SignerIdentifier;
import org.bouncycastle.asn1.cms.SignerInfo;
import org.bouncycastle.asn1.pkcs.Attribute;
import org.bouncycastle.asn1.pkcs.PKCSObjectIdentifiers;
import org.bouncycastle.asn1.x509.DigestInfo;
import org.bouncycastle.cert.X509CertificateHolder;
import org.bouncycastle.cert.jcajce.JcaX509CertificateConverter;
import org.bouncycastle.jce.provider.BouncyCastleProvider;
import org.bouncycastle.util.encoders.Hex;
import org.junit.Before;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import eu.europa.esig.dss.DSSDocument;
import eu.europa.esig.dss.DSSUtils;
import eu.europa.esig.dss.DigestAlgorithm;
import eu.europa.esig.dss.EncryptionAlgorithm;
import eu.europa.esig.dss.FileDocument;
import eu.europa.esig.dss.MimeType;
import eu.europa.esig.dss.SignatureAlgorithm;
import eu.europa.esig.dss.SignatureLevel;
import eu.europa.esig.dss.SignaturePackaging;
import eu.europa.esig.dss.pades.PAdESSignatureParameters;
import eu.europa.esig.dss.signature.AbstractTestSignature;
import eu.europa.esig.dss.signature.DocumentSignatureService;
import eu.europa.esig.dss.test.gen.CertificateService;
import eu.europa.esig.dss.test.mock.MockPrivateKeyEntry;
import eu.europa.esig.dss.validation.CertificateVerifier;
import eu.europa.esig.dss.validation.CommonCertificateVerifier;

public class PAdESLevelBTest extends AbstractTestSignature {

    private static final Logger logger = LoggerFactory.getLogger(PAdESLevelBTest.class);

    private DocumentSignatureService<PAdESSignatureParameters> service;
    private PAdESSignatureParameters signatureParameters;
    private DSSDocument documentToSign;
    private MockPrivateKeyEntry privateKeyEntry;

    @Before
    public void init() throws Exception {
        documentToSign = new FileDocument(new File("src/test/resources/sample.pdf"));

        CertificateService certificateService = new CertificateService();
        privateKeyEntry = certificateService.generateCertificateChain(SignatureAlgorithm.RSA_SHA256);

        signatureParameters = new PAdESSignatureParameters();
        signatureParameters.bLevel().setSigningDate(new Date());
        signatureParameters.setSigningCertificate(privateKeyEntry.getCertificate());
        signatureParameters.setCertificateChain(privateKeyEntry.getCertificateChain());
        signatureParameters.setSignaturePackaging(SignaturePackaging.ENVELOPING);
        signatureParameters.setSignatureLevel(SignatureLevel.PAdES_BASELINE_B);
        signatureParameters.setLocation("Luxembourg");
        signatureParameters.setReason("DSS testing");
        signatureParameters.setContactInfo("Jira");

        CertificateVerifier certificateVerifier = new CommonCertificateVerifier();
        service = new PAdESService(certificateVerifier);
    }

    @Override
    protected void onDocumentSigned(byte[] byteArray) {
        try {
            InputStream inputStream = new ByteArrayInputStream(byteArray);

            PDDocument document = PDDocument.load(inputStream);
            List<PDSignature> signatures = document.getSignatureDictionaries();
            assertEquals(1, signatures.size());

            for (PDSignature pdSignature : signatures) {
                byte[] contents = pdSignature.getContents(byteArray);
                byte[] signedContent = pdSignature.getSignedContent(byteArray);

                logger.info("Byte range : " + Arrays.toString(pdSignature.getByteRange()));

                //IOUtils.write(contents, new FileOutputStream("sig.p7s"));

                ASN1InputStream asn1sInput = new ASN1InputStream(contents);
                ASN1Sequence asn1Seq = (ASN1Sequence) asn1sInput.readObject();

                logger.info("SEQ : " + asn1Seq.toString());

                ASN1ObjectIdentifier oid = ASN1ObjectIdentifier.getInstance(asn1Seq.getObjectAt(0));
                assertEquals(PKCSObjectIdentifiers.signedData, oid);

                SignedData signedData = SignedData
                        .getInstance(DERTaggedObject.getInstance(asn1Seq.getObjectAt(1)).getObject());

                ASN1Set digestAlgorithmSet = signedData.getDigestAlgorithms();
                ASN1ObjectIdentifier oidDigestAlgo = ASN1ObjectIdentifier
                        .getInstance(ASN1Sequence.getInstance(digestAlgorithmSet.getObjectAt(0)).getObjectAt(0));
                DigestAlgorithm digestAlgorithm = DigestAlgorithm.forOID(oidDigestAlgo.getId());
                logger.info("DIGEST ALGO : " + digestAlgorithm);

                ContentInfo encapContentInfo = signedData.getEncapContentInfo();
                ASN1ObjectIdentifier contentTypeOID = encapContentInfo.getContentType();
                logger.info("ENCAPSULATED CONTENT INFO TYPE : " + contentTypeOID);
                assertEquals(PKCSObjectIdentifiers.data, contentTypeOID);

                ASN1Encodable content = encapContentInfo.getContent();
                logger.info("ENCAPSULATED CONTENT INFO CONTENT : " + content);
                assertNull(content);

                List<X509Certificate> certificates = extractCertificates(signedData);

                ASN1Set signerInfosAsn1 = signedData.getSignerInfos();
                logger.info("SIGNER INFO ASN1 : " + signerInfosAsn1.toString());
                SignerInfo signedInfo = SignerInfo
                        .getInstance(ASN1Sequence.getInstance(signerInfosAsn1.getObjectAt(0)));

                ASN1Set authenticatedAttributeSet = signedInfo.getAuthenticatedAttributes();
                logger.info("AUTHENTICATED ATTR : " + authenticatedAttributeSet);

                List<ASN1ObjectIdentifier> attributeOids = new ArrayList<ASN1ObjectIdentifier>();
                for (int i = 0; i < authenticatedAttributeSet.size(); i++) {
                    Attribute attribute = Attribute.getInstance(authenticatedAttributeSet.getObjectAt(i));
                    attributeOids.add(attribute.getAttrType());
                }
                logger.info("List of OID for Auth Attrb : " + attributeOids);

                Attribute attributeDigest = Attribute.getInstance(authenticatedAttributeSet.getObjectAt(1));
                assertEquals(PKCSObjectIdentifiers.pkcs_9_at_messageDigest, attributeDigest.getAttrType());

                ASN1OctetString asn1ObjString = ASN1OctetString
                        .getInstance(attributeDigest.getAttrValues().getObjectAt(0));
                String embeddedDigest = Base64.encode(asn1ObjString.getOctets());
                logger.info("MESSAGE DIGEST : " + embeddedDigest);

                byte[] digestSignedContent = DSSUtils.digest(digestAlgorithm, signedContent);
                String computedDigestSignedContentEncodeBase64 = Base64.encode(digestSignedContent);
                logger.info("COMPUTED DIGEST SIGNED CONTENT BASE64 : " + computedDigestSignedContentEncodeBase64);
                assertEquals(embeddedDigest, computedDigestSignedContentEncodeBase64);

                SignerIdentifier sid = signedInfo.getSID();
                logger.info("SIGNER IDENTIFIER : " + sid.getId());

                IssuerAndSerialNumber issuerAndSerialNumber = IssuerAndSerialNumber
                        .getInstance(signedInfo.getSID());
                ASN1Integer signerSerialNumber = issuerAndSerialNumber.getSerialNumber();
                logger.info("ISSUER AND SN : " + issuerAndSerialNumber.getName() + " " + signerSerialNumber);

                BigInteger serial = issuerAndSerialNumber.getSerialNumber().getValue();
                X509Certificate signerCertificate = null;
                for (X509Certificate x509Certificate : certificates) {
                    if (serial.equals(x509Certificate.getSerialNumber())) {
                        signerCertificate = x509Certificate;
                    }
                }
                assertNotNull(signerCertificate);

                String algorithm = signerCertificate.getPublicKey().getAlgorithm();
                EncryptionAlgorithm encryptionAlgorithm = EncryptionAlgorithm.forName(algorithm);

                ASN1OctetString encryptedInfoOctedString = signedInfo.getEncryptedDigest();
                String signatureValue = Hex.toHexString(encryptedInfoOctedString.getOctets());

                logger.info("SIGNATURE VALUE : " + signatureValue);

                Cipher cipher = Cipher.getInstance(encryptionAlgorithm.getName());
                cipher.init(Cipher.DECRYPT_MODE, signerCertificate);
                byte[] decrypted = cipher.doFinal(encryptedInfoOctedString.getOctets());

                ASN1InputStream inputDecrypted = new ASN1InputStream(decrypted);

                ASN1Sequence seqDecrypt = (ASN1Sequence) inputDecrypted.readObject();
                logger.info("DECRYPTED : " + seqDecrypt);

                DigestInfo digestInfo = new DigestInfo(seqDecrypt);
                assertEquals(oidDigestAlgo, digestInfo.getAlgorithmId().getAlgorithm());

                String decryptedDigestEncodeBase64 = Base64.encode(digestInfo.getDigest());
                logger.info("DECRYPTED BASE64 : " + decryptedDigestEncodeBase64);

                byte[] encoded = authenticatedAttributeSet.getEncoded();
                byte[] digest = DSSUtils.digest(digestAlgorithm, encoded);
                String computedDigestFromSignatureEncodeBase64 = Base64.encode(digest);
                logger.info("COMPUTED DIGEST FROM SIGNATURE BASE64 : " + computedDigestFromSignatureEncodeBase64);

                assertEquals(decryptedDigestEncodeBase64, computedDigestFromSignatureEncodeBase64);

                IOUtils.closeQuietly(inputDecrypted);
                IOUtils.closeQuietly(asn1sInput);
            }

            IOUtils.closeQuietly(inputStream);
            document.close();
        } catch (Exception e) {
            logger.error(e.getMessage(), e);
            fail(e.getMessage());
        }
    }

    private List<X509Certificate> extractCertificates(SignedData signedData) throws Exception {
        ASN1Set certificates = signedData.getCertificates();
        logger.info("CERTIFICATES (" + certificates.size() + ") : " + certificates);

        List<X509Certificate> foundCertificates = new ArrayList<X509Certificate>();
        for (int i = 0; i < certificates.size(); i++) {
            ASN1Sequence seqCertif = ASN1Sequence.getInstance(certificates.getObjectAt(i));

            X509CertificateHolder certificateHolder = new X509CertificateHolder(seqCertif.getEncoded());
            X509Certificate certificate = new JcaX509CertificateConverter()
                    .setProvider(BouncyCastleProvider.PROVIDER_NAME).getCertificate(certificateHolder);

            foundCertificates.add(certificate);
        }
        return foundCertificates;
    }

    @Override
    protected DocumentSignatureService<PAdESSignatureParameters> getService() {
        return service;
    }

    @Override
    protected PAdESSignatureParameters getSignatureParameters() {
        return signatureParameters;
    }

    @Override
    protected MimeType getExpectedMime() {
        return MimeType.PDF;
    }

    @Override
    protected boolean isBaselineT() {
        return false;
    }

    @Override
    protected boolean isBaselineLTA() {
        return false;
    }

    @Override
    protected DSSDocument getDocumentToSign() {
        return documentToSign;
    }

    @Override
    protected MockPrivateKeyEntry getPrivateKeyEntry() {
        return privateKeyEntry;
    }

}