001 /*
002 * Copyright (C) 2011 eXo Platform SAS.
003 *
004 * This is free software; you can redistribute it and/or modify it
005 * under the terms of the GNU Lesser General Public License as
006 * published by the Free Software Foundation; either version 2.1 of
007 * the License, or (at your option) any later version.
008 *
009 * This software is distributed in the hope that it will be useful,
010 * but WITHOUT ANY WARRANTY; without even the implied warranty of
011 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
012 * Lesser General Public License for more details.
013 *
014 * You should have received a copy of the GNU Lesser General Public
015 * License along with this software; if not, write to the Free
016 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
017 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
018 */
019 package org.crsh.standalone;
020
021 import org.crsh.cmdline.ClassDescriptor;
022 import org.crsh.cmdline.CommandFactory;
023 import org.crsh.cmdline.annotations.Argument;
024 import org.crsh.cmdline.annotations.Command;
025 import org.crsh.cmdline.annotations.Option;
026 import org.crsh.cmdline.matcher.CommandMatch;
027 import org.crsh.cmdline.matcher.InvocationContext;
028 import org.crsh.cmdline.matcher.Matcher;
029 import org.crsh.shell.impl.CRaSH;
030 import org.crsh.term.BaseTerm;
031 import org.crsh.term.Term;
032 import org.crsh.term.processor.Processor;
033 import org.crsh.term.spi.net.TermIOClient;
034 import org.slf4j.Logger;
035 import org.slf4j.LoggerFactory;
036
037 import java.io.Closeable;
038 import java.io.File;
039 import java.lang.instrument.Instrumentation;
040 import java.util.Collections;
041 import java.util.List;
042 import java.util.Map;
043 import java.util.Properties;
044
045 /**
046 * @author <a href="mailto:julien.viet@exoplatform.com">Julien Viet</a>
047 */
048 public class Agent {
049
050 /** . */
051 private static Logger log = LoggerFactory.getLogger(Agent.class);
052
053 public static void agentmain(final String agentArgs, final Instrumentation inst) throws Exception {
054 log.info("CRaSH agent loaded");
055
056 //
057 Thread t = new Thread() {
058 @Override
059 public void run() {
060 try {
061 ClassDescriptor<Agent> c = CommandFactory.create(Agent.class);
062 Matcher<Agent> matcher = Matcher.createMatcher("main", c);
063 CommandMatch<Agent, ?, ?> match = matcher.match(agentArgs);
064 match.invoke(new InvocationContext(), new Agent(inst));
065 } catch (Exception e) {
066 e.printStackTrace();
067 }
068 }
069 };
070
071 //
072 t.start();
073 log.info("Spawned CRaSH thread " + t.getId() + " for further processing");
074 }
075
076 /** . */
077 private final Instrumentation instrumentation;
078
079 public Agent(Instrumentation instrumentation) {
080 this.instrumentation = instrumentation;
081 }
082
083 @Command
084 public void main(
085 @Option(names={"j","jar"})
086 List<String> jars,
087 @Option(names={"c","cmd"})
088 List<String> cmds,
089 @Option(names={"conf"})
090 List<String> confs,
091 @Option(names={"p","property"})
092 List<String> properties,
093 @Argument(name = "port")
094 Integer port) throws Exception {
095
096 //
097 Bootstrap bootstrap = new Bootstrap(Thread.currentThread().getContextClassLoader());
098
099 //
100 if (cmds != null) {
101 for (String cmd : cmds) {
102 File cmdPath = new File(cmd);
103 bootstrap.addCmdPath(cmdPath);
104 }
105 }
106
107 //
108 if (confs != null) {
109 for (String conf : confs) {
110 File confPath = new File(conf);
111 bootstrap.addConfPath(confPath);
112 }
113 }
114
115 //
116 if (jars != null) {
117 for (String jar : jars) {
118 File jarFile = new File(jar);
119 bootstrap.addJarPath(jarFile);
120 }
121 }
122
123 //
124 if (properties != null) {
125 Properties config = new Properties();
126 for (String property : properties) {
127 int index = property.indexOf('=');
128 if (index == -1) {
129 config.setProperty(property, "");
130 } else {
131 config.setProperty(property.substring(0, index), property.substring(index + 1));
132 }
133 }
134 bootstrap.setConfig(config);
135 }
136
137 // Set the instrumentation available as an attribute
138 Map<String, Object> attributes = Collections.<String, Object>singletonMap("instrumentation", instrumentation);
139 bootstrap.setAttributes(attributes);
140
141 // Do bootstrap
142 bootstrap.bootstrap();
143
144 //
145 try {
146 final TermIOClient client = new TermIOClient(port);
147 log.info("Callback back remote on port " + port);
148 client.connect();
149
150 // Do stuff
151 Term term = new BaseTerm(client);
152 CRaSH crash = new CRaSH(bootstrap.getContext());
153 Processor processor = new Processor(term, crash.createSession(null));
154 processor.addListener(new Closeable() {
155 public void close() {
156 client.close();
157 }
158 });
159
160 // Run
161 processor.run();
162 }
163 finally {
164 bootstrap.shutdown();
165 }
166 }
167 }