home *** CD-ROM | disk | FTP | other *** search
/ QBasic & Borland Pascal & C / Delphi5.iso / C / BC_502 / ADDON.PAK / MSGTEST.CPP < prev    next >
Encoding:
C/C++ Source or Header  |  1997-05-06  |  6.3 KB  |  248 lines
  1. /*:::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::
  2.  
  3.    msgtest.cpp
  4.    Created: 1/3/96
  5.    Copyright (c) 1987, 1996 Borland International Inc.  All Rights Reserved.
  6.    $Revision:   1.13  $
  7.     
  8. :::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::::*/ 
  9. #ifndef __AOEXPCH_H
  10.   #include "aoexpch.h"
  11. #endif
  12. #pragma hdrstop
  13.  
  14. #include <stdio.h> 
  15. #include <ideaddon\iide.h>
  16. #include "msgtest.h"
  17.  
  18. GenericMessageFolder::GenericMessageFolder(LPCSTR name) {
  19.   IMessageSystem* pMsgSys;
  20.  
  21.   pMsgSys = GET_INTERFACE(IMessageSystem);
  22.  
  23.   d_folder = pMsgSys->NewFolder(MakePolyString(name));
  24.  
  25.   pMsgSys->Release();
  26.   d_msgItem = 0;
  27. };
  28.  
  29. void GenericMessageFolder::OpenFile(LPCSTR fileName, LPCSTR msg) {
  30.   d_msgItem = d_folder->NewFileMessage( fileName ? MakePolyString(fileName) : 0
  31.                                       , MakePolyString(msg));
  32. }
  33.  
  34. void GenericMessageFolder::Log(LPCSTR msg) {
  35.   d_folder->NewMessage(d_msgItem, MakePolyString(msg));
  36. }
  37.  
  38. #define BUFSIZE 4096 
  39.  
  40. VOID GenericMessageFolder::ReadFromPipe(HANDLE hFile) { 
  41.   /* Read output from child, and write it to parent's STDOUT. */ 
  42.  
  43.   DWORD dwRead, dwWritten; 
  44.   CHAR chBuf[BUFSIZE+1]; 
  45.   //
  46.   // append a null terminator. this is needed by strtok
  47.   //
  48.   chBuf[BUFSIZE] = 0;
  49.   for (;;) { 
  50.       if (! ReadFile(hFile, chBuf, BUFSIZE, &dwRead, NULL) || 
  51.           dwRead == 0) break; 
  52.  
  53.       chBuf[dwRead] = 0;
  54.       if (! WriteMessage(chBuf, dwRead, &dwWritten)) 
  55.           break; 
  56.   }
  57. } 
  58.  
  59. BOOL GenericMessageFolder::WriteMessage(LPSTR chBuf, DWORD /*dwRead*/, DWORD* /*dwWritten*/) {
  60.   LPSTR pH = chBuf;
  61.   LPSTR p = strtok(pH, "\n\r");
  62.   while (p != NULL) {
  63.     Log(p);
  64.     p = strtok(NULL, "\n\r");
  65.   };
  66.   return 1;
  67. }
  68.  
  69. DWORD ThreadFunc(LPDWORD lpdwParam) {
  70.   GenericMessageFolder* folder = (GenericMessageFolder*)lpdwParam;
  71.   folder->Run();
  72.   return 0;
  73. }
  74.  
  75. unsigned GenericMessageFolder::DoThread(LPCSTR cmd) {
  76.   d_currentCmd = cmd;
  77.  
  78.   DWORD dwThreadId;
  79.   HANDLE hThread;
  80.   hThread = CreateThread(
  81.       NULL,                       /* no security attributes       */
  82.       0,                           /* use default stack size        */
  83.       (LPTHREAD_START_ROUTINE) ThreadFunc, /* thread function      */
  84.       (LPVOID)this,               /* argument to thread function   */
  85.       0,                           /* use default creation flags    */
  86.       &dwThreadId);               /* returns the thread identifier */
  87.  
  88.   /* Check the return value for success. */
  89.  
  90.   if (hThread == NULL)
  91.     return 0;
  92.  
  93.   return 1;
  94. };
  95.  
  96. void GenericMessageFolder::Run() {
  97.   CreatePipedProcess((LPSTR)d_currentCmd);
  98. };
  99.  
  100. DWORD GenericMessageFolder::CreatePipedProcess(LPSTR commandLine) { 
  101.  
  102.   HRESULT                hr     = NOERROR;
  103.   SECURITY_ATTRIBUTES    sAttr   =
  104.   {
  105.     sizeof(SECURITY_ATTRIBUTES),
  106.     0,
  107.     TRUE
  108.   };
  109.  
  110.   STARTUPINFO    siStartInfo;
  111.   BOOL            bSuccess;
  112.  
  113.   memset(&siStartInfo, 0, sizeof(STARTUPINFO));
  114.   siStartInfo.cb = sizeof(STARTUPINFO);
  115.  
  116.   // Create a pipe for the child's stdout.
  117.   HANDLE          hPipeRead;      // The child's stdin.
  118.   HANDLE          hPipeWrite;    // The child's stdout.
  119.  
  120.   if (!CreatePipe(  &hPipeRead,
  121.                     &hPipeWrite,
  122.                     &sAttr,
  123.                     0)) {
  124.     return  ResultFromScode(E_FAIL);
  125.   }
  126.  
  127.   // Don't allow the child program to inherit the read end of the pipe.
  128.   DuplicateHandle(GetCurrentProcess(),
  129.                   hPipeRead,
  130.                   GetCurrentProcess(),
  131.                   0,
  132.                   0,
  133.                   FALSE,
  134.                   DUPLICATE_SAME_ACCESS);
  135.  
  136.   siStartInfo.dwFlags    = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;
  137.   siStartInfo.hStdOutput  = hPipeWrite;
  138.   siStartInfo.wShowWindow = SW_HIDE;
  139.  
  140.   // Start the child process.
  141.   PROCESS_INFORMATION piProcInfo; 
  142.   bSuccess = CreateProcess(NULL,
  143.                            commandLine,
  144.                            &sAttr,
  145.                            &sAttr,
  146.                            TRUE,
  147.                            CREATE_NEW_PROCESS_GROUP,
  148.                            0,
  149.                            0,
  150.                            &siStartInfo,
  151.                            &piProcInfo);
  152.  
  153.   CloseHandle(hPipeWrite);
  154.   
  155.   if (bSuccess)
  156.   {
  157.     // Close the handles to the child process and thread.
  158.     CloseHandle(piProcInfo.hThread);
  159.     CloseHandle(piProcInfo.hProcess);
  160.     ReadFromPipe(hPipeRead); 
  161.   }
  162.  
  163.   CloseHandle(hPipeRead);
  164.   
  165.   return  hr;
  166. }
  167.  
  168. /* ****************************************************************************
  169.  *
  170.  *
  171.  */
  172. MsgSystemTest::MsgSystemTest() {
  173.    d_msgFolder1 = NULL;
  174.    d_msgFolder2 = NULL;
  175. }
  176.  
  177. MsgSystemTest::~MsgSystemTest() {
  178.   UnInit();
  179. }
  180. //.............................................................................
  181. BOOL MsgSystemTest::Init() 
  182. {
  183.   d_msgFolder1 = new GenericMessageFolder("Addon");
  184.   d_msgFolder2 = new GenericMessageFolder("Addon");
  185.  
  186.   OutStr( "MsgSystemTest::Init()" );
  187.   return FALSE;
  188. }
  189.  
  190. void MsgSystemTest::UnInit() 
  191. {
  192.   OutStr( "MsgSystemTest::UnInit()" );
  193.   if ( d_msgFolder1 ) {
  194.      delete d_msgFolder1;
  195.   }
  196.   if ( d_msgFolder2 ) {
  197.      delete d_msgFolder2;
  198.   }    
  199. }
  200.  
  201. const char * MsgSystemTest::GetName() 
  202. {
  203.   return "Message System Test";
  204. }
  205.  
  206. const char * MsgSystemTest::GetTestDescription( int testNum) 
  207. {
  208.   switch ( testNum )
  209.   {
  210.     case 1:
  211.       return "Create a message folder associated with a file";
  212.  
  213.     case 2:
  214.       return "Display a meesage in folder created by test1";
  215.  
  216.     case 3:
  217.       return "Create a message folder";
  218.  
  219.     case 4:
  220.       return "Display a meesage in folder created by test3";
  221.   }
  222.   return "This test not implemented.";
  223. }
  224.  
  225. void MsgSystemTest::DoTest( int testNum ) 
  226. {
  227.   switch ( testNum ) 
  228.   {
  229.     case 1:
  230.       d_msgFolder1->OpenFile("msgtest.cpp", "Click this line to open file msgtest.cpp");
  231.       break;
  232.     case 2:
  233.       // my test here 
  234.       d_msgFolder1->DoThread("where find.*");
  235.       d_msgFolder1->Log("Test output");
  236.       break;
  237.     case 3:
  238.       d_msgFolder2->OpenFile(NULL, "Click this line to expand/collapse node");
  239.       break;
  240.     case 4:
  241.       d_msgFolder2->Log("Test output");
  242.       break;
  243.     default: 
  244.       OutStr( FormatStr( "Test #%d Not Implemented!", testNum ) );
  245.       break;
  246.   }
  247. }
  248.