/**
* This file is part of Waarp Project.
*
* Copyright 2009, Frederic Bregier, and individual contributors by the @author tags. See the
* COPYRIGHT.txt in the distribution for a full listing of individual contributors.
*
* All Waarp Project is free software: you can redistribute it and/or modify it under the terms of
* the GNU General Public License as published by the Free Software Foundation, either version 3 of
* the License, or (at your option) any later version.
*
* Waarp is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even
* the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
* Public License for more details.
*
* You should have received a copy of the GNU General Public License along with Waarp . If not, see
* <http://www.gnu.org/licenses/>.
*/
package org.waarp.openr66.server;
import java.net.SocketAddress;
import org.waarp.common.database.exception.WaarpDatabaseException;
import org.waarp.common.logging.WaarpLogger;
import org.waarp.common.logging.WaarpLoggerFactory;
import org.waarp.common.logging.WaarpSlf4JLoggerFactory;
import org.waarp.openr66.configuration.FileBasedConfiguration;
import org.waarp.openr66.context.ErrorCode;
import org.waarp.openr66.context.R66FiniteDualStates;
import org.waarp.openr66.context.R66Result;
import org.waarp.openr66.database.DbConstant;
import org.waarp.openr66.database.data.DbHostAuth;
import org.waarp.openr66.protocol.configuration.Configuration;
import org.waarp.openr66.protocol.configuration.PartnerConfiguration;
import org.waarp.openr66.protocol.exception.OpenR66ProtocolNoConnectionException;
import org.waarp.openr66.protocol.exception.OpenR66ProtocolPacketException;
import org.waarp.openr66.protocol.localhandler.LocalChannelReference;
import org.waarp.openr66.protocol.localhandler.packet.AbstractLocalPacket;
import org.waarp.openr66.protocol.localhandler.packet.JsonCommandPacket;
import org.waarp.openr66.protocol.localhandler.packet.LocalPacketFactory;
import org.waarp.openr66.protocol.localhandler.packet.ValidPacket;
import org.waarp.openr66.protocol.localhandler.packet.json.ConfigImportJsonPacket;
import org.waarp.openr66.protocol.networkhandler.NetworkTransaction;
import org.waarp.openr66.protocol.utils.ChannelUtils;
import org.waarp.openr66.protocol.utils.R66Future;
/**
* Config Import from a local client without database connection
*
* @author Frederic Bregier
*
*/
public class ConfigImport implements Runnable {
/**
* Internal Logger
*/
static volatile WaarpLogger logger;
protected static String _INFO_ARGS = "Need at least the configuration file as first argument then at least one from\n"
+
" -hosts file\n" +
" -rules file\n" +
" -business file (if compatible)\n" +
" -alias file (if compatible)\n" +
" -roles file (if compatible)\n" +
" -purgehosts\n" +
" -purgerules\n" +
" -purgebusiness (if compatible)\n" +
" -purgealias (if compatible)\n" +
" -purgeroles (if compatible)\n" +
" -hostid file transfer id (if compatible)\n" +
" -ruleid file transfer id (if compatible)\n" +
" -businessid file transfer id (if compatible)\n" +
" -aliasid file transfer id (if compatible)\n" +
" -roleid file transfer id (if compatible)\n" +
" -host host (optional)";
protected final R66Future future;
protected final String host;
protected final boolean hostPurge;
protected final String rule;
protected final boolean rulePurge;
protected final String business;
protected final boolean businessPurge;
protected final String alias;
protected final boolean aliasPurge;
protected final String role;
protected final boolean rolePurge;
protected long hostid = DbConstant.ILLEGALVALUE, ruleid = DbConstant.ILLEGALVALUE,
businessid = DbConstant.ILLEGALVALUE, aliasid = DbConstant.ILLEGALVALUE,
roleid = DbConstant.ILLEGALVALUE;
protected final NetworkTransaction networkTransaction;
protected DbHostAuth dbhost;
public ConfigImport(R66Future future, boolean hostPurge, boolean rulePurge,
String host, String rule,
NetworkTransaction networkTransaction) {
this.future = future;
this.host = host;
this.rule = rule;
this.hostPurge = hostPurge;
this.rulePurge = rulePurge;
this.business = null;
this.businessPurge = false;
this.alias = null;
this.aliasPurge = false;
this.role = null;
this.rolePurge = false;
this.networkTransaction = networkTransaction;
this.dbhost = Configuration.configuration.getHOST_SSLAUTH();
}
public ConfigImport(R66Future future, boolean hostPurge, boolean rulePurge,
boolean businessPurge, boolean aliasPurge, boolean rolePurge,
String host, String rule,
String business, String alias, String role,
NetworkTransaction networkTransaction) {
this.future = future;
this.host = host;
this.rule = rule;
this.hostPurge = hostPurge;
this.rulePurge = rulePurge;
this.business = business;
this.businessPurge = businessPurge;
this.alias = alias;
this.aliasPurge = aliasPurge;
this.role = role;
this.rolePurge = rolePurge;
this.networkTransaction = networkTransaction;
this.dbhost = Configuration.configuration.getHOST_SSLAUTH();
}
/**
* Used when the filenames are not compliant with remote filenames.
*
* @param hostid
* @param ruleid
* @param businessid
* @param aliasid
* @param roleid
*/
public void setSpecialIds(long hostid, long ruleid, long businessid, long aliasid, long roleid) {
this.hostid = hostid;
this.ruleid = ruleid;
this.businessid = businessid;
this.aliasid = aliasid;
this.roleid = roleid;
}
public void setHost(DbHostAuth host) {
this.dbhost = host;
}
/**
* Prior to call this method, the pipeline and NetworkTransaction must have been initialized. It
* is the responsibility of the caller to finish all network resources.
*/
public void run() {
if (logger == null) {
logger = WaarpLoggerFactory.getLogger(ConfigImport.class);
}
SocketAddress socketAddress;
try {
socketAddress = dbhost.getSocketAddress();
} catch (IllegalArgumentException e) {
logger.error("Cannot Connect to " + dbhost.getHostid());
future.setResult(new R66Result(
new OpenR66ProtocolNoConnectionException("Cannot connect to server " + dbhost.getHostid()),
null, true, ErrorCode.ConnectionImpossible, null));
dbhost = null;
future.setFailure(future.getResult().getException());
return;
}
boolean isSSL = dbhost.isSsl();
LocalChannelReference localChannelReference = networkTransaction
.createConnectionWithRetry(socketAddress, isSSL, future);
socketAddress = null;
if (localChannelReference == null) {
logger.error("Cannot Connect to " + dbhost.getHostid());
future.setResult(new R66Result(
new OpenR66ProtocolNoConnectionException("Cannot connect to server " + dbhost.getHostid()),
null, true, ErrorCode.ConnectionImpossible, null));
dbhost = null;
future.setFailure(future.getResult().getException());
return;
}
localChannelReference.sessionNewState(R66FiniteDualStates.VALIDOTHER);
boolean useJson = PartnerConfiguration.useJson(dbhost.getHostid());
logger.debug("UseJson: " + useJson);
AbstractLocalPacket valid = null;
if (useJson) {
ConfigImportJsonPacket node = new ConfigImportJsonPacket();
node.setHost(host);
node.setRule(rule);
node.setBusiness(business);
node.setAlias(alias);
node.setRoles(role);
node.setPurgehost(hostPurge);
node.setPurgerule(rulePurge);
node.setPurgebusiness(businessPurge);
node.setPurgealias(aliasPurge);
node.setPurgeroles(rolePurge);
node.setHostid(hostid);
node.setRuleid(ruleid);
node.setBusinessid(businessid);
node.setAliasid(aliasid);
node.setRolesid(roleid);
valid = new JsonCommandPacket(node, LocalPacketFactory.CONFIMPORTPACKET);
} else {
valid = new ValidPacket((hostPurge ? "1 " : "0 ") + host,
(rulePurge ? "1 " : "0 ") + rule,
LocalPacketFactory.CONFIMPORTPACKET);
}
try {
ChannelUtils.writeAbstractLocalPacket(localChannelReference, valid, false);
} catch (OpenR66ProtocolPacketException e) {
logger.error("Bad Protocol", e);
localChannelReference.getLocalChannel().close();
localChannelReference = null;
dbhost = null;
valid = null;
future.setResult(new R66Result(e, null, true,
ErrorCode.TransferError, null));
future.setFailure(e);
return;
}
dbhost = null;
future.awaitUninterruptibly();
logger.debug("Request done with " + (future.isSuccess() ? "success" : "error"));
localChannelReference.getLocalChannel().close();
localChannelReference = null;
}
protected static String shost = null;
protected static String srule = null;
protected static String sbusiness = null;
protected static String salias = null;
protected static String srole = null;
protected static boolean shostpurge = false;
protected static boolean srulepurge = false;
protected static boolean sbusinesspurge = false;
protected static boolean saliaspurge = false;
protected static boolean srolepurge = false;
protected static String stohost = null;
protected static long lhost = DbConstant.ILLEGALVALUE;
protected static long lrule = DbConstant.ILLEGALVALUE;
protected static long lbusiness = DbConstant.ILLEGALVALUE;
protected static long lalias = DbConstant.ILLEGALVALUE;
protected static long lrole = DbConstant.ILLEGALVALUE;
protected static boolean getParams(String[] args) {
if (args.length < 3) {
logger.error(_INFO_ARGS);
return false;
}
if (!FileBasedConfiguration
.setClientConfigurationFromXml(Configuration.configuration, args[0])) {
logger.error(_INFO_ARGS);
return false;
}
for (int i = 1; i < args.length; i++) {
if (args[i].equalsIgnoreCase("-hosts")) {
i++;
if (args.length <= i) {
return false;
}
shost = args[i];
} else if (args[i].equalsIgnoreCase("-rules")) {
i++;
if (args.length <= i) {
return false;
}
srule = args[i];
} else if (args[i].equalsIgnoreCase("-business")) {
i++;
if (args.length <= i) {
return false;
}
sbusiness = args[i];
} else if (args[i].equalsIgnoreCase("-alias")) {
i++;
if (args.length <= i) {
return false;
}
salias = args[i];
} else if (args[i].equalsIgnoreCase("-roles")) {
i++;
if (args.length <= i) {
return false;
}
srole = args[i];
} else if (args[i].equalsIgnoreCase("-purgehosts")) {
shostpurge = true;
} else if (args[i].equalsIgnoreCase("-purgerules")) {
srulepurge = true;
} else if (args[i].equalsIgnoreCase("-purgebusiness")) {
sbusinesspurge = true;
} else if (args[i].equalsIgnoreCase("-purgealias")) {
saliaspurge = true;
} else if (args[i].equalsIgnoreCase("-purgeroles")) {
srolepurge = true;
} else if (args[i].equalsIgnoreCase("-host")) {
i++;
stohost = args[i];
} else if (args[i].equalsIgnoreCase("-hostid")) {
i++;
if (args.length <= i) {
return false;
}
try {
lhost = Long.parseLong(args[i]);
} catch (NumberFormatException e) {
return false;
}
} else if (args[i].equalsIgnoreCase("-ruleid")) {
i++;
if (args.length <= i) {
return false;
}
try {
lrule = Long.parseLong(args[i]);
} catch (NumberFormatException e) {
return false;
}
} else if (args[i].equalsIgnoreCase("-businessid")) {
i++;
if (args.length <= i) {
return false;
}
try {
lbusiness = Long.parseLong(args[i]);
} catch (NumberFormatException e) {
return false;
}
} else if (args[i].equalsIgnoreCase("-aliasid")) {
i++;
if (args.length <= i) {
return false;
}
try {
lalias = Long.parseLong(args[i]);
} catch (NumberFormatException e) {
return false;
}
} else if (args[i].equalsIgnoreCase("-roleid")) {
i++;
if (args.length <= i) {
return false;
}
try {
lrole = Long.parseLong(args[i]);
} catch (NumberFormatException e) {
return false;
}
}
}
return true;
}
public static void main(String[] args) {
WaarpLoggerFactory.setDefaultFactory(new WaarpSlf4JLoggerFactory(null));
if (logger == null) {
logger = WaarpLoggerFactory.getLogger(ConfigImport.class);
}
if (!getParams(args)) {
logger.error("Wrong initialization");
if (DbConstant.admin != null && DbConstant.admin.isActive()) {
DbConstant.admin.close();
}
System.exit(1);
}
long time1 = System.currentTimeMillis();
R66Future future = new R66Future(true);
Configuration.configuration.pipelineInit();
NetworkTransaction networkTransaction = new NetworkTransaction();
try {
ConfigImport transaction = new ConfigImport(future, shostpurge, srulepurge,
sbusinesspurge, saliaspurge, srolepurge, shost, srule,
sbusiness, salias, srole, networkTransaction);
transaction.setSpecialIds(lhost, lrule, lbusiness, lalias, lrole);
if (stohost != null) {
try {
transaction.setHost(new DbHostAuth(DbConstant.admin.getSession(), stohost));
} catch (WaarpDatabaseException e) {
logger.error("ConfigImport in FAILURE since Host is not found: " + stohost, e);
networkTransaction.closeAll();
System.exit(10);
}
} else {
stohost = Configuration.configuration.getHOST_SSLID();
}
transaction.run();
future.awaitUninterruptibly();
long time2 = System.currentTimeMillis();
long delay = time2 - time1;
R66Result result = future.getResult();
if (future.isSuccess()) {
boolean useJson = PartnerConfiguration.useJson(stohost);
logger.debug("UseJson: " + useJson);
String message = null;
if (useJson) {
message = (result.getOther() != null ? ((JsonCommandPacket) result.getOther()).getRequest() :
"no file");
} else {
message = (result.getOther() != null ? ((ValidPacket) result.getOther()).getSheader() :
"no file");
}
if (result.getCode() == ErrorCode.Warning) {
logger.warn("WARNED on import: " +
message
+ " delay: " + delay);
} else {
logger.warn("SUCCESS on import: " +
message
+ " delay: " + delay);
}
} else {
if (result.getCode() == ErrorCode.Warning) {
logger.warn("ConfigImport is WARNED", future.getCause());
networkTransaction.closeAll();
System.exit(result.getCode().ordinal());
} else {
logger.error("ConfigImport in FAILURE", future.getCause());
networkTransaction.closeAll();
System.exit(result.getCode().ordinal());
}
}
} finally {
networkTransaction.closeAll();
System.exit(0);
}
}
}