TokenManagerIssuerImpl.java

//* Licensed Materials - Property of                                  *
//* IBM                                                               *
//* Miracle A/S                                                       *
//* Alexandra Instituttet A/S                                         *
//*                                                                   *
//* eu.abc4trust.pabce.1.34                                           *
//*                                                                   *
//* (C) Copyright IBM Corp. 2014. All Rights Reserved.                *
//* (C) Copyright Miracle A/S, Denmark. 2014. All Rights Reserved.    *
//* (C) Copyright Alexandra Instituttet A/S, Denmark. 2014. All       *
//* Rights Reserved.                                                  *
//* US Government Users Restricted Rights - Use, duplication or       *
//* disclosure restricted by GSA ADP Schedule Contract with IBM Corp. *
//*                                                                   *
//* This file is licensed under the Apache License, Version 2.0 (the  *
//* "License"); you may not use this file except in compliance with   *
//* the License. You may obtain a copy of the License at:             *
//*   http://www.apache.org/licenses/LICENSE-2.0                      *
//* Unless required by applicable law or agreed to in writing,        *
//* software distributed under the License is distributed on an       *
//* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY            *
//* KIND, either express or implied.  See the License for the         *
//* specific language governing permissions and limitations           *
//* under the License.                                                *
//*/**/****************************************************************

package eu.abc4trust.abce.internal.issuer.tokenManagerIssuer;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.ObjectInput;
import java.io.ObjectInputStream;
import java.io.ObjectOutput;
import java.io.ObjectOutputStream;
import java.net.URI;
import java.util.List;
import java.util.UUID;

import javax.xml.bind.DatatypeConverter;

import com.google.inject.Inject;

import eu.abc4trust.xml.IssuanceLogEntry;
import eu.abc4trust.xml.IssuanceToken;
import eu.abc4trust.xml.PseudonymInToken;

public class TokenManagerIssuerImpl implements TokenManagerIssuer {

    private final TokenStorageIssuer storage;
    private static final String URI_PREFIX = "issuanceToken:";
    private static final String URI_PREFIX_LOG_ENTRY = "logEntry:";

    @Inject
    public TokenManagerIssuerImpl(TokenStorageIssuer storage) {
        this.storage = storage;
    }

    @Override @Deprecated
    public IssuanceToken getToken(URI tokenuid) {

        IssuanceToken tokenResult = null;

        try {
            byte[] tokenBytes = this.storage.getToken(tokenuid);
            if(tokenBytes != null) {
                ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(tokenBytes);
                ObjectInput objectInput = new ObjectInputStream(byteArrayInputStream);
                tokenResult = (IssuanceToken)objectInput.readObject();

                // Close the streams..
                objectInput.close();
                byteArrayInputStream.close();
            }

        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }

        return tokenResult;
    }


    @Override
    public boolean isEstablishedPseudonym(PseudonymInToken p) {
        String primaryKey = DatatypeConverter.printBase64Binary(p.getPseudonymValue());
        try {
            return this.storage.checkForPseudonym(primaryKey);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    @Override
    public URI storeToken(IssuanceToken t) {

        URI oldTokenUid = t.getIssuanceTokenDescription().getPresentationTokenDescription().getTokenUID();
        UUID uuid = UUID.randomUUID();
        URI tokenuid = URI.create(URI_PREFIX + uuid);

        // Check if a pseudonym is present in the PresentationToken
        List<PseudonymInToken> pseudonyms = t.getIssuanceTokenDescription().getPresentationTokenDescription().getPseudonym();

        try {
            t.getIssuanceTokenDescription().getPresentationTokenDescription().setTokenUID(tokenuid);
            ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
            ObjectOutput objectOutput = new ObjectOutputStream(byteArrayOutputStream);
            objectOutput.writeObject(t);
            byte[] tokenBytes = byteArrayOutputStream.toByteArray();
            this.storage.addToken(tokenuid, tokenBytes);

            // Close the streams..
            objectOutput.close();
            byteArrayOutputStream.close();
            
            // Token is not immutable, we cannot change the tokenUid
            t.getIssuanceTokenDescription().getPresentationTokenDescription().setTokenUID(oldTokenUid);

            // For faster lookup in isEstablishedPseudonym(): Store the pseudonym value lexical representation of xsd:base64Binary if a pseudonym was present.
            for(PseudonymInToken p: pseudonyms) {
                String primaryKey = DatatypeConverter.printBase64Binary(p.getPseudonymValue());
                this.storage.addPseudonymPrimaryKey(primaryKey);
            }

        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }

        return tokenuid;
    }

    @Override
    public boolean deleteToken(URI tokenuid) {
        try {
            return this.storage.deleteToken(tokenuid);
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    // @Override
    // public URI storeIssuerAttributes(IssuerAttributes iats, URI
    // issuanceLogEntryURI) {
    // // TODO Auto-generated method stub
    // throw new UnsupportedOperationException();
    // }

    @Override
    public IssuanceLogEntry getIssuanceLogEntry(URI issuanceLogEntryUid) {
    	
        IssuanceLogEntry result = null;

        try {
            byte[] entryBytes = this.storage.getIssuanceLogEntry(issuanceLogEntryUid);
            if(entryBytes != null) {
                ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(entryBytes);
                ObjectInput objectInput = new ObjectInputStream(byteArrayInputStream);
                result = (IssuanceLogEntry)objectInput.readObject();

                // Close the streams..
                objectInput.close();
                byteArrayInputStream.close();
            }

        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }

        return result;
    }

    @Override
    public boolean deleteIssuanceLogEntry(URI issuanceDataUid) {
        try {
            return this.storage.deleteIssuanceLogEntry(issuanceDataUid);
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }
    
    @Override
    public URI storeIssuanceLogEntry(IssuanceLogEntry entry){
    	try{
    	    if(entry.getIssuanceLogEntryUID() == null) {
    	      UUID uuid = UUID.randomUUID();
              entry.setIssuanceLogEntryUID(URI.create(URI_PREFIX_LOG_ENTRY + uuid.toString()));
    	    }
    		ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
    		ObjectOutput objectOutput = new ObjectOutputStream(byteArrayOutputStream);
    		objectOutput.writeObject(entry);
    		byte[] data = byteArrayOutputStream.toByteArray();
    		this.storage.addIssuanceLogEntry(entry.getIssuanceLogEntryUID(), data);
    		// 	Close the streams..
    		objectOutput.close();
    		byteArrayOutputStream.close();
    	}catch(Exception e){
    		throw new RuntimeException(e);
    	}
    	return entry.getIssuanceLogEntryUID();
    }

    @Override
    public void addPeudonymForTest(byte[] pseudonymValue) {
      try {
        String primaryKey = DatatypeConverter.printBase64Binary(pseudonymValue);
        storage.addPseudonymPrimaryKey(primaryKey);
      } catch (IOException e) {
        throw new RuntimeException(e);
      }
    }

}