[MYSQL] 尽可能的从坏块中提取数据 (实战篇)

#!/usr/bin/env python
# write by ddcw @https://github.com/ddcw
# 模拟mysql坏块的, 此脚本非常危险, 是在原数据文件上修改(节省空间). 不要用于任何生产/测试.
# 为了保险期间, 每次操作都要输入一个随机数确认.

import struct
import sys
import os
import random
import argparse

def _argparse():
	parser = argparse.ArgumentParser(add_help=True,description="模拟坏块的脚本.不要用于任何生产/测试")
	parser.add_argument("--set",dest="SET_OPTIONS",action='append',help="options:fil_header,page_header,page_dir,random,pageno")
	parser.add_argument(dest='FILENAME', help='将原地破坏这个ibd文件!!!!!', nargs='*')
	parser = parser.parse_args()
	return parser

# 破坏fil_header
def damage_fil_header(data):
	return b'0xff'*38+data[38:]

# 破坏index page的page header
def damage_page_header(data):
	return data[:38] + b'0xff'*56 + data[94:]

# 破坏page directory
def damage_page_dir(data,n=1):
	return data[:-16] + b'0xff'*8 + data[-8:]

# 破坏user record, 随机行,随机位置
def damage_random(data,n=1):
	start = random.randint(120,8000)
	size = random.randint(10,200)
	return data[:start] + b'0xff'*size + data[start+size:]

def main():
	parser = _argparse()
	opt = {}
	if parser.SET_OPTIONS is not None:
		for x in parser.SET_OPTIONS:
			for y in x.split(';'):
				for z in y.split(','):
					kv = z.split('=')
					if len(kv) == 2:
						opt[kv[0]] = kv[1]
					elif len(kv) == 1 and z != '':
						opt[kv[0]] = True

	filename = parser.FILENAME[0]
	flag = str(random.randint(1000,9999))
	if 'skip' not in opt:
		print('输入如下数字表示开始破坏文件xxx',flag,'\n我的输入: ',end='')
		user_input = sys.stdin.readline()
		if user_input[:-1] != flag:
			print('输错了, 就没有后续了')
			sys.exit(1)
		else:
			print('既然你已求死, 那便整起!')
	pages = os.path.getsize(filename)//16384
	pageno = random.randint(4,pages) if 'pageno' not in opt else int(opt['pageno'])
	pageno = min(pages,pageno)
	print('BAD PAGE:',pageno)
	with open(filename,'r+b') as f:
		f.seek(16384*pageno,0)
		data = f.read(16384)
		if 'fil_header' in opt:
			data = damage_fil_header(data)
		if 'page_header' in opt:
			data = damage_page_header(data)
		if 'page_dir' in opt:
			data = damage_page_dir(data)
		if 'random' in opt:
			data = damage_random(data)
		f.seek(16384*pageno,0)
		f.write(data)

if __name__ == '__main__':
	main()

# 备份下
cp -ra /data/mysql_3314/mysqldata/db1/sbtest2.ibd /tmp/sbtest2.ibd

# 记录下正常的数据
python3 main.py /tmp/sbtest2.ibd --ddl --sql > /tmp/sbtest2.sql

# 随机破坏下(可以多整几次, 怕万一没坏到关键处.)
python3 mysql_ibd_corruption_danger.py /tmp/sbtest2.ibd --set random

python3 main.py /tmp/sbtest2.ibd --ddl --sql --force --set bad-pages=fast > /tmp/sbtest2_fast.sql

# 少了的数据
diff /tmp/sbtest2_fast.sql /tmp/sbtest2.sql | grep '^>' | wc -l
# 多了的数据
diff /tmp/sbtest2_fast.sql /tmp/sbtest2.sql | grep '^<' | wc -l

python3 main.py /tmp/sbtest2.ibd --ddl --sql --force --set bad-pages=try > /tmp/sbtest2_try.sql

python3 main.py /tmp/sbtest2.ibd --ddl --sql --force --set bad-pages=skip > /tmp/sbtest2_skip.sql