-/* 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 <stdio.h>
+#include <locale.h>
+
+#include <captive/client-vfs.h>
+#include <captive/client.h>
+
+#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_info.h"
+#include "cmd_volume.h"
+#include "cmd_rm.h"
+#include "cmd_mv.h"
+#include "cmd_mkdir.h"
+#include "cmd_rmdir.h"
+#include "cmd_commit.h"
+#include "cmd_open.h"
+#include "cmd_create.h"
+#include "cmd_close.h"
+#include "cmd_quit.h"
+#include "cmd_help.h"
+#include "utf8.h"
+
+
+CaptiveVfsObject *cmdline_captive_vfs_object;
+
+
+GQuark cmdline_main_error_quark(void)
+{
+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 },
+ { "info" ,N_("Query information about guest-os item[1].") ,cmd_info_table ,cmd_info ,1,1 },
+ { "volume",N_("Query information about guest-os volume.") ,cmd_volume_table,cmd_volume,0,0 },
+ { "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 },
+ { "commit",N_("Write any pending changes and remount the volume.") ,cmd_commit_table,cmd_commit,0,0 },
+ { "open" ,N_("Open as[1] file[2] in mode; see 'open --help'") ,cmd_open_table ,cmd_open ,2,2 },
+ { "create",N_("Create as[1] file[2] in mode with perm; see 'create --help'"),cmd_create_table,cmd_create,2,2 },
+ { "close" ,N_("Close handle[1]") ,cmd_close_table ,cmd_close ,1,1 },
+ { "quit" ,N_("Quit this program.") ,cmd_quit_table ,cmd_quit ,0,0 },
+ { "help" ,N_("Show this list of commands or help for command[1].") ,cmd_help_table ,cmd_help ,0,1 },
+ { NULL }, /* G_N_ELEMENTS() not usable as sizeof() is not visible for 'extern' */
+ };
+
+
+static gboolean displayArgs_hit;
+
+static void displayArgs(poptContext con,enum poptCallbackReason foo,struct poptOption *key,const char *arg,void *data)
+{
+ displayArgs_hit=TRUE;
+
+ if (key->shortName=='?')
+ poptPrintHelp(con,stdout,0);
+ else
+ poptPrintUsage(con,stdout,0);
+}
+
+const struct poptOption cmdline_poptHelpOptions[]={
+ { argInfo:POPT_ARG_INTL_DOMAIN,arg:"popt" },
+ { NULL ,'\0',POPT_ARG_CALLBACK,(void *)&displayArgs,'\0',NULL,NULL },
+ { "help" ,'?' ,0 ,NULL,'?',/* N_ */("Show this help message"), NULL },
+ { "usage",'\0',0 ,NULL,'u',/* N_ */("Display brief usage message"),NULL },
+ POPT_TABLEEND
+ };
+
+
+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,argci;
+const char **csp,*cs;
+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;
+ }
+
+ for (argci=0;argci<cmd_argc;argci++) {
+ if ((cs=CMD_LOCALE_TO_UTF8_ALLOCA(cmd_argv[argci])))
+ cmd_argv[argci]=cs;
+ }
+
+ 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_LOCALE_FROM_UTF8_ALLOCA(cmd_name));
+ return;
+ }
+ displayArgs_hit=FALSE;
+ 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_LOCALE_FROM_UTF8_ALLOCA(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 '%s' reading default configuration for command: %s"),
+ poptStrerror(errint),CMD_LOCALE_FROM_UTF8_ALLOCA(cmd_name));
+ goto err_free_context;
+ }
+ 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_LOCALE_FROM_UTF8_ALLOCA(cmd_name));
+ goto err_free_context;
+ }
+ if (!(cmdarg_argv=poptGetArgs(cmd_context)))
+ cmdarg_argv=&emptyargv_NULL;
+
+ if (displayArgs_hit)
+ goto err_free_context;
+
+ 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_LOCALE_FROM_UTF8_ALLOCA(cmd_name),cmdarg_argc,commandp->argsn_min,commandp->argsn_max);
+ goto err_free_context;
+ }
-#include <captive/client.h> /* for captive_init() */
+ (*commandp->func)(cmdarg_argv,errp);
+
+err_free_context:
+ poptFreeContext(cmd_context);
+}
+
+
+void err_cleanup(GError **errp)
+{
+ g_return_if_fail(errp!=NULL);
+
+ if (!*errp)
+ return;
+ 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;
+}
+
+void main_exit(void) G_GNUC_NORETURN;
+void main_exit(void)
+{
+ if (cmdline_captive_vfs_object) {
+ g_object_unref(cmdline_captive_vfs_object);
+ cmdline_captive_vfs_object=NULL;
+ }
+ exit(EXIT_SUCCESS);
+}
+
int main(int argc,char **argv)
{
- if (!captive_init(argv[1],argv[2]))
- g_error(_("captive_init FAIL"));
+poptContext context;
+int errint;
+const char **cmd_argv,**csp;
+int cmd_argc;
+GError *gerr=NULL;
+struct captive_options options;
+
+ /* Do not set g_log_set_always_fatal() here as we would not be able
+ * to restart failed children due to communication-failure alarms.
+ */
+
+ captive_standalone_init();
+
+ captive_options_init(&options);
+ captive_options=&options; /* for parsing by 'CAPTIVE_POPT_INCLUDE' */
+
+ 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_error(_("Error parsing command-line arguments"));
+ return EXIT_FAILURE;
+ }
+ errint=poptReadDefaultConfig(context,
+ TRUE); /* useEnv */
+ if (errint!=0)
+ g_warning(_("Error reading default popt configuration"));
+ errint=poptGetNextOpt(context);
+ if (errint!=-1) {
+ g_error(_("Error parsing (dash-prefixed) command-line argument"));
+ return EXIT_FAILURE;
+ }
+ cmd_argv=poptGetArgs(context);
+ for (csp=cmd_argv,cmd_argc=0;csp && *csp;csp++)
+ cmd_argc++;
+
+ captive_options=NULL; /* already parsed by 'CAPTIVE_POPT_INCLUDE' */
+
+ /* image_iochannel */
+ if (cmd_argc<=0) {
+ g_error(_("File/device disk image pathname command-line argument required"));
+ return EXIT_FAILURE;
+ }
+ g_assert(options.image_iochannel==NULL);
+ if (!(options.image_iochannel=g_io_channel_new_file(
+ cmd_argv[0], /* filename */
+ (options.rwmode==CAPTIVE_OPTION_RWMODE_RW ? "r+" : "r"), /* mode */
+ NULL))) { /* error */
+ g_error(_("image_iochannel open failed"));
+ return EXIT_FAILURE;
+ }
+ cmd_argc--;
+ cmd_argv++;
+
+ if (options.filesystem.type==CAPTIVE_OPTIONS_MODULE_TYPE_EMPTY) {
+ g_error(_("'--filesystem' option required ('ntfs.sys' pathname suggested)"));
+ return EXIT_FAILURE;
+ }
+ if (!options.load_module) {
+ g_warning(_("'--load-module' option required ('ntoskrnl.exe' pathname suggested)"));
+ return EXIT_FAILURE;
+ }
+
+ if (GNOME_VFS_OK!=captive_vfs_new(&cmdline_captive_vfs_object,&options)) {
+ g_error(_("captive_vfs_new() failed"));
+ return EXIT_FAILURE;
+ }
+ captive_options_free(&options);
+
+ cmd_cd_internal("/",&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;
+ main_exit(); /* unref 'cmdline_captive_vfs_object' */
+ /* NOTREACHED */
}
git://git.jankratochvil.net / captive.git / blobdiff