| /* |
| * Copyright 2015 Red Hat, Inc. |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library 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 |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, see <http://www.gnu.org/licenses/>. |
| * |
| * Author: Matthias Clasen <mclasen@redhat.com> |
| */ |
| |
| #include "config.h" |
| |
| #include <gio/gio.h> |
| #include <gi18n.h> |
| |
| #ifdef G_OS_WIN32 |
| #include <io.h> |
| #endif |
| |
| #ifndef STDIN_FILENO |
| #define STDIN_FILENO 0 |
| #endif |
| |
| #ifdef HAVE_UNISTD_H |
| #include <unistd.h> |
| #endif |
| |
| #include "gio-tool.h" |
| |
| static char *etag = NULL; |
| static gboolean backup = FALSE; |
| static gboolean create = FALSE; |
| static gboolean append = FALSE; |
| static gboolean priv = FALSE; |
| static gboolean replace_dest = FALSE; |
| static gboolean print_etag = FALSE; |
| |
| static const GOptionEntry entries[] = |
| { |
| { "backup", 'b', 0, G_OPTION_ARG_NONE, &backup, N_("Backup existing destination files"), NULL }, |
| { "create", 'c', 0, G_OPTION_ARG_NONE, &create, N_("Only create if not existing"), NULL }, |
| { "append", 'a', 0, G_OPTION_ARG_NONE, &append, N_("Append to end of file"), NULL }, |
| { "private", 'p', 0, G_OPTION_ARG_NONE, &priv, N_("When creating, restrict access to the current user"), NULL }, |
| { "unlink", 'u', 0, G_OPTION_ARG_NONE, &replace_dest, N_("When replacing, replace as if the destination did not exist"), NULL }, |
| /* Translators: The "etag" is a token allowing to verify whether a file has been modified */ |
| { "print-etag", 'v', 0, G_OPTION_ARG_NONE, &print_etag, N_("Print new etag at end"), NULL }, |
| /* Translators: The "etag" is a token allowing to verify whether a file has been modified */ |
| { "etag", 'e', 0, G_OPTION_ARG_STRING, &etag, N_("The etag of the file being overwritten"), N_("ETAG") }, |
| { NULL } |
| }; |
| |
| /* 256k minus malloc overhead */ |
| #define STREAM_BUFFER_SIZE (1024*256 - 2*sizeof(gpointer)) |
| |
| static gboolean |
| save (GFile *file) |
| { |
| GOutputStream *out; |
| GFileCreateFlags flags; |
| char *buffer; |
| gssize res; |
| gboolean close_res; |
| GError *error; |
| gboolean save_res; |
| |
| error = NULL; |
| |
| flags = priv ? G_FILE_CREATE_PRIVATE : G_FILE_CREATE_NONE; |
| flags |= replace_dest ? G_FILE_CREATE_REPLACE_DESTINATION : 0; |
| |
| if (create) |
| out = (GOutputStream *)g_file_create (file, flags, NULL, &error); |
| else if (append) |
| out = (GOutputStream *)g_file_append_to (file, flags, NULL, &error); |
| else |
| out = (GOutputStream *)g_file_replace (file, etag, backup, flags, NULL, &error); |
| if (out == NULL) |
| { |
| print_file_error (file, error->message); |
| g_error_free (error); |
| return FALSE; |
| } |
| |
| buffer = g_malloc (STREAM_BUFFER_SIZE); |
| save_res = TRUE; |
| |
| while (1) |
| { |
| res = read (STDIN_FILENO, buffer, STREAM_BUFFER_SIZE); |
| if (res > 0) |
| { |
| g_output_stream_write_all (out, buffer, res, NULL, NULL, &error); |
| if (error != NULL) |
| { |
| save_res = FALSE; |
| print_file_error (file, error->message); |
| g_clear_error (&error); |
| goto out; |
| } |
| } |
| else if (res < 0) |
| { |
| save_res = FALSE; |
| print_error ("%s", _("Error reading from standard input")); |
| break; |
| } |
| else if (res == 0) |
| break; |
| } |
| |
| out: |
| |
| close_res = g_output_stream_close (out, NULL, &error); |
| if (!close_res) |
| { |
| save_res = FALSE; |
| print_file_error (file, error->message); |
| g_error_free (error); |
| } |
| |
| if (close_res && print_etag) |
| { |
| char *etag; |
| etag = g_file_output_stream_get_etag (G_FILE_OUTPUT_STREAM (out)); |
| |
| if (etag) |
| g_print ("Etag: %s\n", etag); |
| else |
| /* Translators: The "etag" is a token allowing to verify whether a file has been modified */ |
| g_print (_("Etag not available\n")); |
| g_free (etag); |
| } |
| |
| g_object_unref (out); |
| g_free (buffer); |
| |
| return save_res; |
| } |
| |
| int |
| handle_save (int argc, char *argv[], gboolean do_help) |
| { |
| GOptionContext *context; |
| GError *error = NULL; |
| GFile *file; |
| gboolean res; |
| |
| g_set_prgname ("gio save"); |
| |
| /* Translators: commandline placeholder */ |
| context = g_option_context_new (_("DESTINATION")); |
| g_option_context_set_help_enabled (context, FALSE); |
| g_option_context_set_summary (context, |
| _("Read from standard input and save to DEST.")); |
| g_option_context_add_main_entries (context, entries, GETTEXT_PACKAGE); |
| |
| if (do_help) |
| { |
| show_help (context, NULL); |
| g_option_context_free (context); |
| return 0; |
| } |
| |
| if (!g_option_context_parse (context, &argc, &argv, &error)) |
| { |
| show_help (context, error->message); |
| g_error_free (error); |
| g_option_context_free (context); |
| return 1; |
| } |
| |
| if (argc < 2) |
| { |
| show_help (context, _("No destination given")); |
| g_option_context_free (context); |
| return 1; |
| } |
| |
| if (argc > 2) |
| { |
| show_help (context, _("Too many arguments")); |
| g_option_context_free (context); |
| return 1; |
| } |
| |
| g_option_context_free (context); |
| |
| file = g_file_new_for_commandline_arg (argv[1]); |
| res = save (file); |
| g_object_unref (file); |
| |
| return res ? 0 : 2; |
| } |
| |