1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
/*
* Copyright (C) 2011 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.android.dx.merge;
import com.android.dex.DexException;
import com.android.dex.DexIndexOverflowException;
import com.android.dx.io.CodeReader;
import com.android.dx.io.Opcodes;
import com.android.dx.io.instructions.DecodedInstruction;
import com.android.dx.io.instructions.ShortArrayCodeOutput;
final class InstructionTransformer {
private final IndexMap indexMap;
private final CodeReader reader;
private DecodedInstruction[] mappedInstructions;
private int mappedAt;
public InstructionTransformer(IndexMap indexMap) {
this.indexMap = indexMap;
this.reader = new CodeReader();
this.reader.setAllVisitors(new GenericVisitor());
this.reader.setStringVisitor(new StringVisitor());
this.reader.setTypeVisitor(new TypeVisitor());
this.reader.setFieldVisitor(new FieldVisitor());
this.reader.setMethodVisitor(new MethodVisitor());
}
public short[] transform(short[] encodedInstructions) throws DexException {
DecodedInstruction[] decodedInstructions =
DecodedInstruction.decodeAll(encodedInstructions);
int size = decodedInstructions.length;
mappedInstructions = new DecodedInstruction[size];
mappedAt = 0;
reader.visitAll(decodedInstructions);
ShortArrayCodeOutput out = new ShortArrayCodeOutput(size);
for (DecodedInstruction instruction : mappedInstructions) {
if (instruction != null) {
instruction.encode(out);
}
}
return out.getArray();
}
private class GenericVisitor implements CodeReader.Visitor {
public void visit(DecodedInstruction[] all, DecodedInstruction one) {
mappedInstructions[mappedAt++] = one;
}
}
private class StringVisitor implements CodeReader.Visitor {
public void visit(DecodedInstruction[] all, DecodedInstruction one) {
int stringId = one.getIndex();
int mappedId = indexMap.adjustString(stringId);
boolean isJumbo = (one.getOpcode() == Opcodes.CONST_STRING_JUMBO);
jumboCheck(isJumbo, mappedId);
mappedInstructions[mappedAt++] = one.withIndex(mappedId);
}
}
private class FieldVisitor implements CodeReader.Visitor {
public void visit(DecodedInstruction[] all, DecodedInstruction one) {
int fieldId = one.getIndex();
int mappedId = indexMap.adjustField(fieldId);
boolean isJumbo = (one.getOpcode() == Opcodes.CONST_STRING_JUMBO);
jumboCheck(isJumbo, mappedId);
mappedInstructions[mappedAt++] = one.withIndex(mappedId);
}
}
private class TypeVisitor implements CodeReader.Visitor {
public void visit(DecodedInstruction[] all, DecodedInstruction one) {
int typeId = one.getIndex();
int mappedId = indexMap.adjustType(typeId);
boolean isJumbo = (one.getOpcode() == Opcodes.CONST_STRING_JUMBO);
jumboCheck(isJumbo, mappedId);
mappedInstructions[mappedAt++] = one.withIndex(mappedId);
}
}
private class MethodVisitor implements CodeReader.Visitor {
public void visit(DecodedInstruction[] all, DecodedInstruction one) {
int methodId = one.getIndex();
int mappedId = indexMap.adjustMethod(methodId);
boolean isJumbo = (one.getOpcode() == Opcodes.CONST_STRING_JUMBO);
jumboCheck(isJumbo, mappedId);
mappedInstructions[mappedAt++] = one.withIndex(mappedId);
}
}
private static void jumboCheck(boolean isJumbo, int newIndex) {
if (!isJumbo && (newIndex > 0xffff)) {
throw new DexIndexOverflowException("Cannot merge new index " + newIndex +
" into a non-jumbo instruction!");
}
}
}
|