rosapps/applications/sysutils/gettype/gettype.c

   1 /*
   2  * ReactOS Win32 Applications
   3  * Copyright (C) 2005 ReactOS Team
   4  *
   5  * COPYRIGHT:   See COPYING in the top level directory
   6  * PROJECT:     ReactOS arp utility
   7  * FILE:        apps/utils/gettype/gettype.c
   8  * PURPOSE:
   9  * PROGRAMMERS: Brandon Turner (turnerb7@msu.edu)
  10  * REVISIONS:
  11  *   GM 30/10/05 Created
  12  *
  13  * FIXME: gettype only supports local computer.
  14  */
  15 #include <windows.h>
  16 #include <tchar.h>
  17 #include <stdio.h>
  18 #include <lm.h>
  19 #include <shellapi.h>
  20
  21 enum
  22 {
  23         GETTYPE_ROLE    = 0x001,
  24         GETTYPE_HELP    = 0x002,
  25         GETTYPE_SP              = 0x004,
  26         GETTYPE_VER             = 0x008,
  27         GETTYPE_MINV    = 0x010,
  28         GETTYPE_MAJV    = 0x020,
  29         GETTYPE_TYPE    = 0x040,
  30         GETTYPE_BUILD   = 0x080
  31 };
  32
  33 INT
  34 GetVersionNumber(BOOL bLocal, LPOSVERSIONINFOEX osvi, LPSERVER_INFO_102 pBuf102)
  35 {
  36         INT VersionNumber = 255;
  37         if(pBuf102 != NULL && !bLocal)
  38         {
  39                 VersionNumber = pBuf102->sv102_version_major * 1000;
  40                 VersionNumber += (pBuf102->sv102_version_minor * 100);
  41         }
  42         else if(bLocal)
  43         {
  44                 VersionNumber = osvi->dwMajorVersion * 1000;
  45                 VersionNumber += (osvi->dwMinorVersion * 100);
  46         }
  47
  48         return VersionNumber;
  49 }
  50
  51 INT
  52 GetMajValue(BOOL Major, BOOL bLocal, LPOSVERSIONINFOEX osvi, LPSERVER_INFO_102 pBuf102)
  53 {
  54         INT VersionNumber = 255;
  55         if(pBuf102 != NULL && !bLocal)
  56         {
  57                 if(Major)
  58                         VersionNumber = pBuf102->sv102_version_major * 1000;
  59                 else
  60                         VersionNumber = pBuf102->sv102_version_minor * 100;
  61         }
  62         else
  63         {
  64                 if(Major)
  65                         VersionNumber = osvi->dwMajorVersion * 1000;
  66                 else
  67                         VersionNumber = osvi->dwMinorVersion * 100;
  68         }
  69         return VersionNumber;
  70 }
  71
  72 INT
  73 GetSystemRole(BOOL bLocal, LPOSVERSIONINFOEX osvi, LPSERVER_INFO_102 pBuf102)
  74 {
  75
  76         if(pBuf102 != NULL && !bLocal)
  77         {
  78                 if ((pBuf102->sv102_type & SV_TYPE_DOMAIN_CTRL) ||
  79                         (pBuf102->sv102_type & SV_TYPE_DOMAIN_BAKCTRL))
  80                         return 1;
  81                 else if(pBuf102->sv102_type & SV_TYPE_SERVER_NT)
  82                         return 2;
  83                 else
  84                         return 3;
  85         }
  86         else
  87         {
  88                 if(osvi->wProductType == VER_NT_DOMAIN_CONTROLLER)
  89                         return 1;
  90                 else if(osvi->wProductType == VER_NT_SERVER)
  91                         return 2;
  92                 else if(osvi->wProductType == VER_NT_WORKSTATION)
  93                         return 3;
  94         }
  95         return 255;
  96
  97 }
  98
  99 INT
 100 GetServicePack(BOOL bLocal, LPOSVERSIONINFOEX osvi, LPSERVER_INFO_102 pBuf102, TCHAR * Server)
 101 {
 102         INT SPNumber = 255;
 103         if(!bLocal)
 104         {
 105                 /* FIXME: Use Registry to get value */
 106         }
 107         else
 108         {
 109                 SPNumber = osvi->wServicePackMajor;
 110         }
 111         return SPNumber;
 112 }
 113
 114 INT
 115 GetBuildNumber(BOOL bLocal, LPOSVERSIONINFOEX osvi, LPSERVER_INFO_102 pBuf102)
 116 {
 117         INT BuildNum = 255;
 118         if(!bLocal)
 119         {
 120                 /* FIXME: Use Registry to get value */
 121         }
 122         else
 123         {
 124                 BuildNum = osvi->dwBuildNumber;
 125         }
 126         return BuildNum;
 127 }
 128
 129 INT GetType(BOOL bLocal, LPOSVERSIONINFOEX osvi, LPSERVER_INFO_102 pBuf102)
 130 {
 131         if(bLocal)
 132         {
 133                 if(osvi->dwMajorVersion == 5)
 134                 {
 135                         if(osvi->dwMinorVersion == 1)
 136                         {
 137                                 if(osvi->wSuiteMask & VER_SUITE_PERSONAL)
 138                                         return 1;
 139                                 else
 140                                         return 2;
 141                         }
 142                         else if(osvi->dwMinorVersion == 2)
 143                         {
 144                                 if(osvi->wSuiteMask & VER_SUITE_BLADE)
 145                                         return 6;
 146                                 else if(osvi->wSuiteMask & VER_SUITE_DATACENTER)
 147                                         return 5;
 148                                 else if(osvi->wSuiteMask & VER_SUITE_ENTERPRISE)
 149                                         return 4;
 150                                 else
 151                                         return 3;
 152                         }
 153                 }
 154         }
 155         else
 156         {
 157                 /* FIXME: Get this value from registry */
 158         }
 159         return 255;
 160 }
 161
 162 VOID
 163 GetBasicInfo(LPOSVERSIONINFOEX osvi, TCHAR * HostName, TCHAR * OSName, TCHAR * Version, TCHAR * Role, TCHAR * Components)
 164 {
 165         /* Host Name - COMPUTERNAME*/
 166         DWORD  bufCharCount = 1024;
 167         GetComputerName(HostName, &bufCharCount);
 168
 169
 170         /* OSName - Windows XP Home Editition */
 171         if(osvi->dwMajorVersion == 4)
 172         {
 173                 _tcscpy(OSName, _T("Microsoft Windows NT 4.0 "));
 174         }
 175         else if(osvi->dwMajorVersion == 5)
 176         {
 177                 if(osvi->dwMajorVersion == 0)
 178                 {
 179                         _tcscpy(OSName, _T("Microsoft Windows 2000 "));
 180                 }
 181                 else if(osvi->dwMinorVersion == 1)
 182                 {
 183                         _tcscpy(OSName, _T("Microsoft Windows XP "));
 184                 }
 185                 else if(osvi->dwMinorVersion == 2)
 186                 {
 187                         _tcscpy(OSName, _T("Microsoft Windows Server 2003 "));
 188                 }
 189         }
 190         else if(osvi->dwMajorVersion == 6)
 191         {
 192                 _tcscpy(OSName, _T("Microsoft Windows Vista "));
 193         }
 194         else
 195         {
 196                 _tcscpy(OSName, _T("Microsoft Windows "));
 197         }
 198
 199         if(osvi->wSuiteMask & VER_SUITE_BLADE)
 200                 _tcscat(OSName, _T("Web Edition"));
 201         if(osvi->wSuiteMask & VER_SUITE_DATACENTER)
 202                 _tcscat(OSName, _T("Datacenter"));
 203         if(osvi->wSuiteMask & VER_SUITE_ENTERPRISE)
 204                 _tcscat(OSName, _T("Enterprise"));
 205         if(osvi->wSuiteMask & VER_SUITE_EMBEDDEDNT)
 206                 _tcscat(OSName, _T("Embedded"));
 207         if(osvi->wSuiteMask & VER_SUITE_PERSONAL)
 208                 _tcscat(OSName, _T("Home Edition"));
 209         if(osvi->wSuiteMask & VER_SUITE_SMALLBUSINESS_RESTRICTED && osvi->wSuiteMask & VER_SUITE_SMALLBUSINESS)
 210                 _tcscat(OSName, _T("Small Bussiness Edition"));
 211
 212         /* Version - 5.1 Build 2600 Serivce Pack 2 */
 213         _stprintf(Version, _T("%d.%d Build %d %s"),(int)osvi->dwMajorVersion,(int)osvi->dwMinorVersion,(int)osvi->dwBuildNumber, osvi->szCSDVersion);
 214
 215         /* Role - Workgroup / Server / Domain Controller */
 216         if(osvi->wProductType == VER_NT_DOMAIN_CONTROLLER)
 217                 _tcscpy(Role, _T("Domain Controller"));
 218         else if(osvi->wProductType == VER_NT_SERVER)
 219                 _tcscpy(Role, _T("Server"));
 220         else if(osvi->wProductType == VER_NT_WORKSTATION)
 221                 _tcscpy(Role, _T("Workgroup"));
 222
 223         /* Components - FIXME: what is something that might be installed? */
 224         _tcscat(Components, _T("Not Installed"));
 225
 226 }
 227
 228 INT
 229 _tmain (VOID)
 230 {
 231         DWORD Operations = 0;
 232         INT ret = 255;
 233         INT i = 0;
 234         INT argc = 0;
 235         /* True if the target is local host */
 236         BOOL bLocal = TRUE;
 237         DWORD nStatus = 0;
 238         TCHAR ServerName[32];
 239         TCHAR RemoteResource[32];
 240         TCHAR UserName[32] = _T("");
 241         TCHAR Password[32] = _T("");
 242         LPOSVERSIONINFOEX osvi = NULL;
 243         LPSERVER_INFO_102 pBuf102 = NULL;
 244         LPTSTR * argv;
 245         osvi = (LPOSVERSIONINFOEX)malloc(sizeof(LPOSVERSIONINFOEX));
 246         pBuf102 = (LPSERVER_INFO_102)malloc(sizeof(LPSERVER_INFO_102));
 247
 248         /* Get the command line correctly since it is unicode */
 249         argv = CommandLineToArgvW(GetCommandLineW(), &argc);
 250
 251
 252         /* Process flags */
 253         if(argc)
 254         {
 255                 for (i = 1; i < argc; i++)
 256                 {
 257                         if(!_tcsicmp(argv[i], _T("/ROLE")) && !Operations)
 258                                 Operations |= GETTYPE_ROLE;
 259                         else if(!_tcsicmp(argv[i], _T("/VER")) && !Operations)
 260                                 Operations |= GETTYPE_VER;
 261                         else if(!_tcsicmp(argv[i], _T("/MAJV")) && !Operations)
 262                                 Operations |= GETTYPE_MAJV;
 263                         else if(!_tcsicmp(argv[i], _T("/MINV")) && !Operations)
 264                                 Operations |= GETTYPE_MINV;
 265                         else if(!_tcsicmp(argv[i], _T("/SP")) && !Operations)
 266                                 Operations |= GETTYPE_SP;
 267                         else if(!_tcsicmp(argv[i], _T("/BUILD")) && !Operations)
 268                                 Operations |= GETTYPE_BUILD;
 269                         else if(!_tcsicmp(argv[i], _T("/TYPE")) && !Operations)
 270                                 Operations |= GETTYPE_TYPE;
 271                         else if(!_tcsicmp(argv[i], _T("/?")) && !Operations)
 272                                 Operations |= GETTYPE_HELP;
 273                         else if(!_tcsicmp(argv[i], _T("/S")) && i + 1 < argc)
 274                         {
 275                                 _tcscpy(ServerName,argv[++i]);
 276                                 bLocal = FALSE;
 277                         }
 278                         else if(!_wcsicmp(argv[i], L"/U") && i + 1 < argc)
 279                                 _tcscpy(UserName,argv[++i]);
 280                         else if(!_wcsicmp(argv[i], L"/P") && i + 1 < argc)
 281                                 _tcscpy(Password,argv[++i]);
 282                         else
 283                         {
 284                                 wprintf(L"Error in paramters, please see usage\n");
 285                                 return 255;
 286                         }
 287                 }
 288         }
 289
 290         /* Some debug info */
 291         //_tprintf(_T("%s - %s - %s - %d"), ServerName, UserName, Password, (int)Operations);
 292
 293         if(!bLocal)
 294         {
 295                 NETRESOURCE nr;
 296
 297
 298                 /* \\*IP or Computer Name*\*Share* */
 299                 _stprintf(RemoteResource, _T("\\\\%s\\IPC$"), ServerName);
 300
 301                 nr.dwType = RESOURCETYPE_ANY;
 302                 nr.lpLocalName = NULL;
 303                 nr.lpProvider= NULL;
 304                 nr.lpRemoteName = RemoteResource;
 305
 306                 /* open a connection to the server with difference user/pass. */
 307                 nStatus = WNetAddConnection2(&nr, UserName[0]?UserName:NULL,Password[0]?Password:NULL, CONNECT_INTERACTIVE | CONNECT_COMMANDLINE);
 308
 309                 if(nStatus != NO_ERROR)
 310                 {
 311                         _tprintf(_T("Error:%d-%d\n"),(int)nStatus,GetLastError());
 312                         return 255;
 313                 }
 314         }
 315
 316         /* Use GetVersionEx for anything that we are looking for locally */
 317         if(bLocal)
 318         {
 319                 osvi->dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);
 320                 if(!GetVersionEx((LPOSVERSIONINFO)osvi))
 321                 {
 322                         _tprintf(_T("Failed to get local information\n"));
 323                         return 255;
 324                 }
 325         }
 326         else
 327         {
 328                 nStatus = NetServerGetInfo(NULL,102,(LPBYTE *)&pBuf102);
 329                 if (nStatus != NERR_Success)
 330                 {
 331                         _tprintf(_T("Failed to connection to remote machine\n"));
 332                         return 255;
 333                 }
 334
 335         }
 336
 337         if(Operations & GETTYPE_VER)
 338         {
 339                 ret = GetVersionNumber(bLocal, osvi, pBuf102);
 340         }
 341         else if(Operations & GETTYPE_MAJV)
 342         {
 343                 ret = GetMajValue(TRUE, bLocal, osvi, pBuf102);
 344         }
 345         else if(Operations & GETTYPE_MINV)
 346         {
 347                 ret = GetMajValue(FALSE, bLocal, osvi, pBuf102);
 348         }
 349         else if(Operations & GETTYPE_ROLE)
 350         {
 351                 ret = GetSystemRole(bLocal, osvi, pBuf102);
 352         }
 353         else if(Operations & GETTYPE_SP)
 354         {
 355                 ret = GetServicePack(bLocal, osvi, pBuf102, ServerName);
 356         }
 357         else if(Operations & GETTYPE_BUILD)
 358         {
 359                 ret = GetBuildNumber(bLocal, osvi, pBuf102);
 360         }
 361         else if(Operations & GETTYPE_TYPE)
 362         {
 363                 ret = GetType(bLocal, osvi, pBuf102);
 364         }
 365         else if(Operations & GETTYPE_HELP)
 366         {
 367                 wprintf(L"GETTYPE  [/ROLE | /SP | /VER | /MAJV | /MINV | /TYPE | /BUILD]");
 368                 ret = 0;
 369         }
 370         else if(!Operations && bLocal)
 371         {
 372                 /* FIXME: what happens when no flags except remote machine, does it
 373                 it print this info for the remote server? */
 374                 TCHAR HostName[1024] = _T("");
 375                 TCHAR OSName[1024] = _T("");
 376                 TCHAR VersionInfo[1024] = _T("");
 377                 TCHAR Role[1024] = _T("");
 378                 TCHAR Components[1024] = _T("");
 379                 GetBasicInfo(osvi, HostName, OSName, VersionInfo, Role, Components);
 380                 _tprintf(_T("\nHostname: %s\nName: %s\nVersion:%s\n") ,HostName, OSName, VersionInfo);
 381                 _tprintf(_T("Role: %s\nComponent: %s\n"), Role, Components);
 382                 ret = 0;
 383         }
 384
 385         /* Clean up some stuff that that was opened */
 386         if(pBuf102)
 387                 NetApiBufferFree(pBuf102);
 388         LocalFree(argv);
 389         if(!bLocal)
 390         {
 391                 WNetCancelConnection2(RemoteResource,0,TRUE);
 392         }
 393         return ret;
 394 }