155 lines
3.7 KiB
C
155 lines
3.7 KiB
C
|
/*
|
||
|
* Copyright (C) 2003-2010 The Music Player Daemon Project
|
||
|
* http://www.musicpd.org
|
||
|
*
|
||
|
* This program is free software; you can redistribute it and/or modify
|
||
|
* it under the terms of the GNU General Public License as published by
|
||
|
* the Free Software Foundation; either version 2 of the License, or
|
||
|
* (at your option) any later version.
|
||
|
*
|
||
|
* This program is distributed in the hope that it will be useful,
|
||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
* GNU General Public License for more details.
|
||
|
*
|
||
|
* You should have received a copy of the GNU General Public License along
|
||
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
||
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
||
|
*/
|
||
|
|
||
|
#include "config.h"
|
||
|
#include "main.h"
|
||
|
|
||
|
#ifdef WIN32
|
||
|
|
||
|
#include "event_pipe.h"
|
||
|
|
||
|
#include <glib.h>
|
||
|
|
||
|
#define WINVER 0x0501
|
||
|
#include <windows.h>
|
||
|
|
||
|
static int service_argc;
|
||
|
static char **service_argv;
|
||
|
static char service_name[] = "";
|
||
|
static BOOL ignore_console_events;
|
||
|
static SERVICE_STATUS_HANDLE service_handle;
|
||
|
|
||
|
static void WINAPI
|
||
|
service_main(DWORD argc, CHAR *argv[]);
|
||
|
|
||
|
static SERVICE_TABLE_ENTRY service_registry[] = {
|
||
|
{service_name, service_main},
|
||
|
{NULL, NULL}
|
||
|
};
|
||
|
|
||
|
static void
|
||
|
service_notify_status(DWORD status_code)
|
||
|
{
|
||
|
SERVICE_STATUS current_status;
|
||
|
|
||
|
current_status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
|
||
|
current_status.dwControlsAccepted = status_code == SERVICE_START_PENDING
|
||
|
? 0
|
||
|
: SERVICE_ACCEPT_SHUTDOWN | SERVICE_ACCEPT_STOP;
|
||
|
|
||
|
current_status.dwCurrentState = status_code;
|
||
|
current_status.dwWin32ExitCode = NO_ERROR;
|
||
|
current_status.dwCheckPoint = 0;
|
||
|
current_status.dwWaitHint = 1000;
|
||
|
|
||
|
SetServiceStatus(service_handle, ¤t_status);
|
||
|
}
|
||
|
|
||
|
static DWORD WINAPI
|
||
|
service_dispatcher(G_GNUC_UNUSED DWORD control, G_GNUC_UNUSED DWORD event_type,
|
||
|
G_GNUC_UNUSED void *event_data, G_GNUC_UNUSED void *context)
|
||
|
{
|
||
|
switch (control) {
|
||
|
case SERVICE_CONTROL_SHUTDOWN:
|
||
|
case SERVICE_CONTROL_STOP:
|
||
|
event_pipe_emit(PIPE_EVENT_SHUTDOWN);
|
||
|
return NO_ERROR;
|
||
|
default:
|
||
|
return NO_ERROR;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void WINAPI
|
||
|
service_main(G_GNUC_UNUSED DWORD argc, G_GNUC_UNUSED CHAR *argv[])
|
||
|
{
|
||
|
DWORD error_code;
|
||
|
gchar* error_message;
|
||
|
|
||
|
service_handle =
|
||
|
RegisterServiceCtrlHandlerEx(service_name,
|
||
|
service_dispatcher, NULL);
|
||
|
|
||
|
if (service_handle == 0) {
|
||
|
error_code = GetLastError();
|
||
|
error_message = g_win32_error_message(error_code);
|
||
|
g_error("RegisterServiceCtrlHandlerEx() failed: %s",
|
||
|
error_message);
|
||
|
}
|
||
|
|
||
|
service_notify_status(SERVICE_START_PENDING);
|
||
|
mpd_main(service_argc, service_argv);
|
||
|
service_notify_status(SERVICE_STOPPED);
|
||
|
}
|
||
|
|
||
|
static BOOL WINAPI
|
||
|
console_handler(DWORD event)
|
||
|
{
|
||
|
switch (event) {
|
||
|
case CTRL_C_EVENT:
|
||
|
case CTRL_CLOSE_EVENT:
|
||
|
if (!ignore_console_events)
|
||
|
event_pipe_emit(PIPE_EVENT_SHUTDOWN);
|
||
|
return TRUE;
|
||
|
default:
|
||
|
return FALSE;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
int win32_main(int argc, char *argv[])
|
||
|
{
|
||
|
DWORD error_code;
|
||
|
gchar* error_message;
|
||
|
|
||
|
service_argc = argc;
|
||
|
service_argv = argv;
|
||
|
|
||
|
if (StartServiceCtrlDispatcher(service_registry))
|
||
|
return 0; /* run as service successefully */
|
||
|
|
||
|
error_code = GetLastError();
|
||
|
if (error_code == ERROR_FAILED_SERVICE_CONTROLLER_CONNECT) {
|
||
|
/* running as console app */
|
||
|
SetConsoleTitle("Music Player Daemon");
|
||
|
ignore_console_events = TRUE;
|
||
|
SetConsoleCtrlHandler(console_handler, TRUE);
|
||
|
return mpd_main(argc, argv);
|
||
|
}
|
||
|
|
||
|
error_message = g_win32_error_message(error_code);
|
||
|
g_error("StartServiceCtrlDispatcher() failed: %s", error_message);
|
||
|
}
|
||
|
|
||
|
void win32_app_started()
|
||
|
{
|
||
|
if (service_handle != 0)
|
||
|
service_notify_status(SERVICE_RUNNING);
|
||
|
else
|
||
|
ignore_console_events = FALSE;
|
||
|
}
|
||
|
|
||
|
void win32_app_stopping()
|
||
|
{
|
||
|
if (service_handle != 0)
|
||
|
service_notify_status(SERVICE_STOP_PENDING);
|
||
|
else
|
||
|
ignore_console_events = TRUE;
|
||
|
}
|
||
|
|
||
|
#endif
|