벽에게 말걸기

// 버퍼 사이즈 정의
#define BUFFER_SIZE 4096

void CTestprocessDlg::OnOk()
{
  STARTUPINFO si;
  PROCESS_INFORMATION pi;
  HANDLE hPipeRead, hPipeWrite;
  HANDLE hOutputFile;
  SECURITY_ATTRIBUTES sa;

  TCHAR szCmdLine[MAX_PATH];
  TCHAR szBuffer[BUFFER_SIZE + 1];
  DWORD dwBytesRead = 0, dwBytesWrite;
  DWORD dwExitCode = 0, dwAvail = 0, dwResult = 0;

  lstrcpy(szCmdLine, "c:\\windows\\system32\\cmd.exe /?"); //실행할 프로그램 경로...
  sa.nLength = sizeof(SECURITY_ATTRIBUTES);
  sa.bInheritHandle = TRUE;
  sa.lpSecurityDescriptor = NULL;

  // 파이프를 생성
  if(!CreatePipe( &hPipeRead, &hPipeWrite, &sa, 0))
  {
      AfxMessageBox("Error");
      return;
  }

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

  // 시작 플래그로 표준 입,출력 핸들 지정 및 윈도우 표시 방식을 설정한다.
  si.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;

  // 프로세스의 출력을 파이프로 설정한다.
  si.hStdInput  = GetStdHandle(STD_INPUT_HANDLE);
  si.hStdOutput = hPipeWrite;
  si.hStdError = GetStdHandle(STD_ERROR_HANDLE);

  // 윈도우는 숨겨서 시작하게 한다.
  si.wShowWindow = SW_HIDE;

//파일 출력 안할란다.
/*
  // 출력 파일을 생성한다.
  hOutputFile = CreateFile("C:\\report.txt",
                      GENERIC_WRITE,
                      0,
                      NULL,
                      CREATE_ALWAYS,
                      FILE_ATTRIBUTE_NORMAL,
                      NULL);
  if(hOutputFile == INVALID_HANDLE_VALUE)
  {
      AfxMessageBox("Failed to create file.");
      return;
  }
*/

  // 프로세스를 실행한다.
  if(!CreateProcess(NULL, szCmdLine, NULL, NULL, TRUE, 0, NULL, NULL, &si, &pi))
  {
      AfxMessageBox("Failed to execute CreateProcess().");
      return;
  }

  // 프로세스가 살아있는 동안만 파이프의 내용을 출력파일로 쓴다.
  while(dwExitCode == 0)
  {
      // 프로세스가 종료하기를 기다린다.
      dwResult = WaitForSingleObject(pi.hProcess, 500);
      // 파이프에 내요이 있는지 확인한다.
      if(PeekNamedPipe(hPipeRead, NULL, 0, NULL, &dwAvail, NULL))
      {
          if(dwAvail > 0)
          {
              // 파이프의 내용을 버퍼 만큼 읽는다.
              ReadFile(hPipeRead, &szBuffer, BUFFER_SIZE, &dwBytesRead, NULL);
              // 출력 파일에 파이프에서 읽은 데이터를 쓴다.
  //         WriteFile(hOutputFile, szBuffer, dwBytesRead, &dwBytesWrite, NULL);
  // 파일출력 안하고 화면상에 메시지 박스로 보기
  unsigned char temp[BUFFER_SIZE];
  memcpy(temp,szBuffer,BUFFER_SIZE);
  AfxMessageBox((LPCTSTR)temp);
          }
      }
      // Time Out이 걸린 것이 아니면 파이프 읽기를 종료한다.
      if(dwResult != WAIT_TIMEOUT)
          dwExitCode = 1;
  }

  // 출력파일 안 만든다..
  // 출력파일을 닫는다.
/*
  if(hOutputFile != INVALID_HANDLE_VALUE)
      CloseHandle(hOutputFile);
*/

  // 프로세스 종료코드를 얻는다.
  GetExitCodeProcess(pi.hProcess, &dwExitCode);
  // 프로세스 핸들을 닫는다.
  CloseHandle( pi.hProcess );
  CloseHandle( pi.hThread );

}