int main(int argc, char* argv[])
{
SC_HANDLE hSCManager = NULL, hService = NULL;
char szImagePath[MAX_PATH];
SERVICE_TABLE_ENTRY ServiceTable[] =
{
{ SERVICE_NAME, ServiceMain },
{ NULL, NULL }
};
if(argc==1)
{
// Attempt to start service. If this fails we're probably
// not running as a service
if(!StartServiceCtrlDispatcher(ServiceTable))
return 0;
}
if(argc!=2 || (strcmp(argv[1],"-i") && strcmp(argv[1],"-u") && strcmp(argv[1],"-test") && strcmp(argv[1],"-v") && strcmp(argv[1],"-systray") ))
{
fprintf(stderr, "NT CVS Service Handler\n\n"
"Arguments:\n"
"\t%s -i\tInstall\n"
"\t%s -u\tUninstall\n"
"\t%s -test\tInteractive run"
"\t%s -v\tReport version number",
basename(argv[0]),basename(argv[0]),
basename(argv[0]), basename(argv[0]));
return -1;
}
if (!strcmp(argv[1],"-v"))
{
puts(NTSERVICE_VERSION_STRING);
return 0;
}
if(!strcmp(argv[1],"-i"))
{
// connect to the service control manager
if((hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_CREATE_SERVICE)) == NULL)
{
fprintf(stderr,"OpenSCManager Failed\n");
return -1;
}
if((hService=OpenService(hSCManager,SERVICE_NAME,DELETE))!=NULL)
{
DeleteService(hService);
CloseServiceHandle(hService);
}
GetModuleFileName(NULL,szImagePath,MAX_PATH);
if ((hService = CreateService(hSCManager,SERVICE_NAME,DISPLAY_NAME,
STANDARD_RIGHTS_REQUIRED|SERVICE_CHANGE_CONFIG, SERVICE_WIN32_OWN_PROCESS,
SERVICE_AUTO_START, SERVICE_ERROR_NORMAL,
szImagePath, NULL, NULL, NULL, NULL, NULL)) == NULL)
{
fprintf(stderr,"CreateService Failed: %s\n",GetErrorString());
return -1;
}
{
BOOL (WINAPI *pChangeServiceConfig2)(SC_HANDLE,DWORD,LPVOID);
pChangeServiceConfig2=(BOOL (WINAPI *)(SC_HANDLE,DWORD,LPVOID))GetProcAddress(GetModuleHandle("advapi32"),"ChangeServiceConfig2A");
if(pChangeServiceConfig2)
{
SERVICE_DESCRIPTION sd = { NTSERVICE_VERSION_STRING };
pChangeServiceConfig2(hService,SERVICE_CONFIG_DESCRIPTION,&sd);
}
}
CloseServiceHandle(hService);
CloseServiceHandle(hSCManager);
ReportError(FALSE,DISPLAY_NAME " installed successfully");
printf(DISPLAY_NAME " installed successfully\n");
}
if(!strcmp(argv[1],"-u"))
{
// connect to the service control manager
if((hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_CREATE_SERVICE)) == NULL)
{
fprintf(stderr,"OpenSCManager Failed\n");
return -1;
}
if((hService=OpenService(hSCManager,SERVICE_NAME,DELETE))==NULL)
{
fprintf(stderr,"OpenService Failed: %s\n",GetErrorString());
return -1;
}
if(!DeleteService(hService))
{
fprintf(stderr,"DeleteService Failed: %s\n",GetErrorString());
return -1;
}
CloseServiceHandle(hService);
CloseServiceHandle(hSCManager);
ReportError(FALSE,DISPLAY_NAME " uninstalled successfully");
printf(DISPLAY_NAME " uninstalled successfully\n");
}
else if(!strcmp(argv[1],"-test"))
{
ServiceMain(999,NULL);
}
return 0;
}
int main(int argc, char* argv[])
{
SC_HANDLE hSCManager = NULL, hService = NULL;
TCHAR szImagePath[MAX_PATH];
HKEY hk;
DWORD dwType;
SERVICE_TABLE_ENTRY ServiceTable[] =
{
{ SERVICE_NAME, ServiceMain },
{ NULL, NULL }
};
LPSTR szRoot;
if(argc==1)
{
// Attempt to start service. If this fails we're probably
// not running as a service
if(!StartServiceCtrlDispatcher(ServiceTable)) return 0;
}
if(argc<2 || (strcmp(argv[1],"-i") && strcmp(argv[1],"-reglsa") && strcmp(argv[1],"-u") && strcmp(argv[1],"-unreglsa") && strcmp(argv[1],"-test") && strcmp(argv[1],"-v") ))
{
fprintf(stderr, "CVSNT Service Handler\n\n"
"Arguments:\n"
"\t%s -i [cvsroot]\tInstall\n"
"\t%s -reglsa\tRegister LSA helper\n"
"\t%s -u\tUninstall\n"
"\t%s -unreglsa\tUnregister LSA helper\n"
"\t%s -test\tInteractive run\n"
"\t%s -v\tReport version number\n",
basename(argv[0]),basename(argv[0]),
basename(argv[0]), basename(argv[0]),
basename(argv[0]), basename(argv[0])
);
return -1;
}
if(!strcmp(argv[1],"-reglsa"))
{
TCHAR lsaBuf[10240];
DWORD dwLsaBuf;
if(RegOpenKeyEx(HKEY_LOCAL_MACHINE,_T("SYSTEM\\CurrentControlSet\\Control\\Lsa"),0,KEY_ALL_ACCESS,&hk))
{
fprintf(stderr,"Couldn't open LSA registry key, error %d\n",GetLastError());
return -1;
}
dwLsaBuf=sizeof(lsaBuf);
if(RegQueryValueEx(hk,_T("Authentication Packages"),NULL,&dwType,(BYTE*)lsaBuf,&dwLsaBuf))
{
fprintf(stderr,"Couldn't read LSA registry key, error %d\n",GetLastError());
return -1;
}
if(dwType!=REG_MULTI_SZ)
{
fprintf(stderr,"LSA key isn't REG_MULTI_SZ!!!\n");
return -1;
}
lsaBuf[dwLsaBuf]='\0';
TCHAR *p = lsaBuf;
while(*p)
{
if(!_tcscmp(p,"setuid"))
break;
p+=strlen(p)+1;
}
if(!*p)
{
strcpy(p,"setuid");
dwLsaBuf+=strlen(p)+1;
lsaBuf[dwLsaBuf]='\0';
if(RegSetValueEx(hk,_T("Authentication Packages"),NULL,dwType,(BYTE*)lsaBuf,dwLsaBuf))
{
fprintf(stderr,"Couldn't write LSA registry key, error %d\n",GetLastError());
return -1;
}
}
return 0;
}
if(!strcmp(argv[1],"-unreglsa"))
{
TCHAR lsaBuf[10240];
DWORD dwLsaBuf;
if(RegOpenKeyEx(HKEY_LOCAL_MACHINE,_T("SYSTEM\\CurrentControlSet\\Control\\Lsa"),0,KEY_ALL_ACCESS,&hk))
{
fprintf(stderr,"Couldn't open LSA registry key, error %d\n",GetLastError());
return -1;
}
dwLsaBuf=sizeof(lsaBuf);
if(RegQueryValueEx(hk,_T("Authentication Packages"),NULL,&dwType,(BYTE*)lsaBuf,&dwLsaBuf))
{
fprintf(stderr,"Couldn't read LSA registry key, error %d\n",GetLastError());
return -1;
}
if(dwType!=REG_MULTI_SZ)
{
fprintf(stderr,"LSA key isn't REG_MULTI_SZ!!!\n");
return -1;
}
lsaBuf[dwLsaBuf]='\0';
TCHAR *p = lsaBuf;
while(*p)
{
if(!_tcscmp(p,"setuid"))
break;
p+=strlen(p)+1;
}
if(*p)
{
size_t l = strlen(p)+1;
memcpy(p,p+l,(dwLsaBuf-((p+l)-lsaBuf))+1);
dwLsaBuf-=l;
if(RegSetValueEx(hk,_T("Authentication Packages"),NULL,dwType,(BYTE*)lsaBuf,dwLsaBuf))
{
fprintf(stderr,"Couldn't write LSA registry key, error %d\n",GetLastError());
return -1;
}
}
return 0;
}
if(RegCreateKeyEx(HKEY_LOCAL_MACHINE,_T("Software\\CVS\\Pserver"),NULL,_T(""),REG_OPTION_NON_VOLATILE,KEY_ALL_ACCESS,NULL,&hk,NULL))
{
fprintf(stderr,"Couldn't create HKLM\\Software\\CVS\\Pserver key, error %d\n",GetLastError());
return -1;
}
if (!strcmp(argv[1],"-v")) {
puts(NTSERVICE_VERSION_STRING);
return 0;
}
if(!strcmp(argv[1],"-i"))
{
if(argc==3)
{
szRoot = argv[2];
if(GetFileAttributesA(szRoot)==(DWORD)-1)
{
fprintf(stderr,"Repository directory '%s' not found\n",szRoot);
return -1;
}
dwType=REG_SZ;
RegSetValueExA(hk,"Repository0",NULL,dwType,(BYTE*)szRoot,strlen(szRoot)+1);
}
// connect to the service control manager
if((hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_CREATE_SERVICE)) == NULL)
{
fprintf(stderr,"OpenSCManager Failed\n");
return -1;
}
if((hService=OpenService(hSCManager,SERVICE_NAME,DELETE))!=NULL)
{
DeleteService(hService);
CloseServiceHandle(hService);
}
GetModuleFileName(NULL,szImagePath,MAX_PATH);
if ((hService = CreateService(hSCManager,SERVICE_NAME,DISPLAY_NAME,
STANDARD_RIGHTS_REQUIRED|SERVICE_CHANGE_CONFIG, SERVICE_WIN32_OWN_PROCESS|SERVICE_INTERACTIVE_PROCESS,
SERVICE_AUTO_START, SERVICE_ERROR_NORMAL,
szImagePath, NULL, NULL, NULL, NULL, NULL)) == NULL)
{
fprintf(stderr,"CreateService Failed: %s\n",GetErrorString());
return -1;
}
{
BOOL (WINAPI *pChangeServiceConfig2)(SC_HANDLE,DWORD,LPVOID);
pChangeServiceConfig2=(BOOL (WINAPI *)(SC_HANDLE,DWORD,LPVOID))GetProcAddress(GetModuleHandle("advapi32"),"ChangeServiceConfig2A");
if(pChangeServiceConfig2)
{
SERVICE_DESCRIPTION sd = { NTSERVICE_VERSION_STRING };
if(!pChangeServiceConfig2(hService,SERVICE_CONFIG_DESCRIPTION,&sd))
{
0;
}
}
}
CloseServiceHandle(hService);
CloseServiceHandle(hSCManager);
ReportError(FALSE,DISPLAY_NAMEA " installed successfully");
printf(DISPLAY_NAMEA " installed successfully\n");
}
RegCloseKey(hk);
if(!strcmp(argv[1],"-u"))
{
// connect to the service control manager
if((hSCManager = OpenSCManager(NULL, NULL, SC_MANAGER_CREATE_SERVICE)) == NULL)
{
fprintf(stderr,"OpenSCManager Failed\n");
return -1;
}
if((hService=OpenService(hSCManager,SERVICE_NAME,DELETE))==NULL)
{
fprintf(stderr,"OpenService Failed: %s\n",GetErrorString());
return -1;
}
if(!DeleteService(hService))
{
fprintf(stderr,"DeleteService Failed: %s\n",GetErrorString());
return -1;
}
CloseServiceHandle(hService);
CloseServiceHandle(hSCManager);
ReportError(FALSE,DISPLAY_NAMEA " uninstalled successfully");
printf(DISPLAY_NAMEA " uninstalled successfully\n");
}
else if(!strcmp(argv[1],"-test"))
{
ServiceMain(999,NULL);
}
return 0;
}
