#include "stdafx.h"
#include "windows.h"
#pragma comment(lib,"crypt32.lib")
LPBYTE MyBase64Decode(LPSTR lpBase64Str , LPDWORD lpdwLen)
{
	DWORD dwLen ;
	DWORD dwNeed ;
	LPBYTE lpBuffer = NULL;

	dwLen = strlen(lpBase64Str);
	dwNeed = 0;
	CryptStringToBinaryA(lpBase64Str , 0 , CRYPT_STRING_BASE64 , NULL , &dwNeed , NULL , NULL);
	if(dwNeed)
	{
		lpBuffer = (LPBYTE)malloc(dwNeed);
		CryptStringToBinaryA(lpBase64Str , 0 , CRYPT_STRING_BASE64 , lpBuffer , &dwNeed , NULL , NULL);
		*lpdwLen = dwNeed ;
	}

	return lpBuffer ;
}
int main()
{
	// 这里写要变化的shellcode
	//char * shellcode = "\x89\x75\xFC\xEB\x0E\x33\xC0\x40\xC3\x8B\x65\xE8\xC7\x45\xFC\xFE\xFF\xFF\xFF\xE8\xD5\xCE\xF8\xFF\xC3\x90\x90\x90\x90\x90";
	

	// 这里传入的是shellcode BASE64加密之后的
	LPBYTE shellcode;
	char * base64Str = new char[ 10000 ];
	memset(base64Str , 0 , 10000);
	printf("请输入shellcode的base64编码\n");
	scanf("%s", base64Str);
	DWORD dwShellCodeSize;

	
	shellcode = MyBase64Decode(base64Str , &dwShellCodeSize);
	printf("请输入指向shellcode char*变量名 DWORD*变量名 ；函数名\n");
	char shellcodechName[ 256 ] = { 0 };
	char shellcodedwName[ 256 ] = { 0 };
	char funcName[ 256 ] = { 0 };
	scanf("%s%s%s", shellcodechName, shellcodedwName, funcName);
	// 判断可以转化为4字节的有多少个
	DWORD dwDWORDNum = dwShellCodeSize / 4;
	// 计算最后以char方式传入的有多少个
 
 
	DWORD * shellcodeTmp = (DWORD*)shellcode;
 
	//这是我们想要生成的代码文本
	//char shellcode[ 28 ];
	//DWORD * shellcodeTmp = (DWORD *)shellcode;
	//shellcodeTmp[ 0 ] = 0x12345678;
	//shellcodeTmp[ 1 ] = 0x55555555;
	////.....
	//shellcode[ 26 ] = 0x1;
	//shellcode[ 27 ] = 0x1;
	char *codeText = new char[ 55555 ];
	memset(codeText , 0, 55555);
	sprintf(codeText, "char %s[ 0x%X ];\nDWORD * %s = (DWORD *)%s;\n", shellcodechName,dwShellCodeSize,shellcodedwName,shellcodechName);
	for (int i = 0; i<dwDWORDNum; i++)
	{
		char tmp[ 256 ] = {0};
		sprintf(tmp,"%s[ 0x%X ] = 0x%X;\n",shellcodedwName ,i, shellcodeTmp[i]);
		strcat(codeText , tmp);
	}
	for (int i = dwDWORDNum*4; i<dwShellCodeSize; i++)
	{
		char tmp[ 256 ] = { 0 };
		sprintf(tmp , "%s[ 0x%X ] = 0x%X;\n" , shellcodechName,i , (unsigned char)shellcode[ i ]);
		strcat(codeText , tmp);
	}

	// 这里顺带把裸函数的形式输出
	strcat(codeText , "\n\n\n\n\n\n\n\n");
	char begin[ 256 ] = {0};
	sprintf(begin , "void __declspec(naked) %s()\n{\n" , funcName);
	strcat(codeText , begin);
	
	for (int i = 0; i<dwShellCodeSize; i++)
	{
		char tmp[ 256 ] = { 0 };
		sprintf(tmp , "\t_asm _emit(0x%X)\n" , (unsigned char)shellcode[ i ]);
		strcat(codeText , tmp);
	}
	strcat(codeText , "}");

	FILE * fp = fopen("code.txt","w");
	fwrite(codeText, 55555 ,1,fp);
	fclose(fp);
	

	free(shellcode);
    return 0;
}