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