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 (resolv == NULL) {
 202                 DBG("No online service.");
 203                 return;
 204         }
 205
 206         /* Cancel current lookup */
 207         if(resolv_id > 0)
 208                 g_resolv_cancel_lookup(resolv, resolv_id);
 209
 210         /* Reload system timeserver list */
 211         if (system_timeservers != NULL) {
 212                 g_strfreev(system_timeservers);
 213                 system_timeservers = NULL;
 214         }
 215
 216         system_timeservers = load_timeservers();
 217
 218         if (system_timeservers == NULL)
 219                 return;
 220
 221         __connman_ntp_stop();
 222
 223         count = 0;
 224
 225         __connman_timeserver_sync_next();
 226 }
 227
 228 static void resolv_result(GResolvResultStatus status, char **results, gpointer user_data)
 229 {
 230         int i;
 231
 232         DBG("status %d", status);
 233
 234         __sync_fetch_and_add(&count, 1);
 235
 236         if (status == G_RESOLV_RESULT_STATUS_SUCCESS) {
 237                 if (results != NULL) {
 238                         for (i = 0; results[i]; i++)
 239                                 DBG("result: %s", results[i]);
 240
 241                         __connman_ntp_start(results[0]);
 242
 243                         return;
 244                 }
 245         }
 246
 247         __connman_timeserver_sync_next();
 248 }
 249
 250 void __connman_timeserver_sync_next()
 251 {
 252         if (system_timeservers == NULL ||
 253         system_timeservers[count] == NULL)
 254                 return;
 255
 256         DBG("Trying timeserver %s", system_timeservers[count]);
 257
 258         if (resolv)
 259                 resolv_id = g_resolv_lookup_hostname(resolv,
 260                         system_timeservers[count], resolv_result,
 261                                                 NULL);
 262 }
 263
 264 int __connman_timeserver_sync(struct connman_service *service)
 265 {
 266         char **nameservers = NULL;
 267         int i;
 268
 269         DBG("service %p", service);
 270
 271         i = __connman_service_get_index(service);
 272         if (i < 0)
 273                 return -EINVAL;
 274
 275         nameservers = connman_service_get_nameservers(service);
 276         if (nameservers == NULL)
 277                 return -EINVAL;
 278
 279         resolv = g_resolv_new(i);
 280         if (resolv == NULL)
 281                 return -ENOMEM;
 282
 283         if (getenv("CONNMAN_RESOLV_DEBUG"))
 284                 g_resolv_set_debug(resolv, resolv_debug, "RESOLV");
 285
 286         for (i = 0; nameservers[i] != NULL; i++)
 287                 g_resolv_add_nameserver(resolv, nameservers[i], 53, 0);
 288
 289         count = 0;
 290
 291         system_timeservers = load_timeservers();
 292
 293         timeservers = connman_service_get_timeservers(service);
 294
 295         if (timeservers != NULL && timeservers[0] != NULL) {
 296                 DBG("Using service tiemservers");
 297                 __connman_ntp_start(timeservers[0]);
 298                 return 0;
 299         }
 300
 301         if (system_timeservers == NULL || system_timeservers[count] == NULL) {
 302                 DBG("No timeservers set.");
 303                 return 0;
 304         }
 305
 306         DBG("Trying server %s", system_timeservers[count]);
 307
 308         resolv_id = g_resolv_lookup_hostname(resolv, system_timeservers[count],
 309                                                 resolv_result, NULL);
 310         return 0;
 311 }
 312
 313 void __connman_timeserver_stop()
 314 {
 315         DBG(" ");
 316
 317         if (resolv != NULL) {
 318                 g_resolv_unref(resolv);
 319                 resolv = NULL;
 320         }
 321
 322         if (system_timeservers != NULL) {
 323                 g_strfreev(system_timeservers);
 324                 system_timeservers = NULL;
 325         }
 326
 327         timeservers = NULL;
 328
 329         count = 0;
 330
 331         __connman_ntp_stop();
 332 }
 333
 334 int __connman_timeserver_system_append(const char *server)
 335 {
 336         int len;
 337         char **servers = NULL;
 338
 339         if (server == NULL) {
 340                 save_timeservers(servers);
 341                 goto restart;
 342         }
 343
 344         DBG("server %s", server);
 345
 346         servers = load_timeservers();
 347
 348         if (servers != NULL) {
 349                 int i;
 350
 351                 for (i = 0; servers[i] != NULL; i++)
 352                         if (g_strcmp0(servers[i], server) == 0) {
 353                                 g_strfreev(servers);
 354                                 return -EEXIST;
 355                         }
 356
 357                 len = g_strv_length(servers);
 358                 servers = g_try_renew(char *, servers, len + 2);
 359         } else {
 360                 len = 0;
 361                 servers = g_try_new0(char *, len + 2);
 362         }
 363
 364         if (servers == NULL)
 365                 return -ENOMEM;
 366
 367         servers[len] = g_strdup(server);
 368         servers[len + 1] = NULL;
 369
 370         save_timeservers(servers);
 371
 372         g_strfreev(servers);
 373 restart:
 374         connman_timeserver_restart();
 375
 376         return 0;
 377 }
 378
 379 int __connman_timeserver_system_remove(const char *server)
 380 {
 381         char **servers;
 382         char **temp;
 383         int len, i, j;
 384
 385         if (server == NULL)
 386                 return -EINVAL;
 387
 388         DBG("server %s", server);
 389
 390         servers = load_timeservers();
 391
 392         if (servers == NULL)
 393                 return 0;
 394
 395         len = g_strv_length(servers);
 396         if (len == 1) {
 397                 if (g_strcmp0(servers[0], server) != 0) {
 398                         g_strfreev(servers);
 399                         return 0;
 400                 }
 401
 402                 g_strfreev(servers);
 403                 servers = NULL;
 404                 save_timeservers(servers);
 405                 return 0;
 406         }
 407
 408         temp = g_try_new0(char *, len - 1);
 409         if (temp == NULL) {
 410                         g_strfreev(servers);
 411                         return -ENOMEM;
 412         }
 413
 414         for (i = 0, j = 0; i < len; i++) {
 415                 if (g_strcmp0(servers[i], server) != 0) {
 416                         temp[j] = g_strdup(servers[i]);
 417                         j++;
 418                 }
 419         }
 420         temp[len - 1] = NULL;
 421
 422         g_strfreev(servers);
 423         servers = g_strdupv(temp);
 424         g_strfreev(temp);
 425
 426         save_timeservers(servers);
 427         g_strfreev(servers);
 428
 429         connman_timeserver_restart();
 430
 431         return 0;
 432 }
 433
 434 char **__connman_timeserver_system_get()
 435 {
 436         char **servers;
 437
 438         servers = load_timeservers();
 439         return servers;
 440 }
 441
 442 int __connman_timeserver_init(void)
 443 {
 444         DBG("");
 445
 446         server_hash = g_hash_table_new_full(g_str_hash, g_str_equal,
 447                                                 g_free, NULL);
 448
 449         return 0;
 450 }
 451
 452 void __connman_timeserver_cleanup(void)
 453 {
 454         DBG("");
 455
 456         g_hash_table_destroy(server_hash);
 457 }