/*
|
* CDDL HEADER START
|
*
|
* The contents of this file are subject to the terms of the Common Development and
|
* Distribution License (the License). You may not use this file except in compliance with the
|
* License.
|
*
|
* You can obtain a copy of the License at legal/CDDLv1.0.txt. See the License for the
|
* specific language governing permission and limitations under the License.
|
*
|
* When distributing Covered Software, include this CDDL Header Notice in each file and include
|
* the License file at legal/CDDLv1.0.txt. If applicable, add the following below the CDDL
|
* Header, with the fields enclosed by brackets [] replaced by your own identifying
|
* information: "Portions Copyright [year] [name of copyright owner]".
|
*
|
* CDDL HEADER END
|
*
|
* Copyright 2013 ForgeRock AS.
|
* Portions Copyright 2013 IS4U.
|
*/
|
|
package org.forgerock.opendj.virtual;
|
|
import java.io.IOException;
|
import java.sql.DriverManager;
|
import java.sql.ResultSet;
|
import java.sql.ResultSetMetaData;
|
import java.sql.SQLException;
|
import java.sql.Statement;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.Iterator;
|
import java.util.List;
|
import java.util.ListIterator;
|
import java.util.Map;
|
|
import org.forgerock.opendj.ldap.AbstractSynchronousConnection;
|
import org.forgerock.opendj.ldap.Attribute;
|
import org.forgerock.opendj.ldap.ByteString;
|
import org.forgerock.opendj.ldap.ConnectionEventListener;
|
import org.forgerock.opendj.ldap.DN;
|
import org.forgerock.opendj.ldap.Entry;
|
import org.forgerock.opendj.ldap.ErrorResultException;
|
import org.forgerock.opendj.ldap.Filter;
|
import org.forgerock.opendj.ldap.IntermediateResponseHandler;
|
import org.forgerock.opendj.ldap.LinkedHashMapEntry;
|
import org.forgerock.opendj.ldap.Modification;
|
import org.forgerock.opendj.ldap.ModificationType;
|
import org.forgerock.opendj.ldap.RDN;
|
import org.forgerock.opendj.ldap.ResultCode;
|
import org.forgerock.opendj.ldap.SearchResultHandler;
|
import org.forgerock.opendj.ldap.requests.AddRequest;
|
import org.forgerock.opendj.ldap.requests.BindRequest;
|
import org.forgerock.opendj.ldap.requests.CompareRequest;
|
import org.forgerock.opendj.ldap.requests.DeleteRequest;
|
import org.forgerock.opendj.ldap.requests.ExtendedRequest;
|
import org.forgerock.opendj.ldap.requests.ModifyDNRequest;
|
import org.forgerock.opendj.ldap.requests.ModifyRequest;
|
import org.forgerock.opendj.ldap.requests.SearchRequest;
|
import org.forgerock.opendj.ldap.requests.SimpleBindRequest;
|
import org.forgerock.opendj.ldap.requests.UnbindRequest;
|
import org.forgerock.opendj.ldap.responses.BindResult;
|
import org.forgerock.opendj.ldap.responses.CompareResult;
|
import org.forgerock.opendj.ldap.responses.ExtendedResult;
|
import org.forgerock.opendj.ldap.responses.Responses;
|
import org.forgerock.opendj.ldap.responses.Result;
|
|
/**
|
* JDBC connection implementation.
|
*/
|
public final class JDBCConnection extends AbstractSynchronousConnection {
|
//For connection to h2 database, use driverName = "org.h2.driver";
|
private final String driverName = "com.mysql.jdbc.Driver";
|
private java.sql.Connection connection;
|
private String connectionUrl;
|
private JDBCMapper jdbcm;
|
private MappingConfigurationManager mcm;
|
private List<Entry> searchEntries = new ArrayList<Entry>();
|
|
/**
|
* Creates a new JDBC connection.
|
*
|
* @param connectionURL
|
* The URL to connect to the SQL Database Server consisting of the
|
* host name, port number and name of the database.
|
*/
|
JDBCConnection(final String connectionURL)
|
{
|
this.connectionUrl = connectionURL;
|
}
|
|
/**
|
* Initializes the mapping component for this connection.
|
*
|
* @param jdbcMapper
|
* The JDBCMapper object used to map the directory and database
|
* structure.
|
* @throws SQLException
|
* If the SQL query has an invalid format.
|
* @throws ErrorResultException
|
* If the result code indicates that the request failed for some
|
* reason.
|
* @throws IOException
|
* If an I/O exception error occurs.
|
*/
|
public void initializeMapper(final JDBCMapper jdbcMapper) throws SQLException, ErrorResultException, IOException
|
{
|
//Load the jdbc mapper with the mapping from the properties file using the Mapping Configuration Manager.
|
jdbcm = jdbcMapper;
|
jdbcm.fillMaps();
|
mcm = new MappingConfigurationManager(jdbcm);
|
jdbcm.loadMappingConfig(mcm.loadMapping());
|
}
|
|
/**
|
* Returns the SQL connection used by the SQL driver to connect with the database.
|
*
|
* @return A connection to the database used by the SQL driver.
|
*/
|
public java.sql.Connection getSqlConnection()
|
{
|
return connection;
|
}
|
|
/**
|
* Adds a record to the Database Server using the provided add request.
|
*
|
* @param request
|
* The add request.
|
* @return The result of the operation.
|
* @throws ErrorResultException
|
* If the result code indicates that the request failed for some
|
* reason.
|
* @throws SQLException
|
* If the SQL query has an invalid format.
|
* @throws NullPointerException
|
* If {@code request} was {@code null}, or if
|
* a corresponding mapping value could not be found in the mapping component.
|
*/
|
@Override
|
public Result add(final AddRequest request) throws ErrorResultException
|
{
|
Result r;
|
try {
|
//Split up the DN the of the request.
|
final DN DN = request.getName();
|
final RDN rDN = DN.rdn();
|
final String rDNName = rDN.getFirstAVA().getAttributeType().getNameOrOID();
|
final String rDNValue = rDN.getFirstAVA().getAttributeValue().toString();
|
final RDN OU = DN.parent().rdn();
|
final String organizationalUnitName = OU.getFirstAVA().getAttributeValue().toString();
|
final String baseDN = DN.parent(2).toString();
|
|
//Search mapping for the corresponding table and column names.
|
final String tableName = jdbcm.getTableNameFromMapping(baseDN, organizationalUnitName);
|
|
if(tableName == null) throw new NullPointerException("JDBC Error: Could not find matching table name for OU: '" + organizationalUnitName + "' in DN: '" + baseDN + "'. Please check if mapping was succesful.");
|
final String rDNColumnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, organizationalUnitName, rDNName);
|
|
if(rDNColumnName == null) throw new NullPointerException("JDBC Error: Could not find matching column name for attribute: '" + rDNName + "'. Please check if mapping was succesful.");
|
final Map<String, Object> columnValuesMap = new HashMap<String, Object>();
|
final Iterable<Attribute> attributesCollection = request.getAllAttributes();
|
final Iterator<Attribute> attributeIter = attributesCollection.iterator();
|
|
//Search mapping for the corresponding column name for each attribute.
|
while(attributeIter.hasNext()){
|
final Attribute att = attributeIter.next();
|
final Iterator<ByteString> valueIter = att.iterator();
|
final String attributeName = att.getAttributeDescriptionAsString();
|
|
if(attributeName == "objectClass") continue;
|
final String columnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, organizationalUnitName, attributeName);
|
|
if(columnName == null) throw new NullPointerException("JDBC Error: Could not find matching column name for attribute: '" + attributeName + "'. Please check if mapping was succesful.");
|
String columnValue = "";
|
|
while(valueIter.hasNext()){
|
columnValue = columnValue.concat(valueIter.next().toString());
|
}
|
columnValuesMap.put(columnName, columnValue);
|
}
|
|
//Build the SQL values String.
|
final ArrayList<String>columnList = jdbcm.getTableColumns(tableName);
|
String columnNamesString = " (" + rDNColumnName;
|
String columnValuesString = " ('" + rDNValue + "'";
|
|
for(int i = 0; i < columnList.size(); i++){
|
final String columnName = columnList.get(i);
|
|
if(columnNamesString.contains(columnName)) continue;
|
final Object columnDataType = jdbcm.getTableColumnDataType(tableName, columnName);
|
Object columnValue = columnValuesMap.get(columnName);
|
|
if(columnValue == null){
|
boolean columnNullable = jdbcm.getTableColumnNullable(tableName, columnName);
|
|
if(columnNullable) continue;
|
|
if(columnDataType.equals(Integer.class)) columnValue = "0";
|
else columnValue = "Default Value";
|
}
|
|
if(columnDataType.equals(Integer.class)) columnValue = Integer.parseInt(columnValue.toString());
|
columnNamesString = columnNamesString.concat(", " + columnName);
|
columnValuesString = columnValuesString.concat(", '" + columnValue + "'");
|
}
|
columnNamesString = columnNamesString.concat(")");
|
columnValuesString = columnValuesString.concat(")");
|
|
//Build the SQL query.
|
final Statement st = connection.createStatement();
|
final String sql = "INSERT INTO " + tableName + columnNamesString + " VALUES" + columnValuesString;
|
st.executeUpdate(sql);
|
r = Responses.newResult(ResultCode.SUCCESS);
|
}catch (SQLException e) {
|
System.out.println(e.toString());
|
r = Responses.newResult(ResultCode.OPERATIONS_ERROR);
|
r.setCause(e);
|
}catch (NullPointerException e){
|
System.out.println(e.toString());
|
r = Responses.newResult(ResultCode.OPERATIONS_ERROR);
|
r.setCause(e);
|
}
|
return r;
|
}
|
|
@Override
|
public void addConnectionEventListener(ConnectionEventListener listener)
|
{
|
// TODO Auto-generated method stub
|
}
|
|
/**
|
* Authenticates to the Database Server using the provided bind request.
|
*
|
* @param request
|
* The bind request.
|
* @return The result of the operation.
|
* @throws ErrorResultException
|
* If the result code indicates that the request failed for some
|
* reason.
|
* @throws ClassNotFoundException
|
* If no definition for the class with the specified driver name could be found.
|
* @throws SQLException
|
* If the driver could not establish a connection with the
|
* Database Server.
|
*/
|
@Override
|
public BindResult bind(final BindRequest request) throws ErrorResultException
|
{
|
BindResult r;
|
|
//Extract user name and password from the bind request.
|
final String userName = request.getName();
|
final String userPass;
|
byte[] password = null;
|
|
if (request instanceof SimpleBindRequest) {
|
password = ((SimpleBindRequest) request).getPassword();
|
} else {
|
r = Responses.newBindResult(ResultCode.PROTOCOL_ERROR);
|
return r;
|
}
|
userPass = new String(password);
|
|
//Establish SQL connection to the Database Server.
|
try {
|
Class.forName(driverName);
|
this.connection = DriverManager
|
.getConnection(this.connectionUrl,userName , userPass);
|
}catch (ClassNotFoundException e) {
|
System.out.println(e.toString());
|
r = Responses.newBindResult(ResultCode.OTHER);
|
return r;
|
}catch (SQLException e) {
|
System.out.println(e.toString());
|
r = Responses.newBindResult(ResultCode.CLIENT_SIDE_CONNECT_ERROR);
|
return r;
|
}
|
r = Responses.newBindResult(ResultCode.SUCCESS);
|
return r;
|
}
|
|
/**
|
* {@inheritDoc}
|
*/
|
@Override
|
public void close(final UnbindRequest request, final String reason)
|
{
|
if(connection != null){
|
try
|
{
|
connection.close();
|
}
|
catch (SQLException e)
|
{
|
e.printStackTrace();
|
}
|
}
|
}
|
|
/**
|
* Compares a record in the Database Server using the provided compare request.
|
*
|
* @param request
|
* The compare request.
|
* @return The result of the operation.
|
* @throws ErrorResultException
|
* If the result code indicates that the request failed for some
|
* reason.
|
* @throws SQLException
|
* If the SQL query has an invalid format.
|
* @throws NullPointerException
|
* If {@code request} was {@code null}, or if
|
* a corresponding mapping value could not be found in the mapping component.
|
*/
|
@Override
|
public CompareResult compare(final CompareRequest request) throws ErrorResultException
|
{
|
CompareResult cr;
|
try
|
{
|
//Split up the DN the of the request.
|
final DN DN = request.getName();
|
final RDN rDN = DN.rdn();
|
final String filterAttributeName = rDN.getFirstAVA().getAttributeType().getNameOrOID();
|
final String filterAttributeValue = rDN.getFirstAVA().getAttributeValue().toString();
|
final RDN OU = DN.parent(1).rdn();
|
final String OUName = OU.getFirstAVA().getAttributeValue().toString();
|
final String baseDN = DN.parent(2).toString();
|
|
//Search mapping for the corresponding table and column names.
|
final String tableName = jdbcm.getTableNameFromMapping(baseDN, OUName);
|
|
if(tableName == null) throw new NullPointerException("JDBC Error: Could not find matching table name for OU: '" + OUName + "' in DN: '" + baseDN + "'. Please check if mapping was succesful.");
|
final String columnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, OUName, filterAttributeName);
|
|
if(columnName == null) throw new NullPointerException("JDBC Error: Could not find matching column name for attribute: '" + filterAttributeName + "'. Please check if mapping was succesful.");
|
|
//Search mapping for the corresponding column name for the comparing attribute.
|
final String compareAttributeName = request.getAttributeDescription().toString();
|
final String compareAttributeValue = request.getAssertionValueAsString();
|
final String compareColumnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, OUName, compareAttributeName);
|
|
if(compareColumnName == null) throw new NullPointerException("JDBC Error: Could not find matching column name for attribute: '" + compareAttributeName + "'. Please check if mapping was succesful.");
|
|
//Build the SQL query.
|
final Statement st = connection.createStatement();
|
final String sql = "SELECT * FROM " + tableName + " WHERE " + columnName + "='" + filterAttributeValue + "' AND " + compareColumnName + "='" + compareAttributeValue + "'";
|
final ResultSet rs = st.executeQuery(sql);
|
if(rs.first()) cr = Responses.newCompareResult(ResultCode.COMPARE_TRUE);
|
else cr = Responses.newCompareResult(ResultCode.COMPARE_FALSE);
|
}catch (SQLException e) {
|
System.out.println(e.toString());
|
cr = Responses.newCompareResult(ResultCode.OPERATIONS_ERROR);
|
cr.setCause(e);
|
}catch (NullPointerException e){
|
System.out.println(e.toString());
|
cr = Responses.newCompareResult(ResultCode.OPERATIONS_ERROR);
|
cr.setCause(e);
|
}
|
return cr;
|
}
|
|
/**
|
* Deletes a record from the Database Server using the provided delete request.
|
*
|
* @param request
|
* The delete request.
|
* @return The result of the operation.
|
* @throws ErrorResultException
|
* If the result code indicates that the request failed for some
|
* reason.
|
* @throws SQLException
|
* If the SQL query has an invalid format.
|
* @throws NullPointerException
|
* If {@code request} was {@code null}, or if
|
* a corresponding mapping value could not be found in the mapping component.
|
*/
|
@Override
|
public Result delete(DeleteRequest request) throws ErrorResultException
|
{
|
Result r;
|
try{
|
//Split up the DN the of the request.
|
final DN DN = request.getName();
|
final RDN rDN = DN.rdn();
|
final String filterAttributeName = rDN.getFirstAVA().getAttributeType().getNameOrOID();
|
final String filterAttributeValue = rDN.getFirstAVA().getAttributeValue().toString();
|
final RDN OU = DN.parent(1).rdn();
|
final String OUName = OU.getFirstAVA().getAttributeValue().toString();
|
final String baseDN = DN.parent(2).toString();
|
|
//Search mapping for the corresponding table and column names.
|
final String tableName = jdbcm.getTableNameFromMapping(baseDN, OUName);
|
|
if(tableName == null) throw new NullPointerException("JDBC Error: Could not find matching table name for OU: '" + OUName + "' in DN: '" + baseDN + "'. Please check if mapping was succesful.");
|
final String columnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, OUName, filterAttributeName);
|
|
if(columnName == null) throw new NullPointerException("JDBC Error: Could not find matching column name for attribute: '" + filterAttributeName + "'. Please check if mapping was succesful.");
|
|
//Build the SQL query.
|
final Statement st = connection.createStatement();
|
final String sql = "DELETE FROM " + tableName + " WHERE " + columnName + "='" + filterAttributeValue + "'";
|
st.executeUpdate(sql);
|
r = Responses.newResult(ResultCode.SUCCESS);
|
}catch (SQLException e) {
|
System.out.println(e.toString());
|
r = Responses.newResult(ResultCode.OPERATIONS_ERROR);
|
r.setCause(e);
|
}catch (NullPointerException e){
|
System.out.println(e.toString());
|
r = Responses.newCompareResult(ResultCode.OPERATIONS_ERROR);
|
r.setCause(e);
|
}
|
return r;
|
}
|
|
@Override
|
public <R extends ExtendedResult> R extendedRequest(
|
ExtendedRequest<R> request, IntermediateResponseHandler handler)
|
throws ErrorResultException
|
{
|
// TODO Auto-generated method stub
|
return null;
|
}
|
|
/**
|
* {@inheritDoc}
|
*/
|
@Override
|
public boolean isClosed()
|
{
|
try
|
{
|
return connection.isClosed();
|
}
|
catch (SQLException e)
|
{
|
e.printStackTrace();
|
return true;
|
}
|
}
|
|
/**
|
* {@inheritDoc}
|
*/
|
@Override
|
public boolean isValid()
|
{
|
try
|
{
|
return connection.isValid(0);
|
}
|
catch (SQLException e)
|
{
|
e.printStackTrace();
|
return false;
|
}
|
}
|
|
/**
|
* Modifies a record in the Database Server using the provided modify request.
|
*
|
* @param request
|
* The modify request.
|
* @return The result of the operation.
|
* @throws ErrorResultException
|
* If the result code indicates that the request failed for some
|
* reason.
|
* @throws SQLException
|
* If the SQL query has an invalid format.
|
* @throws NullPointerException
|
* If {@code request} was {@code null}, or if
|
* a corresponding mapping value could not be found in the mapping component.
|
*/
|
@Override
|
public Result modify(final ModifyRequest request)
|
{
|
Result r;
|
try{
|
//Split up the DN the of the request.
|
final DN DN = request.getName();
|
final RDN rDN = DN.rdn();
|
final String filterAttributeName = rDN.getFirstAVA().getAttributeType().getNameOrOID();
|
final String filterAttributeValue = rDN.getFirstAVA().getAttributeValue().toString();
|
final RDN OU = DN.parent(1).rdn();
|
final String OUName = OU.getFirstAVA().getAttributeValue().toString();
|
final String baseDN = DN.parent(2).toString();
|
|
//Search mapping for the corresponding table and column names.
|
final String tableName = jdbcm.getTableNameFromMapping(baseDN, OUName);
|
|
if(tableName == null) throw new NullPointerException("JDBC Error: Could not find matching table name for OU: '" + OUName + "' in DN: '" + baseDN + "'. Please check if mapping was succesful.");
|
final String columnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, OUName, filterAttributeName);
|
|
if(columnName == null) throw new NullPointerException("JDBC Error: Could not find matching column name for attribute: '" + filterAttributeName + "'. Please check if mapping was succesful.");
|
|
//Get attribute and modificationtype for each modification.
|
//Search mapping for the corresponding column name for the modification attribute.
|
final List<Modification> modificationList = request.getModifications();
|
final ListIterator<Modification> listIter = modificationList.listIterator();
|
String modificationString = "";
|
|
while(listIter.hasNext()){
|
final Modification modification = listIter.next();
|
final ModificationType modificationType = modification.getModificationType();
|
final Attribute modificationAttribute = modification.getAttribute();
|
final String modificationAttributeName = modificationAttribute.getAttributeDescription().toString();
|
final String modificationColumnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, OUName, modificationAttributeName);
|
|
if(modificationColumnName == null) throw new NullPointerException("JDBC Error: Could not find matching column name for attribute: '" + modificationAttributeName + "'. Please check if mapping was succesful.");
|
String modificationAttributeValue = "";
|
|
if(modificationType == ModificationType.ADD){
|
final Iterator<ByteString> iter = modificationAttribute.iterator();
|
|
int counter = 0;
|
while (iter.hasNext()){
|
if(counter > 0) modificationAttributeValue = modificationAttributeValue.concat(", ");
|
modificationAttributeValue = modificationAttributeValue.concat(iter.next().toString());
|
counter++;
|
}
|
final Object classType = jdbcm.getTableColumnDataType(tableName, modificationColumnName);
|
|
if(classType == Integer.class) modificationAttributeValue = "(CASE WHEN (" + modificationColumnName + " = 0) THEN ' "
|
+ modificationAttributeValue + "' ELSE concat(" + modificationColumnName + ", ', " + modificationAttributeValue + "') END)";
|
else modificationAttributeValue = "(CASE WHEN (" + modificationColumnName + " = 'Default Value') THEN ' "
|
+ modificationAttributeValue + "' ELSE concat(" + modificationColumnName + ", ', " + modificationAttributeValue + "') END)";
|
}
|
else if(modificationType == ModificationType.REPLACE){
|
final Iterator<ByteString> iter = modificationAttribute.iterator();
|
|
while (iter.hasNext()){
|
modificationAttributeValue = "'" + iter.next().toString() + "'";
|
}
|
}
|
else{
|
final boolean nullable = jdbcm.getTableColumnNullable(tableName, modificationColumnName);
|
|
if(nullable == false) throw new SQLException("Cannot delete data from not-nullable column.");
|
final Object classType = jdbcm.getTableColumnDataType(tableName, modificationColumnName);
|
|
if(classType == Integer.class) modificationAttributeValue = "'" + Integer.toString(0) + "'";
|
else modificationAttributeValue = "'Default Value'";
|
}
|
modificationString = modificationString.concat(modificationColumnName + "=" + modificationAttributeValue + ", ");
|
}
|
modificationString = modificationString.substring(0, modificationString.length() - 2);
|
|
//Build the SQL query.
|
final Statement st = connection.createStatement();
|
final String sql = "UPDATE " + tableName + " SET " + modificationString + " WHERE " + columnName + "='" + filterAttributeValue + "'";
|
st.executeUpdate(sql);
|
r = Responses.newResult(ResultCode.SUCCESS);
|
}catch (SQLException e) {
|
System.out.println(e.toString());
|
r = Responses.newResult(ResultCode.OPERATIONS_ERROR);
|
r.setCause(e);
|
}catch (NullPointerException e){
|
System.out.println(e.toString());
|
r = Responses.newCompareResult(ResultCode.OPERATIONS_ERROR);
|
r.setCause(e);
|
}
|
return r;
|
}
|
|
@Override
|
public Result modifyDN(ModifyDNRequest request) throws ErrorResultException
|
{
|
// TODO Auto-generated method stub
|
return null;
|
}
|
|
@Override
|
public void removeConnectionEventListener(ConnectionEventListener listener)
|
{
|
// TODO Auto-generated method stub
|
}
|
|
/**
|
* Converts the LDAP filter to an SQL filter.
|
*
|
* @param filter
|
* The filter of the search request.
|
* @param baseDN
|
* The top domain name.
|
* @param OUName
|
* The name of the organizational unit.
|
* @return The converted SQL filter.
|
* @throws NullPointerException
|
* If the result code indicates that the request failed for some
|
* reason.
|
* @throws SQLException
|
* If the SQL query has an invalid format.
|
* @throws NullPointerException
|
* If {@code filter}, {@code baseDN} or {@code OUName} was {@code null}, or if
|
* a corresponding mapping value could not be found in the mapping component.
|
*/
|
private String convertSearchFilter(final Filter filter, final String baseDN, String OUName)
|
{
|
try{
|
String filterString = filter.toString();
|
String convertedFilterString = "";
|
|
if(filterString.isEmpty() || filterString.contains("objectClass=*")) return "";
|
int stringIndex = 0;
|
int subStringCount = 0;
|
|
|
//Set the amount of subfilters if operators are present.
|
while(filterString.charAt(stringIndex) == '('){
|
if(filterString.charAt(stringIndex + 2) == '('){
|
subStringCount++;
|
}
|
stringIndex += 2;
|
}
|
|
if(subStringCount == 0){
|
//Replace escape characters.
|
String subString = filterString.substring(1, filterString.length()-1);
|
subString = subString.replace("\\02a", "*");
|
subString = subString.replace("\\028", "(");
|
subString = subString.replace("\\029", ")");
|
|
String[] subStringSplitter;
|
|
if(subString.contains("<=")) subStringSplitter = subString.split("<=");
|
else if(subString.contains(">=")) subStringSplitter = subString.split(">=");
|
else subStringSplitter = subString.split("=");
|
|
final String filterAttributeName = subStringSplitter[0];
|
String filterColumnName = null;;
|
Object columnDataType = null;
|
|
//Search mapping for the corresponding table and column names.
|
if(OUName.isEmpty()){
|
final List<String> OUList = jdbcm.getOrganizationalUnits(baseDN);
|
|
for(Iterator<String> iter = OUList.iterator(); iter.hasNext();){
|
OUName = iter.next();
|
final String tableName = jdbcm.getTableNameFromMapping(baseDN, OUName);
|
|
if(tableName == null) continue;
|
filterColumnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, OUName, filterAttributeName);
|
|
if(filterColumnName == null) continue;
|
else columnDataType = jdbcm.getTableColumnDataType(tableName, filterColumnName);
|
break;
|
}
|
|
if(filterColumnName == null) throw new NullPointerException("JDBC Error: Could not find matching column name for attribute: '" + filterAttributeName + "'. Please check if mapping was succesful.");
|
}
|
else{
|
final String tableName = jdbcm.getTableNameFromMapping(baseDN, OUName);
|
|
if(tableName == null) throw new NullPointerException("JDBC Error: Could not find matching table name for OU: '" + OUName + "' in DN: '" + baseDN + "'. Please check if mapping was succesful.");
|
filterColumnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, OUName, filterAttributeName);
|
|
if(filterColumnName == null) throw new NullPointerException("JDBC Error: Could not find matching column name for attribute: '" + filterAttributeName + "'. Please check if mapping was succesful.");
|
columnDataType = jdbcm.getTableColumnDataType(tableName, filterColumnName);
|
}
|
|
//Set values to right format.
|
if(columnDataType != Integer.class){
|
String filterAttributeValue = subStringSplitter[1];
|
String filterColumnValue = "'" + filterAttributeValue + "'";
|
filterAttributeValue = filterAttributeValue.replace("*", "\\*");
|
filterAttributeValue = filterAttributeValue.replace("(", "\\(");
|
filterAttributeValue = filterAttributeValue.replace(")", "\\)");
|
|
if(filterColumnValue.length() > 3 && filterColumnValue.contains("*")){
|
filterColumnValue = filterColumnValue.replace("*", "%");
|
|
subString = subString.replace("=", " like ");
|
}
|
subString = subString.replaceFirst(filterAttributeValue, filterColumnValue);
|
}
|
subString = subString.replaceFirst(filterAttributeName, filterColumnName);
|
convertedFilterString = convertedFilterString.concat(subString);
|
}
|
else
|
{
|
int subStringStartIndex = 2 * subStringCount;
|
int subStringEndIndex = 0;
|
final String[] subStrings = new String[subStringCount];
|
final char[] operationChars = new char[subStringCount];
|
|
//Separate operation characters and subfilters.
|
while(subStringStartIndex > 0){
|
final char operationChar = filterString.charAt(subStringStartIndex - 1);
|
subStringEndIndex = filterString.indexOf("))") + 1;
|
String subString = filterString.substring(subStringStartIndex, subStringEndIndex);
|
subString = subString.replace("()","");
|
|
operationChars[subStringCount - subStringStartIndex / 2] = operationChar;
|
subStrings[subStringCount - subStringStartIndex / 2] = subString;
|
|
final String replaceString = filterString.substring(subStringStartIndex - 1, subStringEndIndex);
|
filterString = filterString.replace(replaceString, "");
|
|
subStringStartIndex-=2;
|
}
|
|
//Handle the not operator.
|
for(int i = 0; i < subStringCount; i++){
|
final char operationChar = operationChars[i];
|
|
if(operationChar == '!'){
|
String subString = subStrings[i];
|
|
if(subString.isEmpty()){
|
subString = subStrings[i-1];
|
subString = subString.replace(">=", ">");
|
subString = subString.replace("<=", "<");
|
subString = subString.replace("=", "!=");
|
subString = subString.replace(">", ">=");
|
subString = subString.replace("<", "<=");
|
subStrings[i-1] = subString;
|
}
|
else{
|
subString = subString.replace(">=", ">");
|
subString = subString.replace("<=", "<");
|
subString = subString.replace("=", "!=");
|
subString = subString.replace(">", ">=");
|
subString = subString.replace("<", "<=");
|
subStrings[i] = subString;
|
}
|
}
|
}
|
boolean multipleSubStrings = false;
|
|
if(subStringCount > 1) multipleSubStrings = true;
|
|
//Convert each subfilter with corresponding operator.
|
for(int i = 0; i < subStringCount; i++){
|
final char operationChar = operationChars[i];
|
String operationString = "";
|
|
if(operationChar == '&'){
|
operationString = " AND ";
|
}
|
else if (operationChar == '|'){
|
operationString = " OR ";
|
}
|
|
String subString = subStrings[i];
|
if (subString.isEmpty()) continue;
|
|
subString = subString.substring(1, subString.length() - 1);
|
String[] subStringSplitter = subString.split("\\)\\(");
|
|
for(int j = 0; j < subStringSplitter.length; j++){
|
String subStringFilter = subStringSplitter[j];
|
final String[] subStringFilterSplitter;
|
|
subStringFilter = subStringFilter.replace("\\02a", "*");
|
subStringFilter = subStringFilter.replace("\\028", "(");
|
subStringFilter = subStringFilter.replace("\\029", ")");
|
|
if(subStringFilter.contains("!=")) subStringFilterSplitter = subStringFilter.split("!=");
|
else if(subStringFilter.contains("<=")) subStringFilterSplitter = subStringFilter.split("<=");
|
else if(subStringFilter.contains(">=")) subStringFilterSplitter = subStringFilter.split(">=");
|
else subStringFilterSplitter = subStringFilter.split("=");
|
|
final String filterAttributeName = subStringFilterSplitter[0];
|
String filterColumnName = null;;
|
Object columnDataType = null;
|
|
//Search mapping for the corresponding column name for each filter attribute.
|
if(OUName.isEmpty()){
|
final List<String> OUList = jdbcm.getOrganizationalUnits(baseDN);
|
|
for(Iterator<String> iter = OUList.iterator(); iter.hasNext();){
|
OUName = iter.next();
|
final String tableName = jdbcm.getTableNameFromMapping(baseDN, OUName);
|
|
if(tableName == null) continue;
|
filterColumnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, OUName, filterAttributeName);
|
|
if(filterColumnName == null) continue;
|
else columnDataType = jdbcm.getTableColumnDataType(tableName, filterColumnName);
|
break;
|
}
|
if(filterColumnName == null) throw new NullPointerException("JDBC Error: Could not find matching column name for attribute: '" + filterAttributeName + "'. Please check if mapping was succesful.");
|
}
|
else{
|
final String tableName = jdbcm.getTableNameFromMapping(baseDN, OUName);
|
|
if(tableName == null) throw new NullPointerException("JDBC Error: Could not find matching table name for OU: '" + OUName + "' in DN: '" + baseDN + "'. Please check if mapping was succesful.");
|
filterColumnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, OUName, filterAttributeName);
|
|
if(filterColumnName == null) throw new NullPointerException("JDBC Error: Could not find matching column name for attribute: '" + filterAttributeName + "'. Please check if mapping was succesful.");
|
columnDataType = jdbcm.getTableColumnDataType(tableName, filterColumnName);
|
}
|
|
//Handle the like operator.
|
if(columnDataType != Integer.class){
|
String filterAttributeValue = subStringFilterSplitter[1];
|
String filterColumnValue = "'" + filterAttributeValue + "'";
|
filterAttributeValue = filterAttributeValue.replace("*", "\\*");
|
filterAttributeValue = filterAttributeValue.replace("(", "\\(");
|
filterAttributeValue = filterAttributeValue.replace(")", "\\)");
|
|
if(filterColumnValue.length() > 3 && filterColumnValue.contains("*")){
|
filterColumnValue = filterColumnValue.replace("*", "%");
|
|
if(subStringFilter.contains("!=")) subStringFilter = subStringFilter.replace("!=", " not like ");
|
else subStringFilter = subStringFilter.replace("=", " like ");
|
}
|
subStringFilter = subStringFilter.replaceFirst(filterAttributeValue, filterColumnValue);
|
}
|
subStringFilter = subStringFilter.replaceFirst(filterAttributeName, filterColumnName);
|
|
if(j != 0 || i != 0) convertedFilterString = convertedFilterString.concat(operationString);
|
convertedFilterString = convertedFilterString.concat(subStringFilter);
|
}
|
|
if(multipleSubStrings && i < subStringCount -1) convertedFilterString = "(" + convertedFilterString + ")";
|
}
|
}
|
return convertedFilterString;
|
}
|
catch(NullPointerException e){
|
return "Err:" + e.getMessage();
|
}
|
}
|
|
/**
|
* Returns the search entries of the last search request.
|
*
|
* @return A list of entries of the last search request.
|
*/
|
public List<Entry> getSearchEntries()
|
{
|
return searchEntries;
|
}
|
|
/**
|
* Search the Database Server using the provided search request.
|
*
|
* @param request
|
* The search request.
|
* @param handler
|
* A search result handler which can be used to asynchronously process the
|
* search result entries and references as they are received, may be null.
|
* @return The result of the operation.
|
* @throws ErrorResultException
|
* If the result code indicates that the request failed for some
|
* reason.
|
* @throws SQLException
|
* If the SQL query has an invalid format.
|
* @throws NullPointerException
|
* If {@code request} was {@code null}, or if
|
* a corresponding mapping value could not be found in the mapping component.
|
*/
|
@Override
|
public Result search(SearchRequest request, SearchResultHandler handler) throws ErrorResultException
|
{
|
Result r;
|
try{
|
//Split up the DN the of the request.
|
final DN DN = request.getName();
|
final int DNSize = DN.size();
|
final String baseDN = DN.parent(DNSize - 2).toString();
|
RDN OU = null;
|
RDN rDN = null;
|
String OUName = "";
|
String rDNAttributeName = "";
|
String rDNAttributeValue = "";
|
|
if(DNSize > 3){
|
OU = DN.parent(1).rdn();
|
OUName = OU.getFirstAVA().getAttributeValue().toString();
|
rDN = DN.parent(0).rdn();
|
rDNAttributeName = rDN.getFirstAVA().getAttributeType().getNameOrOID();
|
rDNAttributeValue = rDN.getFirstAVA().getAttributeValue().toString();
|
}
|
else if(DNSize > 2){
|
OU = DN.parent(0).rdn();
|
OUName = OU.getFirstAVA().getAttributeValue().toString();
|
}
|
List<String> returnAttributeNames = request.getAttributes();
|
final List<String> removeAttributesList = new ArrayList<String>();
|
|
//Get all column names to be returned to return full records.
|
if(returnAttributeNames == null){
|
if(OU != null){
|
returnAttributeNames = jdbcm.getOrganizationalUnitAttributes(baseDN, OUName);
|
|
if(returnAttributeNames == null) throw new NullPointerException("JDBC Error: Could not find any matching attributes in OU: '" + OUName + "'.");
|
final String tableName = jdbcm.getTableNameFromMapping(baseDN, OUName);
|
|
for(Iterator<String> iterAttributes = returnAttributeNames.iterator(); iterAttributes.hasNext();){
|
String returnAttributeName = iterAttributes.next();
|
final String returnColumnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, OUName, returnAttributeName);
|
|
if(returnColumnName == null) removeAttributesList.add(returnAttributeName);
|
}
|
returnAttributeNames.removeAll(removeAttributesList);
|
}
|
else{
|
final List<String> OUList = jdbcm.getOrganizationalUnits(baseDN);
|
|
if(OUList == null) throw new NullPointerException("JDBC Error: Could not find any matching organizational units in DN: '" + baseDN + "'.");
|
|
for(Iterator<String> iterOU = OUList.iterator(); iterOU.hasNext();){
|
OUName = iterOU.next();
|
final String tableName = jdbcm.getTableNameFromMapping(baseDN, OUName);
|
returnAttributeNames = jdbcm.getOrganizationalUnitAttributes(baseDN, OUName);
|
|
for(Iterator<String> iterAttributes = returnAttributeNames.iterator(); iterAttributes.hasNext();){
|
String returnAttributeName = iterAttributes.next();
|
final String returnColumnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, OUName, returnAttributeName);
|
|
if(returnColumnName == null) removeAttributesList.add(returnAttributeName);
|
}
|
returnAttributeNames.removeAll(removeAttributesList);
|
}
|
}
|
}
|
//Convert the search filter.
|
final Filter searchFilter = request.getFilter();
|
final String convertedFilterString = convertSearchFilter(searchFilter, baseDN, OUName);
|
|
if(convertedFilterString.startsWith("Err:")){
|
String[] errorStringSplitter = convertedFilterString.split("Err:");
|
throw new NullPointerException(errorStringSplitter[1]);
|
}
|
final List<String> returnColumnNames = new ArrayList<String>();
|
|
//Search mapping for the corresponding column name for each return attribute.
|
for(Iterator<String> iter = returnAttributeNames.iterator();iter.hasNext();){
|
|
if(OU == null){
|
final String returnAttributeName = iter.next();
|
final List<String> OUList = jdbcm.getOrganizationalUnits(baseDN);
|
|
if(OUList == null) throw new NullPointerException("JDBC Error: Could not find any matching organizational units in DN: '" + baseDN + "'.");
|
|
for(Iterator<String> iterOU = OUList.iterator(); iterOU.hasNext();){
|
OUName = iterOU.next();
|
final String tableName = jdbcm.getTableNameFromMapping(baseDN, OUName);
|
final String returnColumnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, OUName, returnAttributeName);
|
|
if(returnColumnName == null) continue;
|
else returnColumnNames.add(returnColumnName);
|
break;
|
}
|
}
|
else {
|
final String returnAttributeName = iter.next();
|
final String tableName = jdbcm.getTableNameFromMapping(baseDN, OUName);
|
|
if(tableName == null) throw new NullPointerException("JDBC Error: Could not find matching table name for OU: '" + OUName + "' in DN: '" + baseDN + "'. Please check if mapping was succesful.");
|
final String returnColumnName = jdbcm.getColumnNameFromMapping(tableName, baseDN, OUName, returnAttributeName);
|
|
if(returnColumnName == null) throw new NullPointerException("JDBC Error: Could not find matching column name for attribute: '" + returnAttributeName + "'. Please check if mapping was succesful.");
|
returnColumnNames.add(returnColumnName);
|
}
|
}
|
|
String selectString = "";
|
String fromString = "";
|
String whereString = convertedFilterString;
|
|
if(returnAttributeNames.isEmpty()) selectString = "*";
|
|
//Build the SQL SELECT string.
|
if(OU == null){
|
final List<String> OUList = jdbcm.getOrganizationalUnits(baseDN);
|
|
if(OUList == null) throw new NullPointerException("JDBC Error: Could not find any matching organizational units in DN: '" + baseDN + "'.");
|
|
for(int i = 0; i < OUList.size(); i++){
|
final String currentOU = OUList.get(i);
|
final String currentTable = jdbcm.getTableNameFromMapping(baseDN, currentOU);
|
|
if(currentTable == null) continue;
|
final List<String> currentTableColumnNames = jdbcm.getTableColumns(currentTable);
|
|
if(!fromString.isEmpty()) fromString = fromString.concat(",");
|
fromString = fromString.concat(currentTable);
|
|
for(int j = 0; j < returnColumnNames.size(); j++){
|
final String returnColumnName = returnColumnNames.get(j);
|
|
if(currentTableColumnNames.contains(returnColumnName)){
|
|
if(!selectString.isEmpty()) selectString = selectString.concat(",");
|
selectString = selectString.concat(currentTable + "." + returnColumnName);
|
}
|
}
|
}
|
}
|
else{
|
final String selectTable = jdbcm.getTableNameFromMapping(baseDN, OUName);
|
if(selectTable == null) throw new NullPointerException("JDBC Error: Could not find matching table name for OU: '" + OUName + "' in DN: '" + baseDN + "'. Please check if mapping was succesful.");
|
|
fromString = fromString.concat(selectTable);
|
|
for(int j = 0; j < returnColumnNames.size(); j++){
|
final String returnColumnName = returnColumnNames.get(j);
|
|
if(returnColumnName == null) continue;
|
|
if(!selectString.isEmpty()) selectString = selectString.concat(",");
|
selectString = selectString.concat(returnColumnName);
|
}
|
|
if(rDN != null){
|
final String columnName = jdbcm.getColumnNameFromMapping(selectTable, baseDN, OUName, rDNAttributeName);
|
|
if(columnName == null) throw new NullPointerException("JDBC Error: Could not find matching column name for attribute: '" + rDNAttributeName + "'. Please check if mapping was succesful.");
|
whereString = columnName + "='" + rDNAttributeValue + "'";
|
}
|
}
|
|
//Build the SQL query.
|
final Statement st = connection.createStatement();
|
String sql = "SELECT " + selectString + " FROM " + fromString;
|
|
if(!whereString.isEmpty()) sql = sql.concat(" WHERE " + whereString);
|
ResultSet rs = st.executeQuery(sql);
|
ResultSetMetaData rsmd = rs.getMetaData();
|
|
//Fill the entries result for this query.
|
searchEntries.clear();
|
while (rs.next()) {
|
Entry e = new LinkedHashMapEntry();
|
|
for(int i = 1; i <= rsmd.getColumnCount(); i++){
|
e.addAttribute(rsmd.getTableName(i) + "_" + rsmd.getColumnName(i), rs.getObject(i));
|
}
|
searchEntries.add(e);
|
}
|
r = Responses.newResult(ResultCode.SUCCESS);
|
}catch (SQLException e) {
|
System.out.println(e.toString());
|
r = Responses.newResult(ResultCode.OPERATIONS_ERROR);
|
r.setCause(e);
|
}catch (NullPointerException e){
|
System.out.println(e.toString());
|
r = Responses.newCompareResult(ResultCode.OPERATIONS_ERROR);
|
r.setCause(e);
|
}
|
return r;
|
}
|
|
/**
|
* {@inheritDoc}
|
*/
|
@Override
|
public String toString()
|
{
|
return connection.toString();
|
}
|
}
|
