通过 CreateProcess() 运行批处理文件的多线程服务

Question

请参阅最后更新

我正在尝试运行以下批处理文件（名为 boot_time.bat，与 .exe 位于同一目录中）：

@echo off
cd %1
For /F %%I in ('Cscript boot_time.vbs //Nologo') Do Set var=%%I
set DATETIME=%var:~0,4%/%var:~4,2%/%var:~6,2% %var:~8,2%:%var:~10,2%:%var:~12,2%.%var:~15,3%
echo %DATETIME%

该批处理文件采用一个参数，它是当前工作目录。其目的是通过 boot_time.vbs 脚本检索系统启动时间并将其格式化为通用日期格式。为了完整起见，以下是 vbs 文件的内容：

set objWMI = GetObject("winmgmts:\\.\root\cimv2")
set colOS = objWMI.InstancesOf("Win32_OperatingSystem")
for each objOS in colOS
    Wscript.Echo objOS.LastBootUpTime
NEXT

虽然我很欣赏有关检索系统启动时间的替代（阅读：更简单）方法的评论，但请放心，我已经探索了所有途径（我遇到过），并且没有一个与对系统的要求。

现在，问题的关键来了。当我尝试通过 C++ 运行 .bat 文件时，CreateProcess() 返回 1，但批处理文件未运行（我已通过将 boot_time.bat 的内容替换为简单的“start calc”来验证这一点，但它仍然拒绝运行）。有问题的代码：

//run batch file
stringstream commandStream;
commandStream << "/C "                           //close window on termination
         << "\"" << batchFile.c_str() << "\" "   //batch file path inside ""
         << "\"" << processPath.c_str() << "\" " //working directory as argument inside ""
         << ">"                                  //redirect
         << "\"" << outFile.c_str() << "\"";     //output file inside ""

STARTUPINFO si;
PROCESS_INFORMATION pi;

ZeroMemory( &si, sizeof(si) );
si.cb = sizeof(si);
ZeroMemory( &pi, sizeof(pi) );

BOOL ret = CreateProcess("cmd.exe", const_cast<char*>(commandStream.str().c_str()), NULL, NULL, TRUE, NULL, NULL, processPath.c_str(), &si, &pi);

WaitForSingleObject( pi.hProcess, INFINITE );

CloseHandle( pi.hProcess );
CloseHandle( pi.hThread );

路径和命令行参数都很好（我可以调试打印它们并粘贴到 cmd.exe 中，它们按预期工作）。单步执行代码，CreateProcess() 返回 1，一切正常运行，但批处理文件永远不会运行。

在我在互联网上搜索的许多小时中，我在从服务运行批处理文件时遇到了问题的低语，但我：

a）不再找到所说的低语，
b) 会认为 CreateProcess("cmd.exe",...) 不是运行批处理文件，而是运行 exe，并将其余部分留给命令提示符。

那么，知道发生了什么事吗？

哦，我正在运行 VC++6

UPDATE 1：
如果我勾选“允许服务与桌面交互”，并将 CreateProcess() 更改为不隐藏控制台窗口（通过 CREATE_NEW_CONSOLE），我会立即弹出命令提示符。因此，CreateProcess() 正在创建一个 cmd.exe 进程，但 cmd.exe 拒绝运行批处理文件。

更新2：
在加布的建议之后，我重新考虑了我的流程并对其进行了一些简化，以试图找出罪魁祸首。我已经删除了批处理文件，并将其合并到 VBS 文件中，现在是：

set objWMI = GetObject("winmgmts:\\.\root\cimv2")
set colOS = objWMI.InstancesOf("Win32_OperatingSystem")
Dim bootTime
for each objOS in colOS
   bootTime = objOS.LastBootUpTime

bootTime = mid(bootTime,1,4) & "/" & Mid(bootTime,5,2) & "/" & Mid(bootTime,7,2) & " " & Mid(bootTime,9,2) & ":" & Mid(bootTime,11,2) & ":" & Mid(bootTime,13,2) & "." & Mid(bootTime,16,3)
Wscript.Echo bootTime
NEXT

C++ 代码基本上没有变化，只是修改了命令行以调用 WScript.exe 而不是 cmd.exe。我正在获取所有文件的完全限定路径，这会导致以下字符串：

"C:\WINDOWS\System32\Wscript.exe" //Nologo "c:\<repository_dir>\boot_time.vbs" >"C:\WINDOWS\TEMP\rts5FD.tmp"

（repository_dir 有意省略名称）。另外，我只使用 Wscript 进行测试，因为它会弹出一个消息框。正确的用法是使用 Cscript.exe，并将输出重定向到临时文件。

CreateProcess 不会导致弹出消息框，直接从命令行运行会弹出消息框。

Answer 1

仔细阅读 MSDN 对 CreateProcess 的描述中的这一行：

由于 argv[0] 是模块名称，C 程序员通常会重复模块名称作为命令行中的第一个标记。

这意味着您的代码应该是：

commandStream << "cmd.exe /C "

另请参阅 http:// /blogs.msdn.com/b/oldnewthing/archive/2006/05/15/597984.aspx。

然而，我必须补充一点，我会射击我发现的使用 C++ 运行批处理文件来执行 VBScript 的人，以处理其输出。既然批处理文件只执行字符串操作，为什么不能在 VBS 或 C++ 中执行字符串操作呢？

或者，您可以避免使用批处理文件，而直接使用以下代码调用 cscript：

set objWMI = GetObject("winmgmts:\\.\root\cimv2") 
set colOS = objWMI.InstancesOf("Win32_OperatingSystem") 
for each objOS in colOS 
    t = objOS.LastBootUpTime 
next
wscript.echo left(t, 4) & "/" & mid(t, 5, 2) & "/" & mid(t, 7, 2) & " " & _
             mid(t, 9, 2) & ":" & mid(t, 11, 2) & ":" & mid(t, 13, 2) & "." & _
             mid(t, 16, 3)

另一种选择是查询 System\System Up Time 性能计数器，并从当前时间。

Answer 2

我设法找出 C++ 代码来完成这一切。

与大多数 C++ COM 代码一样，它变得相当冗长。但你看吧：

#define _WIN32_DCOM
#include <iostream>
#include <exception>
#include <string>

#include <windows.h>
#include <wbemidl.h>
# pragma comment(lib, "wbemuuid.lib")

void com_init() {
  HRESULT hr;
  hr = CoInitializeEx(0, COINIT_MULTITHREADED); 
  if (FAILED(hr)) {
    throw std::runtime_error("COM Error");
  }
  hr =  CoInitializeSecurity(NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_DEFAULT, RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE, NULL);
  if (FAILED(hr)) {
    CoUninitialize();
    throw std::runtime_error("COM Error");
  }
}

std::wstring wmi_last_boot_time() {
  IWbemLocator *pLoc = NULL;
  IWbemServices *pSvc = NULL;
  IEnumWbemClassObject *pEnum = NULL;
  IWbemClassObject *pEach = NULL;

  std::wstring lastBootUpTime;

  try {
    HRESULT hr;
    // Get WMI object
    hr = CoCreateInstance(CLSID_WbemLocator, 0, CLSCTX_INPROC_SERVER, IID_IWbemLocator, (LPVOID *) &pLoc);
    if (FAILED(hr)) {
      throw std::runtime_error("WMI: Unable to create WbemLocator");
    }
    hr = pLoc->ConnectServer(L"root\\cimv2", NULL, NULL, 0, NULL, 0, 0, &pSvc);
    if (FAILED(hr))  {
      throw std::runtime_error("WMI: Unable to connect");
    }

    // Exec query
    hr = pSvc->ExecQuery(L"WQL", L"SELECT LastBootUpTime FROM Win32_OperatingSystem", 0, 0, &pEnum);
    if (FAILED(hr))  {
      throw std::runtime_error("WMI: Query failed");
    }

    // Fetch result
    VARIANT value;
    ULONG uCount;
    hr = pEnum->Next(WBEM_INFINITE, 1, &pEach, &uCount);
    if (FAILED(hr) || uCount == 0)  {
      throw std::runtime_error("WMI: Can't fetch result");
    }

    hr = pEach->Get(L"LastBootUpTime", 0, &value, 0, 0);
    if (FAILED(hr))  {
      throw std::runtime_error("WMI: Can't get LastBootUpTime");
    }

    if (value.vt != VT_BSTR) {
      throw std::runtime_error("Expected string");
    }

    lastBootUpTime = value.bstrVal;

  } catch (std::runtime_error &) {
    if (pLoc)  pLoc->Release();
    if (pSvc)  pSvc->Release();
    if (pEnum) pEnum->Release();
    if (pEach) pEach->Release();
    throw;
  }

  pLoc->Release();
  pSvc->Release();
  pEnum->Release();
  pEach->Release();

  return lastBootUpTime;
}


int main() {
  try {
    // Initialize COM. Do this only once
    com_init();

    // Get last boot time
    std::wstring str = wmi_last_boot_time();

    // Reformat time
    std::wcout << str.substr(0, 4) << "/" << str.substr(4, 2) << "/" << str.substr(6, 2) << " "
               << str.substr(8, 2) << ":" << str.substr(10, 2) << ":" << str.substr(12, 2) << "."
               << str.substr(15, 3) << std::endl;


    // Uninitialize COM
    CoUninitialize();

  } catch (std::exception &e) {
    std::cerr << e.what() << std::endl;
    return 1;
  }
  return 0;
}

我认为你可以在 vc6 上运行它。 wbemidl.h 包含 COM 接口定义，您可以使用 midl 生成这些定义，甚至可以使用最新 SDK 中的标头。

不幸的是我什至没有安装vc6。

Answer 3

感谢您的意见，不过我已经设法解决了这个问题。问题似乎源于我调用 CreateProcess() 的服务作为 LocalSystem 运行的事实。 CreateProcess() 在同一用户下创建新进程，我需要它作为登录用户运行。以下是获取当前用户并在 CreateProcessAsUser() 中使用其令牌的代码：

//retrieve the user token via an open process
HANDLE hToken = NULL;                
HANDLE hProcess = NULL;
DWORD dwProcessId = NULL;

//user 'explorer.exe' as the process to search for
PROCESSENTRY32 pe32;
ZeroMemory(&pe32,sizeof(pe32));

HANDLE hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,NULL);

pe32.dwSize = sizeof(PROCESSENTRY32); 

if(Process32First(hSnapshot,&pe32))
{
    do
    {
        if(!strcmp(pe32.szExeFile,"explorer.exe"))
        {
            dwProcessId = pe32.th32ProcessID;
            break;
        }

    }while(Process32Next(hSnapshot,&pe32));
}

if( dwProcessId ) 
{
    hProcess = OpenProcess(PROCESS_ALL_ACCESS,TRUE, dwProcessId );
    if( hProcess) 
    {
        OpenProcessToken(hProcess,  TOKEN_EXECUTE | TOKEN_READ | TOKEN_QUERY | 
                                    TOKEN_ASSIGN_PRIMARY | TOKEN_QUERY_SOURCE |
                                    TOKEN_WRITE | TOKEN_DUPLICATE, 
                                    &hToken);
        CloseHandle( hProcess );
    }
    else
    {
        error = _T("Could not open process 'explorer.exe'");
        return false;
    }
}
else
{
    error = _T("Could not retrieve process id for 'explorer.exe'");
    return false;
}

if (hToken != NULL)
{
    STARTUPINFO si;
    PROCESS_INFORMATION pi;

    ZeroMemory( &si, sizeof(si) );
    si.cb = sizeof(si);
    ZeroMemory( &pi, sizeof(pi) );
    if (CreateProcessAsUser(hToken, "cmd.exe", const_cast<char*>(command.c_str()), NULL, NULL, TRUE, 
                                            CREATE_NO_WINDOW, NULL, NULL, &si, &pi) == 0)
    {
        tstringstream err;
        err << "CreateProcessAsUser() failed with error: " << GetLastError();
        error = err.str();
        CloseHandle(hToken);
        return false;
    }

    WaitForSingleObject( pi.hProcess, INFINITE );

    CloseHandle( pi.hProcess );
    CloseHandle( pi.hThread );
    CloseHandle( hToken );
}
else
{
    error = _T("Token retrieved from 'explorer.exe' is NULL");
    return false;
}

return true;

在我的例子中，command.c_str() 指向以下字符串：

/C ""C:\WINDOWS\System32\Cscript.exe" //Nologo "c:\<repository_dir>\boot_time.vbs" >"C:\WINDOWS\TEMP\rts452.tmp""

感谢您的所有帮助！