/* ========================================================================= * Handle the processing for a single directory, optionally recursing into * subdirectories if needed. * Parameters: * rootdir [I] The directory to search in * filespec [I] The filespec to search for * recurse [I] Whether to recurse (search subdirectories before * current directory) * includedirs [I] Whether to set directory attributes as well * attrib_set [I] Attributes to set * attrib_clear [I] Attributes to clear * * Returns TRUE if at least one file displayed / modified * ========================================================================= */ static BOOL ATTRIB_processdirectory(const WCHAR *rootdir, const WCHAR *filespec, BOOL recurse, BOOL includedirs, DWORD attrib_set, DWORD attrib_clear) { BOOL found = FALSE; WCHAR buffer[MAX_PATH]; HANDLE hff; WIN32_FIND_DATAW fd; WCHAR flags[] = {' ',' ',' ',' ',' ',' ',' ',' ','\0'}; static const WCHAR slashW[] = {'\\','\0'}; WINE_TRACE("Processing dir '%s', spec '%s', %d,%x,%x\n", wine_dbgstr_w(rootdir), wine_dbgstr_w(filespec), recurse, attrib_set, attrib_clear); if (recurse) { /* Build spec to search for */ strcpyW(buffer, rootdir); strcatW(buffer, starW); /* Search for directories in the location and recurse if necessary */ WINE_TRACE("Searching for directories with '%s'\n", wine_dbgstr_w(buffer)); hff = FindFirstFileW(buffer, &fd); if (hff != INVALID_HANDLE_VALUE) { do { const WCHAR dot[] = {'.', 0}; const WCHAR dotdot[] = {'.', '.', 0}; /* Only interested in directories, and not . nor .. */ if (!(fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) || !strcmpW(fd.cFileName, dot) || !strcmpW(fd.cFileName, dotdot)) continue; /* Build new root dir to go searching in */ strcpyW(buffer, rootdir); strcatW(buffer, fd.cFileName); strcatW(buffer, slashW); ATTRIB_processdirectory(buffer, filespec, recurse, includedirs, attrib_set, attrib_clear); } while (FindNextFileW(hff, &fd) != 0); } FindClose (hff); } /* Build spec to search for */ strcpyW(buffer, rootdir); strcatW(buffer, filespec); WINE_TRACE("Searching for files as '%s'\n", wine_dbgstr_w(buffer)); /* Search for files in the location with the filespec supplied */ hff = FindFirstFileW(buffer, &fd); if (hff != INVALID_HANDLE_VALUE) { do { const WCHAR dot[] = {'.', 0}; const WCHAR dotdot[] = {'.', '.', 0}; DWORD count; WINE_TRACE("Found '%s'\n", wine_dbgstr_w(fd.cFileName)); if (!strcmpW(fd.cFileName, dot) || !strcmpW(fd.cFileName, dotdot)) continue; if (!includedirs && (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) continue; if (attrib_set || attrib_clear) { fd.dwFileAttributes &= ~attrib_clear; fd.dwFileAttributes |= attrib_set; if (!fd.dwFileAttributes) fd.dwFileAttributes |= FILE_ATTRIBUTE_NORMAL; strcpyW(buffer, rootdir); strcatW(buffer, fd.cFileName); SetFileAttributesW(buffer, fd.dwFileAttributes); found = TRUE; } else { static const WCHAR fmt[] = {'%','1',' ',' ',' ',' ',' ','%','2','\n','\0'}; if (fd.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN) { flags[4] = 'H'; } if (fd.dwFileAttributes & FILE_ATTRIBUTE_SYSTEM) { flags[1] = 'S'; } if (fd.dwFileAttributes & FILE_ATTRIBUTE_ARCHIVE) { flags[0] = 'A'; } if (fd.dwFileAttributes & FILE_ATTRIBUTE_READONLY) { flags[5] = 'R'; } if (fd.dwFileAttributes & FILE_ATTRIBUTE_TEMPORARY) { flags[6] = 'T'; } if (fd.dwFileAttributes & FILE_ATTRIBUTE_COMPRESSED) { flags[7] = 'C'; } strcpyW(buffer, rootdir); strcatW(buffer, fd.cFileName); ATTRIB_wprintf(fmt, flags, buffer); for (count = 0; count < (sizeof(flags)/sizeof(WCHAR) - 1); count++) flags[count] = ' '; found = TRUE; } } while (FindNextFileW(hff, &fd) != 0); } FindClose (hff); return found; }
int wmain(int argc, WCHAR *argv[]) { WCHAR name[MAX_PATH]; WCHAR *namepart; WCHAR curdir[MAX_PATH]; WCHAR originalname[MAX_PATH]; DWORD attrib_set = 0; DWORD attrib_clear = 0; BOOL attrib_recurse = FALSE; BOOL attrib_includedirs = FALSE; static const WCHAR help_option[] = {'/','?','\0'}; static const WCHAR wildcardsW[] = {'*','?','\0'}; int i = 1; BOOL found = FALSE; if ((argc >= 2) && !strcmpW(argv[1], help_option)) { ATTRIB_wprintf(ATTRIB_LoadMessage(STRING_HELP)); return 0; } /* By default all files from current directory are taken into account */ strcpyW(name, starW); while (i < argc) { WCHAR *param = argv[i++]; WINE_TRACE("Processing arg: '%s'\n", wine_dbgstr_w(param)); if ((param[0] == '+') || (param[0] == '-')) { DWORD attrib = 0; switch (param[1]) { case 'H': case 'h': attrib |= FILE_ATTRIBUTE_HIDDEN; break; case 'S': case 's': attrib |= FILE_ATTRIBUTE_SYSTEM; break; case 'R': case 'r': attrib |= FILE_ATTRIBUTE_READONLY; break; case 'A': case 'a': attrib |= FILE_ATTRIBUTE_ARCHIVE; break; default: ATTRIB_wprintf(ATTRIB_LoadMessage(STRING_NYI)); return 0; } switch (param[0]) { case '+': attrib_set = attrib; break; case '-': attrib_clear = attrib; break; } } else if (param[0] == '/') { if (((param[1] == 'D') || (param[1] == 'd')) && !param[2]) { attrib_includedirs = TRUE; } else if (((param[1] == 'S') || (param[1] == 's')) && !param[2]) { attrib_recurse = TRUE; } else { WINE_FIXME("Unknown option %s\n", debugstr_w(param)); } } else if (param[0]) { strcpyW(originalname, param); } } /* Name may be a relative or explicit path, so calculate curdir based on current locations, stripping off the filename */ WINE_TRACE("Supplied name: '%s'\n", wine_dbgstr_w(originalname)); GetFullPathNameW(originalname, sizeof(curdir)/sizeof(WCHAR), curdir, &namepart); WINE_TRACE("Result: '%s'\n", wine_dbgstr_w(curdir)); if (namepart) { strcpyW(name, namepart); *namepart = 0; } else { name[0] = 0; } /* If a directory is explicitly supplied on the command line, and no wildcards are in the name, then allow it to be changed/displayed */ if (strpbrkW(originalname, wildcardsW) == NULL) attrib_includedirs = TRUE; /* Do all the processing based on the filename arg */ found = ATTRIB_processdirectory(curdir, name, attrib_recurse, attrib_includedirs, attrib_set, attrib_clear); if (!found) { ATTRIB_wprintf(ATTRIB_LoadMessage(STRING_FILENOTFOUND), originalname); } return 0; }
int wmain(int argc, WCHAR *argv[]) { DWORD count; HANDLE hff; WIN32_FIND_DATAW fd; WCHAR flags[9] = {' ',' ',' ',' ',' ',' ',' ',' ','\0'}; WCHAR name[128]; WCHAR *param = argc >= 2 ? argv[1] : NULL; DWORD attrib_set = 0; DWORD attrib_clear = 0; WCHAR help_option[] = {'/','?','\0'}; if (param && !strcmpW(param, help_option)) { ATTRIB_wprintf(ATTRIB_LoadMessage(STRING_HELP)); return 0; } if (param && (param[0] == '+' || param[0] == '-')) { DWORD attrib = 0; /* FIXME: the real cmd can handle many more than two args; this should be in a loop */ switch (param[1]) { case 'H': case 'h': attrib |= FILE_ATTRIBUTE_HIDDEN; break; case 'S': case 's': attrib |= FILE_ATTRIBUTE_SYSTEM; break; case 'R': case 'r': attrib |= FILE_ATTRIBUTE_READONLY; break; case 'A': case 'a': attrib |= FILE_ATTRIBUTE_ARCHIVE; break; default: ATTRIB_wprintf(ATTRIB_LoadMessage(STRING_NYI)); return 0; } switch (param[0]) { case '+': attrib_set = attrib; break; case '-': attrib_clear = attrib; break; } param = argc >= 3 ? argv[2] : NULL; } if (!param || strlenW(param) == 0) { static const WCHAR slashStarW[] = {'\\','*','\0'}; GetCurrentDirectoryW(sizeof(name)/sizeof(WCHAR), name); strcatW (name, slashStarW); } else { strcpyW(name, param); } hff = FindFirstFileW(name, &fd); if (hff == INVALID_HANDLE_VALUE) { ATTRIB_wprintf(ATTRIB_LoadMessage(STRING_FILENOTFOUND), name); } else { do { if (attrib_set || attrib_clear) { fd.dwFileAttributes &= ~attrib_clear; fd.dwFileAttributes |= attrib_set; if (!fd.dwFileAttributes) fd.dwFileAttributes |= FILE_ATTRIBUTE_NORMAL; SetFileAttributesW(name, fd.dwFileAttributes); } else { static const WCHAR fmt[] = {'%','1',' ',' ',' ','%','2','\n','\0'}; if (fd.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN) { flags[0] = 'H'; } if (fd.dwFileAttributes & FILE_ATTRIBUTE_SYSTEM) { flags[1] = 'S'; } if (fd.dwFileAttributes & FILE_ATTRIBUTE_ARCHIVE) { flags[2] = 'A'; } if (fd.dwFileAttributes & FILE_ATTRIBUTE_READONLY) { flags[3] = 'R'; } if (fd.dwFileAttributes & FILE_ATTRIBUTE_TEMPORARY) { flags[4] = 'T'; } if (fd.dwFileAttributes & FILE_ATTRIBUTE_COMPRESSED) { flags[5] = 'C'; } ATTRIB_wprintf(fmt, flags, fd.cFileName); for (count=0; count < 8; count++) flags[count] = ' '; } } while (FindNextFileW(hff, &fd) != 0); } FindClose (hff); return 0; }