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/dl.h>
  21 #include <grub/err.h>
  22 #include <grub/misc.h>
  23 #include <grub/datetime.h>
  24 #include <grub/command.h>
  25
  26 #define GRUB_DATETIME_SET_YEAR          1
  27 #define GRUB_DATETIME_SET_MONTH         2
  28 #define GRUB_DATETIME_SET_DAY           4
  29 #define GRUB_DATETIME_SET_HOUR          8
  30 #define GRUB_DATETIME_SET_MINUTE        16
  31 #define GRUB_DATETIME_SET_SECOND        32
  32
  33 static grub_err_t
  34 grub_cmd_date (grub_command_t cmd __attribute__ ((unused)),
  35                int argc, char **args)
  36 {
  37   struct grub_datetime datetime;
  38   int limit[6][2] = {{1980, 2079}, {1, 12}, {1, 31}, {0, 23}, {0, 59}, {0, 59}};
  39   int value[6], mask;
  40
  41   if (argc == 0)
  42     {
  43       if (grub_get_datetime (&datetime))
  44         return grub_errno;
  45
  46       grub_printf ("%d-%02d-%02d %02d:%02d:%02d %s\n",
  47                    datetime.year, datetime.month, datetime.day,
  48                    datetime.hour, datetime.minute, datetime.second,
  49                    grub_get_weekday_name (&datetime));
  50
  51       return 0;
  52     }
  53
  54   grub_memset (&value, 0, sizeof (value));
  55   mask = 0;
  56
  57   for (; argc; argc--, args++)
  58     {
  59       char *p, c;
  60       int m1, ofs, n, cur_mask;
  61
  62       p = args[0];
  63       m1 = grub_strtoul (p, &p, 10);
  64
  65       c = *p;
  66       if (c == '-')
  67         ofs = 0;
  68       else if (c == ':')
  69         ofs = 3;
  70       else
  71         goto fail;
  72
  73       value[ofs] = m1;
  74       cur_mask = (1 << ofs);
  75       mask &= ~(cur_mask * (1 + 2 + 4));
  76
  77       for (n = 1; (n < 3) && (*p); n++)
  78         {
  79           if (*p != c)
  80             goto fail;
  81
  82           value[ofs + n] = grub_strtoul (p + 1, &p, 10);
  83           cur_mask |= (1 << (ofs + n));
  84         }
  85
  86       if (*p)
  87         goto fail;
  88
  89       if ((ofs == 0) && (n == 2))
  90         {
  91           value[ofs + 2] = value[ofs + 1];
  92           value[ofs + 1] = value[ofs];
  93           ofs++;
  94           cur_mask <<= 1;
  95         }
  96
  97       for (; n; n--, ofs++)
  98         if ((value [ofs] < limit[ofs][0]) ||
  99             (value [ofs] > limit[ofs][1]))
 100           goto fail;
 101
 102       mask |= cur_mask;
 103     }
 104
 105   if (grub_get_datetime (&datetime))
 106     return grub_errno;
 107
 108   if (mask & GRUB_DATETIME_SET_YEAR)
 109     datetime.year = value[0];
 110
 111   if (mask & GRUB_DATETIME_SET_MONTH)
 112     datetime.month = value[1];
 113
 114   if (mask & GRUB_DATETIME_SET_DAY)
 115     datetime.day = value[2];
 116
 117   if (mask & GRUB_DATETIME_SET_HOUR)
 118     datetime.hour = value[3];
 119
 120   if (mask & GRUB_DATETIME_SET_MINUTE)
 121     datetime.minute = value[4];
 122
 123   if (mask & GRUB_DATETIME_SET_SECOND)
 124     datetime.second = value[5];
 125
 126   return grub_set_datetime (&datetime);
 127
 128 fail:
 129   return grub_error (GRUB_ERR_BAD_ARGUMENT, "invalid datetime");
 130 }
 131
 132 static grub_command_t cmd;
 133
 134 GRUB_MOD_INIT(date)
 135 {
 136   cmd =
 137     grub_register_command ("date", grub_cmd_date,
 138                            "date [[year-]month-day] [hour:minute[:second]]",
 139                            "Command to display/set current datetime.");
 140 }
 141
 142 GRUB_MOD_FINI(date)
 143 {
 144   grub_unregister_command (cmd);
 145 }