mirror of
https://github.com/reactos/reactos.git
synced 2024-07-09 06:05:11 +00:00
e2b04fe75d
Fix arguments parsing and redundant code in case no file specification has been given. (Handles both "attrib" and "attrib +h /s" as given in example.)
433 lines
12 KiB
C
433 lines
12 KiB
C
/*
|
|
* ATTRIB.C - attrib internal command.
|
|
*
|
|
*
|
|
* History:
|
|
*
|
|
* 04-Dec-1998 Eric Kohl
|
|
* started
|
|
*
|
|
* 09-Dec-1998 Eric Kohl
|
|
* implementation works, except recursion ("attrib /s").
|
|
*
|
|
* 05-Jan-1999 Eric Kohl
|
|
* major rewrite.
|
|
* fixed recursion ("attrib /s").
|
|
* started directory support ("attrib /s /d").
|
|
* updated help text.
|
|
*
|
|
* 14-Jan-1999 Eric Kohl
|
|
* Unicode ready!
|
|
*
|
|
* 19-Jan-1999 Eric Kohl
|
|
* Redirection ready!
|
|
*
|
|
* 21-Jan-1999 Eric Kohl
|
|
* Added check for invalid filenames.
|
|
*
|
|
* 23-Jan-1999 Eric Kohl
|
|
* Added handling of multiple filenames.
|
|
*
|
|
* 02-Apr-2005 (Magnus Olsen <magnus@greatlord.com>)
|
|
* Remove all hardcoded strings in En.rc
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#include <windef.h>
|
|
#include <winbase.h>
|
|
#include <wincon.h>
|
|
#include <winuser.h>
|
|
|
|
#include <conutils.h>
|
|
|
|
#include "resource.h"
|
|
|
|
CON_SCREEN StdOutScreen = INIT_CON_SCREEN(StdOut);
|
|
|
|
static
|
|
VOID
|
|
ErrorMessage(
|
|
_In_ DWORD dwErrorCode,
|
|
_In_opt_ PCWSTR pszMsg,
|
|
...)
|
|
{
|
|
INT Len;
|
|
va_list arg_ptr;
|
|
|
|
if (dwErrorCode == ERROR_SUCCESS)
|
|
return;
|
|
|
|
va_start(arg_ptr, pszMsg);
|
|
Len = ConMsgPrintfV(StdErr,
|
|
FORMAT_MESSAGE_FROM_SYSTEM,
|
|
NULL,
|
|
dwErrorCode,
|
|
LANG_USER_DEFAULT,
|
|
&arg_ptr);
|
|
va_end(arg_ptr);
|
|
|
|
/* Fall back just in case the error is not defined */
|
|
if (Len <= 0)
|
|
ConResPrintf(StdErr, STRING_CONSOLE_ERROR, dwErrorCode);
|
|
|
|
/* Display the extra optional message if necessary */
|
|
if (pszMsg)
|
|
ConPrintf(StdErr, L" %s\n", pszMsg);
|
|
}
|
|
|
|
/* Returns TRUE if anything is printed, FALSE otherwise */
|
|
static
|
|
BOOL
|
|
PrintAttribute(
|
|
LPWSTR pszPath,
|
|
LPWSTR pszFile,
|
|
BOOL bRecurse,
|
|
BOOL bDirectories)
|
|
{
|
|
WIN32_FIND_DATAW findData;
|
|
HANDLE hFind;
|
|
WCHAR szFullName[MAX_PATH];
|
|
LPWSTR pszFileName;
|
|
BOOL bFound = FALSE;
|
|
BOOL bIsDir;
|
|
BOOL bExactMatch;
|
|
DWORD Error;
|
|
|
|
/* prepare full file name buffer */
|
|
wcscpy(szFullName, pszPath);
|
|
pszFileName = szFullName + wcslen(szFullName);
|
|
|
|
/* display all subdirectories */
|
|
if (bRecurse)
|
|
{
|
|
/* append *.* */
|
|
wcscpy(pszFileName, L"*.*");
|
|
|
|
hFind = FindFirstFileW(szFullName, &findData);
|
|
if (hFind == INVALID_HANDLE_VALUE)
|
|
{
|
|
Error = GetLastError();
|
|
if ((Error != ERROR_DIRECTORY) && (Error != ERROR_SHARING_VIOLATION)
|
|
&& (Error != ERROR_FILE_NOT_FOUND))
|
|
{
|
|
ErrorMessage(Error, pszFile);
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
do
|
|
{
|
|
if (!(findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
|
|
continue;
|
|
|
|
if (!wcscmp(findData.cFileName, L".") ||
|
|
!wcscmp(findData.cFileName, L".."))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
wcscpy(pszFileName, findData.cFileName);
|
|
wcscat(pszFileName, L"\\");
|
|
bFound |= PrintAttribute(szFullName, pszFile, bRecurse, bDirectories);
|
|
}
|
|
while (FindNextFileW(hFind, &findData));
|
|
FindClose(hFind);
|
|
}
|
|
|
|
/* append file name */
|
|
wcscpy(pszFileName, pszFile);
|
|
|
|
/* search current directory */
|
|
hFind = FindFirstFileW(szFullName, &findData);
|
|
if (hFind == INVALID_HANDLE_VALUE)
|
|
{
|
|
return bFound;
|
|
}
|
|
|
|
do
|
|
{
|
|
bIsDir = findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
|
|
bExactMatch = wcsicmp(findData.cFileName, pszFile) == 0;
|
|
|
|
if (bIsDir && !bDirectories && !bExactMatch)
|
|
continue;
|
|
|
|
if (!wcscmp(findData.cFileName, L".") ||
|
|
!wcscmp(findData.cFileName, L".."))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
wcscpy(pszFileName, findData.cFileName);
|
|
|
|
ConPrintf(StdOut,
|
|
L"%c %c%c%c %s\n",
|
|
(findData.dwFileAttributes & FILE_ATTRIBUTE_ARCHIVE) ? L'A' : L' ',
|
|
(findData.dwFileAttributes & FILE_ATTRIBUTE_SYSTEM) ? L'S' : L' ',
|
|
(findData.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN) ? L'H' : L' ',
|
|
(findData.dwFileAttributes & FILE_ATTRIBUTE_READONLY) ? L'R' : L' ',
|
|
szFullName);
|
|
bFound = TRUE;
|
|
}
|
|
while (FindNextFileW(hFind, &findData));
|
|
FindClose(hFind);
|
|
|
|
return bFound;
|
|
}
|
|
|
|
|
|
/* Returns TRUE if anything changed, FALSE otherwise */
|
|
static
|
|
BOOL
|
|
ChangeAttribute(
|
|
LPWSTR pszPath,
|
|
LPWSTR pszFile,
|
|
BOOL bRecurse,
|
|
BOOL bDirectories,
|
|
DWORD dwMask,
|
|
DWORD dwAttrib)
|
|
{
|
|
WIN32_FIND_DATAW findData;
|
|
HANDLE hFind;
|
|
WCHAR szFullName[MAX_PATH];
|
|
LPWSTR pszFileName;
|
|
BOOL bFound = FALSE;
|
|
BOOL bIsDir;
|
|
BOOL bExactMatch;
|
|
DWORD dwAttribute;
|
|
DWORD Error;
|
|
|
|
/* prepare full file name buffer */
|
|
wcscpy(szFullName, pszPath);
|
|
pszFileName = szFullName + wcslen(szFullName);
|
|
|
|
/* display all subdirectories */
|
|
if (bRecurse)
|
|
{
|
|
/* append *.* */
|
|
wcscpy(pszFileName, L"*.*");
|
|
|
|
hFind = FindFirstFileW(szFullName, &findData);
|
|
if (hFind == INVALID_HANDLE_VALUE)
|
|
{
|
|
Error = GetLastError();
|
|
if ((Error != ERROR_DIRECTORY) && (Error != ERROR_SHARING_VIOLATION)
|
|
&& (Error != ERROR_FILE_NOT_FOUND))
|
|
{
|
|
ErrorMessage(Error, pszFile);
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
do
|
|
{
|
|
if (!(findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
|
|
continue;
|
|
|
|
if (!wcscmp(findData.cFileName, L".") ||
|
|
!wcscmp(findData.cFileName, L".."))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
wcscpy(pszFileName, findData.cFileName);
|
|
wcscat(pszFileName, L"\\");
|
|
bFound |= ChangeAttribute(szFullName, pszFile, bRecurse, bDirectories,
|
|
dwMask, dwAttrib);
|
|
}
|
|
while (FindNextFileW(hFind, &findData));
|
|
FindClose(hFind);
|
|
}
|
|
|
|
/* append file name */
|
|
wcscpy(pszFileName, pszFile);
|
|
|
|
/* search current directory */
|
|
hFind = FindFirstFileW(szFullName, &findData);
|
|
if (hFind == INVALID_HANDLE_VALUE)
|
|
{
|
|
return bFound;
|
|
}
|
|
|
|
do
|
|
{
|
|
bIsDir = findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY;
|
|
bExactMatch = wcsicmp(findData.cFileName, pszFile) == 0;
|
|
|
|
if (bIsDir && !bDirectories && !bExactMatch)
|
|
continue;
|
|
|
|
if (!wcscmp(findData.cFileName, L".") ||
|
|
!wcscmp(findData.cFileName, L".."))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
if (bRecurse && bIsDir && !bDirectories)
|
|
continue;
|
|
|
|
wcscpy(pszFileName, findData.cFileName);
|
|
|
|
dwAttribute = (findData.dwFileAttributes & ~dwMask) | dwAttrib;
|
|
|
|
SetFileAttributes(szFullName, dwAttribute);
|
|
bFound = TRUE;
|
|
}
|
|
while (FindNextFileW(hFind, &findData));
|
|
FindClose(hFind);
|
|
|
|
return bFound;
|
|
}
|
|
|
|
int wmain(int argc, WCHAR *argv[])
|
|
{
|
|
INT i;
|
|
BOOL bRecurse = FALSE;
|
|
BOOL bDirectories = FALSE;
|
|
DWORD dwAttrib = 0;
|
|
DWORD dwMask = 0;
|
|
BOOL bFound = FALSE;
|
|
PWSTR pszFileName;
|
|
WCHAR szFilePath[MAX_PATH + 2] = L""; // + 2 to reserve an extra path separator and a NULL-terminator.
|
|
|
|
/* Initialize the Console Standard Streams */
|
|
ConInitStdStreams();
|
|
|
|
/* Check for options and file specifications */
|
|
for (i = 1; i < argc; i++)
|
|
{
|
|
if (*argv[i] == L'/')
|
|
{
|
|
/* Print help and bail out if needed */
|
|
if (wcscmp(argv[i], L"/?") == 0)
|
|
{
|
|
ConResPuts(StdOut, STRING_ATTRIB_HELP);
|
|
return 0;
|
|
}
|
|
else
|
|
/* Retrieve the enumeration modes */
|
|
if (wcsicmp(argv[i], L"/s") == 0)
|
|
bRecurse = TRUE;
|
|
else if (wcsicmp(argv[i], L"/d") == 0)
|
|
bDirectories = TRUE;
|
|
else
|
|
{
|
|
/* Unknown option */
|
|
ConResPrintf(StdErr, STRING_ERROR_INVALID_PARAM_FORMAT, argv[i]);
|
|
return -1;
|
|
}
|
|
}
|
|
else
|
|
/* Build attributes and mask */
|
|
if ((*argv[i] == L'+') || (*argv[i] == L'-'))
|
|
{
|
|
BOOL bAdd = (*argv[i] == L'+');
|
|
|
|
if (wcslen(argv[i]) != 2)
|
|
{
|
|
ConResPrintf(StdErr, STRING_ERROR_INVALID_PARAM_FORMAT, argv[i]);
|
|
return -1;
|
|
}
|
|
|
|
switch (towupper(argv[i][1]))
|
|
{
|
|
case L'A':
|
|
dwMask |= FILE_ATTRIBUTE_ARCHIVE;
|
|
if (bAdd)
|
|
dwAttrib |= FILE_ATTRIBUTE_ARCHIVE;
|
|
else
|
|
dwAttrib &= ~FILE_ATTRIBUTE_ARCHIVE;
|
|
break;
|
|
|
|
case L'S':
|
|
dwMask |= FILE_ATTRIBUTE_SYSTEM;
|
|
if (bAdd)
|
|
dwAttrib |= FILE_ATTRIBUTE_SYSTEM;
|
|
else
|
|
dwAttrib &= ~FILE_ATTRIBUTE_SYSTEM;
|
|
break;
|
|
|
|
case L'H':
|
|
dwMask |= FILE_ATTRIBUTE_HIDDEN;
|
|
if (bAdd)
|
|
dwAttrib |= FILE_ATTRIBUTE_HIDDEN;
|
|
else
|
|
dwAttrib &= ~FILE_ATTRIBUTE_HIDDEN;
|
|
break;
|
|
|
|
case L'R':
|
|
dwMask |= FILE_ATTRIBUTE_READONLY;
|
|
if (bAdd)
|
|
dwAttrib |= FILE_ATTRIBUTE_READONLY;
|
|
else
|
|
dwAttrib &= ~FILE_ATTRIBUTE_READONLY;
|
|
break;
|
|
|
|
default:
|
|
ConResPrintf(StdErr, STRING_ERROR_INVALID_PARAM_FORMAT, argv[i]);
|
|
return -1;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* At least one file specification found */
|
|
bFound = TRUE;
|
|
}
|
|
}
|
|
|
|
/* If no file specification was found, operate on all files of the current directory */
|
|
if (!bFound)
|
|
{
|
|
DWORD len = GetCurrentDirectoryW(_countof(szFilePath) - 2, szFilePath);
|
|
if (szFilePath[len - 1] != L'\\')
|
|
{
|
|
szFilePath[len] = L'\\';
|
|
szFilePath[len + 1] = UNICODE_NULL;
|
|
}
|
|
pszFileName = L"*.*";
|
|
|
|
if (dwMask == 0)
|
|
bFound = PrintAttribute(szFilePath, pszFileName, bRecurse, bDirectories);
|
|
else
|
|
bFound = ChangeAttribute(szFilePath, pszFileName, bRecurse, bDirectories, dwMask, dwAttrib);
|
|
|
|
if (!bFound)
|
|
ConResPrintf(StdOut, STRING_FILE_NOT_FOUND, pszFileName);
|
|
|
|
return 0;
|
|
}
|
|
|
|
/* Operate on each file specification */
|
|
for (i = 1; i < argc; i++)
|
|
{
|
|
/* Skip options */
|
|
if (*argv[i] == L'/' || *argv[i] == L'+' || *argv[i] == L'-')
|
|
continue;
|
|
|
|
GetFullPathNameW(argv[i], _countof(szFilePath) - 2, szFilePath, &pszFileName);
|
|
if (pszFileName)
|
|
{
|
|
/* Move the file part so as to separate and NULL-terminate the directory */
|
|
MoveMemory(pszFileName + 1, pszFileName,
|
|
sizeof(szFilePath) - (pszFileName -szFilePath + 1) * sizeof(*szFilePath));
|
|
*pszFileName++ = UNICODE_NULL;
|
|
}
|
|
else
|
|
{
|
|
pszFileName = L"";
|
|
}
|
|
|
|
if (dwMask == 0)
|
|
bFound = PrintAttribute(szFilePath, pszFileName, bRecurse, bDirectories);
|
|
else
|
|
bFound = ChangeAttribute(szFilePath, pszFileName, bRecurse, bDirectories, dwMask, dwAttrib);
|
|
|
|
if (!bFound)
|
|
ConResPrintf(StdOut, STRING_FILE_NOT_FOUND, argv[i]);
|
|
}
|
|
|
|
return 0;
|
|
}
|