1 /*
2 * Copyright 2015 Red Hat, Inc.
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * This library 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 GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, see <http://www.gnu.org/licenses/>.
16 *
17 * Author: Matthias Clasen <mclasen@redhat.com>
18 */
19
20 #include "config.h"
21
22 #include <gio/gio.h>
23 #include <gi18n.h>
24
25 #ifdef G_OS_WIN32
26 #include <io.h>
27 #endif
28
29 #ifndef STDIN_FILENO
30 #define STDIN_FILENO 0
31 #endif
32
33 #ifdef HAVE_UNISTD_H
34 #include <unistd.h>
35 #endif
36
37 #include "gio-tool.h"
38
39 static char *etag = NULL;
40 static gboolean backup = FALSE;
41 static gboolean create = FALSE;
42 static gboolean append = FALSE;
43 static gboolean priv = FALSE;
44 static gboolean replace_dest = FALSE;
45 static gboolean print_etag = FALSE;
46
47 static const GOptionEntry entries[] =
48 {
49 { "backup", 'b', 0, G_OPTION_ARG_NONE, &backup, N_("Backup existing destination files"), NULL },
50 { "create", 'c', 0, G_OPTION_ARG_NONE, &create, N_("Only create if not existing"), NULL },
51 { "append", 'a', 0, G_OPTION_ARG_NONE, &append, N_("Append to end of file"), NULL },
52 { "private", 'p', 0, G_OPTION_ARG_NONE, &priv, N_("When creating, restrict access to the current user"), NULL },
53 { "unlink", 'u', 0, G_OPTION_ARG_NONE, &replace_dest, N_("When replacing, replace as if the destination did not exist"), NULL },
54 /* Translators: The "etag" is a token allowing to verify whether a file has been modified */
55 { "print-etag", 'v', 0, G_OPTION_ARG_NONE, &print_etag, N_("Print new etag at end"), NULL },
56 /* Translators: The "etag" is a token allowing to verify whether a file has been modified */
57 { "etag", 'e', 0, G_OPTION_ARG_STRING, &etag, N_("The etag of the file being overwritten"), N_("ETAG") },
58 { NULL }
59 };
60
61 /* 256k minus malloc overhead */
62 #define STREAM_BUFFER_SIZE (1024*256 - 2*sizeof(gpointer))
63
64 static gboolean
save(GFile * file)65 save (GFile *file)
66 {
67 GOutputStream *out;
68 GFileCreateFlags flags;
69 char *buffer;
70 gssize res;
71 gboolean close_res;
72 GError *error;
73 gboolean save_res;
74
75 error = NULL;
76
77 flags = priv ? G_FILE_CREATE_PRIVATE : G_FILE_CREATE_NONE;
78 flags |= replace_dest ? G_FILE_CREATE_REPLACE_DESTINATION : 0;
79
80 if (create)
81 out = (GOutputStream *)g_file_create (file, flags, NULL, &error);
82 else if (append)
83 out = (GOutputStream *)g_file_append_to (file, flags, NULL, &error);
84 else
85 out = (GOutputStream *)g_file_replace (file, etag, backup, flags, NULL, &error);
86 if (out == NULL)
87 {
88 print_file_error (file, error->message);
89 g_error_free (error);
90 return FALSE;
91 }
92
93 buffer = g_malloc (STREAM_BUFFER_SIZE);
94 save_res = TRUE;
95
96 while (1)
97 {
98 res = read (STDIN_FILENO, buffer, STREAM_BUFFER_SIZE);
99 if (res > 0)
100 {
101 g_output_stream_write_all (out, buffer, res, NULL, NULL, &error);
102 if (error != NULL)
103 {
104 save_res = FALSE;
105 print_file_error (file, error->message);
106 g_clear_error (&error);
107 goto out;
108 }
109 }
110 else if (res < 0)
111 {
112 save_res = FALSE;
113 print_error ("%s", _("Error reading from standard input"));
114 break;
115 }
116 else if (res == 0)
117 break;
118 }
119
120 out:
121
122 close_res = g_output_stream_close (out, NULL, &error);
123 if (!close_res)
124 {
125 save_res = FALSE;
126 print_file_error (file, error->message);
127 g_error_free (error);
128 }
129
130 if (close_res && print_etag)
131 {
132 char *etag;
133 etag = g_file_output_stream_get_etag (G_FILE_OUTPUT_STREAM (out));
134
135 if (etag)
136 g_print ("Etag: %s\n", etag);
137 else
138 /* Translators: The "etag" is a token allowing to verify whether a file has been modified */
139 g_print (_("Etag not available\n"));
140 g_free (etag);
141 }
142
143 g_object_unref (out);
144 g_free (buffer);
145
146 return save_res;
147 }
148
149 int
handle_save(int argc,char * argv[],gboolean do_help)150 handle_save (int argc, char *argv[], gboolean do_help)
151 {
152 GOptionContext *context;
153 GError *error = NULL;
154 GFile *file;
155 gboolean res;
156
157 g_set_prgname ("gio save");
158
159 /* Translators: commandline placeholder */
160 context = g_option_context_new (_("DESTINATION"));
161 g_option_context_set_help_enabled (context, FALSE);
162 g_option_context_set_summary (context,
163 _("Read from standard input and save to DEST."));
164 g_option_context_add_main_entries (context, entries, GETTEXT_PACKAGE);
165
166 if (do_help)
167 {
168 show_help (context, NULL);
169 g_option_context_free (context);
170 return 0;
171 }
172
173 if (!g_option_context_parse (context, &argc, &argv, &error))
174 {
175 show_help (context, error->message);
176 g_error_free (error);
177 g_option_context_free (context);
178 return 1;
179 }
180
181 if (argc < 2)
182 {
183 show_help (context, _("No destination given"));
184 g_option_context_free (context);
185 return 1;
186 }
187
188 if (argc > 2)
189 {
190 show_help (context, _("Too many arguments"));
191 g_option_context_free (context);
192 return 1;
193 }
194
195 g_option_context_free (context);
196
197 file = g_file_new_for_commandline_arg (argv[1]);
198 res = save (file);
199 g_object_unref (file);
200
201 return res ? 0 : 2;
202 }
203