小白也能看懂的插件化DroidPlugin原理(三)-- 如何拦截startActivity方法

时间:2022-10-17 16:02:36

  前言:在前两篇文章中分别介绍了动态代理、反射机制和Hook机制,如果对这些还不太了解的童鞋建议先去参考一下前两篇文章。经过了前面两篇文章的铺垫,终于可以玩点真刀实弹的了,本篇将会通过 Hook 掉 startActivity 方法的一个小例子来介绍如何找出合适的 Hook 切入点。 开始之前我们需要知道的一点就是,其实在 Android 里面启动一个 Activity 可以通过两种方式实现,一种是我们常用的调用 Activity.startActivity 方法,一种是调用 Context.startActivity 方法,两种方法相比之下, 第一种启动Activity的方式更为简单,所以先以第一种为例。

  本系列文章的代码已经上传至github,下载地址:https://github.com/lgliuwei/DroidPluginStudy 本篇文章对应的代码在 com.liuwei.proxy_hook.hook.activityhook 包内,下载下来对照代码看文章效果会更好!

一、Hook 掉 Activity 的 startActivity 的方法

  在 Hook Activity 的 startActivity 方法之前,我们首先明确一下我们的目标,我们先通过追踪源码找出 startActivity 调用的真正起作用的方法,然后想办法把目标方法拦截掉,并输出我们的一条 Log 信息。

  我们先来一步步分析 startActivity 的源码,随手写一个 startActivity 的示例,按住 command 键( windows 下按住 control )用鼠标点击 startActivity的方法即可跳转到方法里面。

  startActivity(Intent intent) 源码如下:

 public void startActivity(Intent intent) {
this.startActivity(intent, null);
}

  接着看 this.startActivity(intent, null) 方法源码:

 public void startActivity(Intent intent, @Nullable Bundle options) {
if (options != null) {
startActivityForResult(intent, -1, options);
} else {
// Note we want to go through this call for compatibility with
// applications that may have overridden the method.
startActivityForResult(intent, -1);
}
}

  从上一步传入的参数 options 为 null 我们就可以知道这一步调用了 startActivityForResult(intent, -1) 的代码。

  startActivityForResult(Intent intent, int requestCode) 源码如下:

 public void startActivityForResult(@RequiresPermission Intent intent, int requestCode) {
startActivityForResult(intent, requestCode, null);
}

  startActivityForResult(Intent intent, int requestCode, Bundle options) 源码如下:

 public void startActivityForResult(@RequiresPermission Intent intent, int requestCode,
@Nullable Bundle options) {
if (mParent == null) {
options = transferSpringboardActivityOptions(options);
Instrumentation.ActivityResult ar =
6 mInstrumentation.execStartActivity(
7 this, mMainThread.getApplicationThread(), mToken, this,
8 intent, requestCode, options);
if (ar != null) {
mMainThread.sendActivityResult(
mToken, mEmbeddedID, requestCode, ar.getResultCode(),
ar.getResultData());
}
if (requestCode >= 0) {
// If this start is requesting a result, we can avoid making
// the activity visible until the result is received. Setting
// this code during onCreate(Bundle savedInstanceState) or onResume() will keep the
// activity hidden during this time, to avoid flickering.
// This can only be done when a result is requested because
// that guarantees we will get information back when the
// activity is finished, no matter what happens to it.
mStartedActivity = true;
} cancelInputsAndStartExitTransition(options);
// TODO Consider clearing/flushing other event sources and events for child windows.
} else {
if (options != null) {
mParent.startActivityFromChild(this, intent, requestCode, options);
} else {
// Note we want to go through this method for compatibility with
// existing applications that may have overridden it.
mParent.startActivityFromChild(this, intent, requestCode);
}
}
}

  到这一步我们已经看到了关键点,注意上面代码块中红色的代码,其实 startActivity 真正调用的是 mInstrumentation.execStartActivity(...) 方法,mInstrumentation 是 Activity 的一个私有变量。接下来的任务将变得非常简单,回忆一下上一篇博文《小白也能看懂插件化DroidPlugin原理(二)-- 反射机制和Hook入门》中的方案一,在替换汽车引擎时我们继承原来的汽车引擎类创建了一个新类,然后在新引擎类中拦截了最大速度的方法,这里的思路是一样的,我们直接新建一个继承 Instrumentation 的新类,然后重写 execStartActivity() 。对此有不明白的童鞋建议再看一遍上一篇博文《小白也能看懂插件化DroidPlugin原理(二)-- 反射机制和Hook入门》。代码如下:

 public class EvilInstrumentation extends Instrumentation {
private Instrumentation instrumentation;
public EvilInstrumentation(Instrumentation instrumentation) {
this.instrumentation = instrumentation;
}
public ActivityResult execStartActivity(
Context who, IBinder contextThread, IBinder token, Activity target,
Intent intent, int requestCode, Bundle options) {
Logger.i(EvilInstrumentation.class, "请注意! startActivity已经被hook了!");
try {
Method execStartActivity = Instrumentation.class.getDeclaredMethod("execStartActivity", Context.class,
IBinder.class, IBinder.class, Activity.class,
Intent.class, int.class, Bundle.class);
return (ActivityResult)execStartActivity.invoke(instrumentation, who, contextThread, token, target,
intent, requestCode, options);
} catch (Exception e) {
e.printStackTrace();
} return null;
}
}

  重写工作已经做完了,接着我们通过反射机制用新建的 EvilInstrumentation 替换掉 Activity 的 mInstrumentation 变量,具体代码如下:

 public static void doActivityStartHook(Activity activity){
try {
Field mInstrumentationField = Activity.class.getDeclaredField("mInstrumentation");
mInstrumentationField.setAccessible(true);
Instrumentation originalInstrumentation = (Instrumentation)mInstrumentationField.get(activity);
mInstrumentationField.set(activity, new EvilInstrumentation(originalInstrumentation));
} catch (Exception e) {
e.printStackTrace();
}
}

  这对于我们来说已经很是轻车熟路了,很快就写完了,然后我们在 Activity 的 onCreate() 方法中需要调用一下 doActivityStartHook 即可完成对 Activity.startActivity 的 hook。MainActivity 的代码如下:

 public class MainActivity extends Activity {
private Button btn_start_by_activity;
@Override
protected void onCreate(Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
setContentView(R.layout.activity_main);
// hook Activity.startActivity()的方法时不知道这行代码为什么放在attachBaseContext里面不行?
// 调试发现,被hook的Instrumentation后来又会被替换掉原来的。
ActivityThreadHookHelper.doActivityStartHook(this);
btn_start_by_activity = (Button) findViewById(R.id.btn_start_by_activity);
btn_start_by_activity.setOnClickListener(new View.OnClickListener() {
@Override
public void onClick(View v) {
Intent intent = new Intent(MainActivity.this, OtherActivity.class);
startActivity(intent);
}
});
}
}

  程序运行之后,点击启动 Activity 的按钮将输出以下 Log:

   [EvilInstrumentation] : 请注意! startActivity已经被hook了!

  到此为止我们已经 hook 了 Activity 的 startActivity 方法,非常简单,代码量也很少,但我们也很轻易的发现这种方法需要在每一个 Activity 的 onCreate 方法里面调用一次 doActivityStartHook 方法,显然这不是一个好的方案,所以我们在寻找 hook 点时一定要注意尽量找一些在进程中保持不变或不容易被改变的变量,就像单例和静态变量。

  问题1:在这里有一点值得一提,我们将 doActivityStartHook(...) 方法的调用如果放到  MainActivity 的 attachBaseContext(...) 方法中替换工作将不会生效,为什么?

  调试发现,我们在 attachBaseContext(..) 里面执行完毕 doActivityStartHook(...) 方法后确实将 Activity 的 mInstrumentation 变量换成了我们自己的 EvilInstrumentation,但程序执行到 onCreate() 方法后就会发现这时候 mInstrumentation 变成了系统自己的 Instrumentation 对象了。这时候我们可以确信的是 mInstrumentation 变量一定是在 attachBaseContext() 之后被初始化或者赋值的。带着这个目标我们很轻松就在 Activity 源码的 attach() 方法中找到如下代码:

  Activity.attach() 的源码如下(注意第8行和第26行):

   final void attach(Context context, ActivityThread aThread,
Instrumentation instr, IBinder token, int ident,
Application application, Intent intent, ActivityInfo info,
CharSequence title, Activity parent, String id,
NonConfigurationInstances lastNonConfigurationInstances,
Configuration config, String referrer, IVoiceInteractor voiceInteractor,
Window window) {
8 attachBaseContext(context); mFragments.attachHost(null /*parent*/); mWindow = new PhoneWindow(this, window);
mWindow.setWindowControllerCallback(this);
mWindow.setCallback(this);
mWindow.setOnWindowDismissedCallback(this);
mWindow.getLayoutInflater().setPrivateFactory(this);
if (info.softInputMode != WindowManager.LayoutParams.SOFT_INPUT_STATE_UNSPECIFIED) {
mWindow.setSoftInputMode(info.softInputMode);
}
if (info.uiOptions != 0) {
mWindow.setUiOptions(info.uiOptions);
}
mUiThread = Thread.currentThread(); mMainThread = aThread;
26 mInstrumentation = instr;
mToken = token;
mIdent = ident;
mApplication = application;
mIntent = intent;
mReferrer = referrer;
mComponent = intent.getComponent();
mActivityInfo = info;
mTitle = title;
mParent = parent;
mEmbeddedID = id;
mLastNonConfigurationInstances = lastNonConfigurationInstances;
if (voiceInteractor != null) {
if (lastNonConfigurationInstances != null) {
mVoiceInteractor = lastNonConfigurationInstances.voiceInteractor;
} else {
mVoiceInteractor = new VoiceInteractor(voiceInteractor, this, this,
Looper.myLooper());
}
} mWindow.setWindowManager(
(WindowManager)context.getSystemService(Context.WINDOW_SERVICE),
mToken, mComponent.flattenToString(),
(info.flags & ActivityInfo.FLAG_HARDWARE_ACCELERATED) != 0);
if (mParent != null) {
mWindow.setContainer(mParent.getWindow());
}
mWindowManager = mWindow.getWindowManager();
mCurrentConfig = config;
}

  至此,问题1算是找到了答案。

二、Hook 掉 Context 的 startActivity 的方法

  文章开头我们就说 Android 中有个两种启动 Activity 的方式,一种是 Activity.startActivity 另一种是 Context.startActivity,但需要注意的时,我们在使用 Context.startActivity 启动一个 Activity 的时候将 flags 指定为 FLAG_ACTIVITY_NEW_TASK。

  在接下来的分析中需要查看 Android 源码,先推荐两个查看 Android 源码的网站:

  http://androidxref.com

  http://grepcode.com/project/repository.grepcode.com/java/ext/com.google.android/android/

  我们试着 hook 掉 Context.startActivity 方法,我们依然随手写一个 Context 方式启动 Activity 的示例,如下:

 Intent intent = new Intent(MainActivity.this, OtherActivity.class);
intent.setFlags(Intent.FLAG_ACTIVITY_NEW_TASK);
getApplicationContext().startActivity(intent);

  照着(一)中的姿势点入 startActivity() 方法里面,由于 Context 是一个抽象类,所以我们需要找到它的实现类才能看到具体的代码,通过查看 Android 源码我们可以在 ActivityTread 中可知 Context 的实现类是 ContextImpl。(在这里大家先知道这一点就行,具体的调用细节将会在下一篇博文中详细介绍)

  源码地址:

  http://grepcode.com/file/repository.grepcode.com/java/ext/com.google.android/android/5.1.1_r1/android/app/ActivityThread.java#2338

 private Activity performLaunchActivity(ActivityClientRecord r, Intent customIntent) {
...
if (activity != null) {
4 Context appContext = createBaseContextForActivity(r, activity);
CharSequence title = r.activityInfo.loadLabel(appContext.getPackageManager());
Configuration config = new Configuration(mCompatConfiguration);
...
}
...
private Context createBaseContextForActivity(ActivityClientRecord r, final Activity activity) {
11 ContextImpl appContext = ContextImpl.createActivityContext(this, r.packageInfo, r.token);
12 appContext.setOuterContext(activity);
13 Context baseContext = appContext;
...
}

   现在我们来查看 ContextImpl.startActivity() 的源码。

  源码地址:

  http://grepcode.com/file/repository.grepcode.com/java/ext/com.google.android/android/5.1.1_r1/android/app/ContextImpl.java#ContextImpl.startActivity%28android.content.Intent%29

 @Override
public void startActivity(Intent intent) {
warnIfCallingFromSystemProcess();
startActivity(intent, null);
}

  再进入 startActivity(intent, null) 查看源码如下:

 @Override
public void startActivity(Intent intent, Bundle options) {
warnIfCallingFromSystemProcess();
if ((intent.getFlags()&Intent.FLAG_ACTIVITY_NEW_TASK) == 0) {
throw new AndroidRuntimeException(
"Calling startActivity() from outside of an Activity "
+ " context requires the FLAG_ACTIVITY_NEW_TASK flag."
+ " Is this really what you want?");
}
10 mMainThread.getInstrumentation().execStartActivity(
11 getOuterContext(), mMainThread.getApplicationThread(), null,
12 (Activity)null, intent, -1, options);
}

  由上面第四行代码可以看出在代码中判断了 intent 的 flag 类型,如果非 FLAG_ACTIVITY_NEW_TASK 类型就会抛出异常。接着看红色部分的关键代码,可以看出先从 ActivityTread 中获取到了 Instrumentation 最后还是调用了 Instrumentation 的 execStartActivity(...) 方法,我们现在需要做的就是分析 ActivityTread 类,并想办法用我们自己写的 EvilInstrumentation 类将 ActivityTread 的 mInstrumentation 替换掉。

  源码地址:

  http://grepcode.com/file/repository.grepcode.com/java/ext/com.google.android/android/5.1.1_r1/android/app/ActivityThread.java#ActivityThread.0sCurrentActivityThread

  ActivityTread 部分代码如下:

206     private static ActivityThread sCurrentActivityThread;
207 Instrumentation mInstrumentation;
...
1597 public static ActivityThread currentActivityThread() {
1598 return sCurrentActivityThread;
1599 }
...
1797 public Instrumentation getInstrumentation()
1798 {
1799 return mInstrumentation;
1800 }

  这里需要告诉大家是,ActivityTread 即代表应用的主线程,而一个应用中只有一个主线程,并且由源码可知,ActivityTreadd 的对象又是以静态变量的形式存在的,太好了,这正是我们要找的 Hook 点。废话不多说了,现在我们只需利用反射通过 currentActivityThread() 方法拿到 ActivityThread 的对象,然后在将 mInstrumentation 替换成 EvilInstrumentation 即可,代码如下:

 1   public static void doContextStartHook(){
try {
Class<?> activityThreadClass = Class.forName("android.app.ActivityThread");
Method currentActivityThreadMethod = activityThreadClass.getDeclaredMethod("currentActivityThread");
Object activityThread = currentActivityThreadMethod.invoke(null); Field mInstrumentationField = activityThreadClass.getDeclaredField("mInstrumentation");
mInstrumentationField.setAccessible(true);
Instrumentation originalInstrumentation = (Instrumentation)mInstrumentationField.get(activityThread);
mInstrumentationField.set(activityThread, new EvilInstrumentation(originalInstrumentation));
} catch (Exception e) {
e.printStackTrace();
}
}

  其实代码也不难理解,跟 Hook Activity 的 startActivity() 方法是一个思路,只是 Hook 的点不同而已。下面我们在 MainActivity 的 attachBaseContext() 方法中调用 doContextStartHook() 方法,并添加相关测试代码,具体代码如下:

 public class MainActivity extends Activity {
private Button btn_start_by_context;
@Override
protected void onCreate(Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
setContentView(R.layout.activity_main);
btn_start_by_context = (Button) findViewById(R.id.btn_start_by_context);
btn_start_by_context.setOnClickListener(new View.OnClickListener() {
@Override
public void onClick(View v) {
Intent intent = new Intent(MainActivity.this, OtherActivity.class);
intent.setFlags(Intent.FLAG_ACTIVITY_NEW_TASK);
getApplicationContext().startActivity(intent);
}
});
}
@Override
protected void attachBaseContext(Context newBase) {
super.attachBaseContext(newBase);
ActivityThreadHookHelper.doContextStartHook();
}
}

  点击按钮后查看 Log 输出如下:

   [EvilInstrumentation] : 请注意! startActivity已经被hook了!

  看到这样的 Log,说明我们已经成功的 Hook 了 Context.startActivity()。而且 doContextStartHook() 方法只在程序开始的时候调用一次即可,后面在程序其他的 Activity 中调用 Context.startActivity() 时此拦截工作均可生效,这是因为 Context.startActivity() 在执行启动 Activity 的操作时调是通过 ActivityTread 获取到 Instrumentation,然后再调用 Instrumentation.execStartActivity() 方法,而 ActivityTread 在程序中是以单例的形式存在的,这就是原因。所以说调用 doContextStartHook() 方法最好的时机应该是放在 Application 中。

  注意!前方惊现彩蛋一枚!!

  将 doContextStartHook() 方法放入到了 MyApplication 的 attachBaseContext() 里面后,代码如下:

 public class MyApplication extends Application {
@Override
protected void attachBaseContext(Context base) {
super.attachBaseContext(base);
5 ActivityThreadHookHelper.doContextStartHook();
}

  MainActivity 的代码如下:

 public class MainActivity extends Activity {
private final static String TAG = MainActivity.class.getSimpleName();
private Button btn_start_by_activity;
private Button btn_start_by_context;
@Override
protected void onCreate(Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
setContentView(R.layout.activity_main);
btn_start_by_activity = (Button) findViewById(R.id.btn_start_by_activity);
btn_start_by_context = (Button) findViewById(R.id.btn_start_by_context);
11 ActivityThreadHookHelper.doActivityStartHook(this);
btn_start_by_activity.setOnClickListener(new View.OnClickListener() {
@Override
public void onClick(View v) {
Log.i(TAG, "onClick: Activity.startActivity()");
Intent intent = new Intent(MainActivity.this, OtherActivity.class);
startActivity(intent);
}
}); btn_start_by_context.setOnClickListener(new View.OnClickListener() {
@Override
public void onClick(View v) {
Log.i(TAG, "onClick: Context.startActivity()");
Intent intent = new Intent(MainActivity.this, OtherActivity.class);
intent.setFlags(Intent.FLAG_ACTIVITY_NEW_TASK);
getApplicationContext().startActivity(intent);
}
});
}
}

  代码如上,布局文件很简单就不贴出来了,就是两个按钮,一个测试 Activity.startActivity() 方法,一个测试 Context.startActivity() 方法,然后在 MainActivity 的 onCreate() 中调用了 doActivityStartHook() 在 MyApplication 里面调用了 doContextStartHook(), 目前看来代码很正常,符合我们上面的思路,但楼主在点击按钮发现 Log 输出如下:

小白也能看懂的插件化DroidPlugin原理(三)-- 如何拦截startActivity方法

  是的,Activity.startActivity 被 hook 的信息输出了两次!为什么?

  我们不妨先猜想一下,一定是 Activity 的 mInstrumentation 对象在我们替换之前就已经变成了 EvilInstrumentation, 然后我们又在 Activity.onCreate 方法调用了一次 doActivityStartHook(), 相当于我们又用 EvilInstrumentation 又重写了 EvilInstrumentation 的 startActivity() 方法,所以导致 log 信息输出了两次。

  那问题又来了,为什么 Activity 的 mInstrumentation 对象在我们替换之前就已经变成了 EvilInstrumentation?

  纵观代码,只有一个地方有疑点,那就是我们放到 MyApplication.attachBaseContext() 方法里面的 doContextStartHook() 起的作用!

  还是先直接简单说一下事实的真相吧,结合上文所说,一个应用内只存在一个 ActivityTread 对象,也只存在一个 Instrumentation 对象,这个 Instrumentation 是 ActivityTread 的成员变量,并在 ActivityTread 内完成初始化,在启动一个 Activity 的流程中大概在最后的位置 ActivityTread 会回调 Activity 的 attach() 方法,并将自己的 Instrumentation 对象传给 Activity。启动 Activity 的详细流程及调用细节将会在下一篇博文介绍,敬请期待!

三、小结

  本篇文章通过拦截 Context.startActivity() 和 Activity.startActivity() 两个方法,将上一篇文章中介绍的 Hook 技术实践 Activity 的启动流程之中,同时通过这两个小例子初步了解了 Android 源码以及怎么样去选定一个合适的 Hook 点。想要了解插件化的基本原理,熟悉 Activity 的启动流程是必不可少的,下一篇文章将会详细介绍 Activity 的启动流程,感兴趣的同学可以关注一下!

参考文章

  《Android插件化原理解析——Hook机制之动态代理》

  《Android应用程序的Activity启动过程简要介绍和学习计划》

本文链接:http://www.cnblogs.com/codingblock/p/6666239.html