#include <stdio.h>
#include <stdlib.h>

/* run this program using the console pauser or add your own getch, system("pause") or input loop */

const char bios_head[32]=
{
0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0xFF,0x5A,0xA5,0xF0,0x0F,0x03,0x00,0x04,0x03,0x06,0x02,0x10,0x15,0x20,0x01,0x21,0x00
};
char buff[0xffffff];
char id[32];
char outputdir[256];
char inputdir[256];
char chack_str(const char *a,char *b,int len)
{
	int c;
	for(c=0;c<len;c++)
	{
		if(a[c]!=b[c])
		{
			return 1;
		}
	}
	return 0;
}

void tryunlock(char *dir)
{
	int a;
	FILE *bios=NULL;
	FILE *output_bios=NULL;
	bios=fopen(dir,"rb");
	if(bios==NULL)
	{
		printf("ERROR. can't Open file ..\n");
	}else
	{
		fread(id,32,1,bios);
		if(chack_str(bios_head,id,32))
		{
			printf("ERROR. Not Bios File\n");
		}else
		{
			//printf("try it..\n");
			if(fseek(bios,0xf00,SEEK_SET))
			{
				printf("ERROR. -0\n");
			}else
			{
				fread(id,32,1,bios);
				for(a=0;a<32;a++)
				{
					buff[0xf00+a]=id[a];
				}
				for(a=0;a<256;a++)
				{
					outputdir[a]=0;
				}
				printf("id: %s \n",id);
				sprintf(outputdir,"%s%s",dir,"_unlock.bin");
				printf("Output flie %s",outputdir);
				output_bios=fopen(outputdir,"wb");
				if(output_bios==NULL)
				{
					printf("ERROR. can't Output file ..\n");
				}else
				{
					fwrite(buff,0xffffff,1,output_bios);
					fclose(output_bios);
				}
				
				printf("\n\n");
			}
		}
		
	}
	
	fclose(bios);
}
	
int main(int argc, char *argv[])
{
	printf("ThinkPad Bios unlock..v0.1\n\n\n");
	if(argc==1)
	{
		printf("联想ThinkPad Bios超级密码解锁程序（目前仅支持X250）\n\n使用方法：\n");
		printf("1 使用编程器将原机Bios读出\n");
		printf("2 将读出的内容以二进制格式bin保存\n");
		printf("3 将bin文件拖进本程序，程序会自动生成解锁Bios，（支持批量拖入）\n");  
		printf("4 将生成的解锁bios刷入，然后开机，按F1进入Bios，将超级密码改成空密码，按F10保存\n");
		printf("5 完成！ 注意，部分Bios芯片写入之前要先清空，否则无法校验通过。\n\n\n");
		printf("编写日期 2021-12-12 00:46 联系 吴文峰 邮箱 kevin@lmve.net \n\n");
		printf("项目地址 https://git.lmve.net/kevin/thinkpad_unlock_bios \n博客地址 https://wnfed.com \n\n！！！请勿商用！！！\n\n");	
	}
	
	int a;
	FILE *unlock_bios = NULL;	
	unlock_bios=fopen("./X250.bbb","rb");
	if(unlock_bios==NULL)
	{
		printf("ERROR_NOTFUND_THE_BIOS_FILE\n");
		return -1;
	}
	fread(buff,0xffffff,1,unlock_bios);
	
	while(argc>1)
	{
	
		argc--;
		printf("Input flie  %s\n",argv[argc]);
		tryunlock(argv[argc]);
	}
	printf("\n");

	printf("\n");
	
	while(1)
	{
		printf("请输入Bios文件路径，可通过拖拽文件到窗口快速输入，输入exit退出。\ndir:");
		for(a=0;a<256;a++)
		{
			inputdir[a]=0;
		}
		scanf("%s",inputdir);
		if(chack_str("exit",inputdir,4)==0)
		{
			return 0;
		}
		tryunlock(inputdir);
	}
	
	//printf("have %d file\n",argc-1);
	//printf(argv[1]);
	//system("pause");
	return 0;
}