memberMapping) {
39 | this.memberMapping = memberMapping;
40 | }
41 |
42 | @Override
43 | public boolean equals(Object o) {
44 | if (this == o) return true;
45 | if (!(o instanceof ClassMapping)) return false;
46 |
47 | ClassMapping classMapping = (ClassMapping) o;
48 |
49 | return getClassName() != null ? getClassName().equals(classMapping.getClassName()) : classMapping.getClassName() == null;
50 |
51 | }
52 |
53 | @Override
54 | public int hashCode() {
55 | return getClassName() != null ? getClassName().hashCode() : 0;
56 | }
57 | }
58 |
--------------------------------------------------------------------------------
/auto-patch-plugin/src/main/java/com/meituan/robust/autopatch/Config.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust.autopatch;
2 |
3 | import com.meituan.robust.Constants;
4 |
5 | import java.util.ArrayList;
6 | import java.util.HashMap;
7 | import java.util.HashSet;
8 | import java.util.LinkedHashMap;
9 | import java.util.List;
10 | import java.util.Map;
11 | import java.util.Set;
12 |
13 | import javassist.ClassPool;
14 | import javassist.CtMethod;
15 |
16 | import static com.meituan.robust.Constants.DEFAULT_MAPPING_FILE;
17 |
18 | /**
19 | * Created by mivanzhang on 16/12/2.
20 | *
21 | * members read from robust.xml
22 | */
23 |
24 | public final class Config {
25 | public static boolean catchReflectException = false;
26 | public static boolean supportProGuard = true;
27 | public static boolean isLogging = true;
28 | public static boolean isManual = false;
29 | public static String patchPackageName = Constants.PATCH_PACKAGENAME;
30 | public static String mappingFilePath;
31 | public static Set patchMethodSignatureSet = new HashSet<>();
32 | public static List newlyAddedClassNameList = new ArrayList();
33 | public static Set newlyAddedMethodSet = new HashSet();
34 | public static List modifiedClassNameList = new ArrayList();
35 | public static List hotfixPackageList = new ArrayList<>();
36 | public static LinkedHashMap methodMap = new LinkedHashMap<>();
37 | public static String robustGenerateDirectory;
38 | public static Map> invokeSuperMethodMap = new HashMap<>();
39 | public static ClassPool classPool = new ClassPool();
40 | public static Set methodNeedPatchSet = new HashSet();
41 | public static List addedSuperMethodList = new ArrayList<>();
42 | public static Set noNeedReflectClassSet = new HashSet<>();
43 |
44 |
45 | public static void init() {
46 | catchReflectException = false;
47 | isLogging = true;
48 | isManual = false;
49 | patchPackageName = Constants.PATCH_PACKAGENAME;
50 | mappingFilePath = DEFAULT_MAPPING_FILE;
51 | patchMethodSignatureSet = new HashSet<>();
52 | newlyAddedClassNameList = new ArrayList();
53 | modifiedClassNameList = new ArrayList();
54 | hotfixPackageList = new ArrayList<>();
55 | newlyAddedMethodSet = new HashSet<>();
56 | invokeSuperMethodMap = new HashMap<>();
57 | classPool = new ClassPool();
58 | methodNeedPatchSet = new HashSet();
59 | addedSuperMethodList = new ArrayList<>();
60 | noNeedReflectClassSet = new HashSet<>();
61 | noNeedReflectClassSet.addAll(Constants.NO_NEED_REFLECT_CLASS);
62 | supportProGuard=true;
63 | }
64 |
65 | }
66 |
--------------------------------------------------------------------------------
/auto-patch-plugin/src/main/java/com/meituan/robust/autopatch/NameManger.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust.autopatch;
2 |
3 | import com.meituan.robust.Constants;
4 |
5 | import java.util.HashMap;
6 |
7 | /**
8 | * Created by mivanzhang on 17/2/10.
9 | *
10 | * manager patches' name ,inline patch name
11 | */
12 |
13 | public class NameManger {
14 | private static NameManger nameManger;
15 | private HashMap patchNameMap = new HashMap();
16 |
17 | private NameManger() {
18 |
19 | }
20 |
21 | public static void init() {
22 | nameManger = new NameManger();
23 | }
24 |
25 | public static NameManger getInstance() {
26 | if (nameManger == null) {
27 | nameManger = new NameManger();
28 | }
29 | return nameManger;
30 | }
31 |
32 | public HashMap getPatchNameMap() {
33 | return patchNameMap;
34 | }
35 |
36 | /***
37 | * @param className java class name java.lang.object
38 | * @return patchname.object
39 | */
40 | public String getPatchName(String className) {
41 | String patchName = getPatchNamWithoutRecord(className);
42 | patchNameMap.put(patchName, className);
43 | return patchName;
44 | }
45 |
46 | public String getInlinePatchName(String className) {
47 | String patchName = getInlinePatchNameWithoutRecord(className);
48 | patchNameMap.put(patchName, className);
49 | return patchName;
50 | }
51 |
52 | public String getInlinePatchNameWithoutRecord(String className) {
53 | String patchName = Config.patchPackageName + "." + className.substring(className.lastIndexOf(".") + 1) + Constants.INLINE_PATCH_SUFFIX;
54 | return patchName;
55 | }
56 |
57 | public String getPatchNamWithoutRecord(String className) {
58 | String patchName = Config.patchPackageName + "." + className.substring(className.lastIndexOf(".") + 1) + Constants.PATCH_SUFFIX;
59 | return patchName;
60 | }
61 |
62 | public String getPatchControlName(String simpleClassName) {
63 | return Config.patchPackageName + "." + simpleClassName + Constants.PATCH_SUFFIX + Constants.PATCH_CONTROL_SUFFIX;
64 | }
65 |
66 | public String getAssistClassName(String s) {
67 | patchNameMap.put(s + Constants.ROBUST_ASSIST_SUFFIX, s + Constants.ROBUST_ASSIST_SUFFIX);
68 | return s + Constants.ROBUST_ASSIST_SUFFIX;
69 | }
70 | }
71 |
--------------------------------------------------------------------------------
/auto-patch-plugin/src/main/java/com/meituan/robust/autopatch/PatchesControlFactory.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust.autopatch;
2 |
3 | import com.meituan.robust.Constants;
4 | import com.meituan.robust.utils.JavaUtils;
5 |
6 | import javassist.CtClass;
7 | import javassist.CtMethod;
8 | import javassist.NotFoundException;
9 | import javassist.bytecode.AccessFlag;
10 |
11 | import static com.meituan.robust.autopatch.Config.classPool;
12 |
13 | /**
14 | * Created by mivanzhang on 17/2/9.
15 | *
16 | * create patch control classes,which dispatch patch methods
17 | */
18 |
19 | public class PatchesControlFactory {
20 | private static PatchesControlFactory patchesControlFactory = new PatchesControlFactory();
21 |
22 | private PatchesControlFactory() {
23 |
24 | }
25 |
26 | private CtClass createControlClass(CtClass modifiedClass) throws Exception {
27 | CtClass patchClass = classPool.get(NameManger.getInstance().getPatchName(modifiedClass.getName()));
28 | patchClass.defrost();
29 | CtClass controlClass = classPool.getAndRename(Constants.PATCH_TEMPLATE_FULL_NAME, NameManger.getInstance().getPatchControlName(modifiedClass.getSimpleName()));
30 | StringBuilder getRealParameterMethodBody = new StringBuilder();
31 | getRealParameterMethodBody.append("public Object getRealParameter(Object parameter) {");
32 | getRealParameterMethodBody.append("if(parameter instanceof " + modifiedClass.getName() + "){");
33 | getRealParameterMethodBody.
34 | append("return new " + patchClass.getName() + "(parameter);");
35 | getRealParameterMethodBody.append("}");
36 | getRealParameterMethodBody.append("return parameter;}");
37 | controlClass.addMethod(CtMethod.make(getRealParameterMethodBody.toString(), controlClass));
38 | controlClass.getDeclaredMethod("accessDispatch").insertBefore(getAccessDispatchMethodBody(patchClass, modifiedClass.getName()));
39 | controlClass.getDeclaredMethod("isSupport").insertBefore(getIsSupportMethodBody(patchClass, modifiedClass.getName()));
40 | controlClass.defrost();
41 | return controlClass;
42 | }
43 |
44 | private
45 | static String getAccessDispatchMethodBody(CtClass patchClass, String modifiedClassName) throws NotFoundException {
46 | StringBuilder accessDispatchMethodBody = new StringBuilder();
47 | if(Config.catchReflectException){
48 | accessDispatchMethodBody.append("try{");
49 | }
50 | if (Constants.isLogging) {
51 | accessDispatchMethodBody.append(" android.util.Log.d(\"robust\",\"arrivied in AccessDispatch \"+methodName+\" paramArrayOfObject \" +paramArrayOfObject);");
52 | }
53 | //create patch instance
54 | accessDispatchMethodBody.append(patchClass.getName() + " patch= null;\n");
55 | accessDispatchMethodBody.append(" String isStatic=$1.split(\":\")[2];");
56 | accessDispatchMethodBody.append(" if (isStatic.equals(\"false\")) {\n");
57 | accessDispatchMethodBody.append(" if (keyToValueRelation.get(paramArrayOfObject[paramArrayOfObject.length - 1]) == null) {\n");
58 | if (Constants.isLogging) {
59 | accessDispatchMethodBody.append(" android.util.Log.d(\"robust\",\"keyToValueRelation not contain\" );");
60 | }
61 | accessDispatchMethodBody.append("patch=new " + patchClass.getName() + "(paramArrayOfObject[paramArrayOfObject.length - 1]);\n");
62 | accessDispatchMethodBody.append(" keyToValueRelation.put(paramArrayOfObject[paramArrayOfObject.length - 1], null);\n");
63 | accessDispatchMethodBody.append("}else{");
64 | accessDispatchMethodBody.append("patch=(" + patchClass.getName() + ") keyToValueRelation.get(paramArrayOfObject[paramArrayOfObject.length - 1]);\n");
65 | accessDispatchMethodBody.append("}");
66 | accessDispatchMethodBody.append("}\n");
67 | accessDispatchMethodBody.append("else{");
68 | if (Constants.isLogging) {
69 | accessDispatchMethodBody.append(" android.util.Log.d(\"robust\",\"static method forward \" );");
70 | }
71 | accessDispatchMethodBody.append("patch=new " + patchClass.getName() + "(null);\n");
72 | accessDispatchMethodBody.append("}");
73 | accessDispatchMethodBody.append("String methodNo=$1.split(\":\")[3];\n");
74 | if (Constants.isLogging) {
75 | accessDispatchMethodBody.append(" android.util.Log.d(\"robust\",\"assemble method number is \" + methodNo);");
76 | }
77 |
78 | // patchClass.declaredMethods.each {
79 | for (CtMethod method : patchClass.getDeclaredMethods()) {
80 | CtClass[] parametertypes = method.getParameterTypes();
81 | String methodSignure = JavaUtils.getJavaMethodSignure(method).replaceAll(patchClass.getName(), modifiedClassName);
82 | String methodLongName = modifiedClassName + "." + methodSignure;
83 | Integer methodNumber = Config.methodMap.get(methodLongName);
84 | //just Forward methods with methodNumber
85 | if (methodNumber != null) {
86 | accessDispatchMethodBody.append(" if((\"" + methodNumber + "\").equals(methodNo)){\n");
87 | if (Constants.isLogging) {
88 | accessDispatchMethodBody.append(" android.util.Log.d(\"robust\",\"invoke method is " + method.getLongName() + " \" );");
89 | }
90 | String methodName = method.getName();
91 | if (AccessFlag.isPrivate(method.getModifiers())) {
92 | methodName = Constants.ROBUST_PUBLIC_SUFFIX + method.getName();
93 | }
94 | if (method.getReturnType().getName().equals("void")) {
95 | accessDispatchMethodBody.append("(patch." + methodName + "(");
96 | } else {
97 | switch (method.getReturnType().getName()) {
98 | case "boolean":
99 | accessDispatchMethodBody.append("return Boolean.valueOf(patch." + methodName + "(");
100 | break;
101 | case "byte":
102 | accessDispatchMethodBody.append("return Byte.valueOf(patch." + methodName + "(");
103 | break;
104 | case "char":
105 | accessDispatchMethodBody.append("return Character.valueOf(patch." + methodName + "(");
106 | break;
107 | case "double":
108 | accessDispatchMethodBody.append("return Double.valueOf(patch." + methodName + "(");
109 | break;
110 | case "float":
111 | accessDispatchMethodBody.append("return Float.valueOf(patch." + methodName + "(");
112 | break;
113 | case "int":
114 | accessDispatchMethodBody.append("return Integer.valueOf(patch." + methodName + "(");
115 | break;
116 | case "long":
117 | accessDispatchMethodBody.append("return Long.valueOf(patch." + methodName + "(");
118 | break;
119 | case "short":
120 | accessDispatchMethodBody.append("return Short.valueOf(patch." + methodName + "(");
121 | break;
122 | default:
123 | accessDispatchMethodBody.append("return (patch." + methodName + "(");
124 | break;
125 | }
126 | }
127 | for (int index = 0; index < parametertypes.length; index++) {
128 | if (booleanPrimeType(parametertypes[index].getName())){
129 | accessDispatchMethodBody.append("((" + JavaUtils.getWrapperClass(parametertypes[index].getName()) + ") (fixObj(paramArrayOfObject[" + index + "]))");
130 | accessDispatchMethodBody.append(")" + JavaUtils.wrapperToPrime(parametertypes[index].getName()));
131 | if (index != parametertypes.length - 1) {
132 | accessDispatchMethodBody.append(",");
133 | }
134 | } else {
135 | accessDispatchMethodBody.append("((" + JavaUtils.getWrapperClass(parametertypes[index].getName()) + ") (paramArrayOfObject[" + index + "])");
136 | accessDispatchMethodBody.append(")" + JavaUtils.wrapperToPrime(parametertypes[index].getName()));
137 | if (index != parametertypes.length - 1) {
138 | accessDispatchMethodBody.append(",");
139 | }
140 | }
141 | }
142 | accessDispatchMethodBody.append("));}\n");
143 | }
144 | }
145 | if(Config.catchReflectException){
146 | accessDispatchMethodBody.append(" } catch (Throwable e) {");
147 | accessDispatchMethodBody.append(" e.printStackTrace();}");
148 | }
149 | return accessDispatchMethodBody.toString();
150 | }
151 |
152 | private static String getIsSupportMethodBody(CtClass patchClass, String modifiedClassName) throws NotFoundException {
153 | StringBuilder isSupportBuilder = new StringBuilder();
154 | StringBuilder methodsIdBuilder = new StringBuilder();
155 | if (Constants.isLogging) {
156 | isSupportBuilder.append(" android.util.Log.d(\"robust\",\"arrivied in isSupport \"+methodName+\" paramArrayOfObject \" +paramArrayOfObject);");
157 | }
158 | isSupportBuilder.append("String methodNo=$1.split(\":\")[3];\n");
159 | if (Constants.isLogging) {
160 | isSupportBuilder.append(" android.util.Log.d(\"robust\",\"in isSupport assemble method number is \" + methodNo);");
161 | }
162 | for (CtMethod method : patchClass.getDeclaredMethods()) {
163 | String methodSignure = JavaUtils.getJavaMethodSignure(method).replaceAll(patchClass.getName(), modifiedClassName);
164 | String methodLongName = modifiedClassName + "." + methodSignure;
165 | Integer methodNumber = Config.methodMap.get(methodLongName);
166 | //just Forward methods with methodNumber
167 | if (methodNumber != null) {
168 | // 一前一后的冒号作为匹配锚点，只有一边有的话可能会有多重匹配的bug
169 | methodsIdBuilder.append(":" + methodNumber + ":");
170 | }
171 | }
172 |
173 | if (Constants.isLogging) {
174 | isSupportBuilder.append(" android.util.Log.d(\"robust\",\"arrivied in isSupport \"+methodName+\" paramArrayOfObject \" +paramArrayOfObject+\" isSupport result is \"+\"" + methodsIdBuilder.toString() + "\".contains(\":\" + methodNo + \":\"));");
175 | }
176 | isSupportBuilder.append("return \"" + methodsIdBuilder.toString() + "\".contains(\":\" + methodNo + \":\");");
177 | return isSupportBuilder.toString();
178 | }
179 |
180 |
181 | public static CtClass createPatchesControl(CtClass modifiedClass) throws Exception {
182 | return patchesControlFactory.createControlClass(modifiedClass);
183 | }
184 |
185 | public static boolean booleanPrimeType(String typeName) {
186 | return "boolean".equals(typeName);
187 | }
188 |
189 | }
190 |
--------------------------------------------------------------------------------
/auto-patch-plugin/src/main/java/com/meituan/robust/autopatch/PatchesInfoFactory.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust.autopatch;
2 |
3 | import com.meituan.robust.Constants;
4 |
5 | import javassist.CtClass;
6 | import javassist.CtMethod;
7 | import javassist.bytecode.ClassFile;
8 |
9 | import static com.meituan.robust.autopatch.Config.classPool;
10 | import static javassist.CtNewMethod.make;
11 |
12 | /**
13 | * Created by mivanzhang on 17/2/9.
14 | *
15 | * create patch info classes which describes patch dex
16 | */
17 |
18 | public class PatchesInfoFactory {
19 | private static PatchesInfoFactory patchesInfoFactory = new PatchesInfoFactory();
20 |
21 | private PatchesInfoFactory() {
22 |
23 | }
24 |
25 | private CtClass createPatchesInfoClass() {
26 | try {
27 | CtClass ctPatchesInfoImpl = classPool.makeClass(Config.patchPackageName + ".PatchesInfoImpl");
28 | ctPatchesInfoImpl.getClassFile().setMajorVersion(ClassFile.JAVA_7);
29 | ctPatchesInfoImpl.setInterfaces(new CtClass[]{classPool.get("com.meituan.robust.PatchesInfo")});
30 | StringBuilder methodBody = new StringBuilder();
31 | methodBody.append("public java.util.List getPatchedClassesInfo() {");
32 | methodBody.append(" java.util.List patchedClassesInfos = new java.util.ArrayList();");
33 | for (int i = 0; i < Config.modifiedClassNameList.size(); i++) {
34 | if (Constants.OBSCURE) {
35 | methodBody.append("com.meituan.robust.PatchedClassInfo patchedClass" + i + " = new com.meituan.robust.PatchedClassInfo(\"" + ReadMapping.getInstance().getClassMappingOrDefault(Config.modifiedClassNameList.get(i)).getValueName() + "\",\"" + NameManger.getInstance().getPatchControlName(Config.modifiedClassNameList.get(i).substring(Config.modifiedClassNameList.get(i).lastIndexOf('.') + 1)) + "\");");
36 | } else {
37 | methodBody.append("com.meituan.robust.PatchedClassInfo patchedClass" + i + " = new com.meituan.robust.PatchedClassInfo(\"" + Config.modifiedClassNameList.get(i) + "\",\"" + NameManger.getInstance().getPatchControlName(Config.modifiedClassNameList.get(i).substring(Config.modifiedClassNameList.get(i).lastIndexOf('.') + 1)) + "\");");
38 | }
39 | methodBody.append("patchedClassesInfos.add(patchedClass" + i + ");");
40 | }
41 | methodBody.append(Constants.ROBUST_UTILS_FULL_NAME + ".isThrowable=!" + Config.catchReflectException + ";");
42 | methodBody.append("return patchedClassesInfos;\n" +
43 | " }");
44 | CtMethod m = make(methodBody.toString(), ctPatchesInfoImpl);
45 | ctPatchesInfoImpl.addMethod(m);
46 | return ctPatchesInfoImpl;
47 | } catch (Exception e) {
48 | e.printStackTrace();
49 | throw new RuntimeException(e);
50 | }
51 | }
52 |
53 | public static CtClass createPatchesInfo() {
54 | return patchesInfoFactory.createPatchesInfoClass();
55 | }
56 | }
57 |
--------------------------------------------------------------------------------
/auto-patch-plugin/src/main/java/com/meituan/robust/autopatch/ReadMapping.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust.autopatch;
2 |
3 | import java.io.BufferedReader;
4 | import java.io.FileInputStream;
5 | import java.io.IOException;
6 | import java.io.InputStream;
7 | import java.io.InputStreamReader;
8 | import java.util.HashMap;
9 | import java.util.Map;
10 |
11 | /**
12 | * Created by mivanzhang on 17/1/19.
13 | *
14 | * Reading mapping from mapping.txt,which is generated by ProGuard
15 | */
16 |
17 | public class ReadMapping {
18 | private static ReadMapping instance;
19 |
20 | private Map usedInModifiedClassMappingInfo = new HashMap();
21 |
22 | public static ReadMapping getInstance() {
23 | if (instance == null) {
24 | instance = new ReadMapping();
25 | }
26 | return instance;
27 | }
28 |
29 | public static void init() {
30 | instance = new ReadMapping();
31 | }
32 |
33 | private ReadMapping() {
34 |
35 | }
36 |
37 | /***
38 | * read all class mapping info
39 | *
40 | * @return
41 | */
42 | public void initMappingInfo() {
43 | //查找mapping文件
44 | InputStream is = null;
45 | boolean needBacktrace = true;
46 | String line;
47 | try {
48 | is = new FileInputStream(Config.mappingFilePath);
49 | BufferedReader reader = new BufferedReader(new InputStreamReader(is, "UTF-8"), 1024);
50 | // 读取一行，存储于字符串列表中
51 | line = reader.readLine().trim();
52 | while (line != null) {
53 | line = line.trim();
54 | if (!needBacktrace) {
55 | line = reader.readLine();
56 | if (line == null) {
57 | break;
58 | }
59 | line = line.trim();
60 | }
61 | needBacktrace = false;
62 | if (line.indexOf("->") > 0 && line.indexOf(":") == line.length() - 1) {
63 | ClassMapping classMapping = new ClassMapping();
64 | classMapping.setClassName(line.substring(0, line.indexOf("->") - 1).trim());
65 | classMapping.setValueName(line.split("->")[1].substring(0, line.split("->")[1].length() - 1).trim());
66 | line = reader.readLine();
67 | while (line != null) {
68 | line = line.trim();
69 | if (line.endsWith(":")) {
70 | needBacktrace = true;
71 | break;
72 | }
73 | String[] lineinfo = line.split(" ");
74 | if (lineinfo.length != 4) {
75 | throw new RuntimeException("mapping line info is error " + line);
76 | }
77 | if (lineinfo[1].contains("(") && lineinfo[1].contains(")")) {
78 | //methods need return type
79 | classMapping.getMemberMapping().put(getMethodSigureWithReturnTypeInMapping(lineinfo[0].trim(), lineinfo[1].trim()), lineinfo[3].trim());
80 | } else {
81 | //fields
82 | classMapping.getMemberMapping().put(lineinfo[1].trim(), lineinfo[3].trim());
83 | }
84 | line = reader.readLine();
85 | if (line == null) {
86 | break;
87 | }
88 | line = line.trim();
89 | }
90 | usedInModifiedClassMappingInfo.put(classMapping.getClassName(), classMapping);
91 | }
92 | }
93 | } catch (IOException ioe) {
94 | ioe.printStackTrace();
95 | } finally {
96 | try {
97 | if (is != null) {
98 | is.close();
99 | }
100 | } catch (IOException e) {
101 | e.printStackTrace();
102 | }
103 | }
104 | }
105 |
106 | public ClassMapping getClassMapping(String classname) {
107 | return usedInModifiedClassMappingInfo.get(classname);
108 | }
109 |
110 | public void setClassMapping(String classname, ClassMapping classMapping) {
111 | usedInModifiedClassMappingInfo.put(classname, classMapping);
112 | }
113 |
114 | public ClassMapping getClassMappingOrDefault(String classname) {
115 | ClassMapping defaultClassMapping = new ClassMapping();
116 | if (!Config.supportProGuard) {
117 | defaultClassMapping.setValueName(classname);
118 | }
119 | return usedInModifiedClassMappingInfo.getOrDefault(classname, defaultClassMapping);
120 | }
121 |
122 | /***
123 | * @param returnTypeWithNumber
124 | * @param methodSignure
125 | * @return returnType+" "+methodSignure,just one blank
126 | */
127 |
128 | public String getMethodSigureWithReturnTypeInMapping(String returnTypeWithNumber, String methodSignure) {
129 | //初步观察mapping文件，使用":"来截取返回值，还可以通过寻找第一个字符，
130 | if (methodSignure.contains(":")) {
131 | //兼容R8
132 | return getMethodSignureWithReturnType(returnTypeWithNumber.substring(returnTypeWithNumber.lastIndexOf(":") + 1), methodSignure.substring(0, methodSignure.indexOf(":")));
133 | }
134 | return getMethodSignureWithReturnType(returnTypeWithNumber.substring(returnTypeWithNumber.lastIndexOf(":") + 1), methodSignure);
135 | }
136 |
137 | public String getMethodSignureWithReturnType(String returnType, String methodSignure) {
138 | //只有一个空格
139 | return returnType + " " + methodSignure;
140 | }
141 | }
--------------------------------------------------------------------------------
/auto-patch-plugin/src/main/resources/META-INF/gradle-plugins/auto-patch-plugin.properties:
--------------------------------------------------------------------------------
1 | implementation-class=robust.gradle.plugin.AutoPatchTransform
--------------------------------------------------------------------------------
/auto-patch-plugin/src/main/resources/libs/baksmali-2.1.2.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Meituan-Dianping/Robust/955adcc21e4fbcb52054a8f7f4bbb11f462aeb2f/auto-patch-plugin/src/main/resources/libs/baksmali-2.1.2.jar
--------------------------------------------------------------------------------
/auto-patch-plugin/src/main/resources/libs/dx.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Meituan-Dianping/Robust/955adcc21e4fbcb52054a8f7f4bbb11f462aeb2f/auto-patch-plugin/src/main/resources/libs/dx.jar
--------------------------------------------------------------------------------
/auto-patch-plugin/src/main/resources/libs/smali-2.1.2.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Meituan-Dianping/Robust/955adcc21e4fbcb52054a8f7f4bbb11f462aeb2f/auto-patch-plugin/src/main/resources/libs/smali-2.1.2.jar
--------------------------------------------------------------------------------
/autopatchbase/build.gradle:
--------------------------------------------------------------------------------
1 | apply plugin: 'java'
2 | apply from: '../gradle_mvn_push.gradle'
3 | sourceCompatibility= 1.7
4 | targetCompatibility= 1.7
5 |
--------------------------------------------------------------------------------
/autopatchbase/src/main/java/com/meituan/robust/ChangeQuickRedirect.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust;
2 |
3 | /**
4 | * Created by c_kunwu on 16/5/10.
5 | */
6 | public interface ChangeQuickRedirect {
7 | Object accessDispatch(String methodName, Object[] paramArrayOfObject);
8 |
9 | boolean isSupport(String methodName, Object[] paramArrayOfObject);
10 | }
11 |
--------------------------------------------------------------------------------
/autopatchbase/src/main/java/com/meituan/robust/Constants.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust;
2 |
3 | import java.util.Arrays;
4 | import java.util.HashSet;
5 | import java.util.List;
6 | import java.util.Set;
7 |
8 | /**
9 | * Created by mivanzhang on 16/11/3.
10 | */
11 |
12 | public class Constants {
13 |
14 | public static final String ORIGINCLASS = "originClass";
15 | public static final String MODIFY_ANNOTATION = "com.meituan.robust.patch.annotaion.Modify";
16 | // public static final String MODIFY_ANNOTATION = Modify.class.getCanonicalName();
17 | public static final String ADD_ANNOTATION = "com.meituan.robust.patch.annotaion.Add";
18 | // public static final String ADD_ANNOTATION = Add.class.getCanonicalName();
19 | public static final String LAMBDA_MODIFY = "com.meituan.robust.patch.RobustModify";
20 | // public static final String LAMBDA_MODIFY = RobustModify.class.getCanonicalName();
21 |
22 | public static final String PATCH_TEMPLATE_FULL_NAME = "com.meituan.robust.utils.PatchTemplate";
23 |
24 |
25 | public static final String ZIP_FILE_NAME = "meituan.jar";
26 | public static final String PATACH_DEX_NAME = "patch.dex";
27 | public static final String CLASSES_DEX_NAME = "classes.dex";
28 | public static final String PATACH_JAR_NAME = "patch.jar";
29 | public static final String PATCH_SUFFIX = "Patch";
30 | public static final String PATCH_CONTROL_SUFFIX = "Control";
31 | public static final String INLINE_PATCH_SUFFIX = "InLinePatch";
32 | public static final String STATICFLAG = "staticRobust";
33 | public static final String ROBUST_ASSIST_SUFFIX = "RobustAssist";
34 | public static final String ROBUST_PUBLIC_SUFFIX = "RobustPublic";
35 | public static final String GET_REAL_PARAMETER = "getRealParameter";
36 | public static final String ROBUST_UTILS_FULL_NAME = "com.meituan.robust.utils.EnhancedRobustUtils";
37 |
38 | public static final String ROBUST_GENERATE_DIRECTORY = "outputs/robust";
39 |
40 | //FILE_MD5_PATH is a copy from RobustTransform.FILE_MD5_PATH ,please make sure the two is the same
41 | public static final String METHOD_MAP_PATH = "/robust/methodsMap.robust";
42 | public static final String DEFAULT_MAPPING_FILE = "/robust/mapping.txt";
43 |
44 | public static final String SMALI_INVOKE_SUPER_COMMAND = "invoke-super";
45 | public static final String SMALI_INVOKE_VIRTUAL_COMMAND = "invoke-virtual";
46 |
47 | public static Class ModifyAnnotationClass = null;
48 | public static Class AddAnnotationClass = null;
49 |
50 | public final static String[] LIB_NAME_ARRAY = {"baksmali-2.1.2.jar", "smali-2.1.2.jar", "dx.jar"};
51 | public static final String PACKNAME_END = ";";
52 | public final static String PRIMITIVE_TYPE = "ZCBSIJFDV";
53 | public final static String ARRAY_TYPE = "[";
54 | public final static char OBJECT_TYPE = 'L';
55 | public static final String PACKNAME_START = String.valueOf(OBJECT_TYPE);
56 | public static final Boolean OBSCURE = true;
57 | // public static final Boolean OBSCURE = false;
58 | // public static final Boolean isLogging = false;
59 | public static boolean isLogging = true;
60 |
61 | public static final String PATCH_PACKAGENAME = "com.meituan.robust.patch";
62 | public static final Set RFileClassSet = new HashSet();
63 | public final static String ROBUST_XML = "robust.xml";
64 |
65 | static {
66 | RFileClassSet.add("R$array");
67 | RFileClassSet.add("R$xml");
68 | RFileClassSet.add("R$styleable");
69 | RFileClassSet.add("R$style");
70 | RFileClassSet.add("R$string");
71 | RFileClassSet.add("R$raw");
72 | RFileClassSet.add("R$menu");
73 | RFileClassSet.add("R$layout");
74 | RFileClassSet.add("R$integer");
75 | RFileClassSet.add("R$id");
76 | RFileClassSet.add("R$drawable");
77 | RFileClassSet.add("R$dimen");
78 | RFileClassSet.add("R$color");
79 | RFileClassSet.add("R$bool");
80 | RFileClassSet.add("R$attr");
81 | RFileClassSet.add("R$anim");
82 | }
83 |
84 | //=========================RobustTransForm========================
85 | //=========================RobustTransForm========================
86 | public static final String CONSTRUCTOR = "Constructor";
87 | public static final String LANG_VOID = "java.lang.Void";
88 | public static final String VOID = "void";
89 | public static final String LANG_BOOLEAN = "java.lang.Boolean";
90 | public static final String BOOLEAN = "boolean";
91 | public static final String LANG_INT = "java.lang.Integer";
92 | public static final String INT = "int";
93 | public static final String LANG_LONG = "java.lang.Long";
94 | public static final String LONG = "long";
95 | public static final String LANG_DOUBLE = "java.lang.Double";
96 | public static final String DOUBLE = "double";
97 | public static final String LANG_FLOAT = "java.lang.Float";
98 | public static final String FLOAT = "float";
99 | public static final String LANG_SHORT = "java.lang.Short";
100 | public static final String SHORT = "short";
101 | public static final String LANG_BYTE = "java.lang.Byte";
102 | public static final String BYTE = "byte";
103 | public static final String LANG_CHARACTER = "Character";
104 | public static final String CHAR = "char";
105 |
106 | public static final String METHOD_MAP_OUT_PATH = "/outputs/robust/methodsMap.robust";
107 | public static final String INTERFACE_NAME = "com.meituan.robust.ChangeQuickRedirect";
108 | public static final String INSERT_FIELD_NAME = "changeQuickRedirect";
109 | public static final List NO_NEED_REFLECT_CLASS = Arrays.asList("android.os.Bundle","android.os.BaseBundle");
110 |
111 | //robust apk hash : apk's unique id
112 | public static final String ROBUST_APK_HASH_FILE_NAME = "robust.apkhash";
113 | public static final String ASPECTJ_AROUND_CLASS = "org.aspectj.runtime.internal.AroundClosure";
114 | public static final String PATCH_EXECUTE = "patch execute ,other extension will be ignore ";
115 |
116 |
117 |
118 | }
119 |
--------------------------------------------------------------------------------
/autopatchbase/src/main/java/com/meituan/robust/RobustArguments.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust;
2 |
3 | /**
4 | * Created by zhangmeng on 2017/5/9.
5 | */
6 |
7 | public class RobustArguments {
8 | public Object[] paramsArray;
9 | public Object current;
10 | public boolean isStatic;
11 | public int methodNumber;
12 | public Class[] paramsClassTypes;
13 | public Class returnType;
14 |
15 | public RobustArguments(Object[] paramsArray, Object current, boolean isStatic, int methodNumber, Class[] paramsClassTypes, Class returnType) {
16 | this.paramsArray = paramsArray;
17 | this.current = current;
18 | this.isStatic = isStatic;
19 | this.methodNumber = methodNumber;
20 | this.paramsClassTypes = paramsClassTypes;
21 | this.returnType = returnType;
22 | }
23 | }
24 |
--------------------------------------------------------------------------------
/autopatchbase/src/main/java/com/meituan/robust/RobustExtension.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust;
2 |
3 | /**
4 | * Created by zhangmeng on 2017/4/21.
5 | */
6 |
7 | public interface RobustExtension {
8 | /**
9 | *
10 | * @return 关于接入方信息的描述，请保证各个业务方独立不同，且唯一不变。
11 | * 建议使用各个业务方包名+一些功能性描述，比如：com.meituan.robust android热更新系统
12 | */
13 | String describeSelfFunction();
14 |
15 | /**
16 | * 通知监听者信息，通知哪个监听者被执行了
17 | * @param msg，msg是describeSelfFunction中的返回内容
18 | */
19 | void notifyListner(String msg);
20 | /**
21 | * @param paramsArray 原方法的参数列表
22 | * @param current 当前对象的引用，即this对象，如果是static方法，值为null
23 | * @param methodNumber 方法的唯一编号
24 | * @param paramsTypeArray 方法参数类型列表
25 | * @param returnType 方法的返回值类型
26 | * 在方法的方法体最先执行的代码逻辑，请注意这个方法执行之后，原方法体中的其他逻辑不在执行
27 | * @return
28 | *
29 | */
30 | Object accessDispatch(RobustArguments robustArguments);
31 |
32 | /**
33 | *@param paramsArray 原方法的参数列表
34 | * @param current 当前对象的引用，即this对象，如果是static方法，值为null
35 | * @param methodNumber 方法的唯一编号
36 | * @param paramsTypeArray 方法参数类型列表
37 | * @param returnType 方法的返回值类型
38 | *
39 | * @return return true 代表不继续执行原有方法体，只执行accessDispatch方法的逻辑，并把accessDispatch的方法返回值作为原函数的返回值
40 | * return false 代表执行原有方法体，但是可以在isSupport方法里面添加额外的逻辑，比如说记录当前的方法调用栈或者日志等
41 | */
42 | boolean isSupport(RobustArguments robustArguments);
43 |
44 | }
45 |
--------------------------------------------------------------------------------
/autopatchbase/src/main/java/com/meituan/robust/patch/RobustModify.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust.patch;
2 |
3 | /**
4 | * Created by mivanzhang on 16/12/9.
5 | *
6 | * A backup for annotation Modify, in some situation Modify will not work, such as Generic
7 | */
8 |
9 | public final class RobustModify {
10 | public static final void modify() {
11 |
12 | }
13 | }
14 |
--------------------------------------------------------------------------------
/autopatchbase/src/main/java/com/meituan/robust/patch/annotaion/Add.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust.patch.annotaion;
2 |
3 | import java.lang.annotation.Documented;
4 | import java.lang.annotation.ElementType;
5 | import java.lang.annotation.Retention;
6 | import java.lang.annotation.RetentionPolicy;
7 | import java.lang.annotation.Target;
8 |
9 | /**
10 | * Created by mivanzhang on 16/12/19.
11 | * 用来标记新增的类和方法
12 | * annotaion used for add classes or methods,classes and methods will be packed into patch.jar/patch.apk
13 | */
14 |
15 | @Target({ElementType.FIELD, ElementType.METHOD, ElementType.TYPE, ElementType.CONSTRUCTOR})
16 | @Retention(RetentionPolicy.CLASS)
17 | @Documented
18 | public @interface Add {
19 | String value() default "";
20 | }
--------------------------------------------------------------------------------
/autopatchbase/src/main/java/com/meituan/robust/patch/annotaion/Modify.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust.patch.annotaion;
2 |
3 | import java.lang.annotation.Documented;
4 | import java.lang.annotation.ElementType;
5 | import java.lang.annotation.Retention;
6 | import java.lang.annotation.RetentionPolicy;
7 | import java.lang.annotation.Target;
8 |
9 | /**
10 | * Created by mivanzhang on 16/12/9.
11 | * annotaion used for modify classes or methods,classes and methods will be packed into patch.jar/patch.apk
12 | */
13 | @Target({ElementType.FIELD, ElementType.METHOD, ElementType.TYPE, ElementType.CONSTRUCTOR})
14 | @Retention(RetentionPolicy.CLASS)
15 | @Documented
16 | public @interface Modify {
17 | String value() default "";
18 | }
19 |
--------------------------------------------------------------------------------
/autopatchbase/src/main/java/com/meituan/robust/utils/PatchTemplate.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust.utils;
2 |
3 |
4 | import com.meituan.robust.ChangeQuickRedirect;
5 |
6 | import java.util.Map;
7 | import java.util.WeakHashMap;
8 |
9 | /**
10 | * Created by mivanzhang on 16/7/26.
11 | */
12 | public class PatchTemplate implements ChangeQuickRedirect {
13 | public static final String MATCH_ALL_PARAMETER = "(\\w*\\.)*\\w*";
14 |
15 | public PatchTemplate() {
16 | }
17 |
18 | private static final Map keyToValueRelation = new WeakHashMap<>();
19 |
20 | @Override
21 | public Object accessDispatch(String methodName, Object[] paramArrayOfObject) {
22 | return null;
23 | }
24 |
25 | @Override
26 | public boolean isSupport(String methodName, Object[] paramArrayOfObject) {
27 | return true;
28 | }
29 |
30 | //解决boolean被优化成byte的问题
31 | private static Object fixObj(Object booleanObj) {
32 | if (booleanObj instanceof Byte) {
33 | byte byteValue = (Byte) booleanObj;
34 | boolean booleanValue = byteValue != 0x00;
35 | return new Boolean(booleanValue);
36 | }
37 | return booleanObj;
38 | }
39 | }
40 |
--------------------------------------------------------------------------------
/build.gradle:
--------------------------------------------------------------------------------
1 | // Top-level build file where you can add configuration options common to all sub-projects/modules.
2 |
3 | buildscript {
4 | repositories {
5 | mavenLocal()
6 | jcenter()
7 | }
8 | dependencies {
9 | classpath 'com.android.tools.build:gradle:2.1.0'
10 | // NOTE: Do not place your application dependencies here; they belong
11 | // in the individual module build.gradle files
12 | classpath 'com.meituan.robust:gradle-plugin:0.4.96'
13 | classpath 'com.meituan.robust:auto-patch-plugin:0.4.96'
14 | classpath 'me.tatarka:gradle-retrolambda:3.2.0'
15 | classpath 'com.github.dcendents:android-maven-gradle-plugin:1.3'
16 | classpath 'com.jfrog.bintray.gradle:gradle-bintray-plugin:1.7.3'
17 | classpath 'me.tatarka.retrolambda.projectlombok:lombok.ast:0.2.3.a2'
18 | }
19 | configurations.classpath.exclude group: 'com.android.tools.external.lombok'
20 | }
21 | configurations.all {
22 | resolutionStrategy.cacheChangingModulesFor 0, 'seconds'
23 | }
24 | allprojects {
25 | repositories {
26 | mavenLocal()
27 | maven {
28 | url "http://depot.sankuai.com/nexus/content/groups/public/"
29 | }
30 | jcenter()
31 | }
32 | tasks.withType(JavaCompile) { options.encoding = "UTF-8" }
33 | }
34 |
35 | task clean(type: Delete) {
36 | delete rootProject.buildDir
37 | }
38 |
--------------------------------------------------------------------------------
/gradle-plugin/build.gradle:
--------------------------------------------------------------------------------
1 | apply plugin: 'groovy'
2 | compileGroovy {
3 | sourceCompatibility = 1.7
4 | targetCompatibility = 1.7
5 | }
6 | dependencies {
7 | compile gradleApi()
8 | compile localGroovy()
9 | compile 'com.android.tools.build:gradle:2.1.0'
10 | compile 'org.javassist:javassist:3.20.0-GA'
11 | compile fileTree(dir: "./src/main/libs", include: ['*.jar'])
12 | compile project(':autopatchbase')
13 | // compile 'com.meituan.robust:autopatchbase:0.4.93'
14 | compile 'org.ow2.asm:asm:5.0.1'
15 | }
16 |
17 | apply from: '../gradle_mvn_push.gradle'
18 |
--------------------------------------------------------------------------------
/gradle-plugin/gradle/wrapper/gradle-wrapper.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Meituan-Dianping/Robust/955adcc21e4fbcb52054a8f7f4bbb11f462aeb2f/gradle-plugin/gradle/wrapper/gradle-wrapper.jar
--------------------------------------------------------------------------------
/gradle-plugin/gradle/wrapper/gradle-wrapper.properties:
--------------------------------------------------------------------------------
1 | #Wed Dec 16 15:22:26 CST 2015
2 | distributionBase=GRADLE_USER_HOME
3 | distributionPath=wrapper/dists
4 | zipStoreBase=GRADLE_USER_HOME
5 | zipStorePath=wrapper/dists
6 | distributionUrl=https\://services.gradle.org/distributions/gradle-2.10-all.zip
7 |
--------------------------------------------------------------------------------
/gradle-plugin/gradlew:
--------------------------------------------------------------------------------
1 | #!/usr/bin/env bash
2 |
3 | ##############################################################################
4 | ##
5 | ## Gradle start up script for UN*X
6 | ##
7 | ##############################################################################
8 |
9 | # Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
10 | DEFAULT_JVM_OPTS=""
11 |
12 | APP_NAME="Gradle"
13 | APP_BASE_NAME=`basename "$0"`
14 |
15 | # Use the maximum available, or set MAX_FD != -1 to use that value.
16 | MAX_FD="maximum"
17 |
18 | warn ( ) {
19 | echo "$*"
20 | }
21 |
22 | die ( ) {
23 | echo
24 | echo "$*"
25 | echo
26 | exit 1
27 | }
28 |
29 | # OS specific support (must be 'true' or 'false').
30 | cygwin=false
31 | msys=false
32 | darwin=false
33 | case "`uname`" in
34 | CYGWIN* )
35 | cygwin=true
36 | ;;
37 | Darwin* )
38 | darwin=true
39 | ;;
40 | MINGW* )
41 | msys=true
42 | ;;
43 | esac
44 |
45 | # For Cygwin, ensure paths are in UNIX format before anything is touched.
46 | if $cygwin ; then
47 | [ -n "$JAVA_HOME" ] && JAVA_HOME=`cygpath --unix "$JAVA_HOME"`
48 | fi
49 |
50 | # Attempt to set APP_HOME
51 | # Resolve links: $0 may be a link
52 | PRG="$0"
53 | # Need this for relative symlinks.
54 | while [ -h "$PRG" ] ; do
55 | ls=`ls -ld "$PRG"`
56 | link=`expr "$ls" : '.*-> \(.*\)$'`
57 | if expr "$link" : '/.*' > /dev/null; then
58 | PRG="$link"
59 | else
60 | PRG=`dirname "$PRG"`"/$link"
61 | fi
62 | done
63 | SAVED="`pwd`"
64 | cd "`dirname \"$PRG\"`/" >&-
65 | APP_HOME="`pwd -P`"
66 | cd "$SAVED" >&-
67 |
68 | CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
69 |
70 | # Determine the Java command to use to start the JVM.
71 | if [ -n "$JAVA_HOME" ] ; then
72 | if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
73 | # IBM's JDK on AIX uses strange locations for the executables
74 | JAVACMD="$JAVA_HOME/jre/sh/java"
75 | else
76 | JAVACMD="$JAVA_HOME/bin/java"
77 | fi
78 | if [ ! -x "$JAVACMD" ] ; then
79 | die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME
80 |
81 | Please set the JAVA_HOME variable in your environment to match the
82 | location of your Java installation."
83 | fi
84 | else
85 | JAVACMD="java"
86 | which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
87 |
88 | Please set the JAVA_HOME variable in your environment to match the
89 | location of your Java installation."
90 | fi
91 |
92 | # Increase the maximum file descriptors if we can.
93 | if [ "$cygwin" = "false" -a "$darwin" = "false" ] ; then
94 | MAX_FD_LIMIT=`ulimit -H -n`
95 | if [ $? -eq 0 ] ; then
96 | if [ "$MAX_FD" = "maximum" -o "$MAX_FD" = "max" ] ; then
97 | MAX_FD="$MAX_FD_LIMIT"
98 | fi
99 | ulimit -n $MAX_FD
100 | if [ $? -ne 0 ] ; then
101 | warn "Could not set maximum file descriptor limit: $MAX_FD"
102 | fi
103 | else
104 | warn "Could not query maximum file descriptor limit: $MAX_FD_LIMIT"
105 | fi
106 | fi
107 |
108 | # For Darwin, add options to specify how the application appears in the dock
109 | if $darwin; then
110 | GRADLE_OPTS="$GRADLE_OPTS \"-Xdock:name=$APP_NAME\" \"-Xdock:icon=$APP_HOME/media/gradle.icns\""
111 | fi
112 |
113 | # For Cygwin, switch paths to Windows format before running java
114 | if $cygwin ; then
115 | APP_HOME=`cygpath --path --mixed "$APP_HOME"`
116 | CLASSPATH=`cygpath --path --mixed "$CLASSPATH"`
117 |
118 | # We build the pattern for arguments to be converted via cygpath
119 | ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null`
120 | SEP=""
121 | for dir in $ROOTDIRSRAW ; do
122 | ROOTDIRS="$ROOTDIRS$SEP$dir"
123 | SEP="|"
124 | done
125 | OURCYGPATTERN="(^($ROOTDIRS))"
126 | # Add a user-defined pattern to the cygpath arguments
127 | if [ "$GRADLE_CYGPATTERN" != "" ] ; then
128 | OURCYGPATTERN="$OURCYGPATTERN|($GRADLE_CYGPATTERN)"
129 | fi
130 | # Now convert the arguments - kludge to limit ourselves to /bin/sh
131 | i=0
132 | for arg in "$@" ; do
133 | CHECK=`echo "$arg"|egrep -c "$OURCYGPATTERN" -`
134 | CHECK2=`echo "$arg"|egrep -c "^-"` ### Determine if an option
135 |
136 | if [ $CHECK -ne 0 ] && [ $CHECK2 -eq 0 ] ; then ### Added a condition
137 | eval `echo args$i`=`cygpath --path --ignore --mixed "$arg"`
138 | else
139 | eval `echo args$i`="\"$arg\""
140 | fi
141 | i=$((i+1))
142 | done
143 | case $i in
144 | (0) set -- ;;
145 | (1) set -- "$args0" ;;
146 | (2) set -- "$args0" "$args1" ;;
147 | (3) set -- "$args0" "$args1" "$args2" ;;
148 | (4) set -- "$args0" "$args1" "$args2" "$args3" ;;
149 | (5) set -- "$args0" "$args1" "$args2" "$args3" "$args4" ;;
150 | (6) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" ;;
151 | (7) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" ;;
152 | (8) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" ;;
153 | (9) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" "$args8" ;;
154 | esac
155 | fi
156 |
157 | # Split up the JVM_OPTS And GRADLE_OPTS values into an array, following the shell quoting and substitution rules
158 | function splitJvmOpts() {
159 | JVM_OPTS=("$@")
160 | }
161 | eval splitJvmOpts $DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS
162 | JVM_OPTS[${#JVM_OPTS[*]}]="-Dorg.gradle.appname=$APP_BASE_NAME"
163 |
164 | exec "$JAVACMD" "${JVM_OPTS[@]}" -classpath "$CLASSPATH" org.gradle.wrapper.GradleWrapperMain "$@"
165 |
--------------------------------------------------------------------------------
/gradle-plugin/gradlew.bat:
--------------------------------------------------------------------------------
1 | @if "%DEBUG%" == "" @echo off
2 | @rem ##########################################################################
3 | @rem
4 | @rem Gradle startup script for Windows
5 | @rem
6 | @rem ##########################################################################
7 |
8 | @rem Set local scope for the variables with windows NT shell
9 | if "%OS%"=="Windows_NT" setlocal
10 |
11 | @rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
12 | set DEFAULT_JVM_OPTS=
13 |
14 | set DIRNAME=%~dp0
15 | if "%DIRNAME%" == "" set DIRNAME=.
16 | set APP_BASE_NAME=%~n0
17 | set APP_HOME=%DIRNAME%
18 |
19 | @rem Find java.exe
20 | if defined JAVA_HOME goto findJavaFromJavaHome
21 |
22 | set JAVA_EXE=java.exe
23 | %JAVA_EXE% -version >NUL 2>&1
24 | if "%ERRORLEVEL%" == "0" goto init
25 |
26 | echo.
27 | echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
28 | echo.
29 | echo Please set the JAVA_HOME variable in your environment to match the
30 | echo location of your Java installation.
31 |
32 | goto fail
33 |
34 | :findJavaFromJavaHome
35 | set JAVA_HOME=%JAVA_HOME:"=%
36 | set JAVA_EXE=%JAVA_HOME%/bin/java.exe
37 |
38 | if exist "%JAVA_EXE%" goto init
39 |
40 | echo.
41 | echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME%
42 | echo.
43 | echo Please set the JAVA_HOME variable in your environment to match the
44 | echo location of your Java installation.
45 |
46 | goto fail
47 |
48 | :init
49 | @rem Get command-line arguments, handling Windowz variants
50 |
51 | if not "%OS%" == "Windows_NT" goto win9xME_args
52 | if "%@eval[2+2]" == "4" goto 4NT_args
53 |
54 | :win9xME_args
55 | @rem Slurp the command line arguments.
56 | set CMD_LINE_ARGS=
57 | set _SKIP=2
58 |
59 | :win9xME_args_slurp
60 | if "x%~1" == "x" goto execute
61 |
62 | set CMD_LINE_ARGS=%*
63 | goto execute
64 |
65 | :4NT_args
66 | @rem Get arguments from the 4NT Shell from JP Software
67 | set CMD_LINE_ARGS=%$
68 |
69 | :execute
70 | @rem Setup the command line
71 |
72 | set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar
73 |
74 | @rem Execute Gradle
75 | "%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %CMD_LINE_ARGS%
76 |
77 | :end
78 | @rem End local scope for the variables with windows NT shell
79 | if "%ERRORLEVEL%"=="0" goto mainEnd
80 |
81 | :fail
82 | rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of
83 | rem the _cmd.exe /c_ return code!
84 | if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1
85 | exit /b 1
86 |
87 | :mainEnd
88 | if "%OS%"=="Windows_NT" endlocal
89 |
90 | :omega
91 |
--------------------------------------------------------------------------------
/gradle-plugin/src/main/groovy/robust/gradle/plugin/ConvertUtils.groovy:
--------------------------------------------------------------------------------
1 | package robust.gradle.plugin
2 |
3 | import com.android.SdkConstants
4 | import com.android.build.api.transform.TransformInput
5 | import javassist.ClassPool
6 | import javassist.CtClass
7 |
8 | import java.util.jar.JarEntry
9 | import java.util.jar.JarFile
10 | import java.util.regex.Matcher
11 | /**
12 | * Created by mivanzhang on 16/11/3.
13 | */
14 | class ConvertUtils {
15 | static List toCtClasses(Collection inputs, ClassPool classPool) {
16 | List classNames = new ArrayList<>()
17 | List allClass = new ArrayList<>();
18 | def startTime = System.currentTimeMillis()
19 | inputs.each {
20 | it.directoryInputs.each {
21 | def dirPath = it.file.absolutePath
22 | classPool.insertClassPath(it.file.absolutePath)
23 | org.apache.commons.io.FileUtils.listFiles(it.file, null, true).each {
24 | if (it.absolutePath.endsWith(SdkConstants.DOT_CLASS)) {
25 | def className = it.absolutePath.substring(dirPath.length() + 1, it.absolutePath.length() - SdkConstants.DOT_CLASS.length()).replaceAll(Matcher.quoteReplacement(File.separator), '.')
26 | if(classNames.contains(className)){
27 | throw new RuntimeException("You have duplicate classes with the same name : "+className+" please remove duplicate classes ")
28 | }
29 | classNames.add(className)
30 | }
31 | }
32 | }
33 |
34 | it.jarInputs.each {
35 | classPool.insertClassPath(it.file.absolutePath)
36 | def jarFile = new JarFile(it.file)
37 | Enumeration classes = jarFile.entries();
38 | while (classes.hasMoreElements()) {
39 | JarEntry libClass = classes.nextElement();
40 | String className = libClass.getName();
41 | if (className.endsWith(SdkConstants.DOT_CLASS)) {
42 | className = className.substring(0, className.length() - SdkConstants.DOT_CLASS.length()).replaceAll('/', '.')
43 | if(classNames.contains(className)){
44 | throw new RuntimeException("You have duplicate classes with the same name : "+className+" please remove duplicate classes ")
45 | }
46 | classNames.add(className)
47 | }
48 | }
49 | }
50 | }
51 | def cost = (System.currentTimeMillis() - startTime) / 1000
52 | println "read all class file cost $cost second"
53 | classNames.each {
54 | try {
55 | allClass.add(classPool.get(it));
56 | } catch (javassist.NotFoundException e) {
57 | println "class not found exception class name: $it "
58 |
59 | }
60 |
61 | }
62 |
63 | Collections.sort(allClass, new Comparator() {
64 | @Override
65 | int compare(CtClass class1, CtClass class2) {
66 | return class1.getName() <=> class2.getName();
67 | }
68 | });
69 | return allClass;
70 | }
71 |
72 |
73 | }
--------------------------------------------------------------------------------
/gradle-plugin/src/main/groovy/robust/gradle/plugin/InsertcodeStrategy.java:
--------------------------------------------------------------------------------
1 | package robust.gradle.plugin;
2 |
3 | import java.io.File;
4 | import java.io.IOException;
5 | import java.util.ArrayList;
6 | import java.util.HashMap;
7 | import java.util.LinkedHashMap;
8 | import java.util.List;
9 | import java.util.concurrent.atomic.AtomicInteger;
10 | import java.util.zip.ZipEntry;
11 | import java.util.zip.ZipOutputStream;
12 |
13 | import javassist.CannotCompileException;
14 | import javassist.CtClass;
15 | import javassist.NotFoundException;
16 |
17 | /**
18 | * Created by zhangmeng on 2017/5/10.
19 | */
20 |
21 | public abstract class InsertcodeStrategy {
22 | //packnames need to be insert code 需要插桩的包名列表，
23 | protected List hotfixPackageList = new ArrayList<>();
24 | //methods list need to insert code 需要插桩的方法列表
25 | protected List hotfixMethodList = new ArrayList<>();
26 |
27 | //packnames don`t need to be insert code 不需要插桩的包名列表，
28 | protected List exceptPackageList = new ArrayList<>();
29 |
30 | //methods list do not need to insert code 不需要插桩的方法列表
31 | protected List exceptMethodList = new ArrayList<>();
32 | //a switch control whether need to filter method in hotfixMethodList, if false ,hotfixMethodList will be ignored
33 | protected boolean isHotfixMethodLevel = false;
34 |
35 | //a switch control whether need to filter method in exceptMethodList, if false ,exceptMethodList will be ignored
36 | protected boolean isExceptMethodLevel = false;
37 |
38 | //a switch control whether need to insert code into lambda function
39 | protected boolean isForceInsertLambda = false;
40 |
41 | protected AtomicInteger insertMethodCount = new AtomicInteger(0);
42 | //record every method with unique method number, use LinkedHashMap to keep order for printing
43 | public HashMap methodMap = new LinkedHashMap<>();
44 |
45 | public InsertcodeStrategy(List hotfixPackageList, List hotfixMethodList, List exceptPackageList, List exceptMethodList, boolean isHotfixMethodLevel, boolean isExceptMethodLevel, boolean isForceInsertLambda) {
46 | this.hotfixPackageList = hotfixPackageList;
47 | this.hotfixMethodList = hotfixMethodList;
48 | this.exceptPackageList = exceptPackageList;
49 | this.exceptMethodList = exceptMethodList;
50 | this.isHotfixMethodLevel = isHotfixMethodLevel;
51 | this.isExceptMethodLevel = isExceptMethodLevel;
52 | this.isForceInsertLambda = isForceInsertLambda;
53 | insertMethodCount.set(0);
54 | }
55 |
56 | /**
57 | * @param box all classes which will be packed into apk,所有需要打入apk中的类
58 | * @param jarFile 所有的插桩处理过的class都会被输出的jarFile
59 | * @throws CannotCompileException
60 | * @throws IOException
61 | * @throws NotFoundException
62 | */
63 | protected abstract void insertCode(List box, File jarFile) throws CannotCompileException, IOException, NotFoundException;
64 |
65 | protected boolean isNeedInsertClass(String className) {
66 |
67 | //这样子可以在需要埋点的剔除指定的类
68 | for (String exceptName : exceptPackageList) {
69 | if (className.startsWith(exceptName)) {
70 | return false;
71 | }
72 | }
73 | for (String name : hotfixPackageList) {
74 | if (className.startsWith(name)) {
75 | return true;
76 | }
77 | }
78 | return false;
79 | }
80 |
81 | protected void zipFile(byte[] classBytesArray, ZipOutputStream zos, String entryName) {
82 | try {
83 | ZipEntry entry = new ZipEntry(entryName);
84 | zos.putNextEntry(entry);
85 | zos.write(classBytesArray, 0, classBytesArray.length);
86 | zos.closeEntry();
87 | zos.flush();
88 | } catch (Exception e) {
89 | e.printStackTrace();
90 | }
91 | }
92 | }
93 |
--------------------------------------------------------------------------------
/gradle-plugin/src/main/groovy/robust/gradle/plugin/RobustApkHashZipUtils.groovy:
--------------------------------------------------------------------------------
1 | package robust.gradle.plugin
2 |
3 | import com.meituan.robust.Constants
4 |
5 | import java.util.zip.*
6 |
7 | /**
8 | * Created by hedex on 17/2/14.
9 | */
10 | class RobustApkHashZipUtils {
11 | static void packZip(File output, List sources) throws IOException {
12 | ZipOutputStream zipOut = new ZipOutputStream(new FileOutputStream(output));
13 | zipOut.setLevel(Deflater.BEST_SPEED);
14 |
15 | List fileList = new LinkedList();
16 | for (Object source : sources) {
17 | if (source instanceof File){
18 | fileList.add(source)
19 | } else if (source instanceof Collection){
20 | fileList.addAll(source)
21 | } else {
22 | System.err.println("packZip source 4" + source.getClass())
23 | }
24 | }
25 |
26 | for (File source : fileList) {
27 | if (source.isDirectory()) {
28 | zipDir(zipOut, "", source);
29 | } else {
30 | zipFile(zipOut, "", source);
31 | }
32 | }
33 | zipOut.flush();
34 | zipOut.close();
35 | }
36 |
37 | private static String buildPath(String path, String file) {
38 | if (path == null || path == "") {
39 | return file;
40 | } else {
41 | return path + "/" + file;
42 | }
43 | }
44 |
45 | private static void zipDir(ZipOutputStream zos, String path, File dir) throws IOException {
46 | if (!dir.canRead()) {
47 | return;
48 | }
49 |
50 | File[] files = dir.listFiles();
51 | path = buildPath(path, dir.getName());
52 |
53 | for (File source : files) {
54 | if (source.isDirectory()) {
55 | zipDir(zos, path, source);
56 | } else {
57 | zipFile(zos, path, source);
58 | }
59 | }
60 |
61 | }
62 |
63 | def static void zipFile(ZipOutputStream zos, String path, File file) throws IOException {
64 | if (!file.canRead()) {
65 | return;
66 | }
67 |
68 | zos.putNextEntry(new ZipEntry(buildPath(path, file.getAbsolutePath())));
69 |
70 | FileInputStream fis = new FileInputStream(file);
71 |
72 | byte[] buffer = new byte[4092];
73 | int byteCount = 0;
74 | while ((byteCount = fis.read(buffer)) != -1) {
75 | zos.write(buffer, 0, byteCount);
76 | System.out.flush();
77 | }
78 |
79 | fis.close();
80 | zos.closeEntry();
81 | }
82 |
83 | def static void addApkHashFile2ApFile(File apFile, File robustHashFile) {
84 | def tempZipFile = new File(apFile.name + "temp", apFile.parentFile);
85 | if (tempZipFile.exists()) {
86 | tempZipFile.delete();
87 | }
88 |
89 | ZipOutputStream zipOutputStream = new ZipOutputStream(new FileOutputStream(tempZipFile))
90 |
91 | //copy ap file
92 | ZipFile apZipFile = new ZipFile(apFile)
93 | final Enumeration entries = apZipFile.entries();
94 | while (entries.hasMoreElements()) {
95 | ZipEntry originZipEntry = entries.nextElement();
96 | ZipEntry rightZipEntry = getRightZipEntry(originZipEntry);
97 | if (null != rightZipEntry) {
98 | addZipEntry(zipOutputStream, rightZipEntry, apZipFile.getInputStream(originZipEntry))
99 | }
100 | }
101 |
102 | //add hash file
103 | String entryName = "assets/" + Constants.ROBUST_APK_HASH_FILE_NAME;
104 | ZipEntry zipEntry = new ZipEntry(entryName);
105 | zipEntry.setMethod(ZipEntry.STORED);
106 | zipEntry.setSize(robustHashFile.length());
107 | zipEntry.setCompressedSize(robustHashFile.length());
108 | zipEntry.setCrc(computeFileCrc32(robustHashFile));
109 |
110 | FileInputStream hashFileInputStream = new FileInputStream(robustHashFile);
111 |
112 | addZipEntry(zipOutputStream, zipEntry, hashFileInputStream)
113 |
114 | hashFileInputStream.close();
115 | zipOutputStream.close()
116 |
117 | apFile.delete()
118 | tempZipFile.renameTo(apFile.getAbsolutePath())
119 | }
120 |
121 | private static ZipEntry getRightZipEntry(ZipEntry originZipEntry){
122 | ZipEntry rightZipEntry = new ZipEntry(originZipEntry.getName());
123 | if (ZipEntry.STORED == originZipEntry.getMethod()) {
124 | rightZipEntry.setMethod(ZipEntry.STORED)
125 | rightZipEntry.setSize(originZipEntry.getSize())
126 | rightZipEntry.setCompressedSize(originZipEntry.getCompressedSize())
127 | rightZipEntry.setCrc(originZipEntry.getCrc())
128 | } else {
129 | rightZipEntry.setMethod(ZipEntry.DEFLATED)
130 | }
131 | if (originZipEntry.getComment() != null) {
132 | rightZipEntry.setComment(originZipEntry.getComment())
133 | }
134 | if (originZipEntry.getCreationTime() != null) {
135 | rightZipEntry.setCreationTime(originZipEntry.getCreationTime())
136 | }
137 | if (originZipEntry.getLastAccessTime() != null) {
138 | rightZipEntry.setLastAccessTime(originZipEntry.getLastAccessTime())
139 | }
140 | if (originZipEntry.getLastModifiedTime() != null) {
141 | rightZipEntry.setLastModifiedTime(originZipEntry.getLastModifiedTime())
142 | }
143 | if (originZipEntry.getTime() > 0) {
144 | rightZipEntry.setTime(originZipEntry.getTime())
145 | }
146 | return rightZipEntry;
147 | }
148 |
149 | private static long computeFileCrc32(File file) throws IOException {
150 | InputStream inputStream = new FileInputStream(file);
151 | CRC32 crc = new CRC32();
152 | int index;
153 | while ((index = inputStream.read()) != -1) {
154 | crc.update(index);
155 | }
156 | return crc.getValue();
157 | }
158 |
159 | /**
160 | * add zip entry
161 | *
162 | * @param zipOutputStream
163 | * @param zipEntry
164 | * @param inputStream
165 | * @throws Exception
166 | */
167 | private
168 | static void addZipEntry(ZipOutputStream zipOutputStream, ZipEntry zipEntry, InputStream inputStream) throws Exception {
169 | try {
170 | zipOutputStream.putNextEntry(zipEntry);
171 | byte[] buffer = new byte[1024];
172 | int length = -1;
173 | while ((length = inputStream.read(buffer, 0, buffer.length)) != -1) {
174 | zipOutputStream.write(buffer, 0, length);
175 | zipOutputStream.flush();
176 | }
177 | } catch (ZipException e) {
178 | // do nothing
179 | } finally {
180 | if (inputStream != null) {
181 | inputStream.close();
182 | }
183 | zipOutputStream.closeEntry();
184 | }
185 | }
186 | }
--------------------------------------------------------------------------------
/gradle-plugin/src/main/groovy/robust/gradle/plugin/RobustTransform.groovy:
--------------------------------------------------------------------------------
1 | package robust.gradle.plugin
2 |
3 | import com.android.build.api.transform.*
4 | import com.android.build.gradle.internal.pipeline.TransformManager
5 | import com.meituan.robust.Constants
6 | import javassist.ClassPool
7 | import org.gradle.api.Plugin
8 | import org.gradle.api.Project
9 | import org.gradle.api.logging.Logger
10 | import robust.gradle.plugin.asm.AsmInsertImpl
11 | import robust.gradle.plugin.javaassist.JavaAssistInsertImpl
12 |
13 | import java.util.zip.GZIPOutputStream
14 | /**
15 | * Created by mivanzhang on 16/11/3.
16 | *
17 | * insert code
18 | *
19 | */
20 |
21 | class RobustTransform extends Transform implements Plugin {
22 | Project project
23 | static Logger logger
24 | private static List hotfixPackageList = new ArrayList<>();
25 | private static List hotfixMethodList = new ArrayList<>();
26 | private static List exceptPackageList = new ArrayList<>();
27 | private static List exceptMethodList = new ArrayList<>();
28 | private static boolean isHotfixMethodLevel = false;
29 | private static boolean isExceptMethodLevel = false;
30 | // private static boolean isForceInsert = true;
31 | private static boolean isForceInsert = false;
32 | // private static boolean useASM = false;
33 | private static boolean useASM = true;
34 | private static boolean isForceInsertLambda = false;
35 |
36 | def robust
37 | InsertcodeStrategy insertcodeStrategy;
38 |
39 | @Override
40 | void apply(Project target) {
41 | project = target
42 | robust = new XmlSlurper().parse(new File("${project.projectDir}/${Constants.ROBUST_XML}"))
43 | logger = project.logger
44 | initConfig()
45 | //isForceInsert 是true的话，则强制执行插入
46 | if (!isForceInsert) {
47 | def taskNames = project.gradle.startParameter.taskNames
48 | def isDebugTask = false;
49 | for (int index = 0; index < taskNames.size(); ++index) {
50 | def taskName = taskNames[index]
51 | logger.debug "input start parameter task is ${taskName}"
52 | //FIXME: assembleRelease下屏蔽Prepare，这里因为还没有执行Task，没法直接通过当前的BuildType来判断，所以直接分析当前的startParameter中的taskname，
53 | //另外这里有一个小坑task的名字不能是缩写必须是全称 例如assembleDebug不能是任何形式的缩写输入
54 | if (taskName.endsWith("Debug") && taskName.contains("Debug")) {
55 | // logger.warn " Don't register robust transform for debug model !!! task is：${taskName}"
56 | isDebugTask = true
57 | break;
58 | }
59 | }
60 | if (!isDebugTask) {
61 | project.android.registerTransform(this)
62 | project.afterEvaluate(new RobustApkHashAction())
63 | logger.quiet "Register robust transform successful !!!"
64 | }
65 | if (null != robust.switch.turnOnRobust && !"true".equals(String.valueOf(robust.switch.turnOnRobust))) {
66 | return;
67 | }
68 | } else {
69 | project.android.registerTransform(this)
70 | project.afterEvaluate(new RobustApkHashAction())
71 | }
72 | }
73 |
74 | def initConfig() {
75 | hotfixPackageList = new ArrayList<>()
76 | hotfixMethodList = new ArrayList<>()
77 | exceptPackageList = new ArrayList<>()
78 | exceptMethodList = new ArrayList<>()
79 | isHotfixMethodLevel = false;
80 | isExceptMethodLevel = false;
81 | /*对文件进行解析*/
82 | for (name in robust.packname.name) {
83 | hotfixPackageList.add(name.text());
84 | }
85 | for (name in robust.exceptPackname.name) {
86 | exceptPackageList.add(name.text());
87 | }
88 | for (name in robust.hotfixMethod.name) {
89 | hotfixMethodList.add(name.text());
90 | }
91 | for (name in robust.exceptMethod.name) {
92 | exceptMethodList.add(name.text());
93 | }
94 |
95 | if (null != robust.switch.filterMethod && "true".equals(String.valueOf(robust.switch.turnOnHotfixMethod.text()))) {
96 | isHotfixMethodLevel = true;
97 | }
98 |
99 | if (null != robust.switch.useAsm && "false".equals(String.valueOf(robust.switch.useAsm.text()))) {
100 | useASM = false;
101 | }else {
102 | //默认使用asm
103 | useASM = true;
104 | }
105 |
106 | if (null != robust.switch.filterMethod && "true".equals(String.valueOf(robust.switch.turnOnExceptMethod.text()))) {
107 | isExceptMethodLevel = true;
108 | }
109 |
110 | if (robust.switch.forceInsert != null && "true".equals(String.valueOf(robust.switch.forceInsert.text())))
111 | isForceInsert = true
112 | else
113 | isForceInsert = false
114 |
115 | if (robust.switch.forceInsertLambda != null && "true".equals(String.valueOf(robust.switch.forceInsertLambda.text())))
116 | isForceInsertLambda = true;
117 | else
118 | isForceInsertLambda = false;
119 | }
120 |
121 | @Override
122 | String getName() {
123 | return "robust"
124 | }
125 |
126 | @Override
127 | Set getInputTypes() {
128 | return TransformManager.CONTENT_CLASS
129 | }
130 |
131 | @Override
132 | Set getScopes() {
133 | return TransformManager.SCOPE_FULL_PROJECT
134 | }
135 |
136 | @Override
137 | boolean isIncremental() {
138 | return false
139 | }
140 |
141 |
142 | @Override
143 | void transform(Context context, Collection inputs, Collection referencedInputs, TransformOutputProvider outputProvider, boolean isIncremental) throws IOException, TransformException, InterruptedException {
144 | logger.quiet '================robust start================'
145 | def startTime = System.currentTimeMillis()
146 | outputProvider.deleteAll()
147 | File jarFile = outputProvider.getContentLocation("main", getOutputTypes(), getScopes(),
148 | Format.JAR);
149 | if(!jarFile.getParentFile().exists()){
150 | jarFile.getParentFile().mkdirs();
151 | }
152 | if(jarFile.exists()){
153 | jarFile.delete();
154 | }
155 |
156 | ClassPool classPool = new ClassPool()
157 | project.android.bootClasspath.each {
158 | classPool.appendClassPath((String) it.absolutePath)
159 | }
160 |
161 | def box = ConvertUtils.toCtClasses(inputs, classPool)
162 | def cost = (System.currentTimeMillis() - startTime) / 1000
163 | // logger.quiet "check all class cost $cost second, class count: ${box.size()}"
164 | if (useASM) {
165 | insertcodeStrategy = new AsmInsertImpl(hotfixPackageList, hotfixMethodList, exceptPackageList, exceptMethodList, isHotfixMethodLevel, isExceptMethodLevel, isForceInsertLambda);
166 | } else {
167 | insertcodeStrategy = new JavaAssistInsertImpl(hotfixPackageList, hotfixMethodList, exceptPackageList, exceptMethodList, isHotfixMethodLevel, isExceptMethodLevel, isForceInsertLambda);
168 | }
169 | insertcodeStrategy.insertCode(box, jarFile);
170 | writeMap2File(insertcodeStrategy.methodMap, Constants.METHOD_MAP_OUT_PATH)
171 |
172 | logger.quiet "===robust print id start==="
173 | for (String method : insertcodeStrategy.methodMap.keySet()) {
174 | int id = insertcodeStrategy.methodMap.get(method);
175 | System.out.println("key is " + method + " value is " + id);
176 | }
177 | logger.quiet "===robust print id end==="
178 |
179 | cost = (System.currentTimeMillis() - startTime) / 1000
180 | logger.quiet "robust cost $cost second"
181 | logger.quiet '================robust end================'
182 | }
183 |
184 | private void writeMap2File(Map map, String path) {
185 | File file = new File(project.buildDir.path + path);
186 | if (!file.exists() && (!file.parentFile.mkdirs() || !file.createNewFile())) {
187 | // logger.error(path + " file create error!!")
188 | }
189 | FileOutputStream fileOut = new FileOutputStream(file);
190 |
191 | ByteArrayOutputStream byteOut = new ByteArrayOutputStream();
192 | ObjectOutputStream objOut = new ObjectOutputStream(byteOut);
193 | objOut.writeObject(map)
194 | //gzip压缩
195 | GZIPOutputStream gzip = new GZIPOutputStream(fileOut);
196 | gzip.write(byteOut.toByteArray())
197 | objOut.close();
198 | gzip.flush();
199 | gzip.close();
200 | fileOut.flush()
201 | fileOut.close()
202 |
203 | }
204 |
205 | }
--------------------------------------------------------------------------------
/gradle-plugin/src/main/groovy/robust/gradle/plugin/asm/AsmInsertImpl.java:
--------------------------------------------------------------------------------
1 | package robust.gradle.plugin.asm;
2 |
3 | import com.android.utils.AsmUtils;
4 | import com.meituan.robust.ChangeQuickRedirect;
5 | import com.meituan.robust.Constants;
6 |
7 | import org.objectweb.asm.ClassReader;
8 | import org.objectweb.asm.ClassVisitor;
9 | import org.objectweb.asm.ClassWriter;
10 | import org.objectweb.asm.MethodVisitor;
11 | import org.objectweb.asm.Opcodes;
12 | import org.objectweb.asm.Type;
13 | import org.objectweb.asm.commons.GeneratorAdapter;
14 | import org.objectweb.asm.tree.AbstractInsnNode;
15 | import org.objectweb.asm.tree.ClassNode;
16 | import org.objectweb.asm.tree.InsnList;
17 | import org.objectweb.asm.tree.MethodNode;
18 |
19 | import java.io.File;
20 | import java.io.FileOutputStream;
21 | import java.io.IOException;
22 | import java.util.ArrayList;
23 | import java.util.HashMap;
24 | import java.util.List;
25 | import java.util.Map;
26 | import java.util.jar.JarOutputStream;
27 | import java.util.zip.ZipOutputStream;
28 |
29 | import javassist.CannotCompileException;
30 | import javassist.CtClass;
31 | import javassist.bytecode.AccessFlag;
32 | import robust.gradle.plugin.InsertcodeStrategy;
33 |
34 |
35 | /**
36 | * Created by zhangmeng on 2017/5/10.
37 | *
38 | * insert code using asm
39 | */
40 |
41 | public class AsmInsertImpl extends InsertcodeStrategy {
42 |
43 |
44 | public AsmInsertImpl(List hotfixPackageList, List hotfixMethodList, List exceptPackageList, List exceptMethodList, boolean isHotfixMethodLevel, boolean isExceptMethodLevel, boolean isForceInsertLambda) {
45 | super(hotfixPackageList, hotfixMethodList, exceptPackageList, exceptMethodList, isHotfixMethodLevel, isExceptMethodLevel, isForceInsertLambda);
46 | }
47 |
48 | @Override
49 | protected void insertCode(List box, File jarFile) throws IOException, CannotCompileException {
50 | ZipOutputStream outStream = new JarOutputStream(new FileOutputStream(jarFile));
51 | //get every class in the box ,ready to insert code
52 | for (CtClass ctClass : box) {
53 | //change modifier to public ,so all the class in the apk will be public ,you will be able to access it in the patch
54 | ctClass.setModifiers(AccessFlag.setPublic(ctClass.getModifiers()));
55 | if (isNeedInsertClass(ctClass.getName()) && !(ctClass.isInterface() || ctClass.getDeclaredMethods().length < 1)) {
56 | //only insert code into specific classes
57 | zipFile(transformCode(ctClass.toBytecode(), ctClass.getName().replaceAll("\\.", "/")), outStream, ctClass.getName().replaceAll("\\.", "/") + ".class");
58 | } else {
59 | zipFile(ctClass.toBytecode(), outStream, ctClass.getName().replaceAll("\\.", "/") + ".class");
60 |
61 | }
62 | ctClass.defrost();
63 | }
64 | outStream.close();
65 | }
66 |
67 | private class InsertMethodBodyAdapter extends ClassVisitor implements Opcodes {
68 |
69 | public InsertMethodBodyAdapter() {
70 | super(Opcodes.ASM5);
71 | }
72 |
73 | ClassWriter classWriter;
74 | private String className;
75 | //this maybe change in the future
76 | private Map methodInstructionTypeMap;
77 |
78 | public InsertMethodBodyAdapter(ClassWriter cw, String className, Map methodInstructionTypeMap) {
79 | super(Opcodes.ASM5, cw);
80 | this.classWriter = cw;
81 | this.className = className;
82 | this.methodInstructionTypeMap = methodInstructionTypeMap;
83 | //insert the field
84 | classWriter.visitField(Opcodes.ACC_PUBLIC | Opcodes.ACC_STATIC, Constants.INSERT_FIELD_NAME, Type.getDescriptor(ChangeQuickRedirect.class), null, null);
85 | }
86 |
87 |
88 | @Override
89 | public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
90 | if (isProtect(access)) {
91 | access = setPublic(access);
92 | }
93 | MethodVisitor mv = super.visitMethod(access, name,
94 | desc, signature, exceptions);
95 |
96 | if (!isQualifiedMethod(access, name, desc, methodInstructionTypeMap)) {
97 | return mv;
98 | }
99 | StringBuilder parameters = new StringBuilder();
100 | Type[] types = Type.getArgumentTypes(desc);
101 | for (Type type : types) {
102 | parameters.append(type.getClassName()).append(",");
103 | }
104 | //remove the last ","
105 | if (parameters.length() > 0 && parameters.charAt(parameters.length() - 1) == ',') {
106 | parameters.deleteCharAt(parameters.length() - 1);
107 | }
108 | //record method number
109 | methodMap.put(className.replace('/', '.') + "." + name + "(" + parameters.toString() + ")", insertMethodCount.incrementAndGet());
110 | return new MethodBodyInsertor(mv, className, desc, isStatic(access), String.valueOf(insertMethodCount.get()), name, access);
111 | }
112 |
113 | private boolean isProtect(int access) {
114 | return (access & Opcodes.ACC_PROTECTED) != 0;
115 | }
116 |
117 | private int setPublic(int access) {
118 | return (access & ~(Opcodes.ACC_PRIVATE | Opcodes.ACC_PROTECTED)) | Opcodes.ACC_PUBLIC;
119 | }
120 |
121 | private boolean isQualifiedMethod(int access, String name, String desc, Map c) {
122 | //类初始化函数和构造函数过滤
123 | if (AsmUtils.CLASS_INITIALIZER.equals(name) || AsmUtils.CONSTRUCTOR.equals(name)) {
124 | return false;
125 | }
126 | //@warn 这部分代码请重点review一下，判断条件写错会要命
127 | //这部分代码请重点review一下，判断条件写错会要命
128 | // synthetic 方法暂时不aop 比如AsyncTask 会生成一些同名 synthetic方法,对synthetic 以及private的方法也插入的代码，主要是针对lambda表达式
129 | if (!isForceInsertLambda && ((access & Opcodes.ACC_SYNTHETIC) != 0) && ((access & Opcodes.ACC_PRIVATE) == 0)) {
130 | return false;
131 | }
132 | if ((access & Opcodes.ACC_ABSTRACT) != 0) {
133 | return false;
134 | }
135 | if ((access & Opcodes.ACC_NATIVE) != 0) {
136 | return false;
137 | }
138 | if ((access & Opcodes.ACC_INTERFACE) != 0) {
139 | return false;
140 | }
141 |
142 | if ((access & Opcodes.ACC_DEPRECATED) != 0) {
143 | return false;
144 | }
145 |
146 | //方法过滤
147 | if (isExceptMethodLevel && exceptMethodList != null) {
148 | for (String item : exceptMethodList) {
149 | if (name.matches(item)) {
150 | return false;
151 | }
152 | }
153 | }
154 |
155 | if (isHotfixMethodLevel && hotfixMethodList != null) {
156 | for (String item : hotfixMethodList) {
157 | if (name.matches(item)) {
158 | return true;
159 | }
160 | }
161 | }
162 |
163 | boolean isMethodInvoke = methodInstructionTypeMap.getOrDefault(name + desc, false);
164 | //遍历指令类型，
165 | if (!isMethodInvoke) {
166 | return false;
167 | }
168 |
169 | return !isHotfixMethodLevel;
170 |
171 | }
172 |
173 | class MethodBodyInsertor extends GeneratorAdapter implements Opcodes {
174 | private String className;
175 | private Type[] argsType;
176 | private Type returnType;
177 | List paramsTypeClass = new ArrayList();
178 | boolean isStatic;
179 | //目前methodid是int类型的，未来可能会修改为String类型的，这边进行了一次强转
180 | String methodId;
181 |
182 | public MethodBodyInsertor(MethodVisitor mv, String className, String desc, boolean isStatic, String methodId, String name, int access) {
183 | super(Opcodes.ASM5, mv, access, name, desc);
184 | this.className = className;
185 | this.returnType = Type.getReturnType(desc);
186 | Type[] argsType = Type.getArgumentTypes(desc);
187 | for (Type type : argsType) {
188 | paramsTypeClass.add(type);
189 | }
190 | this.isStatic = isStatic;
191 | this.methodId = methodId;
192 | }
193 |
194 |
195 | @Override
196 | public void visitCode() {
197 | //insert code here
198 | RobustAsmUtils.createInsertCode(this, className, paramsTypeClass, returnType, isStatic, Integer.valueOf(methodId));
199 | }
200 |
201 | }
202 |
203 | private boolean isStatic(int access) {
204 | return (access & Opcodes.ACC_STATIC) != 0;
205 | }
206 |
207 | }
208 |
209 | public byte[] transformCode(byte[] b1, String className) throws IOException {
210 | ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_MAXS);
211 | ClassReader cr = new ClassReader(b1);
212 | ClassNode classNode = new ClassNode();
213 | Map methodInstructionTypeMap = new HashMap<>();
214 | cr.accept(classNode, 0);
215 | final List methods = classNode.methods;
216 | for (MethodNode m : methods) {
217 | InsnList inList = m.instructions;
218 | boolean isMethodInvoke = false;
219 | for (int i = 0; i < inList.size(); i++) {
220 | if (inList.get(i).getType() == AbstractInsnNode.METHOD_INSN) {
221 | isMethodInvoke = true;
222 | }
223 | }
224 | methodInstructionTypeMap.put(m.name + m.desc, isMethodInvoke);
225 | }
226 | InsertMethodBodyAdapter insertMethodBodyAdapter = new InsertMethodBodyAdapter(cw, className, methodInstructionTypeMap);
227 | cr.accept(insertMethodBodyAdapter, ClassReader.EXPAND_FRAMES);
228 | return cw.toByteArray();
229 | }
230 |
231 | }
232 |
--------------------------------------------------------------------------------
/gradle-plugin/src/main/resources/META-INF/gradle-plugins/robust.properties:
--------------------------------------------------------------------------------
1 | implementation-class=robust.gradle.plugin.RobustTransform
--------------------------------------------------------------------------------
/gradle.properties:
--------------------------------------------------------------------------------
1 | # Project-wide Gradle settings.
2 |
3 | # IDE (e.g. Android Studio) users:
4 | # Gradle settings configured through the IDE *will override*
5 | # any settings specified in this file.
6 |
7 | # For more details on how to configure your build environment visit
8 | # http://www.gradle.org/docs/current/userguide/build_environment.html
9 |
10 | # Specifies the JVM arguments used for the daemon process.
11 | # The setting is particularly useful for tweaking memory settings.
12 | # Default value: -Xmx10248m -XX:MaxPermSize=256m
13 | # org.gradle.jvmargs=-Xmx2048m -XX:MaxPermSize=512m -XX:+HeapDumpOnOutOfMemoryError -Dfile.encoding=UTF-8
14 |
15 | # When configured, Gradle will run in incubating parallel mode.
16 | # This option should only be used with decoupled projects. More details, visit
17 | # http://www.gradle.org/docs/current/userguide/multi_project_builds.html#sec:decoupled_projects
18 | # org.gradle.parallel=true
19 | VERSION_NAME=0.4.99
20 | GROUP=com.meituan.robust
21 |
22 |
--------------------------------------------------------------------------------
/gradle/wrapper/gradle-wrapper.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Meituan-Dianping/Robust/955adcc21e4fbcb52054a8f7f4bbb11f462aeb2f/gradle/wrapper/gradle-wrapper.jar
--------------------------------------------------------------------------------
/gradle/wrapper/gradle-wrapper.properties:
--------------------------------------------------------------------------------
1 | #Tue May 16 16:20:24 CST 2017
2 | distributionBase=GRADLE_USER_HOME
3 | distributionPath=wrapper/dists
4 | zipStoreBase=GRADLE_USER_HOME
5 | zipStorePath=wrapper/dists
6 | distributionUrl=https\://services.gradle.org/distributions/gradle-2.10-all.zip
7 |
--------------------------------------------------------------------------------
/gradle_mvn_push.gradle:
--------------------------------------------------------------------------------
1 | apply plugin: 'com.github.dcendents.android-maven'
2 | apply plugin: 'com.jfrog.bintray'
3 | group = GROUP
4 | version = VERSION_NAME
5 |
6 |
7 | def getPropertyFromLocalProperties(key) {
8 | File file = project.rootProject.file('local.properties');
9 | if (file.exists()) {
10 | Properties properties = new Properties()
11 | properties.load(file.newDataInputStream());
12 | return properties.getProperty(key)
13 | }
14 | }
15 |
16 | def getRepositoryUrl() {
17 | return isSnapshot() ? getPropertyFromLocalProperties("SNAPSHOT_REPOSITORY_URL") : getPropertyFromLocalProperties("RELEASE_REPOSITORY_URL")
18 | // return isSnapshot() ? SNAPSHOT_REPOSITORY_URL : RELEASE_REPOSITORY_URL
19 | }
20 |
21 | def isSnapshot() {
22 | return version.endsWith("SNAPSHOT");
23 | }
24 |
25 | def hasAndroidPlugin() {
26 | return getPlugins().inject(false) { a, b->
27 | def classStr = b.getClass().name
28 | def isAndroid = ("com.android.build.gradle.LibraryPlugin" == classStr) || ("com.android.build.gradle.AppPlugin" == classStr)
29 | a || isAndroid
30 | }
31 | }
32 | task sourcesJar(type: Jar) {
33 | if (hasAndroidPlugin()) {
34 | from android.sourceSets.main.java.srcDirs
35 | classifier = 'sources'
36 | } else {
37 | from sourceSets.main.allSource
38 | classifier = 'sources'
39 | }
40 | }
41 | artifacts {
42 | archives sourcesJar
43 | }
44 |
45 | uploadArchives {
46 | repositories.mavenDeployer {
47 | repository(url: repositoryUrl) {
48 | authentication(userName: getPropertyFromLocalProperties("USER"), password: getPropertyFromLocalProperties("PASSWORD"))
49 | }
50 | }
51 | }
52 | bintray {
53 | user = getPropertyFromLocalProperties("bintray.user")
54 | key = getPropertyFromLocalProperties("bintray.apikey")
55 | configurations = ['archives']
56 | pkg {
57 | repo = 'maven'
58 | name = "${project.group}:${project.name}"
59 | userOrg = 'meituan'
60 | licenses = ['Apache-2.0']
61 | websiteUrl = 'http://tech.meituan.com/android_autopatch.html'
62 | vcsUrl = 'https://github.com/Meituan-Dianping/Robust'
63 | publish = true
64 | }
65 | }
66 |
--------------------------------------------------------------------------------
/gradlew:
--------------------------------------------------------------------------------
1 | #!/usr/bin/env bash
2 |
3 | ##############################################################################
4 | ##
5 | ## Gradle start up script for UN*X
6 | ##
7 | ##############################################################################
8 |
9 | # Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
10 | DEFAULT_JVM_OPTS=""
11 |
12 | APP_NAME="Gradle"
13 | APP_BASE_NAME=`basename "$0"`
14 |
15 | # Use the maximum available, or set MAX_FD != -1 to use that value.
16 | MAX_FD="maximum"
17 |
18 | warn ( ) {
19 | echo "$*"
20 | }
21 |
22 | die ( ) {
23 | echo
24 | echo "$*"
25 | echo
26 | exit 1
27 | }
28 |
29 | # OS specific support (must be 'true' or 'false').
30 | cygwin=false
31 | msys=false
32 | darwin=false
33 | case "`uname`" in
34 | CYGWIN* )
35 | cygwin=true
36 | ;;
37 | Darwin* )
38 | darwin=true
39 | ;;
40 | MINGW* )
41 | msys=true
42 | ;;
43 | esac
44 |
45 | # Attempt to set APP_HOME
46 | # Resolve links: $0 may be a link
47 | PRG="$0"
48 | # Need this for relative symlinks.
49 | while [ -h "$PRG" ] ; do
50 | ls=`ls -ld "$PRG"`
51 | link=`expr "$ls" : '.*-> \(.*\)$'`
52 | if expr "$link" : '/.*' > /dev/null; then
53 | PRG="$link"
54 | else
55 | PRG=`dirname "$PRG"`"/$link"
56 | fi
57 | done
58 | SAVED="`pwd`"
59 | cd "`dirname \"$PRG\"`/" >/dev/null
60 | APP_HOME="`pwd -P`"
61 | cd "$SAVED" >/dev/null
62 |
63 | CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
64 |
65 | # Determine the Java command to use to start the JVM.
66 | if [ -n "$JAVA_HOME" ] ; then
67 | if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
68 | # IBM's JDK on AIX uses strange locations for the executables
69 | JAVACMD="$JAVA_HOME/jre/sh/java"
70 | else
71 | JAVACMD="$JAVA_HOME/bin/java"
72 | fi
73 | if [ ! -x "$JAVACMD" ] ; then
74 | die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME
75 |
76 | Please set the JAVA_HOME variable in your environment to match the
77 | location of your Java installation."
78 | fi
79 | else
80 | JAVACMD="java"
81 | which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
82 |
83 | Please set the JAVA_HOME variable in your environment to match the
84 | location of your Java installation."
85 | fi
86 |
87 | # Increase the maximum file descriptors if we can.
88 | if [ "$cygwin" = "false" -a "$darwin" = "false" ] ; then
89 | MAX_FD_LIMIT=`ulimit -H -n`
90 | if [ $? -eq 0 ] ; then
91 | if [ "$MAX_FD" = "maximum" -o "$MAX_FD" = "max" ] ; then
92 | MAX_FD="$MAX_FD_LIMIT"
93 | fi
94 | ulimit -n $MAX_FD
95 | if [ $? -ne 0 ] ; then
96 | warn "Could not set maximum file descriptor limit: $MAX_FD"
97 | fi
98 | else
99 | warn "Could not query maximum file descriptor limit: $MAX_FD_LIMIT"
100 | fi
101 | fi
102 |
103 | # For Darwin, add options to specify how the application appears in the dock
104 | if $darwin; then
105 | GRADLE_OPTS="$GRADLE_OPTS \"-Xdock:name=$APP_NAME\" \"-Xdock:icon=$APP_HOME/media/gradle.icns\""
106 | fi
107 |
108 | # For Cygwin, switch paths to Windows format before running java
109 | if $cygwin ; then
110 | APP_HOME=`cygpath --path --mixed "$APP_HOME"`
111 | CLASSPATH=`cygpath --path --mixed "$CLASSPATH"`
112 | JAVACMD=`cygpath --unix "$JAVACMD"`
113 |
114 | # We build the pattern for arguments to be converted via cygpath
115 | ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null`
116 | SEP=""
117 | for dir in $ROOTDIRSRAW ; do
118 | ROOTDIRS="$ROOTDIRS$SEP$dir"
119 | SEP="|"
120 | done
121 | OURCYGPATTERN="(^($ROOTDIRS))"
122 | # Add a user-defined pattern to the cygpath arguments
123 | if [ "$GRADLE_CYGPATTERN" != "" ] ; then
124 | OURCYGPATTERN="$OURCYGPATTERN|($GRADLE_CYGPATTERN)"
125 | fi
126 | # Now convert the arguments - kludge to limit ourselves to /bin/sh
127 | i=0
128 | for arg in "$@" ; do
129 | CHECK=`echo "$arg"|egrep -c "$OURCYGPATTERN" -`
130 | CHECK2=`echo "$arg"|egrep -c "^-"` ### Determine if an option
131 |
132 | if [ $CHECK -ne 0 ] && [ $CHECK2 -eq 0 ] ; then ### Added a condition
133 | eval `echo args$i`=`cygpath --path --ignore --mixed "$arg"`
134 | else
135 | eval `echo args$i`="\"$arg\""
136 | fi
137 | i=$((i+1))
138 | done
139 | case $i in
140 | (0) set -- ;;
141 | (1) set -- "$args0" ;;
142 | (2) set -- "$args0" "$args1" ;;
143 | (3) set -- "$args0" "$args1" "$args2" ;;
144 | (4) set -- "$args0" "$args1" "$args2" "$args3" ;;
145 | (5) set -- "$args0" "$args1" "$args2" "$args3" "$args4" ;;
146 | (6) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" ;;
147 | (7) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" ;;
148 | (8) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" ;;
149 | (9) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" "$args8" ;;
150 | esac
151 | fi
152 |
153 | # Split up the JVM_OPTS And GRADLE_OPTS values into an array, following the shell quoting and substitution rules
154 | function splitJvmOpts() {
155 | JVM_OPTS=("$@")
156 | }
157 | eval splitJvmOpts $DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS
158 | JVM_OPTS[${#JVM_OPTS[*]}]="-Dorg.gradle.appname=$APP_BASE_NAME"
159 |
160 | exec "$JAVACMD" "${JVM_OPTS[@]}" -classpath "$CLASSPATH" org.gradle.wrapper.GradleWrapperMain "$@"
161 |
--------------------------------------------------------------------------------
/gradlew.bat:
--------------------------------------------------------------------------------
1 | @if "%DEBUG%" == "" @echo off
2 | @rem ##########################################################################
3 | @rem
4 | @rem Gradle startup script for Windows
5 | @rem
6 | @rem ##########################################################################
7 |
8 | @rem Set local scope for the variables with windows NT shell
9 | if "%OS%"=="Windows_NT" setlocal
10 |
11 | @rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
12 | set DEFAULT_JVM_OPTS=
13 |
14 | set DIRNAME=%~dp0
15 | if "%DIRNAME%" == "" set DIRNAME=.
16 | set APP_BASE_NAME=%~n0
17 | set APP_HOME=%DIRNAME%
18 |
19 | @rem Find java.exe
20 | if defined JAVA_HOME goto findJavaFromJavaHome
21 |
22 | set JAVA_EXE=java.exe
23 | %JAVA_EXE% -version >NUL 2>&1
24 | if "%ERRORLEVEL%" == "0" goto init
25 |
26 | echo.
27 | echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
28 | echo.
29 | echo Please set the JAVA_HOME variable in your environment to match the
30 | echo location of your Java installation.
31 |
32 | goto fail
33 |
34 | :findJavaFromJavaHome
35 | set JAVA_HOME=%JAVA_HOME:"=%
36 | set JAVA_EXE=%JAVA_HOME%/bin/java.exe
37 |
38 | if exist "%JAVA_EXE%" goto init
39 |
40 | echo.
41 | echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME%
42 | echo.
43 | echo Please set the JAVA_HOME variable in your environment to match the
44 | echo location of your Java installation.
45 |
46 | goto fail
47 |
48 | :init
49 | @rem Get command-line arguments, handling Windowz variants
50 |
51 | if not "%OS%" == "Windows_NT" goto win9xME_args
52 | if "%@eval[2+2]" == "4" goto 4NT_args
53 |
54 | :win9xME_args
55 | @rem Slurp the command line arguments.
56 | set CMD_LINE_ARGS=
57 | set _SKIP=2
58 |
59 | :win9xME_args_slurp
60 | if "x%~1" == "x" goto execute
61 |
62 | set CMD_LINE_ARGS=%*
63 | goto execute
64 |
65 | :4NT_args
66 | @rem Get arguments from the 4NT Shell from JP Software
67 | set CMD_LINE_ARGS=%$
68 |
69 | :execute
70 | @rem Setup the command line
71 |
72 | set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar
73 |
74 | @rem Execute Gradle
75 | "%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %CMD_LINE_ARGS%
76 |
77 | :end
78 | @rem End local scope for the variables with windows NT shell
79 | if "%ERRORLEVEL%"=="0" goto mainEnd
80 |
81 | :fail
82 | rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of
83 | rem the _cmd.exe /c_ return code!
84 | if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1
85 | exit /b 1
86 |
87 | :mainEnd
88 | if "%OS%"=="Windows_NT" endlocal
89 |
90 | :omega
91 |
--------------------------------------------------------------------------------
/images/Robust结构图.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Meituan-Dianping/Robust/955adcc21e4fbcb52054a8f7f4bbb11f462aeb2f/images/Robust结构图.png
--------------------------------------------------------------------------------
/images/patchsuccess_cn.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Meituan-Dianping/Robust/955adcc21e4fbcb52054a8f7f4bbb11f462aeb2f/images/patchsuccess_cn.png
--------------------------------------------------------------------------------
/images/patchsuccess_en.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Meituan-Dianping/Robust/955adcc21e4fbcb52054a8f7f4bbb11f462aeb2f/images/patchsuccess_en.png
--------------------------------------------------------------------------------
/images/补丁代码结构.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/Meituan-Dianping/Robust/955adcc21e4fbcb52054a8f7f4bbb11f462aeb2f/images/补丁代码结构.png
--------------------------------------------------------------------------------
/patch/build.gradle:
--------------------------------------------------------------------------------
1 | apply plugin: 'com.android.library'
2 | project.archivesBaseName = 'robust'
3 | android {
4 | compileSdkVersion 24
5 | buildToolsVersion '25.0.2'
6 | defaultConfig {
7 | minSdkVersion 9
8 | targetSdkVersion 24
9 | versionCode 1
10 | versionName '1.0'
11 | }
12 | buildTypes {
13 | release {
14 | consumerProguardFiles 'proguard-rules.pro'
15 | proguardFiles getDefaultProguardFile('proguard-android.txt'), 'proguard-rules.pro'
16 | }
17 | }
18 | compileOptions {
19 | sourceCompatibility JavaVersion.VERSION_1_7
20 | targetCompatibility JavaVersion.VERSION_1_7
21 | }
22 | packagingOptions {
23 | exclude 'META-INF/maven/com.google.guava/guava/pom.properties'
24 | exclude 'META-INF/maven/com.google.guava/guava/pom.xml'
25 | exclude 'META-INF/maven/commons-cli/commons-cli/pom.properties'
26 | exclude 'META-INF/maven/commons-cli/commons-cli/pom.xml'
27 | }
28 | }
29 | configurations.all {
30 | resolutionStrategy.cacheChangingModulesFor 0, 'seconds'
31 | }
32 | dependencies {
33 | // compile 'com.meituan.robust:autopatchbase:' + VERSION_NAME
34 | compile project(':autopatchbase')
35 | }
36 |
37 | apply from: '../gradle_mvn_push.gradle'
38 |
--------------------------------------------------------------------------------
/patch/proguard-rules.pro:
--------------------------------------------------------------------------------
1 | -keep class com.meituan.robust.**{*;}
2 | -keep class meituan.robust.**{*;}
3 | -keep class com.google.gson.**{*;}
4 | -keepattributes *Annotation*
5 | -keepclassmembers class **{
6 | public static com.meituan.robust.ChangeQuickRedirect *;
7 | }
8 |
9 |
--------------------------------------------------------------------------------
/patch/src/main/AndroidManifest.xml:
--------------------------------------------------------------------------------
1 |
2 |
5 |
6 |
--------------------------------------------------------------------------------
/patch/src/main/java/com/meituan/robust/Patch.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust;
2 |
3 | import java.io.File;
4 |
5 | /**
6 | * Created by mivanzhang on 15/7/23.
7 | * 补丁定义
8 | */
9 | public class Patch implements Cloneable {
10 | //补丁的编号，补丁的唯一标识符
11 | public String getName() {
12 | return name;
13 | }
14 |
15 | public void setName(String name) {
16 | this.name=name;
17 | }
18 |
19 | public String getUrl() {
20 | return url;
21 | }
22 |
23 | public void setUrl(String url) {
24 | this.url = url;
25 | }
26 | //原始补丁文件的路径，推荐放到私有目录
27 | public String getLocalPath() {
28 | return localPath + ".jar";
29 | }
30 |
31 | public void setLocalPath(String localPath) {
32 | this.localPath = localPath;
33 | }
34 | //原始补丁的md5，确保原始补丁文件没有被篡改
35 | public String getMd5() {
36 | return md5;
37 | }
38 |
39 | public void setMd5(String md5) {
40 | this.md5 = md5;
41 | }
42 |
43 | private String patchesInfoImplClassFullName;
44 | /**
45 | * 补丁名称
46 | */
47 | private String name;
48 |
49 | /**
50 | * 补丁的下载url
51 | */
52 | private String url;
53 | /**
54 | * 补丁本地保存路径
55 | */
56 | private String localPath;
57 |
58 | private String tempPath;
59 |
60 | /**
61 | * 补丁md5值
62 | */
63 | private String md5;
64 |
65 | /**
66 | * app hash值,避免应用内升级导致低版本app的补丁应用到了高版本app上
67 | */
68 | private String appHash;
69 |
70 | public boolean isAppliedSuccess() {
71 | return isAppliedSuccess;
72 | }
73 |
74 | public void setAppliedSuccess(boolean appliedSuccess) {
75 | isAppliedSuccess = appliedSuccess;
76 | }
77 |
78 | /**
79 | * 补丁是否已经applied success
80 | */
81 | private boolean isAppliedSuccess;
82 |
83 | /**
84 | * 删除文件
85 | */
86 | public void delete(String path) {
87 | File f = new File(path);
88 | f.delete();
89 | }
90 |
91 | public String getPatchesInfoImplClassFullName() {
92 | return patchesInfoImplClassFullName;
93 | }
94 |
95 | public void setPatchesInfoImplClassFullName(String patchesInfoImplClassFullName) {
96 | this.patchesInfoImplClassFullName = patchesInfoImplClassFullName;
97 | }
98 |
99 | public String getAppHash() {
100 | return appHash;
101 | }
102 |
103 | public void setAppHash(String appHash) {
104 | this.appHash = appHash;
105 | }
106 | //解密之后的补丁文件，可以直接运行的补丁文件，建议加载之后立刻删除，保证安全性
107 | public String getTempPath() {
108 | return tempPath + "_temp" + ".jar";
109 | }
110 |
111 | public void setTempPath(String tempPath) {
112 | this.tempPath = tempPath;
113 | }
114 |
115 | @Override
116 | public Patch clone() {
117 | Patch clone = null;
118 | try {
119 | clone = (Patch) super.clone();
120 | } catch (CloneNotSupportedException e) {
121 | // throw e;
122 | }
123 | return clone;
124 | }
125 | }
126 |
--------------------------------------------------------------------------------
/patch/src/main/java/com/meituan/robust/PatchExecutor.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust;
2 |
3 | import android.content.Context;
4 | import android.text.TextUtils;
5 | import android.util.Log;
6 |
7 | import java.io.File;
8 | import java.lang.reflect.Field;
9 | import java.util.List;
10 |
11 | import dalvik.system.DexClassLoader;
12 |
13 | /**
14 | * Created by c_kunwu on 16/5/20.
15 | * if you need,you can override PatchExecutor
16 | */
17 | public class PatchExecutor extends Thread {
18 | protected Context context;
19 | protected PatchManipulate patchManipulate;
20 | protected RobustCallBack robustCallBack;
21 |
22 | public PatchExecutor(Context context, PatchManipulate patchManipulate, RobustCallBack robustCallBack) {
23 | this.context = context.getApplicationContext();
24 | this.patchManipulate = patchManipulate;
25 | this.robustCallBack = robustCallBack;
26 | }
27 |
28 | @Override
29 | public void run() {
30 | try {
31 | //拉取补丁列表
32 | List patches = fetchPatchList();
33 | //应用补丁列表
34 | applyPatchList(patches);
35 | } catch (Throwable t) {
36 | Log.e("robust", "PatchExecutor run", t);
37 | robustCallBack.exceptionNotify(t, "class:PatchExecutor,method:run,line:36");
38 | }
39 | }
40 |
41 | /**
42 | * 拉取补丁列表
43 | */
44 | protected List fetchPatchList() {
45 | return patchManipulate.fetchPatchList(context);
46 | }
47 |
48 | /**
49 | * 应用补丁列表
50 | */
51 | protected void applyPatchList(List patches) {
52 | if (null == patches || patches.isEmpty()) {
53 | return;
54 | }
55 | Log.d("robust", " patchManipulate list size is " + patches.size());
56 | for (Patch p : patches) {
57 | if (p.isAppliedSuccess()) {
58 | Log.d("robust", "p.isAppliedSuccess() skip " + p.getLocalPath());
59 | continue;
60 | }
61 | if (patchManipulate.ensurePatchExist(p)) {
62 | boolean currentPatchResult = false;
63 | try {
64 | currentPatchResult = patch(context, p);
65 | } catch (Throwable t) {
66 | robustCallBack.exceptionNotify(t, "class:PatchExecutor method:applyPatchList line:69");
67 | }
68 | if (currentPatchResult) {
69 | //设置patch 状态为成功
70 | p.setAppliedSuccess(true);
71 | //统计PATCH成功率 PATCH成功
72 | robustCallBack.onPatchApplied(true, p);
73 |
74 | } else {
75 | //统计PATCH成功率 PATCH失败
76 | robustCallBack.onPatchApplied(false, p);
77 | }
78 |
79 | Log.d("robust", "patch LocalPath:" + p.getLocalPath() + ",apply result " + currentPatchResult);
80 |
81 | }
82 | }
83 | }
84 |
85 | protected boolean patch(Context context, Patch patch) {
86 | if (!patchManipulate.verifyPatch(context, patch)) {
87 | robustCallBack.logNotify("verifyPatch failure, patch info:" + "id = " + patch.getName() + ",md5 = " + patch.getMd5(), "class:PatchExecutor method:patch line:107");
88 | return false;
89 | }
90 |
91 | ClassLoader classLoader = null;
92 |
93 | try {
94 | File dexOutputDir = getPatchCacheDirPath(context, patch.getName() + patch.getMd5());
95 | classLoader = new DexClassLoader(patch.getTempPath(), dexOutputDir.getAbsolutePath(),
96 | null, PatchExecutor.class.getClassLoader());
97 | } catch (Throwable throwable) {
98 | throwable.printStackTrace();
99 | }
100 | if (null == classLoader) {
101 | return false;
102 | }
103 |
104 | Class patchClass, sourceClass;
105 |
106 | Class patchesInfoClass;
107 | PatchesInfo patchesInfo = null;
108 | try {
109 | Log.d("robust", "patch patch_info_name:" + patch.getPatchesInfoImplClassFullName());
110 | patchesInfoClass = classLoader.loadClass(patch.getPatchesInfoImplClassFullName());
111 | patchesInfo = (PatchesInfo) patchesInfoClass.newInstance();
112 | } catch (Throwable t) {
113 | Log.e("robust", "patch failed 188 ", t);
114 | }
115 |
116 | if (patchesInfo == null) {
117 | robustCallBack.logNotify("patchesInfo is null, patch info:" + "id = " + patch.getName() + ",md5 = " + patch.getMd5(), "class:PatchExecutor method:patch line:114");
118 | return false;
119 | }
120 |
121 | //classes need to patch
122 | List patchedClasses = patchesInfo.getPatchedClassesInfo();
123 | if (null == patchedClasses || patchedClasses.isEmpty()) {
124 | // robustCallBack.logNotify("patchedClasses is null or empty, patch info:" + "id = " + patch.getName() + ",md5 = " + patch.getMd5(), "class:PatchExecutor method:patch line:122");
125 | //手写的补丁有时候会返回一个空list
126 | return true;
127 | }
128 |
129 | boolean isClassNotFoundException = false;
130 | for (PatchedClassInfo patchedClassInfo : patchedClasses) {
131 | String patchedClassName = patchedClassInfo.patchedClassName;
132 | String patchClassName = patchedClassInfo.patchClassName;
133 | if (TextUtils.isEmpty(patchedClassName) || TextUtils.isEmpty(patchClassName)) {
134 | robustCallBack.logNotify("patchedClasses or patchClassName is empty, patch info:" + "id = " + patch.getName() + ",md5 = " + patch.getMd5(), "class:PatchExecutor method:patch line:131");
135 | continue;
136 | }
137 | Log.d("robust", "current path:" + patchedClassName);
138 | try {
139 | try {
140 | sourceClass = classLoader.loadClass(patchedClassName.trim());
141 | } catch (ClassNotFoundException e) {
142 | isClassNotFoundException = true;
143 | // robustCallBack.exceptionNotify(e, "class:PatchExecutor method:patch line:258");
144 | continue;
145 | }
146 |
147 | Field[] fields = sourceClass.getDeclaredFields();
148 | Log.d("robust", "oldClass :" + sourceClass + " fields " + fields.length);
149 | Field changeQuickRedirectField = null;
150 | for (Field field : fields) {
151 | if (TextUtils.equals(field.getType().getCanonicalName(), ChangeQuickRedirect.class.getCanonicalName()) && TextUtils.equals(field.getDeclaringClass().getCanonicalName(), sourceClass.getCanonicalName())) {
152 | changeQuickRedirectField = field;
153 | break;
154 | }
155 | }
156 | if (changeQuickRedirectField == null) {
157 | robustCallBack.logNotify("changeQuickRedirectField is null, patch info:" + "id = " + patch.getName() + ",md5 = " + patch.getMd5(), "class:PatchExecutor method:patch line:147");
158 | Log.d("robust", "current path:" + patchedClassName + " something wrong !! can not find:ChangeQuickRedirect in" + patchClassName);
159 | continue;
160 | }
161 | Log.d("robust", "current path:" + patchedClassName + " find:ChangeQuickRedirect " + patchClassName);
162 | try {
163 | patchClass = classLoader.loadClass(patchClassName);
164 | Object patchObject = patchClass.newInstance();
165 | changeQuickRedirectField.setAccessible(true);
166 | changeQuickRedirectField.set(null, patchObject);
167 | Log.d("robust", "changeQuickRedirectField set success " + patchClassName);
168 | } catch (Throwable t) {
169 | Log.e("robust", "patch failed! ");
170 | robustCallBack.exceptionNotify(t, "class:PatchExecutor method:patch line:163");
171 | }
172 | } catch (Throwable t) {
173 | Log.e("robust", "patch failed! ");
174 | // robustCallBack.exceptionNotify(t, "class:PatchExecutor method:patch line:169");
175 | }
176 | }
177 | Log.d("robust", "patch finished ");
178 | if (isClassNotFoundException) {
179 | return false;
180 | }
181 | return true;
182 | }
183 |
184 | private static final String ROBUST_PATCH_CACHE_DIR = "patch_cache";
185 |
186 | /*
187 | * @param c
188 | * @return 返回缓存补丁路径，一般是内部存储,补丁目录
189 | */
190 | private static File getPatchCacheDirPath(Context c, String key) {
191 | File patchTempDir = c.getDir(ROBUST_PATCH_CACHE_DIR + key, Context.MODE_PRIVATE);
192 | if (!patchTempDir.exists()) {
193 | patchTempDir.mkdir();
194 | }
195 |
196 | return patchTempDir;
197 | }
198 |
199 | }
200 |
--------------------------------------------------------------------------------
/patch/src/main/java/com/meituan/robust/PatchManipulate.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust;
2 |
3 | import android.content.Context;
4 |
5 | import java.util.List;
6 |
7 | /**
8 | * Created by hedex on 16/6/20.
9 | */
10 | public abstract class PatchManipulate {
11 | /**
12 | * 获取补丁列表
13 | *
14 | * @param context
15 | * @return 相应的补丁列表
16 | */
17 | protected abstract List fetchPatchList(Context context);
18 |
19 | /**
20 | * 验证补丁文件md5是否一致
21 | * 如果不存在，则动态下载
22 | *
23 | * @param context
24 | * @param patch
25 | * @return 校验结果
26 | */
27 | protected abstract boolean verifyPatch(Context context, Patch patch);
28 |
29 | /**
30 | * 努力确保补丁文件存在，验证md5是否一致。
31 | * 如果不存在，则动态下载
32 | *
33 | * @param patch
34 | * @return 是否存在
35 | */
36 | protected abstract boolean ensurePatchExist(Patch patch);
37 | }
38 |
--------------------------------------------------------------------------------
/patch/src/main/java/com/meituan/robust/PatchProxy.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust;
2 |
3 | import android.text.TextUtils;
4 |
5 | import java.util.concurrent.CopyOnWriteArrayList;
6 |
7 | /**
8 | * Created by c_kunwu on 16/7/5.
9 | */
10 | public class PatchProxy {
11 |
12 | private static CopyOnWriteArrayList registerExtensionList = new CopyOnWriteArrayList<>();
13 | private static ThreadLocal robustExtensionThreadLocal = new ThreadLocal<>();
14 |
15 | /**
16 | * 原来的插桩逻辑为这样：
17 | *
18 | * if (PatchProxy.isSupport()) {
19 | * return PatchProxy.accessDispatch();
20 | * }
21 | *
22 | * 封装一下变成下面这种代码
23 | *
24 | * PatchProxyResult patchProxyResult = PatchProxy.proxy();
25 | * if (patchProxyResult.isSupported) {
26 | * return patchProxyResult.result;
27 | * }
28 | *
29 | * 这样做的好处有两个：
30 | * 1. 减少包大小。 不是开玩笑，虽然后者代码看起来变得复杂，但实质产生的指令更少。
31 | * 之前两个函数调用，每次都需要load 7个参数到栈上，这7个参数还不是简单的基本类型，这意味着比后者多出若干条指令。
32 | * 数据显示在5W个方法的插桩下，后者能比前者节省200KB
33 | *
34 | * 2. fix一个bug。robust其实支持采用将ChangeQuickRedirect置为null的方法实时下线一个patch，那原来的插桩逻辑就存在线程安全的问题。
35 | * 根源在于原来的逻辑中ChangeQuickRedirect是每次都直接去取的static变量值
36 | * 如果在执行isSupport的时候ChangeQuickRedirect有值，但执行到accessDispatch时ChangeQuickRedirect被置为空，那就意味着被patch的方法该次将不执行任何代码
37 | * 这样会带来一系列的不可知问题。
38 | * 封装之后能保证这两个方法读取到的ChangeQuickRedirect是同一份。
39 | */
40 | public static PatchProxyResult proxy(Object[] paramsArray, Object current, ChangeQuickRedirect changeQuickRedirect, boolean isStatic, int methodNumber, Class[] paramsClassTypes, Class returnType) {
41 | PatchProxyResult patchProxyResult = new PatchProxyResult();
42 | if (PatchProxy.isSupport(paramsArray, current, changeQuickRedirect, isStatic, methodNumber, paramsClassTypes, returnType)) {
43 | patchProxyResult.isSupported = true;
44 | patchProxyResult.result = PatchProxy.accessDispatch(paramsArray, current, changeQuickRedirect, isStatic, methodNumber, paramsClassTypes, returnType);
45 | }
46 | return patchProxyResult;
47 | }
48 |
49 | public static boolean isSupport(Object[] paramsArray, Object current, ChangeQuickRedirect changeQuickRedirect, boolean isStatic, int methodNumber, Class[] paramsClassTypes, Class returnType) {
50 | //Robust补丁优先执行，其他功能靠后
51 | if (changeQuickRedirect == null) {
52 | //不执行补丁，轮询其他监听者
53 | if (registerExtensionList == null || registerExtensionList.isEmpty()) {
54 | return false;
55 | }
56 | for (RobustExtension robustExtension : registerExtensionList) {
57 | if (robustExtension.isSupport(new RobustArguments(paramsArray, current, isStatic, methodNumber, paramsClassTypes, returnType))) {
58 | robustExtensionThreadLocal.set(robustExtension);
59 | return true;
60 | }
61 | }
62 | return false;
63 | }
64 | String classMethod = getClassMethod(isStatic, methodNumber);
65 | if (TextUtils.isEmpty(classMethod)) {
66 | return false;
67 | }
68 | Object[] objects = getObjects(paramsArray, current, isStatic);
69 | try {
70 | return changeQuickRedirect.isSupport(classMethod, objects);
71 | } catch (Throwable t) {
72 | return false;
73 | }
74 | }
75 |
76 |
77 | public static Object accessDispatch(Object[] paramsArray, Object current, ChangeQuickRedirect changeQuickRedirect, boolean isStatic, int methodNumber, Class[] paramsClassTypes, Class returnType) {
78 |
79 | if (changeQuickRedirect == null) {
80 | RobustExtension robustExtension = robustExtensionThreadLocal.get();
81 | robustExtensionThreadLocal.remove();
82 | if (robustExtension != null) {
83 | notify(robustExtension.describeSelfFunction());
84 | return robustExtension.accessDispatch(new RobustArguments(paramsArray, current, isStatic, methodNumber, paramsClassTypes, returnType));
85 | }
86 | return null;
87 | }
88 | String classMethod = getClassMethod(isStatic, methodNumber);
89 | if (TextUtils.isEmpty(classMethod)) {
90 | return null;
91 | }
92 | notify(Constants.PATCH_EXECUTE);
93 | Object[] objects = getObjects(paramsArray, current, isStatic);
94 | return changeQuickRedirect.accessDispatch(classMethod, objects);
95 | }
96 |
97 | public static void accessDispatchVoid(Object[] paramsArray, Object current, ChangeQuickRedirect changeQuickRedirect, boolean isStatic, int methodNumber, Class[] paramsClassTypes, Class returnType) {
98 | if (changeQuickRedirect == null) {
99 | RobustExtension robustExtension = robustExtensionThreadLocal.get();
100 | robustExtensionThreadLocal.remove();
101 | if (robustExtension != null) {
102 | notify(robustExtension.describeSelfFunction());
103 | robustExtension.accessDispatch(new RobustArguments(paramsArray, current, isStatic, methodNumber, paramsClassTypes, returnType));
104 | }
105 | return;
106 | }
107 | notify(Constants.PATCH_EXECUTE);
108 | String classMethod = getClassMethod(isStatic, methodNumber);
109 | if (TextUtils.isEmpty(classMethod)) {
110 | return;
111 | }
112 | Object[] objects = getObjects(paramsArray, current, isStatic);
113 | changeQuickRedirect.accessDispatch(classMethod, objects);
114 | }
115 |
116 |
117 | private static Object[] getObjects(Object[] arrayOfObject, Object current, boolean isStatic) {
118 | Object[] objects;
119 | if (arrayOfObject == null) {
120 | return null;
121 | }
122 | int argNum = arrayOfObject.length;
123 | if (isStatic) {
124 | objects = new Object[argNum];
125 | } else {
126 | objects = new Object[argNum + 1];
127 | }
128 | int x = 0;
129 | for (; x < argNum; x++) {
130 | objects[x] = arrayOfObject[x];
131 | }
132 | if (!(isStatic)) {
133 | objects[x] = current;
134 | }
135 | return objects;
136 | }
137 |
138 | private static String getClassMethod(boolean isStatic, int methodNumber) {
139 | String classMethod = "";
140 | try {
141 | //可能过于耗时，这部分需要请自己调用函数
142 | // java.lang.StackTraceElement stackTraceElement = (new java.lang.Throwable()).getStackTrace()[2];
143 | // String methodName = stackTraceElement.getMethodName();
144 | // String className = stackTraceElement.getClassName();
145 | String methodName = "";
146 | String className = "";
147 | classMethod = className + ":" + methodName + ":" + isStatic + ":" + methodNumber;
148 | } catch (Exception e) {
149 |
150 | }
151 | return classMethod;
152 | }
153 |
154 | private static String[] getClassMethodName() {
155 | java.lang.StackTraceElement stackTraceElement = (new java.lang.Throwable()).getStackTrace()[2];
156 | String[] classMethodname = new String[2];
157 | classMethodname[0] = stackTraceElement.getClassName();
158 | classMethodname[1] = stackTraceElement.getMethodName();
159 | return classMethodname;
160 | }
161 |
162 | /***
163 | *
164 | * @param robustExtension
165 | * 注册RobustExtension监听器，通知当前执行程序
166 | * @return
167 | */
168 | public synchronized static boolean register(RobustExtension robustExtension) {
169 | if (registerExtensionList == null) {
170 | registerExtensionList = new CopyOnWriteArrayList();
171 | }
172 | return registerExtensionList.addIfAbsent(robustExtension);
173 | }
174 |
175 | public synchronized static boolean unregister(RobustExtension robustExtension) {
176 | if (registerExtensionList == null) {
177 | return false;
178 | }
179 | return registerExtensionList.remove(robustExtension);
180 | }
181 |
182 | /**
183 | * clear registerExtensionList and executing robustExtension
184 | */
185 | public static void reset() {
186 | registerExtensionList = new CopyOnWriteArrayList<>();
187 | robustExtensionThreadLocal = new ThreadLocal<>();
188 | }
189 |
190 | private static void notify(String info) {
191 | if (registerExtensionList == null) {
192 | return;
193 | }
194 | for (RobustExtension robustExtension : registerExtensionList) {
195 | robustExtension.notifyListner(info);
196 | }
197 | }
198 |
199 | }
200 |
--------------------------------------------------------------------------------
/patch/src/main/java/com/meituan/robust/PatchProxyResult.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust;
2 |
3 | /**
4 | * 辅助model定义
5 | *
6 | * @author limengyun
7 | */
8 | public class PatchProxyResult {
9 | public boolean isSupported;
10 | public Object result;
11 | }
12 |
--------------------------------------------------------------------------------
/patch/src/main/java/com/meituan/robust/PatchedClassInfo.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust;
2 |
3 | /**
4 | * Created by hedex on 16/6/3.
5 | * a map record the class name before ProGuard and after ProGuard
6 | */
7 | public class PatchedClassInfo {
8 | public String patchedClassName;
9 | public String patchClassName;
10 |
11 | public PatchedClassInfo(String patchedClassName, String patchClassName) {
12 | this.patchedClassName = patchedClassName;
13 | this.patchClassName = patchClassName;
14 | }
15 | }
16 |
--------------------------------------------------------------------------------
/patch/src/main/java/com/meituan/robust/PatchesInfo.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust;
2 |
3 | import java.util.List;
4 |
5 | /**
6 | * Created by c_kunwu on 16/5/12.
7 | * an interface describe patch.jar info
8 | */
9 | public interface PatchesInfo {
10 | List getPatchedClassesInfo();
11 | }
12 |
--------------------------------------------------------------------------------
/patch/src/main/java/com/meituan/robust/RobustApkHashUtils.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust;
2 |
3 | import android.content.Context;
4 | import android.text.TextUtils;
5 |
6 | import java.io.BufferedReader;
7 | import java.io.IOException;
8 | import java.io.InputStreamReader;
9 |
10 | /**
11 | * Created by hedex on 17/2/27.
12 | */
13 | public class RobustApkHashUtils {
14 | private static String robustApkHashValue;
15 |
16 | public static String readRobustApkHash(Context context) {
17 |
18 | if (TextUtils.isEmpty(robustApkHashValue)) {
19 | robustApkHashValue = readRobustApkHashFile(context);
20 | }
21 |
22 | return robustApkHashValue;
23 | }
24 |
25 | private static String readRobustApkHashFile(Context context) {
26 | String value = "";
27 | if (null == context) {
28 | return value;
29 | }
30 |
31 | try {
32 | value = readFirstLine(context, Constants.ROBUST_APK_HASH_FILE_NAME);
33 | } catch (Throwable throwable) {
34 |
35 | }
36 |
37 | return value;
38 | }
39 |
40 | private static String readFirstLine(Context context, String fileName) {
41 | BufferedReader reader = null;
42 | try {
43 | reader = new BufferedReader(
44 | new InputStreamReader(context.getAssets().open(fileName)));
45 |
46 | return reader.readLine();
47 | } catch (IOException e) {
48 | return "";
49 | } finally {
50 | if (reader != null) {
51 | try {
52 | reader.close();
53 | } catch (IOException e) {
54 | }
55 | }
56 | }
57 | }
58 |
59 | }
60 |
--------------------------------------------------------------------------------
/patch/src/main/java/com/meituan/robust/RobustCallBack.java:
--------------------------------------------------------------------------------
1 | package com.meituan.robust;
2 |
3 | import java.util.List;
4 |
5 | /**
6 | * Created by hedex on 17/1/22.
7 | */
8 |
9 | public interface RobustCallBack {
10 | /**
11 | * 获取补丁列表后，回调此方法
12 | *
13 | * @param result 补丁
14 | * @param isNet 补丁
15 | */
16 | void onPatchListFetched(boolean result, boolean isNet, List patches);
17 |
18 |
19 | /**
20 | * 在获取补丁后，回调此方法
21 | *
22 | * @param result 结果
23 | * @param patch 补丁
24 | */
25 | void onPatchFetched(boolean result, boolean isNet, Patch patch);
26 |
27 |
28 | /**
29 | * 在补丁应用后，回调此方法
30 | *
31 | * @param result 结果
32 | * @param patch 补丁
33 | */
34 | void onPatchApplied(boolean result, Patch patch);
35 |
36 |
37 | void logNotify(String log, String where);
38 |
39 |
40 | void exceptionNotify(Throwable throwable, String where);
41 | }
42 |
--------------------------------------------------------------------------------
/settings.gradle:
--------------------------------------------------------------------------------
1 | include ':app'
2 | include ':autopatchbase'
3 | include ':gradle-plugin'
4 | include ':patch'
5 | include ':auto-patch-plugin'
6 |
--------------------------------------------------------------------------------