Android 应用流量监控实践

背景

得物Apm系统本身包含网络接口性能监控的能力，但接口监控主要关注的是接口的耗时、异常率等信息，没有流量消耗相关维度的统计信息，并且一部分流量消耗可能来自于音视频等其他特殊场景，在接口监控的盲区外。

为了了解用户目前使用App时的流量消耗情况，并支持分析用户是存在异常的流量使用情况，还需要从流量消耗的角度，进行相关监控功能的建设。本文主要介绍流量监控实现过程中的一些技术实现及平台能力。

流量消耗信息获取

xt_qtaguid模块

xt_qtaguid 模块是android 内核3.0 引入的流量统计模块，在Android系统9以下的版本，可以通过直接读取 /proc/net/xt_qtaguid/stats 文件获取对应应用的流量消耗信息。其内容如下

bash 复制代码

nimdanoob:/ $ cat proc/net/xt_qtaguid/stats | grep 10027
idx iface acct_tag_hex uid_tag_int cnt_set rx_bytes rx_packets tx_bytes tx_packets rx_tcp_bytes rx_tcp_packets rx_udp_bytes rx_udp_packets rx_other_bytes rx_other_packets tx_tcp_bytes tx_tcp_packets tx_udp_bytes tx_udp_packets tx_other_bytes tx_other_packets
10 rmnet_data0 0x0 10027 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
11 rmnet_data0 0x0 10027 1 451662 2320 1193948 3888 451662 2320 0 0 0 0 1193948 3888 0 0 0 0
40 wlan0 0x0 10027 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
41 wlan0 0x0 10027 1 38829 166 77083 208 38829 166 0 0 0 0 77083 208 0 0 0 0

在结果中，根据 uid_tag_int 可以区分不同的应用程序，这个值对应代码中 android.os.Process.myUid()�获取的值。

第二列的 iface 表示所使用的网络接口，比如 wlan 表示Wi-Fi网络适配器的物理接口， rmnet_data 表示手机数据网络接口， l0表示本地回环接口。

通过读取 rx_bytes、tx_bytes 就可以获取不同网络接口累计使用的上下行流量大小。

除了通过直接读取 xt_qtaguid/stats文件的方式，也可以直接通过调用系统封装TrafficStats类提供的相关Api获取流量消耗信息，该类在 Android API8 版本开始就被引入了。

TrafficStats的限制

在Android 9 版本开始，google 逐步取消了对 xt_qtaguid模块的支持，开始采用基于 eBPF的网络流量监控实现，所以在高版本设备中无法再通过读取xt_qtaguid文件获取流量使用信息。而直接使用TrafficStats读取流量消耗信息又由于其提供的Api有限，存在一些限制，比如TrafficStats 虽然支持查看指定程序的整体流量消耗信息，但不支持根据iface端口进行区分，因而无法明确wifi和移动分别消耗了多少流量。

在一些开源项目的实现中通过ConnectivityManager 获取当前网络连接状态，判断采样间隔内流量消耗来自于wifi还是移动。但是实际上在采样间隔内，设备可能同时在使用wifi和移动进行进行网络通信，因此这种方式采集的数据精准度有限。

另外采用基于TrafficStats Api的方案在接入得物后，还发现TrafficStats统计的流量消耗比实际的流量消耗大了非常多（实际流量消耗的5倍），最后定位发现是由于得物的音视频场景使用了本地Socket提供媒体数据,而TrafficStats.getUidTxBytes接口包含了所有iface的数据，本地Socket的通信数据也会被包含在内。

NetworkStatsManager

在Android6.0 (API 23)版中，系统新增了NetworkStatsManager类查询网络历史使用信息，其提供的Api 能力相比TrafficStats 强大很多，比如其可以获取指定时间内的流量消耗，并且通过指定networkType 可以区分流量消耗的类型。

java 复制代码

NetworkStatsManager statsManager = (NetworkStatsManager) getSystemService(Context.NETWORK_STATS_SERVICE);
//查询自身进程 蜂窝网络消耗
statsManager.querySummary(ConnectivityManager.TYPE_MOBILE, "", startTime, endTime);
//查询整机 蜂窝网络消耗
statsManager.querySummaryForDevice(ConnectivityManager.TYPE_MOBILE, null, startTime, endTime);

然而，在实际的测试过程中，以10s为采样间隔为例，经常会发现采样间隔内统计的数值未发生变化，在测试TrafficStats相关Api时发现没有类似的限制，其返回的数值是是实时的。

通过跟踪NetworkStatsService相关的实现后发现，原因可能是2个方面的问题导致的。

当应用层通过本地NetworkStatsManager调用相应Api获取消耗大小时，会创建一个NetworkStats对象，而NetworkStats对象构造时，会创建一个本地的 INetworkStatsSession对象

首先需要了解，NetworkStatsService 主要提供的是历史流量消耗记录的能力，它的计算方式是周期性的读取ebpf相应的虚拟文件中，从中获得最新的流量消耗大小。

NetworkStatsService为了尽量保证应用层获取的是最新的数据在每次openSession时，会尝试进行一次poll，不过这里的poll动作触发存在一定限制，15秒内只能触发一次。

另一个原因是，其底层实际负责流量记录的 NetworkStatsRecorder 会先将变化的流量信息统计在一个mPending对象中，当pending 记录超出一定阈值后，才会将这个数据落盘，而这个阈值默认情况下为2MB。

为了验证这个问题，保证每次采样时获取到的wifi及移动消耗数据是最新的，我们可以创建一组LocalSocket进行本地的端对端通信，在每次采样之前，进行一次数据数据通信，从而触发系统底层的数据合并操作，保证我们获取的数值是最新的。

当然这种方案会消耗不必要的系统资源，包括Socket资源的浪费并且会导致触发系统流量统计服务的数据合并及持久化操作，从节省系统资源的角度并不是一个好的方式，如果对数据采集结果的失效及准确性要求不高的话，不建议进行此类行为。

跟踪流量消耗明细

当发现用户流量消耗不合理时，我们还需要提供流量消耗的明细。比如接口的流量消耗、图片、音视频的流量消耗、直连Socket的流量消耗。

网络库Hook

大部分的网络请求发起依赖于现成的网络框架如 OkHttp、HttpURLConnection,因此需要Hook这些网络库追踪具体的流量消耗信息。

OkHttp

针对OkHttp库，OkHttp本身提供了EventListener机制，通过注册EventListner可以监听获取每个请求的阶段过程，以及请求响应信息。因此这里不做过多阐述。

HttpURLConnection

针对HttpURLConnection进行的网络请求，这里参考的是AndroidStudio中网络监控的方案，具体代码详见aosp,这里简单做下原理阐述。

首先需要字节码插桩代理 URLConnection的 openConnection函数，代理原始URL.openConnection返回的URLConnection对象。这里需要注意由于Https 和Http协议会返回不同的URLConnection类型实例，因此需要做下判断。

以HttpURLConnectionWrapper实现为例，具体跟踪代码

其内部会再创建一个专门负责流量统计的TrackHttpURLConnection实例，这里又包装了一层是因为该类的实现可以在HttpsURLConnection的代理中进行复用。继续跟踪TrackedHttpURLConnection� 实现。

针对请求体、响应体流量的跟踪，其主要实现是返回一个代理包装类型的InputStream或OutputStream

以OutputStream为例，继续代理内部的write函数，当write函数被调用时记录相应的流量消耗。

在整个对HttpURLConnection Hook实现的过程中有一些需要额外注意的点。

拿到URLConnection对象后，业务层会调用connect触发实际的网络请求调用，我们在connect函数中可以进行请求头和请求体流量的记录，但实际上业务层不主动调用connect函数，比如直接调用getInputStream()或getOutputStream时底层也会触发connect操作，但此时它直接触发的是被代理的原始对象的connect函数，而我们自身的connect函数是不会被触发的，因此如果只在connect函数中记录信息，就可能出现遗漏的情况。

因此AS在实现这块逻辑时，会包装一个trackPreConnect�()函数，并同时在这3处进行调度。

类似的直接调用获取响应内容的一些函数，比如获取响应头信息也会触发底层的conenct操作，因此其也封装了一个trackResponse函数，内部会调用conenct函数(如果判断没调用过的话)，并记录响应头的一些信息

基于Socket Api Hook

在部分业务场景中，可能会存在使用Socket进行网络通信的情况，比如自研长连协议，此时需要能够实现直接对原生Java Socket层数据通信的监控。

这里介绍2中针对Java Socket通信的监控方式。

基于字节码替换

和URLConnection监控的方案类似，Socket的数据读写是通过其InputStream和OutputStream对象实现的，因此通过ASM全局替换Socket的getInputStream和getOutPutStream对象，对其数据读写进行监控即可。

自定义SocketImplFactory

在Java层，创建Socket的方式是直接构造 java.net.Socket对象，我们先跟踪其构造函数的实现。

在构造函数中，会调用setImpl()函数，继续跟踪setImple()实现

setImpl()函数先判断 factory是否为空，如果不为空则使用factory创建SocketImpl实例，否则直接通过 SocksSocketImple()构造函数创建。观察整个 java.net.Socket类的代码可以发现，其有关Socket的实际操作其实都是转发到内部的SocksSocketImple对象中，以getInputStream()为例，在java.net.Socket实现中只是简单做了下状态的判断，最后直接调用impl的getInputStream函数。

java.net.Socket 类的factory是一个静态常量，SocketImpleFactory接口的实现如下

而 SocketImple抽象类实现如下

因此可以通过设置自定义的SocketImpleFactory 返回一个包装过的SokcetImpl实例，从而对Socket进行监控。

java 复制代码

//1.反射获取原始的Factory
SocketImplFactory oldFactory = ....
if (oldFacotry!=null){
    //反射修改 Socket.class 的facotry字段，设置为新的Factory
	ReflectionUtil.of(Socket.class).field("factory").set(null, new TrafficSocketImplFactory(oldFactory));
} else {
    //设置新的factory
	Socket.setSocketImplFactory(new TrafficSocketImplFactory(null));
}

java 复制代码

public class TrafficSocketImplFactory implements SocketImplFactory {
	public TrafficSocketImplFactory(SocketImplFactory originalFactory) {
        this.mOldFactory = originalFactory;
    }
     // ... 
     public SocketImpl createSocketImpl() {
        if (this.mOldFactory != null){
            return new TrafficSocketImpl(this.mOldFactory.createSocketImpl());
        }
        try {
            if (isCreateServerSocket())
                return (SocketImpl)sConstructor_SocketImpl.newInstance(new Object[0]);
            return new TrafficSocketImpl((SocketImpl)sConstructor_SocketImpl.newInstance(new Object[0]));
        } catch (Throwable tr) {
            //should never happend
            ApmSafety.handleException(tr);
            return null;
        }
    }
}