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

Introduction

Here is the source code for eu.europa.esig.dss.pades.signature.PAdESLevelLTTest.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.assertTrue;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.util.Date;
import java.util.List;

import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.pdmodel.interactive.digitalsignature.PDSignature;
import org.junit.Before;

import eu.europa.esig.dss.DSSDocument;
import eu.europa.esig.dss.DSSException;
import eu.europa.esig.dss.DSSUtils;
import eu.europa.esig.dss.DigestAlgorithm;
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.pades.PAdESSignatureParameters;
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.test.mock.MockTSPSource;
import eu.europa.esig.dss.utils.Utils;
import eu.europa.esig.dss.validation.CertificateVerifier;
import eu.europa.esig.dss.validation.CommonCertificateVerifier;

public class PAdESLevelLTTest extends AbstractPAdESTestSignature {

    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.setSignatureLevel(SignatureLevel.PAdES_BASELINE_LT);

        CertificateVerifier certificateVerifier = new CommonCertificateVerifier();
        service = new PAdESService(certificateVerifier);
        service.setTspSource(
                new MockTSPSource(certificateService.generateTspCertificate(SignatureAlgorithm.RSA_SHA1)));
    }

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

            PDDocument pdDoc = PDDocument.load(bais);
            List<PDSignature> sigs = pdDoc.getSignatureDictionaries();
            PDSignature pdSignature = sigs.get(0);
            byte[] contents = pdSignature.getContents(byteArray);

            byte[] digest = DSSUtils.digest(DigestAlgorithm.SHA1, contents);
            String hex = Utils.toHex(digest);

            String pdfString = new String(byteArray, "UTF-8");
            assertTrue(pdfString.contains(Utils.upperCase(hex)));
        } catch (Exception e) {
            throw new DSSException(e);
        }
    }

    @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 true;
    }

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

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

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

}