2 * client cmdline interface command "get" for libcaptive
3 * Copyright (C) 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>
23 #include <glib/gerror.h>
25 #include <captive/client-file.h>
29 #include <captive/macros.h>
32 #include "cmd_get.h" /* self */
33 #include "cmd_cd.h" /* for cmdline_path_from_cwd() */
38 #define DEFAULT_TRANSFER_BUFFER_SIZE (0x10000)
41 GQuark cmdline_cmd_get_error_quark(void)
46 r=g_quark_from_static_string("cmdline-cmd-get");
52 static gchar *optarg_transfer_buffer;
54 const struct poptOption cmd_get_table[]={
55 CMDLINE_POPT("buffer-size",'b',POPT_ARG_STRING,&optarg_transfer_buffer,
56 N_("Transfer buffer size"),N_("size")),
62 void cmd_get(const char **cmd_argv,GError **errp)
64 CaptiveFileObject *captive_file_object;
65 const gchar *sourcefile,*targetfile;
68 guint8 *transfer_buffer;
69 long transfer_buffer_size=DEFAULT_TRANSFER_BUFFER_SIZE;
71 g_return_if_fail(!errp || !*errp);
73 if (optarg_transfer_buffer) {
74 const gchar *string=captive_strdup_alloca(optarg_transfer_buffer);
77 free(optarg_transfer_buffer);
78 optarg_transfer_buffer=NULL;
80 transfer_buffer_size=strtol(string,&endptr,0);
81 if (transfer_buffer_size<=0 || transfer_buffer_size>=LONG_MAX || (endptr && *endptr)) {
82 g_set_error(errp,CMDLINE_CMD_GET_ERROR,CMDLINE_CMD_GET_ERROR_PARSING_TRANSFER_BUFFER_SIZE,
83 _("Error parsing transfer buffer size: %s"),string);
88 sourcefile=cmdline_path_from_cwd(cmd_argv[0]);
90 targetfile=captive_strdup_alloca(cmdline_path_from_cwd(cmd_argv[1]));
94 targetfile=captive_strdup_alloca(sourcefile);
95 if ((s=strrchr(targetfile,G_DIR_SEPARATOR)))
99 if (-1==(fdtgt=open(targetfile,
100 O_CREAT|O_EXCL|O_WRONLY /* flags */
103 #endif /* O_BINARY */
106 #endif /* O_LARGEFILE */
109 g_set_error(errp,CMDLINE_CMD_GET_ERROR,CMDLINE_CMD_GET_ERROR_CANNOT_CREATE_TARGET_HOSTOS_FILE,
110 _("Cannot create target host-os file '%s': %s"),targetfile,g_strerror(errno));
114 if (!errvfsresult_to_gerr(errp,captive_file_new_open(
115 &captive_file_object, /* captive_file_object_return */
116 cmdline_captive_vfs_object, /* captive_vfs_object */
117 sourcefile, /* pathname */
118 GNOME_VFS_OPEN_READ))) { /* mode */
120 g_set_error(errp,CMDLINE_CMD_GET_ERROR,CMDLINE_CMD_GET_ERROR_OPENING_SOURCE_FILE,
121 _("Error opening source guest-os file: %s"),sourcefile);
122 goto err_close_fdtgt;
125 transfer_buffer=g_malloc(transfer_buffer_size);
126 g_assert(transfer_buffer!=NULL); /* Should not happen. */
129 GnomeVFSFileSize bytes_read;
130 GnomeVFSResult errvfsresult;
133 errvfsresult=captive_file_read(
134 captive_file_object, /* captive_file_object */
135 transfer_buffer, /* buffer */
136 transfer_buffer_size, /* num_bytes */
137 &bytes_read); /* bytes_read_return */
138 g_assert(errvfsresult==GNOME_VFS_OK || errvfsresult==GNOME_VFS_ERROR_EOF);
139 g_assert((errvfsresult==GNOME_VFS_ERROR_EOF)==(bytes_read==0));
140 if (errvfsresult==GNOME_VFS_ERROR_EOF)
142 if (errvfsresult!=GNOME_VFS_OK) {
145 errbool=errvfsresult_to_gerr(errp,errvfsresult);
146 g_assert(errbool==FALSE);
148 g_set_error(errp,CMDLINE_CMD_GET_ERROR,CMDLINE_CMD_GET_ERROR_READING_SOURCE_FILE,
149 _("Error reading source guest-os file '%s': %s"),sourcefile,g_strerror(errno));
150 goto err_free_transfer_buffer;
153 if (bytes_read!=(GnomeVFSFileSize)(gotssize=write(fdtgt,transfer_buffer,bytes_read))) {
154 g_set_error(errp,CMDLINE_CMD_GET_ERROR,CMDLINE_CMD_GET_ERROR_WRITING_TARGET_HOSTOS_FILE,
155 _("Error writing target host-os file: %s"),targetfile);
156 goto err_free_transfer_buffer;
160 err_free_transfer_buffer:
161 g_free(transfer_buffer);
162 /* err_unref_captive_file_object: */
163 g_object_unref(captive_file_object);
166 err_cleanup(errp); /* may be clean */
167 g_set_error(errp,CMDLINE_CMD_GET_ERROR,CMDLINE_CMD_GET_ERROR_CLOSING_TARGET_HOSTOS_FILE,
168 _("Error closing target host-os file '%s': %s"),targetfile,g_strerror(errno));