Nicksxs's Blog

java的agent继续体验

发表于 2024-12-29 分类于 java ，技巧阅读次数： Disqus：

在上次的基础上我们可以通过一些方法来获取参数的参数名，以此我们又可以达到类似于切面的功能逻辑，
首先我在之前的代码里做一点修改

public static void start(String input) {
    try {
        System.out.println("Demo starting ...");
        System.out.println(input);
        Thread.sleep(123);
    } catch (InterruptedException ignore) {
    }
}

就是带了个参数，这边是我把参数打了出来，如果没有打，我想通过agent来知道这个参数值是啥就可以用agent技术
获取代码中的变量名以及类型

MethodInfo methodInfo = declaredMethod.getMethodInfo();
CodeAttribute codeAttribute = methodInfo.getCodeAttribute();
LocalVariableAttribute attribute = (LocalVariableAttribute) codeAttribute.getAttribute(LocalVariableAttribute.tag);
System.out.println(attribute.variableName(0));

这边就是个string类型，我们就能够获取到 input 这个变量名，然后我们就可以在这里插入一段代码来打印这个参数

1	declaredMethod.insertAfter("System.out.println( " + attribute.variableName(0)+ " );");

这样我们就能做到类似于java中切面的技术

java的agent初体验

发表于 2024-12-22 分类于 java ，技巧阅读次数： Disqus：

之前在用到arthas就想到过可以研究下java的agent，这里算是个初入门
首先我们有个应用，需要挂上agent来探测一些事情
比如就是简单的主方法

public class Main {
    public static void main(String[] args) {
        System.out.println("Hello world!");
        Demo.start();
        Demo.start2();
    }
}

然后有个Demo类

public class Demo {
    public static void start() {
        try {
            System.out.println("Demo starting ...");
            Thread.sleep(123);
        } catch (InterruptedException ignore) {
        }
    }

    public static void start2() {
        try {
            System.out.println("Demo starting 2...");
            Thread.sleep(567);
        } catch (InterruptedException ignore) {
        }
    }
}

然后我们想要像切面一样在start方法前后织入一些前后置的逻辑
我们需要将一个maven的module
然后agent代码是这样, premain 方法会在我们指定挂载agent执行，并且就是入口方法

public class DemoAgent {

    public static void premain(String arg, Instrumentation instrumentation) {
        System.out.println("agent start");
        System.out.println("arg is " + arg);
        instrumentation.addTransformer(new DemoTransformer());
    }

}

然后这个 transformer 就是我要怎么在目标类里如何织入代码的逻辑

public class DemoTransformer implements ClassFileTransformer {
    // 这里是我们要织入代码的目标类
    private static final String INJECTED_CLASS = "org.example.Demo";

    @Override
    public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {
        String realClassName = className.replace("/", ".");
        // 匹配到目标类
        if (realClassName.equals(INJECTED_CLASS)) {
            System.out.println("injected class ：" + realClassName);
            CtClass ctClass;
            try {
                // 使用javassist,获取字节码类
                ClassPool classPool = ClassPool.getDefault();
                ctClass = classPool.get(realClassName);

                // 获取目标类所有的方法，也可指定方法，进行增强
                CtMethod[] declaredMethods = ctClass.getDeclaredMethods();
                for (CtMethod declaredMethod : declaredMethods) {
                    System.out.println(declaredMethod.getName() + "method be inject");
                    declaredMethod.addLocalVariable("time", CtClass.longType);
                    declaredMethod.insertBefore("System.out.println(\"--- start ---\");");
                    declaredMethod.insertBefore("time = System.currentTimeMillis();");
                    declaredMethod.insertAfter("System.out.println(\"--- end ---\");");
                    declaredMethod.insertAfter("System.out.println(\"cost: \" + (System.currentTimeMillis() - time));");
                }
                return ctClass.toBytecode();
            } catch (Throwable ignore) {
            }
        }
        return classfileBuffer;
    }
}

接下去要在pom.xml里配置如何打包agent的命令

<build>
        <plugins>
            <plugin>
                <groupId>org.apache.maven.plugins</groupId>
                <artifactId>maven-compiler-plugin</artifactId>
                <version>3.5.1</version>
                <!-- 指定maven编译的jdk版本。若不指定,maven3默认用jdk 1.5 maven2默认用jdk1.3 -->
                <configuration>
                    <source>8</source>
                    <target>8</target>
                </configuration>
            </plugin>

            <plugin>
                <groupId>org.apache.maven.plugins</groupId>
                <artifactId>maven-jar-plugin</artifactId>
                <version>3.2.0</version>
                <configuration>
                    <archive>
                        <!--自动添加META-INF/MANIFEST.MF -->
                        <manifest>
                            <addClasspath>true</addClasspath>
                        </manifest>
                        <manifestEntries>
                            <Menifest-Version>1.0</Menifest-Version>
                            <Premain-Class>org.example.DemoAgent</Premain-Class>
                            <Can-Redefine-Classes>true</Can-Redefine-Classes>
                            <Can-Retransform-Classes>true</Can-Retransform-Classes>
                        </manifestEntries>
                    </archive>
                </configuration>
            </plugin>
        </plugins>
    </build>

需要指定 Premain-Class 来指定入口方法所在的类
然后用 mvn package 进行打包
打包后在我们的目标代码的启动vm参数里加上

1	-javaagent:/{path}/demo-agent-1.0-SNAPSHOT.jar

{path} 需要替换成agent的jar包所在路径
然后执行目标main方法

就可以看到结果了，这样我们第一个agent的demo就跑起来了

grep命令小技巧-统计行数

发表于 2024-12-15 分类于技巧阅读次数： Disqus：

之前在使用grep来匹配内容，想看下匹配的结果数量都是用管道，然后用wc命令的
最近看了下似乎没必要这么麻烦
grep命令自带了这个功能
原来在用grep的时候还有个额外的功能就是看匹配行的前后几行，
使用

grep -B 4

是匹配行的前4行，可以用before来记

grep -A 4

是匹配行的后4行，可以用after来记
然后

grep -C 4

是看匹配行的前后4行
而这次的主角是小写的c

1	grep -c 'something' filename.txt

类似于这个效果，这又是个可以少打几个字符的偷懒技巧

tail命令小技巧

发表于 2024-12-08 分类于技巧阅读次数： Disqus：

我们日常在服务器上查看日志的时候用的很多的就是tail命令，使用tail最基本的就是

1	tail filename.log

这种，可以看到日志文件的最后10行
我们一般会使用

1	tail -f filename.log

可以用 -f来看到后续新生成的日志，可以用来观察是否有新增的异常日志或者是否有没有按预期打印新的日志
然后对于我们对输出的日志行数有具体要求的

1	tail -f -n 100 filename.log

可以用 -n 参数，可以指定显示行数，比如想要显示100行
而有一个技巧就是之前看到别的同学用过

1	tail -100f filename.log

可以和上面产生一样的效果
这样可以少输几个字符，也算是一个偷懒小技巧了，如果对于经常需要登录服务器看日志的也算是个效率小窍门
只是这样子对于比如我只想看新增的特定内容的日志来说，好像还不行
这里我们就可以使用组合技，结合 tail 和 grep 命令

1	tail -f filename.log \| grep something

这样就可以关注 something 的新增了

grep小技巧之匹配到二进制文件

发表于 2024-11-24 分类于技巧阅读次数： Disqus：

我们在日常使用grep这个强大的命令行工具时有时候会碰到一个问题，就是grep在识别文件的时候在碰到 ‘\000 NUL’的时候会认为文件是二进制文件，就不进行识别了，碰到这种情况我们可以使用 grep -a 或者 grep --text
比如我们做个小实验

<?php
$content = file_get_contents('input.txt');
$binary = "\0" . $content;  // 添加 NUL 字符到开头
file_put_contents('output.bin', $binary);

然后我们就有了 output.bin 这个二进制文件
本身 input.txt 的内容也很简单

1 2	12345 12345

我们尝试用 grep 查找其中的内容，

就会出现这个问题
当我们加上了 -a 或者 --text 就是指定 grep 需要在包含二进制文件格式中或者将文件当做是文本文件来识别匹配