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,
 113                                 int exit_code, void *user_data)
 114 {
 115         struct ntpdate_task *ntpdate = user_data;
 116
 117         DBG("");
 118
 119         unlink(ntpdate->conf_path);
 120         g_free(ntpdate->conf_path);
 121         connman_task_destroy(ntpdate->task);
 122 }
 123
 124 static void ntpdate_add_peer(struct ntpdate_task *ntpdate, char *peer)
 125 {
 126         FILE *conf_file;
 127
 128         DBG("%s", peer);
 129
 130         conf_file = fdopen(ntpdate->conf_fd, "a+");
 131         if (conf_file == NULL) {
 132                 connman_error("fdopen failed");
 133                 return;
 134         }
 135
 136         fprintf(conf_file, "server %s iburst\n", peer);
 137
 138         fclose(conf_file);
 139 }
 140
 141 static int ntpdate(void)
 142 {
 143         int err;
 144         GError *g_err;
 145         GList *list;
 146         struct ntpd_peer *peer;
 147         struct ntpdate_task *ntpdate;
 148
 149         DBG("");
 150
 151         ntpdate = g_try_new0(struct ntpdate_task, 1);
 152         if (ntpdate == NULL)
 153                 return -ENOMEM;
 154
 155         /* ntpdate is deprecated, we use ntpd -q instead */
 156         ntpdate->task = connman_task_create(NTPD);
 157         if (ntpdate->task == NULL) {
 158                 err = -ENOMEM;
 159                 goto error_task;
 160         }
 161
 162         connman_task_add_argument(ntpdate->task, "-g", NULL);
 163         connman_task_add_argument(ntpdate->task, "-q", NULL);
 164
 165         /* The servers are added through a temp configuration file */
 166         ntpdate->conf_fd = g_file_open_tmp("connman.ntp.conf_XXXXXX",
 167                                                 &ntpdate->conf_path, &g_err);
 168         if  (ntpdate->conf_fd == -1) {
 169                 err = g_err->code;
 170                 g_free(g_err);
 171                 goto error_open;
 172         }
 173
 174         connman_task_add_argument(ntpdate->task, "-c", ntpdate->conf_path);
 175
 176         DBG("conf path %s", ntpdate->conf_path);
 177
 178         for (list = pending_peers; list; list = list->next) {
 179                 peer = list->data;
 180
 181                 ntpdate_add_peer(ntpdate, peer->server);
 182         }
 183
 184         for (list = peers; list; list = list->next) {
 185                 peer = list->data;
 186
 187                 ntpdate_add_peer(ntpdate, peer->server);
 188         }
 189
 190         close(ntpdate->conf_fd);
 191
 192         return connman_task_run(ntpdate->task, ntpdate_died, ntpdate,
 193                                                 NULL, NULL, NULL);
 194 error_open:
 195         connman_task_destroy(ntpdate->task);
 196
 197 error_task:
 198         g_free(ntpdate);
 199
 200         return err;
 201 }
 202
 203 static int ntpd_add_peer(char *peer)
 204 {
 205         DBG("%s", peer);
 206
 207         return 0;
 208 }
 209
 210 static void ntpd_sync(void)
 211 {
 212         int err;
 213         GList *list;
 214
 215         DBG("");
 216
 217         if (g_list_length(pending_peers) == 0 &&
 218                         g_list_length(peers) == 0)
 219                 return;
 220
 221         if (!ntpd_running()) {
 222                 ntpdate();
 223                 return;
 224         }
 225
 226         /* TODO Grab ntp keys path */
 227
 228         list = g_list_first(pending_peers);
 229         while(list) {
 230                 struct ntpd_peer *peer = list->data;
 231
 232                 err = ntpd_add_peer(peer->server);
 233                 if (err)
 234                         continue;
 235
 236                 peers = g_list_prepend(peers, peer);
 237
 238                 list = g_list_next(list);
 239
 240                 pending_peers = g_list_remove(pending_peers, peer);
 241         };
 242 }
 243
 244 static int ntpd_append(const char *server)
 245 {
 246         struct ntpd_peer *peer;
 247
 248         DBG("");
 249
 250         if (server == NULL)
 251                 return 0;
 252
 253         if ((peer = find_peer(pending_peers, server)) ||
 254                         (peer = find_peer(peers, server))) {
 255                 g_atomic_int_inc(&peer->refcount);
 256                 return 0;
 257         }
 258
 259         peer = g_try_new0(struct ntpd_peer, 1);
 260         if (peer == NULL)
 261                 return -ENOMEM;
 262
 263         peer->server = g_strdup(server);
 264         if (peer->server == NULL) {
 265                 g_free(peer);
 266                 return -ENOMEM;
 267         }
 268
 269         peer->refcount = 1;
 270
 271         pending_peers = g_list_prepend(pending_peers, peer);
 272
 273         return 0;
 274 }
 275
 276 static int ntpd_remove(const char *server)
 277 {
 278         struct ntpd_peer *peer;
 279
 280         DBG("");
 281
 282         if (server == NULL)
 283                 return 0;
 284
 285         peer = find_peer(peers, server);
 286         if (peer == NULL)
 287                 goto remove;
 288
 289         remove_peer(peers, peer);
 290
 291 remove:
 292         /* TODO: send ntpd remove command */
 293
 294         peer = find_peer(pending_peers, server);
 295         if (peer == NULL)
 296                 return 0;
 297
 298         remove_peer(pending_peers, peer);
 299
 300         return 0;
 301 }
 302
 303 static struct connman_timeserver_driver ntpd_driver = {
 304         .name           = "ntpd",
 305         .priority       = CONNMAN_DRIVER_PRIORITY_DEFAULT,
 306         .append         = ntpd_append,
 307         .remove         = ntpd_remove,
 308         .sync           = ntpd_sync,
 309 };
 310
 311 static int ntpd_init(void)
 312 {
 313         return connman_timeserver_driver_register(&ntpd_driver);
 314 }
 315
 316 static void ntpd_exit(void)
 317 {
 318         connman_timeserver_driver_unregister(&ntpd_driver);
 319 }
 320
 321 CONNMAN_PLUGIN_DEFINE(ntpd, "ntpd plugin", VERSION,
 322                 CONNMAN_PLUGIN_PRIORITY_DEFAULT, ntpd_init, ntpd_exit)