// stdafx.cpp : 只包括标准包含文件的源文件
// FZ_YXJ_ReadDiskSN.pch 将作为预编译头
// stdafx.obj 将包含预编译类型信息
#include "stdafx.h"
#include <iostream>
#include <string>
#include <windows.h>
using namespace std;
LPCWSTR stringToLPCWSTR(std::string orig)
{
size_t origsize = orig.length() + 1;
const size_t newsize = 100;
size_t convertedChars = 0;
wchar_t *wcstring = (wchar_t *)malloc(sizeof(wchar_t)*(orig.length() - 1));
mbstowcs_s(&convertedChars, wcstring, origsize, orig.c_str(), _TRUNCATE);
return wcstring;
}
// TODO: 在 STDAFX.H 中引用任何所需的附加头文件,
//而不是在此文件中引用
int execmd(char* cmd, char* result)
{
char buffer[128]; //定义缓冲区
FILE* pipe = _popen(cmd, "r"); //打开管道,并执行命令
if (!pipe)
return 0; //返回0表示运行失败
while (!feof(pipe)) {
if (fgets(buffer, 128, pipe)) { //将管道输出到result中
strcat(result, buffer);
}
}
_pclose(pipe); //关闭管道
return 1; //返回1表示运行成功
}
int main() {
char result[1024 * 4] = ""; //定义存放结果的字符串数组
if (1 == execmd("wmic diskdrive get serialnumber", result)) {
printf(result);
}
string Path = "C:\\test.dat";
string Disk_SN;
for (int i = 0; i < size(result); i++)
{
if (result[i] != '\r' && result[i] != '\n' && result[i] != ' ')
{
Disk_SN.push_back(result[i]);
}
}
::WritePrivateProfileString(L"配置参数", L"硬盘SN", stringToLPCWSTR(Disk_SN), stringToLPCWSTR(Path));
}