home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / ocl150a.zip / OCL / Source / OFcl.cpp < prev    next >
C/C++ Source or Header  |  1996-08-12  |  8KB  |  337 lines
  1. // OCL - OS/2 Class Library
  2. // (c) Cubus 1995
  3. // All Rights Reserved
  4. // OFcl.CPP
  5.  
  6. // member functions of fcl FileCollector
  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.  
  33. // $Header: W:/Projects/OCL/Source/rcs/OFcl.cpp 1.50 1996/08/11 23:49:15 B.STEIN Release $
  34.  
  35. #define __OCL_SOURCE__
  36.  
  37. #define OINCL_OSTRING
  38. #define OINCL_BASE
  39.  
  40. #include <ocl.hpp>
  41. #include <OFcl.hpp>
  42.  
  43.  
  44. OFcl::OFcl()    // default constructor
  45.    : ulMaxFiles(200), space(0), found(0),
  46.    checkDate(FALSE), checkTime(FALSE),
  47.    AttrMask(FILE_SYSTEM | FILE_HIDDEN | FILE_ARCHIVED | FILE_READONLY),
  48.    DateTime(NULL),
  49.    stopASAP(FALSE)
  50. {
  51.  bufSize = sizeof(FILEFINDBUF3) * ulMaxFiles;
  52.  setupFilebuf();
  53. }
  54.  
  55.  
  56. OFcl::~OFcl()   // destructor
  57. {
  58.  fclstop();
  59.  DosFreeMem(pffb);
  60.  // cleanup the Lists
  61.  dirs.reset();
  62.  files.reset();
  63. }
  64.  
  65.  
  66. PSZ OFcl::isOfType() const
  67. { 
  68.  return("OFcl"); 
  69. }
  70.  
  71.  
  72. void OFcl::setupFilebuf()      // allocate the filefind3 buffer
  73. {
  74.  DosAllocMem((PPVOID)&pffb,  bufSize, PAG_READ | PAG_WRITE | PAG_COMMIT);
  75. }
  76.  
  77.  
  78. void OFcl::fclinit()
  79. {
  80.  // cleanup the Lists
  81.  dirs.reset();
  82.  files.reset();
  83.  space = 0;
  84.  found = 0;
  85.  checkDate = FALSE;
  86.  checkTime = FALSE;
  87.  DateTime = NULL;
  88.  AttrMask = FILE_SYSTEM | FILE_HIDDEN | FILE_ARCHIVED | FILE_READONLY;
  89.  ulMaxFiles = 200;
  90.  stopASAP = FALSE;
  91. }
  92.  
  93.  
  94. void OFcl::setCheck(BOOL date, BOOL time)
  95. {
  96.  checkTime = time;
  97.  checkDate = date; 
  98. }
  99.  
  100.  
  101. void OFcl::fclstop()
  102. {
  103.  stopASAP = TRUE;
  104. }
  105.  
  106.  
  107. // this member handles the collect
  108.  
  109. void OFcl::collect(PCSZ spec, BOOL rec, BOOL local)
  110. {
  111.  ULONG    sDrive;
  112.  ULONG    ulDriveMap, ulDriveNum=0L;
  113.  PSZ      pathname = new CHAR[CCHMAXPATH];
  114.  OString  Spec(spec);
  115.  
  116.  strupr(Spec);
  117.  
  118.  recurse = rec;
  119.  
  120.  if (local)
  121.   {
  122.    finddrives(spec);
  123.    return;
  124.   }
  125.  
  126.  if(*(Spec.getText()+1)==':')
  127.    {
  128.     sDrive = (ULONG) ((*Spec.getText()) - '@');
  129.     if (DosSetDefaultDisk(sDrive))
  130.       return;
  131.     dirs << Spec;
  132.    }
  133.  else
  134.    {
  135.     if (strchr(Spec, '\\'))
  136.       return;
  137.     DosQueryCurrentDisk(&ulDriveNum, &ulDriveMap);
  138.     DosQueryCurrentDir(ulDriveNum, (PBYTE)pathname, &ulDriveMap);
  139.     strcat(pathname, "\\");
  140.     strcat(pathname, Spec);
  141.     dirs << pathname;
  142.    }
  143.  
  144.  ext << (PSZ) strrchr(Spec, '\\');
  145.  findfiles(dirs.last());
  146.  delete[] pathname;
  147.  return;
  148. }
  149.  
  150.  
  151. void OFcl::finddirs(PSZ dir)
  152. {
  153.  HDIR dhdir = HDIR_CREATE;
  154.  
  155.  strcpy(strrchr(dir, '\\'), "\\*");
  156.  ulMaxFiles = 200;
  157.  
  158.  rc = DosFindFirst(dir, &dhdir, MUST_HAVE_DIRECTORY,
  159.                        pffb, bufSize, &ulMaxFiles, FIL_STANDARD);
  160.  while((!rc) && (!stopASAP))
  161.    {
  162.     if (ulMaxFiles)
  163.        adddirs(dir);
  164.     ulMaxFiles = 200;
  165.     rc = DosFindNext(dhdir, pffb, bufSize, &ulMaxFiles);
  166.    }
  167.  
  168.  DosFindClose(dhdir);
  169. }
  170.  
  171.  
  172. void OFcl::findfiles(OListItem<OString> *tmp)  // find all files
  173. {
  174.  OListItem<OString> *strItem = tmp;
  175.  
  176.  while((strItem) && (!stopASAP))
  177.   {
  178.    HDIR hdir = HDIR_CREATE;
  179.  
  180.    ulMaxFiles = 200;
  181.    filespec << strItem->item->getText();
  182.  
  183.    rc = DosFindFirst(strItem->item->getText(), &hdir, (ULONG) AttrMask,
  184.                      pffb, bufSize, &ulMaxFiles, FIL_STANDARD);
  185.  
  186.    while((!rc) && (!stopASAP))
  187.     {
  188.      if (ulMaxFiles)
  189.        addfiles();
  190.      ulMaxFiles = 200;
  191.      rc = DosFindNext(hdir, pffb, bufSize, &ulMaxFiles);
  192.     }
  193.  
  194.    DosFindClose(hdir);
  195.  
  196.    if ((recurse) && (!stopASAP)) {
  197.     finddirs(strItem->item->getText());
  198.     strItem = strItem->next; }
  199.    else
  200.      break;
  201.   }
  202. }
  203.  
  204.  
  205. void OFcl::addfiles()  // add all found files to a List
  206. {
  207.  PFILEFINDBUF3   pffbFile;
  208.  ULONG           i;
  209.  PSZ             Buf = new CHAR[CCHMAXPATH];
  210.  
  211.  strcpy(Buf, filespec);
  212.  
  213.  tmp = (PBYTE) pffb;
  214.  
  215.  for(i = 0; (i < ulMaxFiles) && (!stopASAP); i++)
  216.    {
  217.     BOOL matches = TRUE;
  218.  
  219.     pffbFile = (PFILEFINDBUF3) tmp;
  220.     strcpy(strrchr(Buf, '\\')+1, pffbFile->achName);
  221.     if (checkDate)
  222.       matches = hasDate(pffbFile);
  223.     if (matches)
  224.       if (checkTime)
  225.         matches = hasTime(pffbFile);
  226.     if (matches)
  227.       files << Buf;
  228.     found++;
  229.     space += pffb->cbFileAlloc;
  230.     tmp += pffbFile->oNextEntryOffset;                // Point to the next file in the buffer
  231.    }
  232.  delete[] Buf;
  233. }
  234.  
  235.  
  236. void OFcl::adddirs(PCSZ dir)
  237. {
  238.  PFILEFINDBUF3   pffbFile;
  239.  ULONG           i;
  240.  PSZ             Buf = new CHAR[CCHMAXPATH];
  241.  
  242.  tmp = (PBYTE) pffb;
  243.  
  244.  for(i = 0; (i < ulMaxFiles) && (!stopASAP); i++)
  245.    {
  246.     pffbFile = (PFILEFINDBUF3) tmp;
  247.  
  248.     if (((BOOL)(pffbFile->attrFile&FILE_DIRECTORY)) && (pffbFile->achName[0]!='.'))
  249.      {
  250.       strcpy(Buf, dir);
  251.       strcpy(strrchr(Buf, '\\')+1, pffbFile->achName);
  252.       strcat(Buf, ext.getText());
  253.       dirs << Buf;
  254.      }
  255.     tmp += pffbFile->oNextEntryOffset;                // Point to the next file in the buffer
  256.    }
  257.  
  258.  delete[] Buf;
  259. }
  260.  
  261.  
  262. void OFcl::finddrives(PCSZ spec)
  263. {
  264.  ULONG   DriveNumber; 
  265.  ULONG   LogicalDriveMap;
  266.  UCHAR   Drive[] = "c:";
  267.  ULONG   Temp;
  268.  HFILE   hfDiskHandle;
  269.  ULONG   Action;
  270.  
  271.  if (DosQueryCurrentDisk(&DriveNumber, &LogicalDriveMap))
  272.     return;
  273.  
  274.  for(Temp=(ULONG)Drive[0]-'a', LogicalDriveMap>>=2;
  275.      Temp<=(ULONG)('z'-'a');
  276.      Temp++, Drive[0]++, LogicalDriveMap>>=1)
  277.       {
  278.        if(!(LogicalDriveMap&0x1))
  279.           continue;
  280.  
  281.        if (!DosOpen((PSZ)Drive, &hfDiskHandle, &Action, 0, FILE_NORMAL,
  282.                        OPEN_ACTION_OPEN_IF_EXISTS, OPEN_FLAGS_DASD | 
  283.                        OPEN_FLAGS_FAIL_ON_ERROR | OPEN_FLAGS_NOINHERIT | 
  284.                        OPEN_SHARE_DENYREADWRITE | OPEN_ACCESS_READONLY, 0))
  285.            {
  286.             struct _PPF  
  287.              {
  288.               BYTE        bCommandInformation;
  289.               BYTE        bDriveUnit;
  290.              } PPF={0, 0};
  291.  
  292.             struct _DDF                
  293.              {
  294.               BYTE        bData;
  295.              } DDF;
  296.  
  297.             ULONG       ulParamLengthInOut=sizeof(PPF);
  298.             ULONG       ulDataLengthInOut=sizeof(DDF);
  299.  
  300.             DosDevIOCtl(hfDiskHandle, IOCTL_DISK, DSK_BLOCKREMOVABLE,
  301.                 &PPF, ulParamLengthInOut, &ulParamLengthInOut,
  302.                 &DDF, ulDataLengthInOut, &ulDataLengthInOut);
  303.             DosClose(hfDiskHandle);
  304.             if(DDF.bData)
  305.               {
  306.                OString buffer((PSZ) Drive);
  307.  
  308.                buffer + (PSZ) strchr(spec, '\\');
  309.                collect(buffer, recurse, FALSE);
  310.               }
  311.           }
  312.       }
  313.  return;
  314. }
  315.  
  316.  
  317. BOOL OFcl::hasTime(PFILEFINDBUF3 buf)
  318. {
  319.  if (!DateTime)
  320.    return(FALSE);
  321.  return((buf->ftimeCreation.minutes == DateTime->minutes) &&
  322.          (buf->ftimeCreation.hours == DateTime->hours));
  323. }
  324.  
  325.  
  326. BOOL OFcl::hasDate(PFILEFINDBUF3 buf)
  327. {
  328.  if (!DateTime)
  329.    return(FALSE);
  330.  return((buf->fdateCreation.day == DateTime->day) &&
  331.         (buf->fdateCreation.month == DateTime->month) &&
  332.         (buf->fdateCreation.year == DateTime->year));
  333. }
  334.  
  335.  
  336. // end of source
  337.