+
−/*
+
− * This Source Code Form is subject to the terms of the Mozilla Public
+
− * License, v. 2.0.  If a copy of the MPL was not distributed with this
+
− * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
− *
+
− * Copyright 1997 - July 2008 CWI, August 2008 - 2016 MonetDB B.V.
+
− */
+
−
+
−package nl.cwi.monetdb.merovingian;
+
−
+
−import nl.cwi.monetdb.mcl.connection.MCLException;
+
−import nl.cwi.monetdb.mcl.connection.mapi.MapiConnection;
+
−import nl.cwi.monetdb.mcl.protocol.AbstractProtocol;
+
−import nl.cwi.monetdb.mcl.protocol.ProtocolException;
+
−import nl.cwi.monetdb.mcl.protocol.ServerResponses;
+
−
+
−import java.io.BufferedReader;
+
−import java.io.IOException;
+
−import java.io.InputStreamReader;
+
−import java.io.PrintStream;
+
−import java.net.Socket;
+
−import java.net.URI;
+
−import java.net.URISyntaxException;
+
−import java.util.ArrayList;
+
−import java.util.Collections;
+
−import java.util.List;
+
−import java.util.Properties;
+
−
+
−/**
+
− * A Control class to perform operations on a remote merovingian
+
− * instance, using the TCP control protocol.
+
− * 
+
− * This class implements the protocol specific bits to perform all
+
− * possible actions against a merovingian server that has remote control
+
− * facilities enabled.
+
− * <br />
+
− * In the merovingian world, other merovingians in the vicinity are
+
− * known to each merovingian, allowing to perform cluster wide actions.
+
− * The implementation taken in this class is to require one known
+
− * merovingian to get insight in the entire network.  Note that
+
− * connecting to a merovingian requires a passphrase that is likely to
+
− * be different for each merovingian.
+
− *
+
− * @author Fabian Groffen
+
− * @version 1.0
+
− */
+
−public class Control {
+
−	/** The host to connect to */
+
−	private final String host;
+
−	/** The port to connect to */
+
−	private final int port;
+
−	/** The passphrase to use when connecting */
+
−	private final String passphrase;
+
−	/** The file we should write MapiSocket debuglog to */
+
−
+
−	/**
+
−	 * Constructs a new Control object.
+
−	 *
+
−	 * @throws IllegalArgumentException if host, port or passphrase are
+
−	 * null or &lt;= 0
+
−	 */
+
−	public Control(String host, int port, String passphrase) throws IllegalArgumentException {
+
−		this.host = host;
+
−		this.port = port;
+
−		this.passphrase = passphrase;
+
−	}
+
−
+
−	private String controlHash(String pass, String salt) {
+
−		long ho;
+
−		long h = 0;
+
−
+
−		/* use a very simple hash function designed for a single int val
+
−		 * (hash buckets), we can make this more interesting if necessary in
+
−		 * the future.
+
−		 * http://www.cs.hmc.edu/~geoff/classes/hmc.cs070.200101/homework10/hashfuncs.html */
+
−
+
−		for (int i = 0; i < pass.length(); i++) {
+
−			ho = h & 0xF8000000L;
+
−			h <<= 5;
+
−			h &= 0xFFFFFFFFL;
+
−			h ^= ho >>> 27;
+
−			h ^= (int)(pass.charAt(i));
+
−		}
+
−
+
−		for (int i = 0; i < salt.length(); i++) {
+
−			ho = h & 0xF8000000L;
+
−			h <<= 5;
+
−			h &= 0xFFFFFFFFL;
+
−			h ^= ho >> 27;
+
−			h ^= (int)(salt.charAt(i));
+
−		}
+
−
+
−		return(Long.toString(h));
+
−	}
+
−	
+
−	final static private String RESPONSE_OK = "OK";
+
−
+
−	private List<String> sendCommand(String database, String command, boolean hasOutput)
+
−			throws MerovingianException, IOException {
+
−
+
−		MapiConnection server = new MapiConnection(null, "merovingian",null, "control", false, true, host, port );
+
−		AbstractProtocol protocol = server.getProtocol();
+
−		try {
+
−			server.connect("monetdb", passphrase);
+
−		} catch (ProtocolException | MCLException e) {
+
−			throw new MerovingianException(e.getMessage());
+
−		} catch (AssertionError e) { // mcl panics
+
−			server.close();
+
−			
+
−			// Try old protocol instead
+
−			Socket s;
+
−			PrintStream out;
+
−			BufferedReader in;
+
−			s = new Socket(host, port);
+
−			out = new PrintStream(s.getOutputStream());
+
−			in = new BufferedReader(new InputStreamReader(s.getInputStream()));
+
−			try {
+
−				/* login ritual, step 1: get challenge from server */
+
−				String response = in.readLine();
+
−				if (response == null)
+
−					throw new MerovingianException("server closed the connection");
+
−
+
−				if (!response.startsWith("merovingian:1:") &&
+
−						!response.startsWith("merovingian:2:"))
+
−					throw new MerovingianException("unsupported merovingian server");
+
−
+
−				String[] tokens = response.split(":");
+
−				if (tokens.length < 3)
+
−					throw new MerovingianException("did not understand merovingian server");
+
−				String version = tokens[1];
+
−				String token = tokens[2];
+
−
+
−				response = controlHash(passphrase, token);
+
−				if (version.equals("1")) {
+
−					out.print(response + "\n");
+
−				} else if (version.equals("2")) {
+
−					// we only support control mode for now
+
−					out.print(response + ":control\n");
+
−				}
+
−
+
−				response = in.readLine();
+
−				if (response == null) {
+
−					throw new MerovingianException("server closed the connection");
+
−				}
+
−
+
−				if (!response.equals(RESPONSE_OK)) {
+
−					throw new MerovingianException(response);
+
−				}
+
−
+
−				/* send command, form is simple: "<db> <cmd>\n" */
+
−				out.print(database + " " + command + "\n");
+
−
+
−				/* Response has the first line either "OK\n" or an error
+
−				 * message.  In case of a command with output, the data will
+
−				 * follow the first line */
+
−				response = in.readLine();
+
−				if (response == null) {
+
−					throw new MerovingianException("server closed the connection");
+
−				}
+
−				if (!response.equals(RESPONSE_OK)) {
+
−					throw new MerovingianException(response);
+
−				}
+
−
+
−				if (!hasOutput)
+
−					return null;
+
−
+
−				ArrayList<String> l = new ArrayList<>();
+
−				while ((response = in.readLine()) != null) {
+
−					l.add(response);
+
−				}
+
−				return l;
+
−			} finally {
+
−				in.close();
+
−				out.close();
+
−				s.close();
+
−			}
+
−		}
+
−
+
−		protocol.writeNextQuery(null,database + " " + command, "\n");
+
−		ArrayList<String> l = new ArrayList<>();
+
−		protocol.waitUntilPrompt();
+
−		ServerResponses next = protocol.getCurrentServerResponseHeader();
+
−		String line = protocol.getRemainingStringLine(0);
+
−
+
−		if (next == ServerResponses.ERROR)
+
−			throw new MerovingianException(line.substring(6));
+
−		if (next != ServerResponses.RESULT)
+
−			throw new MerovingianException("unexpected line: " + line);
+
−		if (!line.substring(1).equals(RESPONSE_OK))
+
−			throw new MerovingianException(line.substring(1));
+
−
+
−		next = protocol.getCurrentServerResponseHeader();
+
−		line = protocol.getRemainingStringLine(0);
+
−		while (next != ServerResponses.PROMPT) {
+
−			if (next != ServerResponses.RESULT)
+
−				throw new MerovingianException("unexpected line: " + line);
+
−
+
−			l.add(line.substring(1));
+
−
+
−			next = protocol.getCurrentServerResponseHeader();
+
−			line = protocol.getRemainingStringLine(0);
+
−		}
+
−
+
−		server.close();
+
−		return l;
+
−	}
+
−
+
−	public void start(String database) throws MerovingianException, IOException {
+
−		sendCommand(database, "start", false);
+
−	}
+
−
+
−	public void stop(String database) throws MerovingianException, IOException {
+
−		sendCommand(database, "stop", false);
+
−	}
+
−
+
−	public void kill(String database) throws MerovingianException, IOException {
+
−		sendCommand(database, "kill", false);
+
−	}
+
−
+
−	public void create(String database) throws MerovingianException, IOException {
+
−		sendCommand(database, "create", false);
+
−	}
+
−
+
−	public void destroy(String database) throws MerovingianException, IOException {
+
−		sendCommand(database, "destroy", false);
+
−	}
+
−
+
−	public void lock(String database) throws MerovingianException, IOException {
+
−		sendCommand(database, "lock", false);
+
−	}
+
−
+
−	public void release(String database) throws MerovingianException, IOException {
+
−		sendCommand(database, "release", false);
+
−	}
+
−
+
−	public void rename(String database, String newname) throws MerovingianException, IOException {
+
−		if (newname == null)
+
−			newname = ""; /* force error from merovingian */
+
−
+
−		sendCommand(database, "name=" + newname, false);
+
−	}
+
−
+
−	/**
+
−	 * Sets property for database to value.  If value is null, the
+
−	 * property is unset, and its inherited value becomes active again.
+
−	 *
+
−	 * @param database the target database
+
−	 * @param property the property to set value for
+
−	 * @param value the value to set
+
−	 * @throws MerovingianException if performing the command failed at
+
−	 *         the merovingian side
+
−	 * @throws IOException if connecting to or communicating with
+
−	 *         merovingian failed
+
−	 */
+
−	public void setProperty(String database, String property, String value) throws MerovingianException, IOException {
+
−		/* inherit: set to empty string */
+
−		if (value == null)
+
−			value = "";
+
−
+
−		sendCommand(database, property + "=" + value, false);
+
−	}
+
−
+
−	public void inheritProperty(String database, String property) throws MerovingianException, IOException {
+
−		setProperty(database, property, null);
+
−	}
+
−
+
−	public Properties getProperties(String database) throws MerovingianException, IOException {
+
−		Properties ret = new Properties();
+
−		List<String> response = sendCommand(database, "get", true);
+
−		for (String responseLine : response) {
+
−			if (responseLine.startsWith("#"))
+
−				continue;
+
−			int pos = responseLine.indexOf("=");
+
−			if (pos > 0) {
+
−				ret.setProperty(
+
−						responseLine.substring(0, pos),
+
−						responseLine.substring(pos + 1, responseLine.length()));
+
−			}
+
−		}
+
−		return ret;
+
−	}
+
−
+
−	public Properties getDefaultProperties() throws MerovingianException, IOException {
+
−		return(getProperties("#defaults"));
+
−	}
+
−
+
−	public SabaothDB getStatus(String database) throws MerovingianException, IOException {
+
−		List<String> response = sendCommand(database, "status", true);
+
−		if (response.isEmpty())
+
−			throw new MerovingianException("communication error");
+
−		return new SabaothDB(response.get(0));
+
−	}
+
−	
+
−	/**
+
−	 * Test whether a specific database exists. 
+
−	 * 
+
−	 * @param database
+
−	 * @return true, iff database already exists.
+
−	 * @throws MerovingianException
+
−	 * @throws IOException
+
−	 */
+
−	public boolean exists(String database) throws MerovingianException, IOException {
+
−		List<SabaothDB> all = getAllStatuses();
+
−		for (SabaothDB db : all) {
+
−			if (db.getName().equals(database)) {
+
−				return true;
+
−			}
+
−		}
+
−		return false;
+
−	}
+
−
+
−	public List<SabaothDB> getAllStatuses() throws MerovingianException, IOException {
+
−		List<SabaothDB> l = new ArrayList<>();
+
−		List<String> response = sendCommand("#all", "status", true);
+
−		try {
+
−			for (String responseLine : response) {
+
−				l.add(new SabaothDB(responseLine));
+
−			}
+
−		} catch (IllegalArgumentException e) {
+
−			throw new MerovingianException(e.getMessage());
+
−		}
+
−		return Collections.unmodifiableList(l);
+
−	}
+
−
+
−	public List<URI> getAllNeighbours() throws MerovingianException, IOException {
+
−		List<URI> l = new ArrayList<>();
+
−		List<String> response = sendCommand("anelosimus", "eximius", true);
+
−		try {
+
−			for (String responseLine : response) {
+
−				// format is <db>\t<uri>
+
−				String[] parts = responseLine.split("\t", 2);
+
−				if (parts.length != 2)
+
−					throw new MerovingianException("invalid entry: " +
+
−							responseLine);
+
−				if (parts[0].equals("*")) {
+
−					l.add(new URI(parts[1]));
+
−				} else {
+
−					l.add(new URI(parts[1] + parts[0]));
+
−				}
+
−			}
+
−		} catch (URISyntaxException e) {
+
−			throw new MerovingianException(e.getMessage());
+
−		}
+
−		return Collections.unmodifiableList(l);
+
−	}
+
−}