org.jboss.arquillian.drone.configuration.SecurityActions Maven / Gradle / Ivy
/*
* JBoss, Home of Professional Open Source
* Copyright 2009, Red Hat Middleware LLC, and individual contributors
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* 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.jboss.arquillian.drone.configuration;
import java.lang.reflect.Field;
import java.lang.reflect.Modifier;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.security.AccessController;
import java.security.PrivilegedAction;
import java.security.PrivilegedActionException;
import java.security.PrivilegedExceptionAction;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Properties;
/**
* SecurityActions
*
* A set of privileged actions that are not to leak out of this package
*
* @author Karel Piwko
* @version $Revision: $
*/
final class SecurityActions {
// -------------------------------------------------------------------------------||
// Constructor
// ------------------------------------------------------------------||
// -------------------------------------------------------------------------------||
/**
* No instantiation
*/
private SecurityActions() {
throw new UnsupportedOperationException("No instantiation");
}
// -------------------------------------------------------------------------------||
// Utility Methods
// --------------------------------------------------------------||
// -------------------------------------------------------------------------------||
static Map getAccessableFields(final Class source) {
Map declaredAccessableFields =
AccessController.doPrivileged(new PrivilegedAction>() {
public Map run() {
Map foundFields = new LinkedHashMap();
for (Field field : source.getDeclaredFields()) {
// omit final fields
if (Modifier.isFinal(field.getModifiers())) {
continue;
}
if (!field.isAccessible()) {
field.setAccessible(true);
}
foundFields.put(field.getName(), field);
}
return foundFields;
}
});
return declaredAccessableFields;
}
static Map getProperties(final String prefix) {
try {
Map value =
AccessController.doPrivileged(new PrivilegedExceptionAction>() {
public Map run() {
Properties props = System.getProperties();
Map subset = new LinkedHashMap();
for (Map.Entry entry : props.entrySet()) {
String name = entry.getKey().toString();
if (name.startsWith(prefix)) {
subset.put(name, entry.getValue().toString());
}
}
return subset;
}
});
return value;
}
// Unwrap
catch (final PrivilegedActionException pae) {
final Throwable t = pae.getCause();
// Rethrow
if (t instanceof SecurityException) {
throw (SecurityException) t;
}
if (t instanceof NullPointerException) {
throw (NullPointerException) t;
} else if (t instanceof IllegalArgumentException) {
throw (IllegalArgumentException) t;
} else {
// No other checked Exception thrown by System.getProperty
try {
throw (RuntimeException) t;
}
// Just in case we've really messed up
catch (final ClassCastException cce) {
throw new RuntimeException("Obtained unchecked Exception; this code should never be reached", t);
}
}
}
}
// Finds all fields of source class that are of generic type implementing java.util.Map parameterized by parameters.
static List getMapFields(final Class source, final Class... parameters) {
try {
List value = AccessController.doPrivileged(new PrivilegedExceptionAction>() {
public List run() {
Map accesableFields = getAccessableFields(source);
List fields = new ArrayList();
for (Field field : accesableFields.values()) {
// check for map
if (Map.class.isAssignableFrom(field.getType())) {
Type genericType = field.getGenericType();
boolean parameterMatched = true;
// check generic parameters
if (genericType instanceof ParameterizedType) {
ParameterizedType parameterizedType = (ParameterizedType) genericType;
if (parameterizedType.getActualTypeArguments().length != parameters.length) {
parameterMatched = false;
} else {
for (int i = 0; i < parameterizedType.getActualTypeArguments().length; i++) {
Class actualType = (Class) parameterizedType.getActualTypeArguments()[i];
if (!actualType.equals(parameters[i])) {
parameterMatched = false;
break;
}
}
}
}
if (parameterMatched == true) {
fields.add(field);
}
}
}
return fields;
}
});
return value;
}
// Unwrap
catch (final PrivilegedActionException pae) {
final Throwable t = pae.getCause();
// Rethrow
if (t instanceof SecurityException) {
throw (SecurityException) t;
}
if (t instanceof NullPointerException) {
throw (NullPointerException) t;
} else if (t instanceof IllegalArgumentException) {
throw (IllegalArgumentException) t;
} else {
// No other checked Exception thrown by System.getProperty
try {
throw (RuntimeException) t;
}
// Just in case we've really messed up
catch (final ClassCastException cce) {
throw new RuntimeException("Obtained unchecked Exception; this code should never be reached", t);
}
}
}
}
static String getProperty(final String key) {
try {
String value = AccessController.doPrivileged(new PrivilegedExceptionAction() {
public String run() {
return System.getProperty(key);
}
});
return value;
}
// Unwrap
catch (final PrivilegedActionException pae) {
final Throwable t = pae.getCause();
// Rethrow
if (t instanceof SecurityException) {
throw (SecurityException) t;
}
if (t instanceof NullPointerException) {
throw (NullPointerException) t;
} else if (t instanceof IllegalArgumentException) {
throw (IllegalArgumentException) t;
} else {
// No other checked Exception thrown by System.getProperty
try {
throw (RuntimeException) t;
}
// Just in case we've really messed up
catch (final ClassCastException cce) {
throw new RuntimeException("Obtained unchecked Exception; this code should never be reached", t);
}
}
}
}
}