?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
quartz: Check hr so we are not incrementing a NULL (Coverity 106).
[wine/multimedia.git] / programs / cmd / directory.c
blob2082d6e7af3f019d97d9e205782810d95bdc59e9
1 /*
2 * CMD - Wine-compatible command line interface - Directory functions.
3 *
4 * Copyright (C) 1999 D A Pickles
5 * Copyright (C) 2007 J Edmeades
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20 */
21
22 /*
23 * NOTES:
24 * On entry, global variables quals, param1, param2 contain
25 * the qualifiers (uppercased and concatenated) and parameters entered, with
26 * environment-variable and batch parameter substitution already done.
27 */
28
29 #define WIN32_LEAN_AND_MEAN
30
31 #include "wcmd.h"
32 #include "wine/debug.h"
33
34 WINE_DEFAULT_DEBUG_CHANNEL(cmd);
35
36 int WCMD_dir_sort (const void *a, const void *b);
37 WCHAR * WCMD_filesize64 (ULONGLONG free);
38 WCHAR * WCMD_strrev (WCHAR *buff);
39 static void WCMD_getfileowner(WCHAR *filename, WCHAR *owner, int ownerlen);
40 static void WCMD_dir_trailer(WCHAR drive);
41
42 extern int echo_mode;
43 extern WCHAR quals[MAX_PATH], param1[MAX_PATH], param2[MAX_PATH];
44 extern DWORD errorlevel;
45
46 typedef enum _DISPLAYTIME
47 {
48 Creation = 0,
49 Access,
50 Written
51 } DISPLAYTIME;
52
53 typedef enum _DISPLAYORDER
54 {
55 Name = 0,
56 Extension,
57 Size,
58 Date
59 } DISPLAYORDER;
60
61 static DIRECTORY_STACK *WCMD_list_directory (DIRECTORY_STACK *parms, int level);
62 static int file_total, dir_total, recurse, wide, bare, max_width, lower;
63 static int shortname, usernames;
64 static ULONGLONG byte_total;
65 static DISPLAYTIME dirTime;
66 static DISPLAYORDER dirOrder;
67 static BOOL orderReverse, orderGroupDirs, orderGroupDirsReverse, orderByCol;
68 static BOOL separator;
69 static ULONG showattrs, attrsbits;
70
71 static const WCHAR dotW[] = {'.','\0'};
72 static const WCHAR dotdotW[] = {'.','.','\0'};
73 static const WCHAR starW[] = {'*','\0'};
74 static const WCHAR slashW[] = {'\\','\0'};
75 static const WCHAR emptyW[] = {'\0'};
76 static const WCHAR spaceW[] = {' ','\0'};
77
78 /*****************************************************************************
79 * WCMD_directory
80 *
81 * List a file directory.
82 *
83 */
84
85 void WCMD_directory (WCHAR *cmd) {
86
87 WCHAR path[MAX_PATH], cwd[MAX_PATH];
88 int status, paged_mode;
89 CONSOLE_SCREEN_BUFFER_INFO consoleInfo;
90 WCHAR *p;
91 WCHAR string[MAXSTRING];
92 int argno = 0;
93 int argsProcessed = 0;
94 WCHAR *argN = cmd;
95 WCHAR lastDrive;
96 BOOL trailerReqd = FALSE;
97 DIRECTORY_STACK *fullParms = NULL;
98 DIRECTORY_STACK *prevEntry = NULL;
99 DIRECTORY_STACK *thisEntry = NULL;
100 WCHAR drive[10];
101 WCHAR dir[MAX_PATH];
102 WCHAR fname[MAX_PATH];
103 WCHAR ext[MAX_PATH];
104 static const WCHAR dircmdW[] = {'D','I','R','C','M','D','\0'};
105
106 errorlevel = 0;
107
108 /* Prefill quals with (uppercased) DIRCMD env var */
109 if (GetEnvironmentVariable (dircmdW, string, sizeof(string)/sizeof(WCHAR))) {
110 p = string;
111 while ( (*p = toupper(*p)) ) ++p;
112 strcatW(string,quals);
113 strcpyW(quals, string);
114 }
115
116 byte_total = 0;
117 file_total = dir_total = 0;
118
119 /* Initialize all flags to their defaults as if no DIRCMD or quals */
120 paged_mode = FALSE;
121 recurse = FALSE;
122 wide = FALSE;
123 bare = FALSE;
124 lower = FALSE;
125 shortname = FALSE;
126 usernames = FALSE;
127 orderByCol = FALSE;
128 separator = TRUE;
129 dirTime = Written;
130 dirOrder = Name;
131 orderReverse = FALSE;
132 orderGroupDirs = FALSE;
133 orderGroupDirsReverse = FALSE;
134 showattrs = 0;
135 attrsbits = 0;
136
137 /* Handle args - Loop through so right most is the effective one */
138 /* Note: /- appears to be a negate rather than an off, eg. dir
139 /-W is wide, or dir /w /-w /-w is also wide */
140 p = quals;
141 while (*p && (*p=='/' || *p==' ')) {
142 BOOL negate = FALSE;
143 if (*p++==' ') continue; /* Skip / and blanks introduced through DIRCMD */
144
145 if (*p=='-') {
146 negate = TRUE;
147 p++;
148 }
149
150 WINE_TRACE("Processing arg '%c' (in %s)\n", *p, wine_dbgstr_w(quals));
151 switch (*p) {
152 case 'P': if (negate) paged_mode = !paged_mode;
153 else paged_mode = TRUE;
154 break;
155 case 'S': if (negate) recurse = !recurse;
156 else recurse = TRUE;
157 break;
158 case 'W': if (negate) wide = !wide;
159 else wide = TRUE;
160 break;
161 case 'B': if (negate) bare = !bare;
162 else bare = TRUE;
163 break;
164 case 'L': if (negate) lower = !lower;
165 else lower = TRUE;
166 break;
167 case 'X': if (negate) shortname = !shortname;
168 else shortname = TRUE;
169 break;
170 case 'Q': if (negate) usernames = !usernames;
171 else usernames = TRUE;
172 break;
173 case 'D': if (negate) orderByCol = !orderByCol;
174 else orderByCol = TRUE;
175 break;
176 case 'C': if (negate) separator = !separator;
177 else separator = TRUE;
178 break;
179 case 'T': p = p + 1;
180 if (*p==':') p++; /* Skip optional : */
181
182 if (*p == 'A') dirTime = Access;
183 else if (*p == 'C') dirTime = Creation;
184 else if (*p == 'W') dirTime = Written;
185
186 /* Support /T and /T: with no parms, default to written */
187 else if (*p == 0x00 || *p == '/') {
188 dirTime = Written;
189 p = p - 1; /* So when step on, move to '/' */
190 } else {
191 SetLastError(ERROR_INVALID_PARAMETER);
192 WCMD_print_error();
193 errorlevel = 1;
194 return;
195 }
196 break;
197 case 'O': p = p + 1;
198 if (*p==':') p++; /* Skip optional : */
199 while (*p && *p != '/') {
200 WINE_TRACE("Processing subparm '%c' (in %s)\n", *p, wine_dbgstr_w(quals));
201 switch (*p) {
202 case 'N': dirOrder = Name; break;
203 case 'E': dirOrder = Extension; break;
204 case 'S': dirOrder = Size; break;
205 case 'D': dirOrder = Date; break;
206 case '-': if (*(p+1)=='G') orderGroupDirsReverse=TRUE;
207 else orderReverse = TRUE;
208 break;
209 case 'G': orderGroupDirs = TRUE; break;
210 default:
211 SetLastError(ERROR_INVALID_PARAMETER);
212 WCMD_print_error();
213 errorlevel = 1;
214 return;
215 }
216 p++;
217 }
218 p = p - 1; /* So when step on, move to '/' */
219 break;
220 case 'A': p = p + 1;
221 showattrs = 0;
222 attrsbits = 0;
223 if (*p==':') p++; /* Skip optional : */
224 while (*p && *p != '/') {
225 BOOL anegate = FALSE;
226 ULONG mask;
227
228 /* Note /A: - options are 'offs' not toggles */
229 if (*p=='-') {
230 anegate = TRUE;
231 p++;
232 }
233
234 WINE_TRACE("Processing subparm '%c' (in %s)\n", *p, wine_dbgstr_w(quals));
235 switch (*p) {
236 case 'D': mask = FILE_ATTRIBUTE_DIRECTORY; break;
237 case 'H': mask = FILE_ATTRIBUTE_HIDDEN; break;
238 case 'S': mask = FILE_ATTRIBUTE_SYSTEM; break;
239 case 'R': mask = FILE_ATTRIBUTE_READONLY; break;
240 case 'A': mask = FILE_ATTRIBUTE_ARCHIVE; break;
241 default:
242 SetLastError(ERROR_INVALID_PARAMETER);
243 WCMD_print_error();
244 errorlevel = 1;
245 return;
246 }
247
248 /* Keep running list of bits we care about */
249 attrsbits |= mask;
250
251 /* Mask shows what MUST be in the bits we care about */
252 if (anegate) showattrs = showattrs & ~mask;
253 else showattrs |= mask;
254
255 p++;
256 }
257 p = p - 1; /* So when step on, move to '/' */
258 WINE_TRACE("Result: showattrs %x, bits %x\n", showattrs, attrsbits);
259 break;
260 default:
261 SetLastError(ERROR_INVALID_PARAMETER);
262 WCMD_print_error();
263 errorlevel = 1;
264 return;
265 }
266 p = p + 1;
267 }
268
269 /* Handle conflicting args and initialization */
270 if (bare || shortname) wide = FALSE;
271 if (bare) shortname = FALSE;
272 if (wide) usernames = FALSE;
273 if (orderByCol) wide = TRUE;
274
275 if (wide) {
276 if (GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &consoleInfo))
277 max_width = consoleInfo.dwSize.X;
278 else
279 max_width = 80;
280 }
281 if (paged_mode) {
282 WCMD_enter_paged_mode(NULL);
283 }
284
285 argno = 0;
286 argsProcessed = 0;
287 argN = cmd;
288 GetCurrentDirectory (MAX_PATH, cwd);
289 strcatW(cwd, slashW);
290
291 /* Loop through all args, calculating full effective directory */
292 fullParms = NULL;
293 prevEntry = NULL;
294 while (argN) {
295 WCHAR fullname[MAXSTRING];
296 WCHAR *thisArg = WCMD_parameter (cmd, argno++, &argN);
297 if (argN && argN[0] != '/') {
298
299 WINE_TRACE("Found parm '%s'\n", wine_dbgstr_w(thisArg));
300 if (thisArg[1] == ':' && thisArg[2] == '\\') {
301 strcpyW(fullname, thisArg);
302 } else if (thisArg[1] == ':' && thisArg[2] != '\\') {
303 WCHAR envvar[4];
304 static const WCHAR envFmt[] = {'=','%','c',':','\0'};
305 wsprintf(envvar, envFmt, thisArg[0]);
306 if (!GetEnvironmentVariable(envvar, fullname, MAX_PATH)) {
307 static const WCHAR noEnvFmt[] = {'%','c',':','\0'};
308 wsprintf(fullname, noEnvFmt, thisArg[0]);
309 }
310 strcatW(fullname, slashW);
311 strcatW(fullname, &thisArg[2]);
312 } else if (thisArg[0] == '\\') {
313 memcpy(fullname, cwd, 2 * sizeof(WCHAR));
314 strcpyW(fullname+2, thisArg);
315 } else {
316 strcpyW(fullname, cwd);
317 strcatW(fullname, thisArg);
318 }
319 WINE_TRACE("Using location '%s'\n", wine_dbgstr_w(fullname));
320
321 status = GetFullPathName (fullname, sizeof(path)/sizeof(WCHAR), path, NULL);
322
323 /*
324 * If the path supplied does not include a wildcard, and the endpoint of the
325 * path references a directory, we need to list the *contents* of that
326 * directory not the directory file itself.
327 */
328 if ((strchrW(path, '*') == NULL) && (strchrW(path, '%') == NULL)) {
329 status = GetFileAttributes (path);
330 if ((status != INVALID_FILE_ATTRIBUTES) && (status & FILE_ATTRIBUTE_DIRECTORY)) {
331 if (path[strlenW(path)-1] == '\\') {
332 strcatW (path, starW);
333 }
334 else {
335 const WCHAR slashStarW[] = {'\\','*','\0'};
336 strcatW (path, slashStarW);
337 }
338 }
339 } else {
340 /* Special case wildcard search with no extension (ie parameters ending in '.') as
341 GetFullPathName strips off the additional '.' */
342 if (fullname[strlenW(fullname)-1] == '.') strcatW(path, dotW);
343 }
344
345 WINE_TRACE("Using path '%s'\n", wine_dbgstr_w(path));
346 thisEntry = HeapAlloc(GetProcessHeap(),0,sizeof(DIRECTORY_STACK));
347 if (fullParms == NULL) fullParms = thisEntry;
348 if (prevEntry != NULL) prevEntry->next = thisEntry;
349 prevEntry = thisEntry;
350 thisEntry->next = NULL;
351
352 /* Split into components */
353 WCMD_splitpath(path, drive, dir, fname, ext);
354 WINE_TRACE("Path Parts: drive: '%s' dir: '%s' name: '%s' ext:'%s'\n",
355 wine_dbgstr_w(drive), wine_dbgstr_w(dir),
356 wine_dbgstr_w(fname), wine_dbgstr_w(ext));
357
358 thisEntry->dirName = HeapAlloc(GetProcessHeap(),0,
359 sizeof(WCHAR) * (strlenW(drive)+strlenW(dir)+1));
360 strcpyW(thisEntry->dirName, drive);
361 strcatW(thisEntry->dirName, dir);
362
363 thisEntry->fileName = HeapAlloc(GetProcessHeap(),0,
364 sizeof(WCHAR) * (strlenW(fname)+strlenW(ext)+1));
365 strcpyW(thisEntry->fileName, fname);
366 strcatW(thisEntry->fileName, ext);
367
368 }
369 }
370
371 /* If just 'dir' entered, a '*' parameter is assumed */
372 if (fullParms == NULL) {
373 WINE_TRACE("Inserting default '*'\n");
374 fullParms = HeapAlloc(GetProcessHeap(),0, sizeof(DIRECTORY_STACK));
375 fullParms->next = NULL;
376 fullParms->dirName = HeapAlloc(GetProcessHeap(),0,sizeof(WCHAR) * (strlenW(cwd)+1));
377 strcpyW(fullParms->dirName, cwd);
378 fullParms->fileName = HeapAlloc(GetProcessHeap(),0,sizeof(WCHAR) * 2);
379 strcpyW(fullParms->fileName, starW);
380 }
381
382 lastDrive = '?';
383 prevEntry = NULL;
384 thisEntry = fullParms;
385 trailerReqd = FALSE;
386
387 while (thisEntry != NULL) {
388
389 /* Output disk free (trailer) and volume information (header) if the drive
390 letter changes */
391 if (lastDrive != toupper(thisEntry->dirName[0])) {
392
393 /* Trailer Information */
394 if (lastDrive != '?') {
395 trailerReqd = FALSE;
396 WCMD_dir_trailer(prevEntry->dirName[0]);
397 }
398
399 lastDrive = toupper(thisEntry->dirName[0]);
400
401 if (!bare) {
402 WCHAR drive[3];
403
404 WINE_TRACE("Writing volume for '%c:'\n", thisEntry->dirName[0]);
405 memcpy(drive, thisEntry->dirName, 2 * sizeof(WCHAR));
406 drive[2] = 0x00;
407 status = WCMD_volume (0, drive);
408 trailerReqd = TRUE;
409 if (!status) {
410 errorlevel = 1;
411 goto exit;
412 }
413 }
414 } else {
415 static const WCHAR newLine2[] = {'\n','\n','\0'};
416 if (!bare) WCMD_output (newLine2);
417 }
418
419 /* Clear any errors from previous invocations, and process it */
420 errorlevel = 0;
421 prevEntry = thisEntry;
422 thisEntry = WCMD_list_directory (thisEntry, 0);
423 }
424
425 /* Trailer Information */
426 if (trailerReqd) {
427 WCMD_dir_trailer(prevEntry->dirName[0]);
428 }
429
430 exit:
431 if (paged_mode) WCMD_leave_paged_mode();
432
433 /* Free storage allocated for parms */
434 while (fullParms != NULL) {
435 prevEntry = fullParms;
436 fullParms = prevEntry->next;
437 HeapFree(GetProcessHeap(),0,prevEntry->dirName);
438 HeapFree(GetProcessHeap(),0,prevEntry->fileName);
439 HeapFree(GetProcessHeap(),0,prevEntry);
440 }
441 }
442
443 /*****************************************************************************
444 * WCMD_list_directory
445 *
446 * List a single file directory. This function (and those below it) can be called
447 * recursively when the /S switch is used.
448 *
449 * FIXME: Assumes 24-line display for the /P qualifier.
450 */
451
452 static DIRECTORY_STACK *WCMD_list_directory (DIRECTORY_STACK *inputparms, int level) {
453
454 WCHAR string[1024], datestring[32], timestring[32];
455 WCHAR real_path[MAX_PATH];
456 WIN32_FIND_DATA *fd;
457 FILETIME ft;
458 SYSTEMTIME st;
459 HANDLE hff;
460 int dir_count, file_count, entry_count, i, widest, cur_width, tmp_width;
461 int numCols, numRows;
462 int rows, cols;
463 ULARGE_INTEGER byte_count, file_size;
464 DIRECTORY_STACK *parms;
465 int concurrentDirs = 0;
466 BOOL done_header = FALSE;
467
468 static const WCHAR fmtDir[] = {'%','1','0','s',' ',' ','%','8','s',' ',' ',
469 '<','D','I','R','>',' ',' ',' ',' ',' ',' ',' ',' ',' ','\0'};
470 static const WCHAR fmtFile[] = {'%','1','0','s',' ',' ','%','8','s',' ',' ',
471 ' ',' ','%','1','0','s',' ',' ','\0'};
472 static const WCHAR fmt2[] = {'%','-','1','3','s','\0'};
473 static const WCHAR fmt3[] = {'%','-','2','3','s','\0'};
474 static const WCHAR fmt4[] = {'%','s','\0'};
475 static const WCHAR fmt5[] = {'%','s','%','s','\0'};
476
477 dir_count = 0;
478 file_count = 0;
479 entry_count = 0;
480 byte_count.QuadPart = 0;
481 widest = 0;
482 cur_width = 0;
483
484 /* Loop merging all the files from consecutive parms which relate to the
485 same directory. Note issuing a directory header with no contents
486 mirrors what windows does */
487 parms = inputparms;
488 fd = HeapAlloc(GetProcessHeap(),0,sizeof(WIN32_FIND_DATA));
489 while (parms && strcmpW(inputparms->dirName, parms->dirName) == 0) {
490 concurrentDirs++;
491
492 /* Work out the full path + filename */
493 strcpyW(real_path, parms->dirName);
494 strcatW(real_path, parms->fileName);
495
496 /* Load all files into an in memory structure */
497 WINE_TRACE("Looking for matches to '%s'\n", wine_dbgstr_w(real_path));
498 hff = FindFirstFile (real_path, (fd+entry_count));
499 if (hff != INVALID_HANDLE_VALUE) {
500 do {
501 /* Skip any which are filtered out by attribute */
502 if (((fd+entry_count)->dwFileAttributes & attrsbits) != showattrs) continue;
503
504 entry_count++;
505
506 /* Keep running track of longest filename for wide output */
507 if (wide || orderByCol) {
508 int tmpLen = strlenW((fd+(entry_count-1))->cFileName) + 3;
509 if ((fd+(entry_count-1))->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) tmpLen = tmpLen + 2;
510 if (tmpLen > widest) widest = tmpLen;
511 }
512
513 fd = HeapReAlloc(GetProcessHeap(),0,fd,(entry_count+1)*sizeof(WIN32_FIND_DATA));
514 if (fd == NULL) {
515 FindClose (hff);
516 WINE_ERR("Out of memory\n");
517 errorlevel = 1;
518 return parms->next;
519 }
520 } while (FindNextFile(hff, (fd+entry_count)) != 0);
521 FindClose (hff);
522 }
523
524 /* Work out the actual current directory name without a trailing \ */
525 strcpyW(real_path, parms->dirName);
526 real_path[strlenW(parms->dirName)-1] = 0x00;
527
528 /* Output the results */
529 if (!bare) {
530 if (level != 0 && (entry_count > 0)) WCMD_output (newline);
531 if (!recurse || ((entry_count > 0) && done_header==FALSE)) {
532 static const WCHAR headerW[] = {'D','i','r','e','c','t','o','r','y',' ','o','f',
533 ' ','%','s','\n','\n','\0'};
534 WCMD_output (headerW, real_path);
535 done_header = TRUE;
536 }
537 }
538
539 /* Move to next parm */
540 parms = parms->next;
541 }
542
543 /* Handle case where everything is filtered out */
544 if (entry_count > 0) {
545
546 /* Sort the list of files */
547 qsort (fd, entry_count, sizeof(WIN32_FIND_DATA), WCMD_dir_sort);
548
549 /* Work out the number of columns */
550 WINE_TRACE("%d entries, maxwidth=%d, widest=%d\n", entry_count, max_width, widest);
551 if (wide || orderByCol) {
552 numCols = max(1, (int)max_width / widest);
553 numRows = entry_count / numCols;
554 if (entry_count % numCols) numRows++;
555 } else {
556 numCols = 1;
557 numRows = entry_count;
558 }
559 WINE_TRACE("cols=%d, rows=%d\n", numCols, numRows);
560
561 for (rows=0; rows<numRows; rows++) {
562 BOOL addNewLine = TRUE;
563 for (cols=0; cols<numCols; cols++) {
564 WCHAR username[24];
565
566 /* Work out the index of the entry being pointed to */
567 if (orderByCol) {
568 i = (cols * numRows) + rows;
569 if (i >= entry_count) continue;
570 } else {
571 i = (rows * numCols) + cols;
572 if (i >= entry_count) continue;
573 }
574
575 /* /L convers all names to lower case */
576 if (lower) {
577 WCHAR *p = (fd+i)->cFileName;
578 while ( (*p = tolower(*p)) ) ++p;
579 }
580
581 /* /Q gets file ownership information */
582 if (usernames) {
583 strcpyW (string, inputparms->dirName);
584 strcatW (string, (fd+i)->cFileName);
585 WCMD_getfileowner(string, username, sizeof(username)/sizeof(WCHAR));
586 }
587
588 if (dirTime == Written) {
589 FileTimeToLocalFileTime (&(fd+i)->ftLastWriteTime, &ft);
590 } else if (dirTime == Access) {
591 FileTimeToLocalFileTime (&(fd+i)->ftLastAccessTime, &ft);
592 } else {
593 FileTimeToLocalFileTime (&(fd+i)->ftCreationTime, &ft);
594 }
595 FileTimeToSystemTime (&ft, &st);
596 GetDateFormat (0, DATE_SHORTDATE, &st, NULL, datestring,
597 sizeof(datestring)/sizeof(WCHAR));
598 GetTimeFormat (0, TIME_NOSECONDS, &st,
599 NULL, timestring, sizeof(timestring)/sizeof(WCHAR));
600
601 if (wide) {
602
603 tmp_width = cur_width;
604 if ((fd+i)->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
605 static const WCHAR fmt[] = {'[','%','s',']','\0'};
606 WCMD_output (fmt, (fd+i)->cFileName);
607 dir_count++;
608 tmp_width = tmp_width + strlenW((fd+i)->cFileName) + 2;
609 } else {
610 static const WCHAR fmt[] = {'%','s','\0'};
611 WCMD_output (fmt, (fd+i)->cFileName);
612 tmp_width = tmp_width + strlenW((fd+i)->cFileName) ;
613 file_count++;
614 file_size.u.LowPart = (fd+i)->nFileSizeLow;
615 file_size.u.HighPart = (fd+i)->nFileSizeHigh;
616 byte_count.QuadPart += file_size.QuadPart;
617 }
618 cur_width = cur_width + widest;
619
620 if ((cur_width + widest) > max_width) {
621 cur_width = 0;
622 } else {
623 int padding = cur_width - tmp_width;
624 int toWrite = 0;
625 WCHAR temp[101];
626
627 /* Note: WCMD_output uses wvsprintf which does not allow %*
628 so manually pad with spaces to appropriate width */
629 strcpyW(temp, emptyW);
630 while (padding > 0) {
631 strcatW(&temp[toWrite], spaceW);
632 toWrite++;
633 if (toWrite > 99) {
634 WCMD_output(temp);
635 toWrite = 0;
636 strcpyW(temp, emptyW);
637 }
638 padding--;
639 }
640 WCMD_output(temp);
641 }
642
643 } else if ((fd+i)->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
644 dir_count++;
645
646 if (!bare) {
647 WCMD_output (fmtDir, datestring, timestring);
648 if (shortname) WCMD_output (fmt2, (fd+i)->cAlternateFileName);
649 if (usernames) WCMD_output (fmt3, username);
650 WCMD_output(fmt4,(fd+i)->cFileName);
651 } else {
652 if (!((strcmpW((fd+i)->cFileName, dotW) == 0) ||
653 (strcmpW((fd+i)->cFileName, dotdotW) == 0))) {
654 WCMD_output (fmt5, recurse?inputparms->dirName:emptyW, (fd+i)->cFileName);
655 } else {
656 addNewLine = FALSE;
657 }
658 }
659 }
660 else {
661 file_count++;
662 file_size.u.LowPart = (fd+i)->nFileSizeLow;
663 file_size.u.HighPart = (fd+i)->nFileSizeHigh;
664 byte_count.QuadPart += file_size.QuadPart;
665 if (!bare) {
666 WCMD_output (fmtFile, datestring, timestring,
667 WCMD_filesize64(file_size.QuadPart));
668 if (shortname) WCMD_output (fmt2, (fd+i)->cAlternateFileName);
669 if (usernames) WCMD_output (fmt3, username);
670 WCMD_output(fmt4,(fd+i)->cFileName);
671 } else {
672 WCMD_output (fmt5, recurse?inputparms->dirName:emptyW, (fd+i)->cFileName);
673 }
674 }
675 }
676 if (addNewLine) WCMD_output (newline);
677 cur_width = 0;
678 }
679
680 if (!bare) {
681 if (file_count == 1) {
682 static const WCHAR fmt[] = {' ',' ',' ',' ',' ',' ',' ','1',' ','f','i','l','e',' ',
683 '%','2','5','s',' ','b','y','t','e','s','\n','\0'};
684 WCMD_output (fmt, WCMD_filesize64 (byte_count.QuadPart));
685 }
686 else {
687 static const WCHAR fmt[] = {'%','8','d',' ','f','i','l','e','s',' ','%','2','4','s',
688 ' ','b','y','t','e','s','\n','\0'};
689 WCMD_output (fmt, file_count, WCMD_filesize64 (byte_count.QuadPart));
690 }
691 }
692 byte_total = byte_total + byte_count.QuadPart;
693 file_total = file_total + file_count;
694 dir_total = dir_total + dir_count;
695
696 if (!bare && !recurse) {
697 if (dir_count == 1) {
698 static const WCHAR fmt[] = {'%','8','d',' ','d','i','r','e','c','t','o','r','y',
699 ' ',' ',' ',' ',' ',' ',' ',' ',' ','\0'};
700 WCMD_output (fmt, 1);
701 } else {
702 static const WCHAR fmt[] = {'%','8','d',' ','d','i','r','e','c','t','o','r','i',
703 'e','s','\0'};
704 WCMD_output (fmt, dir_count);
705 }
706 }
707 }
708 HeapFree(GetProcessHeap(),0,fd);
709
710 /* When recursing, look in all subdirectories for matches */
711 if (recurse) {
712 DIRECTORY_STACK *dirStack = NULL;
713 DIRECTORY_STACK *lastEntry = NULL;
714 WIN32_FIND_DATA finddata;
715
716 /* Build path to search */
717 strcpyW(string, inputparms->dirName);
718 strcatW(string, starW);
719
720 WINE_TRACE("Recursive, looking for '%s'\n", wine_dbgstr_w(string));
721 hff = FindFirstFile (string, &finddata);
722 if (hff != INVALID_HANDLE_VALUE) {
723 do {
724 if ((finddata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) &&
725 (strcmpW(finddata.cFileName, dotdotW) != 0) &&
726 (strcmpW(finddata.cFileName, dotW) != 0)) {
727
728 DIRECTORY_STACK *thisDir;
729 int dirsToCopy = concurrentDirs;
730
731 /* Loop creating list of subdirs for all concurrent entries */
732 parms = inputparms;
733 while (dirsToCopy > 0) {
734 dirsToCopy--;
735
736 /* Work out search parameter in sub dir */
737 strcpyW (string, inputparms->dirName);
738 strcatW (string, finddata.cFileName);
739 strcatW (string, slashW);
740 WINE_TRACE("Recursive, Adding to search list '%s'\n", wine_dbgstr_w(string));
741
742 /* Allocate memory, add to list */
743 thisDir = HeapAlloc(GetProcessHeap(),0,sizeof(DIRECTORY_STACK));
744 if (dirStack == NULL) dirStack = thisDir;
745 if (lastEntry != NULL) lastEntry->next = thisDir;
746 lastEntry = thisDir;
747 thisDir->next = NULL;
748 thisDir->dirName = HeapAlloc(GetProcessHeap(),0,
749 sizeof(WCHAR) * (strlenW(string)+1));
750 strcpyW(thisDir->dirName, string);
751 thisDir->fileName = HeapAlloc(GetProcessHeap(),0,
752 sizeof(WCHAR) * (strlenW(parms->fileName)+1));
753 strcpyW(thisDir->fileName, parms->fileName);
754 parms = parms->next;
755 }
756 }
757 } while (FindNextFile(hff, &finddata) != 0);
758 FindClose (hff);
759
760 while (dirStack != NULL) {
761 DIRECTORY_STACK *thisDir = dirStack;
762 dirStack = WCMD_list_directory (thisDir, 1);
763 while (thisDir != dirStack) {
764 DIRECTORY_STACK *tempDir = thisDir->next;
765 HeapFree(GetProcessHeap(),0,thisDir->dirName);
766 HeapFree(GetProcessHeap(),0,thisDir->fileName);
767 HeapFree(GetProcessHeap(),0,thisDir);
768 thisDir = tempDir;
769 }
770 }
771 }
772 }
773
774 /* Handle case where everything is filtered out */
775 if ((file_total + dir_total == 0) && (level == 0)) {
776 SetLastError (ERROR_FILE_NOT_FOUND);
777 WCMD_print_error ();
778 errorlevel = 1;
779 }
780
781 return parms;
782 }
783
784 /*****************************************************************************
785 * WCMD_filesize64
786 *
787 * Convert a 64-bit number into a WCHARacter string, with commas every three digits.
788 * Result is returned in a static string overwritten with each call.
789 * FIXME: There must be a better algorithm!
790 */
791
792 WCHAR * WCMD_filesize64 (ULONGLONG n) {
793
794 ULONGLONG q;
795 unsigned int r, i;
796 WCHAR *p;
797 static WCHAR buff[32];
798
799 p = buff;
800 i = -3;
801 do {
802 if (separator && ((++i)%3 == 1)) *p++ = ',';
803 q = n / 10;
804 r = n - (q * 10);
805 *p++ = r + '0';
806 *p = '\0';
807 n = q;
808 } while (n != 0);
809 WCMD_strrev (buff);
810 return buff;
811 }
812
813 /*****************************************************************************
814 * WCMD_strrev
815 *
816 * Reverse a WCHARacter string in-place (strrev() is not available under unixen :-( ).
817 */
818
819 WCHAR * WCMD_strrev (WCHAR *buff) {
820
821 int r, i;
822 WCHAR b;
823
824 r = strlenW (buff);
825 for (i=0; i<r/2; i++) {
826 b = buff[i];
827 buff[i] = buff[r-i-1];
828 buff[r-i-1] = b;
829 }
830 return (buff);
831 }
832
833
834 /*****************************************************************************
835 * WCMD_dir_sort
836 *
837 * Sort based on the /O options supplied on the command line
838 */
839 int WCMD_dir_sort (const void *a, const void *b)
840 {
841 WIN32_FIND_DATA *filea = (WIN32_FIND_DATA *)a;
842 WIN32_FIND_DATA *fileb = (WIN32_FIND_DATA *)b;
843 int result = 0;
844
845 /* If /OG or /O-G supplied, dirs go at the top or bottom, ignoring the
846 requested sort order for the directory components */
847 if (orderGroupDirs &&
848 ((filea->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) ||
849 (fileb->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)))
850 {
851 BOOL aDir = filea->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
852 if (aDir) result = -1;
853 else result = 1;
854 if (orderGroupDirsReverse) result = -result;
855 return result;
856
857 /* Order by Name: */
858 } else if (dirOrder == Name) {
859 result = lstrcmpiW(filea->cFileName, fileb->cFileName);
860
861 /* Order by Size: */
862 } else if (dirOrder == Size) {
863 ULONG64 sizea = (((ULONG64)filea->nFileSizeHigh) << 32) + filea->nFileSizeLow;
864 ULONG64 sizeb = (((ULONG64)fileb->nFileSizeHigh) << 32) + fileb->nFileSizeLow;
865 if( sizea < sizeb ) result = -1;
866 else if( sizea == sizeb ) result = 0;
867 else result = 1;
868
869 /* Order by Date: (Takes into account which date (/T option) */
870 } else if (dirOrder == Date) {
871
872 FILETIME *ft;
873 ULONG64 timea, timeb;
874
875 if (dirTime == Written) {
876 ft = &filea->ftLastWriteTime;
877 timea = (((ULONG64)ft->dwHighDateTime) << 32) + ft->dwLowDateTime;
878 ft = &fileb->ftLastWriteTime;
879 timeb = (((ULONG64)ft->dwHighDateTime) << 32) + ft->dwLowDateTime;
880 } else if (dirTime == Access) {
881 ft = &filea->ftLastAccessTime;
882 timea = (((ULONG64)ft->dwHighDateTime) << 32) + ft->dwLowDateTime;
883 ft = &fileb->ftLastAccessTime;
884 timeb = (((ULONG64)ft->dwHighDateTime) << 32) + ft->dwLowDateTime;
885 } else {
886 ft = &filea->ftCreationTime;
887 timea = (((ULONG64)ft->dwHighDateTime) << 32) + ft->dwLowDateTime;
888 ft = &fileb->ftCreationTime;
889 timeb = (((ULONG64)ft->dwHighDateTime) << 32) + ft->dwLowDateTime;
890 }
891 if( timea < timeb ) result = -1;
892 else if( timea == timeb ) result = 0;
893 else result = 1;
894
895 /* Order by Extension: (Takes into account which date (/T option) */
896 } else if (dirOrder == Extension) {
897 WCHAR drive[10];
898 WCHAR dir[MAX_PATH];
899 WCHAR fname[MAX_PATH];
900 WCHAR extA[MAX_PATH];
901 WCHAR extB[MAX_PATH];
902
903 /* Split into components */
904 WCMD_splitpath(filea->cFileName, drive, dir, fname, extA);
905 WCMD_splitpath(fileb->cFileName, drive, dir, fname, extB);
906 result = lstrcmpiW(extA, extB);
907 }
908
909 if (orderReverse) result = -result;
910 return result;
911 }
912
913 /*****************************************************************************
914 * WCMD_getfileowner
915 *
916 * Reverse a WCHARacter string in-place (strrev() is not available under unixen :-( ).
917 */
918 void WCMD_getfileowner(WCHAR *filename, WCHAR *owner, int ownerlen) {
919
920 ULONG sizeNeeded = 0;
921 DWORD rc;
922 WCHAR name[MAXSTRING];
923 WCHAR domain[MAXSTRING];
924
925 /* In case of error, return empty string */
926 *owner = 0x00;
927
928 /* Find out how much space we need for the owner security descriptor */
929 GetFileSecurity(filename, OWNER_SECURITY_INFORMATION, 0, 0, &sizeNeeded);
930 rc = GetLastError();
931
932 if(rc == ERROR_INSUFFICIENT_BUFFER && sizeNeeded > 0) {
933
934 LPBYTE secBuffer;
935 PSID pSID = NULL;
936 BOOL defaulted = FALSE;
937 ULONG nameLen = MAXSTRING;
938 ULONG domainLen = MAXSTRING;
939 SID_NAME_USE nameuse;
940
941 secBuffer = (LPBYTE) HeapAlloc(GetProcessHeap(),0,sizeNeeded * sizeof(BYTE));
942 if(!secBuffer) return;
943
944 /* Get the owners security descriptor */
945 if(!GetFileSecurity(filename, OWNER_SECURITY_INFORMATION, secBuffer,
946 sizeNeeded, &sizeNeeded)) {
947 HeapFree(GetProcessHeap(),0,secBuffer);
948 return;
949 }
950
951 /* Get the SID from the SD */
952 if(!GetSecurityDescriptorOwner(secBuffer, &pSID, &defaulted)) {
953 HeapFree(GetProcessHeap(),0,secBuffer);
954 return;
955 }
956
957 /* Convert to a username */
958 if (LookupAccountSid(NULL, pSID, name, &nameLen, domain, &domainLen, &nameuse)) {
959 static const WCHAR fmt[] = {'%','s','%','c','%','s','\0'};
960 snprintfW(owner, ownerlen, fmt, domain, '\\', name);
961 }
962 HeapFree(GetProcessHeap(),0,secBuffer);
963 }
964 return;
965 }
966
967 /*****************************************************************************
968 * WCMD_dir_trailer
969 *
970 * Print out the trailer for the supplied drive letter
971 */
972 static void WCMD_dir_trailer(WCHAR drive) {
973 ULARGE_INTEGER avail, total, freebytes;
974 DWORD status;
975 WCHAR driveName[4] = {'c',':','\\','\0'};
976
977 driveName[0] = drive;
978 status = GetDiskFreeSpaceEx (driveName, &avail, &total, &freebytes);
979 WINE_TRACE("Writing trailer for '%s' gave %d(%d)\n", wine_dbgstr_w(driveName),
980 status, GetLastError());
981
982 if (errorlevel==0 && !bare) {
983 if (recurse) {
984 static const WCHAR fmt1[] = {'\n',' ',' ',' ',' ',' ','T','o','t','a','l',' ','f','i','l','e','s',
985 ' ','l','i','s','t','e','d',':','\n','%','8','d',' ','f','i','l','e',
986 's','%','2','5','s',' ','b','y','t','e','s','\n','\0'};
987 static const WCHAR fmt2[] = {'%','8','d',' ','d','i','r','e','c','t','o','r','i','e','s',' ','%',
988 '1','8','s',' ','b','y','t','e','s',' ','f','r','e','e','\n','\n',
989 '\0'};
990 WCMD_output (fmt1, file_total, WCMD_filesize64 (byte_total));
991 WCMD_output (fmt2, dir_total, WCMD_filesize64 (freebytes.QuadPart));
992 } else {
993 static const WCHAR fmt[] = {' ','%','1','8','s',' ','b','y','t','e','s',' ','f','r','e','e',
994 '\n','\n','\0'};
995 WCMD_output (fmt, WCMD_filesize64 (freebytes.QuadPart));
996 }
997 }
998 }
Maarten's multimedia branch