| /* | |
| * CMD - Wine-compatible command line interface - Directory functions. | |
| * | |
| * Copyright (C) 1999 D A Pickles | |
| * Copyright (C) 2007 J Edmeades | |
| * | |
| * This library is free software; you can redistribute it and/or | |
| * modify it under the terms of the GNU Lesser General Public | |
| * License as published by the Free Software Foundation; either | |
| * version 2.1 of the License, or (at your option) any later version. | |
| * | |
| * This library is distributed in the hope that it will be useful, | |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
| * Lesser General Public License for more details. | |
| * | |
| * You should have received a copy of the GNU Lesser General Public | |
| * License along with this library; if not, write to the Free Software | |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA | |
| */ | |
| #define WIN32_LEAN_AND_MEAN | |
| #include "wcmd.h" | |
| #include "wine/debug.h" | |
| WINE_DEFAULT_DEBUG_CHANNEL(cmd); | |
| typedef enum _DISPLAYTIME | |
| { | |
| Creation = 0, | |
| Access, | |
| Written | |
| } DISPLAYTIME; | |
| typedef enum _DISPLAYORDER | |
| { | |
| Name = 0, | |
| Extension, | |
| Size, | |
| Date | |
| } DISPLAYORDER; | |
| static int file_total, dir_total, max_width; | |
| static ULONGLONG byte_total; | |
| static DISPLAYTIME dirTime; | |
| static DISPLAYORDER dirOrder; | |
| static BOOL orderReverse, orderGroupDirs, orderGroupDirsReverse, orderByCol; | |
| static BOOL paged_mode, recurse, wide, bare, lower, shortname, usernames, separator; | |
| static ULONG showattrs, attrsbits; | |
| /***************************************************************************** | |
| * WCMD_strrev | |
| * | |
| * Reverse a WCHARacter string in-place (strrev() is not available under unixen :-( ). | |
| */ | |
| static WCHAR * WCMD_strrev (WCHAR *buff) { | |
| int r, i; | |
| WCHAR b; | |
| r = strlenW (buff); | |
| for (i=0; i<r/2; i++) { | |
| b = buff[i]; | |
| buff[i] = buff[r-i-1]; | |
| buff[r-i-1] = b; | |
| } | |
| return (buff); | |
| } | |
| /***************************************************************************** | |
| * WCMD_filesize64 | |
| * | |
| * Convert a 64-bit number into a WCHARacter string, with commas every three digits. | |
| * Result is returned in a static string overwritten with each call. | |
| * FIXME: There must be a better algorithm! | |
| */ | |
| static WCHAR * WCMD_filesize64 (ULONGLONG n) { | |
| ULONGLONG q; | |
| unsigned int r, i; | |
| WCHAR *p; | |
| static WCHAR buff[32]; | |
| p = buff; | |
| i = -3; | |
| do { | |
| if (separator && ((++i)%3 == 1)) *p++ = ','; | |
| q = n / 10; | |
| r = n - (q * 10); | |
| *p++ = r + '0'; | |
| *p = '\0'; | |
| n = q; | |
| } while (n != 0); | |
| WCMD_strrev (buff); | |
| return buff; | |
| } | |
| /***************************************************************************** | |
| * WCMD_dir_sort | |
| * | |
| * Sort based on the /O options supplied on the command line | |
| */ | |
| static int WCMD_dir_sort (const void *a, const void *b) | |
| { | |
| const WIN32_FIND_DATAW *filea = (const WIN32_FIND_DATAW *)a; | |
| const WIN32_FIND_DATAW *fileb = (const WIN32_FIND_DATAW *)b; | |
| int result = 0; | |
| /* If /OG or /O-G supplied, dirs go at the top or bottom, ignoring the | |
| requested sort order for the directory components */ | |
| if (orderGroupDirs && | |
| ((filea->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) || | |
| (fileb->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))) | |
| { | |
| BOOL aDir = filea->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY; | |
| if (aDir) result = -1; | |
| else result = 1; | |
| if (orderGroupDirsReverse) result = -result; | |
| return result; | |
| /* Order by Name: */ | |
| } else if (dirOrder == Name) { | |
| result = lstrcmpiW(filea->cFileName, fileb->cFileName); | |
| /* Order by Size: */ | |
| } else if (dirOrder == Size) { | |
| ULONG64 sizea = (((ULONG64)filea->nFileSizeHigh) << 32) + filea->nFileSizeLow; | |
| ULONG64 sizeb = (((ULONG64)fileb->nFileSizeHigh) << 32) + fileb->nFileSizeLow; | |
| if( sizea < sizeb ) result = -1; | |
| else if( sizea == sizeb ) result = 0; | |
| else result = 1; | |
| /* Order by Date: (Takes into account which date (/T option) */ | |
| } else if (dirOrder == Date) { | |
| const FILETIME *ft; | |
| ULONG64 timea, timeb; | |
| if (dirTime == Written) { | |
| ft = &filea->ftLastWriteTime; | |
| timea = (((ULONG64)ft->dwHighDateTime) << 32) + ft->dwLowDateTime; | |
| ft = &fileb->ftLastWriteTime; | |
| timeb = (((ULONG64)ft->dwHighDateTime) << 32) + ft->dwLowDateTime; | |
| } else if (dirTime == Access) { | |
| ft = &filea->ftLastAccessTime; | |
| timea = (((ULONG64)ft->dwHighDateTime) << 32) + ft->dwLowDateTime; | |
| ft = &fileb->ftLastAccessTime; | |
| timeb = (((ULONG64)ft->dwHighDateTime) << 32) + ft->dwLowDateTime; | |
| } else { | |
| ft = &filea->ftCreationTime; | |
| timea = (((ULONG64)ft->dwHighDateTime) << 32) + ft->dwLowDateTime; | |
| ft = &fileb->ftCreationTime; | |
| timeb = (((ULONG64)ft->dwHighDateTime) << 32) + ft->dwLowDateTime; | |
| } | |
| if( timea < timeb ) result = -1; | |
| else if( timea == timeb ) result = 0; | |
| else result = 1; | |
| /* Order by Extension: (Takes into account which date (/T option) */ | |
| } else if (dirOrder == Extension) { | |
| WCHAR drive[10]; | |
| WCHAR dir[MAX_PATH]; | |
| WCHAR fname[MAX_PATH]; | |
| WCHAR extA[MAX_PATH]; | |
| WCHAR extB[MAX_PATH]; | |
| /* Split into components */ | |
| WCMD_splitpath(filea->cFileName, drive, dir, fname, extA); | |
| WCMD_splitpath(fileb->cFileName, drive, dir, fname, extB); | |
| result = lstrcmpiW(extA, extB); | |
| } | |
| if (orderReverse) result = -result; | |
| return result; | |
| } | |
| /***************************************************************************** | |
| * WCMD_getfileowner | |
| * | |
| * Reverse a WCHARacter string in-place (strrev() is not available under unixen :-( ). | |
| */ | |
| static void WCMD_getfileowner(WCHAR *filename, WCHAR *owner, int ownerlen) { | |
| ULONG sizeNeeded = 0; | |
| DWORD rc; | |
| WCHAR name[MAXSTRING]; | |
| WCHAR domain[MAXSTRING]; | |
| /* In case of error, return empty string */ | |
| *owner = 0x00; | |
| /* Find out how much space we need for the owner security descriptor */ | |
| GetFileSecurityW(filename, OWNER_SECURITY_INFORMATION, 0, 0, &sizeNeeded); | |
| rc = GetLastError(); | |
| if(rc == ERROR_INSUFFICIENT_BUFFER && sizeNeeded > 0) { | |
| LPBYTE secBuffer; | |
| PSID pSID = NULL; | |
| BOOL defaulted = FALSE; | |
| ULONG nameLen = MAXSTRING; | |
| ULONG domainLen = MAXSTRING; | |
| SID_NAME_USE nameuse; | |
| secBuffer = heap_alloc(sizeNeeded * sizeof(BYTE)); | |
| /* Get the owners security descriptor */ | |
| if(!GetFileSecurityW(filename, OWNER_SECURITY_INFORMATION, secBuffer, | |
| sizeNeeded, &sizeNeeded)) { | |
| heap_free(secBuffer); | |
| return; | |
| } | |
| /* Get the SID from the SD */ | |
| if(!GetSecurityDescriptorOwner(secBuffer, &pSID, &defaulted)) { | |
| heap_free(secBuffer); | |
| return; | |
| } | |
| /* Convert to a username */ | |
| if (LookupAccountSidW(NULL, pSID, name, &nameLen, domain, &domainLen, &nameuse)) { | |
| static const WCHAR fmt[] = {'%','s','%','c','%','s','\0'}; | |
| snprintfW(owner, ownerlen, fmt, domain, '\\', name); | |
| } | |
| heap_free(secBuffer); | |
| } | |
| return; | |
| } | |
| /***************************************************************************** | |
| * WCMD_list_directory | |
| * | |
| * List a single file directory. This function (and those below it) can be called | |
| * recursively when the /S switch is used. | |
| * | |
| * FIXME: Assumes 24-line display for the /P qualifier. | |
| */ | |
| static DIRECTORY_STACK *WCMD_list_directory (DIRECTORY_STACK *inputparms, int level) { | |
| WCHAR string[1024], datestring[32], timestring[32]; | |
| WCHAR real_path[MAX_PATH]; | |
| WIN32_FIND_DATAW *fd; | |
| FILETIME ft; | |
| SYSTEMTIME st; | |
| HANDLE hff; | |
| int dir_count, file_count, entry_count, i, widest, cur_width, tmp_width; | |
| int numCols, numRows; | |
| int rows, cols; | |
| ULARGE_INTEGER byte_count, file_size; | |
| DIRECTORY_STACK *parms; | |
| int concurrentDirs = 0; | |
| BOOL done_header = FALSE; | |
| static const WCHAR fmtDir[] = {'%','1','!','1','0','s','!',' ',' ','%','2','!','8','s','!',' ',' ', | |
| '<','D','I','R','>',' ',' ',' ',' ',' ',' ',' ',' ',' ','\0'}; | |
| static const WCHAR fmtFile[] = {'%','1','!','1','0','s','!',' ',' ','%','2','!','8','s','!',' ',' ', | |
| ' ',' ','%','3','!','1','0','s','!',' ',' ','\0'}; | |
| static const WCHAR fmt2[] = {'%','1','!','-','1','3','s','!','\0'}; | |
| static const WCHAR fmt3[] = {'%','1','!','-','2','3','s','!','\0'}; | |
| static const WCHAR fmt4[] = {'%','1','\0'}; | |
| static const WCHAR fmt5[] = {'%','1','%','2','\0'}; | |
| dir_count = 0; | |
| file_count = 0; | |
| entry_count = 0; | |
| byte_count.QuadPart = 0; | |
| widest = 0; | |
| cur_width = 0; | |
| /* Loop merging all the files from consecutive parms which relate to the | |
| same directory. Note issuing a directory header with no contents | |
| mirrors what windows does */ | |
| parms = inputparms; | |
| fd = heap_alloc(sizeof(WIN32_FIND_DATAW)); | |
| while (parms && strcmpW(inputparms->dirName, parms->dirName) == 0) { | |
| concurrentDirs++; | |
| /* Work out the full path + filename */ | |
| strcpyW(real_path, parms->dirName); | |
| strcatW(real_path, parms->fileName); | |
| /* Load all files into an in memory structure */ | |
| WINE_TRACE("Looking for matches to '%s'\n", wine_dbgstr_w(real_path)); | |
| hff = FindFirstFileW(real_path, &fd[entry_count]); | |
| if (hff != INVALID_HANDLE_VALUE) { | |
| do { | |
| /* Skip any which are filtered out by attribute */ | |
| if ((fd[entry_count].dwFileAttributes & attrsbits) != showattrs) continue; | |
| entry_count++; | |
| /* Keep running track of longest filename for wide output */ | |
| if (wide || orderByCol) { | |
| int tmpLen = strlenW(fd[entry_count-1].cFileName) + 3; | |
| if (fd[entry_count-1].dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) tmpLen = tmpLen + 2; | |
| if (tmpLen > widest) widest = tmpLen; | |
| } | |
| fd = HeapReAlloc(GetProcessHeap(),0,fd,(entry_count+1)*sizeof(WIN32_FIND_DATAW)); | |
| if (fd == NULL) { | |
| FindClose (hff); | |
| WINE_ERR("Out of memory\n"); | |
| errorlevel = 1; | |
| return parms->next; | |
| } | |
| } while (FindNextFileW(hff, &fd[entry_count]) != 0); | |
| FindClose (hff); | |
| } | |
| /* Work out the actual current directory name without a trailing \ */ | |
| strcpyW(real_path, parms->dirName); | |
| real_path[strlenW(parms->dirName)-1] = 0x00; | |
| /* Output the results */ | |
| if (!bare) { | |
| if (level != 0 && (entry_count > 0)) WCMD_output_asis (newlineW); | |
| if (!recurse || ((entry_count > 0) && done_header==FALSE)) { | |
| static const WCHAR headerW[] = {'D','i','r','e','c','t','o','r','y',' ','o','f', | |
| ' ','%','1','\n','\n','\0'}; | |
| WCMD_output (headerW, real_path); | |
| done_header = TRUE; | |
| } | |
| } | |
| /* Move to next parm */ | |
| parms = parms->next; | |
| } | |
| /* Handle case where everything is filtered out */ | |
| if (entry_count > 0) { | |
| /* Sort the list of files */ | |
| qsort (fd, entry_count, sizeof(WIN32_FIND_DATAW), WCMD_dir_sort); | |
| /* Work out the number of columns */ | |
| WINE_TRACE("%d entries, maxwidth=%d, widest=%d\n", entry_count, max_width, widest); | |
| if (wide || orderByCol) { | |
| numCols = max(1, (int)max_width / widest); | |
| numRows = entry_count / numCols; | |
| if (entry_count % numCols) numRows++; | |
| } else { | |
| numCols = 1; | |
| numRows = entry_count; | |
| } | |
| WINE_TRACE("cols=%d, rows=%d\n", numCols, numRows); | |
| for (rows=0; rows<numRows; rows++) { | |
| BOOL addNewLine = TRUE; | |
| for (cols=0; cols<numCols; cols++) { | |
| WCHAR username[24]; | |
| /* Work out the index of the entry being pointed to */ | |
| if (orderByCol) { | |
| i = (cols * numRows) + rows; | |
| if (i >= entry_count) continue; | |
| } else { | |
| i = (rows * numCols) + cols; | |
| if (i >= entry_count) continue; | |
| } | |
| /* /L convers all names to lower case */ | |
| if (lower) { | |
| WCHAR *p = fd[i].cFileName; | |
| while ( (*p = tolower(*p)) ) ++p; | |
| } | |
| /* /Q gets file ownership information */ | |
| if (usernames) { | |
| strcpyW (string, inputparms->dirName); | |
| strcatW (string, fd[i].cFileName); | |
| WCMD_getfileowner(string, username, sizeof(username)/sizeof(WCHAR)); | |
| } | |
| if (dirTime == Written) { | |
| FileTimeToLocalFileTime (&fd[i].ftLastWriteTime, &ft); | |
| } else if (dirTime == Access) { | |
| FileTimeToLocalFileTime (&fd[i].ftLastAccessTime, &ft); | |
| } else { | |
| FileTimeToLocalFileTime (&fd[i].ftCreationTime, &ft); | |
| } | |
| FileTimeToSystemTime (&ft, &st); | |
| GetDateFormatW(0, DATE_SHORTDATE, &st, NULL, datestring, | |
| sizeof(datestring)/sizeof(WCHAR)); | |
| GetTimeFormatW(0, TIME_NOSECONDS, &st, | |
| NULL, timestring, sizeof(timestring)/sizeof(WCHAR)); | |
| if (wide) { | |
| tmp_width = cur_width; | |
| if (fd[i].dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) { | |
| static const WCHAR fmt[] = {'[','%','1',']','\0'}; | |
| WCMD_output (fmt, fd[i].cFileName); | |
| dir_count++; | |
| tmp_width = tmp_width + strlenW(fd[i].cFileName) + 2; | |
| } else { | |
| static const WCHAR fmt[] = {'%','1','\0'}; | |
| WCMD_output (fmt, fd[i].cFileName); | |
| tmp_width = tmp_width + strlenW(fd[i].cFileName) ; | |
| file_count++; | |
| file_size.u.LowPart = fd[i].nFileSizeLow; | |
| file_size.u.HighPart = fd[i].nFileSizeHigh; | |
| byte_count.QuadPart += file_size.QuadPart; | |
| } | |
| cur_width = cur_width + widest; | |
| if ((cur_width + widest) > max_width) { | |
| cur_width = 0; | |
| } else { | |
| static const WCHAR padfmt[] = {'%','1','!','*','s','!','\0'}; | |
| WCMD_output(padfmt, cur_width - tmp_width, nullW); | |
| } | |
| } else if (fd[i].dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) { | |
| dir_count++; | |
| if (!bare) { | |
| WCMD_output (fmtDir, datestring, timestring); | |
| if (shortname) WCMD_output (fmt2, fd[i].cAlternateFileName); | |
| if (usernames) WCMD_output (fmt3, username); | |
| WCMD_output(fmt4,fd[i].cFileName); | |
| } else { | |
| if (!((strcmpW(fd[i].cFileName, dotW) == 0) || | |
| (strcmpW(fd[i].cFileName, dotdotW) == 0))) { | |
| WCMD_output (fmt5, recurse?inputparms->dirName:nullW, fd[i].cFileName); | |
| } else { | |
| addNewLine = FALSE; | |
| } | |
| } | |
| } | |
| else { | |
| file_count++; | |
| file_size.u.LowPart = fd[i].nFileSizeLow; | |
| file_size.u.HighPart = fd[i].nFileSizeHigh; | |
| byte_count.QuadPart += file_size.QuadPart; | |
| if (!bare) { | |
| WCMD_output (fmtFile, datestring, timestring, | |
| WCMD_filesize64(file_size.QuadPart)); | |
| if (shortname) WCMD_output (fmt2, fd[i].cAlternateFileName); | |
| if (usernames) WCMD_output (fmt3, username); | |
| WCMD_output(fmt4,fd[i].cFileName); | |
| } else { | |
| WCMD_output (fmt5, recurse?inputparms->dirName:nullW, fd[i].cFileName); | |
| } | |
| } | |
| } | |
| if (addNewLine) WCMD_output_asis (newlineW); | |
| cur_width = 0; | |
| } | |
| if (!bare) { | |
| if (file_count == 1) { | |
| static const WCHAR fmt[] = {' ',' ',' ',' ',' ',' ',' ','1',' ','f','i','l','e',' ', | |
| '%','1','!','2','5','s','!',' ','b','y','t','e','s','\n','\0'}; | |
| WCMD_output (fmt, WCMD_filesize64 (byte_count.QuadPart)); | |
| } | |
| else { | |
| static const WCHAR fmt[] = {'%','1','!','8','d','!',' ','f','i','l','e','s',' ','%','2','!','2','4','s','!', | |
| ' ','b','y','t','e','s','\n','\0'}; | |
| WCMD_output (fmt, file_count, WCMD_filesize64 (byte_count.QuadPart)); | |
| } | |
| } | |
| byte_total = byte_total + byte_count.QuadPart; | |
| file_total = file_total + file_count; | |
| dir_total = dir_total + dir_count; | |
| if (!bare && !recurse) { | |
| if (dir_count == 1) { | |
| static const WCHAR fmt[] = {'%','1','!','8','d','!',' ','d','i','r','e','c','t','o','r','y', | |
| ' ',' ',' ',' ',' ',' ',' ',' ',' ','\0'}; | |
| WCMD_output (fmt, 1); | |
| } else { | |
| static const WCHAR fmt[] = {'%','1','!','8','d','!',' ','d','i','r','e','c','t','o','r','i', | |
| 'e','s','\0'}; | |
| WCMD_output (fmt, dir_count); | |
| } | |
| } | |
| } | |
| heap_free(fd); | |
| /* When recursing, look in all subdirectories for matches */ | |
| if (recurse) { | |
| DIRECTORY_STACK *dirStack = NULL; | |
| DIRECTORY_STACK *lastEntry = NULL; | |
| WIN32_FIND_DATAW finddata; | |
| /* Build path to search */ | |
| strcpyW(string, inputparms->dirName); | |
| strcatW(string, starW); | |
| WINE_TRACE("Recursive, looking for '%s'\n", wine_dbgstr_w(string)); | |
| hff = FindFirstFileW(string, &finddata); | |
| if (hff != INVALID_HANDLE_VALUE) { | |
| do { | |
| if ((finddata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) && | |
| (strcmpW(finddata.cFileName, dotdotW) != 0) && | |
| (strcmpW(finddata.cFileName, dotW) != 0)) { | |
| DIRECTORY_STACK *thisDir; | |
| int dirsToCopy = concurrentDirs; | |
| /* Loop creating list of subdirs for all concurrent entries */ | |
| parms = inputparms; | |
| while (dirsToCopy > 0) { | |
| dirsToCopy--; | |
| /* Work out search parameter in sub dir */ | |
| strcpyW (string, inputparms->dirName); | |
| strcatW (string, finddata.cFileName); | |
| strcatW (string, slashW); | |
| WINE_TRACE("Recursive, Adding to search list '%s'\n", wine_dbgstr_w(string)); | |
| /* Allocate memory, add to list */ | |
| thisDir = heap_alloc(sizeof(DIRECTORY_STACK)); | |
| if (dirStack == NULL) dirStack = thisDir; | |
| if (lastEntry != NULL) lastEntry->next = thisDir; | |
| lastEntry = thisDir; | |
| thisDir->next = NULL; | |
| thisDir->dirName = heap_strdupW(string); | |
| thisDir->fileName = heap_strdupW(parms->fileName); | |
| parms = parms->next; | |
| } | |
| } | |
| } while (FindNextFileW(hff, &finddata) != 0); | |
| FindClose (hff); | |
| while (dirStack != NULL) { | |
| DIRECTORY_STACK *thisDir = dirStack; | |
| dirStack = WCMD_list_directory (thisDir, 1); | |
| while (thisDir != dirStack) { | |
| DIRECTORY_STACK *tempDir = thisDir->next; | |
| heap_free(thisDir->dirName); | |
| heap_free(thisDir->fileName); | |
| heap_free(thisDir); | |
| thisDir = tempDir; | |
| } | |
| } | |
| } | |
| } | |
| /* Handle case where everything is filtered out */ | |
| if ((file_total + dir_total == 0) && (level == 0)) { | |
| SetLastError (ERROR_FILE_NOT_FOUND); | |
| WCMD_print_error (); | |
| errorlevel = 1; | |
| } | |
| return parms; | |
| } | |
| /***************************************************************************** | |
| * WCMD_dir_trailer | |
| * | |
| * Print out the trailer for the supplied drive letter | |
| */ | |
| static void WCMD_dir_trailer(WCHAR drive) { | |
| ULARGE_INTEGER avail, total, freebytes; | |
| DWORD status; | |
| WCHAR driveName[] = {'c',':','\\','\0'}; | |
| driveName[0] = drive; | |
| status = GetDiskFreeSpaceExW(driveName, &avail, &total, &freebytes); | |
| WINE_TRACE("Writing trailer for '%s' gave %d(%d)\n", wine_dbgstr_w(driveName), | |
| status, GetLastError()); | |
| if (errorlevel==0 && !bare) { | |
| if (recurse) { | |
| static const WCHAR fmt1[] = {'\n',' ',' ',' ',' ',' ','T','o','t','a','l',' ','f','i','l','e','s', | |
| ' ','l','i','s','t','e','d',':','\n','%','1','!','8','d','!',' ','f','i','l','e', | |
| 's','%','2','!','2','5','s','!',' ','b','y','t','e','s','\n','\0'}; | |
| static const WCHAR fmt2[] = {'%','1','!','8','d','!',' ','d','i','r','e','c','t','o','r','i','e','s',' ','%', | |
| '2','!','1','8','s','!',' ','b','y','t','e','s',' ','f','r','e','e','\n','\n', | |
| '\0'}; | |
| WCMD_output (fmt1, file_total, WCMD_filesize64 (byte_total)); | |
| WCMD_output (fmt2, dir_total, WCMD_filesize64 (freebytes.QuadPart)); | |
| } else { | |
| static const WCHAR fmt[] = {' ','%','1','!','1','8','s','!',' ','b','y','t','e','s',' ','f','r','e','e', | |
| '\n','\n','\0'}; | |
| WCMD_output (fmt, WCMD_filesize64 (freebytes.QuadPart)); | |
| } | |
| } | |
| } | |
| /***************************************************************************** | |
| * WCMD_directory | |
| * | |
| * List a file directory. | |
| * | |
| */ | |
| void WCMD_directory (WCHAR *args) | |
| { | |
| WCHAR path[MAX_PATH], cwd[MAX_PATH]; | |
| DWORD status; | |
| CONSOLE_SCREEN_BUFFER_INFO consoleInfo; | |
| WCHAR *p; | |
| WCHAR string[MAXSTRING]; | |
| int argno = 0; | |
| WCHAR *argN = args; | |
| WCHAR lastDrive; | |
| BOOL trailerReqd = FALSE; | |
| DIRECTORY_STACK *fullParms = NULL; | |
| DIRECTORY_STACK *prevEntry = NULL; | |
| DIRECTORY_STACK *thisEntry = NULL; | |
| WCHAR drive[10]; | |
| WCHAR dir[MAX_PATH]; | |
| WCHAR fname[MAX_PATH]; | |
| WCHAR ext[MAX_PATH]; | |
| static const WCHAR dircmdW[] = {'D','I','R','C','M','D','\0'}; | |
| errorlevel = 0; | |
| /* Prefill quals with (uppercased) DIRCMD env var */ | |
| if (GetEnvironmentVariableW(dircmdW, string, sizeof(string)/sizeof(WCHAR))) { | |
| p = string; | |
| while ( (*p = toupper(*p)) ) ++p; | |
| strcatW(string,quals); | |
| strcpyW(quals, string); | |
| } | |
| byte_total = 0; | |
| file_total = dir_total = 0; | |
| /* Initialize all flags to their defaults as if no DIRCMD or quals */ | |
| paged_mode = FALSE; | |
| recurse = FALSE; | |
| wide = FALSE; | |
| bare = FALSE; | |
| lower = FALSE; | |
| shortname = FALSE; | |
| usernames = FALSE; | |
| orderByCol = FALSE; | |
| separator = TRUE; | |
| dirTime = Written; | |
| dirOrder = Name; | |
| orderReverse = FALSE; | |
| orderGroupDirs = FALSE; | |
| orderGroupDirsReverse = FALSE; | |
| showattrs = 0; | |
| attrsbits = FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM; | |
| /* Handle args - Loop through so right most is the effective one */ | |
| /* Note: /- appears to be a negate rather than an off, eg. dir | |
| /-W is wide, or dir /w /-w /-w is also wide */ | |
| p = quals; | |
| while (*p && (*p=='/' || *p==' ')) { | |
| BOOL negate = FALSE; | |
| if (*p++==' ') continue; /* Skip / and blanks introduced through DIRCMD */ | |
| if (*p=='-') { | |
| negate = TRUE; | |
| p++; | |
| } | |
| WINE_TRACE("Processing arg '%c' (in %s)\n", *p, wine_dbgstr_w(quals)); | |
| switch (*p) { | |
| case 'P': if (negate) paged_mode = !paged_mode; | |
| else paged_mode = TRUE; | |
| break; | |
| case 'S': if (negate) recurse = !recurse; | |
| else recurse = TRUE; | |
| break; | |
| case 'W': if (negate) wide = !wide; | |
| else wide = TRUE; | |
| break; | |
| case 'B': if (negate) bare = !bare; | |
| else bare = TRUE; | |
| break; | |
| case 'L': if (negate) lower = !lower; | |
| else lower = TRUE; | |
| break; | |
| case 'X': if (negate) shortname = !shortname; | |
| else shortname = TRUE; | |
| break; | |
| case 'Q': if (negate) usernames = !usernames; | |
| else usernames = TRUE; | |
| break; | |
| case 'D': if (negate) orderByCol = !orderByCol; | |
| else orderByCol = TRUE; | |
| break; | |
| case 'C': if (negate) separator = !separator; | |
| else separator = TRUE; | |
| break; | |
| case 'T': p = p + 1; | |
| if (*p==':') p++; /* Skip optional : */ | |
| if (*p == 'A') dirTime = Access; | |
| else if (*p == 'C') dirTime = Creation; | |
| else if (*p == 'W') dirTime = Written; | |
| /* Support /T and /T: with no parms, default to written */ | |
| else if (*p == 0x00 || *p == '/') { | |
| dirTime = Written; | |
| p = p - 1; /* So when step on, move to '/' */ | |
| } else { | |
| SetLastError(ERROR_INVALID_PARAMETER); | |
| WCMD_print_error(); | |
| errorlevel = 1; | |
| return; | |
| } | |
| break; | |
| case 'O': p = p + 1; | |
| if (*p==':') p++; /* Skip optional : */ | |
| while (*p && *p != '/') { | |
| WINE_TRACE("Processing subparm '%c' (in %s)\n", *p, wine_dbgstr_w(quals)); | |
| switch (*p) { | |
| case 'N': dirOrder = Name; break; | |
| case 'E': dirOrder = Extension; break; | |
| case 'S': dirOrder = Size; break; | |
| case 'D': dirOrder = Date; break; | |
| case '-': if (*(p+1)=='G') orderGroupDirsReverse=TRUE; | |
| else orderReverse = TRUE; | |
| break; | |
| case 'G': orderGroupDirs = TRUE; break; | |
| default: | |
| SetLastError(ERROR_INVALID_PARAMETER); | |
| WCMD_print_error(); | |
| errorlevel = 1; | |
| return; | |
| } | |
| p++; | |
| } | |
| p = p - 1; /* So when step on, move to '/' */ | |
| break; | |
| case 'A': p = p + 1; | |
| showattrs = 0; | |
| attrsbits = 0; | |
| if (*p==':') p++; /* Skip optional : */ | |
| while (*p && *p != '/') { | |
| BOOL anegate = FALSE; | |
| ULONG mask; | |
| /* Note /A: - options are 'offs' not toggles */ | |
| if (*p=='-') { | |
| anegate = TRUE; | |
| p++; | |
| } | |
| WINE_TRACE("Processing subparm '%c' (in %s)\n", *p, wine_dbgstr_w(quals)); | |
| switch (*p) { | |
| case 'D': mask = FILE_ATTRIBUTE_DIRECTORY; break; | |
| case 'H': mask = FILE_ATTRIBUTE_HIDDEN; break; | |
| case 'S': mask = FILE_ATTRIBUTE_SYSTEM; break; | |
| case 'R': mask = FILE_ATTRIBUTE_READONLY; break; | |
| case 'A': mask = FILE_ATTRIBUTE_ARCHIVE; break; | |
| default: | |
| SetLastError(ERROR_INVALID_PARAMETER); | |
| WCMD_print_error(); | |
| errorlevel = 1; | |
| return; | |
| } | |
| /* Keep running list of bits we care about */ | |
| attrsbits |= mask; | |
| /* Mask shows what MUST be in the bits we care about */ | |
| if (anegate) showattrs = showattrs & ~mask; | |
| else showattrs |= mask; | |
| p++; | |
| } | |
| p = p - 1; /* So when step on, move to '/' */ | |
| WINE_TRACE("Result: showattrs %x, bits %x\n", showattrs, attrsbits); | |
| break; | |
| default: | |
| SetLastError(ERROR_INVALID_PARAMETER); | |
| WCMD_print_error(); | |
| errorlevel = 1; | |
| return; | |
| } | |
| p = p + 1; | |
| } | |
| /* Handle conflicting args and initialization */ | |
| if (bare || shortname) wide = FALSE; | |
| if (bare) shortname = FALSE; | |
| if (wide) usernames = FALSE; | |
| if (orderByCol) wide = TRUE; | |
| if (wide) { | |
| if (GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &consoleInfo)) | |
| max_width = consoleInfo.dwSize.X; | |
| else | |
| max_width = 80; | |
| } | |
| if (paged_mode) { | |
| WCMD_enter_paged_mode(NULL); | |
| } | |
| argno = 0; | |
| argN = args; | |
| GetCurrentDirectoryW(MAX_PATH, cwd); | |
| strcatW(cwd, slashW); | |
| /* Loop through all args, calculating full effective directory */ | |
| fullParms = NULL; | |
| prevEntry = NULL; | |
| while (argN) { | |
| WCHAR fullname[MAXSTRING]; | |
| WCHAR *thisArg = WCMD_parameter(args, argno++, &argN, FALSE, FALSE); | |
| if (argN && argN[0] != '/') { | |
| WINE_TRACE("Found parm '%s'\n", wine_dbgstr_w(thisArg)); | |
| if (thisArg[1] == ':' && thisArg[2] == '\\') { | |
| strcpyW(fullname, thisArg); | |
| } else if (thisArg[1] == ':' && thisArg[2] != '\\') { | |
| WCHAR envvar[4]; | |
| static const WCHAR envFmt[] = {'=','%','c',':','\0'}; | |
| wsprintfW(envvar, envFmt, thisArg[0]); | |
| if (!GetEnvironmentVariableW(envvar, fullname, MAX_PATH)) { | |
| static const WCHAR noEnvFmt[] = {'%','c',':','\0'}; | |
| wsprintfW(fullname, noEnvFmt, thisArg[0]); | |
| } | |
| strcatW(fullname, slashW); | |
| strcatW(fullname, &thisArg[2]); | |
| } else if (thisArg[0] == '\\') { | |
| memcpy(fullname, cwd, 2 * sizeof(WCHAR)); | |
| strcpyW(fullname+2, thisArg); | |
| } else { | |
| strcpyW(fullname, cwd); | |
| strcatW(fullname, thisArg); | |
| } | |
| WINE_TRACE("Using location '%s'\n", wine_dbgstr_w(fullname)); | |
| status = GetFullPathNameW(fullname, sizeof(path)/sizeof(WCHAR), path, NULL); | |
| /* | |
| * If the path supplied does not include a wildcard, and the endpoint of the | |
| * path references a directory, we need to list the *contents* of that | |
| * directory not the directory file itself. | |
| */ | |
| if ((strchrW(path, '*') == NULL) && (strchrW(path, '%') == NULL)) { | |
| status = GetFileAttributesW(path); | |
| if ((status != INVALID_FILE_ATTRIBUTES) && (status & FILE_ATTRIBUTE_DIRECTORY)) { | |
| if (!ends_with_backslash( path )) strcatW( path, slashW ); | |
| strcatW (path, starW); | |
| } | |
| } else { | |
| /* Special case wildcard search with no extension (ie parameters ending in '.') as | |
| GetFullPathName strips off the additional '.' */ | |
| if (fullname[strlenW(fullname)-1] == '.') strcatW(path, dotW); | |
| } | |
| WINE_TRACE("Using path '%s'\n", wine_dbgstr_w(path)); | |
| thisEntry = heap_alloc(sizeof(DIRECTORY_STACK)); | |
| if (fullParms == NULL) fullParms = thisEntry; | |
| if (prevEntry != NULL) prevEntry->next = thisEntry; | |
| prevEntry = thisEntry; | |
| thisEntry->next = NULL; | |
| /* Split into components */ | |
| WCMD_splitpath(path, drive, dir, fname, ext); | |
| WINE_TRACE("Path Parts: drive: '%s' dir: '%s' name: '%s' ext:'%s'\n", | |
| wine_dbgstr_w(drive), wine_dbgstr_w(dir), | |
| wine_dbgstr_w(fname), wine_dbgstr_w(ext)); | |
| thisEntry->dirName = heap_alloc(sizeof(WCHAR) * (strlenW(drive)+strlenW(dir)+1)); | |
| strcpyW(thisEntry->dirName, drive); | |
| strcatW(thisEntry->dirName, dir); | |
| thisEntry->fileName = heap_alloc(sizeof(WCHAR) * (strlenW(fname)+strlenW(ext)+1)); | |
| strcpyW(thisEntry->fileName, fname); | |
| strcatW(thisEntry->fileName, ext); | |
| } | |
| } | |
| /* If just 'dir' entered, a '*' parameter is assumed */ | |
| if (fullParms == NULL) { | |
| WINE_TRACE("Inserting default '*'\n"); | |
| fullParms = heap_alloc(sizeof(DIRECTORY_STACK)); | |
| fullParms->next = NULL; | |
| fullParms->dirName = heap_strdupW(cwd); | |
| fullParms->fileName = heap_strdupW(starW); | |
| } | |
| lastDrive = '?'; | |
| prevEntry = NULL; | |
| thisEntry = fullParms; | |
| trailerReqd = FALSE; | |
| while (thisEntry != NULL) { | |
| /* Output disk free (trailer) and volume information (header) if the drive | |
| letter changes */ | |
| if (lastDrive != toupper(thisEntry->dirName[0])) { | |
| /* Trailer Information */ | |
| if (lastDrive != '?') { | |
| trailerReqd = FALSE; | |
| WCMD_dir_trailer(prevEntry->dirName[0]); | |
| } | |
| lastDrive = toupper(thisEntry->dirName[0]); | |
| if (!bare) { | |
| WCHAR drive[3]; | |
| WINE_TRACE("Writing volume for '%c:'\n", thisEntry->dirName[0]); | |
| memcpy(drive, thisEntry->dirName, 2 * sizeof(WCHAR)); | |
| drive[2] = 0x00; | |
| status = WCMD_volume (0, drive); | |
| trailerReqd = TRUE; | |
| if (!status) { | |
| errorlevel = 1; | |
| goto exit; | |
| } | |
| } | |
| } else { | |
| static const WCHAR newLine2[] = {'\n','\n','\0'}; | |
| if (!bare) WCMD_output_asis (newLine2); | |
| } | |
| /* Clear any errors from previous invocations, and process it */ | |
| errorlevel = 0; | |
| prevEntry = thisEntry; | |
| thisEntry = WCMD_list_directory (thisEntry, 0); | |
| } | |
| /* Trailer Information */ | |
| if (trailerReqd) { | |
| WCMD_dir_trailer(prevEntry->dirName[0]); | |
| } | |
| exit: | |
| if (paged_mode) WCMD_leave_paged_mode(); | |
| /* Free storage allocated for parms */ | |
| while (fullParms != NULL) { | |
| prevEntry = fullParms; | |
| fullParms = prevEntry->next; | |
| heap_free(prevEntry->dirName); | |
| heap_free(prevEntry->fileName); | |
| heap_free(prevEntry); | |
| } | |
| } |