Skip to content

Commit

Permalink
feat:添加对字节码中存放参数的寄存器的解析
Browse files Browse the repository at this point in the history
  • Loading branch information
@Yricky
Yricky committed Aug 30, 2024
1 parent 24ad99b commit d72d6b4
Show file tree
Hide file tree
Showing 6 changed files with 119 additions and 79 deletions.
161 changes: 82 additions & 79 deletions abcdecoder/src/jvmMain/kotlin/me/yricky/abcde/page/CodeView.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,11 +12,8 @@ import androidx.compose.ui.Alignment
import androidx.compose.ui.Modifier
import androidx.compose.ui.draw.alpha
import androidx.compose.ui.draw.clip
import androidx.compose.ui.draw.drawBehind
import androidx.compose.ui.geometry.Size
import androidx.compose.ui.graphics.Color
import androidx.compose.ui.platform.LocalClipboardManager
import androidx.compose.ui.platform.LocalDensity
import androidx.compose.ui.text.*
import androidx.compose.ui.unit.dp
import kotlinx.coroutines.flow.collectLatest
Expand All @@ -31,19 +28,21 @@ import me.yricky.oh.abcd.code.Code
import me.yricky.oh.abcd.code.TryBlock
import me.yricky.oh.abcd.isa.*
import me.yricky.oh.abcd.isa.util.ExternModuleParser
import me.yricky.oh.abcd.isa.util.V2AInstParser

class CodeView(val code: Code,override val hap:HapView? = null):AttachHapPage() {
companion object{
const val ANNO_TAG = "ANNO_TAG"
const val ANNO_ASM_NAME = "ANNO_ASM_NAME"

val operandParser = listOf(ExternModuleParser)
val operandParser = listOf(V2AInstParser,ExternModuleParser)
fun tryBlockString(tb:TryBlock):String = "TryBlock[0x${tb.startPc.toString(16)},0x${(tb.startPc + tb.length).toString(16)})"
}
override val navString: String = "${hap?.navString ?: ""}${asNavString("ASM", code.method.defineStr(true))}"
override val name: String = "${hap?.name ?: ""}/${code.method.abc.tag}/${code.method.clazz.name}/${code.method.name}"

private val asmViewInfo:List<Pair<Asm.AsmItem, AnnotatedString>> by lazy{
code.asm.list.map {
private fun genAsmViewInfo():List<Pair<Asm.AsmItem, AnnotatedString>> {
return code.asm.list.map {
buildAnnotatedString {
append(buildAnnotatedString {
val asmName = it.asmName
Expand Down Expand Up @@ -82,6 +81,9 @@ class CodeView(val code: Code,override val hap:HapView? = null):AttachHapPage()
}
}

private var asmViewInfo:List<Pair<Asm.AsmItem, AnnotatedString>> by mutableStateOf(genAsmViewInfo())
private var showLabel:Boolean by mutableStateOf(false)

@OptIn(ExperimentalFoundationApi::class)
@Composable
override fun Page(modifier: Modifier, hapSession: HapSession, appState: AppState) {
Expand All @@ -91,21 +93,32 @@ class CodeView(val code: Code,override val hap:HapView? = null):AttachHapPage()
Image(Icons.asm(), null, Modifier.fillMaxSize())
} to composeContent {
Column(Modifier.fillMaxSize()) {
Text(
"寄存器数量:${code.numVRegs}, 参数数量:${code.numArgs}, 指令字节数:${code.codeSize}, TryCatch数:${code.triesSize}",
modifier = Modifier.padding(horizontal = 4.dp)
)
Row(
modifier = Modifier.padding(vertical = 4.dp).padding(end = 8.dp),
verticalAlignment = Alignment.CenterVertically
) {
when(val clz = code.method.clazz){
is FieldType.ClassType -> Image(Icons.clazz(),null, modifier = Modifier.clickable {
hapSession.openClass(hap,clz.clazz as AbcClass)
})
else -> {}
}

Spacer(Modifier.weight(1f))
if(code.tryBlocks.isNotEmpty()){
Checkbox(showLabel,{ showLabel = it },Modifier.size(24.dp))
Text("展示TryCatch标签")
}
}
Box(
Modifier.fillMaxWidth().weight(1f).padding(8.dp)
Modifier.fillMaxWidth().weight(1f).padding(end = 8.dp, bottom = 8.dp)
.clip(RoundedCornerShape(8.dp))
.border(2.dp, MaterialTheme.colorScheme.primaryContainer, RoundedCornerShape(8.dp))
.padding(8.dp)
) {
MultiNodeSelectionContainer {
var tryBlock by remember {
mutableStateOf<TryBlock?>(null)
}
val range = multiNodeSelectionState.rememberSelectionChange()
//处理文本选择等操作
LaunchedEffect(null){
textActionFlow.collectLatest { when(it){
is TextAction.Click -> {
Expand Down Expand Up @@ -140,7 +153,14 @@ class CodeView(val code: Code,override val hap:HapView? = null):AttachHapPage()
else -> { }
} }
}
//汇编代码展示
LazyColumnWithScrollBar {
item {
Text(
"寄存器个数:${code.numVRegs}, 参数个数:${code.numArgs}, 指令字节数:${code.codeSize}, TryCatch数:${code.triesSize}",
style = codeStyle
)
}
item {
Text(code.method.defineStr(true), style = codeStyle)
}
Expand All @@ -152,58 +172,20 @@ class CodeView(val code: Code,override val hap:HapView? = null):AttachHapPage()
}
}
Text(line, style = codeStyle)
val tb = remember(item) { item.tryBlocks }
ContextMenuArea(
items = {
buildList<ContextMenuItem> {
if (tryBlock != null) {
add(ContextMenuItem("隐藏行高亮") {
tryBlock = null
})
}
code.tryBlocks.forEach {
add(
ContextMenuItem(
"高亮 TryBlock[0x${
it.startPc.toString(
16
)
},0x${(it.startPc + it.length).toString(16)}]"
) {
tryBlock = it
}
)
}
buildList {
item.calledMethods.forEach {
add(ContextMenuItem("跳转到${it.name}"){
hapSession.openCode(hap,it)
})
}
}
}
},
) {
Text(
String.format("%04X ", item.codeOffset),
style = codeStyle.copy(color = commentColor),
modifier = with(Modifier) {
val density = LocalDensity.current
if (tryBlock != null) {
drawBehind {
if (tb.contains(tryBlock)) {
drawRect(
Color.Yellow,
size = Size(density.density * 2, size.height)
)
}
}
} else this
}.let { m ->
if (tryBlock?.catchBlocks?.find { item.codeOffset in (it.handlerPc until (it.handlerPc + it.codeSize)) } != null) {
m.background(MaterialTheme.colorScheme.errorContainer)
} else {
m
}
}
style = codeStyle.copy(color = commentColor)
)
}
TooltipArea(tooltip = {
Expand All @@ -217,32 +199,53 @@ class CodeView(val code: Code,override val hap:HapView? = null):AttachHapPage()
val layoutResult = remember { mutableStateOf<TextLayoutResult?>(null) }
val selectColor = LocalTextSelectionColors.current.backgroundColor
val thisRange = range?.rangeOf(index,asmStr)
Text(
text = remember(thisRange) {
if(thisRange == null || thisRange.isEmpty()){
asmStr
} else {
val sp = asmStr.spanStyles.plus(
AnnotatedString.Range(
SpanStyle(background = selectColor),
thisRange.start,
thisRange.endExclusive,
Column {
if(showLabel) code.tryBlocks.asSequence().filter { it.length > 0 }.forEach {
if(it.startPc == item.codeOffset){
Text("${tryBlockString(it)}_begin", style = MaterialTheme.typography.labelSmall)
}
it.catchBlocks.forEach { cb ->
if(cb.handlerPc == item.codeOffset){
Text("${tryBlockString(it)}_catch${if(cb.codeSize > 0) "_begin" else ""}", style = MaterialTheme.typography.labelSmall)
}
}
}
Text(
text = remember(thisRange) {
if(thisRange == null || thisRange.isEmpty()){
asmStr
} else {
val sp = asmStr.spanStyles.plus(
AnnotatedString.Range(
SpanStyle(background = selectColor),
thisRange.start,
thisRange.endExclusive,
)
)
AnnotatedString(
asmStr.text,
sp,
asmStr.paragraphStyles,
)
)
AnnotatedString(
asmStr.text,
sp,
asmStr.paragraphStyles,
)
}
},
style = codeStyle,
modifier = Modifier
.withMultiNodeSelection({ layoutResult.value },index)
.fillMaxWidth(),
onTextLayout = { layoutResult.value = it },
)
if (showLabel) code.tryBlocks.asSequence().filter { it.length > 0 }.forEach {
it.catchBlocks.forEach { cb ->
if(cb.codeSize > 0 && cb.handlerPc + cb.codeSize == item.nextOffset){
Text("${tryBlockString(it)}_catch_end", style = MaterialTheme.typography.labelSmall)
}
}
},
style = codeStyle,
modifier = Modifier
.withMultiNodeSelection({ layoutResult.value },index)
.fillMaxWidth(),
onTextLayout = { layoutResult.value = it },
)
Text("\n", maxLines = 1, style = codeStyle)
if(it.startPc + it.length == item.nextOffset){
Text("${tryBlockString(it)}_end", style = MaterialTheme.typography.labelSmall)
}
}
}
}
}
}
Expand Down
1 change: 1 addition & 0 deletions modules/abcde/src/commonMain/kotlin/me/yricky/oh/abcd/isa/util/BaseInstParser.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ import me.yricky.oh.abcd.isa.Asm
import me.yricky.oh.abcd.isa.InstFmt

object BaseInstParser:InstDisAsmParser {
override fun id(): String = "${InstDisAsmParser.ID_PREFIX}.base"
override fun title(): String = "基础反汇编器"
override fun description(): String = "提供基础的反汇编功能"

Expand Down
1 change: 1 addition & 0 deletions modules/abcde/src/commonMain/kotlin/me/yricky/oh/abcd/isa/util/ExternModuleParser.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ import me.yricky.oh.abcd.isa.Asm
import me.yricky.oh.abcd.isa.InstFmt

object ExternModuleParser:InstDisAsmParser {
override fun id(): String = "${InstDisAsmParser.ID_PREFIX}.extern_module"
override fun title(): String = "ExternModule拓展"
override fun description(): String = "将字节码中的模块导入指令解析为可读格式"

Expand Down
2 changes: 2 additions & 0 deletions modules/abcde/src/commonMain/kotlin/me/yricky/oh/abcd/isa/util/InstCommentParser.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,8 @@ interface InstCommentParser {

fun parse(asmItem: Asm.AsmItem):String?
companion object{
const val ID_PREFIX = "abcde.asm.comment.parser"

fun commentString(item: Asm.AsmItem, externalParser:List<InstCommentParser> = listOf(RawByteCommentParser,LineNumberCommentParser)):String{
val sb = StringBuilder()
externalParser.forEach {
Expand Down
5 changes: 5 additions & 0 deletions modules/abcde/src/commonMain/kotlin/me/yricky/oh/abcd/isa/util/InstDisAsmParser.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@ package me.yricky.oh.abcd.isa.util
import me.yricky.oh.abcd.isa.Asm

interface InstDisAsmParser {
fun id():String
fun title():String? = null
fun description():String? = null

Expand All @@ -11,4 +12,8 @@ interface InstDisAsmParser {
* @param index [Asm.AsmItem.opUnits]的下标
*/
fun parseArg(asmItem: Asm.AsmItem, index:Int):String?

companion object{
const val ID_PREFIX = "abcde.asm.parser"
}
}
28 changes: 28 additions & 0 deletions modules/abcde/src/commonMain/kotlin/me/yricky/oh/abcd/isa/util/V2AInstParser.kt
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
package me.yricky.oh.abcd.isa.util

import me.yricky.oh.abcd.isa.Asm
import me.yricky.oh.abcd.isa.InstFmt

object V2AInstParser:InstDisAsmParser {
override fun id(): String = "${InstDisAsmParser.ID_PREFIX}.v2a"
override fun title(): String = "参数寄存器解析插件"
override fun description(): String = "解析字节码中的参数寄存器"

override fun parseArg(asmItem: Asm.AsmItem, index: Int): String? {
val arg = asmItem.opUnits[index]
val format = asmItem.ins.format
return when(format[index]){
is InstFmt.RegV -> {
val v = arg.toInt()
val vRegs = asmItem.asm.code.numVRegs
if (v < vRegs) "v${v}" else when(val aIndex = v - vRegs){
0 -> "FunctionObject"
1 -> "NewTarget"
2 -> "this"
else -> "arg${aIndex - 3}"
}
}
else -> null
}
}
}

0 comments on commit d72d6b4

Please sign in to comment.