src/timeserver.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 <errno.h>
  27
  28 #include <glib.h>
  29 #include <stdlib.h>
  30 #include <gweb/gresolv.h>
  31
  32 #include "connman.h"
  33
  34 static GSList *driver_list = NULL;
  35 static GHashTable *server_hash = NULL;
  36
  37 static char **system_timeservers = NULL;
  38 static char **timeservers = NULL;
  39
  40 static GResolv *resolv = NULL;
  41 static int resolv_id = 0;
  42 static volatile int count;
  43
  44 static void resolv_debug(const char *str, void *data)
  45 {
  46         connman_info("%s: %s\n", (const char *) data, str);
  47 }
  48 static void save_timeservers(char **servers)
  49 {
  50         GKeyFile *keyfile;
  51         int cnt;
  52
  53         keyfile = __connman_storage_load_global();
  54         if (keyfile == NULL)
  55                 keyfile = g_key_file_new();
  56
  57         for (cnt = 0; servers != NULL && servers[cnt] != NULL; cnt++);
  58
  59         g_key_file_set_string_list(keyfile, "global", "Timeservers",
  60                            (const gchar **)servers, cnt);
  61
  62         __connman_storage_save_global(keyfile);
  63
  64         g_key_file_free(keyfile);
  65
  66         return;
  67 }
  68
  69 static char **load_timeservers()
  70 {
  71         GKeyFile *keyfile;
  72         GError *error = NULL;
  73         char **servers = NULL;
  74
  75         keyfile = __connman_storage_load_global();
  76         if (keyfile == NULL)
  77                 return NULL;
  78
  79         servers = g_key_file_get_string_list(keyfile, "global",
  80                                                 "Timeservers", NULL, &error);
  81         if (error) {
  82                 DBG("Error loading timeservers: %s", error->message);
  83                 g_error_free(error);
  84         }
  85
  86         g_key_file_free(keyfile);
  87
  88         return servers;
  89 }
  90
  91 static gint compare_priority(gconstpointer a, gconstpointer b)
  92 {
  93         const struct connman_timeserver_driver *driver1 = a;
  94         const struct connman_timeserver_driver *driver2 = b;
  95
  96         return driver2->priority - driver1->priority;
  97 }
  98
  99 /**
 100  * connman_timeserver_driver_register:
 101  * @driver: timeserver driver definition
 102  *
 103  * Register a new timeserver driver
 104  *
 105  * Returns: %0 on success
 106  */
 107 int connman_timeserver_driver_register(struct connman_timeserver_driver *driver)
 108 {
 109         DBG("driver %p name %s", driver, driver->name);
 110
 111         driver_list = g_slist_insert_sorted(driver_list, driver,
 112                                                         compare_priority);
 113
 114         return 0;
 115 }
 116
 117 /**
 118  * connman_timeserver_driver_unregister:
 119  * @driver: timeserver driver definition
 120  *
 121  * Remove a previously registered timeserver driver
 122  */
 123 void connman_timeserver_driver_unregister(struct connman_timeserver_driver *driver)
 124 {
 125         DBG("driver %p name %s", driver, driver->name);
 126
 127         driver_list = g_slist_remove(driver_list, driver);
 128 }
 129
 130 /**
 131  * connman_timeserver_append:
 132  * @server: server address
 133  *
 134  * Append time server server address to current list
 135  */
 136 int connman_timeserver_append(const char *server)
 137 {
 138         GSList *list;
 139
 140         DBG("server %s", server);
 141
 142         if (server == NULL)
 143                 return -EINVAL;
 144
 145         /* This server is already handled by a driver */
 146         if (g_hash_table_lookup(server_hash, server))
 147                 return 0;
 148
 149         for (list = driver_list; list; list = list->next) {
 150                 struct connman_timeserver_driver *driver = list->data;
 151                 char *new_server;
 152
 153                 if (driver->append == NULL)
 154                         continue;
 155
 156                 new_server = g_strdup(server);
 157                 if (new_server == NULL)
 158                         return -ENOMEM;
 159
 160                 if (driver->append(server) == 0) {
 161                         g_hash_table_insert(server_hash, new_server, driver);
 162                         return 0;
 163                 } else {
 164                         g_free(new_server);
 165                 }
 166         }
 167
 168         return -ENOENT;
 169 }
 170
 171 /**
 172  * connman_timeserver_remove:
 173  * @server: server address
 174  *
 175  * Remover time server server address from current list
 176  */
 177 int connman_timeserver_remove(const char *server)
 178 {
 179         struct connman_timeserver_driver *driver;
 180
 181         DBG("server %s", server);
 182
 183         if (server == NULL)
 184                 return -EINVAL;
 185
 186         driver = g_hash_table_lookup(server_hash, server);
 187         if (driver == NULL)
 188                 return -EINVAL;
 189
 190         g_hash_table_remove(server_hash, server);
 191
 192         if (driver->remove == NULL)
 193                 return -ENOENT;
 194
 195         return driver->remove(server);
 196 }
 197
 198 /* Restart NTP procedure */
 199 static void connman_timeserver_restart()
 200 {
 201         /* If service timeservers are in use, dont restart ntp */
 202         if (timeservers != NULL)
 203                 return;
 204
 205         if (resolv == NULL) {
 206                 DBG("No online service.");
 207                 return;
 208         }
 209
 210         /* Cancel current lookup */
 211         if(resolv_id > 0)
 212                 g_resolv_cancel_lookup(resolv, resolv_id);
 213
 214         /* Reload system timeserver list */
 215         if (system_timeservers != NULL) {
 216                 g_strfreev(system_timeservers);
 217                 system_timeservers = NULL;
 218         }
 219
 220         system_timeservers = load_timeservers();
 221
 222         if (system_timeservers == NULL)
 223                 return;
 224
 225         __connman_ntp_stop();
 226
 227         count = 0;
 228
 229         __connman_timeserver_sync_next();
 230 }
 231
 232 static void resolv_result(GResolvResultStatus status, char **results, gpointer user_data)
 233 {
 234         int i;
 235
 236         DBG("status %d", status);
 237
 238         __sync_fetch_and_add(&count, 1);
 239
 240         if (status == G_RESOLV_RESULT_STATUS_SUCCESS) {
 241                 if (results != NULL) {
 242                         for (i = 0; results[i]; i++)
 243                                 DBG("result: %s", results[i]);
 244
 245                         __connman_ntp_start(results[0]);
 246
 247                         return;
 248                 }
 249         }
 250
 251         __connman_timeserver_sync_next();
 252 }
 253
 254 void __connman_timeserver_sync_next()
 255 {
 256         if (system_timeservers == NULL ||
 257         system_timeservers[count] == NULL)
 258                 return;
 259
 260         DBG("Trying timeserver %s", system_timeservers[count]);
 261
 262         if (resolv)
 263                 resolv_id = g_resolv_lookup_hostname(resolv,
 264                         system_timeservers[count], resolv_result,
 265                                                 NULL);
 266 }
 267
 268 int __connman_timeserver_sync(struct connman_service *service)
 269 {
 270         char **nameservers = NULL;
 271         int i;
 272
 273         DBG("service %p", service);
 274
 275         i = __connman_service_get_index(service);
 276         if (i < 0)
 277                 return -EINVAL;
 278
 279         nameservers = connman_service_get_nameservers(service);
 280         if (nameservers == NULL)
 281                 return -EINVAL;
 282
 283         resolv = g_resolv_new(i);
 284         if (resolv == NULL)
 285                 return -ENOMEM;
 286
 287         if (getenv("CONNMAN_RESOLV_DEBUG"))
 288                 g_resolv_set_debug(resolv, resolv_debug, "RESOLV");
 289
 290         for (i = 0; nameservers[i] != NULL; i++)
 291                 g_resolv_add_nameserver(resolv, nameservers[i], 53, 0);
 292
 293         count = 0;
 294
 295         system_timeservers = load_timeservers();
 296
 297         timeservers = connman_service_get_timeservers(service);
 298
 299         if (timeservers != NULL && timeservers[0] != NULL) {
 300                 DBG("Using service tiemservers");
 301                 __connman_ntp_start(timeservers[0]);
 302                 return 0;
 303         }
 304
 305         if (system_timeservers == NULL || system_timeservers[count] == NULL) {
 306                 DBG("No timeservers set.");
 307                 return 0;
 308         }
 309
 310         DBG("Trying server %s", system_timeservers[count]);
 311
 312         resolv_id = g_resolv_lookup_hostname(resolv, system_timeservers[count],
 313                                                 resolv_result, NULL);
 314         return 0;
 315 }
 316
 317 void __connman_timeserver_stop()
 318 {
 319         DBG(" ");
 320
 321         if (resolv != NULL) {
 322                 g_resolv_unref(resolv);
 323                 resolv = NULL;
 324         }
 325
 326         if (system_timeservers != NULL) {
 327                 g_strfreev(system_timeservers);
 328                 system_timeservers = NULL;
 329         }
 330
 331         timeservers = NULL;
 332
 333         count = 0;
 334
 335         __connman_ntp_stop();
 336 }
 337
 338 int __connman_timeserver_system_append(const char *server)
 339 {
 340         int len;
 341         char **servers = NULL;
 342
 343         if (server == NULL) {
 344                 save_timeservers(servers);
 345                 goto restart;
 346         }
 347
 348         DBG("server %s", server);
 349
 350         servers = load_timeservers();
 351
 352         if (servers != NULL) {
 353                 int i;
 354
 355                 for (i = 0; servers[i] != NULL; i++)
 356                         if (g_strcmp0(servers[i], server) == 0) {
 357                                 g_strfreev(servers);
 358                                 return -EEXIST;
 359                         }
 360
 361                 len = g_strv_length(servers);
 362                 servers = g_try_renew(char *, servers, len + 2);
 363         } else {
 364                 len = 0;
 365                 servers = g_try_new0(char *, len + 2);
 366         }
 367
 368         if (servers == NULL)
 369                 return -ENOMEM;
 370
 371         servers[len] = g_strdup(server);
 372         servers[len + 1] = NULL;
 373
 374         save_timeservers(servers);
 375
 376         g_strfreev(servers);
 377 restart:
 378         connman_timeserver_restart();
 379
 380         return 0;
 381 }
 382
 383 int __connman_timeserver_system_remove(const char *server)
 384 {
 385         char **servers;
 386         char **temp;
 387         int len, i, j;
 388
 389         if (server == NULL)
 390                 return -EINVAL;
 391
 392         DBG("server %s", server);
 393
 394         servers = load_timeservers();
 395
 396         if (servers == NULL)
 397                 return 0;
 398
 399         len = g_strv_length(servers);
 400         if (len == 1) {
 401                 if (g_strcmp0(servers[0], server) != 0) {
 402                         g_strfreev(servers);
 403                         return 0;
 404                 }
 405
 406                 g_strfreev(servers);
 407                 servers = NULL;
 408                 save_timeservers(servers);
 409                 return 0;
 410         }
 411
 412         temp = g_try_new0(char *, len - 1);
 413         if (temp == NULL) {
 414                         g_strfreev(servers);
 415                         return -ENOMEM;
 416         }
 417
 418         for (i = 0, j = 0; i < len; i++) {
 419                 if (g_strcmp0(servers[i], server) != 0) {
 420                         temp[j] = g_strdup(servers[i]);
 421                         j++;
 422                 }
 423         }
 424         temp[len - 1] = NULL;
 425
 426         g_strfreev(servers);
 427         servers = g_strdupv(temp);
 428         g_strfreev(temp);
 429
 430         save_timeservers(servers);
 431         g_strfreev(servers);
 432
 433         connman_timeserver_restart();
 434
 435         return 0;
 436 }
 437
 438 char **__connman_timeserver_system_get()
 439 {
 440         char **servers;
 441
 442         servers = load_timeservers();
 443         return servers;
 444 }
 445
 446 int __connman_timeserver_init(void)
 447 {
 448         DBG("");
 449
 450         server_hash = g_hash_table_new_full(g_str_hash, g_str_equal,
 451                                                 g_free, NULL);
 452
 453         return 0;
 454 }
 455
 456 void __connman_timeserver_cleanup(void)
 457 {
 458         DBG("");
 459
 460         g_hash_table_destroy(server_hash);
 461 }