static DWORD scmdatabase_load_services(struct scmdatabase *db)
{
DWORD err;
int i;
for (i = 0; TRUE; i++)
{
WCHAR szName[MAX_SERVICE_NAME];
struct service_entry *entry;
HKEY hServiceKey;
err = RegEnumKeyW(db->root_key, i, szName, MAX_SERVICE_NAME);
if (err == ERROR_NO_MORE_ITEMS)
break;
if (err != 0)
{
WINE_ERR("Error %d reading key %d name - skipping\n", err, i);
continue;
}
err = service_create(szName, &entry);
if (err != ERROR_SUCCESS)
break;
WINE_TRACE("Loading service %s\n", wine_dbgstr_w(szName));
err = RegOpenKeyExW(db->root_key, szName, 0, KEY_READ, &hServiceKey);
if (err == ERROR_SUCCESS)
{
err = load_service_config(hServiceKey, entry);
RegCloseKey(hServiceKey);
}
if (err != ERROR_SUCCESS)
{
WINE_ERR("Error %d reading registry key for service %s - skipping\n", err, wine_dbgstr_w(szName));
free_service_entry(entry);
continue;
}
if (entry->config.dwServiceType == 0)
{
/* Maybe an application only wrote some configuration in the service key. Continue silently */
WINE_TRACE("Even the service type not set for service %s - skipping\n", wine_dbgstr_w(szName));
free_service_entry(entry);
continue;
}
if (!validate_service_config(entry))
{
WINE_ERR("Invalid configuration of service %s - skipping\n", wine_dbgstr_w(szName));
free_service_entry(entry);
continue;
}
entry->status.dwServiceType = entry->config.dwServiceType;
entry->db = db;
list_add_tail(&db->services, &entry->entry);
}
return ERROR_SUCCESS;
}
/* ---------------------------------------------------------------------[<]-
Function: service_begin
Synopsis: depending on arguments (from the command line):
-i: install service (windows)
-u: remove service (windows)
-d: runs service in console mode (windows)
-h: basic help information (windows)
-d: runs service in background (UNIX / Linux)
if no arguments, the service is actually started.
NOTE: with Windows, the working directory is set to the one where the
binary program stands.
Returns: 0 is everything is OK, negative error code otherwise
---------------------------------------------------------------------[>]-*/
int
service_begin (
int argc,
char **argv,
SMT_AGENTS_INIT_FCT *init_fct,
SMT_AGENTS_TERM_FCT *term_fct,
const char *appl_version)
{
int
action;
int
rc = 0;
#if (defined(WIN32))
static char
buffer [LINE_MAX];
char
*p_char;
SERVICE_TABLE_ENTRY
dispatch_table [] = {
{ NULL, (LPSERVICE_MAIN_FUNCTION) service_main },
{ NULL, NULL }
};
/* Change to the correct working directory, where config file stands */
GetModuleFileName (NULL, buffer, LINE_MAX);
if ((p_char = strrchr (buffer, '\\')) != NULL)
*p_char = '\0';
SetCurrentDirectory (buffer);
#endif
rc = init_resources (argv[0], appl_version, init_fct, term_fct);
if (rc != 0)
return (1);
ASSERT (application_config); /* init_resources post condition */
if (load_service_config (application_config) != 0)
{
free_resources ();
return (1);
}
ASSERT (service_trace_file); /* load_service_config postcondition */
console_set_mode (CONSOLE_DATETIME);
console_capture (service_trace_file, 'a');
#if (defined(WIN32))
dispatch_table [0].lpServiceName = service_name;
win_version = get_windows_version ();
#endif
action = parse_command_line (argc, argv);
if (action == ACTION_HELP)
{
puts (USAGE);
}
#if (defined(WIN32))
else
if (action == ACTION_INSTALL)
{
if (win_version == WINDOWS_95)
set_win95_service (TRUE);
else
install_service ();
}
else
if (action == ACTION_UNINSTALL)
{
if (win_version == WINDOWS_95)
set_win95_service (FALSE);
else
remove_service ();
}
else
if (action == ACTION_CONSOLE)
{
console_mode = TRUE;
console_service (argc, argv);
}
else
if (action == ACTION_NOARG)
{
console_send (NULL, FALSE);
if (win_version == WINDOWS_95)
{
hide_window ();
console_mode = TRUE;
console_service (argc, argv);
}
else
if (win_version == WINDOWS_NT_3X
|| win_version == WINDOWS_NT_4
|| win_version == WINDOWS_2000)
{
log_printf ("%s: initialising service ...", application_name);
if (!StartServiceCtrlDispatcher (dispatch_table))
add_to_message_log ("StartServiceCtrlDispatcher failed");
}
}
#elif (defined(__UNIX__))
else
if (action == ACTION_BACKGROUND)
{
const char
*background_args [] = { "-s", NULL };
log_printf ("Moving into the background");
if (process_server (NULL, NULL, argc, argv, background_args) != 0)
{
log_printf ("Backgrounding failed. Giving up.");
rc = -1;
}
else
action = ACTION_NOARG;
}
if (action == ACTION_NOARG)
{
rc = smt_init ();
if (!rc && (init_fct != NULL))
rc = (*init_fct)(application_config);
if (!rc)
smt_exec_full ();
if (term_fct != NULL)
(*term_fct)();
smt_term ();
}
#endif
else
if (action == ACTION_ERROR)
puts (USAGE);
free_resources ();
return rc;
}