plugins/ntpd.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 #include <stdio.h>
  28 #include <unistd.h>
  29 #include <stdlib.h>
  30 #include <string.h>
  31 #include <arpa/inet.h>
  32
  33 #define CONNMAN_API_SUBJECT_TO_CHANGE
  34 #include <connman/plugin.h>
  35 #include <connman/task.h>
  36 #include <connman/timeserver.h>
  37 #include <connman/driver.h>
  38 #include <connman/log.h>
  39
  40 /*
  41  * The peers list are the peers currently added to a running ntpd,
  42  * while pending_peers are the one appended but not used by ntpd yet.
  43  */
  44 static GList *peers = NULL;
  45 static GList *pending_peers = NULL;
  46
  47 #define NTPD_PORT 123
  48
  49 struct ntpd_peer {
  50         char *server;
  51         gint refcount;
  52 };
  53
  54 struct ntpdate_task {
  55         struct connman_task *task;
  56         gint conf_fd;
  57         char *conf_path;
  58 };
  59
  60 static struct ntpd_peer *find_peer(GList *peer_list, const char* server)
  61 {
  62         GList *list;
  63         struct ntpd_peer *peer;
  64
  65         for (list = peer_list; list; list = list->next) {
  66                 peer = list->data;
  67
  68                 if (g_str_equal(peer->server, server))
  69                         return peer;
  70         }
  71
  72         return NULL;
  73 }
  74
  75 static void remove_peer(GList *peer_list, struct ntpd_peer *peer)
  76 {
  77         if (!g_atomic_int_dec_and_test(&peer->refcount))
  78                 return;
  79
  80         g_free(peer->server);
  81         g_free(peer);
  82         peer_list = g_list_remove(peer_list, peer);
  83 }
  84
  85 static connman_bool_t ntpd_running(void)
  86 {
  87         int sock;
  88         connman_bool_t ret;
  89         struct sockaddr_in server_addr;
  90
  91         if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
  92                 return FALSE;
  93
  94         server_addr.sin_family = AF_INET;
  95         server_addr.sin_port = htons(NTPD_PORT);
  96         server_addr.sin_addr.s_addr = INADDR_ANY;
  97         memset(&(server_addr.sin_zero), 0, 8);
  98
  99         if (bind(sock, (struct sockaddr *)&server_addr,
 100                         sizeof(struct sockaddr)) == -1) {
 101                 if (errno == EADDRINUSE)
 102                         ret = TRUE;
 103                 else
 104                         ret = FALSE;
 105         }
 106
 107         close(sock);
 108
 109         return ret;
 110 }
 111
 112 static void ntpdate_died(struct connman_task *task, void *user_data)
 113 {
 114         struct ntpdate_task *ntpdate = user_data;
 115
 116         DBG("");
 117
 118         unlink(ntpdate->conf_path);
 119         g_free(ntpdate->conf_path);
 120         connman_task_destroy(ntpdate->task);
 121 }
 122
 123 static void ntpdate_add_peer(struct ntpdate_task *ntpdate, char *peer)
 124 {
 125         FILE *conf_file;
 126
 127         DBG("%s", peer);
 128
 129         conf_file = fdopen(ntpdate->conf_fd, "a+");
 130         if (conf_file == NULL) {
 131                 connman_error("fdopen failed");
 132                 return;
 133         }
 134
 135         fprintf(conf_file, "server %s iburst\n", peer);
 136
 137         fclose(conf_file);
 138 }
 139
 140 static int ntpdate(void)
 141 {
 142         int err;
 143         GError *g_err;
 144         GList *list;
 145         struct ntpd_peer *peer;
 146         struct ntpdate_task *ntpdate;
 147
 148         DBG("");
 149
 150         ntpdate = g_try_new0(struct ntpdate_task, 1);
 151         if (ntpdate == NULL)
 152                 return -ENOMEM;
 153
 154         /* ntpdate is deprecated, we use ntpd -q instead */
 155         ntpdate->task = connman_task_create(NTPD);
 156         if (ntpdate->task == NULL) {
 157                 err = -ENOMEM;
 158                 goto error_task;
 159         }
 160
 161         connman_task_add_argument(ntpdate->task, "-q", NULL);
 162
 163         /* The servers are added through a temp configuration file */
 164         ntpdate->conf_fd = g_file_open_tmp("connman.ntp.conf_XXXXXX",
 165                                                 &ntpdate->conf_path, &g_err);
 166         if  (ntpdate->conf_fd == -1) {
 167                 err = g_err->code;
 168                 g_free(g_err);
 169                 goto error_open;
 170         }
 171
 172         connman_task_add_argument(ntpdate->task, "-c", ntpdate->conf_path);
 173
 174         DBG("conf path %s", ntpdate->conf_path);
 175
 176         for (list = pending_peers; list; list = list->next) {
 177                 peer = list->data;
 178
 179                 ntpdate_add_peer(ntpdate, peer->server);
 180         }
 181
 182         for (list = peers; list; list = list->next) {
 183                 peer = list->data;
 184
 185                 ntpdate_add_peer(ntpdate, peer->server);
 186         }
 187
 188         close(ntpdate->conf_fd);
 189
 190         return connman_task_run(ntpdate->task, ntpdate_died, ntpdate,
 191                                                 NULL, NULL, NULL);
 192 error_open:
 193         connman_task_destroy(ntpdate->task);
 194
 195 error_task:
 196         g_free(ntpdate);
 197
 198         return err;
 199 }
 200
 201 static int ntpd_add_peer(char *peer)
 202 {
 203         DBG("%s", peer);
 204
 205         return 0;
 206 }
 207
 208 static void ntpd_sync(void)
 209 {
 210         int err;
 211         GList *list;
 212
 213         DBG("");
 214
 215         if (g_list_length(pending_peers) == 0 &&
 216                         g_list_length(peers) == 0)
 217                 return;
 218
 219         if (!ntpd_running()) {
 220                 ntpdate();
 221                 return;
 222         }
 223
 224         /* TODO Grab ntp keys path */
 225
 226         list = g_list_first(pending_peers);
 227         while(list) {
 228                 struct ntpd_peer *peer = list->data;
 229
 230                 err = ntpd_add_peer(peer->server);
 231                 if (err)
 232                         continue;
 233
 234                 peers = g_list_prepend(peers, peer);
 235
 236                 list = g_list_next(list);
 237
 238                 pending_peers = g_list_remove(pending_peers, peer);
 239         };
 240 }
 241
 242 static int ntpd_append(const char *server)
 243 {
 244         struct ntpd_peer *peer;
 245
 246         DBG("");
 247
 248         if (server == NULL)
 249                 return 0;
 250
 251         if ((peer = find_peer(pending_peers, server)) ||
 252                         (peer = find_peer(peers, server))) {
 253                 g_atomic_int_inc(&peer->refcount);
 254                 return 0;
 255         }
 256
 257         peer = g_try_new0(struct ntpd_peer, 1);
 258         if (peer == NULL)
 259                 return -ENOMEM;
 260
 261         peer->server = g_strdup(server);
 262         if (peer->server == NULL) {
 263                 g_free(peer);
 264                 return -ENOMEM;
 265         }
 266
 267         peer->refcount = 1;
 268
 269         pending_peers = g_list_prepend(pending_peers, peer);
 270
 271         return 0;
 272 }
 273
 274 static int ntpd_remove(const char *server)
 275 {
 276         struct ntpd_peer *peer;
 277
 278         DBG("");
 279
 280         if (server == NULL)
 281                 return 0;
 282
 283         peer = find_peer(peers, server);
 284         if (peer == NULL)
 285                 goto remove;
 286
 287         remove_peer(peers, peer);
 288
 289 remove:
 290         /* TODO: send ntpd remove command */
 291
 292         peer = find_peer(pending_peers, server);
 293         if (peer == NULL)
 294                 return 0;
 295
 296         remove_peer(pending_peers, peer);
 297
 298         return 0;
 299 }
 300
 301 static struct connman_timeserver_driver ntpd_driver = {
 302         .name           = "ntpd",
 303         .priority       = CONNMAN_DRIVER_PRIORITY_DEFAULT,
 304         .append         = ntpd_append,
 305         .remove         = ntpd_remove,
 306         .sync           = ntpd_sync,
 307 };
 308
 309 static int ntpd_init(void)
 310 {
 311         return connman_timeserver_driver_register(&ntpd_driver);
 312 }
 313
 314 static void ntpd_exit(void)
 315 {
 316         connman_timeserver_driver_unregister(&ntpd_driver);
 317 }
 318
 319 CONNMAN_PLUGIN_DEFINE(ntpd, "ntpd plugin", VERSION,
 320                 CONNMAN_PLUGIN_PRIORITY_DEFAULT, ntpd_init, ntpd_exit)