programs/wcmd/directory.c

   1 /*
   2  * WCMD - Wine-compatible command line interface - Directory functions.
   3  *
   4  * Copyright (C) 1999 D A Pickles
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20
  21 /*
  22  * NOTES:
  23  * On entry, global variables quals, param1, param2 contain
  24  * the qualifiers (uppercased and concatenated) and parameters entered, with
  25  * environment-variable and batch parameter substitution already done.
  26  */
  27
  28 #define NONAMELESSUNION
  29 #define NONAMELESSSTRUCT
  30 #include "wcmd.h"
  31
  32 int WCMD_dir_sort (const void *a, const void *b);
  33 void WCMD_list_directory (char *path, int level);
  34 char * WCMD_filesize64 (ULONGLONG free);
  35 char * WCMD_strrev (char *buff);
  36
  37
  38 extern char nyi[];
  39 extern char newline[];
  40 extern char version_string[];
  41 extern char anykey[];
  42 extern int echo_mode;
  43 extern char quals[MAX_PATH], param1[MAX_PATH], param2[MAX_PATH];
  44 extern DWORD errorlevel;
  45
  46 static int file_total, dir_total, recurse, wide, bare, max_width;
  47 static ULONGLONG byte_total;
  48
  49 /*****************************************************************************
  50  * WCMD_directory
  51  *
  52  * List a file directory.
  53  *
  54  */
  55
  56 void WCMD_directory (void) {
  57
  58 char path[MAX_PATH], drive[8];
  59 int status, paged_mode;
  60 ULARGE_INTEGER avail, total, free;
  61 CONSOLE_SCREEN_BUFFER_INFO consoleInfo;
  62
  63   byte_total = 0;
  64   file_total = dir_total = 0;
  65
  66   /* Handle args */
  67   paged_mode = (strstr(quals, "/P") != NULL);
  68   recurse = (strstr(quals, "/S") != NULL);
  69   wide    = (strstr(quals, "/W") != NULL);
  70   bare    = (strstr(quals, "/B") != NULL);
  71
  72   /* Handle conflicting args and initialization */
  73   if (bare) wide = FALSE;
  74
  75   if (wide) {
  76       if (GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &consoleInfo))
  77           max_width = consoleInfo.dwSize.X;
  78       else
  79           max_width = 80;
  80   }
  81   if (paged_mode) {
  82      WCMD_enter_paged_mode();
  83   }
  84
  85   if (param1[0] == '\0') strcpy (param1, ".");
  86   status = GetFullPathName (param1, sizeof(path), path, NULL);
  87   if (!status) {
  88     WCMD_print_error();
  89     if (paged_mode) WCMD_leave_paged_mode();
  90     return;
  91   }
  92   lstrcpyn (drive, path, 3);
  93
  94   if (!bare) {
  95      status = WCMD_volume (0, drive);
  96      if (!status) {
  97          if (paged_mode) WCMD_leave_paged_mode();
  98        return;
  99      }
 100   }
 101
 102   WCMD_list_directory (path, 0);
 103   lstrcpyn (drive, path, 4);
 104   GetDiskFreeSpaceEx (drive, &avail, &total, &free);
 105
 106   if (!bare) {
 107      if (recurse) {
 108        WCMD_output ("\n\n     Total files listed:\n%8d files%25s bytes\n",
 109             file_total, WCMD_filesize64 (byte_total));
 110        WCMD_output ("%8d directories %18s bytes free\n\n",
 111             dir_total, WCMD_filesize64 (free.QuadPart));
 112      } else {
 113        WCMD_output (" %18s bytes free\n\n", WCMD_filesize64 (free.QuadPart));
 114      }
 115   }
 116   if (paged_mode) WCMD_leave_paged_mode();
 117 }
 118
 119 /*****************************************************************************
 120  * WCMD_list_directory
 121  *
 122  * List a single file directory. This function (and those below it) can be called
 123  * recursively when the /S switch is used.
 124  *
 125  * FIXME: Entries sorted by name only. Should we support DIRCMD??
 126  * FIXME: Assumes 24-line display for the /P qualifier.
 127  * FIXME: Other command qualifiers not supported.
 128  * FIXME: DIR /S FILENAME fails if at least one matching file is not found in the top level.
 129  */
 130
 131 void WCMD_list_directory (char *search_path, int level) {
 132
 133 char string[1024], datestring[32], timestring[32];
 134 char mem_err[] = "Memory Allocation Error";
 135 char *p;
 136 char real_path[MAX_PATH];
 137 WIN32_FIND_DATA *fd;
 138 FILETIME ft;
 139 SYSTEMTIME st;
 140 HANDLE hff;
 141 int status, dir_count, file_count, entry_count, i, widest, cur_width, tmp_width;
 142 ULARGE_INTEGER byte_count, file_size;
 143
 144   dir_count = 0;
 145   file_count = 0;
 146   entry_count = 0;
 147   byte_count.QuadPart = 0;
 148   widest = 0;
 149   cur_width = 0;
 150
 151 /*
 152  *  If the path supplied does not include a wildcard, and the endpoint of the
 153  *  path references a directory, we need to list the *contents* of that
 154  *  directory not the directory file itself.
 155  */
 156
 157   if ((strchr(search_path, '*') == NULL) && (strchr(search_path, '%') == NULL)) {
 158     status = GetFileAttributes (search_path);
 159     if ((status != INVALID_FILE_ATTRIBUTES) && (status & FILE_ATTRIBUTE_DIRECTORY)) {
 160       if (search_path[strlen(search_path)-1] == '\\') {
 161         strcat (search_path, "*");
 162       }
 163       else {
 164         strcat (search_path, "\\*");
 165       }
 166     }
 167   }
 168
 169   /* Work out the actual current directory name */
 170   p = strrchr (search_path, '\\');
 171   memset(real_path, 0x00, sizeof(real_path));
 172   lstrcpyn (real_path, search_path, (p-search_path+2));
 173
 174   /* Load all files into an in memory structure */
 175   fd = malloc (sizeof(WIN32_FIND_DATA));
 176   hff = FindFirstFile (search_path, fd);
 177   if (hff == INVALID_HANDLE_VALUE) {
 178     SetLastError (ERROR_FILE_NOT_FOUND);
 179     WCMD_print_error ();
 180     free (fd);
 181     return;
 182   }
 183   do {
 184     entry_count++;
 185
 186     /* Keep running track of longest filename for wide output */
 187     if (wide) {
 188        int tmpLen = strlen((fd+(entry_count-1))->cFileName) + 3;
 189        if ((fd+(entry_count-1))->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) tmpLen = tmpLen + 2;
 190        if (tmpLen > widest) widest = tmpLen;
 191     }
 192
 193     fd = realloc (fd, (entry_count+1)*sizeof(WIN32_FIND_DATA));
 194     if (fd == NULL) {
 195       FindClose (hff);
 196       WCMD_output (mem_err);
 197        return;
 198     }
 199   } while (FindNextFile(hff, (fd+entry_count)) != 0);
 200   FindClose (hff);
 201
 202   /* Sort the list of files */
 203   qsort (fd, entry_count, sizeof(WIN32_FIND_DATA), WCMD_dir_sort);
 204
 205   /* Output the results */
 206   if (!bare) {
 207      if (level != 0) WCMD_output ("\n\n");
 208      WCMD_output ("Directory of %s\n\n", real_path);
 209   }
 210
 211   for (i=0; i<entry_count; i++) {
 212     FileTimeToLocalFileTime (&(fd+i)->ftLastWriteTime, &ft);
 213     FileTimeToSystemTime (&ft, &st);
 214     GetDateFormat (0, DATE_SHORTDATE, &st, NULL, datestring,
 215                 sizeof(datestring));
 216     GetTimeFormat (0, TIME_NOSECONDS, &st,
 217                 NULL, timestring, sizeof(timestring));
 218
 219     if (wide) {
 220
 221       tmp_width = cur_width;
 222       if ((fd+i)->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
 223           WCMD_output ("[%s]", (fd+i)->cFileName);
 224           dir_count++;
 225           tmp_width = tmp_width + strlen((fd+i)->cFileName) + 2;
 226       } else {
 227           WCMD_output ("%s", (fd+i)->cFileName);
 228           tmp_width = tmp_width + strlen((fd+i)->cFileName) ;
 229           file_count++;
 230 #ifndef NONAMELESSSTRUCT
 231           file_size.LowPart = (fd+i)->nFileSizeLow;
 232           file_size.HighPart = (fd+i)->nFileSizeHigh;
 233 #else
 234           file_size.u.LowPart = (fd+i)->nFileSizeLow;
 235           file_size.u.HighPart = (fd+i)->nFileSizeHigh;
 236 #endif
 237       byte_count.QuadPart += file_size.QuadPart;
 238       }
 239       cur_width = cur_width + widest;
 240
 241       if ((cur_width + widest) > max_width) {
 242           WCMD_output ("\n");
 243           cur_width = 0;
 244       } else {
 245           WCMD_output ("%*.s", (tmp_width - cur_width) ,"");
 246       }
 247
 248     } else if ((fd+i)->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
 249       dir_count++;
 250
 251       if (!bare) {
 252          WCMD_output ("%10s  %8s  <DIR>         %s\n",
 253              datestring, timestring, (fd+i)->cFileName);
 254       } else {
 255          if (!((strcmp((fd+i)->cFileName, ".") == 0) ||
 256                (strcmp((fd+i)->cFileName, "..") == 0))) {
 257             WCMD_output ("%s%s\n", recurse?real_path:"", (fd+i)->cFileName);
 258          }
 259       }
 260     }
 261     else {
 262       file_count++;
 263 #ifndef NONAMELESSSTRUCT
 264       file_size.LowPart = (fd+i)->nFileSizeLow;
 265       file_size.HighPart = (fd+i)->nFileSizeHigh;
 266 #else
 267       file_size.u.LowPart = (fd+i)->nFileSizeLow;
 268       file_size.u.HighPart = (fd+i)->nFileSizeHigh;
 269 #endif
 270       byte_count.QuadPart += file_size.QuadPart;
 271           if (!bare) {
 272          WCMD_output ("%10s  %8s    %10s  %s\n",
 273              datestring, timestring,
 274                  WCMD_filesize64(file_size.QuadPart), (fd+i)->cFileName);
 275       } else {
 276          WCMD_output ("%s%s\n", recurse?real_path:"", (fd+i)->cFileName);
 277       }
 278     }
 279   }
 280
 281   if (wide && cur_width>0) {
 282       WCMD_output ("\n");
 283   }
 284
 285   if (!bare) {
 286      if (file_count == 1) {
 287        WCMD_output ("       1 file %25s bytes\n", WCMD_filesize64 (byte_count.QuadPart));
 288      }
 289      else {
 290        WCMD_output ("%8d files %24s bytes\n", file_count, WCMD_filesize64 (byte_count.QuadPart));
 291      }
 292   }
 293   byte_total = byte_total + byte_count.QuadPart;
 294   file_total = file_total + file_count;
 295   dir_total = dir_total + dir_count;
 296
 297   if (!bare) {
 298      if (dir_count == 1) WCMD_output ("1 directory         ");
 299      else WCMD_output ("%8d directories", dir_count);
 300   }
 301   for (i=0; i<entry_count; i++) {
 302     if ((recurse) &&
 303           ((fd+i)->cFileName[0] != '.') &&
 304           ((fd+i)->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
 305 #if 0
 306       GetFullPathName ((fd+i)->cFileName, sizeof(string), string, NULL);
 307 #endif
 308       p = strrchr (search_path, '\\');
 309       lstrcpyn (string, search_path, (p-search_path+2));
 310       lstrcat (string, (fd+i)->cFileName);
 311       lstrcat (string, p);
 312       WCMD_list_directory (string, 1);
 313     }
 314   }
 315   free (fd);
 316   return;
 317 }
 318
 319 /*****************************************************************************
 320  * WCMD_filesize64
 321  *
 322  * Convert a 64-bit number into a character string, with commas every three digits.
 323  * Result is returned in a static string overwritten with each call.
 324  * FIXME: There must be a better algorithm!
 325  */
 326
 327 char * WCMD_filesize64 (ULONGLONG n) {
 328
 329 ULONGLONG q;
 330 unsigned int r, i;
 331 char *p;
 332 static char buff[32];
 333
 334   p = buff;
 335   i = -3;
 336   do {
 337     if ((++i)%3 == 1) *p++ = ',';
 338     q = n / 10;
 339     r = n - (q * 10);
 340     *p++ = r + '0';
 341     *p = '\0';
 342     n = q;
 343   } while (n != 0);
 344   WCMD_strrev (buff);
 345   return buff;
 346 }
 347
 348 /*****************************************************************************
 349  * WCMD_strrev
 350  *
 351  * Reverse a character string in-place (strrev() is not available under unixen :-( ).
 352  */
 353
 354 char * WCMD_strrev (char *buff) {
 355
 356 int r, i;
 357 char b;
 358
 359   r = lstrlen (buff);
 360   for (i=0; i<r/2; i++) {
 361     b = buff[i];
 362     buff[i] = buff[r-i-1];
 363     buff[r-i-1] = b;
 364   }
 365   return (buff);
 366 }
 367
 368
 369 int WCMD_dir_sort (const void *a, const void *b) {
 370
 371   return (lstrcmpi(((WIN32_FIND_DATA *)a)->cFileName,
 372         ((WIN32_FIND_DATA *)b)->cFileName));
 373 }