通过自定义Gradle插件修改编译后的class文件

总体思路:通过gradle的transform在class文件转DEX过程中,借助javassist技术修改class文件

 要使用transform和javassist需要添加下面两个依赖

dependencies {
    //gradle sdk
    //groovy sdk
    compile 'com.android.tools.build:gradle:2.1.0'
    compile 'org.javassist:javassist:3.20.0-GA'
}

 

我的简书同步发布:通过自定义Gradle插件修改编译后的class文件

转载请注明出处:【huachao1001的专栏:http://blog.csdn.net/huachao1001】

或许你会觉得没有必要这样做,可是有一种应用场景就是,为每个编译后的class文件添加一行代码。比如:在每个Java类的构造函数中加一句System.out.println("I Love HuaChao!");(PS:莫吐槽~,莫嘲笑~),如果你每次创建一个类的时候都手动加这么一句话,先不谈容易出错,我们说说工作量。或许你觉得,你愿意手动加,那我再跟你提新需求,我现在不要这句代码了,我要的是System.out.println("I Love MaYun!");你给我改去吧~,这时候你会不会想骂人~。忍住!我们上一篇《在AndroidStudio中自定义Gradle插件》不是学过自定义Gradle插件了吗?我们为什么要手动写呢?直接通过Gradle插件来帮我们干!

1 认识Project对象

还记得上一篇文章中,我们自定义的插件类是通过实现Plugin接口,并将org.gradle.api.Project作为模板参数吗?org.gradle.api.Project的实例对象将作为参数传给void apply(Project project)函数。接下来我看看Project类。

根据Gradle官网的介绍,Project是你与Gradle交互的主接口,通过Project你可以通过代码使用所有的Gradle特性,Projectbuild.gradle是一对一的关系。简单来说,你想要通过代码使用Gradle,通过Project这个入口,就可以啦~

我们先看一个简单的通过Project访问的使用场景:Extension。可能你对Extension不熟悉,但是,我给你看一个你熟悉的内容:

<code class="language-java hljs  has-numbering">android {
    compileSdkVersion <span class="hljs-number">24</span>
    buildToolsVersion <span class="hljs-string">"24.0.0"</span>

    defaultConfig {
        applicationId <span class="hljs-string">"com.hc.hcplugin"</span>
        minSdkVersion <span class="hljs-number">15</span>
        targetSdkVersion <span class="hljs-number">24</span>
        versionCode <span class="hljs-number">1</span>
        versionName <span class="hljs-string">"1.0"</span>
    }
    buildTypes {
        release {
            minifyEnabled <span class="hljs-keyword">false</span>
            proguardFiles getDefaultProguardFile(<span class="hljs-string">'proguard-android.txt'</span>), <span class="hljs-string">'proguard-rules.pro'</span>
        }
    }
}</code><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li></ul><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li></ul>

上面的这些你是不是很熟悉呢?你有没有想过,上面的Android{}compileSdkVersiondefaultConfig {}等等这些设置是如何被AndroidGradle插件读取的呢?想必你已经想到了,没错,就是通过Extension。下面我们自定义一个Extension,感受一下~。首先,定义两个Groovy类:AddressHCExtension.注意:为了避免引入插件问题,以下代码全部放入buildsrc模块的build.gradle文件中:

<code class="language-java hljs  has-numbering">class Address{
    String province=<span class="hljs-keyword">null</span>
    String city=<span class="hljs-keyword">null</span>
}
class HCExtension{
    String myName = <span class="hljs-keyword">null</span>;

}
</code><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li></ul><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li></ul>

再新建一个Plugin(同样也放入build.gradle中)

<code class="language-java hljs  has-numbering">class TestExtensionPlugin implements Plugin<Project> {

    <span class="hljs-annotation">@Override</span>
    <span class="hljs-keyword">void</span> apply(Project project) {

        project.extensions.create(<span class="hljs-string">'hc'</span>, HCExtension);
        project.extensions.create(<span class="hljs-string">'address'</span>, Address);


        project.task(<span class="hljs-string">'readExtension'</span>) << {
            def address=project[<span class="hljs-string">'address'</span>]

            println project[<span class="hljs-string">'hc'</span>].myName
            println address.province+<span class="hljs-string">" "</span>+address.city

        }
    }
}</code><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li></ul><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li></ul>

接下来就是把你的配置放进去啦(同样也放入build.gradle中)

<code class="language-java hljs  has-numbering">apply plugin: TestExtensionPlugin

hc {
    address{
       province <span class="hljs-string">"HuBei"</span>
        city <span class="hljs-string">"WuHan"</span>
    }

    myName <span class="hljs-string">"huachao"</span>

}</code><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li></ul><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li></ul>

稍微解释一下,apply plugin: TestExtensionPlugin这一行会导致直接执行TestExtensionPlugin类的apply方法。所以,hc{}这个块必须放在apply plugin: TestExtensionPlugin之后,因为在没有执行project.extensions.create('hc', HCExtension);之前,使用hc{}会报错!address{}也是同理。另外,补充一下:project.extensions相当于project.getExtensions()即返回的是ExtensionContainer对象,而ExtensionContainer对象的create方法就是把hc{}HCExtension对应起来。其他通过project.的方式也是同样的道理。再看看project.task('readExtension'),这是创建一个task。相当于在build.gradle文件中的task xxx <<{}只不过这里是通过代码的方式动态创建.

此时你的buildsrc模块中的build.gradle文件应该如下:

<code class="language-java hljs  has-numbering">apply plugin: <span class="hljs-string">'groovy'</span>

dependencies {
    compile gradleApi()<span class="hljs-comment">//gradle sdk</span>
    compile localGroovy()<span class="hljs-comment">//groovy sdk</span>
    compile <span class="hljs-string">'com.android.tools.build:gradle:2.1.0'</span>
}

repositories {
    jcenter()
}
class Address{
    String province=<span class="hljs-keyword">null</span>
    String city=<span class="hljs-keyword">null</span>
}
class HCExtension{
    String myName = <span class="hljs-keyword">null</span>;

}

class TestExtensionPlugin implements Plugin<Project> {

    <span class="hljs-annotation">@Override</span>
    <span class="hljs-keyword">void</span> apply(Project project) {

        project.extensions.create(<span class="hljs-string">'hc'</span>, HCExtension);
        project.extensions.create(<span class="hljs-string">'address'</span>, Address);


        project.task(<span class="hljs-string">'readExtension'</span>) << {
            def address=project[<span class="hljs-string">'address'</span>]

            println project[<span class="hljs-string">'hc'</span>].myName
            println address.province+<span class="hljs-string">" "</span>+address.city

        }
    }
}

apply plugin: TestExtensionPlugin

hc {
    address{
       province <span class="hljs-string">"HuBei"</span>
        city <span class="hljs-string">"WuHan"</span>
    }

    myName <span class="hljs-string">"huachao"</span>

}</code><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li><li>19</li><li>20</li><li>21</li><li>22</li><li>23</li><li>24</li><li>25</li><li>26</li><li>27</li><li>28</li><li>29</li><li>30</li><li>31</li><li>32</li><li>33</li><li>34</li><li>35</li><li>36</li><li>37</li><li>38</li><li>39</li><li>40</li><li>41</li><li>42</li><li>43</li><li>44</li><li>45</li><li>46</li><li>47</li><li>48</li><li>49</li><li>50</li></ul><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li><li>19</li><li>20</li><li>21</li><li>22</li><li>23</li><li>24</li><li>25</li><li>26</li><li>27</li><li>28</li><li>29</li><li>30</li><li>31</li><li>32</li><li>33</li><li>34</li><li>35</li><li>36</li><li>37</li><li>38</li><li>39</li><li>40</li><li>41</li><li>42</li><li>43</li><li>44</li><li>45</li><li>46</li><li>47</li><li>48</li><li>49</li><li>50</li></ul>

点击buildsrc模块中的readExtension如下图:

readExtension

看看打印信息

<code class="language-java hljs  has-numbering">···
:buildsrc:readExtension
huachao
HuBei WuHan

···</code><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li></ul><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li></ul>

关于Project对象先介绍到这里,更多内容请查看官方网站:https://docs.gradle.org/current/javadoc/org/gradle/api/Project.html

2 修改编译后的class

接下来回到我们的主题,我们需要修改class文件,首先我们得知道什么时候编译完成,并且我们要赶在class文件被转化为dex文件之前去修改。从1.5.0-beta1开始,androidgradle插件引入了com.android.build.api.transform.Transform,可以点击http://tools.android.com/tech-docs/new-build-system/transform-api 查看相关内容。Transform每次都是将一个输入进行处理,然后将处理结果输出,而输出的结果将会作为另一个Transform的输入,过程如下:

Transform处理过程

注意,输出地址不是由你任意指定的。而是根据输入的内容、作用范围等由TransformOutputProvider生成,比如,你要获取输出路径:

<code class="language-java hljs  has-numbering"> String dest = outputProvider.getContentLocation(directoryInput.name,
                        directoryInput.contentTypes, directoryInput.scopes, Format.DIRECTORY)</code><ul class="pre-numbering"><li>1</li><li>2</li></ul><ul class="pre-numbering"><li>1</li><li>2</li></ul>

Transform是一个抽象类,我们先自定义一个Transform,如下:

<code class="language-java hljs  has-numbering"><span class="hljs-keyword">package</span> com.hc.plugin

<span class="hljs-keyword">import</span> com.android.build.api.transform.*
<span class="hljs-keyword">import</span> com.android.build.gradle.internal.pipeline.TransformManager
<span class="hljs-keyword">import</span> org.apache.commons.codec.digest.DigestUtils
<span class="hljs-keyword">import</span> org.apache.commons.io.FileUtils
<span class="hljs-keyword">import</span> org.gradle.api.Project

<span class="hljs-javadoc">/**
 * Created by HuaChao on 2016/7/4.
 */</span>
<span class="hljs-keyword">public</span> <span class="hljs-class"><span class="hljs-keyword">class</span> <span class="hljs-title">MyTransform</span> <span class="hljs-keyword">extends</span> <span class="hljs-title">Transform</span> {</span>

    Project project

    <span class="hljs-comment">// 构造函数,我们将Project保存下来备用</span>
    <span class="hljs-keyword">public</span> <span class="hljs-title">MyTransform</span>(Project project) {
        <span class="hljs-keyword">this</span>.project = project
    }

    <span class="hljs-comment">// 设置我们自定义的Transform对应的Task名称</span>
    <span class="hljs-comment">// 类似:TransformClassesWithPreDexForXXX</span>
    <span class="hljs-annotation">@Override</span>
    String getName() {
        <span class="hljs-keyword">return</span> <span class="hljs-string">"MyTrans"</span>
    }

    <span class="hljs-comment">// 指定输入的类型,通过这里的设定,可以指定我们要处理的文件类型</span>
    <span class="hljs-comment">//这样确保其他类型的文件不会传入</span>
    <span class="hljs-annotation">@Override</span>
    Set<QualifiedContent.ContentType> getInputTypes() {
        <span class="hljs-keyword">return</span> TransformManager.CONTENT_CLASS
    }

    <span class="hljs-comment">// 指定Transform的作用范围</span>
    <span class="hljs-annotation">@Override</span>
    Set<QualifiedContent.Scope> getScopes() {
        <span class="hljs-keyword">return</span> TransformManager.SCOPE_FULL_PROJECT
    }

    <span class="hljs-annotation">@Override</span>
    <span class="hljs-keyword">boolean</span> isIncremental() {
        <span class="hljs-keyword">return</span> <span class="hljs-keyword">false</span>
    }

<span class="hljs-comment">//具体的处理</span>
    <span class="hljs-annotation">@Override</span>
    <span class="hljs-keyword">void</span> transform(Context context, Collection<TransformInput> inputs,
                   Collection<TransformInput> referencedInputs,
                   TransformOutputProvider outputProvider, <span class="hljs-keyword">boolean</span> isIncremental)
            <span class="hljs-keyword">throws</span> IOException, TransformException, InterruptedException {

    }
}</code><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li><li>19</li><li>20</li><li>21</li><li>22</li><li>23</li><li>24</li><li>25</li><li>26</li><li>27</li><li>28</li><li>29</li><li>30</li><li>31</li><li>32</li><li>33</li><li>34</li><li>35</li><li>36</li><li>37</li><li>38</li><li>39</li><li>40</li><li>41</li><li>42</li><li>43</li><li>44</li><li>45</li><li>46</li><li>47</li><li>48</li><li>49</li><li>50</li><li>51</li><li>52</li><li>53</li><li>54</li></ul><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li><li>19</li><li>20</li><li>21</li><li>22</li><li>23</li><li>24</li><li>25</li><li>26</li><li>27</li><li>28</li><li>29</li><li>30</li><li>31</li><li>32</li><li>33</li><li>34</li><li>35</li><li>36</li><li>37</li><li>38</li><li>39</li><li>40</li><li>41</li><li>42</li><li>43</li><li>44</li><li>45</li><li>46</li><li>47</li><li>48</li><li>49</li><li>50</li><li>51</li><li>52</li><li>53</li><li>54</li></ul>

看到函数transform,我们还没有具体实现这个函数。这个函数就是具体如何处理输入和输出。可以运行一下看看,注意,这里的运行时直接编译执行我们的apk,而不是像之前那样直接rebuild,因为rebuild并没有执行到编译这一步。由于我们没有实现transform这个函数,导致没有输出!使得整个过程中断了!最终导致apk运行时找不到MainActivity,所以会报错。接下来我们去实现以下这个函数,我们啥也不干,就是把输入内容写入到作为输出内容,不做任何处理,(下面代码参考自这里)

<code class="language-java hljs  has-numbering"><span class="hljs-annotation">@Override</span>
<span class="hljs-keyword">void</span> transform(Context context, Collection<TransformInput> inputs,
               Collection<TransformInput> referencedInputs,
               TransformOutputProvider outputProvider, <span class="hljs-keyword">boolean</span> isIncremental)
        <span class="hljs-keyword">throws</span> IOException, TransformException, InterruptedException {
    <span class="hljs-comment">// Transform的inputs有两种类型,一种是目录,一种是jar包,要分开遍历</span>
    inputs.each {TransformInput input ->
        <span class="hljs-comment">//对类型为“文件夹”的input进行遍历</span>
            input.directoryInputs.each {DirectoryInput directoryInput->
             <span class="hljs-comment">//文件夹里面包含的是我们手写的类以及R.class、BuildConfig.class以及R$XXX.class等</span>


            <span class="hljs-comment">// 获取output目录</span>
            def dest = outputProvider.getContentLocation(directoryInput.name,
                    directoryInput.contentTypes, directoryInput.scopes, 
                    Format.DIRECTORY)

            <span class="hljs-comment">// 将input的目录复制到output指定目录</span>
            FileUtils.copyDirectory(directoryInput.file, dest)
        }
    <span class="hljs-comment">//对类型为jar文件的input进行遍历</span>
        input.jarInputs.each {JarInput jarInput->

            <span class="hljs-comment">//jar文件一般是第三方依赖库jar文件</span>

            <span class="hljs-comment">// 重命名输出文件(同目录copyFile会冲突)</span>
            def jarName = jarInput.name
            def md5Name = DigestUtils.md5Hex(jarInput.file.getAbsolutePath())
            <span class="hljs-keyword">if</span>(jarName.endsWith(<span class="hljs-string">".jar"</span>)) {
                jarName = jarName.substring(<span class="hljs-number">0</span>,jarName.length()-<span class="hljs-number">4</span>)
            }
            <span class="hljs-comment">//生成输出路径</span>
            def dest = outputProvider.getContentLocation(jarName+md5Name, 
                         jarInput.contentTypes, jarInput.scopes, Format.JAR)
            <span class="hljs-comment">//将输入内容复制到输出</span>
            FileUtils.copyFile(jarInput.file, dest)
        }
    }
}</code><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li><li>19</li><li>20</li><li>21</li><li>22</li><li>23</li><li>24</li><li>25</li><li>26</li><li>27</li><li>28</li><li>29</li><li>30</li><li>31</li><li>32</li><li>33</li><li>34</li><li>35</li><li>36</li><li>37</li><li>38</li><li>39</li></ul><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li><li>19</li><li>20</li><li>21</li><li>22</li><li>23</li><li>24</li><li>25</li><li>26</li><li>27</li><li>28</li><li>29</li><li>30</li><li>31</li><li>32</li><li>33</li><li>34</li><li>35</li><li>36</li><li>37</li><li>38</li><li>39</li></ul>

注意input的类型,分为”文件夹”和“jar文件”,”文件夹”里面的就是我们写的类对应的class文件,jar文件一般为第三方库。此时,能成功运行,但是我们还没有注入代码呢~,下面我们看看如何注入代码~

3 Javassist

要修改class字节码,我们要是自己手动改二进制文件,有点困难,好在有Javassist这个库,可以让我们直接修改编译后的class二进制代码。关于Javassist的使用,这里不介绍,可以自行搜索。要使用到Javassist,我们得在buildsrc模块下的build.gradle添加依赖包:

<code class="language-java hljs  has-numbering">compile <span class="hljs-string">'org.javassist:javassist:3.20.0-GA'</span></code><ul class="pre-numbering"><li>1</li></ul><ul class="pre-numbering"><li>1</li></ul>

使用Javassist也很简单,首先拿到ClassPool对象,通过ClassPool获取已经编译好的类,如:

<code class="language-java hljs  has-numbering">ClassPool pool = ClassPool.getDefault();
CtClass cc = pool.get(<span class="hljs-string">"com.hc.MyClass"</span>);
cc.setSuperclass(pool.get(<span class="hljs-string">"com.hc.ParentClass"</span>));
cc.writeFile();</code><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li></ul><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li></ul>

上面代码就实现了修改MyClass类的父类为ParentClass.

要获取字节码以及加载为Class对象,如下:

<code class="language-java hljs  has-numbering"><span class="hljs-keyword">byte</span>[] b = cc.toBytecode();
Class clazz = cc.toClass();</code><ul class="pre-numbering"><li>1</li><li>2</li></ul><ul class="pre-numbering"><li>1</li><li>2</li></ul>

前面提到,我们自己创建的Java类编译后是放入到文件夹里面的,因此,我们只需针对这个文件夹里面的class文件进行修改即可,新建一个Groovy类:

<code class="language-java hljs  has-numbering"><span class="hljs-keyword">package</span> com.hc.plugin

<span class="hljs-keyword">import</span> javassist.ClassPool
<span class="hljs-keyword">import</span> javassist.CtClass
<span class="hljs-keyword">import</span> javassist.CtConstructor
<span class="hljs-keyword">public</span> <span class="hljs-class"><span class="hljs-keyword">class</span> <span class="hljs-title">MyInject</span> {</span>

    <span class="hljs-keyword">private</span> <span class="hljs-keyword">static</span> ClassPool pool = ClassPool.getDefault()
    <span class="hljs-keyword">private</span> <span class="hljs-keyword">static</span> String injectStr = <span class="hljs-string">"System.out.println(\"I Love HuaChao\" ); "</span>;

    <span class="hljs-keyword">public</span> <span class="hljs-keyword">static</span> <span class="hljs-keyword">void</span> <span class="hljs-title">injectDir</span>(String path, String packageName) {
        pool.appendClassPath(path)
        File dir = <span class="hljs-keyword">new</span> File(path)
        <span class="hljs-keyword">if</span> (dir.isDirectory()) {
            dir.eachFileRecurse { File file ->

                String filePath = file.absolutePath
                <span class="hljs-comment">//确保当前文件是class文件,并且不是系统自动生成的class文件</span>
                <span class="hljs-keyword">if</span> (filePath.endsWith(<span class="hljs-string">".class"</span>)
                        && !filePath.contains(<span class="hljs-string">'R$'</span>)
                        && !filePath.contains(<span class="hljs-string">'R.class'</span>)
                        && !filePath.contains(<span class="hljs-string">"BuildConfig.class"</span>)) {
                    <span class="hljs-comment">// 判断当前目录是否是在我们的应用包里面</span>
                    <span class="hljs-keyword">int</span> index = filePath.indexOf(packageName);
                    <span class="hljs-keyword">boolean</span> isMyPackage = index != -<span class="hljs-number">1</span>;
                    <span class="hljs-keyword">if</span> (isMyPackage) {
                        <span class="hljs-keyword">int</span> end = filePath.length() - <span class="hljs-number">6</span> <span class="hljs-comment">// .class = 6</span>
                        String className = filePath.substring(index, end)
                           .replace(<span class="hljs-string">'\\'</span>, <span class="hljs-string">'.'</span>).replace(<span class="hljs-string">'/'</span>, <span class="hljs-string">'.'</span>)
                        <span class="hljs-comment">//开始修改class文件</span>
                        CtClass c = pool.getCtClass(className)

                        <span class="hljs-keyword">if</span> (c.isFrozen()) {
                            c.defrost()
                        }

                        CtConstructor[] cts = c.getDeclaredConstructors() 
                        <span class="hljs-keyword">if</span> (cts == <span class="hljs-keyword">null</span> || cts.length == <span class="hljs-number">0</span>) {
                            <span class="hljs-comment">//手动创建一个构造函数</span>
                            CtConstructor constructor = <span class="hljs-keyword">new</span> CtConstructor(<span class="hljs-keyword">new</span> CtClass[<span class="hljs-number">0</span>], c)
                            constructor.insertBeforeBody(injectStr)
                            c.addConstructor(constructor)
                        } <span class="hljs-keyword">else</span> {
                            cts[<span class="hljs-number">0</span>].insertBeforeBody(injectStr)
                        }
                        c.writeFile(path)
                        c.detach()
                    }
                }
            }
        }
    }


}</code><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li><li>19</li><li>20</li><li>21</li><li>22</li><li>23</li><li>24</li><li>25</li><li>26</li><li>27</li><li>28</li><li>29</li><li>30</li><li>31</li><li>32</li><li>33</li><li>34</li><li>35</li><li>36</li><li>37</li><li>38</li><li>39</li><li>40</li><li>41</li><li>42</li><li>43</li><li>44</li><li>45</li><li>46</li><li>47</li><li>48</li><li>49</li><li>50</li><li>51</li><li>52</li><li>53</li><li>54</li><li>55</li></ul><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li><li>19</li><li>20</li><li>21</li><li>22</li><li>23</li><li>24</li><li>25</li><li>26</li><li>27</li><li>28</li><li>29</li><li>30</li><li>31</li><li>32</li><li>33</li><li>34</li><li>35</li><li>36</li><li>37</li><li>38</li><li>39</li><li>40</li><li>41</li><li>42</li><li>43</li><li>44</li><li>45</li><li>46</li><li>47</li><li>48</li><li>49</li><li>50</li><li>51</li><li>52</li><li>53</li><li>54</li><li>55</li></ul>

然后就是在 transform函数中,针对“文件夹”里面的class进行注入,而jar文件类型的input依然不做处理。transform函数如下:

<code class="language-java hljs  has-numbering"> <span class="hljs-annotation">@Override</span>
<span class="hljs-keyword">void</span> transform(Context context, Collection<TransformInput> inputs,
               Collection<TransformInput> referencedInputs,
               TransformOutputProvider outputProvider, <span class="hljs-keyword">boolean</span> isIncremental)
        <span class="hljs-keyword">throws</span> IOException, TransformException, InterruptedException {
    <span class="hljs-comment">// Transform的inputs有两种类型,一种是目录,一种是jar包,要分开遍历</span>
    inputs.each { TransformInput input ->
        <span class="hljs-comment">//对类型为“文件夹”的input进行遍历</span>
        input.directoryInputs.each { DirectoryInput directoryInput ->
            <span class="hljs-comment">//文件夹里面包含的是我们手写的类以及R.class、BuildConfig.class以及R$XXX.class等</span>
            MyInject.injectDir(directoryInput.file.absolutePath,<span class="hljs-string">"com\\hc\\hcplugin"</span>)
            <span class="hljs-comment">// 获取output目录</span>
            def dest = outputProvider.getContentLocation(directoryInput.name,
                    directoryInput.contentTypes, directoryInput.scopes,
                    Format.DIRECTORY)

            <span class="hljs-comment">// 将input的目录复制到output指定目录</span>
            FileUtils.copyDirectory(directoryInput.file, dest)
        }
        <span class="hljs-comment">//对类型为jar文件的input进行遍历</span>
        input.jarInputs.each { JarInput jarInput ->

            <span class="hljs-comment">//jar文件一般是第三方依赖库jar文件</span>

            <span class="hljs-comment">// 重命名输出文件(同目录copyFile会冲突)</span>
            def jarName = jarInput.name
            def md5Name = DigestUtils.md5Hex(jarInput.file.getAbsolutePath())
            <span class="hljs-keyword">if</span> (jarName.endsWith(<span class="hljs-string">".jar"</span>)) {
                jarName = jarName.substring(<span class="hljs-number">0</span>, jarName.length() - <span class="hljs-number">4</span>)
            }
            <span class="hljs-comment">//生成输出路径</span>
            def dest = outputProvider.getContentLocation(jarName + md5Name,
                    jarInput.contentTypes, jarInput.scopes, Format.JAR)
            <span class="hljs-comment">//将输入内容复制到输出</span>
            FileUtils.copyFile(jarInput.file, dest)
        }
    }
}</code><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li><li>19</li><li>20</li><li>21</li><li>22</li><li>23</li><li>24</li><li>25</li><li>26</li><li>27</li><li>28</li><li>29</li><li>30</li><li>31</li><li>32</li><li>33</li><li>34</li><li>35</li><li>36</li><li>37</li><li>38</li></ul><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li><li>19</li><li>20</li><li>21</li><li>22</li><li>23</li><li>24</li><li>25</li><li>26</li><li>27</li><li>28</li><li>29</li><li>30</li><li>31</li><li>32</li><li>33</li><li>34</li><li>35</li><li>36</li><li>37</li><li>38</li></ul>

大功告成,接下来测试一下,在app模块中,新建一个Test类,在MainActivity中调用new Test();

Test.java

<code class="language-java hljs  has-numbering"><span class="hljs-keyword">package</span> com.hc.hcplugin;

<span class="hljs-javadoc">/**
 * Created by HuaChao on 2016/7/4.
 */</span>
<span class="hljs-keyword">public</span> <span class="hljs-class"><span class="hljs-keyword">class</span> <span class="hljs-title">Test</span> {</span>
}
</code><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li></ul><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li></ul>

MainActivity.java

<code class="language-java hljs  has-numbering"><span class="hljs-keyword">package</span> com.hc.hcplugin;

<span class="hljs-keyword">import</span> android.os.Bundle;
<span class="hljs-keyword">import</span> android.support.v7.app.AppCompatActivity;
<span class="hljs-keyword">import</span> android.util.Log;

<span class="hljs-keyword">public</span> <span class="hljs-class"><span class="hljs-keyword">class</span> <span class="hljs-title">MainActivity</span> <span class="hljs-keyword">extends</span> <span class="hljs-title">AppCompatActivity</span> {</span>

    <span class="hljs-annotation">@Override</span>
    <span class="hljs-keyword">protected</span> <span class="hljs-keyword">void</span> <span class="hljs-title">onCreate</span>(Bundle savedInstanceState) {
        <span class="hljs-keyword">super</span>.onCreate(savedInstanceState);
        Log.e(<span class="hljs-string">"--->"</span>, <span class="hljs-string">"==================="</span>);
        <span class="hljs-keyword">new</span> Test();
        Log.e(<span class="hljs-string">"--->"</span>, <span class="hljs-string">"==================="</span>);
    }

}
</code><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li></ul><ul class="pre-numbering"><li>1</li><li>2</li><li>3</li><li>4</li><li>5</li><li>6</li><li>7</li><li>8</li><li>9</li><li>10</li><li>11</li><li>12</li><li>13</li><li>14</li><li>15</li><li>16</li><li>17</li><li>18</li></ul>

运行结果如下:

运行结果

第一个打印是MainActivity的构造函数打印的,第二个是Test的构造函数打印的。看到这里,或许你想说,这有什么用啊?难道搞半天就为了打印这么一句话?其实,真的很有用,如果你看过关于热补丁相关内容,你就知道,还真的需要对每个类加上System.out.println(xxx)。不信你看:

https://mp.weixin.qq.com/s?__biz=MzI1MTA1MzM2Nw==&mid=400118620&idx=1&sn=b4fdd5055731290eef12ad0d17f39d4a&scene=1&srcid=1106Imu9ZgwybID13e7y2nEi#wechat_redirect

附上源码:http://download.csdn.net/download/huachao1001/9567113

 

转载自:http://blog.csdn.net/huachao1001/article/details/51819972

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值