commands/date.c

   1 /* date.c - command to display/set current datetime.  */
   2 /*
   3  *  GRUB  --  GRand Unified Bootloader
   4  *  Copyright (C) 2008  Free Software Foundation, Inc.
   5  *
   6  *  GRUB is free software: you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation, either version 3 of the License, or
   9  *  (at your option) any later version.
  10  *
  11  *  GRUB 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 GRUB.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19
  20 #include <grub/normal.h>
  21 #include <grub/dl.h>
  22 #include <grub/arg.h>
  23 #include <grub/err.h>
  24 #include <grub/misc.h>
  25 #include <grub/lib/datetime.h>
  26
  27 #define GRUB_DATETIME_SET_YEAR          1
  28 #define GRUB_DATETIME_SET_MONTH         2
  29 #define GRUB_DATETIME_SET_DAY           4
  30 #define GRUB_DATETIME_SET_HOUR          8
  31 #define GRUB_DATETIME_SET_MINUTE        16
  32 #define GRUB_DATETIME_SET_SECOND        32
  33
  34 static grub_err_t
  35 grub_cmd_date (struct grub_arg_list *state __attribute__ ((unused)),
  36                int argc, char **args)
  37 {
  38   struct grub_datetime datetime;
  39   int limit[6][2] = {{1980, 2079}, {1, 12}, {1, 31}, {0, 23}, {0, 59}, {0, 59}};
  40   int value[6], mask;
  41
  42   if (argc == 0)
  43     {
  44       if (grub_get_datetime (&datetime))
  45         return grub_errno;
  46
  47       grub_printf ("%d-%02d-%02d %02d:%02d:%02d %s\n",
  48                    datetime.year, datetime.month, datetime.day,
  49                    datetime.hour, datetime.minute, datetime.second,
  50                    grub_get_weekday_name (&datetime));
  51
  52       return 0;
  53     }
  54
  55   grub_memset (&value, 0, sizeof (value));
  56   mask = 0;
  57
  58   for (; argc; argc--, args++)
  59     {
  60       char *p, c;
  61       int m1, ofs, n, cur_mask;
  62
  63       p = args[0];
  64       m1 = grub_strtoul (p, &p, 10);
  65
  66       c = *p;
  67       if (c == '-')
  68         ofs = 0;
  69       else if (c == ':')
  70         ofs = 3;
  71       else
  72         goto fail;
  73
  74       value[ofs] = m1;
  75       cur_mask = (1 << ofs);
  76       mask &= ~(cur_mask * (1 + 2 + 4));
  77
  78       for (n = 1; (n < 3) && (*p); n++)
  79         {
  80           if (*p != c)
  81             goto fail;
  82
  83           value[ofs + n] = grub_strtoul (p + 1, &p, 10);
  84           cur_mask |= (1 << (ofs + n));
  85         }
  86
  87       if (*p)
  88         goto fail;
  89
  90       if ((ofs == 0) && (n == 2))
  91         {
  92           value[ofs + 2] = value[ofs + 1];
  93           value[ofs + 1] = value[ofs];
  94           ofs++;
  95           cur_mask <<= 1;
  96         }
  97
  98       for (; n; n--, ofs++)
  99         if ((value [ofs] < limit[ofs][0]) ||
 100             (value [ofs] > limit[ofs][1]))
 101           goto fail;
 102
 103       mask |= cur_mask;
 104     }
 105
 106   if (grub_get_datetime (&datetime))
 107     return grub_errno;
 108
 109   if (mask & GRUB_DATETIME_SET_YEAR)
 110     datetime.year = value[0];
 111
 112   if (mask & GRUB_DATETIME_SET_MONTH)
 113     datetime.month = value[1];
 114
 115   if (mask & GRUB_DATETIME_SET_DAY)
 116     datetime.day = value[2];
 117
 118   if (mask & GRUB_DATETIME_SET_HOUR)
 119     datetime.hour = value[3];
 120
 121   if (mask & GRUB_DATETIME_SET_MINUTE)
 122     datetime.minute = value[4];
 123
 124   if (mask & GRUB_DATETIME_SET_SECOND)
 125     datetime.second = value[5];
 126
 127   return grub_set_datetime (&datetime);
 128
 129 fail:
 130   return grub_error (GRUB_ERR_BAD_ARGUMENT, "invalid datetime");
 131 }
 132
 133 GRUB_MOD_INIT(date)
 134 {
 135   (void) mod;                   /* To stop warning. */
 136   grub_register_command ("date", grub_cmd_date,
 137                          GRUB_COMMAND_FLAG_BOTH,
 138                          "date [[year-]month-day] [hour:minute[:second]]",
 139                          "Command to display/set current datetime.", 0);
 140 }
 141
 142 GRUB_MOD_FINI(date)
 143 {
 144   grub_unregister_command ("date");
 145 }