自定义 Hook，解决 RxJava 异常时堆栈信息显示不全

Posted 2021-12-31 涂程

作者：程序员徐公
转载地址：https://juejin.cn/post/7038789109040545823

前阵子，组内的同事反馈说 RxJava 在 debug 包 crash 了，捕获到的异常信息不全。（即我们捕获到的堆栈没有包含我们自己代码，都是一些系统或者 RxJava 框架的代码）

典型的一些 error 信息如下：

 io.reactivex.exceptions.OnErrorNotImplementedException: The exception was not handled due to missing onError handler in the subscribe() method call. Further reading: https://github.com/ReactiveX/RxJava/wiki/Error-Handling | java.lang.NullPointerException: Callable returned null
     at io.reactivex.internal.functions.Functions$OnErrorMissingConsumer.accept(Functions.java:704)
     at io.reactivex.internal.functions.Functions$OnErrorMissingConsumer.accept(Functions.java:701)
     at io.reactivex.internal.observers.LambdaObserver.onError(LambdaObserver.java:77)
     at io.reactivex.internal.operators.observable.ObservableObserveOn$ObserveOnObserver.checkTerminated(ObservableObserveOn.java:281)
     at io.reactivex.internal.operators.observable.ObservableObserveOn$ObserveOnObserver.drainNormal(ObservableObserveOn.java:172)
     at io.reactivex.internal.operators.observable.ObservableObserveOn$ObserveOnObserver.run(ObservableObserveOn.java:255)
     at io.reactivex.android.schedulers.HandlerScheduler$ScheduledRunnable.run(HandlerScheduler.java:124)
     at android.os.Handler.handleCallback(Handler.java:883)
     at android.os.Handler.dispatchMessage(Handler.java:100)
     at android.os.Looper.loop(Looper.java:214)
     at android.app.ActivityThread.main(ActivityThread.java:7682)
     at java.lang.reflect.Method.invoke(Native Method)
     at com.android.internal.os.RuntimeInit$MethodAndArgsCaller.run(RuntimeInit.java:516)
     at com.android.internal.os.ZygoteInit.main(ZygoteInit.java:950)
 Caused by: java.lang.NullPointerException: Callable returned null
     at io.reactivex.internal.functions.ObjectHelper.requireNonNull(ObjectHelper.java:39)
     at io.reactivex.internal.operators.observable.ObservableFromCallable.subscribeActual(ObservableFromCallable.java:43)
     at io.reactivex.Observable.subscribe(Observable.java:12267)
     at io.reactivex.internal.operators.observable.ObservableSubscribeOn$SubscribeTask.run(ObservableSubscribeOn.java:96)
     at io.reactivex.Scheduler$DisposeTask.run(Scheduler.java:578)
     at io.reactivex.internal.schedulers.ScheduledRunnable.run(ScheduledRunnable.java:66)
     at io.reactivex.internal.schedulers.ScheduledRunnable.call(ScheduledRunnable.java:57)
     at java.util.concurrent.FutureTask.run(FutureTask.java:266)
     at java.util.concurrent.ScheduledThreadPoolExecutor$ScheduledFutureTask.run(ScheduledThreadPoolExecutor.java:301)
     at java.util.concurrent.ThreadPoolExecutor.runWorker(ThreadPoolExecutor.java:1167)
     at java.util.concurrent.ThreadPoolExecutor$Worker.run(ThreadPoolExecutor.java:641)
     at java.lang.Thread.run(Thread.java:919)

可以看到，上面的 Error 堆栈信息中，它并没有给出这个 Error 在实际项目中的调用路径。可以看到，报错的堆栈，提供的有效信息较少， 我们只能知道是由于 callable.call() 这里返回了 Null，导致出错。却不能判断 callable 是哪里创建的，这时候我们只能结合日志上下文，判断当前之前的代码大概在哪里，再逐步排查。

public final class ObservableFromCallable<T> extends Observable<T> implements Callable<T> 

    @Override
    public void subscribeActual(Observer<? super T> observer) 
        DeferredScalarDisposable<T> d = new DeferredScalarDisposable<T>(observer);
        observer.onSubscribe(d);
        if (d.isDisposed()) 
            return;
        
        T value;
        try 
            // callable.call()  这里返回了 Null，并传递给了 RxJavaPlugins 的 errorHandler
            value = ObjectHelper.requireNonNull(callable.call(), "Callable returned null");
         catch (Throwable e) 
            Exceptions.throwIfFatal(e);
            if (!d.isDisposed()) 
                observer.onError(e);
             else 
                RxJavaPlugins.onError(e);
            
            return;
        
        d.complete(value);
    

一顿操作猛如虎，很多，我们结合一些让下文日志，发现是这里返回了 null，导致出错

backgroundTask(Callable<Any> 
    Log.i(TAG, "btn_rx_task: ")
    Thread.sleep(30)
    return@Callable null
)?.subscribe()

---

/**
 * 创建一个rx的子线程任务Observable
 */
private fun <T> backgroundTask(callable: Callable<T>?): Observable<T>? 
    return Observable.fromCallable(callable)
            .compose(IOMain())

如果遇到 callable 比较多的情况下，这时候 一个个排查 callable，估计搞到你吐血。

那有没有什么较好的方法，比如做一些监控？完整打印堆栈信息。

第一种方案，自定义 Hook 解决

首先，我们先来想一下，什么是堆栈？

在我的理解里面，堆栈是用来储存我们程序当前执行的信息。在 Java 当中，我们通过 java.lang.Thread#getStackTrace 可以拿到当前线程的堆栈信息，注意是当前线程的堆栈。

而 RxJava 抛出异常的地方，是在执行 Callable#call 方法中，它打印的自然是 Callable#call 的方法调用栈，而如果 Callable#call 的调用线程跟 callable 的创建线程不一致，那肯定拿不到 创建 callable 时候的堆栈。

而我们实际上需要知道的是 callable 创建的地方，对应到我们我们项目报错的地方，那自然是 Observable.fromCallable 方法的调用栈。

这时候，我们可以采用 Hook 的方式，来 Hook 我们的代码

为了方便，我们这里采用了 wenshu 大神的 Hook 框架, github, 想自己手动去 Hook 的，可以看一下我两年前写的文章 Android Hook 机制之简单实战，里面有介绍介绍一些常用的 Hook 手段。

很快，我们写出了如下代码，对 Observable#fromCallable 方法进行 hook

    fun hookRxFromCallable() 
//        DexposedBridge.findAndHookMethod(ObservableFromCallable::class.java, "subscribeActual", Observer::class.java, RxMethodHook())
        DexposedBridge.findAndHookMethod(
            Observable::class.java,
            "fromCallable",
            Callable::class.java,
            object : XC_MethodHook() 
                override fun beforeHookedMethod(param: MethodHookParam?) 
                    super.beforeHookedMethod(param)
                    val args = param?.args
                    args ?: return

                    val callable = args[0] as Callable<*>
                    args[0] = MyCallable(callable = callable)

                

                override fun afterHookedMethod(param: MethodHookParam?) 
                    super.afterHookedMethod(param)
                
            )
    

    class MyCallable(private val callable: Callable<*>) : Callable<Any> 

        private val TAG = "RxJavaHookActivity"
        val buildStackTrace: String?

        init 
            buildStackTrace = Rx2Utils.buildStackTrace()
        

        override fun call(): Any 
            Log.i(TAG, "call: ")
            val call = callable.call()
            if (call == null) 
                Log.e(TAG, "call should not return null: buildStackTrace is $buildStackTrace")
            
            return call
        

    

再次执行我们的代码

backgroundTask(Callable<Any> 
    Log.i(TAG, "btn_rx_task: ")
    Thread.sleep(30)
    return@Callable null
)?.subscribe()

可以看到，当我们的 Callable 返回为 empty 的时候，这时候报错的信息会含有我们项目的代码， perfect。

第二种方案，RxJavaExtensions

最近，在 Github 上面发现了这一个框架，它也可以帮助我们解决 RxJava 异常过程中信息不全的问题。它的基本使用如下：

使用

github.com/akarnokd/Rx…

第一步，引入依赖库

dependencies 
    implementation "com.github.akarnokd:rxjava2-extensions:0.20.10"

第二步：先启用错误追踪：

RxJavaAssemblyTracking.enable();

第三步：在抛出异常的异常，打印堆栈

    /**
     * 设置全局的 onErrorHandler。
     */
    fun setRxOnErrorHandler() 
        RxJavaPlugins.setErrorHandler  throwable: Throwable ->
            val assembled = RxJavaAssemblyException.find(throwable)
            if (assembled != null) 
                Log.e(TAG, assembled.stacktrace())
            
            throwable.printStackTrace()
            Log.e(TAG, "setRxOnErrorHandler: throwable is $throwable")
        
    

原理

RxJavaAssemblyTracking.enable();

public static void enable() 
    if (lock.compareAndSet(false, true)) 

        // 省略了若干方法

        RxJavaPlugins.setOnObservableAssembly(new Function<Observable, Observable>() 
            @Override
            public Observable apply(Observable f) throws Exception 
                if (f instanceof Callable) 
                    if (f instanceof ScalarCallable) 
                        return new ObservableOnAssemblyScalarCallable(f);
                    
                    return new ObservableOnAssemblyCallable(f);
                
                return new ObservableOnAssembly(f);
            
        );

        lock.set(false);
    

可以看到，它调用了 RxJavaPlugins.setOnObservableAssembly 方法，设置了 RxJavaPlugins onObservableAssembly 变量

而我们上面提到的 Observable#fromCallable 方法，它里面会调用 RxJavaPlugins.onAssembly 方法，当我们的 onObservableAssembly 不为 null 的时候，会调用 apply 方法进行转换。

public static <T> Observable<T> fromCallable(Callable<? extends T> supplier) 
    ObjectHelper.requireNonNull(supplier, "supplier is null");
    return RxJavaPlugins.onAssembly(new ObservableFromCallable<T>(supplier));

public static <T> Observable<T> onAssembly(@NonNull Observable<T> source) 
    Function<? super Observable, ? extends Observable> f = onObservableAssembly;
    if (f != null) 
        return apply(f, source);
    
    return source;

因此，即当我们设置了 RxJavaAssemblyTracking.enable()， Observable#fromCallable 传递进来的 supplier，最终会包裹一层，可能是 ObservableOnAssemblyScalarCallable，ObservableOnAssemblyCallable，ObservableOnAssembly。典型的装饰者模式应用，这里不得不说，RxJava 对外提供的这个点，设计得真巧妙，可以很方便我们做一些 hook。

我们就以 ObservableOnAssemblyCallable 看一下

final class ObservableOnAssemblyCallable<T> extends Observable<T> implements Callable<T> 

    final ObservableSource<T> source;

    // 将在哪里创建的 Callable 的堆栈信息保存下来
    final RxJavaAssemblyException assembled;

    ObservableOnAssemblyCallable(ObservableSource<T> source) 
        this.source = source;
        this.assembled = new RxJavaAssemblyException();
    

    @Override
    protected void subscribeActual(Observer<? super T> observer) 
        source.subscribe(new OnAssemblyObserver<T>(observer, assembled));
    

    @SuppressWarnings("unchecked")
    @Override
    public T call() throws Exception 
        try 
            return ((Callable<T>)source).call();
         catch (Exception ex) 
            Exceptions.throwIfFatal(ex);
            throw (Exception)assembled.appendLast(ex);
        
    


public final class RxJavaAssemblyException extends RuntimeException 

    private static final long serialVersionUID = -6757520270386306081L;

    final String stacktrace;

    public RxJavaAssemblyException() 
        this.stacktrace = buildStackTrace();
    
 

可以看到，他是直接在 ObservableOnAssemblyCallable 的构造方法的时候，直接将 Callable 的堆栈信息保存下来，类为 RxJavaAssemblyException。

而当 error 报错的时候，调用 RxJavaAssemblyException.find(throwable) 方式，判断是不是 RxJavaAssemblyException，是的话，直接返回。

public static RxJavaAssemblyException find(Throwable ex) 
    Set<Throwable> memory = new HashSet<Throwable>();
    while (ex != null) 
        if (ex instanceof RxJavaAssemblyException) 
            return (RxJavaAssemblyException)ex;
        

        if (memory.add(ex)) 
            ex = ex.getCause();
         else 
            return null;
        
    
    return null;

到这里，RxJavaAssemblyTracking 能将 error 信息完整打印出来的流程已经讲明白了，其实就是在创建 Callable 的时候，采用一个包装类，在构造函数的时候，将 error 信息报错下来，等到出错的时候，再将 error 信息，替换成保存下来的 error信息。

我们的自定义 Hook 也是利用这种思路，提前将 callable 创建的堆栈暴露下来，换汤不换药。

一些思考

上述的方案我们一般不会带到线上，为什么呢？ 因为对于每一个 callable，我们需要提前保存堆栈，而获取堆栈是耗时的。那有没有什么方法呢？

如果项目有接入 Matrix 的话，可以考虑借用 Matrix trace 的思想，因为在方法前后插入 AppMethodBeat#i 和 AppMethodBeat#o 这样当我们执行方法的时候，因为插桩了，我们可以方便得获取到方法执行耗时，以及方法的调用栈。

// 第一步：需要在合适的实际先生成 beginRecord
AppMethodBeat.IndexRecord  beginRecord = AppMethodBeat.getInstance().maskIndex("AnrTracer#dispatchBegin");
// 第二步：方法的调用栈信息在 data 里面
long[] data = AppMethodBeat.getInstance().copyData(beginRecord);
第三步：
将 data 转化为我们想要的 stack（初步看了代码，需要我们修改 trace 的代码）