home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / rwp15.zip / FILESRCH.C < prev    next >
C/C++ Source or Header  |  1993-03-24  |  9KB  |  297 lines
  1. /* --------------------------------------------------------------------
  2.                               File Search
  3.                               Chapter 15
  4.  
  5.                     Real World Programming for OS/2
  6.              Copyright (c) 1993 Blain, Delimon, and English
  7. -------------------------------------------------------------------- */
  8.  
  9. #define INCL_DOS
  10. #define INCL_DOSFILEMGR
  11. #define INCL_WIN
  12. #define INCL_ERRORS
  13.  
  14. #include <os2.h>
  15. #include <string.h>
  16. #include "misc.h"
  17. #include "drives.h"
  18. #include "filesrch.h"
  19.  
  20. /* Thread Function */
  21.  
  22. #ifdef __WATCOM__
  23. VOID __stdcall SearchThread (HWND);
  24. #else
  25. VOID SearchThread (HWND);
  26. #endif
  27.  
  28. /* Local Functions    */
  29. VOID SearchForFile (HWND);
  30. VOID SetControlStates (HWND);
  31. VOID StartSearch (HWND);
  32. VOID StopSearch (VOID);
  33.  
  34. /* External variables */
  35. extern HAB       hab;                  /* Handle to anchor block     */
  36. extern DRIVEINFO DriveInfo[];
  37. extern ULONG     ulNumDrives;
  38.  
  39. /* Global variables */
  40. TID  SearchThreadID;                   /* Thread ID of Search thread */
  41. CHAR szFileName[CCHMAXPATHCOMP];       /* Filename to search         */
  42. CHAR szFullPath[CCHMAXPATH] = " :\\";  /* File path buffer           */
  43. HWND hWndFileSearch         = 0;       /* File search dialog window  */
  44. BOOL bSearching             = FALSE;   /* Search in progress flag    */
  45.  
  46. /* ------------------ Search Thread Function ----------------------- */
  47.  
  48. /*
  49.     The Search thread is responsible for searching the selected disk
  50.     for the specified filename.  This function will search all
  51.     subdirectories starting with the directory in szFullPath.
  52.     The field IDC_CURRENTDIR is updated with the name of each directory 
  53.     that is being searched.  Each matching filename is added to
  54.     the IDC_DIRLIST listbox.
  55.  
  56.     Since this thread will be directly updating the dialog box control
  57.     windows it must create a message queue.  Thus it will be a
  58.     Message queue thread.
  59.  
  60.     The thread is automatically terminated when all subdirectories have
  61.     been search or when the bSearching flag is reset.
  62. */
  63.  
  64. #ifdef __IBMC__
  65. #pragma linkage(SearchThread, system)
  66. #endif
  67.  
  68. #ifdef __WATCOM
  69. VOID __stdcall SearchThread (HWND hWnd)
  70. #else
  71. VOID SearchThread (HWND hWnd)
  72. #endif
  73.  
  74. {
  75.     HMQ hmqThread;
  76.     HAB habThread;
  77.  
  78.     /* Create message queue for this thread */
  79.     habThread = WinInitialize (0);
  80.     hmqThread = WinCreateMsgQueue (hab, 0);
  81.  
  82.     bSearching = TRUE;
  83.  
  84.     SetControlStates (hWnd);
  85.  
  86.     /* Perform the search */
  87.     SearchForFile (hWnd);
  88.  
  89.     bSearching = FALSE;
  90.  
  91.     SetControlStates (hWnd);
  92.  
  93.     /* Destroy the message queue for this thread */
  94.     WinDestroyMsgQueue (hmqThread);
  95.     WinTerminate (habThread);
  96.  
  97.     DosExit (EXIT_THREAD, 0L);
  98. }
  99.  
  100. /* ----------------------- Local Functions ----------------------- */
  101.  
  102. VOID SearchForFile (HWND hWnd)
  103. { 
  104.     HDIR         hDir        = HDIR_CREATE;
  105.     ULONG        ulDirPos    = strlen (szFullPath);
  106.     ULONG        ulFileCount = 1L;
  107.     FILEFINDBUF3 FindBuf3;
  108.  
  109.     /* Check for filename length overflow */
  110.     if ((ulDirPos + strlen(szFileName)) > CCHMAXPATH)
  111.         return;
  112.  
  113.     WinSetDlgItemText (hWnd, IDC_CURRENTDIR, szFullPath);
  114.  
  115.     /* Search for all files in directory matching the filename */
  116.     strcat (szFullPath, szFileName);    
  117.     if (!DosFindFirst (szFullPath, &hDir, FILE_NORMAL, &FindBuf3,
  118.             sizeof(FILEFINDBUF3), &ulFileCount, FIL_STANDARD))
  119.     {
  120.         do
  121.         {
  122.             szFullPath[ulDirPos] = '\0';
  123.             strncat (szFullPath, FindBuf3.achName, CCHMAXPATH);
  124.             WinSendDlgItemMsg (hWnd, IDC_DIRLIST, LM_INSERTITEM, (MPARAM)LIT_END,
  125.                 szFullPath);
  126.             ulFileCount = 1L;
  127.         }
  128.         while (bSearching &&
  129.                !DosFindNext (hDir, &FindBuf3, sizeof(FILEFINDBUF3), &ulFileCount));
  130.         DosFindClose (hDir);
  131.     }
  132.  
  133.     /* Has the user cancelled the searching? */
  134.     if (!bSearching)
  135.         return;
  136.  
  137.     /* For each subdirectory in this directory call SearchForFile */
  138.     szFullPath[ulDirPos] = '\0';
  139.     strcat (szFullPath, "*.*");
  140.     ulFileCount = 1L;
  141.     hDir        = HDIR_CREATE;
  142.     if (!DosFindFirst (szFullPath, &hDir, MUST_HAVE_DIRECTORY, &FindBuf3,
  143.             sizeof(FILEFINDBUF3), &ulFileCount, FIL_STANDARD))
  144.     {
  145.         do
  146.         {
  147.             if (!strcmp(FindBuf3.achName, ".") ||
  148.                 !strcmp(FindBuf3.achName, ".."))
  149.                 continue;
  150.  
  151.             /* Check for path name length overflow */
  152.             if ((ulDirPos + FindBuf3.cchName + 1) >= CCHMAXPATH)
  153.                 continue;
  154.  
  155.             szFullPath[ulDirPos] = '\0';
  156.             strcat (szFullPath, FindBuf3.achName);
  157.             strcat (szFullPath, "\\");
  158.             SearchForFile (hWnd);
  159.             ulFileCount = 1L;
  160.         }
  161.         while (bSearching &&
  162.                !DosFindNext (hDir, &FindBuf3, sizeof(FILEFINDBUF3), &ulFileCount));
  163.         DosFindClose (hDir);
  164.     }
  165.  
  166.     return;
  167. }
  168.  
  169. VOID SetControlStates (HWND hWnd)
  170. {
  171.     /* Show the searching fields */
  172.     WinShowWindow (WinWindowFromID (hWnd, IDC_SEARCHING), bSearching);
  173.     WinShowWindow (WinWindowFromID (hWnd, IDC_CURRENTDIR), bSearching);
  174.  
  175.     /* Disable filename and drivelist input controls */
  176.     WinEnableControl (hWnd, IDC_FILENAME, !bSearching);
  177.     WinEnableControl (hWnd, IDC_DRIVELIST, !bSearching);
  178.  
  179.     /* Change button text */
  180.     WinSetDlgItemText (hWnd, IDC_SEARCH, bSearching ? "Stop" : "Search");
  181.  
  182.     return;
  183. }
  184.  
  185. VOID StartSearch (HWND hWnd)
  186. {
  187.     ULONG ulInx;
  188.  
  189.     /* Get file spec to search */
  190.     WinQueryDlgItemText (hWnd, IDC_FILENAME, sizeof(szFileName), szFileName);
  191.     if (!szFileName[0])
  192.     {
  193.         DosBeep (750, 200);
  194.         return;
  195.     }
  196.  
  197.     /* Get drive selection and start at the root directory */
  198.     ulInx = (ULONG)WinSendDlgItemMsg (hWnd, IDC_DRIVELIST, 
  199.                        LM_QUERYSELECTION, (MPARAM)LIT_FIRST, 0L);
  200.     szFullPath[0] = DriveInfo[ulInx].szDrive[0];
  201.     szFullPath[3] = '\0';
  202.  
  203.     /* Clear the directory listbox */
  204.     WinSendDlgItemMsg (hWnd, IDC_DIRLIST, LM_DELETEALL, 0L, 0L);
  205.  
  206.     /* Start the search thread */
  207.     DosCreateThread (&SearchThreadID, (PFNTHREAD)SearchThread, (ULONG)hWnd, 0, 0x10000);
  208.  
  209.     return;
  210. }
  211.  
  212. VOID StopSearch ()
  213. {
  214.     /* Signal search thread to stop */
  215.     bSearching = FALSE;
  216.  
  217.     /* Don't attempt to call DosWaitThread here.  Doing so will hang
  218.        the process since when the search thread returns from the initial
  219.        call to SearchForFile it will call SetControlStates.  The call
  220.        to WinShowWindow, for example, will attempt to send a message to 
  221.        the control window's owner and since the main thread would be
  222.        blocked waiting for the search thread the message could not be
  223.        sent and the search thread would block. 
  224.  
  225.        The thread will terminate as soon as it sees the bSearching
  226.        flag has been set to FALSE. */
  227.  
  228.     return;
  229. }
  230.  
  231. /* ----------------------  Dialog Function ----------------------- */
  232.  
  233. VOID DoFileSearch (HWND hWnd)
  234. {
  235.     if (!hWndFileSearch)
  236.         hWndFileSearch = WinLoadDlg (HWND_DESKTOP, hWnd, FileSearchDlgProc, 
  237.                              0L, IDD_FILESEARCH, NULL);
  238.     else
  239.         WinSetActiveWindow (HWND_DESKTOP, hWndFileSearch);
  240.         
  241.     return;
  242. }
  243.  
  244. MRESULT EXPENTRY FileSearchDlgProc (HWND hWnd, ULONG msg, MPARAM mp1, MPARAM mp2)
  245. {
  246.     BOOL    bHandled = TRUE;
  247.     MRESULT mReturn  = 0;
  248.  
  249.     switch (msg)
  250.     {           
  251.         case WM_INITDLG:
  252.             QueryDrives (hWnd);
  253.             WinShowWindow (WinWindowFromID (hWnd, IDC_SEARCHING), FALSE);
  254.             WinShowWindow (WinWindowFromID (hWnd, IDC_CURRENTDIR), FALSE);
  255.             WinSendDlgItemMsg (hWnd, IDC_FILENAME, EM_SETTEXTLIMIT, 
  256.                 MPFROMSHORT(CCHMAXPATHCOMP), 0L);
  257.             break;
  258.  
  259.         case WM_COMMAND:
  260.         case WM_SYSCOMMAND:
  261.               switch (SHORT1FROMMP(mp1))
  262.             {
  263.                 case IDC_SEARCH:
  264.                     if (bSearching)
  265.                         StopSearch ();
  266.                     else
  267.                         StartSearch (hWnd);
  268.                     break;
  269.  
  270.                     case SC_CLOSE:
  271.                     if (bSearching)
  272.                         StopSearch ();
  273.                     WinDestroyWindow (hWnd);
  274.                     bHandled = TRUE;
  275.                     break;
  276.  
  277.                 default:
  278.                     bHandled = (msg == WM_COMMAND);
  279.             }
  280.             break;
  281.  
  282.         case WM_DESTROY:
  283.             hWndFileSearch = 0;
  284.             break;
  285.  
  286.         default:
  287.             bHandled = FALSE;
  288.             break;
  289.     }
  290.  
  291.     if (!bHandled)
  292.         mReturn = WinDefDlgProc (hWnd, msg, mp1, mp2);
  293.  
  294.     return (mReturn);
  295. }
  296.  
  297.