source: java/net/deterlab/abac/regression/WriteCreds.java @ d2b198c

package net.deterlab.abac.regression;

import java.io.*;
import java.util.*;

import net.deterlab.abac.*;

public class WriteCreds extends RegressionTest {
    protected String className;
    /**
     * Put out an identity and key into dir named by the name of the Identity
     * @param i the Identity to write
     * @param dir a File holding the destination directory
     * @throws IOException if the file writing fails
     */
    public void writeCombinedIdentity(Identity i, File dir) 
            throws IOException {
        FileOutputStream f = new FileOutputStream(
                new File(dir,  i.getName() + ".pem"));

        i.write(f);
        i.writePrivateKey(f);
    }
    /**
     * Create a new Credential/Identity writing test for the given class of
     * credentials.
     * @param cn a String containing the binary name of the class to test
     */
     public WriteCreds(String cn) { 
         super(cn.substring(
                     cn.lastIndexOf('.') == -1 ? 0 : cn.lastIndexOf('.') +1 ) + 
                 " credential writing test"); 
         className = cn;
     }

    /**
     * Carry out the test.  Create credentials for the create_experiment
     * example, run the proof and make sure the correct proof is generated.
     * @param data a File pointing to a directory that contains files the test
     * may need
     * @param scratch a File pointing to a directory that the test can use to
     * store data
     * @return a boolean, true if the test is passed
     */
    public boolean runTest(File data, File scratch) {
        try {
            Context ctxt = new Context();
            Identity acme = new Identity("Acme");
            Identity globotron = new Identity("Globotron");
            Identity alice = new Identity("Alice");
            Identity bob = new Identity("Bob");
            Vector<Identity> ids = new Vector<Identity>();
            Vector<Credential> creds = new Vector<Credential>();
            Collections.addAll(ids, acme, globotron, alice, bob);

            for ( Identity i: ids) 
                writeCombinedIdentity(i, scratch);

            ctxt.setCredentialFactory(new CredentialFactory(
                        new String[] { className}));

            for (Identity id: new Identity[] { acme, globotron, alice, bob } ) 
                ctxt.load_id_chunk(id.getCertificate());

            Credential c = ctxt.newCredential(
                    new Role(acme.getKeyID() + ".experiment_create"),
                    new Role(acme.getKeyID() + ".partner.experiment_create"));
            c.make_cert(acme);
            creds.add(c);
            c = ctxt.newCredential(
                    new Role(acme.getKeyID() + ".partner"),
                    new Role(globotron.getKeyID()));
            c.make_cert(acme);
            creds.add(c);
            c = ctxt.newCredential(
                    new Role(globotron.getKeyID() + ".experiment_create"),
                    new Role(globotron.getKeyID() + ".admin.power_user"));
            c.make_cert(globotron);
            creds.add(c);
            c = ctxt.newCredential(
                    new Role(globotron.getKeyID() + ".admin"),
                    new Role(alice.getKeyID()));
            c.make_cert(globotron);
            creds.add(c);
            c = ctxt.newCredential(
                    new Role(alice.getKeyID() + ".power_user"),
                    new Role(bob.getKeyID()));
            c.make_cert(alice);
            creds.add(c);

            int i =0;
            for (Credential cc: creds) {
                cc.write(new File(scratch,"e" + i + cc.getSuffix()).toString());
                i++;
            }

            Context ctxt2 = new Context();

            ctxt2.load_directory(scratch);
            Collection<Identity> nids = ctxt2.identities();
            Collection<Credential> ncreds = ctxt2.credentials();

            if ( nids.size() != ids.size()) {
                setReason("Different number of identities read " + nids.size() +
                        " expected " + ids.size());
                return false;
            }
            for (Identity ii: ids ) 
                if ( !nids.contains(ii)) {
                    setReason("Identity " + ii + " not read successfully");
                    return false;
                }

            if ( ncreds.size() != creds.size()) {
                setReason("Different number of credentials read " + 
                        ncreds.size() + " expecting " + creds.size());
                return false;
            }
            for (Credential cc: creds ) 
                if ( !ncreds.contains(cc)) {
                    setReason("Credential " + cc + " not read successfully");
                    return false;
                } 

        }
        catch (Exception e) {
            setReason(e.getMessage());
            return false;
        }
        return true;
    }
}