#include "..\include_eapi_header.h"
inline char* GetDiskByCmd()
{
	//diskdrive
	const long MAX_COMMAND_SIZE = 10000; // 命令行输出缓冲大小 
	CHAR szFetCmd[] = "wmic diskdrive get serialnumber"; // 获取DiskDrive命令行 
	const std::string strEnSearch = "SerialNumber"; // DiskDrive序列号的前导信息 
	HANDLE hReadPipe = NULL; //读取管道 
	HANDLE hWritePipe = NULL; //写入管道 
	PROCESS_INFORMATION pi; //进程信息 
	STARTUPINFOA si; //控制命令行窗口信息 
	SECURITY_ATTRIBUTES sa; //安全属性 

	char szBuffer[MAX_COMMAND_SIZE + 1] = { 0 }; // 放置命令行结果的输出缓冲区 
	std::string strBuffer;
	unsigned long count = 0;
	long ipos = 0;

	memset(&pi, 0, sizeof(pi));
	memset(&si, 0, sizeof(si));
	memset(&sa, 0, sizeof(sa));

	pi.hProcess = NULL;
	pi.hThread = NULL;
	si.cb = sizeof(STARTUPINFO);
	sa.nLength = sizeof(SECURITY_ATTRIBUTES);
	sa.lpSecurityDescriptor = NULL;
	sa.bInheritHandle = TRUE;

	//1.0 创建管道 

	if (!CreatePipe(&hReadPipe, &hWritePipe, &sa, 0))
	{
		CloseHandle(hWritePipe);
		CloseHandle(hReadPipe);
		CloseHandle(pi.hProcess);
		CloseHandle(pi.hThread);

		return nullptr;
	}

	//2.0 设置命令行窗口的信息为指定的读写管道 
	GetStartupInfoA(&si);
	si.hStdError = hWritePipe;
	si.hStdOutput = hWritePipe;
	si.wShowWindow = SW_HIDE; //隐藏命令行窗口 
	si.dwFlags = STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES;

	//3.0 创建获取命令行的进程 
	if (!CreateProcessA(NULL, szFetCmd, NULL, NULL, TRUE, 0, NULL, NULL, &si, &pi))
	{
		CloseHandle(hWritePipe);
		CloseHandle(hReadPipe);
		CloseHandle(pi.hProcess);
		CloseHandle(pi.hThread);

		return nullptr;
	}



	//4.0 读取返回的数据 
	WaitForSingleObject(pi.hProcess, 500/*INFINITE*/);

	if (!ReadFile(hReadPipe, szBuffer, MAX_COMMAND_SIZE, &count, 0))
	{
		CloseHandle(hWritePipe);
		CloseHandle(hReadPipe);
		CloseHandle(pi.hProcess);
		CloseHandle(pi.hThread);

		return nullptr;
	}

	//5.0 查找CPU序列号 
	strBuffer = szBuffer;
	ipos = strBuffer.find(strEnSearch);

	if (ipos < 0) // 没有找到 
	{
		CloseHandle(hWritePipe);
		CloseHandle(hReadPipe);
		CloseHandle(pi.hProcess);
		CloseHandle(pi.hThread);

		return nullptr;
	}
	else
	{
		strBuffer = strBuffer.substr(ipos + strEnSearch.length());
	}

	memset(szBuffer, 0x00, sizeof(szBuffer));
	strcpy_s(szBuffer, strBuffer.c_str());

	//modify here
	//去掉中间的空格 \r \n 
	char temp[512];
	memset(temp, 0, sizeof(temp));

	int index = 0;
	for (size_t i = 0; i < strBuffer.size(); i++)
	{
		if (strBuffer[i] != ' ' && strBuffer[i] != '\n' && strBuffer[i] != '\r')
		{
			temp[index] = strBuffer[i];
			index++;
		}
	}

	//关闭所有的句柄 
	CloseHandle(hWritePipe);
	CloseHandle(hReadPipe);
	CloseHandle(pi.hProcess);
	CloseHandle(pi.hThread);

	return CloneTextData(temp);
}


// 调用格式: SDT_TEXT 取硬盘编号, 命令说明: "获取硬盘的编号(出场带的)"
// 无参数
EAPI_EXTERN_C void eapi_GetDiskNumber_93_eapi(PMDATA_INF pRetData, INT nArgCount, PMDATA_INF pArgInf)
{
	pRetData->m_pText = GetDiskByCmd();
}