| 1 | /*
 | 
|---|
| 2 |  * Copyright (c) 2012 Maurizio Lombardi
 | 
|---|
| 3 |  * All rights reserved.
 | 
|---|
| 4 |  *
 | 
|---|
| 5 |  * Redistribution and use in source and binary forms, with or without
 | 
|---|
| 6 |  * modification, are permitted provided that the following conditions
 | 
|---|
| 7 |  * are met:
 | 
|---|
| 8 |  *
 | 
|---|
| 9 |  * - Redistributions of source code must retain the above copyright
 | 
|---|
| 10 |  *   notice, this list of conditions and the following disclaimer.
 | 
|---|
| 11 |  * - Redistributions in binary form must reproduce the above copyright
 | 
|---|
| 12 |  *   notice, this list of conditions and the following disclaimer in the
 | 
|---|
| 13 |  *   documentation and/or other materials provided with the distribution.
 | 
|---|
| 14 |  * - The name of the author may not be used to endorse or promote products
 | 
|---|
| 15 |  *   derived from this software without specific prior written permission.
 | 
|---|
| 16 |  *
 | 
|---|
| 17 |  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 | 
|---|
| 18 |  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 | 
|---|
| 19 |  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 | 
|---|
| 20 |  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 | 
|---|
| 21 |  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 | 
|---|
| 22 |  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 | 
|---|
| 23 |  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 | 
|---|
| 24 |  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 | 
|---|
| 25 |  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 | 
|---|
| 26 |  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 | 
|---|
| 27 |  */
 | 
|---|
| 28 | 
 | 
|---|
| 29 | #include <stdio.h>
 | 
|---|
| 30 | #include <device/clock_dev.h>
 | 
|---|
| 31 | #include <errno.h>
 | 
|---|
| 32 | #include <loc.h>
 | 
|---|
| 33 | #include <time.h>
 | 
|---|
| 34 | #include <stdlib.h>
 | 
|---|
| 35 | #include <getopt.h>
 | 
|---|
| 36 | #include <ctype.h>
 | 
|---|
| 37 | #include <str.h>
 | 
|---|
| 38 | 
 | 
|---|
| 39 | #define NAME   "date"
 | 
|---|
| 40 | 
 | 
|---|
| 41 | static errno_t read_date_from_arg(char *wdate, struct tm *t);
 | 
|---|
| 42 | static errno_t read_time_from_arg(char *wdate, struct tm *t);
 | 
|---|
| 43 | static errno_t tm_sanity_check(struct tm *t);
 | 
|---|
| 44 | static bool is_leap_year(int year);
 | 
|---|
| 45 | 
 | 
|---|
| 46 | static void usage(void);
 | 
|---|
| 47 | 
 | 
|---|
| 48 | static int days_month[12] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
 | 
|---|
| 49 | 
 | 
|---|
| 50 | int
 | 
|---|
| 51 | main(int argc, char **argv)
 | 
|---|
| 52 | {
 | 
|---|
| 53 |         errno_t rc;
 | 
|---|
| 54 |         int c;
 | 
|---|
| 55 |         category_id_t cat_id;
 | 
|---|
| 56 |         size_t        svc_cnt;
 | 
|---|
| 57 |         service_id_t  *svc_ids = NULL;
 | 
|---|
| 58 |         service_id_t  svc_id;
 | 
|---|
| 59 |         char          *svc_name = NULL;
 | 
|---|
| 60 |         bool          read_only = true;
 | 
|---|
| 61 |         char          *wdate = NULL;
 | 
|---|
| 62 |         char          *wtime = NULL;
 | 
|---|
| 63 |         struct tm     t;
 | 
|---|
| 64 |         int           n_args = argc;
 | 
|---|
| 65 | 
 | 
|---|
| 66 |         while ((c = getopt(argc, argv, "hd:t:")) != -1) {
 | 
|---|
| 67 |                 switch (c) {
 | 
|---|
| 68 |                 case 'h':
 | 
|---|
| 69 |                         usage();
 | 
|---|
| 70 |                         return 0;
 | 
|---|
| 71 |                 case 'd':
 | 
|---|
| 72 |                         if (wdate) {
 | 
|---|
| 73 |                                 usage();
 | 
|---|
| 74 |                                 return 1;
 | 
|---|
| 75 |                         }
 | 
|---|
| 76 |                         wdate = (char *)optarg;
 | 
|---|
| 77 |                         read_only = false;
 | 
|---|
| 78 |                         n_args -= 2;
 | 
|---|
| 79 |                         break;
 | 
|---|
| 80 |                 case 't':
 | 
|---|
| 81 |                         if (wtime) {
 | 
|---|
| 82 |                                 usage();
 | 
|---|
| 83 |                                 return 1;
 | 
|---|
| 84 |                         }
 | 
|---|
| 85 |                         wtime = (char *)optarg;
 | 
|---|
| 86 |                         read_only = false;
 | 
|---|
| 87 |                         n_args -= 2;
 | 
|---|
| 88 |                         break;
 | 
|---|
| 89 |                 case '?':
 | 
|---|
| 90 |                         usage();
 | 
|---|
| 91 |                         return 1;
 | 
|---|
| 92 |                 }
 | 
|---|
| 93 |         }
 | 
|---|
| 94 | 
 | 
|---|
| 95 |         if (n_args != 1) {
 | 
|---|
| 96 |                 printf(NAME ": Unrecognized parameter\n");
 | 
|---|
| 97 |                 usage();
 | 
|---|
| 98 |                 return 1;
 | 
|---|
| 99 |         }
 | 
|---|
| 100 | 
 | 
|---|
| 101 |         /* Get the id of the clock category */
 | 
|---|
| 102 |         rc = loc_category_get_id("clock", &cat_id, IPC_FLAG_BLOCKING);
 | 
|---|
| 103 |         if (rc != EOK) {
 | 
|---|
| 104 |                 printf(NAME ": Cannot get clock category id\n");
 | 
|---|
| 105 |                 goto exit;
 | 
|---|
| 106 |         }
 | 
|---|
| 107 | 
 | 
|---|
| 108 |         /* Get the list of available services in the clock category */
 | 
|---|
| 109 |         rc = loc_category_get_svcs(cat_id, &svc_ids, &svc_cnt);
 | 
|---|
| 110 |         if (rc != EOK) {
 | 
|---|
| 111 |                 printf(NAME ": Cannot get the list of services in the clock "
 | 
|---|
| 112 |                     "category\n");
 | 
|---|
| 113 |                 goto exit;
 | 
|---|
| 114 |         }
 | 
|---|
| 115 | 
 | 
|---|
| 116 |         /* Check if there are available services in the clock category */
 | 
|---|
| 117 |         if (svc_cnt == 0) {
 | 
|---|
| 118 |                 printf(NAME ": No available service found in "
 | 
|---|
| 119 |                     "the clock category\n");
 | 
|---|
| 120 |                 goto exit;
 | 
|---|
| 121 |         }
 | 
|---|
| 122 | 
 | 
|---|
| 123 |         /* Get the name of the clock service */
 | 
|---|
| 124 |         rc = loc_service_get_name(svc_ids[0], &svc_name);
 | 
|---|
| 125 |         if (rc != EOK) {
 | 
|---|
| 126 |                 printf(NAME ": Cannot get the name of the service\n");
 | 
|---|
| 127 |                 goto exit;
 | 
|---|
| 128 |         }
 | 
|---|
| 129 | 
 | 
|---|
| 130 |         /* Get the service id for the device */
 | 
|---|
| 131 |         rc = loc_service_get_id(svc_name, &svc_id, 0);
 | 
|---|
| 132 |         if (rc != EOK) {
 | 
|---|
| 133 |                 printf(NAME ": Cannot get the service id for device %s",
 | 
|---|
| 134 |                     svc_name);
 | 
|---|
| 135 |                 goto exit;
 | 
|---|
| 136 |         }
 | 
|---|
| 137 | 
 | 
|---|
| 138 |         /* Connect to the device */
 | 
|---|
| 139 |         async_sess_t *sess = loc_service_connect(svc_id, INTERFACE_DDF, 0);
 | 
|---|
| 140 |         if (!sess) {
 | 
|---|
| 141 |                 printf(NAME ": Cannot connect to the device\n");
 | 
|---|
| 142 |                 goto exit;
 | 
|---|
| 143 |         }
 | 
|---|
| 144 | 
 | 
|---|
| 145 |         /* Read the current date/time */
 | 
|---|
| 146 |         rc = clock_dev_time_get(sess, &t);
 | 
|---|
| 147 |         if (rc != EOK) {
 | 
|---|
| 148 |                 printf(NAME ": Cannot read the current time\n");
 | 
|---|
| 149 |                 goto exit;
 | 
|---|
| 150 |         }
 | 
|---|
| 151 | 
 | 
|---|
| 152 |         if (read_only) {
 | 
|---|
| 153 |                 /* Print the current time and exit */
 | 
|---|
| 154 |                 printf("%02d/%02d/%d ", t.tm_mday,
 | 
|---|
| 155 |                     t.tm_mon + 1, 1900 + t.tm_year);
 | 
|---|
| 156 |                 printf("%02d:%02d:%02d\n", t.tm_hour, t.tm_min, t.tm_sec);
 | 
|---|
| 157 |         } else {
 | 
|---|
| 158 |                 if (wdate) {
 | 
|---|
| 159 |                         rc = read_date_from_arg(wdate, &t);
 | 
|---|
| 160 |                         if (rc != EOK) {
 | 
|---|
| 161 |                                 printf(NAME ": error, date format not "
 | 
|---|
| 162 |                                     "recognized\n");
 | 
|---|
| 163 |                                 usage();
 | 
|---|
| 164 |                                 goto exit;
 | 
|---|
| 165 |                         }
 | 
|---|
| 166 |                 }
 | 
|---|
| 167 |                 if (wtime) {
 | 
|---|
| 168 |                         rc = read_time_from_arg(wtime, &t);
 | 
|---|
| 169 |                         if (rc != EOK) {
 | 
|---|
| 170 |                                 printf(NAME ": error, time format not "
 | 
|---|
| 171 |                                     "recognized\n");
 | 
|---|
| 172 |                                 usage();
 | 
|---|
| 173 |                                 goto exit;
 | 
|---|
| 174 |                         }
 | 
|---|
| 175 |                 }
 | 
|---|
| 176 | 
 | 
|---|
| 177 |                 rc = tm_sanity_check(&t);
 | 
|---|
| 178 |                 if (rc != EOK) {
 | 
|---|
| 179 |                         printf(NAME ": error, invalid date/time\n");
 | 
|---|
| 180 |                         goto exit;
 | 
|---|
| 181 |                 }
 | 
|---|
| 182 | 
 | 
|---|
| 183 |                 rc = clock_dev_time_set(sess, &t);
 | 
|---|
| 184 |                 if (rc != EOK) {
 | 
|---|
| 185 |                         printf(NAME ": error, Unable to set date/time\n");
 | 
|---|
| 186 |                         goto exit;
 | 
|---|
| 187 |                 }
 | 
|---|
| 188 |         }
 | 
|---|
| 189 | 
 | 
|---|
| 190 | exit:
 | 
|---|
| 191 |         free(svc_name);
 | 
|---|
| 192 |         free(svc_ids);
 | 
|---|
| 193 |         return rc;
 | 
|---|
| 194 | }
 | 
|---|
| 195 | 
 | 
|---|
| 196 | /** Read the day, month and year from a string
 | 
|---|
| 197 |  *  with the following format: DD/MM/YYYY
 | 
|---|
| 198 |  */
 | 
|---|
| 199 | static errno_t
 | 
|---|
| 200 | read_date_from_arg(char *wdate, struct tm *t)
 | 
|---|
| 201 | {
 | 
|---|
| 202 |         errno_t rc;
 | 
|---|
| 203 |         uint32_t tmp;
 | 
|---|
| 204 | 
 | 
|---|
| 205 |         if (str_size(wdate) != 10) /* str_size("DD/MM/YYYY") == 10 */
 | 
|---|
| 206 |                 return EINVAL;
 | 
|---|
| 207 | 
 | 
|---|
| 208 |         if (wdate[2] != '/' ||
 | 
|---|
| 209 |             wdate[5] != '/') {
 | 
|---|
| 210 |                 return EINVAL;
 | 
|---|
| 211 |         }
 | 
|---|
| 212 | 
 | 
|---|
| 213 |         rc = str_uint32_t(&wdate[0], NULL, 10, false, &tmp);
 | 
|---|
| 214 |         if (rc != EOK)
 | 
|---|
| 215 |                 return rc;
 | 
|---|
| 216 | 
 | 
|---|
| 217 |         t->tm_mday = tmp;
 | 
|---|
| 218 | 
 | 
|---|
| 219 |         rc = str_uint32_t(&wdate[3], NULL, 10, false, &tmp);
 | 
|---|
| 220 |         if (rc != EOK)
 | 
|---|
| 221 |                 return rc;
 | 
|---|
| 222 | 
 | 
|---|
| 223 |         t->tm_mon = tmp - 1;
 | 
|---|
| 224 | 
 | 
|---|
| 225 |         rc = str_uint32_t(&wdate[6], NULL, 10, false, &tmp);
 | 
|---|
| 226 |         t->tm_year = tmp - 1900;
 | 
|---|
| 227 | 
 | 
|---|
| 228 |         return rc;
 | 
|---|
| 229 | }
 | 
|---|
| 230 | 
 | 
|---|
| 231 | /** Read the hours, minutes and seconds from a string
 | 
|---|
| 232 |  *  with the following format: HH:MM:SS or HH:MM
 | 
|---|
| 233 |  */
 | 
|---|
| 234 | static errno_t
 | 
|---|
| 235 | read_time_from_arg(char *wtime, struct tm *t)
 | 
|---|
| 236 | {
 | 
|---|
| 237 |         errno_t rc;
 | 
|---|
| 238 |         size_t len = str_size(wtime);
 | 
|---|
| 239 |         bool sec_present = len == 8;
 | 
|---|
| 240 |         uint32_t tmp;
 | 
|---|
| 241 | 
 | 
|---|
| 242 |         /* str_size("HH:MM") == 5 */
 | 
|---|
| 243 |         /* str_size("HH:MM:SS") == 8 */
 | 
|---|
| 244 |         if (len != 8 && len != 5)
 | 
|---|
| 245 |                 return EINVAL;
 | 
|---|
| 246 | 
 | 
|---|
| 247 |         if (sec_present && wtime[5] != ':')
 | 
|---|
| 248 |                 return EINVAL;
 | 
|---|
| 249 | 
 | 
|---|
| 250 |         if (wtime[2] != ':')
 | 
|---|
| 251 |                 return EINVAL;
 | 
|---|
| 252 | 
 | 
|---|
| 253 |         rc = str_uint32_t(&wtime[0], NULL, 10, false, &tmp);
 | 
|---|
| 254 |         if (rc != EOK)
 | 
|---|
| 255 |                 return rc;
 | 
|---|
| 256 | 
 | 
|---|
| 257 |         t->tm_hour = tmp;
 | 
|---|
| 258 | 
 | 
|---|
| 259 |         rc = str_uint32_t(&wtime[3], NULL, 10, false, &tmp);
 | 
|---|
| 260 |         if (rc != EOK)
 | 
|---|
| 261 |                 return rc;
 | 
|---|
| 262 | 
 | 
|---|
| 263 |         t->tm_min = tmp;
 | 
|---|
| 264 | 
 | 
|---|
| 265 |         if (sec_present) {
 | 
|---|
| 266 |                 rc = str_uint32_t(&wtime[6], NULL, 10, false, &tmp);
 | 
|---|
| 267 |                 t->tm_sec = tmp;
 | 
|---|
| 268 |         } else
 | 
|---|
| 269 |                 t->tm_sec = 0;
 | 
|---|
| 270 | 
 | 
|---|
| 271 |         return rc;
 | 
|---|
| 272 | }
 | 
|---|
| 273 | 
 | 
|---|
| 274 | /** Check if the tm structure contains valid values
 | 
|---|
| 275 |  *
 | 
|---|
| 276 |  * @param t     The tm structure to check
 | 
|---|
| 277 |  *
 | 
|---|
| 278 |  * @return      EOK on success or EINVAL
 | 
|---|
| 279 |  */
 | 
|---|
| 280 | static errno_t
 | 
|---|
| 281 | tm_sanity_check(struct tm *t)
 | 
|---|
| 282 | {
 | 
|---|
| 283 |         int ndays;
 | 
|---|
| 284 | 
 | 
|---|
| 285 |         if (t->tm_sec < 0 || t->tm_sec > 59)
 | 
|---|
| 286 |                 return EINVAL;
 | 
|---|
| 287 |         else if (t->tm_min < 0 || t->tm_min > 59)
 | 
|---|
| 288 |                 return EINVAL;
 | 
|---|
| 289 |         else if (t->tm_hour < 0 || t->tm_hour > 23)
 | 
|---|
| 290 |                 return EINVAL;
 | 
|---|
| 291 |         else if (t->tm_mday < 1 || t->tm_mday > 31)
 | 
|---|
| 292 |                 return EINVAL;
 | 
|---|
| 293 |         else if (t->tm_mon < 0 || t->tm_mon > 11)
 | 
|---|
| 294 |                 return EINVAL;
 | 
|---|
| 295 |         else if (t->tm_year < 0 || t->tm_year > 199)
 | 
|---|
| 296 |                 return EINVAL;
 | 
|---|
| 297 | 
 | 
|---|
| 298 |         if (t->tm_mon == 1/* FEB */ && is_leap_year(t->tm_year))
 | 
|---|
| 299 |                 ndays = 29;
 | 
|---|
| 300 |         else
 | 
|---|
| 301 |                 ndays = days_month[t->tm_mon];
 | 
|---|
| 302 | 
 | 
|---|
| 303 |         if (t->tm_mday > ndays)
 | 
|---|
| 304 |                 return EINVAL;
 | 
|---|
| 305 | 
 | 
|---|
| 306 |         return EOK;
 | 
|---|
| 307 | }
 | 
|---|
| 308 | 
 | 
|---|
| 309 | /** Check if a year is a leap year
 | 
|---|
| 310 |  *
 | 
|---|
| 311 |  * @param year   The year to check
 | 
|---|
| 312 |  *
 | 
|---|
| 313 |  * @return       true if it is a leap year, false otherwise
 | 
|---|
| 314 |  */
 | 
|---|
| 315 | static bool
 | 
|---|
| 316 | is_leap_year(int year)
 | 
|---|
| 317 | {
 | 
|---|
| 318 |         bool r = false;
 | 
|---|
| 319 | 
 | 
|---|
| 320 |         if (year % 4 == 0) {
 | 
|---|
| 321 |                 if (year % 100 == 0)
 | 
|---|
| 322 |                         r = year % 400 == 0;
 | 
|---|
| 323 |                 else
 | 
|---|
| 324 |                         r = true;
 | 
|---|
| 325 |         }
 | 
|---|
| 326 | 
 | 
|---|
| 327 |         return r;
 | 
|---|
| 328 | }
 | 
|---|
| 329 | 
 | 
|---|
| 330 | static void
 | 
|---|
| 331 | usage(void)
 | 
|---|
| 332 | {
 | 
|---|
| 333 |         printf("Usage: date [-d DD/MM/YYYY] [-t HH:MM[:SS]]\n");
 | 
|---|
| 334 |         printf("       -d   Change the current date\n");
 | 
|---|
| 335 |         printf("       -t   Change the current time\n");
 | 
|---|
| 336 |         printf("       -h   Display this information\n");
 | 
|---|
| 337 | }
 | 
|---|