/*
* File: TestXacmlRequest.java
*
* Copyright 2007 Macquarie E-Learning Centre Of Excellence
*
* 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 org.fcrepo.server.security.xacml.test;
import java.io.File;
import java.io.FileInputStream;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Scanner;
import org.fcrepo.server.security.xacml.util.ContextUtil;
import org.fcrepo.server.security.xacml.util.RIRelationshipResolver;
import org.fcrepo.server.security.xacml.util.RelationshipResolver;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.sun.xacml.attr.AnyURIAttribute;
import com.sun.xacml.attr.AttributeValue;
import com.sun.xacml.attr.StringAttribute;
import com.sun.xacml.ctx.RequestCtx;
/**
* @author nishen@melcoe.mq.edu.au
*/
public class TestXacmlRequest {
private static final Logger logger =
LoggerFactory.getLogger(TestXacmlRequest.class);
private static ContextHandler contextHandler = null;
private static ContextUtil contextUtil = null;
private static RelationshipResolver RELATIONSHIP_RESOLVER = null;
static {
try{
RELATIONSHIP_RESOLVER = new RIRelationshipResolver(null, new HashMap<String,String>());
} catch (Exception e) {}
}
public static void main(String[] args) throws Exception {
contextHandler = new ContextHandler();
contextUtil = new ContextUtil();
StringBuilder request = new StringBuilder();
if (args.length > 0) {
File reqFile = new File(args[0]);
if (logger.isDebugEnabled()) {
logger.debug("Using request file: " + reqFile.getAbsolutePath());
}
Scanner scanner = new Scanner(new FileInputStream(reqFile));
while (scanner.hasNextLine()) {
request.append(scanner.nextLine());
}
testRequest(request.toString());
} else {
testRequest(null);
}
}
public static void testRequest(String request) throws Exception {
String reqs = null;
if (request == null) {
// RequestCtx req = makeRequest("public", "read",
// "demo:SmileyBeerGlass");
RequestCtx req =
makeRequest("administrator",
"urn:fedora:names:fedora:2.1:action:id-ingestObject",
"/coll:accg806/coll:acst305");
reqs = contextUtil.makeRequestCtx(req);
} else {
reqs = request;
}
System.out.println(reqs);
long a = System.nanoTime();
String response = contextHandler.evaluate(reqs);
long b = System.nanoTime();
System.out.println("Total Time (ns): " + (b - a));
System.out.println(response);
a = System.nanoTime();
response = contextHandler.evaluate(reqs);
b = System.nanoTime();
System.out.println("Total Time (ns): " + (b - a));
System.out.println(response);
a = System.nanoTime();
response = contextHandler.evaluate(reqs);
b = System.nanoTime();
System.out.println("Total Time (ns): " + (b - a));
System.out.println(response);
}
public static List<Map<URI, List<AttributeValue>>> getSubjects(String subject) {
// setup the id and value for the requesting subject
List<Map<URI, List<AttributeValue>>> subjects =
new ArrayList<Map<URI, List<AttributeValue>>>();
if (subject == null || subject.equals("")) {
return subjects;
}
Map<URI, List<AttributeValue>> subAttr =
new HashMap<URI, List<AttributeValue>>();
try {
List<AttributeValue> attrList = new ArrayList<AttributeValue>();
attrList.add(new StringAttribute(subject));
subAttr.put(new URI("urn:fedora:names:fedora:2.1:subject:loginId"),
attrList);
attrList = new ArrayList<AttributeValue>();
for (int x = 0; x < 10; x++) {
attrList.add(new StringAttribute("role" + x));
}
subAttr.put(new URI("urn:fedora:names:fedora:2.1:subject:role"),
attrList);
} catch (URISyntaxException use) {
System.out.println(use.getMessage());
}
subjects.add(subAttr);
subAttr = new HashMap<URI, List<AttributeValue>>();
try {
List<AttributeValue> attrList = new ArrayList<AttributeValue>();
attrList.add(new StringAttribute(subject));
subAttr
.put(new URI("urn:oasis:names:tc:xacml:1.0:subject:subject-id"),
attrList);
} catch (URISyntaxException use) {
System.out.println(use.getMessage());
}
subjects.add(subAttr);
return subjects;
}
public static Map<URI, AttributeValue> getEnvironment() {
return new HashMap<URI, AttributeValue>();
}
public static RequestCtx makeRequest(String user, String action, String pid) {
RequestCtx req = null;
Map<URI, AttributeValue> resAttr = new HashMap<URI, AttributeValue>();
Map<URI, AttributeValue> actionAttr =
new HashMap<URI, AttributeValue>();
try {
if (pid != null && !pid.equals("")) {
resAttr
.put(new URI("urn:fedora:names:fedora:2.1:resource:object:pid"),
new StringAttribute(pid));
resAttr
.put(new URI("urn:oasis:names:tc:xacml:1.0:resource:resource-id"),
new AnyURIAttribute(new URI(pid)));
}
if (action != null && !action.equals("")) {
actionAttr
.put(new URI("urn:fedora:names:fedora:2.1:action:id"),
new StringAttribute(action));
actionAttr
.put(new URI("urn:oasis:names:tc:xacml:1.0:action:action-id"),
new StringAttribute(action));
}
} catch (URISyntaxException use) {
System.out.println(use.getMessage());
}
try {
req =
contextUtil.buildRequest(getSubjects(user),
actionAttr,
resAttr,
getEnvironment(),
RELATIONSHIP_RESOLVER);
} catch (Exception e) {
System.out.print(e.getMessage());
}
return req;
}
}