* normal/dyncmd.c (read_command_list): Replace `0' with `NULL'.
        * normal/handler.c (read_handler_list): Likewise.
		
	
			
		
			
				
	
	
		
			158 lines
		
	
	
	
		
			3.2 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			158 lines
		
	
	
	
		
			3.2 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /* dyncmd.c - support dynamic command */
 | |
| /*
 | |
|  *  GRUB  --  GRand Unified Bootloader
 | |
|  *  Copyright (C) 2009  Free Software Foundation, Inc.
 | |
|  *
 | |
|  *  GRUB is free software: you can redistribute it and/or modify
 | |
|  *  it under the terms of the GNU General Public License as published by
 | |
|  *  the Free Software Foundation, either version 3 of the License, or
 | |
|  *  (at your option) any later version.
 | |
|  *
 | |
|  *  GRUB is distributed in the hope that it will be useful,
 | |
|  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 | |
|  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | |
|  *  GNU General Public License for more details.
 | |
|  *
 | |
|  *  You should have received a copy of the GNU General Public License
 | |
|  *  along with GRUB.  If not, see <http://www.gnu.org/licenses/>.
 | |
|  */
 | |
| 
 | |
| #include <grub/dl.h>
 | |
| #include <grub/mm.h>
 | |
| #include <grub/env.h>
 | |
| #include <grub/misc.h>
 | |
| #include <grub/command.h>
 | |
| #include <grub/normal.h>
 | |
| 
 | |
| static grub_err_t
 | |
| grub_dyncmd_dispatcher (struct grub_command *cmd,
 | |
| 			int argc, char **args)
 | |
| {
 | |
|   char *modname = cmd->data;
 | |
|   grub_dl_t mod;
 | |
|   grub_err_t ret;
 | |
| 
 | |
|   mod = grub_dl_load (modname);
 | |
|   if (mod)
 | |
|     {
 | |
|       char *name;
 | |
| 
 | |
|       grub_free (modname);
 | |
|       grub_dl_ref (mod);
 | |
| 
 | |
|       name = (char *) cmd->name;
 | |
|       grub_unregister_command (cmd);
 | |
| 
 | |
|       cmd = grub_command_find (name);
 | |
|       if (cmd)
 | |
| 	ret = (cmd->func) (cmd, argc, args);
 | |
|       else
 | |
| 	ret = grub_errno;
 | |
| 
 | |
|       grub_free (name);
 | |
|     }
 | |
|   else
 | |
|     ret = grub_errno;
 | |
| 
 | |
|   return ret;
 | |
| }
 | |
| 
 | |
| /* Read the file command.lst for auto-loading.  */
 | |
| void
 | |
| read_command_list (void)
 | |
| {
 | |
|   const char *prefix;
 | |
|   static int first_time = 1;
 | |
| 
 | |
|   /* Make sure that this function does not get executed twice.  */
 | |
|   if (! first_time)
 | |
|     return;
 | |
|   first_time = 0;
 | |
| 
 | |
|   prefix = grub_env_get ("prefix");
 | |
|   if (prefix)
 | |
|     {
 | |
|       char *filename;
 | |
| 
 | |
|       filename = grub_malloc (grub_strlen (prefix) + sizeof ("/command.lst"));
 | |
|       if (filename)
 | |
| 	{
 | |
| 	  grub_file_t file;
 | |
| 
 | |
| 	  grub_sprintf (filename, "%s/command.lst", prefix);
 | |
| 	  file = grub_file_open (filename);
 | |
| 	  if (file)
 | |
| 	    {
 | |
| 	      char *buf = NULL;
 | |
| 	      for (;; grub_free (buf))
 | |
| 		{
 | |
| 		  char *p, *name, *modname;
 | |
| 		  grub_command_t cmd;
 | |
| 		  int prio = 0;
 | |
| 
 | |
| 		  buf = grub_file_getline (file);
 | |
| 
 | |
| 		  if (! buf)
 | |
| 		    break;
 | |
| 
 | |
| 		  name = buf;
 | |
| 		  if (*name == '*')
 | |
| 		    {
 | |
| 		      name++;
 | |
| 		      prio++;
 | |
| 		    }
 | |
| 
 | |
| 		  if (! grub_isgraph (name[0]))
 | |
| 		    continue;
 | |
| 
 | |
| 		  p = grub_strchr (name, ':');
 | |
| 		  if (! p)
 | |
| 		    continue;
 | |
| 
 | |
| 		  *p = '\0';
 | |
| 		  while (*++p == ' ')
 | |
| 		    ;
 | |
| 
 | |
| 		  if (! grub_isgraph (*p))
 | |
| 		    continue;
 | |
| 
 | |
| 		  if (grub_dl_get (p))
 | |
| 		    continue;
 | |
| 
 | |
| 		  name = grub_strdup (name);
 | |
| 		  if (! name)
 | |
| 		    continue;
 | |
| 
 | |
| 		  modname = grub_strdup (p);
 | |
| 		  if (! modname)
 | |
| 		    {
 | |
| 		      grub_free (name);
 | |
| 		      continue;
 | |
| 		    }
 | |
| 
 | |
| 		  cmd = grub_register_command_prio (name,
 | |
| 						    grub_dyncmd_dispatcher,
 | |
| 						    0, "not loaded", prio);
 | |
| 		  if (! cmd)
 | |
| 		    {
 | |
| 		      grub_free (name);
 | |
| 		      grub_free (modname);
 | |
| 		      continue;
 | |
| 		    }
 | |
| 		  cmd->flags |= GRUB_COMMAND_FLAG_DYNCMD;
 | |
| 		  cmd->data = modname;
 | |
| 
 | |
| 		  /* Update the active flag.  */
 | |
| 		  grub_command_find (name);
 | |
| 		}
 | |
| 
 | |
| 	      grub_file_close (file);
 | |
| 	    }
 | |
| 
 | |
| 	  grub_free (filename);
 | |
| 	}
 | |
|     }
 | |
| 
 | |
|   /* Ignore errors.  */
 | |
|   grub_errno = GRUB_ERR_NONE;
 | |
| }
 |