/*
|
* Copyright (c) 2002 - 2003
|
* NetGroup, Politecnico di Torino (Italy)
|
* All rights reserved.
|
*
|
* Redistribution and use in source and binary forms, with or without
|
* modification, are permitted provided that the following conditions
|
* are met:
|
*
|
* 1. Redistributions of source code must retain the above copyright
|
* notice, this list of conditions and the following disclaimer.
|
* 2. Redistributions in binary form must reproduce the above copyright
|
* notice, this list of conditions and the following disclaimer in the
|
* documentation and/or other materials provided with the distribution.
|
* 3. Neither the name of the Politecnico di Torino nor the names of its
|
* contributors may be used to endorse or promote products derived from
|
* this software without specific prior written permission.
|
*
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
*
|
*/
|
|
|
#include "rpcapd.h"
|
#include <pcap.h> // for PCAP_ERRBUF_SIZE
|
#include "sockutils.h" // for SOCK_DEBUG_MESSAGE
|
#include "portability.h"
|
#include "fileconf.h"
|
|
static SERVICE_STATUS_HANDLE service_status_handle;
|
static SERVICE_STATUS service_status;
|
|
void svc_geterr(char *str);
|
static void WINAPI svc_main(DWORD argc, char **argv);
|
static void update_svc_status(DWORD state, DWORD progress_indicator);
|
|
int svc_start(void)
|
{
|
int rc;
|
SERVICE_TABLE_ENTRY ste[] =
|
{
|
{ PROGRAM_NAME, svc_main },
|
{ NULL, NULL }
|
};
|
|
// This call is blocking. A new thread is created which will launch
|
// the svc_main() function
|
if ( (rc = StartServiceCtrlDispatcher(ste)) == 0)
|
svc_geterr("StartServiceCtrlDispatcher()");
|
|
return rc; // FALSE if this is not started as a service
|
}
|
|
void svc_geterr(char *str)
|
{
|
char message[PCAP_ERRBUF_SIZE];
|
char string[PCAP_ERRBUF_SIZE];
|
int val;
|
|
val = GetLastError();
|
FormatMessageA(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS |
|
FORMAT_MESSAGE_MAX_WIDTH_MASK,
|
NULL, val, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
|
(LPSTR) string, PCAP_ERRBUF_SIZE, NULL);
|
|
pcap_snprintf(message, PCAP_ERRBUF_SIZE, "%s failed with error %d: %s", str, val, string);
|
|
SOCK_DEBUG_MESSAGE(message);
|
}
|
|
void WINAPI svc_control_handler(DWORD Opcode)
|
{
|
switch(Opcode)
|
{
|
case SERVICE_CONTROL_STOP:
|
//
|
// XXX - is this sufficient to clean up the service?
|
// To be really honest, only the main socket and
|
// such these stuffs are cleared; however the threads
|
// that are running are not stopped.
|
// This can be seen by placing a breakpoint at the
|
// end of svc_main(), in which you will see that is
|
// never reached. However, as soon as you set the
|
// service status to "stopped", the
|
// StartServiceCtrlDispatcher() returns and the main
|
// thread ends. Then, Win32 has a good automatic
|
// cleanup, so that all the threads which are still
|
// running are stopped when the main thread ends.
|
//
|
send_shutdown_notification();
|
|
update_svc_status(SERVICE_STOP_PENDING, 0);
|
break;
|
|
/*
|
Pause and Continue have an usual meaning and they are used just to be able
|
to change the running parameters at run-time. In other words, they act
|
like the SIGHUP signal on UNIX. All the running threads continue to run and
|
they are not paused at all.
|
Particularly,
|
- PAUSE does nothing
|
- CONTINUE re-reads the configuration file and creates the new threads that
|
can be needed according to the new configuration.
|
*/
|
case SERVICE_CONTROL_PAUSE:
|
update_svc_status(SERVICE_PAUSED, 0);
|
break;
|
|
case SERVICE_CONTROL_CONTINUE:
|
update_svc_status(SERVICE_RUNNING, 0);
|
//
|
// Tell the main loop to re-read the configuration.
|
//
|
send_reread_configuration_notification();
|
break;
|
|
case SERVICE_CONTROL_INTERROGATE:
|
// Fall through to send current status.
|
// WARNING: not implemented
|
update_svc_status(SERVICE_RUNNING, 0);
|
MessageBox(NULL, "Not implemented", "warning", MB_OK);
|
break;
|
|
case SERVICE_CONTROL_PARAMCHANGE:
|
//
|
// Tell the main loop to re-read the configuration.
|
//
|
send_reread_configuration_notification();
|
break;
|
}
|
|
// Send current status.
|
return;
|
}
|
|
void WINAPI svc_main(DWORD argc, char **argv)
|
{
|
service_status_handle = RegisterServiceCtrlHandler(PROGRAM_NAME, svc_control_handler);
|
|
if (!service_status_handle)
|
return;
|
|
service_status.dwServiceType = SERVICE_WIN32_OWN_PROCESS | SERVICE_INTERACTIVE_PROCESS;
|
service_status.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_PAUSE_CONTINUE | SERVICE_ACCEPT_PARAMCHANGE;
|
// | SERVICE_ACCEPT_SHUTDOWN ;
|
update_svc_status(SERVICE_RUNNING, 0);
|
|
//
|
// Service requests until we're told to stop.
|
//
|
main_startup();
|
|
//
|
// It returned, so we were told to stop.
|
//
|
update_svc_status(SERVICE_STOPPED, 0);
|
}
|
|
static void
|
update_svc_status(DWORD state, DWORD progress_indicator)
|
{
|
service_status.dwWin32ExitCode = NO_ERROR;
|
service_status.dwCurrentState = state;
|
service_status.dwCheckPoint = progress_indicator;
|
service_status.dwWaitHint = 0;
|
SetServiceStatus(service_status_handle, &service_status);
|
}
|
|
/*
|
sc create rpcapd DisplayName= "Remote Packet Capture Protocol v.0 (experimental)" binpath= "C:\cvsroot\winpcap\wpcap\PRJ\Debug\rpcapd -d -f rpcapd.ini"
|
sc description rpcapd "Allows to capture traffic on this host from a remote machine."
|
*/
|
