/* * 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 . * * Author: Matthias Clasen */ #include "config.h" #include #include #ifdef G_OS_WIN32 #include #endif #ifndef STDIN_FILENO #define STDIN_FILENO 0 #endif #ifdef HAVE_UNISTD_H #include #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; }