home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / ocl150a.zip / OCL / Source / OProcStat.cpp < prev    next >
C/C++ Source or Header  |  1996-08-12  |  5KB  |  201 lines
  1. // OCL - OS/2 Class Library
  2. // (c) Cubus 1995
  3. // All Rights Reserved
  4. // OProcStat.cpp
  5.  
  6.  
  7.  
  8. /*
  9.  * Redistribution and use in source and binary forms, with or without
  10.  * modification, are permitted provided that the following conditions
  11.  * are met:
  12.  * 1. Redistributions of source code must retain the above copyright
  13.  *    notice, this list of conditions and the following disclaimer.
  14.  * 2. Neither the name Cubus nor the name Team OCL may be used to
  15.  *    endorse or promote products derived from this software
  16.  *    without specific prior written permission.
  17.  * 3. See OCL.INF for a detailed copyright notice.
  18.  *
  19.  *              THIS SOFTWARE IS PROVIDED ``AS IS'' AND
  20.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  21.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  22.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  23.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  24.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  25.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  26.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  27.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  28.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  29.  * SUCH DAMAGE.
  30.  */
  31.  
  32. // $Header: W:/Projects/OCL/Source/rcs/OProcStat.cpp 1.50 1996/08/11 23:49:29 B.STEIN Release $
  33.  
  34. #define __OCL_SOURCE__
  35.  
  36. #define OINCL_OSTRING
  37. #define OINCL_BASE
  38.  
  39. #include <ocl.hpp>
  40. #include <OProcStat.hpp>
  41. #include <OBuffer.hpp>
  42.  
  43. #define PROCSTAT_PID  1
  44. #define PROCSTAT_PPID 2
  45.  
  46.  
  47. // member funtions of OProcStat
  48.  
  49. OProcStat::OProcStat()
  50.   : ProcCount(0), TaskCount(0)
  51.  {}
  52.  
  53.  
  54. OProcStat::~OProcStat()
  55. {
  56.  ProcList.reset();
  57. }
  58.  
  59.  
  60. PSZ OProcStat::isOfType() const
  61. { 
  62.  return("OProcStat"); 
  63. }
  64.  
  65.  
  66. BOOL OProcStat::getProcesses()
  67. {
  68.  OProcRecord record;
  69.  qsPtrRec_t  *MainIndex;
  70.  OBuffer     buffer(OBuffer::TILESIZE,
  71.                     OBuffer::ALLOC |
  72.                     OBuffer::TILED);
  73.  
  74. // clean up
  75.  
  76.   ProcList.reset();
  77.   ProcCount = 0;
  78.   TaskCount = 0;
  79.  
  80. // Use DosQuerySysState - a new 32bit-API - for Warp and PPC.
  81.  
  82. if(DosQuerySysState(0x17, 0, 0, 0, (PBYTE)buffer, PROCSTAT_BUFFER_SIZE))
  83.   {
  84.    DosFreeMem(buffer);
  85.    return(FALSE); 
  86.   }
  87. else
  88.   {
  89.    qsLrec_t  *module;
  90.    MainIndex = (qsPtrRec_t*) buffer.getPtr();
  91.    module    = MainIndex->pLibRec;     // Start at first module
  92.  
  93.    while(module)
  94.     {
  95.      qsPrec_t *proc = MainIndex->pProcRec;
  96.  
  97.      while((proc) && (proc->RecType != PROCESS_END_INDICATOR))
  98.        {
  99.         if(proc->hMte == module->hmte)       // Module found
  100.          {
  101.           if (!strcmp((PSZ)module->pName, "SYSINIT"))
  102.             record.Module << (PSZ) "MDOS Process";
  103.           else
  104.             record.Module << (PSZ) module->pName;
  105.           record.Pid     = proc->pid;
  106.           record.PPid    = proc->ppid;
  107.           record.Sid     = proc->sgid;
  108.           record.Threads = proc->cTCB;
  109.           ProcList << &record;
  110.           TaskCount++;
  111.          }
  112.         proc = (qsPrec_t *)(proc->pThrdRec + proc->cTCB);       // point to next process
  113.        }
  114.      ProcCount++;
  115.      module = module->pNextRec;  // point to next module
  116.     }
  117.   }
  118.  return(TRUE);
  119. }
  120.  
  121.  
  122.  
  123. ULONG OProcStat::getPid(PSZ module)
  124. {
  125.  return(getNumInfo(module, PROCSTAT_PID));
  126. }
  127.  
  128.  
  129. ULONG OProcStat::getPPid(PSZ module)
  130. {
  131.  return(getNumInfo(module, PROCSTAT_PPID));
  132. }
  133.  
  134.  
  135. ULONG OProcStat::getNumInfo(PSZ module, ULONG flag)
  136. {
  137.  OString                tmpString;
  138.  OListItem<OProcRecord> *tmp;
  139.  
  140.  if (!getProcesses())
  141.    return(0);
  142.  
  143.  tmp = ProcList.first();
  144.  tmpString << module;
  145.  
  146.  while((tmp) && (tmp->item))
  147.   {
  148.    if (tmpString.getText() == tmp->item->Module.getText())
  149.     {
  150.      switch (flag)
  151.       {
  152.        case PROCSTAT_PID:
  153.          return(tmp->item->Pid);
  154.  
  155.        case PROCSTAT_PPID:
  156.          return(tmp->item->PPid);
  157.       }     
  158.     }
  159.    tmp = tmp->next;
  160.   }
  161.  return(0);
  162. }
  163.  
  164.  
  165. PSZ OProcStat::getModule(ULONG pid)
  166. {
  167.  OListItem<OProcRecord> *tmp;
  168.  
  169.  if (!getProcesses())
  170.    return(0);
  171.  
  172.  tmp = ProcList.first();
  173.  
  174.  while((tmp) && (tmp->item))
  175.   {
  176.    if (tmp->item->Pid == pid)
  177.      return(strdup(tmp->item->Module));
  178.    tmp = tmp->next;
  179.   }
  180.  return(NULL);
  181. }
  182.  
  183.  
  184. ULONG OProcStat::getProcCount() 
  185. {
  186.  if (getProcesses())
  187.     return(ProcCount);
  188.  else return(0); 
  189. }
  190.  
  191.  
  192. ULONG OProcStat::getTaskCount() 
  193. {
  194.  if (getProcesses())
  195.     return(TaskCount);
  196.  else return(0); 
  197. }
  198.  
  199.  
  200. // end of source
  201.