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, "-g", NULL);
 162         connman_task_add_argument(ntpdate->task, "-q", NULL);
 163
 164         /* The servers are added through a temp configuration file */
 165         ntpdate->conf_fd = g_file_open_tmp("connman.ntp.conf_XXXXXX",
 166                                                 &ntpdate->conf_path, &g_err);
 167         if  (ntpdate->conf_fd == -1) {
 168                 err = g_err->code;
 169                 g_free(g_err);
 170                 goto error_open;
 171         }
 172
 173         connman_task_add_argument(ntpdate->task, "-c", ntpdate->conf_path);
 174
 175         DBG("conf path %s", ntpdate->conf_path);
 176
 177         for (list = pending_peers; list; list = list->next) {
 178                 peer = list->data;
 179
 180                 ntpdate_add_peer(ntpdate, peer->server);
 181         }
 182
 183         for (list = peers; list; list = list->next) {
 184                 peer = list->data;
 185
 186                 ntpdate_add_peer(ntpdate, peer->server);
 187         }
 188
 189         close(ntpdate->conf_fd);
 190
 191         return connman_task_run(ntpdate->task, ntpdate_died, ntpdate,
 192                                                 NULL, NULL, NULL);
 193 error_open:
 194         connman_task_destroy(ntpdate->task);
 195
 196 error_task:
 197         g_free(ntpdate);
 198
 199         return err;
 200 }
 201
 202 static int ntpd_add_peer(char *peer)
 203 {
 204         DBG("%s", peer);
 205
 206         return 0;
 207 }
 208
 209 static void ntpd_sync(void)
 210 {
 211         int err;
 212         GList *list;
 213
 214         DBG("");
 215
 216         if (g_list_length(pending_peers) == 0 &&
 217                         g_list_length(peers) == 0)
 218                 return;
 219
 220         if (!ntpd_running()) {
 221                 ntpdate();
 222                 return;
 223         }
 224
 225         /* TODO Grab ntp keys path */
 226
 227         list = g_list_first(pending_peers);
 228         while(list) {
 229                 struct ntpd_peer *peer = list->data;
 230
 231                 err = ntpd_add_peer(peer->server);
 232                 if (err)
 233                         continue;
 234
 235                 peers = g_list_prepend(peers, peer);
 236
 237                 list = g_list_next(list);
 238
 239                 pending_peers = g_list_remove(pending_peers, peer);
 240         };
 241 }
 242
 243 static int ntpd_append(const char *server)
 244 {
 245         struct ntpd_peer *peer;
 246
 247         DBG("");
 248
 249         if (server == NULL)
 250                 return 0;
 251
 252         if ((peer = find_peer(pending_peers, server)) ||
 253                         (peer = find_peer(peers, server))) {
 254                 g_atomic_int_inc(&peer->refcount);
 255                 return 0;
 256         }
 257
 258         peer = g_try_new0(struct ntpd_peer, 1);
 259         if (peer == NULL)
 260                 return -ENOMEM;
 261
 262         peer->server = g_strdup(server);
 263         if (peer->server == NULL) {
 264                 g_free(peer);
 265                 return -ENOMEM;
 266         }
 267
 268         peer->refcount = 1;
 269
 270         pending_peers = g_list_prepend(pending_peers, peer);
 271
 272         return 0;
 273 }
 274
 275 static int ntpd_remove(const char *server)
 276 {
 277         struct ntpd_peer *peer;
 278
 279         DBG("");
 280
 281         if (server == NULL)
 282                 return 0;
 283
 284         peer = find_peer(peers, server);
 285         if (peer == NULL)
 286                 goto remove;
 287
 288         remove_peer(peers, peer);
 289
 290 remove:
 291         /* TODO: send ntpd remove command */
 292
 293         peer = find_peer(pending_peers, server);
 294         if (peer == NULL)
 295                 return 0;
 296
 297         remove_peer(pending_peers, peer);
 298
 299         return 0;
 300 }
 301
 302 static struct connman_timeserver_driver ntpd_driver = {
 303         .name           = "ntpd",
 304         .priority       = CONNMAN_DRIVER_PRIORITY_DEFAULT,
 305         .append         = ntpd_append,
 306         .remove         = ntpd_remove,
 307         .sync           = ntpd_sync,
 308 };
 309
 310 static int ntpd_init(void)
 311 {
 312         return connman_timeserver_driver_register(&ntpd_driver);
 313 }
 314
 315 static void ntpd_exit(void)
 316 {
 317         connman_timeserver_driver_unregister(&ntpd_driver);
 318 }
 319
 320 CONNMAN_PLUGIN_DEFINE(ntpd, "ntpd plugin", VERSION,
 321                 CONNMAN_PLUGIN_PRIORITY_DEFAULT, ntpd_init, ntpd_exit)