+Real implementation of 'captive-cmdline' client
[captive.git] / src / client / cmdline / main.c
index 3b7f81e..9976446 100644 (file)
-/* FIXME: test source only! Don't pass into CVS! */
+/* $Id$
+ * client cmdline interface for libcaptive
+ * Copyright (C) 2002-2003 Jan Kratochvil <project-captive@jankratochvil.net>
+ * 
+ * This program 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; exactly version 2 of June 1991 is required
+ * 
+ * This program 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 this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
 
 #include "config.h"
 
 #include <glib/gmessages.h>
-#include <stdio.h>
 #include <stdlib.h>
 #include <glib/giochannel.h>
+#include <glib/gerror.h>
+#include <popt.h>
+#include <string.h>
+#include <glib/gprintf.h>
 
 #include <captive/client.h>    /* for captive_init() */
 
+#include "main.h"      /* self */
+#include "cmd_shell.h"
+#include "cmd_cd.h"
+#include "cmd_lcd.h"
+#include "cmd_ls.h"
+#include "cmd_get.h"
+#include "cmd_put.h"
+#include "cmd_rm.h"
+#include "cmd_mv.h"
+#include "cmd_mkdir.h"
+#include "cmd_rmdir.h"
+#include "cmd_quit.h"
+#include "cmd_help.h"
 
-void captive_sandbox_init(void);
 
-int main(int argc,char **argv)
+GQuark cmdline_main_error_quark(void)
 {
-       if (argc>3) {
-               fprintf(stderr,"Syntax: captivecmdline <captive_args> <image_filename>");
-               exit(EXIT_FAILURE);
+GQuark r=0;
+
+       if (!r)
+               r=g_quark_from_static_string("cmdline-main");
+
+       return r;
+}
+
+
+static const struct poptOption popt_table[]={
+               CAPTIVE_POPT_INCLUDE,
+               POPT_AUTOHELP
+               POPT_TABLEEND
+               };
+
+const struct cmdline_command cmdline_command_table[]={
+               /* First entry is the default if no command name was specified. */
+               { "shell",N_("Interactive commands shell.")                        ,cmd_shell_table,cmd_shell,0,0 },
+               { "cd"   ,N_("Print or change current guest-os directory[1].")     ,cmd_cd_table   ,cmd_cd   ,0,1 },
+               { "lcd"  ,N_("Print or change current host-os  directory[1].")     ,cmd_lcd_table  ,cmd_lcd  ,0,1 },
+               { "ls"   ,N_("Directory[1] listing.")                              ,cmd_ls_table   ,cmd_ls   ,0,1 },
+               { "get"  ,N_("Copy guest-os file[1] to host-os (opt. file[2]).")   ,cmd_get_table  ,cmd_get  ,1,2 },
+               { "put"  ,N_("Copy host-os file[1] to guest-os (opt. file[2]).")   ,cmd_put_table  ,cmd_put  ,1,2 },
+               { "rm"   ,N_("Remove guest-os file[1].")                           ,cmd_rm_table   ,cmd_rm   ,1,1 },
+               { "mv"   ,N_("Move (rename) guest-os item[1] to guest-os item[2]."),cmd_mv_table   ,cmd_mv   ,2,2 },
+               { "mkdir",N_("Create guest-os directory[1].")                      ,cmd_mkdir_table,cmd_mkdir,1,1 },
+               { "rmdir",N_("Remove guest-os directory[1].")                      ,cmd_rmdir_table,cmd_rmdir,1,1 },
+               { "quit" ,N_("Quit this program.")                                 ,cmd_quit_table ,cmd_quit ,0,0 },
+               { "help" ,N_("Show this list of commands.")                        ,cmd_help_table ,cmd_help ,0,0 },
+               { NULL },       /* G_N_ELEMENTS() not usable as sizeof() is not visible for 'extern' */
+               };
+
+
+static void invoke_cmd_err(int cmd_argc,const char **cmd_argv,GError **errp)
+{
+const struct cmdline_command *commandp;
+const char *cmd_name=NULL;
+poptContext cmd_context;
+int errint;
+const char **cmdarg_argv;
+int cmdarg_argc;
+const char **csp;
+const char *emptyargv_NULL=NULL;
+
+       g_return_if_fail(cmd_argc>=0);
+       g_return_if_fail(!errp || !*errp);
+
+       /* poptGetContext() cannot be passed argc==0 even if we lass POPT_CONTEXT_KEEP_FIRST
+        * as it is buggy. Workaround it by keeping the command name as argv[0].
+        */
+       if (!cmd_argc) {
+const char *stub_shell[]={ cmdline_command_table[0].name,NULL };
+               
+               cmd_argc=1;
+               cmd_argv=stub_shell;
                }
 
-       if (FALSE!=captive_init(
-                       argv[1],        /* captive_args */
-                       NULL))  /* image_iochannel */
-               g_error(_("captive_init captive_args phase FAIL"));
+       cmd_name=*cmd_argv;
+       for (commandp=cmdline_command_table;commandp->name;commandp++) {
+               if (!cmd_name   /* NULL cmd_name fallback to the first table entry - "shell" */
+                               || !strcasecmp(cmd_name,commandp->name))
+                       break;
+               }
+       if (!commandp->name) {
+               g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_UNKNOWN_COMMAND,
+                               _("Unknown command, try 'help': %s"),cmd_name);
+               return;
+               }
+       cmd_context=poptGetContext(
+                       PACKAGE,        /* name */
+                       cmd_argc,cmd_argv,      /* argc,argv */
+                       commandp->table,        /* options */
+                       POPT_CONTEXT_POSIXMEHARDER);    /* flags; !POPT_CONTEXT_KEEP_FIRST */
+       if (cmd_context==NULL) {
+               g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_INVALID_COMMAND_ARGUMENTS,
+                               _("Invalid arguments for command: %s"),cmd_name);
+               return;
+               }
+       errint=poptReadDefaultConfig(cmd_context,
+                       TRUE);  /* useEnv */
+       if (errint!=0) {
+               g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_READING_COMMAND_CONFIG,
+                               _("Error reading default configuration for command: %s"),cmd_name);
+               return;
+               }
+       errint=poptGetNextOpt(cmd_context);
+       if (errint!=-1) {
+               g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_EXCEEDING_COMMAND_OPTION,
+                               _("Exceeding command option for command: %s"),cmd_name);
+               return;
+               }
+       if (!(cmdarg_argv=poptGetArgs(cmd_context)))
+               cmdarg_argv=&emptyargv_NULL;
+       
+       for (csp=cmdarg_argv,cmdarg_argc=0;*csp;csp++)
+               cmdarg_argc++;
+       if (cmdarg_argc<commandp->argsn_min || cmdarg_argc>commandp->argsn_max) {
+               g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_INVALID_COMMAND_ARGUMENT_COUNT,
+                               _("Invalid number of command '%s' arguments: %d; expected from %d to %d incl."),
+                               cmd_name,cmdarg_argc,commandp->argsn_min,commandp->argsn_max);
+               return;
+               }
+
+       (*commandp->func)(cmdarg_argv,errp);
+
+       poptFreeContext(cmd_context);
+}
+
+
+void err_cleanup(GError **errp)
+{
+       g_return_if_fail(errp!=NULL);
 
-       if (TRUE!=captive_init(NULL,    /* captive_args */
+       if (!*errp)
+               return;
+       g_printf("\nERROR: %s\n",(*errp)->message);
+       g_clear_error(errp);
+}
+
+
+void invoke_cmd(int cmd_argc,const char **cmd_argv)
+{
+GError *gerr=NULL;
+
+       invoke_cmd_err(cmd_argc,cmd_argv,&gerr);
+       err_cleanup(&gerr);
+}
+
+
+/* Returns: Success (no error occured). */
+gboolean errvfsresult_to_gerr(GError **errp,GnomeVFSResult errvfsresult)
+{
+       g_return_val_if_fail(!errp || !*errp,FALSE);
+
+       if (errvfsresult==GNOME_VFS_OK)
+               return TRUE;
+
+       g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_GENERIC_ERROR,
+                       _("Generic error: %s"),gnome_vfs_result_to_string(errvfsresult));
+       return FALSE;
+}
+
+int main(int argc,char **argv)
+{
+poptContext context;
+int errint;
+const char **cmd_argv,**csp;
+int cmd_argc;
+GError *gerr=NULL;
+const char *cmd_cd_root_args[]={"/",NULL};
+
+       context=poptGetContext(
+                       PACKAGE,        /* name */
+                       argc,(/*en-const*/const char **)argv,   /* argc,argv */
+                       popt_table,     /* options */
+                       POPT_CONTEXT_POSIXMEHARDER);    /* flags; && !POPT_CONTEXT_KEEP_FIRST */
+       if (context==NULL) {
+               g_assert_not_reached(); /* argument recognization args_error */
+               return EXIT_FAILURE;
+               }
+       errint=poptReadDefaultConfig(context,
+                       TRUE);  /* useEnv */
+       if (errint!=0) {
+               g_assert_not_reached(); /* argument recognization args_error */
+               return EXIT_FAILURE;
+               }
+       errint=poptGetNextOpt(context);
+       if (errint!=-1) {
+               g_assert_not_reached(); /* some non-callbacked argument reached */
+               return EXIT_FAILURE;
+               }
+       cmd_argv=poptGetArgs(context);
+       for (csp=cmd_argv,cmd_argc=0;csp && *csp;csp++)
+               cmd_argc++;
+
+       if (TRUE!=captive_init(NULL,    /* captive_args; already parsed above */
                        (               /* image_iochannel */
-                                       !argv[2] ? NULL : g_io_channel_new_file(        /* FIXME: g_io_channel_new_file() is NOT 64-bit compliant! */
-                                                       argv[2],        /* filename */
-                                                       (captive_option_rwmode==CAPTIVE_OPTION_RWMODE_RW ? "w+" : "r"), /* mode */
+                                       !cmd_argc ? NULL : g_io_channel_new_file(       /* FIXME: g_io_channel_new_file() is NOT 64-bit compliant! */
+                                                       cmd_argv[cmd_argc-1],   /* filename */
+                                                       (captive_option_rwmode==CAPTIVE_OPTION_RWMODE_RW ? "r+" : "r"), /* mode */
                                                        NULL))))        /* error */
                g_error(_("captive_init image_iochannel FAIL"));
+       cmd_argc--;     /* image file */
+
+       cmd_cd(cmd_cd_root_args,&gerr);
+       if (gerr) {
+               err_cleanup(&gerr);
+               return EXIT_FAILURE;
+               }
+
+       invoke_cmd(cmd_argc,cmd_argv);
+
+       /* 'cmd_argv' gets cleared by 'poptFreeContext(context);' below */
+       poptFreeContext(context);
 
        return EXIT_SUCCESS;
 }