view/CopalPro/KExecProc.cpp

//---------------------------------------------------------------------------
// 子プロセスを起動し、標準入出力を取るクラス
//---------------------------------------------------------------------------
#include <vcl.h>
#pragma hdrstop

#include "KExecProc.h"
//---------------------------------------------------------------------------
#pragma package(smart_init)


const int BUFFER_SIZE = 4097;//パイプのバッファサイズ
//---------------------------------------------------------------------------
/**
 * コンストラクタ
 */
KExecProc::KExecProc(AnsiString cmd) {
  kThread = new KThread(true);
  CommandLine = cmd;
  msStdOut = new TMemoryStream;//標準出力用
  msStdErr = new TMemoryStream;//標準エラー用
  lpEnvironment = NULL;
}
//---------------------------------------------------------------------------
/**
 * デストラクタ
 */
KExecProc::~KExecProc(void) {
  if(lpEnvironment != NULL)
    delete lpEnvironment;
  delete kThread;
  delete msStdOut;
  delete msStdErr;
}
//---------------------------------------------------------------------------
/**
 * 環境変数データを受け取る
 */
void
KExecProc::SetEnvironment(char *b) {
  if(lpEnvironment != NULL) {
    ShowMessage("KExecProc::SetEnvironment\nAsertionFailed.");
  }
  lpEnvironment = b;
}
//---------------------------------------------------------------------------
/**
 * 子プロセスを起動する
 */
bool
KExecProc::Execute(void) {
  //実行監視スレッドを立ち上げる

  //SECURITY_ATTRIBUTEの設定

  SECURITY_ATTRIBUTES saStdOut,saStdErr;
  saStdOut.nLength = sizeof( SECURITY_ATTRIBUTES );
  saStdOut.bInheritHandle = true;
  saStdOut.lpSecurityDescriptor = NULL;

  saStdErr.nLength = sizeof( SECURITY_ATTRIBUTES );
  saStdErr.bInheritHandle = true;
  saStdErr.lpSecurityDescriptor = NULL;

  //このプロセスのSTDOUT,STDERRハンドルの保存
  HANDLE hSaveStdout = GetStdHandle(STD_OUTPUT_HANDLE);
  HANDLE hSaveStderr = GetStdHandle(STD_ERROR_HANDLE);

  //子プロセスへのパイプをつくる。
  HANDLE hStdOutWriteEnd,hStdOutReadEnd;
  HANDLE hStdErrWriteEnd,hStdErrReadEnd;

  if (!CreatePipe(&hStdOutReadEnd, &hStdOutWriteEnd, &saStdOut, 0)) {
    SetErrorMessage("Cannot create a Pipe(StdOut)");
    return false;
  }
  if (!CreatePipe(&hStdErrReadEnd, &hStdErrWriteEnd, &saStdErr, 0)) {
    SetErrorMessage("Cannot create a Pipe(StdErr)");
    return false;
  }

  PROCESS_INFORMATION pi;
  STARTUPINFO si;
  GetStartupInfo(&si);
  si.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
  si.wShowWindow = SW_HIDE;
  si.hStdOutput = hStdOutWriteEnd;
  si.hStdError = hStdErrWriteEnd;

  //プロセスの作成
  if (!CreateProcess(NULL,CommandLine.c_str(), NULL, NULL, TRUE, 0, lpEnvironment, NULL, &si, &pi)) {
    SetErrorMessage("Cannot create a proccess.");
    return false;
  }
  // このプロセスの標準出力を元に戻す。
  if (!SetStdHandle(STD_OUTPUT_HANDLE, hSaveStdout)) {
    SetErrorMessage("Cannot restore STDOUT");
    return false;
  }
  if (!SetStdHandle(STD_ERROR_HANDLE, hSaveStderr)) {
    SetErrorMessage("Cannot restore STDERR");
    return false;
  }
  //パイプから読み込む前に書き込み端を閉じる。
  if ( !CloseHandle( hStdOutWriteEnd) ) {
    SetErrorMessage("Cannot close stdout write end.");
    return false;
  }
  if ( !CloseHandle( hStdErrWriteEnd) ) {
    SetErrorMessage("Cannot close stderr write end.");
    return false;
  }

  //スレッドにハンドルを渡す

  kThread->SetHandle(pi.hProcess);
  kThread->Resume();

  //パイプ読み込みの準備

  DWORD exitcode = 1;//終了フラグ
  char bufStdOut[BUFFER_SIZE],bufStdErr[BUFFER_SIZE];//読み込みbuffer
  DWORD readbufStdOut = 0,readbufStdErr = 0;//読み込んだバイト数

  SetLastError(-1);

  //ストリームの準備
  msStdOut->Clear();
  msStdErr->Clear();

  //プロセスを監視する無限メインループ

  while(1) {

    //exitcodeを調べる
    if(GetExitCodeProcess(pi.hProcess,&exitcode)==false) {
      SetErrorMessage("Error:Fail to ExitCodeProcess\n");
      break;
    }

    if (exitcode !=STILL_ACTIVE)
      break;

    if (GetAsyncKeyState(VK_ESCAPE)<0) {
      ShowMessage("処理は中断されました");
      break;
    }

    do {
      if(!ReadFile(hStdOutReadEnd,bufStdOut,sizeof(bufStdOut),&readbufStdOut,NULL))
        break;
      if(readbufStdOut) {
        msStdOut->Write(bufStdOut,readbufStdOut);
      }
    } while(readbufStdOut!=0);


    if(!ReadFile(hStdErrReadEnd,bufStdErr,sizeof(bufStdErr),&readbufStdErr,NULL))
      break;
    if(readbufStdErr) {
      msStdErr->Write(bufStdErr,readbufStdErr);
    }
  }

  //プロセス終了後の処理

  CloseHandle(hStdOutReadEnd);
  CloseHandle(hStdErrReadEnd);

  //スレッドを殺す
  kThread->Terminate();

  if(GetExitCodeProcess(pi.hProcess,&exitcode)==false) {
    SetErrorMessage("Error:Fail to ExitCodeProcess\n");
    return false;
  }
  ExitCode = exitcode;

  if(ExitCode!=0) {
    return false;
  } else {
    return true;
  }
}
//---------------------------------------------------------------------------