package net.deterlab.abac.regression; import java.io.*; import java.util.*; import net.deterlab.abac.*; public class RocketsTest extends RegressionTest { /** * 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 RocketsTest(String name) { super(name); } /** * 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 warnerBros = new Identity("WarnerBros"); Identity batman = new Identity("Batman"); Identity coyote = new Identity("Coyote"); ArrayList inProof = new ArrayList(); ctxt.load_id_chunk(acme); ctxt.load_id_chunk(warnerBros); ctxt.load_id_chunk(batman); ctxt.load_id_chunk(coyote); Credential c = ctxt.newCredential( new Role(acme.getKeyID() + ".buy_rockets"), new Role(acme.getKeyID() + ".preferred_customer & " + warnerBros.getKeyID() + ".character")); c.make_cert(acme); ctxt.load_attribute_chunk(c); inProof.add(c); c = ctxt.newCredential( new Role(acme.getKeyID() + ".preferred_customer"), new Role(coyote.getKeyID())); c.make_cert(acme); ctxt.load_attribute_chunk(c); inProof.add(c); c = ctxt.newCredential( new Role(acme.getKeyID() + ".preferred_customer"), new Role(batman.getKeyID())); c.make_cert(acme); ctxt.load_attribute_chunk(c); c = ctxt.newCredential( new Role(warnerBros.getKeyID() + ".character"), new Role(coyote.getKeyID())); c.make_cert(warnerBros); ctxt.load_attribute_chunk(c); inProof.add(c); Context.QueryResult q = ctxt.query( acme.getKeyID() + ".buy_rockets", coyote.getKeyID()); if ( !q.getSuccess()) { setReason("Coyote can buy rockets proof failed?"); return false; } if ( inProof.size() != q.getCredentials().size()) { setReason("Coyote can buy rockets proof is the wrong size"); return false; } for (Credential cc: inProof) if ( !q.getCredentials().contains(cc)) { setReason("Coyote can buy rockets proof missing cred "+cc); return false; } q = ctxt.query(acme.getKeyID() + ".buy_rockets", batman.getKeyID()); if ( q.getSuccess()) { setReason("Batman can buy rockets ?"); return false; } } catch (Exception e) { setReason(e.getMessage()); return false; } return true; } }