Форум программистов [Powered by Invision Power Board]

Код

#define bzero(a) memset(a,0,sizeof(a))

Pipes::Pipes()
{
    Started = false;
}

void Pipes::Start(String Param)
{
  try
  {
    if (this->IsWinNT())
    {
      InitializeSecurityDescriptor(&sd,SECURITY_DESCRIPTOR_REVISION);
      SetSecurityDescriptorDacl(&sd, true, NULL, false);
      sa.lpSecurityDescriptor = &sd;
    } else sa.lpSecurityDescriptor = NULL;

    sa.nLength = sizeof(SECURITY_ATTRIBUTES);
    sa.bInheritHandle = true;

    // Ñîçäàåì òðóáó STDIN
    if (!CreatePipe(&newstdin,&write_stdin,&sa,0))
    throw Exception("Íå óäàëîñü ñîçäàòü òðóáó STDIN");

    // Ñîçäàåì òðóáó STDOUT
    if (!CreatePipe(&read_stdout,&newstdout,&sa,0))
    {
      CloseHandle(newstdin);
      CloseHandle(write_stdin);
      throw Exception("Íå óäàëîñü ñîçäàòü òðóáó STDOUT");
    }

    GetStartupInfo(&si);
    si.dwFlags = STARTF_USESTDHANDLES|STARTF_USESHOWWINDOW;
    si.wShowWindow = SW_HIDE;
    si.hStdOutput = newstdout;
    si.hStdError = newstdout;
    si.hStdInput = newstdin;

    char *app_spawn = "c:\\windows\\system32\\cmd.exe";

    //ñîçäàåì äî÷åðíèé ïðîöåññ
    if (!CreateProcess(NULL,app_spawn,NULL,NULL,TRUE,CREATE_NEW_CONSOLE,NULL,NULL,&si,&pi))
    {
      CloseHandle(newstdin);
      CloseHandle(newstdout);
      CloseHandle(read_stdout);
      CloseHandle(write_stdin);
      throw Exception("Íå óäàëîñü ñîçäàòü äî÷åðíèé ïðîöåññ " + (String) app_spawn);
    }

    exit = 0;
    bzero(buf);
    MainForm->RefreshTimer->Enabled = true;
    Started = true;
  }
  catch(Exception &E)
  {
    Started = false;
    MainForm->RefreshTimer->Enabled = false;
    ShowMessage(E.Message);
  }
}

bool Pipes::IsWinNT()
{
  OSVERSIONINFO osv;
  osv.dwOSVersionInfoSize = sizeof(osv);
  GetVersionEx(&osv);
  return (osv.dwPlatformId == VER_PLATFORM_WIN32_NT);
}


bool Pipes::Write(String Data)
{
  try
  {
    if(Data.IsEmpty()) return 1;
    bzero(buf);
    StrCopy(buf, (Data+"\r\n").c_str());
    WriteFile(write_stdin,buf,sizeof(buf),&bread,NULL);
    return 1;
  }
  catch(...)
  {
    return 0;
  }
}

String Pipes::Read()
{
  try
  {
    if(!Started) throw Exception("Server is NOT STARTED");
    String Line;
    GetExitCodeProcess(pi.hProcess, &exit);
    if (exit != STILL_ACTIVE) throw Exception("Server is DEAD");
    PeekNamedPipe(read_stdout, buf, 1023, &bread, &avail, NULL);

    //Ïðîâåðÿåì, åñòü ëè äàííûå äëÿ ÷òåíèÿ â stdout
    if (bread != 0)
    {
      bzero(buf);
      if (avail > 1023)
      {
        while (bread >= 1023)
        {
          ReadFile(read_stdout,buf,1023,&bread,NULL);
          OemToAnsi(buf, buf);
          Line = buf;
          bzero(buf);
        }
      }
      else
      {
        ReadFile(read_stdout,buf,1023,&bread,NULL);
        OemToAnsi(buf, buf);
        Line = buf;
      }
    // Ýòî òàê âûðàâíèâàåì êîíåö ñòðîêè è âîçâðàò êîðåòêè (äëÿ ðàçíûõ ïðèëîæåíèé)
    Line = AnsiReplaceStr(Line, "\r\n", "\n");
    Line = AnsiReplaceStr(Line, "\n", "\r\n");
    return Line;
    }
  }
  catch(Exception &E)
  {
    this->Stop();
    MainForm->RefreshTimer->Enabled = false;
    ShowMessage("READ: " + E.Message);
  }
}

void Pipes::Stop()
{
  this->Write("exit");
  WaitForSingleObject(pi.hProcess, INFINITE);
  CloseHandle(pi.hThread);
  CloseHandle(pi.hProcess);
  CloseHandle(newstdin);
  CloseHandle(newstdout);
  CloseHandle(read_stdout);
  CloseHandle(write_stdin);
  MainForm->RefreshTimer->Enabled=false;
}
//---------------------------------------------------------------------------