From b086f57f56bf0eb9dab9cf321a0f69aaaae84347 Mon Sep 17 00:00:00 2001 From: Andrew Branson Date: Wed, 30 May 2012 08:37:45 +0200 Subject: Initial Maven Conversion --- .../java/com/c2kernel/scripting/ScriptConsole.java | 229 +++++++++++++++++++++ 1 file changed, 229 insertions(+) create mode 100644 src/main/java/com/c2kernel/scripting/ScriptConsole.java (limited to 'src/main/java/com/c2kernel/scripting/ScriptConsole.java') diff --git a/src/main/java/com/c2kernel/scripting/ScriptConsole.java b/src/main/java/com/c2kernel/scripting/ScriptConsole.java new file mode 100644 index 0000000..be3f79f --- /dev/null +++ b/src/main/java/com/c2kernel/scripting/ScriptConsole.java @@ -0,0 +1,229 @@ +package com.c2kernel.scripting; + +import java.io.BufferedReader; +import java.io.IOException; +import java.io.InputStreamReader; +import java.io.InterruptedIOException; +import java.io.PrintStream; +import java.net.InetAddress; +import java.net.Socket; +import java.net.UnknownHostException; +import java.util.ArrayList; +import java.util.StringTokenizer; + +import javax.script.Bindings; +import javax.script.ScriptEngine; +import javax.script.ScriptEngineManager; + +import org.tanukisoftware.wrapper.WrapperManager; + +import com.c2kernel.process.Gateway; +import com.c2kernel.utils.Logger; +import com.c2kernel.utils.Resource; +import com.c2kernel.utils.server.SocketHandler; + +/************************************************************************** + * + * $Revision: 1.16 $ + * $Date: 2005/08/31 07:20:40 $ + * + * Copyright (C) 2003 CERN - European Organization for Nuclear Research + * All rights reserved. + **************************************************************************/ + + + +public class ScriptConsole implements SocketHandler { + BufferedReader input; + PrintStream output; + Socket socket = null; + ScriptEngine engine; + Bindings beans; + static ArrayList securityHosts = new ArrayList(); + public static final short NONE = 0; + public static final short ALLOW = 1; + public static final short DENY = 2; + static short securityMode; + + static { + securityMode = ALLOW; + String hosts = Gateway.getProperty("ItemServer.Console.allow"); + if (hosts == null || hosts.equals("")) { + securityMode = DENY; + hosts = Gateway.getProperty("ItemServer.Console.deny"); + } + if (hosts == null || hosts.equals("")) { // by default only allow localhost + securityMode = ALLOW; + securityHosts.add("127.0.0.1"); + } + else { + StringTokenizer tok = new StringTokenizer(hosts, ","); + while(tok.hasMoreTokens()) { + try { + securityHosts.add(InetAddress.getByName(tok.nextToken()).getHostAddress()); + } catch (UnknownHostException ex) { + Logger.error("Host not found "+ex.getMessage()); + } + } + } + } + public ScriptConsole() { + } + + @Override + public String getName() { + return "Script Console"; + } + + @Override + public boolean isBusy() { + return (socket != null); + } + + @Override + public void setSocket(Socket newSocket) { + try { + input = new BufferedReader(new InputStreamReader(newSocket.getInputStream())); + output = new PrintStream(newSocket.getOutputStream()); + newSocket.setSoTimeout(0); + socket = newSocket; + } catch (IOException ex) { + try { + newSocket.close(); + } catch (IOException ex2) { + } + socket = null; + return; + } + } + + @Override + public void shutdown() { + Socket closingSocket = socket; + socket = null; + if (closingSocket == null) + return; + try { + Logger.removeLogStream(output); + closingSocket.shutdownInput(); + closingSocket.shutdownOutput(); + closingSocket.close(); + Logger.msg("Script console to "+closingSocket.getInetAddress()+" closed"); + } catch (IOException e) { + Logger.error("Script Console to " + closingSocket.getInetAddress() + " - Error closing."); + Logger.error(e); + } + } + + + @Override + public void run() { + // check permission + boolean allowed = true; + if (securityMode!=NONE) { + if (securityHosts.contains(socket.getInetAddress().getHostAddress())) { + if (securityMode==DENY) + allowed = false; + } + else if (securityMode==ALLOW) + allowed = false; + } + + if (!allowed) { + Logger.error("Host "+socket.getInetAddress()+" access denied"); + output.println("Host "+socket.getInetAddress()+" access denied"); + shutdown(); + return; + } + + // get system objects + try { + Logger.addLogStream(output, 0); + + engine = new ScriptEngineManager().getEngineByName("JavaScript"); + beans = engine.createBindings(); + beans.put("storage", Gateway.getStorage()); + beans.put("db", Gateway.getStorage().getDb()); + beans.put("proxy", Gateway.getProxyManager()); + beans.put("lookup", Gateway.getLDAPLookup()); + beans.put("orb", Gateway.getORB()); + try { + beans.put("system", Gateway.getProxyManager().getProxy( + Gateway.getLDAPLookup().getRoleManager().getAgentPath("system"))); + } catch (Exception ex) { + output.println("System agent unavailable"); + } + + StringBuffer commandBuffer = new StringBuffer(); + while (socket != null) { + + output.println(); + output.print('>'); + + String command = null; + boolean gotCommand = false; + while (!gotCommand) { + try { + command = input.readLine(); + gotCommand = true; + } catch (InterruptedIOException ex) { + } + } + if (command == null) // disconnected + shutdown(); + else { + if (command.equals("exit")) { + shutdown(); + continue; + } + else if(command.startsWith("log")) { + try { + int newLogLevel = Integer.parseInt(command.substring(4)); + Logger.removeLogStream(output); + Logger.addLogStream(output, newLogLevel); + Logger.msg("Log level for "+socket.getInetAddress()+" set to "+newLogLevel); + continue; + } catch (NumberFormatException ex) { } + } + else if(command.equals("help")) { + output.println(Resource.findTextResource("textFiles/consoleHelp.txt")); + continue; + } + else if(command.equals("version")) { + output.println("Kernel version: "+Resource.getKernelVersion()); + output.println("Modules loaded: "+Gateway.getModuleManager().getModuleVersions()); + continue; + } + try { + if (command.endsWith("\\")) { + commandBuffer.append(command.substring(0,command.length()-1)); + continue; + } + commandBuffer.append(command); + command = commandBuffer.toString(); + commandBuffer = new StringBuffer(); + Logger.msg("Console command from "+socket.getInetAddress()+": "+command); + + // process control + if (command.equals("shutdown")) { + WrapperManager.stop(0); + } + else { + Object response = engine.eval(command, beans); + if (response instanceof sun.org.mozilla.javascript.internal.Undefined) + output.println("Ok"); + else + output.println(response); + } + } catch (Throwable ex) { + ex.printStackTrace(output); + } + output.flush(); + } + } + } catch (IOException ex) { + Logger.error("IO Exception reading from script console socket"); + shutdown(); + } + } +} -- cgit v1.2.3