/*
* Tencent is pleased to support the open source community by making Tinker available.
*
* Copyright (C) 2016 THL A29 Limited, a Tencent company. All rights reserved.
*
* Licensed under the BSD 3-Clause License (the "License"); you may not use this file except in
* compliance with the License. You may obtain a copy of the License at
*
* https://opensource.org/licenses/BSD-3-Clause
*
* 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.taobao.common.dexpatcher.algorithms.patch;
import com.taobao.common.dexpatcher.struct.DexPatchFile;
import com.taobao.common.dexpatcher.struct.SmallPatchedDexItemFile;
import com.taobao.dex.Dex;
import com.taobao.dex.MethodId;
import com.taobao.dex.TableOfContents;
import com.taobao.dex.io.DexDataBuffer;
import com.taobao.dx.util.IndexMap;
/**
* Created by tangyinsheng on 2016/7/4.
*/
public class MethodIdSectionPatchAlgorithm extends DexSectionPatchAlgorithm<MethodId> {
private TableOfContents.Section patchedMethodIdTocSec = null;
private Dex.Section patchedMethodIdSec = null;
public MethodIdSectionPatchAlgorithm(
DexPatchFile patchFile,
Dex oldDex,
Dex patchedDex,
IndexMap oldToFullPatchedIndexMap,
IndexMap fullPatchedToSmallPatchedIndexMap,
final SmallPatchedDexItemFile extraInfoFile
) {
this(
patchFile,
oldDex,
patchedDex,
oldToFullPatchedIndexMap,
fullPatchedToSmallPatchedIndexMap,
new SmallPatchedDexItemChooser() {
@Override
public boolean isPatchedItemInSmallPatchedDex(
String oldDexSign, int patchedItemIndex
) {
return extraInfoFile.isMethodIdInSmallPatchedDex(
oldDexSign, patchedItemIndex
);
}
}
);
}
public MethodIdSectionPatchAlgorithm(
DexPatchFile patchFile,
Dex oldDex,
Dex patchedDex,
IndexMap oldToFullPatchedIndexMap,
IndexMap fullPatchedToSmallPatchedIndexMap,
SmallPatchedDexItemChooser spdItemChooser
) {
super(
patchFile,
oldDex,
oldToFullPatchedIndexMap,
fullPatchedToSmallPatchedIndexMap,
spdItemChooser
);
if (patchedDex != null) {
this.patchedMethodIdTocSec = patchedDex.getTableOfContents().methodIds;
this.patchedMethodIdSec = patchedDex.openSection(this.patchedMethodIdTocSec);
}
}
@Override
protected TableOfContents.Section getTocSection(Dex dex) {
return dex.getTableOfContents().methodIds;
}
@Override
protected MethodId nextItem(DexDataBuffer section) {
return section.readMethodId();
}
@Override
protected int getItemSize(MethodId item) {
return item.byteCountInDex();
}
@Override
protected int getFullPatchSectionBase() {
if (this.patchFile != null) {
return this.patchFile.getPatchedMethodIdSectionOffset();
} else {
return getTocSection(this.oldDex).off;
}
}
@Override
protected MethodId adjustItem(IndexMap indexMap, MethodId item) {
return indexMap.adjust(item);
}
@Override
protected int writePatchedItem(MethodId patchedItem) {
++this.patchedMethodIdTocSec.size;
return this.patchedMethodIdSec.writeMethodId(patchedItem);
}
@Override
protected void updateIndexOrOffset(IndexMap indexMap, int oldIndex, int oldOffset, int newIndex, int newOffset) {
if (oldIndex != newIndex) {
indexMap.mapMethodIds(oldIndex, newIndex);
}
}
@Override
protected void markDeletedIndexOrOffset(IndexMap indexMap, int deletedIndex, int deletedOffset) {
indexMap.markMethodIdDeleted(deletedIndex);
}
}