arch/all-pc/boot/grub2-aros/grub-core/hook/datehook.c

   1 /* datehook.c - Module to install datetime hooks.  */
   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/types.h>
  21 #include <grub/dl.h>
  22 #include <grub/env.h>
  23 #include <grub/misc.h>
  24 #include <grub/normal.h>
  25 #include <grub/datetime.h>
  26
  27 GRUB_MOD_LICENSE ("GPLv3+");
  28
  29 static const char *grub_datetime_names[] =
  30 {
  31   "YEAR",
  32   "MONTH",
  33   "DAY",
  34   "HOUR",
  35   "MINUTE",
  36   "SECOND",
  37   "WEEKDAY",
  38 };
  39
  40 static const char *
  41 grub_read_hook_datetime (struct grub_env_var *var,
  42                          const char *val __attribute__ ((unused)))
  43 {
  44   struct grub_datetime datetime;
  45   static char buf[6];
  46
  47   buf[0] = 0;
  48   if (! grub_get_datetime (&datetime))
  49     {
  50       int i;
  51
  52       for (i = 0; i < 7; i++)
  53         if (grub_strcmp (var->name, grub_datetime_names[i]) == 0)
  54           {
  55             int n;
  56
  57             switch (i)
  58               {
  59               case 0:
  60                 n = datetime.year;
  61                 break;
  62               case 1:
  63                 n = datetime.month;
  64                 break;
  65               case 2:
  66                 n = datetime.day;
  67                 break;
  68               case 3:
  69                 n = datetime.hour;
  70                 break;
  71               case 4:
  72                 n = datetime.minute;
  73                 break;
  74               case 5:
  75                 n = datetime.second;
  76                 break;
  77               default:
  78                 return grub_get_weekday_name (&datetime);
  79               }
  80
  81             grub_snprintf (buf, sizeof (buf), "%d", n);
  82             break;
  83           }
  84     }
  85
  86   return buf;
  87 }
  88
  89 GRUB_MOD_INIT(datehook)
  90 {
  91   unsigned i;
  92
  93   for (i = 0; i < ARRAY_SIZE (grub_datetime_names); i++)
  94     {
  95       grub_register_variable_hook (grub_datetime_names[i],
  96                                    grub_read_hook_datetime, 0);
  97       grub_env_export (grub_datetime_names[i]);
  98     }
  99 }
 100
 101 GRUB_MOD_FINI(datehook)
 102 {
 103   unsigned i;
 104
 105   for (i = 0; i < ARRAY_SIZE (grub_datetime_names); i++)
 106     {
 107       grub_register_variable_hook (grub_datetime_names[i], 0, 0);
 108       grub_env_unset (grub_datetime_names[i]);
 109     }
 110 }