org.votingsystem.callable.MessageTimeStamper.java Source code

Introduction

Here is the source code for org.votingsystem.callable.MessageTimeStamper.java

Source

package org.votingsystem.callable;

import org.bouncycastle.asn1.ASN1EncodableVector;
import org.bouncycastle.asn1.ASN1OctetString;
import org.bouncycastle.asn1.ASN1Set;
import org.bouncycastle.asn1.DERObject;
import org.bouncycastle.asn1.cms.Attribute;
import org.bouncycastle.asn1.cms.AttributeTable;
import org.bouncycastle.asn1.cms.CMSAttributes;
import org.bouncycastle.cms.CMSSignedData;
import org.bouncycastle.cms.SignerInformation;
import org.bouncycastle.cms.SignerInformationVerifier;
import org.bouncycastle.cms.jcajce.JcaSimpleSignerInfoVerifierBuilder;
import org.bouncycastle.tsp.TimeStampRequest;
import org.bouncycastle.tsp.TimeStampToken;
import org.votingsystem.model.ResponseVS;
import org.votingsystem.signature.smime.SMIMEMessage;
import org.votingsystem.throwable.ExceptionVS;
import org.votingsystem.util.ContentTypeVS;
import org.votingsystem.util.ContextVS;
import org.votingsystem.util.HttpHelper;

import java.security.cert.X509Certificate;
import java.util.Collection;
import java.util.concurrent.Callable;
import java.util.logging.Logger;

/**
* License: https://github.com/votingsystem/votingsystem/wiki/Licencia
*/
public class MessageTimeStamper implements Callable<SMIMEMessage> {

    private static Logger log = Logger.getLogger(MessageTimeStamper.class.getSimpleName());

    private SMIMEMessage smimeMessage;
    private TimeStampToken timeStampToken;
    private TimeStampRequest timeStampRequest;
    private String timeStampServerURL;

    public MessageTimeStamper(SMIMEMessage smimeMessage, String timeStampServerURL) throws Exception {
        this.smimeMessage = smimeMessage;
        this.timeStampRequest = smimeMessage.getTimeStampRequest();
        this.timeStampServerURL = timeStampServerURL;
    }

    public MessageTimeStamper(TimeStampRequest timeStampRequest, String timeStampServerURL) throws Exception {
        this.timeStampRequest = timeStampRequest;
        this.timeStampServerURL = timeStampServerURL;
    }

    @Override
    public SMIMEMessage call() throws Exception {
        ResponseVS responseVS = HttpHelper.getInstance().sendData(timeStampRequest.getEncoded(),
                ContentTypeVS.TIMESTAMP_QUERY, timeStampServerURL);
        if (ResponseVS.SC_OK == responseVS.getStatusCode()) {
            byte[] bytesToken = responseVS.getMessageBytes();
            timeStampToken = new TimeStampToken(new CMSSignedData(bytesToken));
            X509Certificate timeStampCert = ContextVS.getInstance().getTimeStampServerCert();
            if (timeStampCert != null) {
                SignerInformationVerifier timeStampSignerInfoVerifier = new JcaSimpleSignerInfoVerifierBuilder()
                        .build(timeStampCert);
                timeStampToken.validate(timeStampSignerInfoVerifier);
            } else
                log.info("TIMESTAMP RESPONSE NOT VALIDATED");
            if (smimeMessage != null)
                smimeMessage.setTimeStampToken(timeStampToken);
            return smimeMessage;
        } else
            throw new ExceptionVS(responseVS.getMessage());
    }

    public TimeStampToken getTimeStampToken() {
        return timeStampToken;
    }

    public byte[] getDigestToken() {
        if (timeStampToken == null)
            return null;
        CMSSignedData tokenCMSSignedData = timeStampToken.toCMSSignedData();
        Collection signers = tokenCMSSignedData.getSignerInfos().getSigners();
        SignerInformation tsaSignerInfo = (SignerInformation) signers.iterator().next();
        AttributeTable signedAttrTable = tsaSignerInfo.getSignedAttributes();
        ASN1EncodableVector v = signedAttrTable.getAll(CMSAttributes.messageDigest);
        Attribute t = (Attribute) v.get(0);
        ASN1Set attrValues = t.getAttrValues();
        DERObject validMessageDigest = attrValues.getObjectAt(0).getDERObject();
        ASN1OctetString signedMessageDigest = (ASN1OctetString) validMessageDigest;
        byte[] digestToken = signedMessageDigest.getOctets();
        //String digestTokenStr = new String(Base64.encode(digestToken));
        //log.info(" digestTokenStr: " + digestTokenStr);
        return digestToken;
    }

    public SMIMEMessage getSMIME() {
        return smimeMessage;
    }

}