private Message parseMessage(byte[] b) throws WireParseException {
try {
return (new Message(b));
} catch (IOException e) {
if (Options.check("verbose")) e.printStackTrace();
if (!(e instanceof WireParseException)) e = new WireParseException("Error parsing message");
throw (WireParseException) e;
}
}
/*
* Receive an exception. If the resolution has been completed,
* do nothing. Otherwise make progress.
*/
public void handleException(Object id, Exception e) {
if (Options.check("verbose")) System.err.println("ExtendedResolver: got " + e);
synchronized (this) {
outstanding--;
if (done) return;
int n;
for (n = 0; n < inprogress.length; n++) if (inprogress[n] == id) break;
/* If we don't know what this is, do nothing. */
if (n == inprogress.length) return;
boolean startnext = false;
boolean waiting = false;
/*
* If this is the first response from server n,
* we should start sending queries to server n + 1.
*/
if (sent[n] == 1 && n < resolvers.length - 1) startnext = true;
if (e instanceof InterruptedIOException) {
/* Got a timeout; resend */
if (sent[n] < retries) send(n);
if (thrown == null) thrown = e;
} else if (e instanceof SocketException) {
/*
* Problem with the socket; don't resend
* on it
*/
if (thrown == null || thrown instanceof InterruptedIOException) thrown = e;
} else {
/*
* Problem with the response; don't resend
* on the same socket.
*/
thrown = e;
}
if (done) return;
if (startnext) send(n + 1);
if (done) return;
if (outstanding == 0) {
/*
* If we're done and this is synchronous,
* wake up the blocking thread.
*/
done = true;
if (listener == null) {
notifyAll();
return;
}
}
if (!done) return;
}
/* If we're done and this is asynchronous, call the callback. */
if (!(thrown instanceof Exception)) thrown = new RuntimeException(thrown.getMessage());
listener.handleException(this, (Exception) thrown);
}
/*
* Receive a response. If the resolution hasn't been completed,
* either wake up the blocking thread or call the callback.
*/
public void receiveMessage(Object id, Message m) {
if (Options.check("verbose")) System.err.println("ExtendedResolver: " + "received message");
synchronized (this) {
if (done) return;
response = m;
done = true;
if (listener == null) {
notifyAll();
return;
}
}
listener.receiveMessage(this, response);
}
/**
* Initializes the database context, based on the initial servlet context. Parses all context
* parameters and passes them on to the database context.
*
* @param sc servlet context
* @throws IOException I/O exception
*/
public static synchronized void init(final ServletContext sc) throws IOException {
// check if HTTP context has already been initialized
if (init) return;
init = true;
// set web application path as home directory and HTTPPATH
final String webapp = sc.getRealPath("/");
Options.setSystem(Prop.PATH, webapp);
Options.setSystem(GlobalOptions.WEBPATH, webapp);
// bind all parameters that start with "org.basex." to system properties
final Enumeration<String> en = sc.getInitParameterNames();
while (en.hasMoreElements()) {
final String key = en.nextElement();
if (!key.startsWith(Prop.DBPREFIX)) continue;
String val = sc.getInitParameter(key);
if (key.endsWith("path") && !new File(val).isAbsolute()) {
// prefix relative path with absolute servlet path
Util.debug(key.toUpperCase(Locale.ENGLISH) + ": " + val);
val = new IOFile(webapp, val).path();
}
Options.setSystem(key, val);
}
// create context, update options
if (context == null) {
context = new Context(false);
} else {
context.globalopts.setSystem();
context.options.setSystem();
}
// start server instance
if (!context.globalopts.get(GlobalOptions.HTTPLOCAL)) new BaseXServer(context);
}
/**
* Sends a message to a single server and waits for a response. No checking is done to ensure that
* the response is associated with the query.
*
* @param query The query to send.
* @return The response.
* @throws IOException An error occurred while sending or receiving.
*/
public Message send(Message query) throws IOException {
if (Options.check("verbose"))
System.err.println(
"Sending to " + address.getAddress().getHostAddress() + ":" + address.getPort());
if (query.getHeader().getOpcode() == Opcode.QUERY) {
Record question = query.getQuestion();
if (question != null && question.getType() == Type.AXFR) return sendAXFR(query);
}
query = (Message) query.clone();
applyEDNS(query);
if (tsig != null) {
tsig.apply(query, null);
}
byte[] out = query.toWire(Message.MAXLENGTH);
int udpSize = maxUDPSize(query);
boolean tcp = false;
long endTime = System.currentTimeMillis() + timeoutValue;
do {
byte[] b_in;
if (useTCP || out.length > udpSize) {
tcp = true;
}
if (tcp) {
b_in = TCPClient.sendrecv(localAddress, address, out, endTime);
} else {
b_in = UDPClient.sendrecv(localAddress, address, out, udpSize, endTime);
}
/*
* Check that the response is long enough.
*/
if (b_in.length < Header.LENGTH) {
throw new WireParseException("invalid DNS header - " + "too short");
}
/*
* Check that the response ID matches the query ID. We want
* to check this before actually parsing the message, so that
* if there's a malformed response that's not ours, it
* doesn't confuse us.
*/
int id = ((b_in[0] & 0xFF) << 8) + (b_in[1] & 0xFF);
int qid = query.getHeader().getID();
if (id != qid) {
String error = "invalid message id: expected " + qid + "; got id " + id;
if (tcp) {
throw new WireParseException(error);
} else {
if (Options.check("verbose")) {
System.err.println(error);
}
continue;
}
}
Message response = parseMessage(b_in);
verifyTSIG(query, response, b_in, tsig);
if (!tcp && !ignoreTruncation && response.getHeader().getFlag(Flags.TC)) {
tcp = true;
continue;
}
return response;
} while (true);
}
private void verifyTSIG(Message query, Message response, byte[] b, TSIG tsig) {
if (tsig == null) return;
int error = tsig.verify(response, b, query.getTSIG());
if (Options.check("verbose")) System.err.println("TSIG verify: " + Rcode.TSIGstring(error));
}
public static void main(String[] args) throws Exception {
boolean isInteractive = false;
classUrl = MynaInstaller.class.getResource("MynaInstaller.class").toString();
isJar = (classUrl.indexOf("jar") == 0);
if (!isJar) {
System.err.println("Installer can only be run from inside a Myna distribution war file");
System.exit(1);
}
Thread.sleep(1000);
Console console = System.console();
String response = null;
CommandLineParser parser = new PosixParser();
// create the Options
Options options = new Options();
options.addOption(
"c", "context", true, "Webapp context. Must Start with \"/\" Default: " + webctx);
options.addOption("h", "help", false, "Displays help.");
options.addOption(
"w",
"webroot",
true,
"Webroot to use. Will be created if webroot/WEB-INF does not exist. Default: " + webroot);
options.addOption(
"l",
"logfile",
true,
"Log file to use. Will be created if it does not exist. Default: ./<context>.log");
options.addOption(
"s",
"servername",
true,
"Name of this instance. Will also be the name of the init script. Defaults to either \"myna\" or the value of <context> if defined");
// options.addOption( "P", "purpose", true, "Purpose of the Server, such as DEV,PROD,TRAIN, etc.
// Defaults to DEV" );
options.addOption("p", "port", true, "HTTP port. Set to 0 to disable HTTP. Default: " + port);
options.addOption(
"sp", "ssl-port", true, "SSL (HTTPS) port. Set to 0 to disable SSL, Default: 0");
options.addOption(
"ks", "keystore", true, "keystore path. Default: <webroot>/WEB-INF/myna/myna_keystore");
options.addOption("ksp", "ks-pass", true, "keystore password. Default: " + ksPass);
options.addOption("ksa", "ks-alias", true, "certificate alias. Default: " + ksAlias);
modeOptions.add("upgrade");
modeOptions.add("install");
options.addOption(
"m",
"mode",
true,
"Mode: one of "
+ modeOptions.toString()
+ ". \n"
+ "\"upgrade\": Upgrades myna installation in webroot and exits. "
+ "\"install\": Unpacks to webroot, and installs startup files");
options.addOption(
"u",
"user",
true,
"User to own and run the Myna installation. Only applies to unix installs. Default: nobody");
HelpFormatter formatter = new HelpFormatter();
String cmdSyntax = "java -jar myna-X.war -m <mode> [options]";
try {
CommandLine line = parser.parse(options, args);
Option option;
if (args.length == 0) {
formatter.printHelp(cmdSyntax, options);
response = console.readLine("\nContinue with Interactive Install? (y/N)");
if (response.toLowerCase().equals("y")) {
isInteractive = true;
} else System.exit(1);
}
// Help
if (line.hasOption("help")) {
formatter.printHelp(cmdSyntax, options);
System.exit(1);
}
// mode
if (line.hasOption("mode")) {
mode = line.getOptionValue("mode");
if (!modeOptions.contains(mode)) {
System.err.println(
"Invalid Arguments. Reason: Mode must be in " + modeOptions.toString());
formatter.printHelp(cmdSyntax, options);
System.exit(1);
}
} else if (isInteractive) {
option = options.getOption("mode");
console.printf("\n" + option.getDescription());
do {
response = console.readLine("\nEnter " + option.getLongOpt() + "(" + mode + "): ");
if (!response.isEmpty()) mode = response;
} while (!modeOptions.contains(mode));
}
// webroot
if (line.hasOption("webroot")) {
webroot = line.getOptionValue("webroot");
} else if (isInteractive) {
option = options.getOption("webroot");
console.printf("\n" + option.getDescription());
response = console.readLine("\nEnter " + option.getLongOpt() + "(" + webroot + "): ");
if (!response.isEmpty()) webroot = response;
}
// port
if (line.hasOption("port")) {
port = Integer.parseInt(line.getOptionValue("port"));
} else if (isInteractive && mode.equals("install")) {
option = options.getOption("port");
console.printf("\n" + option.getDescription());
response = console.readLine("\nEnter " + option.getLongOpt() + "(" + port + "): ");
if (!response.isEmpty()) port = Integer.parseInt(response);
}
// context
if (line.hasOption("context")) {
webctx = line.getOptionValue("context");
} else if (isInteractive && mode.equals("install")) {
option = options.getOption("context");
console.printf("\n" + option.getDescription());
response = console.readLine("\nEnter " + option.getLongOpt() + "(" + webctx + "): ");
if (!response.isEmpty()) webctx = response;
}
if (!webctx.startsWith("/")) {
webctx = "/" + webctx;
}
// servername (depends on context)
if (!webctx.equals("/")) {
serverName = webctx.substring(1);
}
if (line.hasOption("servername")) {
serverName = line.getOptionValue("servername");
} else if (isInteractive && mode.equals("install")) {
option = options.getOption("servername");
console.printf("\n" + option.getDescription());
response = console.readLine("\nEnter " + option.getLongOpt() + "(" + serverName + "): ");
if (!response.isEmpty()) serverName = response;
}
// user
if (line.hasOption("user")) {
user = line.getOptionValue("user");
} else if (isInteractive && mode.equals("install")) {
option = options.getOption("user");
console.printf("\n" + option.getDescription());
response = console.readLine("\nEnter " + option.getLongOpt() + "(" + user + "): ");
if (!response.isEmpty()) user = response;
}
// logfile
logFile = "myna.log";
if (!webctx.equals("/")) {
logFile = webctx.substring(1) + ".log";
}
if (line.hasOption("logfile")) {
logFile = line.getOptionValue("logfile");
} else if (isInteractive && mode.equals("install")) {
option = options.getOption("logfile");
console.printf("\n" + option.getDescription());
response = console.readLine("\nEnter " + option.getLongOpt() + "path(" + logFile + "): ");
if (!response.isEmpty()) logFile = response;
}
// ssl-port
if (line.hasOption("ssl-port")) {
sslPort = Integer.parseInt(line.getOptionValue("ssl-port"));
} else if (isInteractive && mode.equals("install")) {
option = options.getOption("ssl-port");
console.printf("\n" + option.getDescription());
response = console.readLine("\nEnter " + option.getLongOpt() + "(" + sslPort + "): ");
if (!response.isEmpty()) sslPort = Integer.parseInt(response);
}
// ks-pass
if (line.hasOption("ks-pass")) {
ksPass = line.getOptionValue("ks-pass");
} else if (isInteractive && mode.equals("install")) {
option = options.getOption("ks-pass");
console.printf("\n" + option.getDescription());
response = console.readLine("\nEnter " + option.getLongOpt() + "(" + ksPass + "): ");
if (!response.isEmpty()) ksPass = response;
}
// ks-alias
if (line.hasOption("ks-alias")) {
ksAlias = line.getOptionValue("ks-alias");
} else if (isInteractive && mode.equals("install")) {
option = options.getOption("ks-alias");
console.printf("\n" + option.getDescription());
response = console.readLine("\nEnter " + option.getLongOpt() + "(" + ksAlias + "): ");
if (!response.isEmpty()) ksAlias = response;
}
// keystore
String appBase = new File(webroot).getCanonicalPath();
if (keystore == null) {
keystore = appBase + "/WEB-INF/myna/myna_keystore";
}
if (line.hasOption("keystore")) {
keystore = line.getOptionValue("keystore");
} else if (isInteractive && mode.equals("install")) {
option = options.getOption("keystore");
console.printf("\n" + option.getDescription());
response = console.readLine("\nEnter " + option.getLongOpt() + "(" + keystore + "): ");
if (!response.isEmpty()) keystore = response;
}
javaOpts = line.getArgList();
} catch (ParseException exp) {
System.err.println("Invalid Arguments. Reason: " + exp.getMessage());
formatter.printHelp(cmdSyntax, options);
System.exit(1);
}
if (isInteractive) {
System.out.println("\nProceeed with the following settings?:\n");
System.out.println("mode = " + mode);
System.out.println("webroot = " + webroot);
if (mode.equals("install")) {
System.out.println("port = " + port);
System.out.println("context = " + webctx);
System.out.println("servername = " + serverName);
System.out.println("user = "******"logfile = " + logFile);
System.out.println("ssl-port = " + sslPort);
System.out.println("ks-pass = "******"ks-alias = " + ksAlias);
System.out.println("keystore = " + keystore);
}
response = console.readLine("Continue? (Y/n)");
if (response.toLowerCase().equals("n")) System.exit(1);
}
File wrFile = new File(webroot);
webroot = wrFile.toString();
if (mode.equals("install")) {
adminPassword = console.readLine("\nCreate an Admin password for this installation: ");
}
// unpack myna if necessary
if (!wrFile.exists() || mode.equals("upgrade") || mode.equals("install")) {
upgrade(wrFile);
}
if (mode.equals("install")) {
File propertiesFile = new File(wrFile.toURI().resolve("WEB-INF/classes/general.properties"));
FileInputStream propertiesFileIS = new FileInputStream(propertiesFile);
Properties generalProperties = new Properties();
generalProperties.load(propertiesFileIS);
propertiesFileIS.close();
if (!adminPassword.isEmpty()) {
org.jasypt.util.password.StrongPasswordEncryptor cryptTool =
new org.jasypt.util.password.StrongPasswordEncryptor();
generalProperties.setProperty("admin_password", cryptTool.encryptPassword(adminPassword));
}
generalProperties.setProperty("instance_id", serverName);
generalProperties.store(
new java.io.FileOutputStream(propertiesFile), "Myna General Properties");
String javaHome = System.getProperty("java.home");
webroot = new File(webroot).getCanonicalPath();
if (serverName.length() == 0) serverName = "myna";
if (java.lang.System.getProperty("os.name").toLowerCase().indexOf("win") >= 0) {
if (!new File(logFile).isAbsolute()) {
logFile = new File(wrFile.toURI().resolve("WEB-INF/" + logFile)).toString();
}
File templateFile =
new File(
wrFile.toURI().resolve("WEB-INF/myna/install/windows/update_myna_service.cmd"));
String initScript =
FileUtils.readFileToString(templateFile)
.replaceAll("\\{webctx\\}", webctx)
.replaceAll("\\{server\\}", Matcher.quoteReplacement(serverName))
.replaceAll("\\{webroot\\}", Matcher.quoteReplacement(webroot))
.replaceAll("\\{logfile\\}", Matcher.quoteReplacement(logFile))
.replaceAll("\\{javahome\\}", Matcher.quoteReplacement(javaHome))
.replaceAll("\\{port\\}", new Integer(port).toString())
.replaceAll("\\{sslPort\\}", new Integer(sslPort).toString())
.replaceAll("\\{keystore\\}", Matcher.quoteReplacement(keystore))
.replaceAll("\\{ksPass\\}", Matcher.quoteReplacement(ksPass))
.replaceAll("\\{ksAlias\\}", Matcher.quoteReplacement(ksAlias));
File scriptFile =
new File(wrFile.toURI().resolve("WEB-INF/myna/install/update_myna_service.cmd"));
FileUtils.writeStringToFile(scriptFile, initScript);
// Runtime.getRuntime().exec("cmd /c start " + scriptFile.toString()).waitFor();
System.out.println(
"\nInstalled Service 'Myna App Server " + serverName + "' the following settings:\n");
System.out.println(
"\nInit script '" + scriptFile + "' created with the following settings:\n");
System.out.println("memory=256MB");
System.out.println("serverName=" + serverName);
System.out.println("javaHome=" + javaHome);
System.out.println("context=" + webctx);
System.out.println("port=" + port);
System.out.println("myna_home=" + webroot);
System.out.println("logfile=" + logFile);
System.out.println("sslPort=" + sslPort);
System.out.println("keyStore=" + keystore);
System.out.println("ksPass="******"ksAlias=" + ksAlias);
System.out.println(
"\nEdit and and run the command file in " + scriptFile + " to update this service");
} else {
String curUser = java.lang.System.getProperty("user.name");
if (!curUser.equals("root")) {
System.out.println("Install mode must be run as root.");
System.exit(1);
}
if (!new File(logFile).isAbsolute()) {
logFile = new File(wrFile.toURI().resolve("WEB-INF/" + logFile)).toString();
}
File templateFile =
new File(wrFile.toURI().resolve("WEB-INF/myna/install/linux/init_script"));
String initScript =
FileUtils.readFileToString(templateFile)
.replaceAll("\\{webctx\\}", webctx)
.replaceAll("\\{server\\}", serverName)
.replaceAll("\\{user\\}", user)
.replaceAll("\\{webroot\\}", webroot)
.replaceAll("\\{javahome\\}", javaHome)
.replaceAll("\\{logfile\\}", logFile)
.replaceAll("\\{port\\}", new Integer(port).toString())
.replaceAll("\\{sslPort\\}", new Integer(sslPort).toString())
.replaceAll("\\{keystore\\}", keystore)
.replaceAll("\\{ksPass\\}", ksPass)
.replaceAll("\\{ksAlias\\}", ksAlias);
File scriptFile = new File(wrFile.toURI().resolve("WEB-INF/myna/install/" + serverName));
FileUtils.writeStringToFile(scriptFile, initScript);
if (new File("/etc/init.d").exists()) {
exec("chown -R " + user + " " + webroot);
exec("chown root " + scriptFile.toString());
exec("chmod 700 " + scriptFile.toString());
exec("cp " + scriptFile.toString() + " /etc/init.d/");
System.out.println(
"\nInit script '/etc/init.d/"
+ serverName
+ "' created with the following settings:\n");
} else {
System.out.println(
"\nInit script '" + scriptFile + "' created with the following settings:\n");
}
System.out.println("user="******"memory=256MB");
System.out.println("server=" + serverName);
System.out.println("context=" + webctx);
System.out.println("port=" + port);
System.out.println("myna_home=" + webroot);
System.out.println("logfile=" + logFile);
System.out.println("sslPort=" + sslPort);
System.out.println("keyStore=" + keystore);
System.out.println("ksPass="******"ksAlias=" + ksAlias);
System.out.println("\nEdit this file to customize startup behavior");
}
}
}
