mirror of
https://github.com/reactos/reactos.git
synced 2024-08-28 14:18:41 +00:00
395 lines
9.3 KiB
C
395 lines
9.3 KiB
C
/*
|
|
* ReactOS Win32 Applications
|
|
* Copyright (C) 2005 ReactOS Team
|
|
*
|
|
* COPYRIGHT: See COPYING in the top level directory
|
|
* PROJECT: ReactOS arp utility
|
|
* FILE: apps/utils/gettype/gettype.c
|
|
* PURPOSE:
|
|
* PROGRAMMERS: Brandon Turner (turnerb7@msu.edu)
|
|
* REVISIONS:
|
|
* GM 30/10/05 Created
|
|
*
|
|
* FIXME: gettype only supports local computer.
|
|
*/
|
|
#include <windows.h>
|
|
#include <tchar.h>
|
|
#include <stdio.h>
|
|
#include <lm.h>
|
|
#include <shellapi.h>
|
|
|
|
enum
|
|
{
|
|
GETTYPE_ROLE = 0x001,
|
|
GETTYPE_HELP = 0x002,
|
|
GETTYPE_SP = 0x004,
|
|
GETTYPE_VER = 0x008,
|
|
GETTYPE_MINV = 0x010,
|
|
GETTYPE_MAJV = 0x020,
|
|
GETTYPE_TYPE = 0x040,
|
|
GETTYPE_BUILD = 0x080
|
|
};
|
|
|
|
INT
|
|
GetVersionNumber(BOOL bLocal, LPOSVERSIONINFOEX osvi, LPSERVER_INFO_102 pBuf102)
|
|
{
|
|
INT VersionNumber = 255;
|
|
if(pBuf102 != NULL && !bLocal)
|
|
{
|
|
VersionNumber = pBuf102->sv102_version_major * 1000;
|
|
VersionNumber += (pBuf102->sv102_version_minor * 100);
|
|
}
|
|
else if(bLocal)
|
|
{
|
|
VersionNumber = osvi->dwMajorVersion * 1000;
|
|
VersionNumber += (osvi->dwMinorVersion * 100);
|
|
}
|
|
|
|
return VersionNumber;
|
|
}
|
|
|
|
INT
|
|
GetMajValue(BOOL Major, BOOL bLocal, LPOSVERSIONINFOEX osvi, LPSERVER_INFO_102 pBuf102)
|
|
{
|
|
INT VersionNumber = 255;
|
|
if(pBuf102 != NULL && !bLocal)
|
|
{
|
|
if(Major)
|
|
VersionNumber = pBuf102->sv102_version_major * 1000;
|
|
else
|
|
VersionNumber = pBuf102->sv102_version_minor * 100;
|
|
}
|
|
else
|
|
{
|
|
if(Major)
|
|
VersionNumber = osvi->dwMajorVersion * 1000;
|
|
else
|
|
VersionNumber = osvi->dwMinorVersion * 100;
|
|
}
|
|
return VersionNumber;
|
|
}
|
|
|
|
INT
|
|
GetSystemRole(BOOL bLocal, LPOSVERSIONINFOEX osvi, LPSERVER_INFO_102 pBuf102)
|
|
{
|
|
|
|
if(pBuf102 != NULL && !bLocal)
|
|
{
|
|
if ((pBuf102->sv102_type & SV_TYPE_DOMAIN_CTRL) ||
|
|
(pBuf102->sv102_type & SV_TYPE_DOMAIN_BAKCTRL))
|
|
return 1;
|
|
else if(pBuf102->sv102_type & SV_TYPE_SERVER_NT)
|
|
return 2;
|
|
else
|
|
return 3;
|
|
}
|
|
else
|
|
{
|
|
if(osvi->wProductType == VER_NT_DOMAIN_CONTROLLER)
|
|
return 1;
|
|
else if(osvi->wProductType == VER_NT_SERVER)
|
|
return 2;
|
|
else if(osvi->wProductType == VER_NT_WORKSTATION)
|
|
return 3;
|
|
}
|
|
return 255;
|
|
|
|
}
|
|
|
|
INT
|
|
GetServicePack(BOOL bLocal, LPOSVERSIONINFOEX osvi, LPSERVER_INFO_102 pBuf102, TCHAR * Server)
|
|
{
|
|
INT SPNumber = 255;
|
|
if(!bLocal)
|
|
{
|
|
/* FIXME: Use Registry to get value */
|
|
}
|
|
else
|
|
{
|
|
SPNumber = osvi->wServicePackMajor;
|
|
}
|
|
return SPNumber;
|
|
}
|
|
|
|
INT
|
|
GetBuildNumber(BOOL bLocal, LPOSVERSIONINFOEX osvi, LPSERVER_INFO_102 pBuf102)
|
|
{
|
|
INT BuildNum = 255;
|
|
if(!bLocal)
|
|
{
|
|
/* FIXME: Use Registry to get value */
|
|
}
|
|
else
|
|
{
|
|
BuildNum = osvi->dwBuildNumber;
|
|
}
|
|
return BuildNum;
|
|
}
|
|
|
|
INT GetType(BOOL bLocal, LPOSVERSIONINFOEX osvi, LPSERVER_INFO_102 pBuf102)
|
|
{
|
|
if(bLocal)
|
|
{
|
|
if(osvi->dwMajorVersion == 5)
|
|
{
|
|
if(osvi->dwMinorVersion == 1)
|
|
{
|
|
if(osvi->wSuiteMask & VER_SUITE_PERSONAL)
|
|
return 1;
|
|
else
|
|
return 2;
|
|
}
|
|
else if(osvi->dwMinorVersion == 2)
|
|
{
|
|
if(osvi->wSuiteMask & VER_SUITE_BLADE)
|
|
return 6;
|
|
else if(osvi->wSuiteMask & VER_SUITE_DATACENTER)
|
|
return 5;
|
|
else if(osvi->wSuiteMask & VER_SUITE_ENTERPRISE)
|
|
return 4;
|
|
else
|
|
return 3;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* FIXME: Get this value from registry */
|
|
}
|
|
return 255;
|
|
}
|
|
|
|
VOID
|
|
GetBasicInfo(LPOSVERSIONINFOEX osvi, TCHAR * HostName, TCHAR * OSName, TCHAR * Version, TCHAR * Role, TCHAR * Components)
|
|
{
|
|
/* Host Name - COMPUTERNAME*/
|
|
DWORD bufCharCount = 1024;
|
|
GetComputerName(HostName, &bufCharCount);
|
|
|
|
|
|
/* OSName - Windows XP Home Editition */
|
|
if(osvi->dwMajorVersion == 4)
|
|
{
|
|
_tcscpy(OSName, _T("Microsoft Windows NT 4.0 "));
|
|
}
|
|
else if(osvi->dwMajorVersion == 5)
|
|
{
|
|
if(osvi->dwMajorVersion == 0)
|
|
{
|
|
_tcscpy(OSName, _T("Microsoft Windows 2000 "));
|
|
}
|
|
else if(osvi->dwMinorVersion == 1)
|
|
{
|
|
_tcscpy(OSName, _T("Microsoft Windows XP "));
|
|
}
|
|
else if(osvi->dwMinorVersion == 2)
|
|
{
|
|
_tcscpy(OSName, _T("Microsoft Windows Server 2003 "));
|
|
}
|
|
}
|
|
else if(osvi->dwMajorVersion == 6)
|
|
{
|
|
_tcscpy(OSName, _T("Microsoft Windows Vista "));
|
|
}
|
|
else
|
|
{
|
|
_tcscpy(OSName, _T("Microsoft Windows "));
|
|
}
|
|
|
|
if(osvi->wSuiteMask & VER_SUITE_BLADE)
|
|
_tcscat(OSName, _T("Web Edition"));
|
|
if(osvi->wSuiteMask & VER_SUITE_DATACENTER)
|
|
_tcscat(OSName, _T("Datacenter"));
|
|
if(osvi->wSuiteMask & VER_SUITE_ENTERPRISE)
|
|
_tcscat(OSName, _T("Enterprise"));
|
|
if(osvi->wSuiteMask & VER_SUITE_EMBEDDEDNT)
|
|
_tcscat(OSName, _T("Embedded"));
|
|
if(osvi->wSuiteMask & VER_SUITE_PERSONAL)
|
|
_tcscat(OSName, _T("Home Edition"));
|
|
if(osvi->wSuiteMask & VER_SUITE_SMALLBUSINESS_RESTRICTED && osvi->wSuiteMask & VER_SUITE_SMALLBUSINESS)
|
|
_tcscat(OSName, _T("Small Bussiness Edition"));
|
|
|
|
/* Version - 5.1 Build 2600 Serivce Pack 2 */
|
|
_stprintf(Version, _T("%d.%d Build %d %s"),(int)osvi->dwMajorVersion,(int)osvi->dwMinorVersion,(int)osvi->dwBuildNumber, osvi->szCSDVersion);
|
|
|
|
/* Role - Workgroup / Server / Domain Controller */
|
|
if(osvi->wProductType == VER_NT_DOMAIN_CONTROLLER)
|
|
_tcscpy(Role, _T("Domain Controller"));
|
|
else if(osvi->wProductType == VER_NT_SERVER)
|
|
_tcscpy(Role, _T("Server"));
|
|
else if(osvi->wProductType == VER_NT_WORKSTATION)
|
|
_tcscpy(Role, _T("Workgroup"));
|
|
|
|
/* Components - FIXME: what is something that might be installed? */
|
|
_tcscat(Components, _T("Not Installed"));
|
|
|
|
}
|
|
|
|
INT
|
|
_tmain (VOID)
|
|
{
|
|
DWORD Operations = 0;
|
|
INT ret = 255;
|
|
INT i = 0;
|
|
INT argc = 0;
|
|
/* True if the target is local host */
|
|
BOOL bLocal = TRUE;
|
|
DWORD nStatus = 0;
|
|
TCHAR ServerName[32];
|
|
TCHAR RemoteResource[32];
|
|
TCHAR UserName[32] = _T("");
|
|
TCHAR Password[32] = _T("");
|
|
LPOSVERSIONINFOEX osvi = NULL;
|
|
LPSERVER_INFO_102 pBuf102 = NULL;
|
|
LPTSTR * argv;
|
|
osvi = (LPOSVERSIONINFOEX)malloc(sizeof(LPOSVERSIONINFOEX));
|
|
pBuf102 = (LPSERVER_INFO_102)malloc(sizeof(LPSERVER_INFO_102));
|
|
|
|
/* Get the command line correctly since it is unicode */
|
|
argv = CommandLineToArgvW(GetCommandLineW(), &argc);
|
|
|
|
|
|
/* Process flags */
|
|
if(argc)
|
|
{
|
|
for (i = 1; i < argc; i++)
|
|
{
|
|
if(!_tcsicmp(argv[i], _T("/ROLE")) && !Operations)
|
|
Operations |= GETTYPE_ROLE;
|
|
else if(!_tcsicmp(argv[i], _T("/VER")) && !Operations)
|
|
Operations |= GETTYPE_VER;
|
|
else if(!_tcsicmp(argv[i], _T("/MAJV")) && !Operations)
|
|
Operations |= GETTYPE_MAJV;
|
|
else if(!_tcsicmp(argv[i], _T("/MINV")) && !Operations)
|
|
Operations |= GETTYPE_MINV;
|
|
else if(!_tcsicmp(argv[i], _T("/SP")) && !Operations)
|
|
Operations |= GETTYPE_SP;
|
|
else if(!_tcsicmp(argv[i], _T("/BUILD")) && !Operations)
|
|
Operations |= GETTYPE_BUILD;
|
|
else if(!_tcsicmp(argv[i], _T("/TYPE")) && !Operations)
|
|
Operations |= GETTYPE_TYPE;
|
|
else if(!_tcsicmp(argv[i], _T("/?")) && !Operations)
|
|
Operations |= GETTYPE_HELP;
|
|
else if(!_tcsicmp(argv[i], _T("/S")) && i + 1 < argc)
|
|
{
|
|
_tcscpy(ServerName,argv[++i]);
|
|
bLocal = FALSE;
|
|
}
|
|
else if(!wcsicmp(argv[i], L"/U") && i + 1 < argc)
|
|
_tcscpy(UserName,argv[++i]);
|
|
else if(!wcsicmp(argv[i], L"/P") && i + 1 < argc)
|
|
_tcscpy(Password,argv[++i]);
|
|
else
|
|
{
|
|
wprintf(L"Error in paramters, please see usage\n");
|
|
return 255;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Some debug info */
|
|
//_tprintf(_T("%s - %s - %s - %d"), ServerName, UserName, Password, (int)Operations);
|
|
|
|
if(!bLocal)
|
|
{
|
|
NETRESOURCE nr;
|
|
|
|
|
|
/* \\*IP or Computer Name*\*Share* */
|
|
_stprintf(RemoteResource, _T("\\\\%s\\IPC$"), ServerName);
|
|
|
|
nr.dwType = RESOURCETYPE_ANY;
|
|
nr.lpLocalName = NULL;
|
|
nr.lpProvider= NULL;
|
|
nr.lpRemoteName = RemoteResource;
|
|
|
|
/* open a connection to the server with difference user/pass. */
|
|
nStatus = WNetAddConnection2(&nr, UserName[0]?UserName:NULL,Password[0]?Password:NULL, CONNECT_INTERACTIVE | CONNECT_COMMANDLINE);
|
|
|
|
if(nStatus != NO_ERROR)
|
|
{
|
|
_tprintf(_T("Error:%d-%d\n"),(int)nStatus,GetLastError());
|
|
return 255;
|
|
}
|
|
}
|
|
|
|
/* Use GetVersionEx for anything that we are looking for locally */
|
|
if(bLocal)
|
|
{
|
|
osvi->dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);
|
|
if(!GetVersionEx((LPOSVERSIONINFO)osvi))
|
|
{
|
|
_tprintf(_T("Failed to get local information\n"));
|
|
return 255;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
nStatus = NetServerGetInfo(NULL,102,(LPBYTE *)&pBuf102);
|
|
if (nStatus != NERR_Success)
|
|
{
|
|
_tprintf(_T("Failed to connection to remote machine\n"));
|
|
return 255;
|
|
}
|
|
|
|
}
|
|
|
|
if(Operations & GETTYPE_VER)
|
|
{
|
|
ret = GetVersionNumber(bLocal, osvi, pBuf102);
|
|
}
|
|
else if(Operations & GETTYPE_MAJV)
|
|
{
|
|
ret = GetMajValue(TRUE, bLocal, osvi, pBuf102);
|
|
}
|
|
else if(Operations & GETTYPE_MINV)
|
|
{
|
|
ret = GetMajValue(FALSE, bLocal, osvi, pBuf102);
|
|
}
|
|
else if(Operations & GETTYPE_ROLE)
|
|
{
|
|
ret = GetSystemRole(bLocal, osvi, pBuf102);
|
|
}
|
|
else if(Operations & GETTYPE_SP)
|
|
{
|
|
ret = GetServicePack(bLocal, osvi, pBuf102, ServerName);
|
|
}
|
|
else if(Operations & GETTYPE_BUILD)
|
|
{
|
|
ret = GetBuildNumber(bLocal, osvi, pBuf102);
|
|
}
|
|
else if(Operations & GETTYPE_TYPE)
|
|
{
|
|
ret = GetType(bLocal, osvi, pBuf102);
|
|
}
|
|
else if(Operations & GETTYPE_HELP)
|
|
{
|
|
wprintf(L"GETTYPE [/ROLE | /SP | /VER | /MAJV | /MINV | /TYPE | /BUILD]");
|
|
ret = 0;
|
|
}
|
|
else if(!Operations && bLocal)
|
|
{
|
|
/* FIXME: what happens when no flags except remote machine, does it
|
|
it print this info for the remote server? */
|
|
TCHAR HostName[1024] = _T("");
|
|
TCHAR OSName[1024] = _T("");
|
|
TCHAR VersionInfo[1024] = _T("");
|
|
TCHAR Role[1024] = _T("");
|
|
TCHAR Components[1024] = _T("");
|
|
GetBasicInfo(osvi, HostName, OSName, VersionInfo, Role, Components);
|
|
_tprintf(_T("\nHostname: %s\nName: %s\nVersion:%s\n") ,HostName, OSName, VersionInfo);
|
|
_tprintf(_T("Role: %s\nComponent: %s\n"), Role, Components);
|
|
ret = 0;
|
|
}
|
|
|
|
/* Clean up some stuff that that was opened */
|
|
if(pBuf102)
|
|
NetApiBufferFree(pBuf102);
|
|
LocalFree(argv);
|
|
if(!bLocal)
|
|
{
|
|
WNetCancelConnection2(RemoteResource,0,TRUE);
|
|
}
|
|
return ret;
|
|
}
|