429 lines
14 KiB
C++
429 lines
14 KiB
C++
|
#include <tchar.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <windows.h>
|
||
|
|
|
||
|
|
#include <winnetp.h>
|
||
|
|
|
||
|
|
#define PRINTCASE(a) case (a): _tprintf(TEXT("%s"), TEXT(#a)); break;
|
||
|
|
|
||
|
|
#define FLAG_HELP 0x00000001
|
||
|
|
#define FLAG_DEBUG 0x00000002
|
||
|
|
#define FLAG_DRIVE 0x00000004
|
||
|
|
#define FLAG_FILE 0x00000008
|
||
|
|
#define FLAG_NETWORK 0x00000010
|
||
|
|
#define FLAG_LOGDRIVE 0x00000020
|
||
|
|
#define FLAG_MOUNTPOINT 0x00000040
|
||
|
|
#define FLAG_VOLUMEGUID 0x00000080
|
||
|
|
|
||
|
|
#define ARRAYSIZE(a) (sizeof((a))/sizeof((a)[0]))
|
||
|
|
|
||
|
|
typedef struct _tagFLAGASSOC
|
||
|
|
{
|
||
|
|
TCHAR szName;
|
||
|
|
DWORD dwFlag;
|
||
|
|
} FLAGASSOC;
|
||
|
|
|
||
|
|
FLAGASSOC FlagList[] =
|
||
|
|
{
|
||
|
|
{ TEXT('?'), FLAG_HELP },
|
||
|
|
{ TEXT('b'), FLAG_DEBUG },
|
||
|
|
{ TEXT('d'), FLAG_DRIVE },
|
||
|
|
{ TEXT('f'), FLAG_FILE },
|
||
|
|
{ TEXT('n'), FLAG_NETWORK },
|
||
|
|
{ TEXT('l'), FLAG_LOGDRIVE },
|
||
|
|
#if 0
|
||
|
|
{ TEXT('m'), FLAG_MOUNTPOINT },
|
||
|
|
{ TEXT('g'), FLAG_VOLUMEGUID },
|
||
|
|
#endif
|
||
|
|
};
|
||
|
|
|
||
|
|
#ifdef UNICODE
|
||
|
|
extern "C"
|
||
|
|
{
|
||
|
|
int __cdecl wmain(int argc, wchar_t* argv[])
|
||
|
|
#else
|
||
|
|
int __cdecl main(int argc, char* argv[])
|
||
|
|
#endif
|
||
|
|
{
|
||
|
|
TCHAR szName[MAX_PATH];
|
||
|
|
DWORD dwFlags = 0;
|
||
|
|
BOOL fGotFileName = FALSE;
|
||
|
|
|
||
|
|
// process the args
|
||
|
|
for (int i = 1; i < argc; ++i)
|
||
|
|
{
|
||
|
|
if ((TEXT('/') == argv[i][0]) || (TEXT('-') == argv[i][0]))
|
||
|
|
{
|
||
|
|
// This is a flag
|
||
|
|
for (int j = 0; j < ARRAYSIZE(FlagList); ++j)
|
||
|
|
{
|
||
|
|
if ((FlagList[j].szName == argv[i][1]) ||
|
||
|
|
((FlagList[j].szName + (TEXT('a') - TEXT('A'))) == argv[i][1]))
|
||
|
|
{
|
||
|
|
dwFlags |= FlagList[j].dwFlag;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// This is the filename
|
||
|
|
lstrcpyn(szName, argv[i], ARRAYSIZE(szName));
|
||
|
|
fGotFileName = TRUE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!fGotFileName)
|
||
|
|
{
|
||
|
|
if (!GetCurrentDirectory(ARRAYSIZE(szName), szName))
|
||
|
|
{
|
||
|
|
dwFlags = FLAG_HELP;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!dwFlags)
|
||
|
|
{
|
||
|
|
dwFlags = FLAG_FILE;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (dwFlags & FLAG_HELP)
|
||
|
|
{
|
||
|
|
dwFlags = FLAG_HELP;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (dwFlags & FLAG_DEBUG)
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("DBG: %d\n"), argc);
|
||
|
|
|
||
|
|
for (int i = 0; i < argc; ++i)
|
||
|
|
_tprintf(TEXT("DBG: Arg #%d is \"%s\"\n"), i, argv[i]);
|
||
|
|
|
||
|
|
_tprintf(TEXT("DBG: Flags\n"));
|
||
|
|
for (DWORD j = 0; j < 32; ++j)
|
||
|
|
{
|
||
|
|
DWORD dw = 1 << j;
|
||
|
|
|
||
|
|
if (dw & dwFlags)
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("\n DBG: "));
|
||
|
|
|
||
|
|
switch (dw)
|
||
|
|
{
|
||
|
|
PRINTCASE(FLAG_HELP );
|
||
|
|
PRINTCASE(FLAG_DEBUG );
|
||
|
|
PRINTCASE(FLAG_DRIVE );
|
||
|
|
PRINTCASE(FLAG_FILE );
|
||
|
|
PRINTCASE(FLAG_NETWORK );
|
||
|
|
PRINTCASE(FLAG_LOGDRIVE );
|
||
|
|
PRINTCASE(FLAG_MOUNTPOINT );
|
||
|
|
PRINTCASE(FLAG_VOLUMEGUID );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (dwFlags & FLAG_HELP)
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("\nDRV [/F] [/D] [/B] [/?] [path]"));
|
||
|
|
_tprintf(TEXT("\n\n [path]"));
|
||
|
|
_tprintf(TEXT("\n Specifies a drive, file, drive mounted on a folder, or Volume GUID."));
|
||
|
|
_tprintf(TEXT("\n\n /F Retrieves file information (GetFileAttributes)."));
|
||
|
|
_tprintf(TEXT("\n /D Retrieves drive information (GetDriveType + GetVolumeInformation)."));
|
||
|
|
_tprintf(TEXT("\n /N Retrieves Network share information (WNetGetConnection)"));
|
||
|
|
_tprintf(TEXT("\n /L Retrieves Logical drives info (GetLogicalDrives)"));
|
||
|
|
_tprintf(TEXT("\n /B Dumps debug info."));
|
||
|
|
_tprintf(TEXT("\n /? Displays this message."));
|
||
|
|
_tprintf(TEXT("\n\nIf no switches are provided, '/F' is assumed. For mounted volumes on\nfolder problems, try appending a backslash.\n\n(Source code: shell\\tools\\drv)"));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("\n---------------------------------------------------\n--- For: \"%s\""), szName);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (dwFlags & FLAG_FILE)
|
||
|
|
{
|
||
|
|
// GetFileAttributes
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("\n............................................................."));
|
||
|
|
_tprintf(TEXT("\n... GetFileAttributes()\n"));
|
||
|
|
|
||
|
|
DWORD dwFA = GetFileAttributes(szName);
|
||
|
|
|
||
|
|
_tprintf(TEXT("\n Return Value: 0x%08X"), dwFA);
|
||
|
|
|
||
|
|
if (0xFFFFFFFF != dwFA)
|
||
|
|
{
|
||
|
|
for (DWORD i = 0; i < 32; ++i)
|
||
|
|
{
|
||
|
|
DWORD dw = 1 << i;
|
||
|
|
|
||
|
|
if (dw & dwFA)
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("\n "));
|
||
|
|
|
||
|
|
switch (dw)
|
||
|
|
{
|
||
|
|
PRINTCASE(FILE_ATTRIBUTE_READONLY );
|
||
|
|
PRINTCASE(FILE_ATTRIBUTE_HIDDEN );
|
||
|
|
PRINTCASE(FILE_ATTRIBUTE_SYSTEM );
|
||
|
|
PRINTCASE(FILE_ATTRIBUTE_DIRECTORY );
|
||
|
|
PRINTCASE(FILE_ATTRIBUTE_ARCHIVE );
|
||
|
|
PRINTCASE(FILE_ATTRIBUTE_DEVICE );
|
||
|
|
PRINTCASE(FILE_ATTRIBUTE_NORMAL );
|
||
|
|
PRINTCASE(FILE_ATTRIBUTE_TEMPORARY );
|
||
|
|
PRINTCASE(FILE_ATTRIBUTE_SPARSE_FILE );
|
||
|
|
PRINTCASE(FILE_ATTRIBUTE_REPARSE_POINT );
|
||
|
|
PRINTCASE(FILE_ATTRIBUTE_COMPRESSED );
|
||
|
|
PRINTCASE(FILE_ATTRIBUTE_OFFLINE );
|
||
|
|
PRINTCASE(FILE_ATTRIBUTE_NOT_CONTENT_INDEXED);
|
||
|
|
PRINTCASE(FILE_ATTRIBUTE_ENCRYPTED );
|
||
|
|
|
||
|
|
default:
|
||
|
|
_tprintf(TEXT(" %08X"), i);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (dwFlags & FLAG_DRIVE)
|
||
|
|
{
|
||
|
|
// GetDriveType
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("\n............................................................."));
|
||
|
|
_tprintf(TEXT("\n... GetDriveType()\n"));
|
||
|
|
|
||
|
|
UINT u = GetDriveType(szName);
|
||
|
|
|
||
|
|
_tprintf(TEXT("\n Return Value: 0x%08X"), u);
|
||
|
|
_tprintf(TEXT("\n "));
|
||
|
|
|
||
|
|
switch (u)
|
||
|
|
{
|
||
|
|
PRINTCASE(DRIVE_UNKNOWN);
|
||
|
|
PRINTCASE(DRIVE_NO_ROOT_DIR);
|
||
|
|
PRINTCASE(DRIVE_REMOVABLE);
|
||
|
|
PRINTCASE(DRIVE_FIXED);
|
||
|
|
PRINTCASE(DRIVE_REMOTE);
|
||
|
|
PRINTCASE(DRIVE_CDROM);
|
||
|
|
PRINTCASE(DRIVE_RAMDISK);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (dwFlags & FLAG_DRIVE)
|
||
|
|
{
|
||
|
|
// GetVolumeInformation
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("\n............................................................."));
|
||
|
|
_tprintf(TEXT("\n... GetVolumeInformation()\n"));
|
||
|
|
|
||
|
|
{
|
||
|
|
TCHAR szVolumeName[MAX_PATH];
|
||
|
|
DWORD dwVolSerialNumber;
|
||
|
|
DWORD dwMaxCompName;
|
||
|
|
DWORD dwFileSysFlags;
|
||
|
|
TCHAR szFileSysName[MAX_PATH];
|
||
|
|
|
||
|
|
BOOL b = GetVolumeInformation(
|
||
|
|
szName,
|
||
|
|
szVolumeName,
|
||
|
|
MAX_PATH,
|
||
|
|
&dwVolSerialNumber,
|
||
|
|
&dwMaxCompName,
|
||
|
|
&dwFileSysFlags,
|
||
|
|
szFileSysName,
|
||
|
|
MAX_PATH);
|
||
|
|
|
||
|
|
_tprintf(TEXT("\n Return Value: 0x%08X"), b);
|
||
|
|
|
||
|
|
if (!b)
|
||
|
|
{
|
||
|
|
_tprintf(TEXT(" (FALSE)"));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
if (TRUE == b)
|
||
|
|
{
|
||
|
|
_tprintf(TEXT(" (TRUE)"));
|
||
|
|
}
|
||
|
|
|
||
|
|
_tprintf(TEXT("\n Volume Name: \"%s\""), szVolumeName);
|
||
|
|
_tprintf(TEXT("\n Volume Serial#: %04X-%04X"), HIWORD(dwVolSerialNumber),
|
||
|
|
LOWORD(dwVolSerialNumber));
|
||
|
|
_tprintf(TEXT("\n Volume Max Comp Name: %d"), dwMaxCompName);
|
||
|
|
_tprintf(TEXT("\n File System Name: \"%s\""), szFileSysName);
|
||
|
|
_tprintf(TEXT("\n File System Flags:"));
|
||
|
|
|
||
|
|
for (DWORD i = 0; i < 32; ++i)
|
||
|
|
{
|
||
|
|
DWORD dw = 1 << i;
|
||
|
|
|
||
|
|
if (dw & dwFileSysFlags)
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("\n "));
|
||
|
|
|
||
|
|
switch (dw)
|
||
|
|
{
|
||
|
|
PRINTCASE(FILE_CASE_SENSITIVE_SEARCH );
|
||
|
|
PRINTCASE(FILE_CASE_PRESERVED_NAMES );
|
||
|
|
PRINTCASE(FILE_UNICODE_ON_DISK );
|
||
|
|
PRINTCASE(FILE_PERSISTENT_ACLS );
|
||
|
|
PRINTCASE(FILE_FILE_COMPRESSION );
|
||
|
|
PRINTCASE(FILE_VOLUME_QUOTAS );
|
||
|
|
PRINTCASE(FILE_SUPPORTS_SPARSE_FILES );
|
||
|
|
PRINTCASE(FILE_SUPPORTS_REPARSE_POINTS);
|
||
|
|
PRINTCASE(FILE_SUPPORTS_REMOTE_STORAGE);
|
||
|
|
PRINTCASE(FILE_VOLUME_IS_COMPRESSED );
|
||
|
|
PRINTCASE(FILE_SUPPORTS_OBJECT_IDS );
|
||
|
|
PRINTCASE(FILE_SUPPORTS_ENCRYPTION );
|
||
|
|
|
||
|
|
default:
|
||
|
|
_tprintf(TEXT(" %08X"), i);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (dwFlags & FLAG_NETWORK)
|
||
|
|
{
|
||
|
|
TCHAR szRemote[MAX_PATH];
|
||
|
|
lstrcpyn(szRemote, szName, ARRAYSIZE(szRemote));
|
||
|
|
|
||
|
|
szRemote[2] = 0;
|
||
|
|
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("\n............................................................."));
|
||
|
|
_tprintf(TEXT("\n... WNetGetConnection()\n"));
|
||
|
|
|
||
|
|
TCHAR szRemoteName[MAX_PATH];
|
||
|
|
DWORD cchRemoteName = ARRAYSIZE(szRemoteName);
|
||
|
|
|
||
|
|
DWORD dw = WNetGetConnection(szRemote, szRemoteName, &cchRemoteName);
|
||
|
|
|
||
|
|
_tprintf(TEXT("\n Return Value: 0x%08X ("), dw);
|
||
|
|
|
||
|
|
switch (dw)
|
||
|
|
{
|
||
|
|
PRINTCASE(NO_ERROR );
|
||
|
|
PRINTCASE(WN_NOT_CONNECTED );
|
||
|
|
PRINTCASE(WN_OPEN_FILES );
|
||
|
|
PRINTCASE(WN_DEVICE_IN_USE );
|
||
|
|
PRINTCASE(WN_BAD_NETNAME );
|
||
|
|
PRINTCASE(WN_BAD_LOCALNAME );
|
||
|
|
PRINTCASE(WN_ALREADY_CONNECTED );
|
||
|
|
PRINTCASE(WN_DEVICE_ERROR );
|
||
|
|
PRINTCASE(WN_CONNECTION_CLOSED );
|
||
|
|
PRINTCASE(WN_NO_NET_OR_BAD_PATH );
|
||
|
|
PRINTCASE(WN_BAD_PROVIDER );
|
||
|
|
PRINTCASE(WN_CANNOT_OPEN_PROFILE );
|
||
|
|
PRINTCASE(WN_BAD_PROFILE );
|
||
|
|
PRINTCASE(WN_BAD_DEV_TYPE );
|
||
|
|
PRINTCASE(WN_DEVICE_ALREADY_REMEMBERED );
|
||
|
|
PRINTCASE(WN_CONNECTED_OTHER_PASSWORD );
|
||
|
|
}
|
||
|
|
|
||
|
|
if (WN_CONNECTION_CLOSED == dw)
|
||
|
|
{
|
||
|
|
_tprintf(TEXT(" == ERROR_CONNECTION_UNAVAIL)"));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
_tprintf(TEXT(")"));
|
||
|
|
}
|
||
|
|
|
||
|
|
if (NO_ERROR == dw)
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("\n RemoteName: \"%s\""), szRemoteName);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("\n\n............................................................."));
|
||
|
|
_tprintf(TEXT("\n... WNetGetConnection3(..., WNGC_INFOLEVEL_DISCONNECTED, ...)\n"));
|
||
|
|
|
||
|
|
WNGC_CONNECTION_STATE wngc;
|
||
|
|
DWORD dwSize = sizeof(wngc.dwState);
|
||
|
|
|
||
|
|
DWORD dw = WNetGetConnection3(szRemote, NULL, WNGC_INFOLEVEL_DISCONNECTED,
|
||
|
|
&wngc.dwState, &dwSize);
|
||
|
|
|
||
|
|
_tprintf(TEXT("\n Return Value: 0x%08X ("), dw);
|
||
|
|
|
||
|
|
switch (dw)
|
||
|
|
{
|
||
|
|
PRINTCASE(NO_ERROR );
|
||
|
|
PRINTCASE(WN_NOT_CONNECTED );
|
||
|
|
PRINTCASE(WN_OPEN_FILES );
|
||
|
|
PRINTCASE(WN_DEVICE_IN_USE );
|
||
|
|
PRINTCASE(WN_BAD_NETNAME );
|
||
|
|
PRINTCASE(WN_BAD_LOCALNAME );
|
||
|
|
PRINTCASE(WN_ALREADY_CONNECTED );
|
||
|
|
PRINTCASE(WN_DEVICE_ERROR );
|
||
|
|
PRINTCASE(WN_CONNECTION_CLOSED );
|
||
|
|
PRINTCASE(WN_NO_NET_OR_BAD_PATH );
|
||
|
|
PRINTCASE(WN_BAD_PROVIDER );
|
||
|
|
PRINTCASE(WN_CANNOT_OPEN_PROFILE );
|
||
|
|
PRINTCASE(WN_BAD_PROFILE );
|
||
|
|
PRINTCASE(WN_BAD_DEV_TYPE );
|
||
|
|
PRINTCASE(WN_DEVICE_ALREADY_REMEMBERED );
|
||
|
|
PRINTCASE(WN_CONNECTED_OTHER_PASSWORD );
|
||
|
|
}
|
||
|
|
|
||
|
|
if (WN_CONNECTION_CLOSED == dw)
|
||
|
|
{
|
||
|
|
_tprintf(TEXT(" == ERROR_CONNECTION_UNAVAIL)"));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
_tprintf(TEXT(")"));
|
||
|
|
}
|
||
|
|
|
||
|
|
if (NO_ERROR == dw)
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("\n CONNECTIONSTATUS.dwState: 0x%08X ("), wngc.dwState);
|
||
|
|
|
||
|
|
switch (wngc.dwState)
|
||
|
|
{
|
||
|
|
PRINTCASE(WNGC_CONNECTED);
|
||
|
|
PRINTCASE(WNGC_DISCONNECTED);
|
||
|
|
}
|
||
|
|
|
||
|
|
_tprintf(TEXT(")"));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (dwFlags & FLAG_LOGDRIVE)
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("\n\n............................................................."));
|
||
|
|
_tprintf(TEXT("\n... GetLogicalDrives()\n"));
|
||
|
|
|
||
|
|
DWORD dwLG = GetLogicalDrives();
|
||
|
|
|
||
|
|
_tprintf(TEXT("\n Return Value: 0x%08X"), dwLG);
|
||
|
|
|
||
|
|
for (DWORD i = 0; i < 32; ++i)
|
||
|
|
{
|
||
|
|
DWORD dw = 1 << i;
|
||
|
|
|
||
|
|
if (dw & dwLG)
|
||
|
|
{
|
||
|
|
_tprintf(TEXT("\n %c:"), TEXT('A') + (TCHAR)i);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
_tprintf(TEXT("\n"));
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
#ifdef UNICODE
|
||
|
|
}
|
||
|
|
#endif
|