* limitations under the License.
*/
+#include <signal.h>
+#include <errno.h>
+#include <sys/wait.h>
#include "stc-manager.h"
-#include "stc-statistics.h"
-#include "stc-restriction.h"
#include "stc-manager-gdbus.h"
#include "stc-db.h"
#include "counter.h"
#include "table-restrictions.h"
#include "helper-cgroup.h"
#include "helper-nfacct-rule.h"
-#include "stc-monitor.h"
-#include "stc-manager-plugin.h"
+#include "helper-iptables.h"
+#include "helper-inotify.h"
+#include "stc-firewall.h"
+#include "stc-manager-plugin-appstatus.h"
+#include "stc-manager-plugin-exception.h"
+#include "stc-manager-plugin-procfs.h"
+#include "stc-manager-plugin-tether.h"
+#include "stc-manager-plugin-pcap.h"
+#include "stc-manager-plugin-monitor.h"
+#include "stc-manager-plugin-firewall.h"
+
+#define BUF_SIZE_FOR_ERR 100
static stc_s *g_stc = NULL;
+static gboolean g_is_fail_exit = FALSE;
+
+static gboolean __validate_ident(const char *ident)
+{
+ unsigned int i;
+
+ if (!ident)
+ return FALSE;
+
+ for (i = 0; i < strlen(ident); ++i)
+ if (!g_ascii_isprint(ident[i]))
+ return FALSE;
+
+ return TRUE;
+}
+
+static void __stc_inotify_handler(struct inotify_event *event, const char *ident)
+{
+ if (!ident)
+ return;
+
+ if (!__validate_ident(ident)) {
+ STC_LOGE("Invalid ident [%s]", ident);
+ return;
+ }
+
+ if (!g_strcmp0(ident, INFO_CONFIG))
+ stc_util_update_log_state();
+}
static void __stc_manager_deinit(void)
{
return;
}
- stc_monitor_deinit();
stc_deinit_db_guard();
stc_db_deinitialize();
+
+ iptables_flush_chains();
+ iptables_deinit();
+
stc_manager_gdbus_deinit((gpointer)g_stc);
- stc_manager_plugin_deinit();
+
+ stc_plugin_appstatus_deinit();
+ stc_plugin_exception_deinit();
+ stc_plugin_procfs_deinit();
+ stc_plugin_tether_deinit();
+ stc_plugin_pcap_deinit();
+ stc_plugin_monitor_deinit();
+ stc_plugin_firewall_deinit();
+
+ inotify_deregister(INFO_STORAGE_DIR);
+ inotify_deinitialize();
STC_LOGI("stc manager deinitialized");
FREE(g_stc);
__STC_LOG_FUNC_EXIT__;
}
+void __stc_manager_stop_with_fail(void)
+{
+ STC_LOGI("plugin needs stc-manager to exit");
+
+ g_is_fail_exit = TRUE;
+
+ stc_stop_manager();
+}
+
static stc_s *__stc_manager_init(void)
{
__STC_LOG_FUNC_ENTER__;
stc_s *stc;
+ stc_error_e err = STC_ERROR_NONE;
stc = MALLOC0(stc_s, 1);
if (!stc) {
- STC_LOGE("Failed to allocate memory for manager structure");
- return NULL;
+ STC_LOGE("Failed to allocate memory for manager structure"); //LCOV_EXCL_LINE
+ return NULL; //LCOV_EXCL_LINE
}
g_stc = stc;
+ stc_util_initialize_config();
+
+ inotify_initialize();
+ inotify_register(INFO_STORAGE_DIR, __stc_inotify_handler);
+
cgroup_set_release_agent(NET_CLS_SUBSYS, NET_RELEASE_AGENT);
- EXEC(STC_ERROR_NONE, stc_db_initialize());
+ err = stc_db_initialize();
+ if (err != STC_ERROR_NONE) {
+ STC_LOGD("Failed to initialize stc db"); //LCOV_EXCL_LINE
+ return NULL; //LCOV_EXCL_LINE
+ }
+
+ g_stc->ondemand_mode = TRUE;
+
+ stc_plugin_appstatus_init();
+ stc_plugin_exception_init();
+ stc_plugin_procfs_init();
+ stc_plugin_tether_init();
+ if (stc_plugin_pcap_init() == STC_ERROR_NONE)
+ g_stc->ondemand_mode = FALSE;
+ if (stc_plugin_monitor_init(__stc_manager_stop_with_fail) == STC_ERROR_NONE)
+ g_stc->ondemand_mode = FALSE;
+ stc_plugin_firewall_init();
+
+ stc_plugin_procfs_load_pid();
- stc_monitor_init();
stc_manager_gdbus_init((gpointer)stc);
- stc_manager_plugin_init();
STC_LOGI("stc manager initialized");
__STC_LOG_FUNC_EXIT__;
return stc;
+}
-handle_error:
- STC_LOGD("Failed to initialize stc manager");
- __stc_manager_deinit();
- return NULL;
+static gboolean __stc_timer_expired(gpointer data)
+{
+ if (g_stc->keep_alive) {
+ g_stc->keep_alive = FALSE;
+ return TRUE;
+ }
+
+ g_main_loop_quit(g_stc->main_loop);
+
+ return FALSE;
}
-stc_s *stc_get_manager(void)
+API stc_s *stc_get_manager(void)
{
return g_stc;
}
+void stc_stop_manager(void)
+{
+ if (g_stc && g_stc->main_loop)
+ g_main_loop_quit(g_stc->main_loop);
+}
+
+int stc_commit_iptables(char *cmd, int *err_num, char **err_str)
+{
+ pid_t pid = 0;
+ int status = 0;
+ int ret = 0;
+ char err_buf[BUF_SIZE_FOR_ERR] = { 0, };
+ gchar **args = NULL;
+
+ if (cmd == NULL) {
+ STC_LOGE("Invalid arguments");
+ return STC_ERROR_INVALID_PARAMETER;
+ }
+
+ args = g_strsplit_set(cmd, " ", -1);
+
+ errno = 0;
+ pid = fork();
+
+ if (pid == 0) {
+ errno = 0;
+
+ if (!g_strcmp0(args[1], STC_CMD_INSERT)) {
+ STC_LOGE("Invalid arguments");
+ g_strfreev(args);
+ exit(-1);
+ }
+
+ if (execv(args[0], args) == -1) {
+ STC_LOGE("Failed to execute [%s]", *err_str);
+ g_strfreev(args);
+ exit(-1);
+ }
+ } else if (pid > 0) {
+ if (waitpid(pid, &status, 0) == -1)
+ STC_LOGD("wait pid [%u] status [%d] ", pid, status);
+
+ if (WIFEXITED(status)) {
+ ret = WEXITSTATUS(status);
+ STC_LOGD("exited, status [%d]", status);
+ } else if (WIFSIGNALED(status)) {
+ STC_LOGD("killed by signal [%d]", WTERMSIG(status));
+ } else if (WIFSTOPPED(status)) {
+ STC_LOGD("stopped by signal [%d]", WSTOPSIG(status));
+ } else if (WIFCONTINUED(status)) {
+ STC_LOGD("continued");
+ }
+
+ *err_num = ret;
+ *err_str = strerror_r(ret, err_buf, BUF_SIZE_FOR_ERR);
+ STC_LOGD("return err_num [%d] err_str [%s]", *err_num, *err_str);
+
+ g_strfreev(args);
+ if (ret == 0)
+ return STC_ERROR_NONE;
+ else
+ return STC_ERROR_FAIL;
+ }
+
+ *err_num = errno;
+ *err_str = strerror_r(errno, err_buf, BUF_SIZE_FOR_ERR);
+ STC_LOGD("Failed to fork [%d:%s]", *err_num, *err_str);
+
+ g_strfreev(args);
+ return STC_ERROR_FAIL;
+}
+
+void stc_set_keep_alive(gboolean keep_alive)
+{
+ g_stc->keep_alive = keep_alive;
+}
+
gint32 main(gint32 argc, gchar *argv[])
{
GMainLoop *main_loop = NULL;
- gint32 ret = -1;
STC_LOGI("Smart Traffic Control Manager");
+#ifdef TIZEN_GTESTS
+ setenv("GCOV_PREFIX", "/tmp/daemon", 1);
+#endif
+
+/*
if (daemon(0, 0) != 0)
- STC_LOGE("Can't start daemon");
+ STC_LOGE("Can't start daemon"); //LCOV_EXCL_LINE
+*/
/* Initialize required subsystems */
#if !GLIB_CHECK_VERSION(2, 35, 0)
if (!g_stc)
goto fail;
+ if (g_is_fail_exit == TRUE)
+ goto fail;
+
/* Crate the GLIB main loop */
main_loop = g_main_loop_new(NULL, FALSE);
g_stc->main_loop = main_loop;
+ if (g_stc->ondemand_mode) {
+ g_stc->timer = g_timeout_add_seconds(10, __stc_timer_expired, NULL);
+ g_stc->keep_alive = FALSE;
+ }
+
/* Run the main loop */
g_main_loop_run(main_loop);
- ret = 0;
-
fail:
__stc_manager_deinit();
if (main_loop)
g_main_loop_unref(main_loop);
- return ret;
+ if (g_is_fail_exit == TRUE)
+ exit(-1);
+
+ return 0;
}
projects / platform / core / connectivity / stc-manager.git / blobdiff