<!--
* @Author: wenling
* @Date: 2022-06-03 19:18:11
* @LastEditors: wenling
* @LastEditTime: 2022-06-05 16:39:50
* @Description: 请填写简介
-->

2022国赛babytree wp

题目给了一个txt文件，里面是swift的抽象语法树

文本很长而且是我不熟悉的swift语言，所以先了解一下swift的抽象语法树吧
swift开发环境搭建：https://www.fatbobman.com/posts/swift-in-linux/

例子：

var str = "hello"

使用如下命令将switf文件转换为ast文件

![图 2]

如图就是一个初始化语句的ast结构：

• pattern_binding_decl：初始化语句
• pattern_named type='String' 'str'：变量名称
• value="hello"：变量值

在题目给出的ast中，有类似的结构

所以这一块结构的意思就是：

var key = "345y"

再看一个例子：

var num = 10
if num == 10 {
    num = 100
}

![图 3]

if语句

• if_stmt：if语句
• declref_expr：比较的对象
• integer_literal_expr：比较的值

赋值语句

• assign_expr：赋值语句
• declref_expr：赋值的对象
• integer_literal_expr：赋值的值

其他的对应关系就不列出了，可以自己写代码对照

• call_expr：函数调用
• declref_expr：调用的函数，decl=Swift.(file).print(_:separator:terminator:)，这里调用的是print函数
• argument：参数

print(result)

这段代码中，调用了print函数，打印的是result的值，result是Bool类型的数据，所以程序会输出true或者false

result变量就在call语句块的上方，可以分析出result是初始化的变量，用来初始化result变量的是一个call语句，调用的函数是check，参数分别是data和key

var result = check(data, key)

再往上可以找到key的定义

var key = "345y"

这里是data的定义，data是有一个subscript_expr语句块定义的，subscript_expr其实是在去数组的下标，这里取的是arguments[1]，也就是第二个命令行参数

var data = CommandLine.arguments[1]

在往上是一个if语句，操作符是 >=（extension.>=），操作数一个是argument.count()，一个是2,所以这一块的意思就是比较命令行参数是否大于等于2

if CommandLine.arguments.count >= 2 {

}

![图 4]

在往上就是top_level_code_decl，顶层表达式了，如图所示，这个程序真是由一个func_decl和一个top_level_code_decl组成，我们已经分析完了下面的top_level_code_decl：

func check(encoded: String, keyValue:String) -> Bool {
    
}

if CommandLine.arguments.count >= 2 {
    let data = CommandLine.arguments[1]
    let key = "345y"
    let result = check(encoded: data, keyValue: key)
    print(result)
}

上方函数的大体结构为：参数，返回值和一个表达式

参数分别为encoded和keyValue，都是String类型

返回值为Bool类型

下方的语句块中又包含了3个定义变量的语句，1个for循环，1个返回语句

第一个定义变量的语句，变量名为b，类型是一个[UInt8]数组，这个数组是由ebcoded（decl=re.(file).check(::).encoded@re.swift）转换过来的

var b = [UInt8](encoded.utf8)

同上，接着定义一个变量k

var k = [UInt8](keyValue.utf8)

接下来的语句块中定义了四个未初始化的变量r0, r1, r2, r3，类型是UInt8

var r0, r1, r2, r3:UInt8

可以看到(decl=Swift.(file).Comparable extension....),说明这是一个for-in循环

循环的范围是从0到b.count-4

for i in 0...b.count-4 {

}

![图 5]

紧跟着的是六条赋值语句
![图 6]

展开发现首尾的赋值语句格式相似,中间四条赋值语句格式相似,所以我们应该只需要仔细分析两种不同格式中的各一种
![图 7]

这是第一个赋值语句的结构,做的事情是用b数组的值填充r0到r3

r0, r1, r2, r3 = b[i], b[i+1], b[i+2], b[i+3]

![图 8]

第二个赋值语句,首先是一个取下标,取的是b[i+0]
![图 9]

![图 10]

接着是一个运算语句块
![图 11]

初步看出是用r2异或一个表达式
该表达式是另一个表达式 & 0xff得到
![图 13]

接着又是加运算
一个是k[0]
![图 15]

一个是r0 >> 4

综上,第二个赋值语句:

b[i+0] = r2 ^ ((k[0] + (r0 >> 4)) & 0xff)

同理可以分析出接下来的几条赋值语句

b[i+1] = r3 ^ ((k[1] + (r1 >> 2)) & 0xff)
b[i+2] = r0 ^ k[2]
b[i+3] = r1 ^ k[3]
k[0], k[1], k[2], k[3] = k[1], k[2], k[3], k[0]

for循环之后就是返回语句
![图 16]

返回语句由一条判断语句组成,操作符是==
![图 17]

第一个参数是b数组
![图 18]
第二个参数是一个给定的数组
[88, 35, 88, 225, 7, 201, 57, 94, 77, 56, 75, 168, 72, 218, 64, 91, 16, 101, 32, 207, 73, 130, 74, 128, 76, 201, 16, 248, 41, 205, 103, 84, 91, 99, 79, 202, 22, 131, 63, 255, 20, 16]

return (b == [88, 35, 88, 225, 7,  201, 57, 94, 77, 56, 75, 168, 72, 218, 64, 91, 16, 101, 32, 207, 73, 130, 74, 128, 76, 201, 16, 248, 41, 205, 103, 84, 91, 99, 79, 202, 22, 131, 63, 255, 20, 16])

re.swift

func check(encoded: String, keyValue:String) -> Bool {
    var b = [UInt8](encoded.utf8)
    var k = [UInt8](keyValue.utf8)
    var r0, r1, r2, r3:UInt8
    for i in 0...b.count-4 {
        r0 = b[i]
        r1 = b[i+1]
        r2 = b[i+2]
        r3 = b[i+3]
        b[i+0] = r2 ^ ((k[0] + (r0 >> 4)) & 0xff)
        b[i+1] = r3 ^ ((k[1] + (r1 >> 2)) & 0xff)
        b[i+2] = r0 ^ k[2]
        b[i+3] = r1 ^ k[3]
        let temp = k[0]
        k[0] = k[1]
        k[1] = k[2]
        k[2] = k[3]
        k[3] = temp
    }
    return (b == [88, 35, 88, 225, 7,  201, 57, 94, 77, 56, 75, 168, 72, 218, 64, 91, 16, 101, 32, 207, 73, 130, 74, 128, 76, 201, 16, 248, 41, 205, 103, 84, 91, 99, 79, 202, 22, 131, 63, 255, 20, 16])
}

if CommandLine.arguments.count >= 2 {
    let data = CommandLine.arguments[1]
    let key = "345y"
    let result = check(encoded: data, keyValue: key)
    print(result)
}

解题脚本

[CTF入门培训]顶尖高校博士及硕士团队亲授《30小时教你玩转CTF》，视频+靶场+题目！助力进入CTF世界