scouter.agent.netio.request.handle.AgentClassHandle Maven / Gradle / Ivy
/*
* Copyright 2015 the original author or authors.
* @https://github.com/scouter-project/scouter
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package scouter.agent.netio.request.handle;
import java.io.File;
import java.io.InputStream;
import java.net.JarURLConnection;
import java.net.URL;
import scouter.agent.JavaAgent;
import scouter.agent.Logger;
import scouter.agent.netio.request.anotation.RequestHandler;
import scouter.lang.pack.MapPack;
import scouter.lang.pack.Pack;
import scouter.lang.value.BlobValue;
import scouter.lang.value.ListValue;
import scouter.lang.value.MapValue;
import scouter.lang.value.Value;
import scouter.net.RequestCmd;
import scouter.util.ClassUtil;
import scouter.util.FileUtil;
public class AgentClassHandle {
@RequestHandler(RequestCmd.OBJECT_LOAD_CLASS_BY_STREAM)
public Pack loadClassAsStream(Pack param) {
MapPack p = (MapPack) param;
String className = p.getText("class");
InputStream is = null;
try {
Class clazz = getClass(className);
if (clazz == null) {
p.put("error", "Not found class " + className);
return p;
}
String clsAsResource = "/" + className.replace('.', '/').concat(".class");
is = clazz.getResourceAsStream(clsAsResource);
p.put("class", new BlobValue(FileUtil.readAll(is)));
} catch (Throwable th) {
Logger.println("A126", th);
p.put("error", th.getMessage());
return p;
} finally {
FileUtil.close(is);
}
return p;
}
@RequestHandler(RequestCmd.OBJECT_CLASS_DESC)
public Pack getClassInfo(Pack param) {
MapPack p = (MapPack) param;
String className = p.getText("class");
try {
Class clazz = getClass(className);
if (clazz == null) {
p.put("error", "Not found class " + className);
return p;
}
p.put("class", ClassUtil.getClassDescription(clazz));
} catch (Throwable th) {
Logger.println("A904", th);
p.put("error", th.getMessage());
return p;
}
return p;
}
private ListValue toValue(Class[] inf) {
ListValue v = new ListValue();
for (int i = 0; i < inf.length; i++) {
v.add(inf[i].getName());
}
return v;
}
private Class getClass(String className) {
Class[] loadedClasses = JavaAgent.getInstrumentation().getAllLoadedClasses();
for (Class c : loadedClasses) {
if (c.getName().equals(className)) {
return c;
}
}
return null;
}
@RequestHandler(RequestCmd.OBJECT_CHECK_RESOURCE_FILE)
public Pack checkJarFile(Pack param) {
MapPack p = (MapPack) param;
String resource = p.getText("resource");
MapPack m = new MapPack();
try {
URL url = new URL(resource);
JarURLConnection connection = (JarURLConnection) url.openConnection();
File file = new File(connection.getJarFileURL().toURI());
if (file.exists() == false) {
m.put("error", "Cannot find jar file.");
} else {
if (file.canRead()) {
m.put("name", file.getName());
m.put("size", file.length());
} else {
m.put("error", "Cannot read jar file.");
}
}
} catch (Exception e) {
m.put("error", e.toString());
}
return m;
}
@RequestHandler(RequestCmd.OBJECT_DOWNLOAD_JAR)
public Pack downloadJar(Pack param) {
MapPack p = (MapPack) param;
String resource = p.getText("resource");
MapPack m = new MapPack();
try {
URL url = new URL(resource);
JarURLConnection connection = (JarURLConnection) url.openConnection();
File file = new File(connection.getJarFileURL().toURI());
if (file.exists() == false) {
m.put("error", "Cannot find jar file.");
} else {
if (file.canRead()) {
m.put("jar", new BlobValue(FileUtil.readAll(file)));
} else {
m.put("error", "Cannot read jar file.");
}
}
} catch (Exception e) {
m.put("error", e.toString());
}
return m;
}
}