src/task.c

   1 /*
   2  *
   3  *  Connection Manager
   4  *
   5  *  Copyright (C) 2007-2010  Intel Corporation. All rights reserved.
   6  *
   7  *  This program is free software; you can redistribute it and/or modify
   8  *  it under the terms of the GNU General Public License version 2 as
   9  *  published by the Free Software Foundation.
  10  *
  11  *  This program is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with this program; if not, write to the Free Software
  18  *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  19  *
  20  */
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include <config.h>
  24 #endif
  25
  26 #include <unistd.h>
  27 #include <stdarg.h>
  28 #include <sys/wait.h>
  29
  30 #include <glib.h>
  31
  32 #include "connman.h"
  33
  34 struct notify_data {
  35         connman_task_notify_t func;
  36         void *data;
  37 };
  38
  39 struct connman_task {
  40         char *path;
  41         pid_t pid;
  42         guint child_watch;
  43         GPtrArray *argv;
  44         GPtrArray *envp;
  45         connman_task_exit_t exit_func;
  46         void *exit_data;
  47         GHashTable *notify;
  48 };
  49
  50 static GHashTable *task_hash = NULL;
  51
  52 static volatile gint task_counter;
  53
  54 static DBusConnection *connection;
  55
  56 static void free_pointer(gpointer data, gpointer user_data)
  57 {
  58         g_free(data);
  59 }
  60
  61 static void free_task(gpointer data)
  62 {
  63         struct connman_task *task = data;
  64
  65         DBG("task %p", task);
  66
  67         g_hash_table_destroy(task->notify);
  68         task->notify = NULL;
  69
  70         if (task->pid > 0)
  71                 kill(task->pid, SIGTERM);
  72
  73         if (task->child_watch > 0)
  74                 g_source_remove(task->child_watch);
  75
  76         g_ptr_array_foreach(task->envp, free_pointer, NULL);
  77         g_ptr_array_free(task->envp, TRUE);
  78
  79         g_ptr_array_foreach(task->argv, free_pointer, NULL);
  80         g_ptr_array_free(task->argv, TRUE);
  81
  82         g_free(task->path);
  83         g_free(task);
  84 }
  85
  86 /**
  87  * connman_task_create:
  88  * @program: name of executable
  89  *
  90  * Allocate a new task of given #program
  91  *
  92  * Returns: a newly-allocated #connman_task structure
  93  */
  94 struct connman_task *connman_task_create(const char *program)
  95 {
  96         struct connman_task *task;
  97         gint counter;
  98         char *str;
  99
 100         DBG("");
 101
 102         task = g_try_new0(struct connman_task, 1);
 103         if (task == NULL)
 104                 return NULL;
 105
 106         counter = g_atomic_int_exchange_and_add(&task_counter, 1);
 107
 108         task->path = g_strdup_printf("/task/%d", counter);
 109         task->pid = -1;
 110
 111         task->argv = g_ptr_array_new();
 112         task->envp = g_ptr_array_new();
 113
 114         str = g_strdup(program);
 115         g_ptr_array_add(task->argv, str);
 116
 117         task->notify = g_hash_table_new_full(g_str_hash, g_str_equal,
 118                                                         g_free, g_free);
 119
 120         DBG("task %p", task);
 121
 122         g_hash_table_insert(task_hash, task->path, task);
 123
 124         return task;
 125 }
 126
 127 /**
 128  * connman_task_destory:
 129  * @task: task structure
 130  *
 131  * Remove and destory #task
 132  */
 133 void connman_task_destroy(struct connman_task *task)
 134 {
 135         DBG("task %p", task);
 136
 137         g_hash_table_remove(task_hash, task->path);
 138 }
 139
 140 /**
 141  * connman_task_get_path:
 142  * @task: task structure
 143  *
 144  * Get object path
 145  */
 146 const char *connman_task_get_path(struct connman_task *task)
 147 {
 148         return task->path;
 149 }
 150
 151 /**
 152  * connman_task_add_argument:
 153  * @task: task structure
 154  * @name: argument name
 155  * @format: format string
 156  * @Varargs: list of arguments
 157  *
 158  * Add a new command line argument
 159  */
 160 int connman_task_add_argument(struct connman_task *task,
 161                                 const char *name, const char *format, ...)
 162 {
 163         va_list ap;
 164         char *str;
 165
 166         DBG("task %p arg %s", task, name);
 167
 168         if (name == NULL)
 169                 return -EINVAL;
 170
 171         str = g_strdup(name);
 172         g_ptr_array_add(task->argv, str);
 173
 174         va_start(ap, format);
 175
 176         if (format != NULL) {
 177                 str = g_strdup_vprintf(format, ap);
 178                 g_ptr_array_add(task->argv, str);
 179         }
 180
 181         va_end(ap);
 182
 183         return 0;
 184 }
 185
 186 /**
 187  * connman_task_add_variable:
 188  * @task: task structure
 189  * @key: variable name
 190  * @format: format string
 191  * @Varargs: list of arguments
 192  *
 193  * Add a new environment variable
 194  */
 195 int connman_task_add_variable(struct connman_task *task,
 196                                 const char *key, const char *format, ...)
 197 {
 198         va_list ap;
 199         char *str, *val;
 200
 201         DBG("task %p key %s", task, key);
 202
 203         if (key == NULL)
 204                 return -EINVAL;
 205
 206         va_start(ap, format);
 207
 208         val = g_strdup_vprintf(format, ap);
 209         str = g_strdup_printf("%s=%s", key, format ? format : "");
 210         g_ptr_array_add(task->envp, str);
 211         g_free(val);
 212
 213         va_end(ap);
 214
 215         return 0;
 216 }
 217
 218 /**
 219  * connman_task_set_notify:
 220  * @task: task structure
 221  * @member: notifcation method name
 222  * @function: notification callback
 223  * @user_data: optional notification user data
 224  *
 225  * Set notification handler for #member
 226  */
 227 int connman_task_set_notify(struct connman_task *task, const char *member,
 228                         connman_task_notify_t function, void *user_data)
 229 {
 230         struct notify_data *notify;
 231
 232         DBG("task %p", task);
 233
 234         notify = g_try_new0(struct notify_data, 1);
 235         if (notify == NULL)
 236                 return -ENOMEM;
 237
 238         notify->func = function;
 239         notify->data = user_data;
 240
 241         g_hash_table_insert(task->notify, g_strdup(member), notify);
 242
 243         return 0;
 244 }
 245
 246 static void task_died(GPid pid, gint status, gpointer user_data)
 247 {
 248         struct connman_task *task = user_data;
 249
 250         if (WIFEXITED(status))
 251                 DBG("task %p exit status %d", task, WEXITSTATUS(status));
 252         else
 253                 DBG("task %p signal %d", task, WTERMSIG(status));
 254
 255         g_spawn_close_pid(pid);
 256         task->pid = -1;
 257
 258         task->child_watch = 0;
 259
 260         if (task->exit_func)
 261                 task->exit_func(task, task->exit_data);
 262 }
 263
 264 static void task_setup(gpointer user_data)
 265 {
 266         struct connman_task *task = user_data;
 267
 268         DBG("task %p", task);
 269 }
 270
 271 /**
 272  * connman_task_run:
 273  * @task: task structure
 274  * @function: exit callback
 275  * @user_data: optional exit user data
 276  * @fd: optional spawn with pipe
 277  *
 278  * Execute program specified by #task
 279  */
 280 int connman_task_run(struct connman_task *task,
 281                         connman_task_exit_t function, void *user_data,
 282                         int *stdin_fd, int *stdout_fd, int *stderr_fd)
 283 {
 284         GSpawnFlags flags = G_SPAWN_DO_NOT_REAP_CHILD;
 285         gboolean result;
 286         char **argv, **envp;
 287
 288         DBG("task %p", task);
 289
 290         if (task->pid > 0)
 291                 return -EALREADY;
 292
 293         if (stdout_fd == NULL)
 294                 flags |= G_SPAWN_STDOUT_TO_DEV_NULL;
 295
 296         if (stderr_fd == NULL)
 297                 flags |= G_SPAWN_STDERR_TO_DEV_NULL;
 298
 299         task->exit_func = function;
 300         task->exit_data = user_data;
 301
 302         if (g_ptr_array_index(task->argv, task->argv->len - 1) != NULL)
 303                 g_ptr_array_add(task->argv, NULL);
 304
 305         if (task->envp->len == 0 || g_ptr_array_index(task->envp,
 306                                         task->envp->len - 1) != NULL) {
 307                 if (g_hash_table_size(task->notify) > 0) {
 308                         const char *busname;
 309                         char *str;
 310
 311                         busname = dbus_bus_get_unique_name(connection);
 312                         str = g_strdup_printf("CONNMAN_BUSNAME=%s", busname);
 313                         g_ptr_array_add(task->envp, str);
 314
 315                         str = g_strdup_printf("CONNMAN_INTERFACE=%s",
 316                                                 CONNMAN_TASK_INTERFACE);
 317                         g_ptr_array_add(task->envp, str);
 318
 319                         str = g_strdup_printf("CONNMAN_PATH=%s", task->path);
 320                         g_ptr_array_add(task->envp, str);
 321                 }
 322
 323                 g_ptr_array_add(task->envp, NULL);
 324         }
 325
 326         argv = (char **) task->argv->pdata;
 327         envp = (char **) task->envp->pdata;
 328
 329         result = g_spawn_async_with_pipes(NULL, argv, envp, flags,
 330                                         task_setup, task, &task->pid,
 331                                         stdin_fd, stdout_fd, stderr_fd, NULL);
 332         if (result == FALSE) {
 333                 connman_error("Failed to spawn %s", argv[0]);
 334                 return -EIO;
 335         }
 336
 337         task->child_watch = g_child_watch_add(task->pid, task_died, task);
 338
 339         return 0;
 340 }
 341
 342 /**
 343  * connman_task_stop:
 344  * @task: task structure
 345  *
 346  * Stop program specified by #task
 347  */
 348 int connman_task_stop(struct connman_task *task)
 349 {
 350         DBG("task %p", task);
 351
 352         if (task->pid > 0)
 353                 kill(task->pid, SIGTERM);
 354
 355         return 0;
 356 }
 357
 358 static DBusHandlerResult task_filter(DBusConnection *connection,
 359                                         DBusMessage *message, void *user_data)
 360 {
 361         struct connman_task *task;
 362         struct notify_data *notify;
 363         const char *path, *member;
 364
 365         if (dbus_message_get_type(message) != DBUS_MESSAGE_TYPE_METHOD_CALL)
 366                 return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
 367
 368         if (dbus_message_has_interface(message,
 369                                         CONNMAN_TASK_INTERFACE) == FALSE)
 370                 return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
 371
 372         path = dbus_message_get_path(message);
 373         if (path == NULL)
 374                 return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
 375
 376         task = g_hash_table_lookup(task_hash, path);
 377         if (task == NULL)
 378                 return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
 379
 380         if (dbus_message_get_no_reply(message) == FALSE) {
 381                 DBusMessage *reply;
 382                 dbus_bool_t result;
 383
 384                 reply = dbus_message_new_method_return(message);
 385                 if (reply == NULL)
 386                         return DBUS_HANDLER_RESULT_NEED_MEMORY;
 387
 388                 result = dbus_connection_send(connection, reply, NULL);
 389
 390                 dbus_message_unref(reply);
 391         }
 392
 393         member = dbus_message_get_member(message);
 394         if (member == NULL)
 395                 return DBUS_HANDLER_RESULT_HANDLED;
 396
 397         notify = g_hash_table_lookup(task->notify, member);
 398         if (notify == NULL)
 399                 return DBUS_HANDLER_RESULT_HANDLED;
 400
 401         if (notify->func)
 402                 notify->func(task, message, notify->data);
 403
 404         return DBUS_HANDLER_RESULT_HANDLED;
 405 }
 406
 407 static const char *task_rule = "type=method_call"
 408                                         ",interface=" CONNMAN_TASK_INTERFACE;
 409
 410 int __connman_task_init(void)
 411 {
 412         DBG("");
 413
 414         connection = connman_dbus_get_connection();
 415
 416         dbus_connection_add_filter(connection, task_filter, NULL, NULL);
 417
 418         g_atomic_int_set(&task_counter, 0);
 419
 420         task_hash = g_hash_table_new_full(g_str_hash, g_str_equal,
 421                                                         NULL, free_task);
 422
 423         dbus_bus_add_match(connection, task_rule, NULL);
 424         dbus_connection_flush(connection);
 425
 426         return 0;
 427 }
 428
 429 void __connman_task_cleanup(void)
 430 {
 431         DBG("");
 432
 433         dbus_bus_remove_match(connection, task_rule, NULL);
 434         dbus_connection_flush(connection);
 435
 436         g_hash_table_destroy(task_hash);
 437         task_hash = NULL;
 438
 439         dbus_connection_remove_filter(connection, task_filter, NULL);
 440
 441         dbus_connection_unref(connection);
 442 }