networkd: Enable the service watchdog

[network.git] / src / networkd / daemon.c

/*#############################################################################
#                                                                             #
# IPFire.org - A linux based firewall                                         #
# Copyright (C) 2023 IPFire Network Development Team                          #
#                                                                             #
# 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 3 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, see <http://www.gnu.org/licenses/>.       #
#                                                                             #
#############################################################################*/

#include <stdlib.h>

#include <systemd/sd-event.h>

#include "daemon.h"

struct nw_daemon {
	int nrefs;

	// Event Loop
	sd_event* loop;
};

static int nw_daemon_setup_loop(struct nw_daemon* daemon) {
	int r;

	// Fetch a reference to the default event loop
	r = sd_event_default(&daemon->loop);
	if (r < 0) {
		//ERROR("Could not setup event loop: %m\n");
		return 1;
	}

	// Enable the watchdog
	r = sd_event_set_watchdog(daemon->loop, 1);
	if (r < 0) {
		//ERROR("Could not activate watchdog: %m\n");
		return 1;
	}

	return 0;
}

static int nw_daemon_setup(struct nw_daemon* daemon) {
	int r;

	// Setup the event loop
	r = nw_daemon_setup_loop(daemon);
	if (r)
		return r;

	return 0;
}

int nw_daemon_create(struct nw_daemon** daemon) {
	int r;

	struct nw_daemon* d = calloc(1, sizeof(*d));
	if (!d)
		return 1;

	// Initialize reference counter
	d->nrefs = 1;

	// Setup the daemon
	r = nw_daemon_setup(d);
	if (r)
		goto ERROR;

	// Set the reference
	*daemon = d;

	return 0;

ERROR:
	nw_daemon_unref(d);

	return r;
}

static void nw_daemon_free(struct nw_daemon* daemon) {
	if (daemon->loop)
		sd_event_unref(daemon->loop);

	free(daemon);
}

struct nw_daemon* nw_daemon_ref(struct nw_daemon* daemon) {
	daemon->nrefs++;

	return daemon;
}

struct nw_daemon* nw_daemon_unref(struct nw_daemon* daemon) {
	if (--daemon->nrefs > 0)
		return daemon;

	nw_daemon_free(daemon);
	return NULL;
}

/*
	This function contains the main loop of the daemon...
*/
int nw_daemon_run(struct nw_daemon* daemon) {
	int r;

	// Launch the event loop
	r = sd_event_loop(daemon->loop);
	if (r) {
		//ERROR("Could not run the event loop: %m\n");
		return r;
	}

	return 0;
}