/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License, Version 1.0 only
* (the "License"). You may not use this file except in compliance
* with the License.
*
* You can obtain a copy of the license at
* trunk/nasutekds/resource/legal-notices/NasuTekDS.LICENSE
* or https://NasuTekDS.dev.java.net/NasuTekDS.LICENSE.
* See the License for the specific language governing permissions
* and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at
* trunk/nasutekds/resource/legal-notices/NasuTekDS.LICENSE. If applicable,
* add the following below this CDDL HEADER, with the fields enclosed
* by brackets "[]" replaced with your own identifying information:
* Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*
*
* Copyright 2006-2009 Sun Microsystems, Inc.
*/
package org.nasutekds.server.extensions;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.security.MessageDigest;
import java.util.Arrays;
import java.util.Date;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.TreeSet;
import java.util.TreeMap;
import java.util.zip.Deflater;
import java.util.zip.GZIPInputStream;
import java.util.zip.GZIPOutputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CopyOnWriteArrayList;
import javax.crypto.Mac;
import org.nasutekds.messages.Message;
import org.nasutekds.messages.MessageBuilder;
import org.nasutekds.server.admin.Configuration;
import org.nasutekds.server.api.AlertGenerator;
import org.nasutekds.server.api.ClientConnection;
import org.nasutekds.server.api.ConfigAddListener;
import org.nasutekds.server.api.ConfigChangeListener;
import org.nasutekds.server.api.ConfigDeleteListener;
import org.nasutekds.server.api.ConfigHandler;
import org.nasutekds.server.config.ConfigEntry;
import org.nasutekds.server.config.ConfigException;
import org.nasutekds.server.core.AddOperation;
import org.nasutekds.server.core.DeleteOperation;
import org.nasutekds.server.core.DirectoryServer;
import org.nasutekds.server.core.ModifyOperation;
import org.nasutekds.server.core.ModifyDNOperation;
import org.nasutekds.server.core.SearchOperation;
import org.nasutekds.server.loggers.debug.DebugTracer;
import org.nasutekds.server.schema.GeneralizedTimeSyntax;
import org.nasutekds.server.tools.LDIFModify;
import org.nasutekds.server.types.*;
import org.nasutekds.server.util.DynamicConstants;
import org.nasutekds.server.util.LDIFException;
import org.nasutekds.server.util.LDIFReader;
import org.nasutekds.server.util.LDIFWriter;
import org.nasutekds.server.util.TimeThread;
import static org.nasutekds.server.config.ConfigConstants.*;
import static org.nasutekds.server.extensions.ExtensionsConstants.*;
import static org.nasutekds.server.loggers.ErrorLogger.*;
import static org.nasutekds.server.loggers.debug.DebugLogger.*;
import static org.nasutekds.messages.ConfigMessages.*;
import static org.nasutekds.server.util.ServerConstants.*;
import static org.nasutekds.server.util.StaticUtils.*;
/**
* This class defines a simple configuration handler for the Directory Server
* that will read the server configuration from an LDIF file.
*/
public class ConfigFileHandler
extends ConfigHandler
implements AlertGenerator
{
/**
* The tracer object for the debug logger.
*/
private static final DebugTracer TRACER = getTracer();
/**
* The fully-qualified name of this class.
*/
private static final String CLASS_NAME =
"org.nasutekds.server.extensions.ConfigFileHandler";
/**
* The set of supported control OIDs for this backend.
*/
private static final HashSet<String> SUPPORTED_CONTROLS =
new HashSet<String>(0);
/**
* The set of supported feature OIDs for this backend.
*/
private static final HashSet<String> SUPPORTED_FEATURES =
new HashSet<String>(0);
/**
* The privilege array containing both the CONFIG_READ and CONFIG_WRITE
* privileges.
*/
private static final Privilege[] CONFIG_READ_AND_WRITE =
{
Privilege.CONFIG_READ,
Privilege.CONFIG_WRITE
};
// Indicates whether to maintain a configuration archive.
private boolean maintainConfigArchive;
// Indicates whether to start using the last known good configuration.
private boolean useLastKnownGoodConfig;
// A SHA-1 digest of the last known configuration. This should only be
// incorrect if the server configuration file has been manually edited with
// the server online, which is a bad thing.
private byte[] configurationDigest;
// The mapping that holds all of the configuration entries that have been read
// from the LDIF file.
private ConcurrentHashMap<DN,ConfigEntry> configEntries;
// The reference to the configuration root entry.
private ConfigEntry configRootEntry;
// The set of base DNs for this config handler backend.
private DN[] baseDNs;
// The maximum config archive size to maintain.
private int maxConfigArchiveSize;
// The write lock used to ensure that only one thread can apply a
// configuration update at any given time.
private Object configLock;
// The path to the configuration file.
private String configFile;
// The install root directory for the Directory Server.
private String serverRoot;
// The instance root directory for the Directory Server.
private String instanceRoot;
/**
* Creates a new instance of this config file handler. No initialization
* should be performed here, as all of that work should be done in the
* <CODE>initializeConfigHandler</CODE> method.
*/
public ConfigFileHandler()
{
super();
}
/**
* {@inheritDoc}
*/
@Override()
public void initializeConfigHandler(String configFile, boolean checkSchema)
throws InitializationException
{
// Initialize the config lock.
configLock = new Object();
// Determine whether we should try to start using the last known good
// configuration. If so, then only do so if such a file exists. If it
// doesn't exist, then fall back on the active configuration file.
this.configFile = configFile;
DirectoryEnvironmentConfig envConfig =
DirectoryServer.getEnvironmentConfig();
useLastKnownGoodConfig = envConfig.useLastKnownGoodConfiguration();
File f = null;
if (useLastKnownGoodConfig)
{
f = new File(configFile + ".startok");
if (! f.exists())
{
logError(WARN_CONFIG_FILE_NO_STARTOK_FILE.get(f.getAbsolutePath(),
configFile));
useLastKnownGoodConfig = false;
f = new File(configFile);
}
else
{
logError(NOTE_CONFIG_FILE_USING_STARTOK_FILE.get(f.getAbsolutePath(),
configFile));
}
}
else
{
f = new File(configFile);
}
try
{
if (! f.exists())
{
Message message = ERR_CONFIG_FILE_DOES_NOT_EXIST.get(
f.getAbsolutePath());
throw new InitializationException(message);
}
}
catch (InitializationException ie)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, ie);
}
throw ie;
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_FILE_CANNOT_VERIFY_EXISTENCE.get(
f.getAbsolutePath(), String.valueOf(e));
throw new InitializationException(message);
}
// Check to see if a configuration archive exists. If not, then create one.
// If so, then check whether the current configuration matches the last
// configuration in the archive. If it doesn't, then archive it.
maintainConfigArchive = envConfig.maintainConfigArchive();
maxConfigArchiveSize = envConfig.getMaxConfigArchiveSize();
if (maintainConfigArchive & (! useLastKnownGoodConfig))
{
try
{
configurationDigest = calculateConfigDigest();
}
catch (DirectoryException de)
{
throw new InitializationException(de.getMessageObject(), de.getCause());
}
File archiveDirectory = new File(f.getParent(), CONFIG_ARCHIVE_DIR_NAME);
if (archiveDirectory.exists())
{
try
{
byte[] lastDigest = getLastConfigDigest(archiveDirectory);
if (! Arrays.equals(configurationDigest, lastDigest))
{
writeConfigArchive();
}
} catch (Exception e) {}
}
else
{
writeConfigArchive();
}
}
// Fixme -- Should we add a hash or signature check here?
// See if there is a config changes file. If there is, then try to apply
// the changes contained in it.
File changesFile = new File(f.getParent(), CONFIG_CHANGES_NAME);
try
{
if (changesFile.exists())
{
applyChangesFile(f, changesFile);
if (maintainConfigArchive)
{
configurationDigest = calculateConfigDigest();
writeConfigArchive();
}
}
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_UNABLE_TO_APPLY_STARTUP_CHANGES.get(
changesFile.getAbsolutePath(), String.valueOf(e));
throw new InitializationException(message, e);
}
// We will use the LDIF reader to read the configuration file. Create an
// LDIF import configuration to do this and then get the reader.
LDIFReader reader;
try
{
LDIFImportConfig importConfig = new LDIFImportConfig(f.getAbsolutePath());
// FIXME -- Should we support encryption or compression for the config?
reader = new LDIFReader(importConfig);
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_FILE_CANNOT_OPEN_FOR_READ.get(
f.getAbsolutePath(), String.valueOf(e));
throw new InitializationException(message, e);
}
// Read the first entry from the configuration file.
Entry entry;
try
{
entry = reader.readEntry(checkSchema);
}
catch (LDIFException le)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, le);
}
try
{
reader.close();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
}
Message message = ERR_CONFIG_FILE_INVALID_LDIF_ENTRY.get(
le.getLineNumber(), f.getAbsolutePath(), String.valueOf(le));
throw new InitializationException(message, le);
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
try
{
reader.close();
}
catch (Exception e2)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e2);
}
}
Message message =
ERR_CONFIG_FILE_READ_ERROR.get(f.getAbsolutePath(),
String.valueOf(e));
throw new InitializationException(message, e);
}
// Make sure that the provide LDIF file is not empty.
if (entry == null)
{
try
{
reader.close();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
}
Message message = ERR_CONFIG_FILE_EMPTY.get(f.getAbsolutePath());
throw new InitializationException(message);
}
// Make sure that the DN of this entry is equal to the config root DN.
try
{
DN configRootDN = DN.decode(DN_CONFIG_ROOT);
if (! entry.getDN().equals(configRootDN))
{
Message message = ERR_CONFIG_FILE_INVALID_BASE_DN.get(
f.getAbsolutePath(), entry.getDN().toString(),
DN_CONFIG_ROOT);
throw new InitializationException(message);
}
}
catch (InitializationException ie)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, ie);
}
try
{
reader.close();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
}
throw ie;
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
try
{
reader.close();
}
catch (Exception e2)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e2);
}
}
// This should not happen, so we can use a generic error here.
Message message = ERR_CONFIG_FILE_GENERIC_ERROR.get(f.getAbsolutePath(),
String.valueOf(e));
throw new InitializationException(message, e);
}
// Convert the entry to a configuration entry and put it in the config
// hash.
configEntries = new ConcurrentHashMap<DN,ConfigEntry>();
configRootEntry = new ConfigEntry(entry, null);
configEntries.put(entry.getDN(), configRootEntry);
// Iterate through the rest of the configuration file and process the
// remaining entries.
while (true)
{
// Read the next entry from the configuration.
try
{
entry = reader.readEntry(checkSchema);
}
catch (LDIFException le)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, le);
}
try
{
reader.close();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
}
Message message = ERR_CONFIG_FILE_INVALID_LDIF_ENTRY.get(
le.getLineNumber(), f.getAbsolutePath(),
String.valueOf(le));
throw new InitializationException(message, le);
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
try
{
reader.close();
}
catch (Exception e2)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e2);
}
}
Message message = ERR_CONFIG_FILE_READ_ERROR.get(f.getAbsolutePath(),
String.valueOf(e));
throw new InitializationException(message, e);
}
// If the entry is null, then we have reached the end of the configuration
// file.
if (entry == null)
{
try
{
reader.close();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
}
break;
}
// Make sure that the DN of the entry read doesn't already exist.
DN entryDN = entry.getDN();
if (configEntries.containsKey(entryDN))
{
try
{
reader.close();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
}
Message message = ERR_CONFIG_FILE_DUPLICATE_ENTRY.get(
entryDN.toString(),
String.valueOf(reader.getLastEntryLineNumber()),
f.getAbsolutePath());
throw new InitializationException(message);
}
// Make sure that the parent DN of the entry read does exist.
DN parentDN = entryDN.getParent();
if (parentDN == null)
{
try
{
reader.close();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
}
Message message = ERR_CONFIG_FILE_UNKNOWN_PARENT.get(
entryDN.toString(),
reader.getLastEntryLineNumber(),
f.getAbsolutePath());
throw new InitializationException(message);
}
ConfigEntry parentEntry = configEntries.get(parentDN);
if (parentEntry == null)
{
try
{
reader.close();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
}
Message message = ERR_CONFIG_FILE_NO_PARENT.get(entryDN.toString(),
reader.getLastEntryLineNumber(),
f.getAbsolutePath(), parentDN.toString());
throw new InitializationException(message);
}
// Create the new configuration entry, add it as a child of the provided
// parent entry, and put it into the entry has.
try
{
ConfigEntry configEntry = new ConfigEntry(entry, parentEntry);
parentEntry.addChild(configEntry);
configEntries.put(entryDN, configEntry);
}
catch (Exception e)
{
// This should not happen.
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
try
{
reader.close();
}
catch (Exception e2)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e2);
}
}
Message message = ERR_CONFIG_FILE_GENERIC_ERROR.get(f.getAbsolutePath(),
String.valueOf(e));
throw new InitializationException(message, e);
}
}
// Determine the appropriate server root. If it's not defined in the
// environment config, then try to figure it out from the location of the
// configuration file.
File rootFile = envConfig.getServerRoot();
if (rootFile == null)
{
try
{
File configDirFile = f.getParentFile();
if ((configDirFile != null) &&
configDirFile.getName().equals(CONFIG_DIR_NAME))
{
/*
* Do a best effort to avoid having a relative representation (for
* instance to avoid having ../../../).
*/
try
{
serverRoot = configDirFile.getParentFile().getCanonicalPath();
}
catch (IOException ioe)
{
// Best effort
serverRoot = configDirFile.getParentFile().getAbsolutePath();
}
}
if (serverRoot == null)
{
Message message = ERR_CONFIG_CANNOT_DETERMINE_SERVER_ROOT.get(
ENV_VAR_INSTALL_ROOT);
throw new InitializationException(message);
}
}
catch (InitializationException ie)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, ie);
}
throw ie;
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message =
ERR_CONFIG_CANNOT_DETERMINE_SERVER_ROOT.get(ENV_VAR_INSTALL_ROOT);
throw new InitializationException(message);
}
}
else
{
/*
* Do a best effort to avoid having a relative representation (for
* instance to avoid having ../../../).
*/
try
{
serverRoot = rootFile.getCanonicalPath();
}
catch (IOException ioe)
{
// Best effort
serverRoot = rootFile.getAbsolutePath();
}
}
// Determine the appropriate server root. If it's not defined in the
// environment config, then try to figure it out from the location of the
// configuration file.
File instanceFile = envConfig.getInstanceRootFromServerRoot(new File(
serverRoot));
if (instanceFile == null)
{
Message message =
ERR_CONFIG_CANNOT_DETERMINE_SERVER_ROOT.get(ENV_VAR_INSTALL_ROOT);
throw new InitializationException(message);
}
else
{
/*
* Do a best effort to avoid having a relative representation (for
* instance to avoid having ../../../).
*/
try
{
instanceRoot = instanceFile.getCanonicalPath();
}
catch (IOException ioe)
{
// Best effort
instanceRoot = instanceFile.getAbsolutePath();
}
}
// Register with the Directory Server as an alert generator.
DirectoryServer.registerAlertGenerator(this);
// Register with the Directory Server as the backend that should be used
// when accessing the configuration.
baseDNs = new DN[] { configRootEntry.getDN() };
try
{
// Set a backend ID for the config backend. Try to avoid potential
// conflict with user backend identifiers.
setBackendID("__config.ldif__");
DirectoryServer.registerBaseDN(configRootEntry.getDN(), this, true);
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_CANNOT_REGISTER_AS_PRIVATE_SUFFIX.get(
String.valueOf(configRootEntry.getDN()), getExceptionMessage(e));
throw new InitializationException(message, e);
}
}
/**
* Calculates a SHA-1 digest of the current configuration file.
*
* @return The calculated configuration digest.
*
* @throws DirectoryException If a problem occurs while calculating the
* digest.
*/
private byte[] calculateConfigDigest()
throws DirectoryException
{
InputStream inputStream = null;
try
{
MessageDigest sha1Digest =
MessageDigest.getInstance(MESSAGE_DIGEST_ALGORITHM_SHA_1);
inputStream = new FileInputStream(configFile);
byte[] buffer = new byte[8192];
while (true)
{
int bytesRead = inputStream.read(buffer);
if (bytesRead < 0)
{
break;
}
sha1Digest.update(buffer, 0, bytesRead);
}
return sha1Digest.digest();
}
catch (Exception e)
{
Message message = ERR_CONFIG_CANNOT_CALCULATE_DIGEST.get(
configFile, stackTraceToSingleLineString(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
finally
{
if (inputStream != null)
{
try
{
inputStream.close();
}
catch (IOException e) {
// ignore;
}
}
}
}
/**
* Looks at the existing archive directory, finds the latest archive file,
* and calculates a SHA-1 digest of that file.
*
* @return The calculated digest of the most recent archived configuration
* file.
*
* @throws DirectoryException If a problem occurs while calculating the
* digest.
*/
private byte[] getLastConfigDigest(File archiveDirectory)
throws DirectoryException
{
int latestCounter = 0;
long latestTimestamp = -1;
String latestFileName = null;
for (String name : archiveDirectory.list())
{
if (! name.startsWith("config-"))
{
continue;
}
int dotPos = name.indexOf('.', 7);
if (dotPos < 0)
{
continue;
}
int dashPos = name.indexOf('-', 7);
if (dashPos < 0)
{
try
{
ByteString ts = ByteString.valueOf(name.substring(7, dotPos));
long timestamp = GeneralizedTimeSyntax.decodeGeneralizedTimeValue(ts);
if (timestamp > latestTimestamp)
{
latestFileName = name;
latestTimestamp = timestamp;
latestCounter = 0;
continue;
}
}
catch (Exception e)
{
continue;
}
}
else
{
try
{
ByteString ts = ByteString.valueOf(name.substring(7, dashPos));
long timestamp = GeneralizedTimeSyntax.decodeGeneralizedTimeValue(ts);
int counter = Integer.parseInt(name.substring(dashPos+1, dotPos));
if (timestamp > latestTimestamp)
{
latestFileName = name;
latestTimestamp = timestamp;
latestCounter = counter;
continue;
}
else if ((timestamp == latestTimestamp) && (counter > latestCounter))
{
latestFileName = name;
latestTimestamp = timestamp;
latestCounter = counter;
continue;
}
}
catch (Exception e)
{
continue;
}
}
}
if (latestFileName == null)
{
return null;
}
File latestFile = new File(archiveDirectory, latestFileName);
try
{
MessageDigest sha1Digest =
MessageDigest.getInstance(MESSAGE_DIGEST_ALGORITHM_SHA_1);
GZIPInputStream inputStream =
new GZIPInputStream(new FileInputStream(latestFile));
byte[] buffer = new byte[8192];
while (true)
{
int bytesRead = inputStream.read(buffer);
if (bytesRead < 0)
{
break;
}
sha1Digest.update(buffer, 0, bytesRead);
}
return sha1Digest.digest();
}
catch (Exception e)
{
Message message = ERR_CONFIG_CANNOT_CALCULATE_DIGEST.get(
latestFile.getAbsolutePath(), stackTraceToSingleLineString(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
}
/**
* Applies the updates in the provided changes file to the content in the
* specified source file. The result will be written to a temporary file, the
* current source file will be moved out of place, and then the updated file
* will be moved into the place of the original file. The changes file will
* also be renamed so it won't be applied again.
* <BR><BR>
* If any problems are encountered, then the config initialization process
* will be aborted.
*
* @param sourceFile The LDIF file containing the source data.
* @param changesFile The LDIF file containing the changes to apply.
*
* @throws IOException If a problem occurs while performing disk I/O.
*
* @throws LDIFException If a problem occurs while trying to interpret the
* data.
*/
private void applyChangesFile(File sourceFile, File changesFile)
throws IOException, LDIFException
{
// Create the appropriate LDIF readers and writer.
LDIFImportConfig importConfig =
new LDIFImportConfig(sourceFile.getAbsolutePath());
importConfig.setValidateSchema(false);
LDIFReader sourceReader = new LDIFReader(importConfig);
importConfig = new LDIFImportConfig(changesFile.getAbsolutePath());
importConfig.setValidateSchema(false);
LDIFReader changesReader = new LDIFReader(importConfig);
String tempFile = changesFile.getAbsolutePath() + ".tmp";
LDIFExportConfig exportConfig =
new LDIFExportConfig(tempFile, ExistingFileBehavior.OVERWRITE);
LDIFWriter targetWriter = new LDIFWriter(exportConfig);
// Apply the changes and make sure there were no errors.
LinkedList<Message> errorList = new LinkedList<Message>();
boolean successful = LDIFModify.modifyLDIF(sourceReader, changesReader,
targetWriter, errorList);
try
{
sourceReader.close();
} catch (Exception e) {}
try
{
changesReader.close();
} catch (Exception e) {}
try
{
targetWriter.close();
} catch (Exception e) {}
if (! successful)
{
// FIXME -- Log each error message and throw an exception.
for (Message s : errorList)
{
Message message = ERR_CONFIG_ERROR_APPLYING_STARTUP_CHANGE.get(s);
logError(message);
}
Message message = ERR_CONFIG_UNABLE_TO_APPLY_CHANGES_FILE.get();
throw new LDIFException(message);
}
// Move the current config file out of the way and replace it with the
// updated version.
File oldSource = new File(sourceFile.getAbsolutePath() + ".prechanges");
if (oldSource.exists())
{
oldSource.delete();
}
sourceFile.renameTo(oldSource);
new File(tempFile).renameTo(sourceFile);
// Move the changes file out of the way so it doesn't get applied again.
File newChanges = new File(changesFile.getAbsolutePath() + ".applied");
if (newChanges.exists())
{
newChanges.delete();
}
changesFile.renameTo(newChanges);
}
/**
* {@inheritDoc}
*/
@Override()
public void finalizeConfigHandler()
{
try
{
DirectoryServer.deregisterBaseDN(configRootEntry.getDN());
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
}
}
/**
* {@inheritDoc}
*/
@Override()
public void finalizeBackend()
{
// No implementation is required.
}
/**
* {@inheritDoc}
*/
@Override()
public ConfigEntry getConfigRootEntry()
throws ConfigException
{
return configRootEntry;
}
/**
* {@inheritDoc}
*/
@Override()
public ConfigEntry getConfigEntry(DN entryDN)
throws ConfigException
{
return configEntries.get(entryDN);
}
/**
* {@inheritDoc}
*/
@Override()
public String getServerRoot()
{
return serverRoot;
}
/**
* {@inheritDoc}
*/
@Override()
public String getInstanceRoot()
{
return instanceRoot;
}
/**
* {@inheritDoc}
*/
@Override()
public void configureBackend(Configuration cfg)
throws ConfigException
{
// No action is required.
}
/**
* {@inheritDoc}
*/
@Override()
public void initializeBackend()
throws ConfigException, InitializationException
{
// No action is required, since all initialization was performed in the
// initializeConfigHandler method.
}
/**
* {@inheritDoc}
*/
@Override()
public DN[] getBaseDNs()
{
return baseDNs;
}
/**
* {@inheritDoc}
*/
@Override()
public long getEntryCount()
{
return configEntries.size();
}
/**
* {@inheritDoc}
*/
@Override()
public boolean isLocal()
{
// The configuration information will always be local.
return true;
}
/**
* {@inheritDoc}
*/
@Override()
public boolean isIndexed(AttributeType attributeType, IndexType indexType)
{
// All searches in this backend will always be considered indexed.
return true;
}
/**
* {@inheritDoc}
*/
@Override()
public ConditionResult hasSubordinates(DN entryDN)
throws DirectoryException
{
ConfigEntry baseEntry = configEntries.get(entryDN);
if(baseEntry == null)
{
return ConditionResult.UNDEFINED;
}
else if(baseEntry.hasChildren())
{
return ConditionResult.TRUE;
}
else
{
return ConditionResult.FALSE;
}
}
/**
* {@inheritDoc}
*/
@Override()
public long numSubordinates(DN entryDN, boolean subtree)
throws DirectoryException
{
ConfigEntry baseEntry = configEntries.get(entryDN);
if (baseEntry == null)
{
return -1;
}
if(!subtree)
{
return baseEntry.getChildren().size();
}
else
{
long count = 0;
for(ConfigEntry child : baseEntry.getChildren().values())
{
count += numSubordinates(child.getDN(), true);
count ++;
}
return count;
}
}
/**
* {@inheritDoc}
*/
@Override()
public Entry getEntry(DN entryDN)
throws DirectoryException
{
ConfigEntry configEntry = configEntries.get(entryDN);
if (configEntry == null)
{
return null;
}
return configEntry.getEntry().duplicate(true);
}
/**
* {@inheritDoc}
*/
@Override()
public boolean entryExists(DN entryDN)
throws DirectoryException
{
return configEntries.containsKey(entryDN);
}
/**
* {@inheritDoc}
*/
@Override()
public void addEntry(Entry entry, AddOperation addOperation)
throws DirectoryException
{
Entry e = entry.duplicate(false);
// If there is an add operation, then make sure that the associated user has
// both the CONFIG_READ and CONFIG_WRITE privileges.
if (addOperation != null)
{
ClientConnection clientConnection = addOperation.getClientConnection();
if (! (clientConnection.hasAllPrivileges(CONFIG_READ_AND_WRITE,
addOperation)))
{
Message message = ERR_CONFIG_FILE_ADD_INSUFFICIENT_PRIVILEGES.get();
throw new DirectoryException(ResultCode.INSUFFICIENT_ACCESS_RIGHTS,
message);
}
}
// Grab the config lock to ensure that only one config update may be in
// progress at any given time.
synchronized (configLock)
{
// Make sure that the target DN does not already exist. If it does, then
// fail.
DN entryDN = e.getDN();
if (configEntries.containsKey(entryDN))
{
Message message =
ERR_CONFIG_FILE_ADD_ALREADY_EXISTS.get(String.valueOf(entryDN));
throw new DirectoryException(ResultCode.ENTRY_ALREADY_EXISTS, message);
}
// Make sure that the entry's parent exists. If it does not, then fail.
DN parentDN = entryDN.getParent();
if (parentDN == null)
{
// The entry DN doesn't have a parent. This is not allowed.
Message message =
ERR_CONFIG_FILE_ADD_NO_PARENT_DN.get(String.valueOf(entryDN));
throw new DirectoryException(ResultCode.NO_SUCH_OBJECT, message);
}
ConfigEntry parentEntry = configEntries.get(parentDN);
if (parentEntry == null)
{
// The parent entry does not exist. This is not allowed.
Message message = ERR_CONFIG_FILE_ADD_NO_PARENT.get(
String.valueOf(entryDN),
String.valueOf(parentDN));
// Get the matched DN, if possible.
DN matchedDN = null;
parentDN = parentDN.getParent();
while (parentDN != null)
{
if (configEntries.containsKey(parentDN))
{
matchedDN = parentDN;
break;
}
parentDN = parentDN.getParent();
}
throw new DirectoryException(ResultCode.NO_SUCH_OBJECT, message,
matchedDN, null);
}
// Encapsulate the provided entry in a config entry.
ConfigEntry newEntry = new ConfigEntry(e, parentEntry);
// See if the parent entry has any add listeners. If so, then iterate
// through them and make sure the new entry is acceptable.
CopyOnWriteArrayList<ConfigAddListener> addListeners =
parentEntry.getAddListeners();
MessageBuilder unacceptableReason = new MessageBuilder();
for (ConfigAddListener l : addListeners)
{
if (! l.configAddIsAcceptable(newEntry, unacceptableReason))
{
Message message = ERR_CONFIG_FILE_ADD_REJECTED_BY_LISTENER.
get(String.valueOf(entryDN), String.valueOf(parentDN),
String.valueOf(unacceptableReason));
throw new DirectoryException(
ResultCode.UNWILLING_TO_PERFORM, message);
}
}
// At this point, we will assume that everything is OK and proceed with
// the add.
try
{
parentEntry.addChild(newEntry);
configEntries.put(entryDN, newEntry);
writeUpdatedConfig();
}
catch (ConfigException ce)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, ce);
}
Message message = ERR_CONFIG_FILE_ADD_FAILED.
get(String.valueOf(entryDN), String.valueOf(parentDN),
getExceptionMessage(ce));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message);
}
// Notify all the add listeners that the entry has been added.
ResultCode resultCode = ResultCode.SUCCESS;
LinkedList<Message> messages = new LinkedList<Message>();
for (ConfigAddListener l : addListeners)
{
ConfigChangeResult result = l.applyConfigurationAdd(newEntry);
if (result.getResultCode() != ResultCode.SUCCESS)
{
if (resultCode == ResultCode.SUCCESS)
{
resultCode = result.getResultCode();
}
messages.addAll(result.getMessages());
}
handleConfigChangeResult(result, newEntry.getDN(),
l.getClass().getName(),
"applyConfigurationAdd");
}
if (resultCode != ResultCode.SUCCESS)
{
MessageBuilder buffer = new MessageBuilder();
if (! messages.isEmpty())
{
Iterator<Message> iterator = messages.iterator();
buffer.append(iterator.next());
while (iterator.hasNext())
{
buffer.append(". ");
buffer.append(iterator.next());
}
}
Message message =
ERR_CONFIG_FILE_ADD_APPLY_FAILED.get(String.valueOf(buffer));
throw new DirectoryException(resultCode, message);
}
}
}
/**
* {@inheritDoc}
*/
@Override()
public void deleteEntry(DN entryDN, DeleteOperation deleteOperation)
throws DirectoryException
{
// If there is a delete operation, then make sure that the associated user
// has both the CONFIG_READ and CONFIG_WRITE privileges.
if (deleteOperation != null)
{
ClientConnection clientConnection = deleteOperation.getClientConnection();
if (! (clientConnection.hasAllPrivileges(CONFIG_READ_AND_WRITE,
deleteOperation)))
{
Message message = ERR_CONFIG_FILE_DELETE_INSUFFICIENT_PRIVILEGES.get();
throw new DirectoryException(ResultCode.INSUFFICIENT_ACCESS_RIGHTS,
message);
}
}
// Grab the config lock to ensure that only one config update may be in
// progress at any given time.
synchronized (configLock)
{
// Get the target entry. If it does not exist, then fail.
ConfigEntry entry = configEntries.get(entryDN);
if (entry == null)
{
// Try to find the matched DN if possible.
DN matchedDN = null;
if (entryDN.isDescendantOf(configRootEntry.getDN()))
{
DN parentDN = entryDN.getParent();
while (parentDN != null)
{
if (configEntries.containsKey(parentDN))
{
matchedDN = parentDN;
break;
}
parentDN = parentDN.getParent();
}
}
Message message =
ERR_CONFIG_FILE_DELETE_NO_SUCH_ENTRY.get(String.valueOf(entryDN));
throw new DirectoryException(ResultCode.NO_SUCH_OBJECT, message,
matchedDN, null);
}
// If the entry has children, then fail.
if (entry.hasChildren())
{
Message message =
ERR_CONFIG_FILE_DELETE_HAS_CHILDREN.get(String.valueOf(entryDN));
throw new DirectoryException(ResultCode.NOT_ALLOWED_ON_NONLEAF,
message);
}
// Get the parent entry. If there isn't one, then it must be the config
// root, which we won't allow.
ConfigEntry parentEntry = entry.getParent();
if (parentEntry == null)
{
Message message =
ERR_CONFIG_FILE_DELETE_NO_PARENT.get(String.valueOf(entryDN));
throw new DirectoryException(ResultCode.UNWILLING_TO_PERFORM, message);
}
// Get the delete listeners from the parent and make sure that they are
// all OK with the delete.
CopyOnWriteArrayList<ConfigDeleteListener> deleteListeners =
parentEntry.getDeleteListeners();
MessageBuilder unacceptableReason = new MessageBuilder();
for (ConfigDeleteListener l : deleteListeners)
{
if (! l.configDeleteIsAcceptable(entry, unacceptableReason))
{
Message message = ERR_CONFIG_FILE_DELETE_REJECTED.
get(String.valueOf(entryDN), String.valueOf(parentEntry.getDN()),
String.valueOf(unacceptableReason));
throw new DirectoryException(ResultCode.UNWILLING_TO_PERFORM,
message);
}
}
// At this point, we will assume that everything is OK and proceed with
// the delete.
try
{
parentEntry.removeChild(entryDN);
configEntries.remove(entryDN);
writeUpdatedConfig();
}
catch (ConfigException ce)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, ce);
}
Message message = ERR_CONFIG_FILE_DELETE_FAILED.
get(String.valueOf(entryDN), String.valueOf(parentEntry.getDN()),
getExceptionMessage(ce));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message);
}
// Notify all the delete listeners that the entry has been removed.
ResultCode resultCode = ResultCode.SUCCESS;
LinkedList<Message> messages = new LinkedList<Message>();
for (ConfigDeleteListener l : deleteListeners)
{
ConfigChangeResult result = l.applyConfigurationDelete(entry);
if (result.getResultCode() != ResultCode.SUCCESS)
{
if (resultCode == ResultCode.SUCCESS)
{
resultCode = result.getResultCode();
}
messages.addAll(result.getMessages());
}
handleConfigChangeResult(result, entry.getDN(),
l.getClass().getName(),
"applyConfigurationDelete");
}
if (resultCode != ResultCode.SUCCESS)
{
StringBuilder buffer = new StringBuilder();
if (! messages.isEmpty())
{
Iterator<Message> iterator = messages.iterator();
buffer.append(iterator.next());
while (iterator.hasNext())
{
buffer.append(". ");
buffer.append(iterator.next());
}
}
Message message =
ERR_CONFIG_FILE_DELETE_APPLY_FAILED.get(String.valueOf(buffer));
throw new DirectoryException(resultCode, message);
}
}
}
/**
* {@inheritDoc}
*/
@Override()
public void replaceEntry(Entry oldEntry, Entry newEntry,
ModifyOperation modifyOperation) throws DirectoryException
{
Entry e = newEntry.duplicate(false);
// If there is a modify operation, then make sure that the associated user
// has both the CONFIG_READ and CONFIG_WRITE privileges. Also, if the
// operation targets the set of root privileges then make sure the user has
// the PRIVILEGE_CHANGE privilege.
if (modifyOperation != null)
{
ClientConnection clientConnection = modifyOperation.getClientConnection();
if (! (clientConnection.hasAllPrivileges(CONFIG_READ_AND_WRITE,
modifyOperation)))
{
Message message = ERR_CONFIG_FILE_MODIFY_INSUFFICIENT_PRIVILEGES.get();
throw new DirectoryException(ResultCode.INSUFFICIENT_ACCESS_RIGHTS,
message);
}
AttributeType privType =
DirectoryServer.getAttributeType(ATTR_DEFAULT_ROOT_PRIVILEGE_NAME,
true);
for (Modification m : modifyOperation.getModifications())
{
if (m.getAttribute().getAttributeType().equals(privType))
{
if (! clientConnection.hasPrivilege(Privilege.PRIVILEGE_CHANGE,
modifyOperation))
{
Message message =
ERR_CONFIG_FILE_MODIFY_PRIVS_INSUFFICIENT_PRIVILEGES.get();
throw new DirectoryException(ResultCode.INSUFFICIENT_ACCESS_RIGHTS,
message);
}
break;
}
}
}
// Grab the config lock to ensure that only one config update may be in
// progress at any given time.
synchronized (configLock)
{
// Get the DN of the target entry for future reference.
DN entryDN = e.getDN();
// Get the target entry. If it does not exist, then fail.
ConfigEntry currentEntry = configEntries.get(entryDN);
if (currentEntry == null)
{
// Try to find the matched DN if possible.
DN matchedDN = null;
if (entryDN.isDescendantOf(configRootEntry.getDN()))
{
DN parentDN = entryDN.getParent();
while (parentDN != null)
{
if (configEntries.containsKey(parentDN))
{
matchedDN = parentDN;
break;
}
parentDN = parentDN.getParent();
}
}
Message message =
ERR_CONFIG_FILE_MODIFY_NO_SUCH_ENTRY.get(String.valueOf(entryDN));
throw new DirectoryException(ResultCode.NO_SUCH_OBJECT, message,
matchedDN, null);
}
// If the structural class is different between the current entry and the
// new entry, then reject the change.
if (! currentEntry.getEntry().getStructuralObjectClass().equals(
newEntry.getStructuralObjectClass()))
{
Message message = ERR_CONFIG_FILE_MODIFY_STRUCTURAL_CHANGE_NOT_ALLOWED.
get(String.valueOf(entryDN));
throw new DirectoryException(ResultCode.NO_SUCH_OBJECT, message);
}
// Create a new config entry to use for the validation testing.
ConfigEntry newConfigEntry = new ConfigEntry(e, currentEntry.getParent());
// See if there are any config change listeners registered for this entry.
// If there are, then make sure they are all OK with the change.
CopyOnWriteArrayList<ConfigChangeListener> changeListeners =
currentEntry.getChangeListeners();
MessageBuilder unacceptableReason = new MessageBuilder();
for (ConfigChangeListener l : changeListeners)
{
if (! l.configChangeIsAcceptable(newConfigEntry, unacceptableReason))
{
Message message = ERR_CONFIG_FILE_MODIFY_REJECTED_BY_CHANGE_LISTENER.
get(String.valueOf(entryDN), String.valueOf(unacceptableReason));
throw new DirectoryException(
ResultCode.UNWILLING_TO_PERFORM, message);
}
}
// At this point, it looks like the change is acceptable, so apply it.
// We'll just overwrite the core entry in the current config entry so that
// we keep all the registered listeners, references to the parent and
// children, and other metadata.
currentEntry.setEntry(e);
writeUpdatedConfig();
// Notify all the change listeners of the update.
ResultCode resultCode = ResultCode.SUCCESS;
LinkedList<Message> messages = new LinkedList<Message>();
for (ConfigChangeListener l : changeListeners)
{
ConfigChangeResult result = l.applyConfigurationChange(currentEntry);
if (result.getResultCode() != ResultCode.SUCCESS)
{
if (resultCode == ResultCode.SUCCESS)
{
resultCode = result.getResultCode();
}
messages.addAll(result.getMessages());
}
handleConfigChangeResult(result, currentEntry.getDN(),
l.getClass().getName(),
"applyConfigurationChange");
}
if (resultCode != ResultCode.SUCCESS)
{
MessageBuilder buffer = new MessageBuilder();
if (! messages.isEmpty())
{
Iterator<Message> iterator = messages.iterator();
buffer.append(iterator.next());
while (iterator.hasNext())
{
buffer.append(". ");
buffer.append(iterator.next());
}
}
Message message =
ERR_CONFIG_FILE_MODIFY_APPLY_FAILED.get(String.valueOf(buffer));
throw new DirectoryException(resultCode, message);
}
}
}
/**
* {@inheritDoc}
*/
@Override()
public void renameEntry(DN currentDN, Entry entry,
ModifyDNOperation modifyDNOperation)
throws DirectoryException
{
// If there is a modify DN operation, then make sure that the associated
// user has both the CONFIG_READ and CONFIG_WRITE privileges.
if (modifyDNOperation != null)
{
ClientConnection clientConnection =
modifyDNOperation.getClientConnection();
if (! (clientConnection.hasAllPrivileges(CONFIG_READ_AND_WRITE,
modifyDNOperation)))
{
Message message = ERR_CONFIG_FILE_MODDN_INSUFFICIENT_PRIVILEGES.get();
throw new DirectoryException(ResultCode.INSUFFICIENT_ACCESS_RIGHTS,
message);
}
}
// Modify DN operations will not be allowed in the configuration, so this
// will always throw an exception.
Message message = ERR_CONFIG_FILE_MODDN_NOT_ALLOWED.get();
throw new DirectoryException(ResultCode.UNWILLING_TO_PERFORM, message);
}
/**
* {@inheritDoc}
*/
@Override()
public void search(SearchOperation searchOperation)
throws DirectoryException
{
// Make sure that the associated user has the CONFIG_READ privilege.
ClientConnection clientConnection = searchOperation.getClientConnection();
if (! clientConnection.hasPrivilege(Privilege.CONFIG_READ, searchOperation))
{
Message message = ERR_CONFIG_FILE_SEARCH_INSUFFICIENT_PRIVILEGES.get();
throw new DirectoryException(ResultCode.INSUFFICIENT_ACCESS_RIGHTS,
message);
}
// First, get the base DN for the search and make sure that it exists.
DN baseDN = searchOperation.getBaseDN();
ConfigEntry baseEntry = configEntries.get(baseDN);
if (baseEntry == null)
{
Message message = ERR_CONFIG_FILE_SEARCH_NO_SUCH_BASE.get(
String.valueOf(baseDN));
DN matchedDN = null;
if (baseDN.isDescendantOf(configRootEntry.getDN()))
{
DN parentDN = baseDN.getParent();
while (parentDN != null)
{
if (configEntries.containsKey(parentDN))
{
matchedDN = parentDN;
break;
}
parentDN = parentDN.getParent();
}
}
throw new DirectoryException(ResultCode.NO_SUCH_OBJECT, message,
matchedDN, null);
}
// Get the scope for the search and perform the remainder of the processing
// accordingly. Also get the filter since we will need it in all cases.
SearchScope scope = searchOperation.getScope();
SearchFilter filter = searchOperation.getFilter();
switch (scope)
{
case BASE_OBJECT:
// We are only interested in the base entry itself. See if it matches
// and if so then return the entry.
Entry e = baseEntry.getEntry().duplicate(true);
if (filter.matchesEntry(e))
{
searchOperation.returnEntry(e, null);
}
break;
case SINGLE_LEVEL:
// We are only interested in entries immediately below the base entry.
// Iterate through them and return the ones that match the filter.
for (ConfigEntry child : baseEntry.getChildren().values())
{
e = child.getEntry().duplicate(true);
if (filter.matchesEntry(e))
{
if (! searchOperation.returnEntry(e, null))
{
break;
}
}
}
break;
case WHOLE_SUBTREE:
// We are interested in the base entry and all its children. Use a
// recursive process to achieve this.
searchSubtree(baseEntry, filter, searchOperation);
break;
case SUBORDINATE_SUBTREE:
// We are not interested in the base entry, but we want to check out all
// of its children. Use a recursive process to achieve this.
for (ConfigEntry child : baseEntry.getChildren().values())
{
if (! searchSubtree(child, filter, searchOperation))
{
break;
}
}
break;
default:
// The user provided an invalid scope.
Message message =
ERR_CONFIG_FILE_SEARCH_INVALID_SCOPE.get(String.valueOf(scope));
throw new DirectoryException(ResultCode.PROTOCOL_ERROR, message);
}
}
/**
* Performs a subtree search starting at the provided base entry, returning
* all entries anywhere in that subtree that match the provided filter.
*
* @param baseEntry The base entry below which to perform the search.
* @param filter The filter to use to identify matching entries.
* @param searchOperation The search operation to use to return entries to
* the client.
*
* @return <CODE>true</CODE> if the search should continue, or
* <CODE>false</CODE> if it should stop for some reason (e.g., the
* time limit or size limit has been reached).
*
* @throws DirectoryException If a problem occurs during processing.
*/
private boolean searchSubtree(ConfigEntry baseEntry, SearchFilter filter,
SearchOperation searchOperation)
throws DirectoryException
{
Entry e = baseEntry.getEntry().duplicate(true);
if (filter.matchesEntry(e))
{
if (! searchOperation.returnEntry(e, null))
{
return false;
}
}
for (ConfigEntry child : baseEntry.getChildren().values())
{
if (! searchSubtree(child, filter, searchOperation))
{
return false;
}
}
return true;
}
/**
* {@inheritDoc}
*/
@Override()
public void writeUpdatedConfig()
throws DirectoryException
{
// FIXME -- This needs support for encryption.
// Calculate an archive for the current server configuration file and see if
// it matches what we expect. If not, then the file has been manually
// edited with the server online which is a bad thing. In that case, we'll
// copy the current config off to the side before writing the new config
// so that the manual changes don't get lost but also don't get applied.
// Also, send an admin alert notifying administrators about the problem.
if (maintainConfigArchive)
{
try
{
byte[] currentDigest = calculateConfigDigest();
if (! Arrays.equals(configurationDigest, currentDigest))
{
File existingCfg = new File(configFile);
File newConfigFile = new File(existingCfg.getParent(),
"config.manualedit-" +
TimeThread.getGMTTime() + ".ldif");
int counter = 2;
while (newConfigFile.exists())
{
newConfigFile = new File(newConfigFile.getAbsolutePath() + "." +
counter++);
}
FileInputStream inputStream = new FileInputStream(existingCfg);
FileOutputStream outputStream = new FileOutputStream(newConfigFile);
byte[] buffer = new byte[8192];
while (true)
{
int bytesRead = inputStream.read(buffer);
if (bytesRead < 0)
{
break;
}
outputStream.write(buffer, 0, bytesRead);
}
inputStream.close();
outputStream.close();
Message message = WARN_CONFIG_MANUAL_CHANGES_DETECTED.get(
configFile, newConfigFile.getAbsolutePath());
logError(message);
DirectoryServer.sendAlertNotification(this,
ALERT_TYPE_MANUAL_CONFIG_EDIT_HANDLED, message);
}
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_MANUAL_CHANGES_LOST.get(
configFile, stackTraceToSingleLineString(e));
logError(message);
DirectoryServer.sendAlertNotification(this,
ALERT_TYPE_MANUAL_CONFIG_EDIT_HANDLED, message);
}
}
// Write the new configuration to a temporary file.
String tempConfig = configFile + ".tmp";
try
{
LDIFExportConfig exportConfig =
new LDIFExportConfig(tempConfig, ExistingFileBehavior.OVERWRITE);
// FIXME -- Add all the appropriate configuration options.
writeLDIF(exportConfig);
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_FILE_WRITE_CANNOT_EXPORT_NEW_CONFIG.get(
String.valueOf(tempConfig), stackTraceToSingleLineString(e));
logError(message);
DirectoryServer.sendAlertNotification(this,
ALERT_TYPE_CANNOT_WRITE_CONFIGURATION, message);
return;
}
// Delete the previous version of the configuration and rename the new one.
try
{
File actualConfig = new File(configFile);
File tmpConfig = new File(tempConfig);
renameFile(tmpConfig, actualConfig);
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_FILE_WRITE_CANNOT_RENAME_NEW_CONFIG.
get(String.valueOf(tempConfig), String.valueOf(configFile),
stackTraceToSingleLineString(e));
logError(message);
DirectoryServer.sendAlertNotification(this,
ALERT_TYPE_CANNOT_WRITE_CONFIGURATION, message);
return;
}
configurationDigest = calculateConfigDigest();
// Try to write the archive for the new configuration.
if (maintainConfigArchive)
{
writeConfigArchive();
}
}
/**
* Writes the current configuration to the configuration archive. This will
* be a best-effort attempt.
*/
private void writeConfigArchive()
{
if (! maintainConfigArchive)
{
return;
}
// Determine the path to the directory that will hold the archived
// configuration files.
File configDirectory = new File(configFile).getParentFile();
File archiveDirectory = new File(configDirectory, CONFIG_ARCHIVE_DIR_NAME);
// If the archive directory doesn't exist, then create it.
if (! archiveDirectory.exists())
{
try
{
if (! archiveDirectory.mkdirs())
{
Message message = ERR_CONFIG_FILE_CANNOT_CREATE_ARCHIVE_DIR_NO_REASON.
get(archiveDirectory.getAbsolutePath());
logError(message);
DirectoryServer.sendAlertNotification(this,
ALERT_TYPE_CANNOT_WRITE_CONFIGURATION, message);
return;
}
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_FILE_CANNOT_CREATE_ARCHIVE_DIR.
get(archiveDirectory.getAbsolutePath(),
stackTraceToSingleLineString(e));
logError(message);
DirectoryServer.sendAlertNotification(this,
ALERT_TYPE_CANNOT_WRITE_CONFIGURATION, message);
return;
}
}
// Determine the appropriate name to use for the current configuration.
File archiveFile;
try
{
String timestamp = TimeThread.getGMTTime();
archiveFile = new File(archiveDirectory, "config-" + timestamp + ".gz");
if (archiveFile.exists())
{
int counter = 2;
archiveFile = new File(archiveDirectory,
"config-" + timestamp + "-" + counter + ".gz");
while (archiveFile.exists())
{
counter++;
archiveFile = new File(archiveDirectory,
"config-" + timestamp + "-" + counter + ".gz");
}
}
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_FILE_CANNOT_WRITE_CONFIG_ARCHIVE.get(
stackTraceToSingleLineString(e));
logError(message);
DirectoryServer.sendAlertNotification(this,
ALERT_TYPE_CANNOT_WRITE_CONFIGURATION, message);
return;
}
// Copy the current configuration to the new configuration file.
byte[] buffer = new byte[8192];
FileInputStream inputStream = null;
GZIPOutputStream outputStream = null;
try
{
inputStream = new FileInputStream(configFile);
outputStream = new GZIPOutputStream(new FileOutputStream(archiveFile));
int bytesRead = inputStream.read(buffer);
while (bytesRead > 0)
{
outputStream.write(buffer, 0, bytesRead);
bytesRead = inputStream.read(buffer);
}
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_FILE_CANNOT_WRITE_CONFIG_ARCHIVE.get(
stackTraceToSingleLineString(e));
logError(message);
DirectoryServer.sendAlertNotification(this,
ALERT_TYPE_CANNOT_WRITE_CONFIGURATION, message);
return;
}
finally
{
try
{
inputStream.close();
} catch (Exception e) {}
try
{
outputStream.close();
} catch (Exception e) {}
}
// If we should enforce a maximum number of archived configurations, then
// see if there are any old ones that we need to delete.
if (maxConfigArchiveSize > 0)
{
String[] archivedFileList = archiveDirectory.list();
int numToDelete = archivedFileList.length - maxConfigArchiveSize;
if (numToDelete > 0)
{
TreeSet<String> archiveSet = new TreeSet<String>();
for (String name : archivedFileList)
{
if (! name.startsWith("config-"))
{
continue;
}
// Simply ordering by filename should work, even when there are
// timestamp conflicts, because the dash comes before the period in
// the ASCII character set.
archiveSet.add(name);
}
Iterator<String> iterator = archiveSet.iterator();
for (int i=0; ((i < numToDelete) && iterator.hasNext()); i++)
{
File f = new File(archiveDirectory, iterator.next());
try
{
f.delete();
} catch (Exception e) {}
}
}
}
}
/**
* {@inheritDoc}
*/
@Override()
public void writeSuccessfulStartupConfig()
{
if (useLastKnownGoodConfig)
{
// The server was started with the "last known good" configuration, so we
// shouldn't overwrite it with something that is probably bad.
return;
}
String startOKFilePath = configFile + ".startok";
String tempFilePath = startOKFilePath + ".tmp";
String oldFilePath = startOKFilePath + ".old";
// Copy the current config file to a temporary file.
File tempFile = new File(tempFilePath);
FileInputStream inputStream = null;
try
{
inputStream = new FileInputStream(configFile);
FileOutputStream outputStream = null;
try
{
outputStream = new FileOutputStream(tempFilePath, false);
try
{
byte[] buffer = new byte[8192];
while (true)
{
int bytesRead = inputStream.read(buffer);
if (bytesRead < 0)
{
break;
}
outputStream.write(buffer, 0, bytesRead);
}
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
logError(ERR_STARTOK_CANNOT_WRITE.get(configFile, tempFilePath,
getExceptionMessage(e)));
return;
}
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
logError(ERR_STARTOK_CANNOT_OPEN_FOR_WRITING.get(tempFilePath,
getExceptionMessage(e)));
return;
}
finally
{
try
{
outputStream.close();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
}
}
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
logError(ERR_STARTOK_CANNOT_OPEN_FOR_READING.get(configFile,
getExceptionMessage(e)));
return;
}
finally
{
try
{
inputStream.close();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
}
}
// If a ".startok" file already exists, then move it to an ".old" file.
File oldFile = new File(oldFilePath);
try
{
if (oldFile.exists())
{
oldFile.delete();
}
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
}
File startOKFile = new File(startOKFilePath);
try
{
if (startOKFile.exists())
{
startOKFile.renameTo(oldFile);
}
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
}
// Rename the temp file to the ".startok" file.
try
{
tempFile.renameTo(startOKFile);
} catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
logError(ERR_STARTOK_CANNOT_RENAME.get(tempFilePath, startOKFilePath,
getExceptionMessage(e)));
return;
}
// Remove the ".old" file if there is one.
try
{
if (oldFile.exists())
{
oldFile.delete();
}
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
}
}
/**
* {@inheritDoc}
*/
@Override()
public HashSet<String> getSupportedControls()
{
return SUPPORTED_CONTROLS;
}
/**
* {@inheritDoc}
*/
@Override()
public HashSet<String> getSupportedFeatures()
{
return SUPPORTED_FEATURES;
}
/**
* {@inheritDoc}
*/
@Override()
public boolean supportsLDIFExport()
{
// TODO We would need export-ldif to initialize this backend.
return false;
}
/**
* {@inheritDoc}
*/
@Override()
public void exportLDIF(LDIFExportConfig exportConfig)
throws DirectoryException
{
// TODO We would need export-ldif to initialize this backend.
writeLDIF(exportConfig);
}
/**
* Writes the current configuration to LDIF with the provided export
* configuration.
*
* @param exportConfig The configuration to use for the export.
*
* @throws DirectoryException If a problem occurs while writing the LDIF.
*/
private void writeLDIF(LDIFExportConfig exportConfig)
throws DirectoryException
{
LDIFWriter writer;
try
{
writer = new LDIFWriter(exportConfig);
writer.writeComment(INFO_CONFIG_FILE_HEADER.get(), 80);
writeEntryAndChildren(writer, configRootEntry);
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_LDIF_WRITE_ERROR.get(String.valueOf(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
try
{
writer.close();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_FILE_CLOSE_ERROR.get(String.valueOf(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
}
/**
* Writes the provided entry and any children that it may have to the provided
* LDIF writer.
*
* @param writer The LDIF writer to use to write the entry and its
* children.
* @param configEntry The configuration entry to write, along with its
* children.
*
* @throws DirectoryException If a problem occurs while attempting to write
* the entry or one of its children.
*/
private void writeEntryAndChildren(LDIFWriter writer, ConfigEntry configEntry)
throws DirectoryException
{
try
{
// Write the entry itself to LDIF.
writer.writeEntry(configEntry.getEntry());
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_FILE_WRITE_ERROR.get(
configEntry.getDN().toString(), String.valueOf(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
// See if the entry has any children. If so, then iterate through them and
// write them and their children. We'll copy the entries into a tree map
// so that we have a sensible order in the resulting LDIF.
TreeMap<DN,ConfigEntry> childMap =
new TreeMap<DN,ConfigEntry>(configEntry.getChildren());
for (ConfigEntry childEntry : childMap.values())
{
writeEntryAndChildren(writer, childEntry);
}
}
/**
* {@inheritDoc}
*/
@Override()
public boolean supportsLDIFImport()
{
return false;
}
/**
* {@inheritDoc}
*/
@Override()
public LDIFImportResult importLDIF(LDIFImportConfig importConfig)
throws DirectoryException
{
Message message = ERR_CONFIG_FILE_UNWILLING_TO_IMPORT.get();
throw new DirectoryException(ResultCode.UNWILLING_TO_PERFORM, message);
}
/**
* {@inheritDoc}
*/
@Override()
public boolean supportsBackup()
{
// We do support an online backup mechanism for the configuration.
return true;
}
/**
* {@inheritDoc}
*/
@Override()
public boolean supportsBackup(BackupConfig backupConfig,
StringBuilder unsupportedReason)
{
// We should support online backup for the configuration in any form. This
// implementation does not support incremental backups, but in this case
// even if we're asked to do an incremental we'll just do a full backup
// instead. So the answer to this should always be "true".
return true;
}
/**
* {@inheritDoc}
*/
@Override()
public void createBackup(BackupConfig backupConfig)
throws DirectoryException
{
// Get the properties to use for the backup. We don't care whether or not
// it's incremental, so there's no need to get that.
String backupID = backupConfig.getBackupID();
BackupDirectory backupDirectory = backupConfig.getBackupDirectory();
boolean compress = backupConfig.compressData();
boolean encrypt = backupConfig.encryptData();
boolean hash = backupConfig.hashData();
boolean signHash = backupConfig.signHash();
// Create a hash map that will hold the extra backup property information
// for this backup.
HashMap<String,String> backupProperties = new HashMap<String,String>();
// Get the crypto manager and use it to obtain references to the message
// digest and/or MAC to use for hashing and/or signing.
CryptoManager cryptoManager = DirectoryServer.getCryptoManager();
Mac mac = null;
MessageDigest digest = null;
String digestAlgorithm = null;
String macKeyID = null;
if (hash)
{
if (signHash)
{
try
{
macKeyID = cryptoManager.getMacEngineKeyEntryID();
backupProperties.put(BACKUP_PROPERTY_MAC_KEY_ID, macKeyID);
mac = cryptoManager.getMacEngine(macKeyID);
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_BACKUP_CANNOT_GET_MAC.get(
macKeyID, stackTraceToSingleLineString(e));
throw new DirectoryException(
DirectoryServer.getServerErrorResultCode(), message,
e);
}
}
else
{
digestAlgorithm = cryptoManager.getPreferredMessageDigestAlgorithm();
backupProperties.put(BACKUP_PROPERTY_DIGEST_ALGORITHM, digestAlgorithm);
try
{
digest = cryptoManager.getPreferredMessageDigest();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_BACKUP_CANNOT_GET_DIGEST.get(
digestAlgorithm, stackTraceToSingleLineString(e));
throw new DirectoryException(
DirectoryServer.getServerErrorResultCode(), message,
e);
}
}
}
// Create an output stream that will be used to write the archive file. At
// its core, it will be a file output stream to put a file on the disk. If
// we are to encrypt the data, then that file output stream will be wrapped
// in a cipher output stream. The resulting output stream will then be
// wrapped by a zip output stream (which may or may not actually use
// compression).
String filename = null;
OutputStream outputStream;
try
{
filename = CONFIG_BACKUP_BASE_FILENAME + backupID;
File archiveFile = new File(backupDirectory.getPath() + File.separator +
filename);
if (archiveFile.exists())
{
int i=1;
while (true)
{
archiveFile = new File(backupDirectory.getPath() + File.separator +
filename + "." + i);
if (archiveFile.exists())
{
i++;
}
else
{
filename = filename + "." + i;
break;
}
}
}
outputStream = new FileOutputStream(archiveFile, false);
backupProperties.put(BACKUP_PROPERTY_ARCHIVE_FILENAME, filename);
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_BACKUP_CANNOT_CREATE_ARCHIVE_FILE.
get(String.valueOf(filename), backupDirectory.getPath(),
stackTraceToSingleLineString(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
// If we should encrypt the data, then wrap the output stream in a cipher
// output stream.
if (encrypt)
{
try
{
outputStream
= cryptoManager.getCipherOutputStream(outputStream);
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
Message message = ERR_CONFIG_BACKUP_CANNOT_GET_CIPHER.get(
stackTraceToSingleLineString(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
}
// Wrap the file output stream in a zip output stream.
ZipOutputStream zipStream = new ZipOutputStream(outputStream);
Message message = ERR_CONFIG_BACKUP_ZIP_COMMENT.get(
DynamicConstants.PRODUCT_NAME,
backupID);
zipStream.setComment(message.toString());
if (compress)
{
zipStream.setLevel(Deflater.DEFAULT_COMPRESSION);
}
else
{
zipStream.setLevel(Deflater.NO_COMPRESSION);
}
// This may seem a little weird, but in this context, we only have access to
// this class as a backend and not as the config handler. We need it as a
// config handler to determine the path to the config file, so we can get
// that from the Directory Server object.
String configFile = null;
try
{
configFile =
((ConfigFileHandler) DirectoryServer.getConfigHandler()).configFile;
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
message = ERR_CONFIG_BACKUP_CANNOT_DETERMINE_CONFIG_FILE_LOCATION.
get(getExceptionMessage(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
// Read the Directory Server configuration file and put it in the archive.
byte[] buffer = new byte[8192];
FileInputStream inputStream = null;
try
{
File f = new File(configFile);
ZipEntry zipEntry = new ZipEntry(f.getName());
zipStream.putNextEntry(zipEntry);
inputStream = new FileInputStream(f);
while (true)
{
int bytesRead = inputStream.read(buffer);
if (bytesRead < 0 || backupConfig.isCancelled())
{
break;
}
if (hash)
{
if (signHash)
{
mac.update(buffer, 0, bytesRead);
}
else
{
digest.update(buffer, 0, bytesRead);
}
}
zipStream.write(buffer, 0, bytesRead);
}
inputStream.close();
zipStream.closeEntry();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
try
{
inputStream.close();
} catch (Exception e2) {}
try
{
zipStream.close();
} catch (Exception e2) {}
message = ERR_CONFIG_BACKUP_CANNOT_BACKUP_CONFIG_FILE.get(
configFile, stackTraceToSingleLineString(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
// If an archive directory exists, then add its contents as well.
try
{
File archiveDirectory = new File(new File(configFile).getParent(),
CONFIG_ARCHIVE_DIR_NAME);
if (archiveDirectory.exists())
{
for (File archiveFile : archiveDirectory.listFiles())
{
ZipEntry zipEntry = new ZipEntry(CONFIG_ARCHIVE_DIR_NAME +
File.separator +
archiveFile.getName());
zipStream.putNextEntry(zipEntry);
inputStream = new FileInputStream(archiveFile);
while (true)
{
int bytesRead = inputStream.read(buffer);
if (bytesRead < 0 || backupConfig.isCancelled())
{
break;
}
if (hash)
{
if (signHash)
{
mac.update(buffer, 0, bytesRead);
}
else
{
digest.update(buffer, 0, bytesRead);
}
}
zipStream.write(buffer, 0, bytesRead);
}
inputStream.close();
zipStream.closeEntry();
}
}
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
try
{
inputStream.close();
} catch (Exception e2) {}
try
{
zipStream.close();
} catch (Exception e2) {}
message = ERR_CONFIG_BACKUP_CANNOT_BACKUP_ARCHIVED_CONFIGS.get(
configFile, stackTraceToSingleLineString(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
// We're done writing the file, so close the zip stream (which should also
// close the underlying stream).
try
{
zipStream.close();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
message = ERR_CONFIG_BACKUP_CANNOT_CLOSE_ZIP_STREAM.get(
filename, backupDirectory.getPath(), getExceptionMessage(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
// Get the digest or MAC bytes if appropriate.
byte[] digestBytes = null;
byte[] macBytes = null;
if (hash)
{
if (signHash)
{
macBytes = mac.doFinal();
}
else
{
digestBytes = digest.digest();
}
}
// Create the backup info structure for this backup and add it to the backup
// directory.
// FIXME -- Should I use the date from when I started or finished?
BackupInfo backupInfo = new BackupInfo(backupDirectory, backupID,
new Date(), false, compress,
encrypt, digestBytes, macBytes,
null, backupProperties);
try
{
backupDirectory.addBackup(backupInfo);
backupDirectory.writeBackupDirectoryDescriptor();
}
catch (Exception e)
{
if (debugEnabled())
{
TRACER.debugCaught(DebugLogLevel.ERROR, e);
}
message = ERR_CONFIG_BACKUP_CANNOT_UPDATE_BACKUP_DESCRIPTOR.get(
backupDirectory.getDescriptorPath(), stackTraceToSingleLineString(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
// Remove the backup if this operation was cancelled since the
// backup may be incomplete
if (backupConfig.isCancelled())
{
removeBackup(backupDirectory, backupID);
}
}
/**
* {@inheritDoc}
*/
@Override()
public void removeBackup(BackupDirectory backupDirectory,
String backupID)
throws DirectoryException
{
// NYI
}
/**
* {@inheritDoc}
*/
@Override()
public boolean supportsRestore()
{
// We will provide a restore, but only for offline operations.
return true;
}
/**
* {@inheritDoc}
*/
@Override()
public void restoreBackup(RestoreConfig restoreConfig)
throws DirectoryException
{
// First, make sure that the requested backup exists.
BackupDirectory backupDirectory = restoreConfig.getBackupDirectory();
String backupPath = backupDirectory.getPath();
String backupID = restoreConfig.getBackupID();
BackupInfo backupInfo = backupDirectory.getBackupInfo(backupID);
if (backupInfo == null)
{
Message message =
ERR_CONFIG_RESTORE_NO_SUCH_BACKUP.get(backupID, backupPath);
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message);
}
// Read the backup info structure to determine the name of the file that
// contains the archive. Then make sure that file exists.
String backupFilename =
backupInfo.getBackupProperty(BACKUP_PROPERTY_ARCHIVE_FILENAME);
if (backupFilename == null)
{
Message message =
ERR_CONFIG_RESTORE_NO_BACKUP_FILE.get(backupID, backupPath);
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message);
}
File backupFile = new File(backupPath + File.separator + backupFilename);
try
{
if (! backupFile.exists())
{
Message message =
ERR_CONFIG_RESTORE_NO_SUCH_FILE.get(backupID, backupFile.getPath());
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message);
}
}
catch (DirectoryException de)
{
throw de;
}
catch (Exception e)
{
Message message = ERR_CONFIG_RESTORE_CANNOT_CHECK_FOR_ARCHIVE.get(
backupID, backupFile.getPath(), stackTraceToSingleLineString(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
// If the backup is hashed, then we need to get the message digest to use
// to verify it.
byte[] unsignedHash = backupInfo.getUnsignedHash();
MessageDigest digest = null;
if (unsignedHash != null)
{
String digestAlgorithm =
backupInfo.getBackupProperty(BACKUP_PROPERTY_DIGEST_ALGORITHM);
if (digestAlgorithm == null)
{
Message message = ERR_CONFIG_RESTORE_UNKNOWN_DIGEST.get(backupID);
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message);
}
try
{
digest = DirectoryServer.getCryptoManager().getMessageDigest(
digestAlgorithm);
}
catch (Exception e)
{
Message message =
ERR_CONFIG_RESTORE_CANNOT_GET_DIGEST.get(backupID, digestAlgorithm);
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
}
// If the backup is signed, then we need to get the MAC to use to verify it.
byte[] signedHash = backupInfo.getSignedHash();
Mac mac = null;
if (signedHash != null)
{
String macKeyID =
backupInfo.getBackupProperty(BACKUP_PROPERTY_MAC_KEY_ID);
if (macKeyID == null)
{
Message message = ERR_CONFIG_RESTORE_UNKNOWN_MAC.get(backupID);
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message);
}
try
{
mac = DirectoryServer.getCryptoManager().getMacEngine(macKeyID);
}
catch (Exception e)
{
Message message = ERR_CONFIG_RESTORE_CANNOT_GET_MAC.get(
backupID, macKeyID);
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
}
// Create the input stream that will be used to read the backup file. At
// its core, it will be a file input stream.
InputStream inputStream;
try
{
inputStream = new FileInputStream(backupFile);
}
catch (Exception e)
{
Message message = ERR_CONFIG_RESTORE_CANNOT_OPEN_BACKUP_FILE.get(
backupID, backupFile.getPath(), stackTraceToSingleLineString(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
// If the backup is encrypted, then we need to wrap the file input stream
// in a cipher input stream.
if (backupInfo.isEncrypted())
{
try
{
inputStream = DirectoryServer.getCryptoManager()
.getCipherInputStream(inputStream);
}
catch (Exception e)
{
Message message = ERR_CONFIG_RESTORE_CANNOT_GET_CIPHER.get(
backupFile.getPath(), stackTraceToSingleLineString(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
}
// Now wrap the resulting input stream in a zip stream so that we can read
// its contents. We don't need to worry about whether to use compression or
// not because it will be handled automatically.
ZipInputStream zipStream = new ZipInputStream(inputStream);
// Determine whether we should actually do the restore, or if we should just
// try to verify the archive. If we are going to actually do the restore,
// then create a directory and move the existing config files there so that
// they can be restored in case something goes wrong.
String configFilePath =
((ConfigFileHandler) DirectoryServer.getConfigHandler()).configFile;
File configFile = new File(configFilePath);
File configDir = configFile.getParentFile();
String configDirPath = configDir.getPath();
String backupDirPath = null;
File configBackupDir = null;
boolean verifyOnly = restoreConfig.verifyOnly();
if (! verifyOnly)
{
// Create a new directory to hold the current config files.
try
{
if (configDir.exists())
{
String configBackupDirPath = configDirPath + ".save";
backupDirPath = configBackupDirPath;
configBackupDir = new File(backupDirPath);
if (configBackupDir.exists())
{
int i=2;
while (true)
{
backupDirPath = configBackupDirPath + i;
configBackupDir = new File(backupDirPath);
if (configBackupDir.exists())
{
i++;
}
else
{
break;
}
}
}
configBackupDir.mkdirs();
moveFile(configFile, configBackupDir);
File archiveDirectory = new File(configDir, CONFIG_ARCHIVE_DIR_NAME);
if (archiveDirectory.exists())
{
File archiveBackupPath = new File(configBackupDir,
CONFIG_ARCHIVE_DIR_NAME);
archiveDirectory.renameTo(archiveBackupPath);
}
}
}
catch (Exception e)
{
Message message = ERR_CONFIG_RESTORE_CANNOT_BACKUP_EXISTING_CONFIG.
get(backupID, configDirPath, String.valueOf(backupDirPath),
getExceptionMessage(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
// Create a new directory to hold the restored config files.
try
{
configDir.mkdirs();
}
catch (Exception e)
{
// Try to restore the previous config directory if possible. This will
// probably fail in this case, but try anyway.
if (configBackupDir != null)
{
try
{
configBackupDir.renameTo(configDir);
Message message =
NOTE_CONFIG_RESTORE_RESTORED_OLD_CONFIG.get(configDirPath);
logError(message);
}
catch (Exception e2)
{
Message message = ERR_CONFIG_RESTORE_CANNOT_RESTORE_OLD_CONFIG.get(
configBackupDir.getPath());
logError(message);
}
}
Message message = ERR_CONFIG_RESTORE_CANNOT_CREATE_CONFIG_DIRECTORY.get(
backupID, configDirPath, getExceptionMessage(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
}
// Read through the archive file an entry at a time. For each entry, update
// the digest or MAC if necessary, and if we're actually doing the restore,
// then write the files out into the config directory.
byte[] buffer = new byte[8192];
while (true)
{
ZipEntry zipEntry;
try
{
zipEntry = zipStream.getNextEntry();
}
catch (Exception e)
{
// Tell the user where the previous config was archived.
if (configBackupDir != null)
{
Message message = ERR_CONFIG_RESTORE_OLD_CONFIG_SAVED.get(
configBackupDir.getPath());
logError(message);
}
Message message = ERR_CONFIG_RESTORE_CANNOT_GET_ZIP_ENTRY.get(
backupID, backupFile.getPath(), stackTraceToSingleLineString(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
if (zipEntry == null)
{
break;
}
// Get the filename for the zip entry and update the digest or MAC as
// necessary.
String fileName = zipEntry.getName();
if (digest != null)
{
digest.update(getBytes(fileName));
}
if (mac != null)
{
mac.update(getBytes(fileName));
}
// If we're doing the restore, then create the output stream to write the
// file.
OutputStream outputStream = null;
if (! verifyOnly)
{
File restoreFile = new File(configDirPath + File.separator + fileName);
File parentDir = restoreFile.getParentFile();
try
{
if (! parentDir.exists())
{
parentDir.mkdirs();
}
outputStream = new FileOutputStream(restoreFile);
}
catch (Exception e)
{
// Tell the user where the previous config was archived.
if (configBackupDir != null)
{
Message message = ERR_CONFIG_RESTORE_OLD_CONFIG_SAVED.get(
configBackupDir.getPath());
logError(message);
}
Message message = ERR_CONFIG_RESTORE_CANNOT_CREATE_FILE.
get(backupID, restoreFile.getAbsolutePath(),
stackTraceToSingleLineString(e));
throw new DirectoryException(
DirectoryServer.getServerErrorResultCode(), message,
e);
}
}
// Read the contents of the file and update the digest or MAC as
// necessary. If we're actually restoring it, then write it into the
// new config directory.
try
{
while (true)
{
int bytesRead = zipStream.read(buffer);
if (bytesRead < 0)
{
// We've reached the end of the entry.
break;
}
// Update the digest or MAC if appropriate.
if (digest != null)
{
digest.update(buffer, 0, bytesRead);
}
if (mac != null)
{
mac.update(buffer, 0, bytesRead);
}
// Write the data to the output stream if appropriate.
if (outputStream != null)
{
outputStream.write(buffer, 0, bytesRead);
}
}
// We're at the end of the file so close the output stream if we're
// writing it.
if (outputStream != null)
{
outputStream.close();
}
}
catch (Exception e)
{
// Tell the user where the previous config was archived.
if (configBackupDir != null)
{
Message message = ERR_CONFIG_RESTORE_OLD_CONFIG_SAVED.get(
configBackupDir.getPath());
logError(message);
}
Message message = ERR_CONFIG_RESTORE_CANNOT_PROCESS_ARCHIVE_FILE.get(
backupID, fileName, stackTraceToSingleLineString(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
}
// Close the zip stream since we don't need it anymore.
try
{
zipStream.close();
}
catch (Exception e)
{
Message message = ERR_CONFIG_RESTORE_ERROR_ON_ZIP_STREAM_CLOSE.get(
backupID, backupFile.getPath(), getExceptionMessage(e));
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message, e);
}
// At this point, we should be done with the contents of the ZIP file and
// the restore should be complete. If we were generating a digest or MAC,
// then make sure it checks out.
if (digest != null)
{
byte[] calculatedHash = digest.digest();
if (Arrays.equals(calculatedHash, unsignedHash))
{
Message message = NOTE_CONFIG_RESTORE_UNSIGNED_HASH_VALID.get();
logError(message);
}
else
{
// Tell the user where the previous config was archived.
if (configBackupDir != null)
{
Message message = ERR_CONFIG_RESTORE_OLD_CONFIG_SAVED.get(
configBackupDir.getPath());
logError(message);
}
Message message =
ERR_CONFIG_RESTORE_UNSIGNED_HASH_INVALID.get(backupID);
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message);
}
}
if (mac != null)
{
byte[] calculatedSignature = mac.doFinal();
if (Arrays.equals(calculatedSignature, signedHash))
{
Message message = NOTE_CONFIG_RESTORE_SIGNED_HASH_VALID.get();
logError(message);
}
else
{
// Tell the user where the previous config was archived.
if (configBackupDir != null)
{
Message message = ERR_CONFIG_RESTORE_OLD_CONFIG_SAVED.get(
configBackupDir.getPath());
logError(message);
}
Message message = ERR_CONFIG_RESTORE_SIGNED_HASH_INVALID.get(
configBackupDir.getPath());
throw new DirectoryException(DirectoryServer.getServerErrorResultCode(),
message);
}
}
// If we are just verifying the archive, then we're done.
if (verifyOnly)
{
Message message =
NOTE_CONFIG_RESTORE_VERIFY_SUCCESSFUL.get(backupID, backupPath);
logError(message);
return;
}
// If we've gotten here, then the archive was restored successfully. Get
// rid of the temporary copy we made of the previous config directory and
// exit.
if (configBackupDir != null)
{
recursiveDelete(configBackupDir);
}
Message message = NOTE_CONFIG_RESTORE_SUCCESSFUL.get(backupID, backupPath);
logError(message);
}
/**
* {@inheritDoc}
*/
public DN getComponentEntryDN()
{
return configRootEntry.getDN();
}
/**
* {@inheritDoc}
*/
public String getClassName()
{
return CLASS_NAME;
}
/**
* {@inheritDoc}
*/
public LinkedHashMap<String,String> getAlerts()
{
LinkedHashMap<String,String> alerts = new LinkedHashMap<String,String>();
alerts.put(ALERT_TYPE_CANNOT_WRITE_CONFIGURATION,
ALERT_DESCRIPTION_CANNOT_WRITE_CONFIGURATION);
alerts.put(ALERT_TYPE_MANUAL_CONFIG_EDIT_HANDLED,
ALERT_DESCRIPTION_MANUAL_CONFIG_EDIT_HANDLED);
alerts.put(ALERT_TYPE_MANUAL_CONFIG_EDIT_LOST,
ALERT_DESCRIPTION_MANUAL_CONFIG_EDIT_LOST);
return alerts;
}
/**
* Examines the provided result and logs a message if appropriate. If the
* result code is anything other than {@code SUCCESS}, then it will log an
* error message. If the operation was successful but admin action is
* required, then it will log a warning message. If no action is required but
* messages were generated, then it will log an informational message.
*
* @param result The config change result object that
* @param entryDN The DN of the entry that was added, deleted, or
* modified.
* @param className The name of the class for the object that generated the
* provided result.
* @param methodName The name of the method that generated the provided
* result.
*/
public void handleConfigChangeResult(ConfigChangeResult result, DN entryDN,
String className, String methodName)
{
if (result == null)
{
Message message = ERR_CONFIG_CHANGE_NO_RESULT.
get(String.valueOf(className), String.valueOf(methodName),
String.valueOf(entryDN));
logError(message);
return;
}
ResultCode resultCode = result.getResultCode();
boolean adminActionRequired = result.adminActionRequired();
List<Message> messages = result.getMessages();
MessageBuilder messageBuffer = new MessageBuilder();
if (messages != null)
{
for (Message s : messages)
{
if (messageBuffer.length() > 0)
{
messageBuffer.append(" ");
}
messageBuffer.append(s);
}
}
if (resultCode != ResultCode.SUCCESS)
{
Message message = ERR_CONFIG_CHANGE_RESULT_ERROR.
get(String.valueOf(className), String.valueOf(methodName),
String.valueOf(entryDN), String.valueOf(resultCode),
adminActionRequired, messageBuffer.toString());
logError(message);
}
else if (adminActionRequired)
{
Message message = WARN_CONFIG_CHANGE_RESULT_ACTION_REQUIRED.
get(String.valueOf(className), String.valueOf(methodName),
String.valueOf(entryDN), messageBuffer.toString());
logError(message);
}
else if (messageBuffer.length() > 0)
{
Message message = INFO_CONFIG_CHANGE_RESULT_MESSAGES.
get(String.valueOf(className), String.valueOf(methodName),
String.valueOf(entryDN), messageBuffer.toString());
logError(message);
}
}
/**
* {@inheritDoc}
*/
public void preloadEntryCache() throws UnsupportedOperationException {
throw new UnsupportedOperationException("Operation not supported.");
}
}