z3约束求解器是解决多元问题的好工具
其应用广泛,比如,著名的二进制分析框架angr也内置了修改版的z3
安装下载
虽然z3具有适用于各种编程语言的绑定(由C++开发,提供了 .NET、C、C++、Java、Python 等语言调用接口),但更多以及方便的是python接口。以python接口进行详讲:
GitHub直通车:GitHub - Z3Prover/z3: The Z3 Theorem Prover
准备好Python环境和pip,就可以使用如下命令安装:
pip install z3-solver
# 一般直接下速度很慢,推荐使用清华源等国内镜像源
# 清华源 -i <https://pypi.tuna.tsinghua.edu.cn/simple>
z3_微软开发的SMT求解器
什么是SMT?
- SAT-布尔可满足性问题
- SMT-可满足性模理论
若简化理解的话:SMT = SAT + Theory Solvers(理论求解器)
特点 | SAT | SMT |
---|---|---|
逻辑范围 | 仅限于布尔逻辑 | 支持多种理论,包括整数、实数、数组、位向量等 |
公式表示 | 通常是 CNF 或其他布尔逻辑表示 | 可以是混合的逻辑公式,包含不同理论中的变量和约束 |
求解器框架 | 基于 DPLL 或 CDCL 算法 | 基于 DPLL(T) 框架,结合了 SAT 求解器和理论求解器 |
表达能力 | 只能处理简单的布尔变量关系 | 能处理复杂的约束和多样化的数据类型 |
相比较而言,SMT 能够处理更复杂的约束和数据类型,可以说是SAT的扩展
使用z3
在确保你的python环境以及安装好z3-solver后
python脚本中使用如下导入相应的包
from z3 import *
常用的API
API | 功能 |
---|---|
Solver() | 创建一个通用求解器 |
add() | 添加约束条件 |
check() | 用来判断在添加完约束条件后,来检测解的情况 |
model() | 在存在解的时候,该函数会将每个限制条件所对应的解集取交集,进而得出正解 |
变量声明
z3中有3种类型的变量,分别是整型(Int)、实型(Real)和向量(BitVec)
只有向量类型可以进行位运算(异或,左右移位等)
from z3 import *
# 声明整形
a = Int('a') # 声明单个整型变量
a,b = Ints('a b') # 声明多个整型变量
# 声明实性
a = Real('a') # 声明单个实型变量
a,b = Reals('a b') # 声明多个实型变量
# 声明向量
a = BitVec('a', 8) # 声明单个8位的变量
a, b = BitVec('a b', 8) # 声明多个8位的变量
# tips:更多使用的是循环声明变量
# 更方便创建同类型多元
a1 = [Int(f'a1[{i}]') for i in range(32)]
a2 = [Real(f'a2[{i}]') for i in range(32)]
a3 = [BitVec(f'a3[{i}]', 8) for i in range(32)] # BitVec类型创建时必须给出大小
如何使用z3
使用一些CTF题例来进行讲解(以注释形式)
[HGAME 2023 week3]kunmusic
from z3 import *
taa = [132, 47, 180, 7, 216, 45, 68, 6, 39, 246, 124, 2, 243, 137, 58, 172, 53,
200, 99, 91, 83, 13, 171, 80, 108, 235, 179, 58, 176, 28, 216, 36, 11, 80, 39,
162, 97, 58, 236, 130, 123, 176, 24, 212, 56, 89, 72]
# 涉及位运算,创建BitVec变量(其中32是指的指的是变量的宽度)
num = [BitVec(f'num[{i}]', 32) for i in range(13)]
s = Solver() # 创建一个约束求解器s
# 添加约束条件
s.add(num[0] + 52296 + num[1] - 26211 + num[2] - 11754 + (num[3] ^ 41236) +
num[4] * 63747 + num[5] - 52714 + num[6] - 10512 + num[7] * 12972 + num[8] +
45505 + num[9] - 21713 + num[10] - 59122 + num[11] - 12840 + (num[12] ^ 21087) ==
12702282)
s.add(num[0] - 25228 + (num[1] ^ 20699) + (num[2] ^ 8158) + num[3] - 65307 +
num[4] * 30701 + num[5] * 47555 + num[6] - 2557 + (num[7] ^ 49055) + num[8] -
7992 + (num[9] ^ 57465) + (num[10] ^ 57426) + num[11] + 13299 + num[12] - 50966
== 9946829)
s.add(num[0] - 64801 + num[1] - 60698 + num[2] - 40853 + num[3] - 54907 + num[4]
+ 29882 + (num[5] ^ 13574) + (num[6] ^ 21310) + num[7] + 47366 + num[8] + 41784 +
(num[9] ^ 53690) + num[10] * 58436 + num[11] * 15590 +num[12] + 58225 == 2372055)
s.add(num[0] + 61538 + num[1] - 17121 + num[2] - 58124 + num[3] + 8186 + num[4] +
21253 + num[5] - 38524 + num[6] - 48323 + num[7] - 20556 + num[8] * 56056 +
num[9] + 18568 + num[10] + 12995 + (num[11] ^ 39260) + num[12] + 25329 ==
6732474)
s.add(num[0] - 42567 + num[1] - 17743 + num[2] * 47827 + num[3] - 10246 + (num[4]
^ 16284) + num[5] + 39390 + num[6] * 11803 + num[7] * 60332 + (num[8] ^ 18491) +
(num[9] ^ 4795) + num[10] - 25636 + num[11] - 16780 + num[12] - 62345 ==
14020739)
s.add(num[0] - 10968 + num[1] - 31780 + (num[2] ^ 31857) + num[3] - 61983 +
num[4] * 31048 + num[5] * 20189 + num[6] + 12337 + num[7] * 25945 + (num[8] ^
7064) + num[9] - 25369 + num[10] - 54893 + num[11] * 59949 + (num[12] ^ 12441) ==
14434062)
s.add(num[0] + 16689 + num[1] - 10279 + num[2] - 32918 + num[3] - 57155 + num[4]
* 26571 + num[5] * 15086 + (num[6] ^ 22986) + (num[7] ^ 23349) + (num[8] ^ 16381)
+ (num[9] ^ 23173) + num[10] - 40224 + num[11] + 31751 + num[12] * 8421 ==
7433598)
s.add(num[0] + 28740 + num[1] - 64696 + num[2] + 60470 + num[3] - 14752 + (num[4]
^ 1287) + (num[5] ^ 35272) + num[6] + 49467 + num[7] - 33788 + num[8] + 20606 +
(num[9] ^ 44874) + num[10] * 19764 + num[11] + 48342 + num[12] * 56511 ==
7989404)
s.add((num[0] ^ 28978) + num[1] + 23120 + num[2] + 22802 + num[3] * 31533 +
(num[4] ^ 39287) + num[5] - 48576 + (num[6] ^ 28542) + num[7] - 43265 + num[8] +
22365 + num[9] + 61108 + num[10] * 2823 + num[11] - 30343 + num[12] + 14780 ==
3504803)
s.add(num[0] * 22466 + (num[1] ^ 55999) + num[2] - 53658 + (num[3] ^ 47160) +
(num[4] ^ 12511) + num[5] * 59807 + num[6] + 46242 + num[7] + 3052 + (num[8] ^
25279) + num[9] + 30202 + num[10] * 22698 + num[11] + 33480 + (num[12] ^ 16757)
== 11003580)
s.add(num[0] * 57492 + (num[1] ^ 13421) + num[2] - 13941 + (num[3] ^ 48092) +
num[4] * 38310 + num[5] + 9884 + num[6] - 45500 + num[7] - 19233 + num[8] + 58274
+ num[9] + 36175 + (num[10] ^ 18568) + num[11] * 49694 + (num[12] ^ 9473) ==
25546210)
s.add(num[0] - 23355 + num[1] * 50164 + (num[2] ^ 34618) + num[3] + 52703 +
num[4] + 36245 + num[5] * 46648 + (num[6] ^ 4858) + (num[7] ^ 41846) + num[8] *
27122 + (num[9] ^ 42058) + num[10] * 15676 + num[11] - 31863 +num[12] + 62510 ==
11333836)
s.add(num[0] * 30523 + (num[1] ^ 7990) + num[2] + 39058 + num[3] * 57549 +
(num[4] ^ 53440) + num[5] * 4275 + num[6] - 48863 + (num[7] ^ 55436) + (num[8] ^
2624) + (num[9] ^ 13652) + num[10] + 62231 + num[11] + 19456 + num[12] - 13195 ==
13863722)
s.check() # 尝试求解
m = s.model() # 将每个限制条件所对应的解集取交集,存入m中
a = []
# 将解导入a中
for i in range(len(m)):
a.append(int(str(m[num[i]])))
# 题目额外问题,下面与z3无关
flag = ""
for i in range(47):
flag += chr((taa[i] ^ a[i % 13]) % 0x80)
print(flag)
[NSSRound#X Basic]ez_z3
from z3 import *
# 全是数运算,创建Int变量
a1 = [Int(f'a1[{i}]') for i in range(20)]
x = Solver() # 创建约束求解器x
# 添加约束条件
x.add(20 * a1[19] * 19 * a1[18]
+ 14 * a1[13]
+ 13 * a1[12]
+ 11 * a1[10] * 10 * a1[9]
+ 30 * a1[5]
+ 5 * a1[4]
+ a1[0]
+ 2 * a1[1]
- 3 * a1[2]
- 4 * a1[3]
- 7 * a1[6]
+ 8 * a1[7]
- 9 * a1[8]
- 12 * a1[11]
- 16 * a1[15] * 15 * a1[14]
- 17 * a1[16]
- 18 * a1[17] == 2582239)
x.add(20 * a1[19] * 19 * a1[18]
+ 14 * a1[13]
+ 13 * a1[12]
+ 11 * a1[10] * 10 * a1[9]
+ 30 * a1[5]
- 7 * a1[6]
+ 8 * a1[7]
- 9 * a1[8]
+ 5 * a1[4]
+ 3 * a1[2]
+ 2 * a1[1] * a1[0]
- 4 * a1[3]
- 12 * a1[11]
- 16 * a1[15] * 15 * a1[14]
- (18 * a1[17]
+ 17 * a1[16]) == 2602741)
x.add(19 * a1[18]
+ 18 * a1[17]
+ 14 * a1[13] * 13 * a1[12]
+ 12 * a1[11] * 11 * a1[10]
+ 9 * a1[8]
+ 7 * a1[6] * 30 * a1[5]
+ a1[0]
- 2 * a1[1]
- 4 * a1[3] * 3 * a1[2]
- 5 * a1[4]
+ 8 * a1[7]
- 10 * a1[9]
- 15 * a1[14]
- 17 * a1[16] * 16 * a1[15]
- 20 * a1[19] == 2668123)
x.add(20 * a1[19] * 19 * a1[18]
+ 14 * a1[13]
+ (13 * a1[12] + 11 * a1[10] - 12 * a1[11]) * 10 * a1[9]
+ 30 * a1[5]
+ 5 * a1[4]
+ a1[0]
+ 2 * a1[1]
- 3 * a1[2]
- 4 * a1[3]
- 7 * a1[6]
+ 8 * a1[7]
- 9 * a1[8]
- 16 * a1[15] * 15 * a1[14]
- 17 * a1[16]
- 18 * a1[17] == 2520193)
x.add(18 * a1[17]
+ 17 * a1[16]
+ 15 * a1[14]
+ 13 * a1[12] * 12 * a1[11]
+ 10 * a1[9]
+ 9 * a1[8] * 8 * a1[7]
+ 3 * a1[2] * 2 * a1[1] * a1[0]
- 4 * a1[3]
- 5 * a1[4]
- 30 * a1[5]
- 7 * a1[6]
- 11 * a1[10]
- 14 * a1[13]
- 16 * a1[15]
- 19 * a1[18]
- 20 * a1[19] == 8904587)
x.add(18 * a1[17]
+ 7 * a1[6] * 30 * a1[5] * 5 * a1[4]
+ 4 * a1[3]
+ 8 * a1[7]
+ a1[0]
- 2 * a1[1]
- 3 * a1[2]
- 9 * a1[8]
- 11 * a1[10] * 10 * a1[9]
- 16 * a1[15] * (13 * a1[12] + 12 * a1[11] - 14 * a1[13] - 15 * a1[14])
- 17 * a1[16]
- 19 * a1[18]
- 20 * a1[19] == 1227620874)
x.add(20 * a1[19] * 19 * a1[18]
+ 17 * a1[16]
+ 14 * a1[13]
+ 13 * a1[12]
+ 12 * a1[11] * 11 * a1[10] * 10 * a1[9]
+ 7 * a1[6] * 30 * a1[5]
+ 5 * a1[4]
+ 3 * a1[2]
+ a1[0]
+ 2 * a1[1]
+ 4 * a1[3]
+ 8 * a1[7]
- 9 * a1[8]
- 16 * a1[15] * 15 * a1[14]
- 18 * a1[17] == 1836606059)
x.add(20 * a1[19] * 19 * a1[18]
+ 16 * a1[15] * 15 * a1[14]
+ 14 * a1[13]
+ 13 * a1[12]
+ 12 * a1[11]
+ 7 * a1[6] * 30 * a1[5]
+ 5 * a1[4]
+ 2 * a1[1] * a1[0]
- 3 * a1[2]
+ 4 * a1[3]
+ 8 * a1[7]
- 9 * a1[8]
- 10 * a1[9]
- 11 * a1[10]
- 17 * a1[16]
- 18 * a1[17] == 8720560)
x.add(20 * a1[19] * 19 * a1[18]
+ 14 * a1[13]
+ 13 * a1[12]
+ 11 * a1[10] * (10 * a1[9] + 30 * a1[5] + 5 * a1[4] + 4 * a1[3] - 7 * a1[6] + 8 * a1[7] - 9 * a1[8])
+ a1[0]
+ 2 * a1[1]
- 3 * a1[2]
- 12 * a1[11]
- (16 * a1[15] - 17 * a1[16] - 18 * a1[17]) * 15 * a1[14] == 11387045)
x.add(20 * a1[19] * 19 * a1[18]
+ 16 * a1[15] * 15 * a1[14]
+ 14 * a1[13]
+ 11 * a1[10] * 10 * a1[9]
+ 9 * a1[8]
+ 3 * a1[2]
+ a1[0]
- 2 * a1[1]
+ 4 * a1[3]
- 5 * a1[4]
- 30 * a1[5]
- 7 * a1[6]
+ 8 * a1[7]
- 12 * a1[11]
- 13 * a1[12]
- 17 * a1[16]
- 18 * a1[17] == 7660269)
x.add(20 * a1[19] * 19 * a1[18]
+ 14 * a1[13]
+ 13 * a1[12]
+ 11 * a1[10] * 10 * a1[9]
- 12 * a1[11]
+ a1[0]
+ 2 * a1[1]
- (4 * a1[3] * 3 * a1[2]
- 5 * a1[4]
- 30 * a1[5])
- 7 * a1[6]
+ 8 * a1[7]
- 9 * a1[8]
- 16 * a1[15] * 15 * a1[14]
- 17 * a1[16]
- 18 * a1[17] == 2461883)
x.add(14 * a1[13]
+ 11 * a1[10] * 10 * a1[9]
+ 9 * a1[8] * 8 * a1[7]
+ 7 * a1[6]
+ 2 * a1[1] * a1[0]
- 4 * a1[3] * 3 * a1[2]
- 5 * a1[4]
- 30 * a1[5]
- 12 * a1[11]
- 13 * a1[12]
- 15 * a1[14]
- 17 * a1[16] * 16 * a1[15]
- 18 * a1[17]
- 19 * a1[18]
- 20 * a1[19] == -966296)
x.add(14 * a1[13]
+ 13 * a1[12]
+ (11 * a1[10] * 10 * a1[9] + 30 * a1[5] + 5 * a1[4] + 3 * a1[2] + 4 * a1[3] - 7 * a1[6] + 8 * a1[7] - 9 * a1[8])
* 2
* a1[1]
+ a1[0]
- 12 * a1[11]
- 15 * a1[14]
- 16 * a1[15]
- 17 * a1[16]
- 18 * a1[17]
- 20 * a1[19] * 19 * a1[18] == 254500223)
x.add(16 * a1[15] * 15 * a1[14]
+ 14 * a1[13]
+ 11 * a1[10] * 10 * a1[9]
+ 7 * a1[6] * 30 * a1[5]
+ a1[0]
- 2 * a1[1]
- 3 * a1[2]
- 5 * a1[4] * 4 * a1[3]
+ 8 * a1[7]
- 9 * a1[8]
- 12 * a1[11]
- 13 * a1[12]
- 17 * a1[16]
- 18 * a1[17]
- 19 * a1[18]
- 20 * a1[19] == 6022286)
x.add(18 * a1[17]
+ 16 * a1[15]
- 17 * a1[16]
+ 14 * a1[13]
+ 12 * a1[11]
+ 11 * a1[10] * 10 * a1[9]
+ 30 * a1[5]
+ 5 * a1[4]
+ 4 * a1[3] * 3 * a1[2]
+ 2 * a1[1] * a1[0]
- 9 * a1[8] * 8 * a1[7] * 7 * a1[6]
- 13 * a1[12]
- 15 * a1[14]
- 19 * a1[18]
- 20 * a1[19] == -636956022)
x.add(20 * a1[19] * 19 * a1[18]
+ 13 * a1[12]
+ 12 * a1[11]
+ 11 * a1[10] * 10 * a1[9]
+ 7 * a1[6]
+ 30 * a1[5]
+ 5 * a1[4]
+ 3 * a1[2] * 2 * a1[1] * a1[0]
- 4 * a1[3]
- 9 * a1[8] * 8 * a1[7]
- 14 * a1[13]
- 15 * a1[14]
- 16 * a1[15]
- 17 * a1[16]
- 18 * a1[17] == 10631829)
x.add(20 * a1[19] * 19 * a1[18]
+ 16 * a1[15]
- 17 * a1[16]
- 18 * a1[17]
+ 15 * a1[14] * 14 * a1[13]
+ 13 * a1[12]
+ 11 * a1[10] * 10 * a1[9]
- 12 * a1[11]
+ 7 * a1[6]
+ (4 * a1[3] - 5 * a1[4] - 30 * a1[5]) * 3 * a1[2]
+ a1[0]
+ 2 * a1[1]
+ 8 * a1[7]
- 9 * a1[8] == 6191333)
x.add(14 * a1[13]
+ 10 * a1[9] * 9 * a1[8] * 8 * a1[7]
+ 5 * a1[4]
+ 4 * a1[3] * 3 * a1[2]
+ 2 * a1[1] * a1[0]
- 7 * a1[6] * 30 * a1[5]
- 11 * a1[10]
- 13 * a1[12] * 12 * a1[11]
- 16 * a1[15] * 15 * a1[14]
- 18 * a1[17] * 17 * a1[16]
- 20 * a1[19] * 19 * a1[18] == 890415359)
x.add(20 * a1[19]
+ 19 * a1[18]
+ 18 * a1[17]
+ 16 * a1[15]
- 17 * a1[16]
+ 12 * a1[11]
+ 11 * a1[10]
+ 10 * a1[9]
+ 9 * a1[8]
+ 30 * a1[5]
+ a1[0]
+ 4 * a1[3] * 3 * a1[2] * 2 * a1[1]
- 5 * a1[4]
- 7 * a1[6]
+ 8 * a1[7]
- 13 * a1[12]
- 14 * a1[13]
- 15 * a1[14] == 23493664)
x.add(20 * a1[19] * 19 * a1[18]
+ 13 * a1[12]
+ 12 * a1[11]
+ 10 * a1[9]
+ 3 * a1[2] * 2 * a1[1]
+ a1[0]
- 4 * a1[3]
- 5 * a1[4]
+ 8 * a1[7] * 7 * a1[6] * 30 * a1[5]
- 9 * a1[8]
- 11 * a1[10]
- 14 * a1[13]
- 16 * a1[15] * 15 * a1[14]
- 17 * a1[16]
- 18 * a1[17] == 1967260144)
# 添加约束条件,将解限制在0~256,简化运算范围,加快脚本速度
for i in a1:
x.add(i >= 0, i <= 0xff)
if x.check() == sat: # 表示判断约束是否可满足
m = x.model() # 将解导入m
print(m)
[CISCN 2021初赛]babybc
用z3搓个求解器解出满足的数独
from z3 import *
row = [[0, 0, 0, 1],
[1, 0, 0, 0],
[2, 0, 0, 1],
[0, 0, 0, 0],
[1, 0, 1, 0]]
col = [[0, 0, 2, 0, 2],
[0, 0, 0, 0, 0],
[0, 0, 0, 1, 0],
[0, 1, 0, 0, 1]]
s = Solver() # 创建约束求解器s
maps = [[0] * 5,
[0] * 5,
[0] * 5,
[0] * 5,
[0] * 5]
for i in range(5):
for j in range(5):
maps[i][j] = Int("x%d%d" % (i + 1, j + 1)) # 先初始化设置未知数
s.add(maps[2][2] == 4, maps[3][3] == 3) # 把map数组已知的两个值填入
for i in range(5):
for j in range(5):
s.add(maps[i][j] >= 1, maps[i][j] <= 5) # 约束填入数据在1-5中
for i in range(5):
for j in range(5):
for k in range(j):
s.add(maps[i][j] != maps[i][k]) # 一行中不能有相同数据
for i in range(5):
for j in range(5):
for k in range(i):
s.add(maps[i][j] != maps[k][j]) # 一列中不能有相同数据
for i in range(5):
for j in range(4):
if row[i][j] == 1:
s.add(maps[i][j] > maps[i][j + 1])
elif row[i][j] == 2:
s.add(maps[i][j] < maps[i][j + 1]) # row如果有1,map中对应位置要大于右边;row如果有2,map中对应对应位置要小于右边
for i in range(4):
for j in range(5):
if col[i][j] == 2:
s.add(maps[i][j] > maps[i + 1][j])
elif col[i][j] == 1:
s.add(maps[i][j] < maps[i + 1][j]) # col如果有1,map中对应位置数据小于下边;col如果有2,map中对应位置数据要大于下边
answer = s.check()
if answer == sat:
print(s.model())
num = s.model()
flag = []
for i in maps:
for j in i:
flag.append(int(str(num[j])))
for i in range(len(flag)):
flag[i] += 48
flag[12] = 48
flag[18] = 48
flag = bytes(flag)
print(flag)