blob: 395810010ee75111f52b067d713014cac8e0d858 [file] [log] [blame]
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.apache.openaz.xacml.admin.util;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLEncoder;
import java.util.Collections;
import java.util.Set;
import org.apache.commons.io.IOUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.openaz.xacml.api.pap.PAPEngine;
import org.apache.openaz.xacml.api.pap.PAPException;
import org.apache.openaz.xacml.api.pap.PDP;
import org.apache.openaz.xacml.api.pap.PDPGroup;
import org.apache.openaz.xacml.api.pap.PDPPolicy;
import org.apache.openaz.xacml.api.pap.PDPStatus;
import org.apache.openaz.xacml.rest.XACMLRestProperties;
import org.apache.openaz.xacml.std.pap.StdPDP;
import org.apache.openaz.xacml.std.pap.StdPDPGroup;
import org.apache.openaz.xacml.std.pap.StdPDPItemSetChangeNotifier;
import org.apache.openaz.xacml.std.pap.StdPDPPolicy;
import org.apache.openaz.xacml.std.pap.StdPDPStatus;
import org.apache.openaz.xacml.util.XACMLProperties;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.type.CollectionType;
/**
* Implementation of the PAPEngine interface that communicates with a PAP engine in a remote servlet
* through a RESTful interface
*
* @author glenngriffin
*
*/
public class RESTfulPAPEngine extends StdPDPItemSetChangeNotifier implements PAPEngine {
private static final Log logger = LogFactory.getLog(RESTfulPAPEngine.class);
//
// URL of the PAP Servlet that this Admin Console talks to
//
private String papServletURLString;
/**
* Set up link with PAP Servlet and get our initial set of Groups
* @throws Exception
*/
public RESTfulPAPEngine (String myURLString) throws PAPException, IOException {
//
// Get our URL to the PAP servlet
//
this.papServletURLString = XACMLProperties.getProperty(XACMLRestProperties.PROP_PAP_URL);
if (this.papServletURLString == null || this.papServletURLString.length() == 0) {
String message = "The property 'POLICYENGINE_ADMIN_ACTIVE' was not set during installation. Admin Console cannot call PAP.";
logger.error(message);
throw new PAPException(message);
}
//
// register this Admin Console with the PAP Servlet to get updates
//
Object newURL = sendToPAP("PUT", null, null, null, "adminConsoleURL=" + myURLString);
if (newURL != null) {
// assume this was a re-direct and try again
logger.warn("Redirecting to '" + newURL + "'");
this.papServletURLString = (String)newURL;
newURL = sendToPAP("PUT", null, null, null, "adminConsoleURL=" + myURLString);
if (newURL != null) {
logger.error("Failed to redirect to " + this.papServletURLString);
throw new PAPException("Failed to register with PAP");
}
}
}
//
// High-level commands used by the Admin Console code through the PAPEngine Interface
//
@Override
public PDPGroup getDefaultGroup() throws PAPException {
PDPGroup newGroup = (PDPGroup)sendToPAP("GET", null, null, StdPDPGroup.class, "groupId=", "default=");
return newGroup;
}
@Override
public void SetDefaultGroup(PDPGroup group) throws PAPException {
sendToPAP("POST", null, null, null, "groupId=" + group.getId(), "default=true");
}
@Override
public Set<PDPGroup> getPDPGroups() throws PAPException {
Set<PDPGroup> newGroupSet;
newGroupSet = (Set<PDPGroup>) this.sendToPAP("GET", null, Set.class, StdPDPGroup.class, "groupId=");
return Collections.unmodifiableSet(newGroupSet);
}
@Override
public PDPGroup getGroup(String id) throws PAPException {
PDPGroup newGroup = (PDPGroup)sendToPAP("GET", null, null, StdPDPGroup.class, "groupId=" + id);
return newGroup;
}
@Override
public void newGroup(String name, String description)
throws PAPException, NullPointerException {
String escapedName = null;
String escapedDescription = null;
try {
escapedName = URLEncoder.encode(name, "UTF-8");
escapedDescription = URLEncoder.encode(description, "UTF-8");
} catch (UnsupportedEncodingException e) {
throw new PAPException("Unable to send name or description to PAP: " + e.getMessage());
}
this.sendToPAP("POST", null, null, null, "groupId=", "groupName="+escapedName, "groupDescription=" + escapedDescription);
}
/**
* Update the configuration on the PAP for a single Group.
*
* @param group
* @return
* @throws PAPException
*/
public void updateGroup(PDPGroup group) throws PAPException {
try {
//
// ASSUME that all of the policies mentioned in this group are already located in the correct directory on the PAP!
//
// Whenever a Policy is added to the group, that file must be automatically copied to the PAP from the Workspace.
//
// // Copy all policies from the local machine's workspace to the PAP's PDPGroup directory.
// // This is not efficient since most of the policies will already exist there.
// // However, the policy files are (probably!) not too huge, and this is a good way to ensure that any corrupted files on the PAP get refreshed.
//
//TODO WRONG!!!! The policy.getStream() is based on the location in the PAP directory, not the Workspace.
//TODO
// for (PDPPolicy policy : group.getPolicies()) {
// try (InputStream is = policy.getStream()) {
// copyFile(policy.getId(), group, is);
// }
// }
// now update the group object on the PAP
sendToPAP("PUT", group, null, null, "groupId=" + group.getId());
} catch (Exception e) {
String message = "Unable to PUT policy '" + group.getId() + "', e:" + e;
logger.error(message, e);
throw new PAPException(message);
}
}
@Override
public void removeGroup(PDPGroup group, PDPGroup newGroup)
throws PAPException, NullPointerException {
String moveToGroupString = null;
if (newGroup != null) {
moveToGroupString = "movePDPsToGroupId=" + newGroup.getId();
}
sendToPAP("DELETE", null, null, null, "groupId=" + group.getId(), moveToGroupString);
}
@Override
public PDPGroup getPDPGroup(PDP pdp) throws PAPException {
return getPDPGroup(pdp.getId());
}
@Override
public PDPGroup getPDPGroup(String pdpId) throws PAPException {
PDPGroup newGroup = (PDPGroup)sendToPAP("GET", null, null, StdPDPGroup.class, "groupId=", "pdpId=" + pdpId, "getPDPGroup=");
return newGroup;
}
@Override
public PDP getPDP(String pdpId) throws PAPException {
PDP newPDP = (PDP)sendToPAP("GET", null, null, StdPDP.class, "groupId=", "pdpId=" + pdpId);
return newPDP;
}
@Override
public void newPDP(String id, PDPGroup group, String name, String description) throws PAPException,
NullPointerException {
StdPDP newPDP = new StdPDP(id, name, description);
sendToPAP("PUT", newPDP, null, null, "groupId=" + group.getId(), "pdpId=" + id);
}
@Override
public void movePDP(PDP pdp, PDPGroup newGroup) throws PAPException {
sendToPAP("POST", null, null, null, "groupId=" + newGroup.getId(), "pdpId=" + pdp.getId());
}
@Override
public void updatePDP(PDP pdp) throws PAPException {
PDPGroup group = getPDPGroup(pdp);
sendToPAP("PUT", pdp, null, null, "groupId=" + group.getId(), "pdpId=" + pdp.getId());
}
@Override
public void removePDP(PDP pdp) throws PAPException {
PDPGroup group = getPDPGroup(pdp);
sendToPAP("DELETE", null, null, null, "groupId=" + group.getId(), "pdpId=" + pdp.getId());
}
@Override
public void publishPolicy(String id, String name, boolean isRoot,
InputStream policy, PDPGroup group) throws PAPException {
//TODO - this method should take as input a Policy object, add it to the group, then call updateGroup
//TODO - ?? Where does the Policy object (with the version info) get created?
// copy the (one) file into the target directory on the PAP servlet
copyFile(id, group, policy);
// adjust the local copy of the group to include the new policy
PDPPolicy pdpPolicy = new StdPDPPolicy(id, isRoot, name);
group.getPolicies().add(pdpPolicy);
// tell the PAP servlet to include the policy in the configuration
updateGroup(group);
}
/**
* Copy a single Policy file from the input stream to the PAP Servlet.
* Either this works (silently) or it throws an exception.
*
* @param policyId
* @param group
* @param policy
* @return
* @throws PAPException
*/
public void copyFile(String policyId, PDPGroup group, InputStream policy) throws PAPException {
// send the policy file to the PAP Servlet
try {
sendToPAP("POST", policy, null, null, "groupId=" + group.getId(), "policyId="+policyId);
} catch (Exception e) {
String message = "Unable to PUT policy '" + policyId + "', e:" + e;
logger.error(message, e);
throw new PAPException(message);
}
}
@Override
public void copyPolicy(PDPPolicy policy, PDPGroup group) throws PAPException {
if (policy == null || group == null) {
throw new PAPException("Null input policy="+policy+" group="+group);
}
try (InputStream is = new FileInputStream(new File(policy.getLocation())) ) {
copyFile(policy.getId(), group, is );
} catch (Exception e) {
String message = "Unable to PUT policy '" + policy.getId() + "', e:" + e;
logger.error(message, e);
throw new PAPException(message);
}
}
@Override
public void removePolicy(PDPPolicy policy, PDPGroup group) throws PAPException {
throw new PAPException("NOT IMPLEMENTED");
}
/**
* Special operation - Similar to the normal PAP operations but this one contacts the PDP directly
* to get detailed status info.
*
* @param pdp
* @return
* @throws PAPException
*/
@Override
public PDPStatus getStatus(PDP pdp) throws PAPException {
StdPDPStatus status = (StdPDPStatus)sendToPAP("GET", pdp, null, StdPDPStatus.class);
return status;
}
//
// Internal Operations called by the PAPEngine Interface methods
//
/**
* Send a request to the PAP Servlet and get the response.
*
* The content is either an InputStream to be copied to the Request OutputStream
* OR it is an object that is to be encoded into JSON and pushed into the Request OutputStream.
*
* The Request parameters may be encoded in multiple "name=value" sets, or parameters may be combined by the caller.
*
* @param method
* @param content - EITHER an InputStream OR an Object to be encoded in JSON
* @param collectionTypeClass
* @param responseContentClass
* @param parameters
* @return
* @throws Exception
*/
private Object sendToPAP(String method, Object content, Class collectionTypeClass, Class responseContentClass, String... parameters ) throws PAPException {
HttpURLConnection connection = null;
try {
String fullURL = papServletURLString;
if (parameters != null && parameters.length > 0) {
String queryString = "";
for (String p : parameters) {
queryString += "&" + p;
}
fullURL += "?" + queryString.substring(1);
}
// special case - Status (actually the detailed status) comes from the PDP directly, not the PAP
if (method.equals("GET") &&
content instanceof PDP &&
responseContentClass == StdPDPStatus.class) {
// Adjust the url and properties appropriately
fullURL = ((PDP)content).getId() + "?type=Status";
content = null;
}
URL url = new URL(fullURL);
//
// Open up the connection
//
connection = (HttpURLConnection)url.openConnection();
//
// Setup our method and headers
//
connection.setRequestMethod(method);
// connection.setRequestProperty("Accept", "text/x-java-properties");
// connection.setRequestProperty("Content-Type", "text/x-java-properties");
connection.setUseCaches(false);
//
// Adding this in. It seems the HttpUrlConnection class does NOT
// properly forward our headers for POST re-direction. It does so
// for a GET re-direction.
//
// So we need to handle this ourselves.
//
connection.setInstanceFollowRedirects(false);
connection.setDoOutput(true);
connection.setDoInput(true);
if (content != null) {
if (content instanceof InputStream) {
try {
//
// Send our current policy configuration
//
try (OutputStream os = connection.getOutputStream()) {
int count = IOUtils.copy((InputStream)content, os);
if (logger.isDebugEnabled()) {
logger.debug("copied to output, bytes="+count);
}
}
} catch (Exception e) {
logger.error("Failed to write content in '" + method + "'", e);
throw e;
}
} else {
// The content is an object to be encoded in JSON
ObjectMapper mapper = new ObjectMapper();
mapper.writeValue(connection.getOutputStream(), content);
}
}
//
// Do the connect
//
connection.connect();
if (connection.getResponseCode() == 204) {
logger.info("Success - no content.");
return null;
} else if (connection.getResponseCode() == 200) {
logger.info("Success. We have a return object.");
// get the response content into a String
String json = null;
// read the inputStream into a buffer (trick found online scans entire input looking for end-of-file)
java.util.Scanner scanner = new java.util.Scanner(connection.getInputStream());
scanner.useDelimiter("\\A");
json = scanner.hasNext() ? scanner.next() : "";
scanner.close();
logger.info("JSON response from PAP: " + json);
// convert Object sent as JSON into local object
ObjectMapper mapper = new ObjectMapper();
if (collectionTypeClass != null) {
// collection of objects expected
final CollectionType javaType =
mapper.getTypeFactory().constructCollectionType(collectionTypeClass, responseContentClass);
Object objectFromJSON = mapper.readValue(json, javaType);
return objectFromJSON;
} else {
// single value object expected
Object objectFromJSON = mapper.readValue(json, responseContentClass);
return objectFromJSON;
}
} else if (connection.getResponseCode() >= 300 && connection.getResponseCode() <= 399) {
// redirection
String newURL = connection.getHeaderField("Location");
if (newURL == null) {
logger.error("No Location header to redirect to when response code="+connection.getResponseCode());
throw new IOException("No redirect Location header when response code="+connection.getResponseCode());
}
int qIndex = newURL.indexOf("?");
if (qIndex > 0) {
newURL = newURL.substring(0, qIndex);
}
logger.info("Redirect seen. Redirecting " + fullURL + " to " + newURL);
return newURL;
} else {
logger.warn("Unexpected response code: " + connection.getResponseCode() + " message: " + connection.getResponseMessage());
throw new IOException("Server Response: " + connection.getResponseCode() + ": " + connection.getResponseMessage());
}
} catch (Exception e) {
logger.error("HTTP Request/Response to PAP: " + e,e);
throw new PAPException("Request/Response threw :" + e);
} finally {
// cleanup the connection
if (connection != null) {
try {
// For some reason trying to get the inputStream from the connection
// throws an exception rather than returning null when the InputStream does not exist.
InputStream is = null;
try {
is = connection.getInputStream();
} catch (Exception e1) { //NOPMD
// ignore this
}
if (is != null) {
is.close();
}
} catch (IOException ex) {
logger.error("Failed to close connection: " + ex, ex);
}
connection.disconnect();
}
}
}
}