2 * client cmdline interface for libcaptive
3 * Copyright (C) 2002-2003 Jan Kratochvil <project-captive@jankratochvil.net>
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
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.
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
22 #include <glib/gmessages.h>
24 #include <glib/giochannel.h>
25 #include <glib/gerror.h>
31 #include <captive/client-vfs.h>
33 #include "main.h" /* self */
34 #include "cmd_shell.h"
41 #include "cmd_volume.h"
44 #include "cmd_mkdir.h"
45 #include "cmd_rmdir.h"
46 #include "cmd_commit.h"
51 CaptiveVfsObject *cmdline_captive_vfs_object;
54 GQuark cmdline_main_error_quark(void)
59 r=g_quark_from_static_string("cmdline-main");
65 static const struct poptOption popt_table[]={
71 const struct cmdline_command cmdline_command_table[]={
72 /* First entry is the default if no command name was specified. */
73 { "shell" ,N_("Interactive commands shell.") ,cmd_shell_table ,cmd_shell ,0,0 },
74 { "cd" ,N_("Print or change current guest-os directory[1].") ,cmd_cd_table ,cmd_cd ,0,1 },
75 { "lcd" ,N_("Print or change current host-os directory[1].") ,cmd_lcd_table ,cmd_lcd ,0,1 },
76 { "ls" ,N_("Directory[1] listing.") ,cmd_ls_table ,cmd_ls ,0,1 },
77 { "get" ,N_("Copy guest-os file[1] to host-os (opt. file[2]).") ,cmd_get_table ,cmd_get ,1,2 },
78 { "put" ,N_("Copy host-os file[1] to guest-os (opt. file[2]).") ,cmd_put_table ,cmd_put ,1,2 },
79 { "info" ,N_("Query information about guest-os item[1].") ,cmd_info_table ,cmd_info ,1,1 },
80 { "volume",N_("Query information about guest-os volume.") ,cmd_volume_table,cmd_volume,0,0 },
81 { "rm" ,N_("Remove guest-os file[1].") ,cmd_rm_table ,cmd_rm ,1,1 },
82 { "mv" ,N_("Move (rename) guest-os item[1] to guest-os item[2]."),cmd_mv_table ,cmd_mv ,2,2 },
83 { "mkdir" ,N_("Create guest-os directory[1].") ,cmd_mkdir_table ,cmd_mkdir ,1,1 },
84 { "rmdir" ,N_("Remove guest-os directory[1].") ,cmd_rmdir_table ,cmd_rmdir ,1,1 },
85 { "commit",N_("Write any pending changes and remount the volume.") ,cmd_commit_table,cmd_commit,0,0 },
86 { "quit" ,N_("Quit this program.") ,cmd_quit_table ,cmd_quit ,0,0 },
87 { "help" ,N_("Show this list of commands or help for command[1].") ,cmd_help_table ,cmd_help ,0,1 },
88 { NULL }, /* G_N_ELEMENTS() not usable as sizeof() is not visible for 'extern' */
92 static gboolean displayArgs_hit;
94 static void displayArgs(poptContext con,enum poptCallbackReason foo,struct poptOption *key,const char *arg,void *data)
98 if (key->shortName=='?')
99 poptPrintHelp(con,stdout,0);
101 poptPrintUsage(con,stdout,0);
104 const struct poptOption cmdline_poptHelpOptions[]={
105 { argInfo:POPT_ARG_INTL_DOMAIN,arg:"popt" },
106 { NULL ,'\0',POPT_ARG_CALLBACK,(void *)&displayArgs,'\0',NULL,NULL },
107 { "help" ,'?' ,0 ,NULL,'?',/* N_ */("Show this help message"), NULL },
108 { "usage",'\0',0 ,NULL,'u',/* N_ */("Display brief usage message"),NULL },
113 static void invoke_cmd_err(int cmd_argc,const char **cmd_argv,GError **errp)
115 const struct cmdline_command *commandp;
116 const char *cmd_name=NULL;
117 poptContext cmd_context;
119 const char **cmdarg_argv;
122 const char *emptyargv_NULL=NULL;
124 g_return_if_fail(cmd_argc>=0);
125 g_return_if_fail(!errp || !*errp);
127 /* poptGetContext() cannot be passed argc==0 even if we lass POPT_CONTEXT_KEEP_FIRST
128 * as it is buggy. Workaround it by keeping the command name as argv[0].
131 const char *stub_shell[]={ cmdline_command_table[0].name,NULL };
138 for (commandp=cmdline_command_table;commandp->name;commandp++) {
139 if (!cmd_name /* NULL cmd_name fallback to the first table entry - "shell" */
140 || !strcasecmp(cmd_name,commandp->name))
143 if (!commandp->name) {
144 g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_UNKNOWN_COMMAND,
145 _("Unknown command, try 'help': %s"),cmd_name);
148 displayArgs_hit=FALSE;
149 cmd_context=poptGetContext(
151 cmd_argc,cmd_argv, /* argc,argv */
152 commandp->table, /* options */
153 POPT_CONTEXT_POSIXMEHARDER); /* flags; !POPT_CONTEXT_KEEP_FIRST */
154 if (cmd_context==NULL) {
155 g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_INVALID_COMMAND_ARGUMENTS,
156 _("Invalid arguments for command: %s"),cmd_name);
159 errint=poptReadDefaultConfig(cmd_context,
162 g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_READING_COMMAND_CONFIG,
163 _("Error '%s' reading default configuration for command: %s"),poptStrerror(errint),cmd_name);
164 goto err_free_context;
166 errint=poptGetNextOpt(cmd_context);
168 g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_EXCEEDING_COMMAND_OPTION,
169 _("Exceeding command option for command: %s"),cmd_name);
170 goto err_free_context;
172 if (!(cmdarg_argv=poptGetArgs(cmd_context)))
173 cmdarg_argv=&emptyargv_NULL;
176 goto err_free_context;
178 for (csp=cmdarg_argv,cmdarg_argc=0;*csp;csp++)
180 if (cmdarg_argc<commandp->argsn_min || cmdarg_argc>commandp->argsn_max) {
181 g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_INVALID_COMMAND_ARGUMENT_COUNT,
182 _("Invalid number of command '%s' arguments: %d; expected from %d to %d incl."),
183 cmd_name,cmdarg_argc,commandp->argsn_min,commandp->argsn_max);
184 goto err_free_context;
187 (*commandp->func)(cmdarg_argv,errp);
190 poptFreeContext(cmd_context);
194 void err_cleanup(GError **errp)
196 g_return_if_fail(errp!=NULL);
200 printf("\nERROR: %s\n",(*errp)->message);
205 void invoke_cmd(int cmd_argc,const char **cmd_argv)
209 invoke_cmd_err(cmd_argc,cmd_argv,&gerr);
214 /* Returns: Success (no error occured). */
215 gboolean errvfsresult_to_gerr(GError **errp,GnomeVFSResult errvfsresult)
217 g_return_val_if_fail(!errp || !*errp,FALSE);
219 if (errvfsresult==GNOME_VFS_OK)
222 g_set_error(errp,CMDLINE_MAIN_ERROR,CMDLINE_MAIN_ERROR_GENERIC_ERROR,
223 _("Generic error: %s"),gnome_vfs_result_to_string(errvfsresult));
227 static GIOChannel *main_giochannel;
229 void main_exit(void) G_GNUC_NORETURN;
232 if (cmdline_captive_vfs_object) {
233 g_object_unref(cmdline_captive_vfs_object);
234 cmdline_captive_vfs_object=NULL;
236 if (main_giochannel) {
237 g_io_channel_unref(main_giochannel);
238 main_giochannel=NULL;
243 int main(int argc,char **argv)
247 const char **cmd_argv,**csp;
250 struct captive_options options;
252 /* Do not set g_log_set_always_fatal() here as we would not be able
253 * to restart failed children due to communication-failure alarms.
256 /* Prevent output block buffering if redirecting stdout to file. */
257 setvbuf(stdout,(char *)NULL,_IONBF,0);
258 setvbuf(stderr,(char *)NULL,_IONBF,0);
260 /* Initialize the i18n stuff */
261 setlocale(LC_ALL,"");
262 bindtextdomain(PACKAGE,LOCALEDIR);
265 /* Initialize GObject subsystem of GLib. */
268 captive_options_init(&options);
269 captive_options=&options; /* for parsing by 'CAPTIVE_POPT_INCLUDE' */
271 context=poptGetContext(
273 argc,(/*en-const*/const char **)argv, /* argc,argv */
274 popt_table, /* options */
275 POPT_CONTEXT_POSIXMEHARDER); /* flags; && !POPT_CONTEXT_KEEP_FIRST */
277 g_error(_("Error parsing command-line arguments"));
280 errint=poptReadDefaultConfig(context,
283 g_warning(_("Error reading default popt configuration"));
284 errint=poptGetNextOpt(context);
286 g_error(_("Error parsing (dash-prefixed) command-line argument"));
289 cmd_argv=poptGetArgs(context);
290 for (csp=cmd_argv,cmd_argc=0;csp && *csp;csp++)
293 captive_options=NULL; /* already parsed by 'CAPTIVE_POPT_INCLUDE' */
295 /* image_iochannel */
297 g_error(_("File/device disk image pathname command-line argument required"));
300 g_assert(options.image_iochannel==NULL);
301 if (!(options.image_iochannel=g_io_channel_new_file(
302 cmd_argv[0], /* filename */
303 (options.rwmode==CAPTIVE_OPTION_RWMODE_RW ? "r+" : "r"), /* mode */
304 NULL))) { /* error */
305 g_error(_("image_iochannel open failed"));
311 if (options.filesystem.type==CAPTIVE_OPTIONS_MODULE_TYPE_EMPTY) {
312 g_error(_("'--filesystem' option required ('ntfs.sys' pathname suggested)"));
315 if (!options.load_module) {
316 g_warning(_("'--load-module' option required ('ntoskrnl.exe' pathname suggested)"));
320 if (GNOME_VFS_OK!=captive_vfs_new(&cmdline_captive_vfs_object,&options)) {
321 g_error(_("captive_vfs_new() failed"));
324 captive_options_free(&options);
326 cmd_cd_internal("/",&gerr);
332 invoke_cmd(cmd_argc,cmd_argv);
334 /* 'cmd_argv' gets cleared by 'poptFreeContext(context);' below */
335 poptFreeContext(context);
337 main_exit(); /* unref 'cmdline_captive_vfs_object' */