Changes in uspace/srv/taskmon/taskmon.c [9d58539:1c635d6] in mainline

File:

• uspace/srv/taskmon/taskmon.c (modified) (5 diffs)

uspace/srv/taskmon/taskmon.c

@@ -1 +1 @@
 /*
- * Copyright (c) 2010 Jiri Svoboda
+ * Copyright (c) 2013 Jiri Svoboda
  * All rights reserved.
  *
@@ -40 +40 @@
 #include <sys/typefmt.h>
 #include <task.h>
-#include <event.h>
+#include <ipc/corecfg.h>
+#include <loc.h>
 #include <macros.h>
 #include <errno.h>
@@ -47 +48 @@
 #define NAME  "taskmon"
 
-static void fault_event(ipc_callid_t callid, ipc_call_t *call)
+static bool write_core_files;
+
+static void corecfg_client_conn(ipc_callid_t , ipc_call_t *, void *);
+
+static void fault_event(ipc_callid_t callid, ipc_call_t *call, void *arg)
 {
         const char *fname;
         char *s_taskid;
+        char *dump_fname;
         int rc;
 
@@ -69 +75 @@
         fname = "/app/taskdump";
 
-#ifdef CONFIG_WRITE_CORE_FILES
-        char *dump_fname;
+        if (write_core_files) {
+                if (asprintf(&dump_fname, "/data/core%" PRIu64, taskid) < 0) {
+                        printf("Memory allocation failed.\n");
+                        return;
+                }
 
-        if (asprintf(&dump_fname, "/data/core%" PRIu64, taskid) < 0) {
-                printf("Memory allocation failed.\n");
-                return;
+                printf(NAME ": Executing %s -c %s -t %s\n", fname, dump_fname, s_taskid);
+                rc = task_spawnl(NULL, NULL, fname, fname, "-c", dump_fname, "-t", s_taskid,
+                    NULL);
+        } else {
+                printf(NAME ": Executing %s -t %s\n", fname, s_taskid);
+                rc = task_spawnl(NULL, NULL, fname, fname, "-t", s_taskid, NULL);
         }
 
-        printf(NAME ": Executing %s -c %s -t %s\n", fname, dump_fname, s_taskid);
-        rc = task_spawnl(NULL, fname, fname, "-c", dump_fname, "-t", s_taskid,
-            NULL);
-#else
-        printf(NAME ": Executing %s -t %s\n", fname, s_taskid);
-        rc = task_spawnl(NULL, fname, fname, "-t", s_taskid, NULL);
-#endif
         if (rc != EOK) {
                 printf("%s: Error spawning %s (%s).\n", NAME, fname,
                     str_error(rc));
+        }
+}
+
+static void corecfg_get_enable_srv(ipc_callid_t iid, ipc_call_t *icall)
+{
+        async_answer_1(iid, EOK, write_core_files);
+}
+
+static void corecfg_set_enable_srv(ipc_callid_t iid, ipc_call_t *icall)
+{
+        write_core_files = IPC_GET_ARG1(*icall);
+        async_answer_0(iid, EOK);
+}
+
+static void corecfg_client_conn(ipc_callid_t iid, ipc_call_t *icall, void *arg)
+{
+        /* Accept the connection */
+        async_answer_0(iid, EOK);
+
+        while (true) {
+                ipc_call_t call;
+                ipc_callid_t callid = async_get_call(&call);
+                sysarg_t method = IPC_GET_IMETHOD(call);
+
+                if (!method) {
+                        /* The other side has hung up */
+                        async_answer_0(callid, EOK);
+                        return;
+                }
+
+                switch (method) {
+                case CORECFG_GET_ENABLE:
+                        corecfg_get_enable_srv(callid, &call);
+                        break;
+                case CORECFG_SET_ENABLE:
+                        corecfg_set_enable_srv(callid, &call);
+                        break;
+                }
         }
 }
@@ -94 +137 @@
         printf("%s: Task Monitoring Service\n", NAME);
         
-        if (event_subscribe(EVENT_FAULT, 0) != EOK) {
+#ifdef CONFIG_WRITE_CORE_FILES
+        write_core_files = true;
+#else
+        write_core_files = false;
+#endif
+        if (async_event_subscribe(EVENT_FAULT, fault_event, NULL) != EOK) {
                 printf("%s: Error registering fault notifications.\n", NAME);
                 return -1;
         }
         
-        async_set_interrupt_received(fault_event);
+        async_set_client_connection(corecfg_client_conn);
+        
+        int rc = loc_server_register(NAME);
+        if (rc != EOK) {
+                printf("%s: Failed registering server (%d).\n",
+                    NAME, rc);
+                return -1;
+        }
+        
+        service_id_t sid;
+        rc = loc_service_register(SERVICE_NAME_CORECFG, &sid);
+        if (rc != EOK) {
+                printf("%s: Failed registering service (%d).\n",
+                    NAME, rc);
+                return -1;
+        }
+        
         task_retval(0);
         async_manager();