Annotation of embedaddon/ntp/ports/winnt/instsrv/instsrv.c, revision 1.1
1.1 ! misho 1: /*
! 2: * File: instsrv.c
! 3: * Purpose: To install a new service and to insert registry entries.
! 4: *
! 5: */
! 6: #ifndef __RPCASYNC_H__
! 7: #define __RPCASYNC_H__ /* Skip asynch rpc inclusion */
! 8: #endif
! 9:
! 10: #include <windows.h>
! 11: #include <stdio.h>
! 12:
! 13: #define PERR(api) printf("\n%s: Error %d from %s on line %d", \
! 14: __FILE__, GetLastError(), api, __LINE__);
! 15:
! 16: #define MSG_FOR_ACCESS_DENIED "You aren't authorized to do this - please see your system Administrator"
! 17: #define MSG_1_FOR_BAD_PATH "The fully qualified path name to the .exe must be given, and"
! 18: #define MSG_2_FOR_BAD_PATH " the drive letter must be for a fixed disk (e.g., not a net drive)"
! 19:
! 20: SC_HANDLE schService;
! 21: SC_HANDLE schSCManager;
! 22: int ok2;
! 23:
! 24: VOID DisplayHelp(VOID);
! 25:
! 26: /* --------------------------------------------------------------------------------------- */
! 27:
! 28: int InstallService(LPCTSTR serviceName, LPCTSTR displayName, LPCTSTR serviceExe)
! 29: {
! 30: LPCTSTR lpszBinaryPathName = serviceExe;
! 31: TCHAR lpszRootPathName[] ="?:\\";
! 32:
! 33: if ( (':' != *(lpszBinaryPathName+1)) || ('\\' != *(lpszBinaryPathName+2)) )
! 34: { printf("\n%s",MSG_1_FOR_BAD_PATH);
! 35: printf("\n%s\n",MSG_2_FOR_BAD_PATH);
! 36: return 1;
! 37: }
! 38:
! 39: #define DRIVE_TYPE_INDETERMINATE 0
! 40: #define ROOT_DIR_DOESNT_EXIST 1
! 41:
! 42: *lpszRootPathName = *(lpszBinaryPathName+0) ;
! 43:
! 44: switch ( GetDriveType(lpszRootPathName) )
! 45: {
! 46: case DRIVE_FIXED :
! 47: { // OK
! 48: break;
! 49: }
! 50: case ROOT_DIR_DOESNT_EXIST :
! 51: { printf("\n%s",MSG_1_FOR_BAD_PATH);
! 52: printf("\n the root directory where the .exe is specified to be must exist, and");
! 53: printf("\n%s\n",MSG_2_FOR_BAD_PATH);
! 54: return 1;
! 55: }
! 56: case DRIVE_TYPE_INDETERMINATE :
! 57: case DRIVE_REMOVABLE :
! 58: case DRIVE_REMOTE :
! 59: case DRIVE_CDROM :
! 60: case DRIVE_RAMDISK :
! 61: { printf("\n%s",MSG_1_FOR_BAD_PATH);
! 62: printf("\n%s\n",MSG_2_FOR_BAD_PATH);
! 63: return 1;
! 64: }
! 65: default :
! 66: { printf("\n%s",MSG_1_FOR_BAD_PATH);
! 67: printf("\n%s\n",MSG_2_FOR_BAD_PATH);
! 68: return 1;
! 69: }
! 70: }
! 71:
! 72: if (INVALID_HANDLE_VALUE == CreateFile(lpszBinaryPathName,
! 73: GENERIC_READ,
! 74: FILE_SHARE_READ,
! 75: NULL,
! 76: OPEN_EXISTING,
! 77: FILE_ATTRIBUTE_NORMAL,
! 78: NULL))
! 79: {
! 80: printf("\n%s",MSG_1_FOR_BAD_PATH);
! 81: printf("\n the file must exist, and");
! 82: printf("\n%s\n",MSG_2_FOR_BAD_PATH);
! 83: return 1;
! 84: }
! 85:
! 86: schService = CreateService(
! 87: schSCManager, // SCManager database
! 88: serviceName, // name of service
! 89: displayName, // name to display
! 90: SERVICE_ALL_ACCESS, // desired access
! 91: SERVICE_WIN32_OWN_PROCESS, // service type
! 92: SERVICE_AUTO_START, // start type
! 93: SERVICE_ERROR_NORMAL, // error control type
! 94: lpszBinaryPathName, // service's binary
! 95: NULL, // no load ordering group
! 96: NULL, // no tag identifier
! 97: NULL, // no dependencies
! 98: NULL, // Local System account
! 99: NULL); // null password
! 100:
! 101: if (NULL == schService)
! 102: { switch (GetLastError())
! 103: {
! 104: case ERROR_ACCESS_DENIED :
! 105: { printf("\n%s",MSG_FOR_ACCESS_DENIED);
! 106: break;
! 107: }
! 108: case ERROR_SERVICE_EXISTS :
! 109: { printf("\nThe %s service is already installed",serviceName);
! 110: printf("\nRemove it first if you need to re-install a new version\n");
! 111: break;
! 112: }
! 113: default :
! 114: { PERR("CreateService");
! 115: }
! 116: }
! 117: return 1;
! 118: }
! 119: else
! 120:
! 121: CloseServiceHandle(schService);
! 122: return 0;
! 123: }
! 124:
! 125: /* --------------------------------------------------------------------------------------- */
! 126:
! 127: int RemoveService(LPCTSTR serviceName)
! 128: {
! 129: {
! 130: #define SZ_ENUM_BUF 4096
! 131: ENUM_SERVICE_STATUS essServiceStatus[SZ_ENUM_BUF];
! 132: DWORD dwBufSize = sizeof(essServiceStatus);
! 133: DWORD dwBytesNeeded = 0;
! 134: DWORD dwServicesReturned = 0;
! 135: DWORD dwResumeHandle = 0;
! 136: DWORD dwI = 0;
! 137: BOOLEAN bFound = FALSE;
! 138:
! 139: if (!EnumServicesStatus(schSCManager,
! 140: SERVICE_WIN32,
! 141: SERVICE_ACTIVE,
! 142: (LPENUM_SERVICE_STATUS)&essServiceStatus,
! 143: dwBufSize,
! 144: &dwBytesNeeded,
! 145: &dwServicesReturned,
! 146: &dwResumeHandle))
! 147: { switch (GetLastError())
! 148: {
! 149: case ERROR_ACCESS_DENIED :
! 150: { printf("\n%s",MSG_FOR_ACCESS_DENIED);
! 151: break;
! 152: }
! 153: default :
! 154: { PERR("EnumServicesStatus");
! 155: }
! 156: }
! 157: return 1;
! 158: }
! 159:
! 160: for (dwI=0; dwI<dwServicesReturned; dwI++)
! 161: { if(0 == _stricmp(essServiceStatus[dwI].lpServiceName,serviceName))
! 162: { bFound = TRUE;
! 163: break;
! 164: }
! 165: }
! 166:
! 167: if (bFound)
! 168: { printf("\nThe %s service cannot be removed until it has been stopped.",serviceName);
! 169: printf("\nTo stop the %s service, use the Stop button in the Control",serviceName);
! 170: printf("\n Panel Services applet\n");
! 171: return 1;
! 172: }
! 173: }
! 174:
! 175: schService = OpenService(schSCManager,
! 176: serviceName,
! 177: SERVICE_ALL_ACCESS);
! 178: if (NULL == schService)
! 179: { switch (GetLastError())
! 180: {
! 181: case ERROR_ACCESS_DENIED :
! 182: { printf("\n%s",MSG_FOR_ACCESS_DENIED);
! 183: break;
! 184: }
! 185: case ERROR_SERVICE_DOES_NOT_EXIST :
! 186: { printf("\nThe %s service is not installed, so cannot be removed\n",serviceName);
! 187: break;
! 188: }
! 189: default :
! 190: { PERR("OpenService");
! 191: }
! 192: }
! 193: return 1;
! 194: }
! 195:
! 196: if (DeleteService(schService))
! 197: { printf("\nDelete of Service \"Network Time Protocol\" was SUCCESSFUL\n");
! 198: return 0;
! 199: }
! 200: else
! 201: { switch (GetLastError())
! 202: {
! 203: case ERROR_ACCESS_DENIED :
! 204: { printf("\n%s",MSG_FOR_ACCESS_DENIED);
! 205: break;
! 206: }
! 207: default :
! 208: { PERR("DeleteService");
! 209: }
! 210: }
! 211: return 1;
! 212: }
! 213: }
! 214:
! 215: /* --------------------------------------------------------------------------------------- */
! 216:
! 217: int addSourceToRegistry(LPSTR pszAppname, LPSTR pszMsgDLL)
! 218: {
! 219: HKEY hk; /* registry key handle */
! 220: DWORD dwData;
! 221: BOOL bSuccess;
! 222: char regarray[200];
! 223: char *lpregarray = regarray;
! 224:
! 225: /* When an application uses the RegisterEventSource or OpenEventLog
! 226: function to get a handle of an event log, the event loggging service
! 227: searches for the specified source name in the registry. You can add a
! 228: new source name to the registry by opening a new registry subkey
! 229: under the Application key and adding registry values to the new
! 230: subkey. */
! 231:
! 232: strcpy(lpregarray, "SYSTEM\\CurrentControlSet\\Services\\EventLog\\Application\\");
! 233: strcat(lpregarray, pszAppname);
! 234: /* Create a new key for our application */
! 235: bSuccess = RegCreateKey(HKEY_LOCAL_MACHINE, lpregarray, &hk);
! 236: if(bSuccess != ERROR_SUCCESS)
! 237: {
! 238: PERR("RegCreateKey");
! 239: return 1;
! 240: }
! 241:
! 242: /* Add the Event-ID message-file name to the subkey. */
! 243: bSuccess = RegSetValueEx(hk, /* subkey handle */
! 244: "EventMessageFile", /* value name */
! 245: 0, /* must be zero */
! 246: REG_EXPAND_SZ, /* value type */
! 247: (LPBYTE) pszMsgDLL, /* address of value data */
! 248: strlen(pszMsgDLL) + 1); /* length of value data */
! 249: if(bSuccess != ERROR_SUCCESS)
! 250: {
! 251: PERR("RegSetValueEx");
! 252: return 1;
! 253: }
! 254:
! 255: /* Set the supported types flags and addit to the subkey. */
! 256: dwData = EVENTLOG_ERROR_TYPE | EVENTLOG_WARNING_TYPE |
! 257: EVENTLOG_INFORMATION_TYPE;
! 258: bSuccess = RegSetValueEx(hk, /* subkey handle */
! 259: "TypesSupported", /* value name */
! 260: 0, /* must be zero */
! 261: REG_DWORD, /* value type */
! 262: (LPBYTE) &dwData, /* address of value data */
! 263: sizeof(DWORD)); /* length of value data */
! 264: if(bSuccess != ERROR_SUCCESS)
! 265: {
! 266: PERR("RegSetValueEx");
! 267: return 1;
! 268: }
! 269: RegCloseKey(hk);
! 270: return 0;
! 271: }
! 272:
! 273: /* --------------------------------------------------------------------------------------- */
! 274:
! 275: int addKeysToRegistry()
! 276:
! 277: {
! 278: HKEY hk; /* registry key handle */
! 279: BOOL bSuccess;
! 280: char myarray[200];
! 281: char *lpmyarray = myarray;
! 282: int arsize = 0;
! 283:
! 284: /* now add the depends on service key */
! 285:
! 286: /* Create a new key for our application */
! 287: bSuccess = RegCreateKey(HKEY_LOCAL_MACHINE,
! 288: "SYSTEM\\CurrentControlSet\\Services\\NTP", &hk);
! 289: if(bSuccess != ERROR_SUCCESS)
! 290: {
! 291: PERR("RegCreateKey");
! 292: return 1;
! 293: }
! 294:
! 295: strcpy(lpmyarray,"TcpIp");
! 296: lpmyarray = lpmyarray + 6;
! 297: arsize = arsize + 6;
! 298: strcpy(lpmyarray,"Afd");
! 299: lpmyarray = lpmyarray + 4;
! 300: arsize = arsize + 4;
! 301: arsize = arsize + 2;
! 302: strcpy(lpmyarray,"\0\0");
! 303:
! 304: bSuccess = RegSetValueEx(hk, /* subkey handle */
! 305: "DependOnService", /* value name */
! 306: 0, /* must be zero */
! 307: REG_MULTI_SZ, /* value type */
! 308: (LPBYTE) &myarray, /* address of value data */
! 309: arsize); /* length of value data */
! 310: if(bSuccess != ERROR_SUCCESS)
! 311: {
! 312: PERR("RegSetValueEx");
! 313: return 1;
! 314: }
! 315:
! 316: RegCloseKey(hk);
! 317: return 0;
! 318: }
! 319:
! 320: /* --------------------------------------------------------------------------------------- */
! 321:
! 322: int main(int argc, char *argv[])
! 323: {
! 324: #define SZ_NAME_BUF 270 // 256 is max, add a little
! 325: UCHAR ucNameBuf[SZ_NAME_BUF] = "NTP";
! 326: LPTSTR lpszServName = (LPTSTR)&ucNameBuf;
! 327:
! 328: UCHAR ucDNameBuf[SZ_NAME_BUF] = "Network Time Protocol";
! 329: LPTSTR lpszDispName = (LPTSTR)&ucDNameBuf;
! 330:
! 331:
! 332: UCHAR ucExeNBuf[SZ_NAME_BUF] = "";
! 333: LPTSTR lpszExeName = (LPTSTR)&ucExeNBuf;
! 334:
! 335: BOOL bRemovingService = FALSE;
! 336: char *p;
! 337:
! 338: int ok = 0;
! 339:
! 340: // check if Win32s, if so, display notice and terminate
! 341: if( GetVersion() & 0x80000000 )
! 342: {
! 343: MessageBox( NULL,
! 344: "This application cannot run on Windows 3.1.\n"
! 345: "This application will now terminate.",
! 346: "NAMED",
! 347: MB_OK | MB_ICONSTOP | MB_SETFOREGROUND );
! 348: return( 1 );
! 349: }
! 350: if (argc == 2)
! 351: bRemovingService = (!stricmp(argv[1], "remove"));
! 352:
! 353: if(!bRemovingService)
! 354: {
! 355:
! 356:
! 357: if (argc != 2)
! 358: {
! 359: DisplayHelp();
! 360: return(1);
! 361: }
! 362:
! 363: p=argv[1];
! 364: if ( ('/' == *p)
! 365: || ('-' == *p) )
! 366: {
! 367: DisplayHelp();
! 368: return(1);
! 369: }
! 370:
! 371:
! 372: }
! 373:
! 374: if (strlen(argv[1]) > 256)
! 375: {
! 376: printf("\nThe service name cannot be longer than 256 characters\n");
! 377: return(1);
! 378: }
! 379:
! 380:
! 381:
! 382: bRemovingService = (!stricmp(argv[1], "remove"));
! 383: schSCManager = OpenSCManager(
! 384: NULL, // machine (NULL == local)
! 385: NULL, // database (NULL == default)
! 386: SC_MANAGER_ALL_ACCESS); // access required
! 387:
! 388: if (NULL == schSCManager)
! 389: { switch (GetLastError())
! 390: {
! 391: case ERROR_ACCESS_DENIED :
! 392: { printf("\n%s",MSG_FOR_ACCESS_DENIED);
! 393: break;
! 394: }
! 395: default :
! 396: { PERR("OpenSCManager");
! 397: }
! 398: }
! 399: return (0);
! 400: }
! 401:
! 402: if (bRemovingService)
! 403: {
! 404: ok = RemoveService(lpszServName);
! 405: }
! 406: else
! 407: {
! 408: /* get the exe name */
! 409: strcpy(lpszExeName,argv[1]);
! 410: ok = InstallService(lpszServName, lpszDispName, lpszExeName);
! 411: }
! 412:
! 413: CloseServiceHandle(schSCManager);
! 414:
! 415: if (!bRemovingService)
! 416: {
! 417: if (ok == 0)
! 418: { /* Set the Event-ID message-file name. */
! 419: ok = addSourceToRegistry("NTP", lpszExeName);
! 420: if (ok == 0)
! 421: ok = addKeysToRegistry();
! 422: else return ok;
! 423:
! 424: if (ok == 0)
! 425: {
! 426: printf("\nThe \"Network Time Protocol\" service was successfully created.\n");
! 427: printf("\nDon't forget!!! You must now go to the Control Panel and");
! 428: printf("\n use the Services applet to change the account name and");
! 429: printf("\n password that the NTP Service will use when");
! 430: printf("\n it starts.");
! 431: printf("\nTo do this: use the Startup button in the Services applet,");
! 432: printf("\n and (for example) specify the desired account and");
! 433: printf("\n correct password.");
! 434: printf("\nAlso, use the Services applet to ensure this newly installed");
! 435: printf("\n service starts automatically on bootup.\n");
! 436: return 0;
! 437: }
! 438: }
! 439: else return ok;
! 440: }
! 441: return 0;
! 442: }
! 443:
! 444: /* --------------------------------------------------------------------------------------- */
! 445:
! 446: VOID DisplayHelp(VOID)
! 447: {
! 448: printf("Installs or removes the NTP service.\n");
! 449: printf("To install the NTP service,\n");
! 450: printf("type INSTSRV <path> \n");
! 451: printf("Where:\n");
! 452: printf(" path Absolute path to the NTP service, name.exe. You must\n");
! 453: printf(" use a fully qualified path and the drive letter must be for a\n");
! 454: printf(" fixed, local drive.\n\n");
! 455: printf("For example, INSTSRV i:\\winnt\\system32\\ntpd.exe\n");
! 456: printf("To remove the NTP service,\n");
! 457: printf("type INSTSRV remove \n");
! 458:
! 459: }
! 460:
! 461: /* EOF */
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to instsrv.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / ntp / ports / winnt / instsrv