mirror of
				https://git.proxmox.com/git/grub2
				synced 2025-10-31 10:15:17 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			286 lines
		
	
	
		
			9.5 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			286 lines
		
	
	
		
			9.5 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /*
 | |
|  *  GRUB  --  GRand Unified Bootloader
 | |
|  *  Copyright (C) 2009,2010  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/mm.h>
 | |
| #include <grub/dl.h>
 | |
| #include <grub/command.h>
 | |
| #include <grub/term.h>
 | |
| #include <grub/i18n.h>
 | |
| #include <grub/misc.h>
 | |
| 
 | |
| GRUB_MOD_LICENSE ("GPLv3+");
 | |
| 
 | |
| struct grub_term_autoload *grub_term_input_autoload = NULL;
 | |
| struct grub_term_autoload *grub_term_output_autoload = NULL;
 | |
| 
 | |
| struct abstract_terminal
 | |
| {
 | |
|   struct abstract_terminal *next;
 | |
|   struct abstract_terminal *prev;
 | |
|   const char *name;
 | |
|   grub_err_t (*init) (struct abstract_terminal *term);
 | |
|   grub_err_t (*fini) (struct abstract_terminal *term);
 | |
| };
 | |
| 
 | |
| static grub_err_t
 | |
| handle_command (int argc, char **args, struct abstract_terminal **enabled,
 | |
|                struct abstract_terminal **disabled,
 | |
|                struct grub_term_autoload *autoloads,
 | |
|                const char *active_str,
 | |
|                const char *available_str)
 | |
| {
 | |
|   int i;
 | |
|   struct abstract_terminal *term;
 | |
|   struct grub_term_autoload *aut;
 | |
| 
 | |
|   if (argc == 0)
 | |
|     {
 | |
|       grub_puts_ (active_str);
 | |
|       for (term = *enabled; term; term = term->next)
 | |
|        grub_printf ("%s ", term->name);
 | |
|       grub_printf ("\n");
 | |
|       grub_puts_ (available_str);
 | |
|       for (term = *disabled; term; term = term->next)
 | |
|        grub_printf ("%s ", term->name);
 | |
|       /* This is quadratic but we don't expect mode than 30 terminal
 | |
|         modules ever.  */
 | |
|       for (aut = autoloads; aut; aut = aut->next)
 | |
|        {
 | |
|          for (term = *disabled; term; term = term->next)
 | |
|            if (grub_strcmp (term->name, aut->name) == 0
 | |
| 	       || (aut->name[0] && aut->name[grub_strlen (aut->name) - 1] == '*'
 | |
| 		   && grub_memcmp (term->name, aut->name,
 | |
| 				   grub_strlen (aut->name) - 1) == 0))
 | |
|              break;
 | |
|          if (!term)
 | |
|            for (term = *enabled; term; term = term->next)
 | |
|              if (grub_strcmp (term->name, aut->name) == 0
 | |
| 		 || (aut->name[0] && aut->name[grub_strlen (aut->name) - 1] == '*'
 | |
| 		     && grub_memcmp (term->name, aut->name,
 | |
| 				     grub_strlen (aut->name) - 1) == 0))
 | |
|                break;
 | |
|          if (!term)
 | |
|            grub_printf ("%s ", aut->name);
 | |
|        }
 | |
|       grub_printf ("\n");
 | |
|       return GRUB_ERR_NONE;
 | |
|     }
 | |
|   i = 0;
 | |
| 
 | |
|   if (grub_strcmp (args[0], "--append") == 0
 | |
|       || grub_strcmp (args[0], "--remove") == 0)
 | |
|     i++;
 | |
| 
 | |
|   if (i == argc)
 | |
|     return grub_error (GRUB_ERR_BAD_ARGUMENT, N_ ("no terminal specified"));
 | |
| 
 | |
|   for (; i < argc; i++)
 | |
|     {
 | |
|       int again = 0;
 | |
|       while (1)
 | |
|        {
 | |
|          for (term = *disabled; term; term = term->next)
 | |
|            if (grub_strcmp (args[i], term->name) == 0
 | |
| 	       || (grub_strcmp (args[i], "ofconsole") == 0
 | |
| 		   && grub_strcmp ("console", term->name) == 0))
 | |
|              break;
 | |
|          if (term == 0)
 | |
|            for (term = *enabled; term; term = term->next)
 | |
|              if (grub_strcmp (args[i], term->name) == 0
 | |
| 		 || (grub_strcmp (args[i], "ofconsole") == 0
 | |
| 		     && grub_strcmp ("console", term->name) == 0))
 | |
|                break;
 | |
|          if (term)
 | |
|            break;
 | |
|          if (again)
 | |
| 	   return grub_error (GRUB_ERR_BAD_ARGUMENT,
 | |
| 			      N_("terminal `%s' isn't found"),
 | |
| 			      args[i]);
 | |
|          for (aut = autoloads; aut; aut = aut->next)
 | |
|            if (grub_strcmp (args[i], aut->name) == 0
 | |
| 	       || (grub_strcmp (args[i], "ofconsole") == 0
 | |
| 		   && grub_strcmp ("console", aut->name) == 0)
 | |
| 	       || (aut->name[0] && aut->name[grub_strlen (aut->name) - 1] == '*'
 | |
| 		   && grub_memcmp (args[i], aut->name,
 | |
| 				   grub_strlen (aut->name) - 1) == 0))
 | |
|              {
 | |
|                grub_dl_t mod;
 | |
|                mod = grub_dl_load (aut->modname);
 | |
|                if (mod)
 | |
|                  grub_dl_ref (mod);
 | |
|                grub_errno = GRUB_ERR_NONE;
 | |
|                break;
 | |
|              }
 | |
| 	 if (grub_memcmp (args[i], "serial_usb",
 | |
| 				  sizeof ("serial_usb") - 1) == 0
 | |
| 	     && grub_term_poll_usb)
 | |
| 	   {
 | |
| 	     grub_term_poll_usb (1);
 | |
| 	     again = 1;
 | |
| 	     continue;
 | |
| 	   }
 | |
|          if (!aut)
 | |
|            return grub_error (GRUB_ERR_BAD_ARGUMENT,
 | |
| 			      N_("terminal `%s' isn't found"),
 | |
|                               args[i]);
 | |
|          again = 1;
 | |
|        }
 | |
|     }
 | |
| 
 | |
|   if (grub_strcmp (args[0], "--append") == 0)
 | |
|     {
 | |
|       for (i = 1; i < argc; i++)
 | |
|        {
 | |
|          for (term = *disabled; term; term = term->next)
 | |
|            if (grub_strcmp (args[i], term->name) == 0
 | |
| 	       || (grub_strcmp (args[i], "ofconsole") == 0
 | |
| 		   && grub_strcmp ("console", term->name) == 0))
 | |
|              break;
 | |
|          if (term)
 | |
|            {
 | |
|               if (term->init && term->init (term) != GRUB_ERR_NONE)
 | |
|                 return grub_errno;
 | |
| 
 | |
|              grub_list_remove (GRUB_AS_LIST (term));
 | |
|              grub_list_push (GRUB_AS_LIST_P (enabled), GRUB_AS_LIST (term));
 | |
|            }
 | |
|        }
 | |
|       return GRUB_ERR_NONE;
 | |
|     }
 | |
| 
 | |
|   if (grub_strcmp (args[0], "--remove") == 0)
 | |
|     {
 | |
|       for (i = 1; i < argc; i++)
 | |
|        {
 | |
|          for (term = *enabled; term; term = term->next)
 | |
|            if (grub_strcmp (args[i], term->name) == 0
 | |
| 	       || (grub_strcmp (args[i], "ofconsole") == 0
 | |
| 		   && grub_strcmp ("console", term->name) == 0))
 | |
|              break;
 | |
|          if (term)
 | |
|            {
 | |
|              if (!term->next && term == *enabled)
 | |
|                return grub_error (GRUB_ERR_BAD_ARGUMENT,
 | |
|                                   "can't remove the last terminal");
 | |
|              grub_list_remove (GRUB_AS_LIST (term));
 | |
|              if (term->fini)
 | |
|                term->fini (term);
 | |
|              grub_list_push (GRUB_AS_LIST_P (disabled), GRUB_AS_LIST (term));
 | |
|            }
 | |
|        }
 | |
|       return GRUB_ERR_NONE;
 | |
|     }
 | |
|   for (i = 0; i < argc; i++)
 | |
|     {
 | |
|       for (term = *disabled; term; term = term->next)
 | |
|        if (grub_strcmp (args[i], term->name) == 0
 | |
| 	   || (grub_strcmp (args[i], "ofconsole") == 0
 | |
| 	       && grub_strcmp ("console", term->name) == 0))
 | |
|          break;
 | |
|       if (term)
 | |
|        {
 | |
|          if (term->init && term->init (term) != GRUB_ERR_NONE)
 | |
|            return grub_errno;
 | |
| 
 | |
|          grub_list_remove (GRUB_AS_LIST (term));
 | |
|          grub_list_push (GRUB_AS_LIST_P (enabled), GRUB_AS_LIST (term));
 | |
|        }       
 | |
|     }
 | |
|   
 | |
|   {
 | |
|     struct abstract_terminal *next;
 | |
|     for (term = *enabled; term; term = next)
 | |
|       {
 | |
|        next = term->next;
 | |
|        for (i = 0; i < argc; i++)
 | |
|          if (grub_strcmp (args[i], term->name) == 0
 | |
| 	     || (grub_strcmp (args[i], "ofconsole") == 0
 | |
| 		 && grub_strcmp ("console", term->name) == 0))
 | |
|            break;
 | |
|        if (i == argc)
 | |
|          {
 | |
|            if (!term->next && term == *enabled)
 | |
|              return grub_error (GRUB_ERR_BAD_ARGUMENT,
 | |
|                                 "can't remove the last terminal");
 | |
|            grub_list_remove (GRUB_AS_LIST (term));
 | |
|            if (term->fini)
 | |
|              term->fini (term);
 | |
|            grub_list_push (GRUB_AS_LIST_P (disabled), GRUB_AS_LIST (term));
 | |
|          }
 | |
|       }
 | |
|   }
 | |
| 
 | |
|   return GRUB_ERR_NONE;
 | |
| }
 | |
| 
 | |
| static grub_err_t
 | |
| grub_cmd_terminal_input (grub_command_t cmd __attribute__ ((unused)),
 | |
| 			 int argc, char **args)
 | |
| {
 | |
|   (void) GRUB_FIELD_MATCH (grub_term_inputs, struct abstract_terminal *, next);
 | |
|   (void) GRUB_FIELD_MATCH (grub_term_inputs, struct abstract_terminal *, prev);
 | |
|   (void) GRUB_FIELD_MATCH (grub_term_inputs, struct abstract_terminal *, name);
 | |
|   (void) GRUB_FIELD_MATCH (grub_term_inputs, struct abstract_terminal *, init);
 | |
|   (void) GRUB_FIELD_MATCH (grub_term_inputs, struct abstract_terminal *, fini);
 | |
|   return handle_command (argc, args,
 | |
| 			 (struct abstract_terminal **) (void *) &grub_term_inputs,
 | |
| 			 (struct abstract_terminal **) (void *) &grub_term_inputs_disabled,
 | |
| 			 grub_term_input_autoload,
 | |
| 			 N_ ("Active input terminals:"),
 | |
| 			 N_ ("Available input terminals:"));
 | |
| }
 | |
| 
 | |
| static grub_err_t
 | |
| grub_cmd_terminal_output (grub_command_t cmd __attribute__ ((unused)),
 | |
|                          int argc, char **args)
 | |
| {
 | |
|   (void) GRUB_FIELD_MATCH (grub_term_outputs, struct abstract_terminal *, next);
 | |
|   (void) GRUB_FIELD_MATCH (grub_term_outputs, struct abstract_terminal *, prev);
 | |
|   (void) GRUB_FIELD_MATCH (grub_term_outputs, struct abstract_terminal *, name);
 | |
|   (void) GRUB_FIELD_MATCH (grub_term_outputs, struct abstract_terminal *, init);
 | |
|   (void) GRUB_FIELD_MATCH (grub_term_outputs, struct abstract_terminal *, fini);
 | |
|   return handle_command (argc, args,
 | |
| 			 (struct abstract_terminal **) (void *) &grub_term_outputs,
 | |
| 			 (struct abstract_terminal **) (void *) &grub_term_outputs_disabled,
 | |
| 			 grub_term_output_autoload,
 | |
| 			 N_ ("Active output terminals:"),
 | |
| 			 N_ ("Available output terminals:"));
 | |
| }
 | |
| 
 | |
| static grub_command_t cmd_terminal_input, cmd_terminal_output;
 | |
| 
 | |
| GRUB_MOD_INIT(terminal)
 | |
| {
 | |
|   cmd_terminal_input =
 | |
|     grub_register_command ("terminal_input", grub_cmd_terminal_input,
 | |
| 			   N_("[--append|--remove] "
 | |
| 			      "[TERMINAL1] [TERMINAL2] ..."),
 | |
| 			   N_("List or select an input terminal."));
 | |
|   cmd_terminal_output =
 | |
|     grub_register_command ("terminal_output", grub_cmd_terminal_output,
 | |
| 			   N_("[--append|--remove] "
 | |
| 			      "[TERMINAL1] [TERMINAL2] ..."),
 | |
| 			   N_("List or select an output terminal."));
 | |
| }
 | |
| 
 | |
| GRUB_MOD_FINI(terminal)
 | |
| {
 | |
|   grub_unregister_command (cmd_terminal_input);
 | |
|   grub_unregister_command (cmd_terminal_output);
 | |
| }
 | 
