src/client/cmdline/main.c

   1 /* $Id$
   2  * client cmdline interface for libcaptive
   3  * Copyright (C) 2002-2003 Jan Kratochvil <project-captive@jankratochvil.net>
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; exactly version 2 of June 1991 is required
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17  */
  18
  19
  20 #include "config.h"
  21
  22 #include <glib/gmessages.h>
  23 #include <stdlib.h>
  24 #include <glib/giochannel.h>
  25 #include <glib/gerror.h>
  26 #include <popt.h>
  27 #include <string.h>
  28 #include <stdio.h>
  29
  30 #include <captive/client.h>     /* for captive_init() */
  31
  32 #include "main.h"       /* self */
  33 #include "cmd_shell.h"
  34 #include "cmd_cd.h"
  35 #include "cmd_lcd.h"
  36 #include "cmd_ls.h"
  37 #include "cmd_get.h"
  38 #include "cmd_put.h"
  39 #include "cmd_info.h"
  40 #include "cmd_rm.h"
  41 #include "cmd_mv.h"
  42 #include "cmd_mkdir.h"
  43 #include "cmd_rmdir.h"
  44 #include "cmd_quit.h"
  45 #include "cmd_help.h"
  46
  47
  48 GQuark cmdline_main_error_quark(void)
  49 {
  50 GQuark r=0;
  51
  52         if (!r)
  53                 r=g_quark_from_static_string("cmdline-main");
  54
  55         return r;
  56 }
  57
  58
  59 static const struct poptOption popt_table[]={
  60                 CAPTIVE_POPT_INCLUDE,
  61                 POPT_AUTOHELP
  62                 POPT_TABLEEND
  63                 };
  64
  65 const struct cmdline_command cmdline_command_table[]={
  66                 /* First entry is the default if no command name was specified. */
  67                 { "shell",N_("Interactive commands shell.")                        ,cmd_shell_table,cmd_shell,0,0 },
  68                 { "cd"   ,N_("Print or change current guest-os directory[1].")     ,cmd_cd_table   ,cmd_cd   ,0,1 },
  69                 { "lcd"  ,N_("Print or change current host-os  directory[1].")     ,cmd_lcd_table  ,cmd_lcd  ,0,1 },
  70                 { "ls"   ,N_("Directory[1] listing.")                              ,cmd_ls_table   ,cmd_ls   ,0,1 },
  71                 { "get"  ,N_("Copy guest-os file[1] to host-os (opt. file[2]).")   ,cmd_get_table  ,cmd_get  ,1,2 },
  72                 { "put"  ,N_("Copy host-os file[1] to guest-os (opt. file[2]).")   ,cmd_put_table  ,cmd_put  ,1,2 },
  73                 { "info" ,N_("Query information about guest-os item[1].")          ,cmd_info_table ,cmd_info ,1,1 },
  74                 { "rm"   ,N_("Remove guest-os file[1].")                           ,cmd_rm_table   ,cmd_rm   ,1,1 },
  75                 { "mv"   ,N_("Move (rename) guest-os item[1] to guest-os item[2]."),cmd_mv_table   ,cmd_mv   ,2,2 },
  76                 { "mkdir",N_("Create guest-os directory[1].")                      ,cmd_mkdir_table,cmd_mkdir,1,1 },
  77                 { "rmdir",N_("Remove guest-os directory[1].")                      ,cmd_rmdir_table,cmd_rmdir,1,1 },
  78                 { "quit" ,N_("Quit this program.")                                 ,cmd_quit_table ,cmd_quit ,0,0 },
  79                 { "help" ,N_("Show this list of commands.")                        ,cmd_help_table ,cmd_help ,0,0 },
  80                 { NULL },       /* G_N_ELEMENTS() not usable as sizeof() is not visible for 'extern' */
  81                 };
  82
  83
  84 static void invoke_cmd_err(int cmd_argc,const char **cmd_argv,GError **errp)
  85 {
  86 const struct cmdline_command *commandp;
  87 const char *cmd_name=NULL;
  88 poptContext cmd_context;
  89 int errint;
  90 const char **cmdarg_argv;
  91 int cmdarg_argc;
  92 const char **csp;
  93 const char *emptyargv_NULL=NULL;
  94
  95         g_return_if_fail(cmd_argc>=0);
  96         g_return_if_fail(!errp || !*errp);
  97
  98         /* poptGetContext() cannot be passed argc==0 even if we lass POPT_CONTEXT_KEEP_FIRST
  99          * as it is buggy. Workaround it by keeping the command name as argv[0].
 100          */
 101         if (!cmd_argc) {
 102 const char *stub_shell[]={ cmdline_command_table[0].name,NULL };
 103
 104                 cmd_argc=1;
 105                 cmd_argv=stub_shell;
 106                 }
 107
 108         cmd_name=*cmd_argv;
 109         for (commandp=cmdline_command_table;commandp->name;commandp++) {
 110                 if (!cmd_name   /* NULL cmd_name fallback to the first table entry - "shell" */
 111                                 || !strcasecmp(cmd_name,commandp->name))
 112                         break;
 113                 }
 114         if (!commandp->name) {
 115                 g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_UNKNOWN_COMMAND,
 116                                 _("Unknown command, try 'help': %s"),cmd_name);
 117                 return;
 118                 }
 119         cmd_context=poptGetContext(
 120                         PACKAGE,        /* name */
 121                         cmd_argc,cmd_argv,      /* argc,argv */
 122                         commandp->table,        /* options */
 123                         POPT_CONTEXT_POSIXMEHARDER);    /* flags; !POPT_CONTEXT_KEEP_FIRST */
 124         if (cmd_context==NULL) {
 125                 g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_INVALID_COMMAND_ARGUMENTS,
 126                                 _("Invalid arguments for command: %s"),cmd_name);
 127                 return;
 128                 }
 129         errint=poptReadDefaultConfig(cmd_context,
 130                         TRUE);  /* useEnv */
 131         if (errint!=0) {
 132                 g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_READING_COMMAND_CONFIG,
 133                                 _("Error '%s' reading default configuration for command: %s"),poptStrerror(errint),cmd_name);
 134                 return;
 135                 }
 136         errint=poptGetNextOpt(cmd_context);
 137         if (errint!=-1) {
 138                 g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_EXCEEDING_COMMAND_OPTION,
 139                                 _("Exceeding command option for command: %s"),cmd_name);
 140                 return;
 141                 }
 142         if (!(cmdarg_argv=poptGetArgs(cmd_context)))
 143                 cmdarg_argv=&emptyargv_NULL;
 144
 145         for (csp=cmdarg_argv,cmdarg_argc=0;*csp;csp++)
 146                 cmdarg_argc++;
 147         if (cmdarg_argc<commandp->argsn_min || cmdarg_argc>commandp->argsn_max) {
 148                 g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_INVALID_COMMAND_ARGUMENT_COUNT,
 149                                 _("Invalid number of command '%s' arguments: %d; expected from %d to %d incl."),
 150                                 cmd_name,cmdarg_argc,commandp->argsn_min,commandp->argsn_max);
 151                 return;
 152                 }
 153
 154         (*commandp->func)(cmdarg_argv,errp);
 155
 156         poptFreeContext(cmd_context);
 157 }
 158
 159
 160 void err_cleanup(GError **errp)
 161 {
 162         g_return_if_fail(errp!=NULL);
 163
 164         if (!*errp)
 165                 return;
 166         printf("\nERROR: %s\n",(*errp)->message);
 167         g_clear_error(errp);
 168 }
 169
 170
 171 void invoke_cmd(int cmd_argc,const char **cmd_argv)
 172 {
 173 GError *gerr=NULL;
 174
 175         invoke_cmd_err(cmd_argc,cmd_argv,&gerr);
 176         err_cleanup(&gerr);
 177 }
 178
 179
 180 /* Returns: Success (no error occured). */
 181 gboolean errvfsresult_to_gerr(GError **errp,GnomeVFSResult errvfsresult)
 182 {
 183         g_return_val_if_fail(!errp || !*errp,FALSE);
 184
 185         if (errvfsresult==GNOME_VFS_OK)
 186                 return TRUE;
 187
 188         g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_GENERIC_ERROR,
 189                         _("Generic error: %s"),gnome_vfs_result_to_string(errvfsresult));
 190         return FALSE;
 191 }
 192
 193 int main(int argc,char **argv)
 194 {
 195 poptContext context;
 196 int errint;
 197 const char **cmd_argv,**csp;
 198 int cmd_argc;
 199 GError *gerr=NULL;
 200 const char *cmd_cd_root_args[]={"/",NULL};
 201
 202         context=poptGetContext(
 203                         PACKAGE,        /* name */
 204                         argc,(/*en-const*/const char **)argv,   /* argc,argv */
 205                         popt_table,     /* options */
 206                         POPT_CONTEXT_POSIXMEHARDER);    /* flags; && !POPT_CONTEXT_KEEP_FIRST */
 207         if (context==NULL) {
 208                 g_assert_not_reached(); /* argument recognization args_error */
 209                 return EXIT_FAILURE;
 210                 }
 211         errint=poptReadDefaultConfig(context,
 212                         TRUE);  /* useEnv */
 213         if (errint!=0) {
 214                 g_assert_not_reached(); /* argument recognization args_error */
 215                 return EXIT_FAILURE;
 216                 }
 217         errint=poptGetNextOpt(context);
 218         if (errint!=-1) {
 219                 g_assert_not_reached(); /* some non-callbacked argument reached */
 220                 return EXIT_FAILURE;
 221                 }
 222         cmd_argv=poptGetArgs(context);
 223         for (csp=cmd_argv,cmd_argc=0;csp && *csp;csp++)
 224                 cmd_argc++;
 225
 226         if (TRUE!=captive_init(NULL,    /* captive_args; already parsed above */
 227                         (               /* image_iochannel */
 228                                         !cmd_argc ? NULL : g_io_channel_new_file(       /* FIXME: g_io_channel_new_file() is NOT 64-bit compliant! */
 229                                                         cmd_argv[0],    /* filename */
 230                                                         (captive_option_rwmode==CAPTIVE_OPTION_RWMODE_RW ? "r+" : "r"), /* mode */
 231                                                         NULL))))        /* error */
 232                 g_error(_("captive_init image_iochannel FAIL"));
 233         if (cmd_argc>0) {
 234                 /* image file */
 235                 cmd_argc--;
 236                 cmd_argv++;
 237                 }
 238
 239         cmd_cd(cmd_cd_root_args,&gerr);
 240         if (gerr) {
 241                 err_cleanup(&gerr);
 242                 return EXIT_FAILURE;
 243                 }
 244
 245         invoke_cmd(cmd_argc,cmd_argv);
 246
 247         /* 'cmd_argv' gets cleared by 'poptFreeContext(context);' below */
 248         poptFreeContext(context);
 249
 250         return EXIT_SUCCESS;
 251 }