package at.gv.egiz.param_tests; import static org.junit.Assert.assertNotNull; import java.io.File; import java.io.FileInputStream; import java.io.FileNotFoundException; import java.io.FileOutputStream; import java.io.IOException; import java.security.cert.CertificateException; import java.util.Map; import java.util.UUID; import org.apache.commons.io.IOUtils; import org.junit.Rule; import org.junit.rules.TestRule; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import at.gv.egiz.param_tests.provider.BaseSignatureDataProvider; import at.gv.egiz.param_tests.provider.BaseSignatureTestData; import at.gv.egiz.pdfas.common.exceptions.PdfAsException; import at.gv.egiz.pdfas.common.utils.StreamUtils; import at.gv.egiz.pdfas.lib.api.ByteArrayDataSink; import at.gv.egiz.pdfas.lib.api.ByteArrayDataSource; import at.gv.egiz.pdfas.lib.api.Configuration; import at.gv.egiz.pdfas.lib.api.DataSource; import at.gv.egiz.pdfas.lib.api.PdfAs; import at.gv.egiz.pdfas.lib.api.PdfAsFactory; import at.gv.egiz.pdfas.lib.api.sign.IPlainSigner; import at.gv.egiz.pdfas.lib.api.sign.SignParameter; import at.gv.egiz.pdfas.sigs.pades.PAdESSigner; import at.gv.egiz.pdfas.sigs.pades.PAdESSignerKeystore; import at.gv.egiz.sl.util.BKUSLConnector; import at.gv.egiz.sl.util.MOAConnector; /** * Base class for parameterized signature tests, which provides fields and * methods common to all parameterized signature test. * * @author mtappler * */ public class SignatureTest { /** * test data common to all test types, like connector type */ protected BaseSignatureTestData baseTestData; /** * positioning string which specifies the position of the signature block */ protected String positionString; /** * logger for this class */ private static final Logger logger = LoggerFactory .getLogger(PDFASignatureTest.class); /** * a JUnit TestWatcher, this object gets notified several times a * subclass of this class is run as test */ @Rule public TestRule watchman = new SignatureTestWatcher(); /** * PdfAs used for signing and creating a configuration */ protected PdfAs pdfAs; /** * the sign parameter used for signing in the test */ protected SignParameter signParameter; /** * the connector used for signing in the test */ protected IPlainSigner slConnector; /** * This method signs a file, all parameter like input file name are specified * through fields of this class. * * @return the signed file File-object * @throws IOException * @throws FileNotFoundException * @throws CertificateException * @throws PdfAsException */ protected File signPDFFile() throws IOException, FileNotFoundException, CertificateException, PdfAsException { File inputFile = new File(baseTestData.getPdfFile()); File outputPdfFile = new File(baseTestData.getOutputFile()); DataSource dataSource = new ByteArrayDataSource( StreamUtils.inputStreamToByteArray(new FileInputStream( inputFile))); ByteArrayDataSink dataSink = new ByteArrayDataSink(); pdfAs = null; pdfAs = PdfAsFactory.createPdfAs(new File(baseTestData .getConfigurationFile())); Configuration configuration = pdfAs.getConfiguration(); signParameter = PdfAsFactory.createSignParameter(configuration, dataSource); String id = UUID.randomUUID().toString(); signParameter.setTransactionId(id); logger.debug("Transaction: " + id); slConnector = null; if (baseTestData.getConnectorData() != null) { if (baseTestData.getConnectorData().getConnectorType() .equals("bku")) { slConnector = new PAdESSigner(new BKUSLConnector(configuration)); } else if (baseTestData.getConnectorData().getConnectorType() .equals("moa")) { slConnector = new PAdESSigner(new MOAConnector(configuration)); } else if (baseTestData.getConnectorData().getConnectorType() .equals("ks")) { Map params = baseTestData.getConnectorData() .getConnectorParameters(); String keystoreFilename = params .get(BaseSignatureDataProvider.KS_FILE_NAME); String keystoreAlias = params .get(BaseSignatureDataProvider.KS_ALIAS); String keystoreType = params .get(BaseSignatureDataProvider.KS_TYPE); String keystoreStorepass = params .get(BaseSignatureDataProvider.KS_PASS); String keystoreKeypass = params .get(BaseSignatureDataProvider.KS_KEY_PASS); assertNotNull( "You need to provide a keystore file if using ks connector", keystoreFilename); assertNotNull( "You need to provide a key alias if using ks connector", keystoreAlias); slConnector = new PAdESSignerKeystore(keystoreFilename, keystoreAlias, keystoreStorepass, keystoreKeypass, keystoreType); } } if (slConnector == null) { slConnector = new PAdESSigner(new BKUSLConnector(configuration)); } signParameter.setOutput(dataSink); signParameter.setPlainSigner(slConnector); signParameter.setDataSource(dataSource); // this is not needed for PDF-A test if (positionString != null) signParameter.setSignaturePosition(positionString); signParameter.setSignatureProfileId(baseTestData.getProfilID()); logger.debug("Starting signature for " + baseTestData.getPdfFile()); logger.debug("Selected signature Profile " + baseTestData.getProfilID()); /* SignResult result = */pdfAs.sign(signParameter); FileOutputStream fos = null; try { fos = new FileOutputStream(outputPdfFile, false); fos.write(dataSink.getData()); fos.close(); } catch (IOException e) { logger.debug("IO exception occured while writing PDF output file", e); throw e; } finally { IOUtils.closeQuietly(fos); } return outputPdfFile; } }