* 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, write to the
- * Free Software Foundation, Inc., 59 Temple Place, Suite 330,
- * Boston, MA 02111-1307, USA.
+ * Public License along with this library; if not, see <http://www.gnu.org/licenses/>.
*
* Author: Alexander Larsson <alexl@redhat.com>
*/
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
-#ifdef HAVE_UNISTD_H
-#include <unistd.h>
-#endif
#include <errno.h>
#include <string.h>
#include "gioerror.h"
#include "gcancellable.h"
#include "glocalfileoutputstream.h"
+#include "gfileinfo.h"
#include "glocalfileinfo.h"
+#ifdef G_OS_UNIX
+#include <unistd.h>
+#include "gfiledescriptorbased.h"
+#endif
+
#ifdef G_OS_WIN32
#include <io.h>
#ifndef S_ISDIR
#define O_BINARY 0
#endif
-#include "gioalias.h"
+struct _GLocalFileOutputStreamPrivate {
+ char *tmp_filename;
+ char *original_filename;
+ char *backup_filename;
+ char *etag;
+ guint sync_on_close : 1;
+ guint do_close : 1;
+ int fd;
+};
+
+#ifdef G_OS_UNIX
+static void g_file_descriptor_based_iface_init (GFileDescriptorBasedIface *iface);
+#endif
#define g_local_file_output_stream_get_type _g_local_file_output_stream_get_type
-G_DEFINE_TYPE (GLocalFileOutputStream, g_local_file_output_stream, G_TYPE_FILE_OUTPUT_STREAM);
+#ifdef G_OS_UNIX
+G_DEFINE_TYPE_WITH_CODE (GLocalFileOutputStream, g_local_file_output_stream, G_TYPE_FILE_OUTPUT_STREAM,
+ G_ADD_PRIVATE (GLocalFileOutputStream)
+ G_IMPLEMENT_INTERFACE (G_TYPE_FILE_DESCRIPTOR_BASED,
+ g_file_descriptor_based_iface_init))
+#else
+G_DEFINE_TYPE_WITH_CODE (GLocalFileOutputStream, g_local_file_output_stream, G_TYPE_FILE_OUTPUT_STREAM,
+ G_ADD_PRIVATE (GLocalFileOutputStream))
+#endif
+
/* Some of the file replacement code was based on the code from gedit,
* relicenced to LGPL with permissions from the authors.
#define BACKUP_EXTENSION "~"
-struct _GLocalFileOutputStreamPrivate {
- char *tmp_filename;
- char *original_filename;
- char *backup_filename;
- char *etag;
- gboolean sync_on_close;
- int fd;
-};
-
static gssize g_local_file_output_stream_write (GOutputStream *stream,
const void *buffer,
gsize count,
goffset size,
GCancellable *cancellable,
GError **error);
+#ifdef G_OS_UNIX
+static int g_local_file_output_stream_get_fd (GFileDescriptorBased *stream);
+#endif
static void
g_local_file_output_stream_finalize (GObject *object)
GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
GOutputStreamClass *stream_class = G_OUTPUT_STREAM_CLASS (klass);
GFileOutputStreamClass *file_stream_class = G_FILE_OUTPUT_STREAM_CLASS (klass);
-
- g_type_class_add_private (klass, sizeof (GLocalFileOutputStreamPrivate));
-
+
gobject_class->finalize = g_local_file_output_stream_finalize;
stream_class->write_fn = g_local_file_output_stream_write;
file_stream_class->truncate_fn = g_local_file_output_stream_truncate;
}
+#ifdef G_OS_UNIX
+static void
+g_file_descriptor_based_iface_init (GFileDescriptorBasedIface *iface)
+{
+ iface->get_fd = g_local_file_output_stream_get_fd;
+}
+#endif
+
static void
g_local_file_output_stream_init (GLocalFileOutputStream *stream)
{
- stream->priv = G_TYPE_INSTANCE_GET_PRIVATE (stream,
- G_TYPE_LOCAL_FILE_OUTPUT_STREAM,
- GLocalFileOutputStreamPrivate);
+ stream->priv = g_local_file_output_stream_get_instance_private (stream);
+ stream->priv->do_close = TRUE;
}
static gssize
return res;
}
-static gboolean
-g_local_file_output_stream_close (GOutputStream *stream,
- GCancellable *cancellable,
- GError **error)
+void
+_g_local_file_output_stream_set_do_close (GLocalFileOutputStream *out,
+ gboolean do_close)
{
- GLocalFileOutputStream *file;
- GLocalFileStat final_stat;
- int res;
+ out->priv->do_close = do_close;
+}
- file = G_LOCAL_FILE_OUTPUT_STREAM (stream);
+gboolean
+_g_local_file_output_stream_really_close (GLocalFileOutputStream *file,
+ GCancellable *cancellable,
+ GError **error)
+{
+ GLocalFileStat final_stat;
#ifdef HAVE_FSYNC
if (file->priv->sync_on_close &&
int errsv = errno;
g_set_error (error, G_IO_ERROR,
- g_io_error_from_errno (errno),
+ g_io_error_from_errno (errsv),
_("Error writing to file: %s"),
g_strerror (errsv));
goto err_out;
if (_fstati64 (file->priv->fd, &final_stat) == 0)
file->priv->etag = _g_local_file_info_create_etag (&final_stat);
- res = close (file->priv->fd);
- if (res == -1)
+ if (!g_close (file->priv->fd, NULL))
{
int errsv = errno;
if (g_cancellable_set_error_if_cancelled (cancellable, error))
goto err_out;
-#ifdef HAVE_LINK
+#ifdef G_OS_UNIX
/* create original -> backup link, the original is then renamed over */
if (g_unlink (file->priv->backup_filename) != 0 &&
errno != ENOENT)
int errsv = errno;
g_set_error (error, G_IO_ERROR,
- g_io_error_from_errno (errno),
+ g_io_error_from_errno (errsv),
_("Error renaming temporary file: %s"),
g_strerror (errsv));
goto err_out;
}
+
+ g_clear_pointer (&file->priv->tmp_filename, g_free);
}
if (g_cancellable_set_error_if_cancelled (cancellable, error))
if (fstat (file->priv->fd, &final_stat) == 0)
file->priv->etag = _g_local_file_info_create_etag (&final_stat);
- while (1)
+ if (!g_close (file->priv->fd, NULL))
{
- res = close (file->priv->fd);
- if (res == -1)
- {
- int errsv = errno;
-
- g_set_error (error, G_IO_ERROR,
- g_io_error_from_errno (errsv),
- _("Error closing file: %s"),
- g_strerror (errsv));
- }
- break;
+ int errsv = errno;
+
+ g_set_error (error, G_IO_ERROR,
+ g_io_error_from_errno (errsv),
+ _("Error closing file: %s"),
+ g_strerror (errsv));
+ goto err_out;
}
-
- return res != -1;
-
-#else
-
- return TRUE;
#endif
-
+
+ return TRUE;
err_out:
#ifndef G_OS_WIN32
/* A simple try to close the fd in case we fail before the actual close */
- close (file->priv->fd);
+ (void) g_close (file->priv->fd, NULL);
#endif
+ if (file->priv->tmp_filename)
+ g_unlink (file->priv->tmp_filename);
+
return FALSE;
}
+
+static gboolean
+g_local_file_output_stream_close (GOutputStream *stream,
+ GCancellable *cancellable,
+ GError **error)
+{
+ GLocalFileOutputStream *file;
+
+ file = G_LOCAL_FILE_OUTPUT_STREAM (stream);
+
+ if (file->priv->do_close)
+ return _g_local_file_output_stream_really_close (file,
+ cancellable,
+ error);
+ return TRUE;
+}
+
static char *
g_local_file_output_stream_get_etag (GFileOutputStream *stream)
{
}
GFileOutputStream *
-_g_local_file_output_stream_create (const char *filename,
- GFileCreateFlags flags,
- GCancellable *cancellable,
- GError **error)
+_g_local_file_output_stream_new (int fd)
{
GLocalFileOutputStream *stream;
- int mode;
- int fd;
- if (g_cancellable_set_error_if_cancelled (cancellable, error))
- return NULL;
+ stream = g_object_new (G_TYPE_LOCAL_FILE_OUTPUT_STREAM, NULL);
+ stream->priv->fd = fd;
+ return G_FILE_OUTPUT_STREAM (stream);
+}
- if (flags & G_FILE_CREATE_PRIVATE)
- mode = 0600;
- else
- mode = 0666;
+static void
+set_error_from_open_errno (const char *filename,
+ GError **error)
+{
+ int errsv = errno;
- fd = g_open (filename, O_CREAT | O_EXCL | O_WRONLY | O_BINARY, mode);
- if (fd == -1)
+ if (errsv == EINVAL)
+ /* This must be an invalid filename, on e.g. FAT */
+ g_set_error_literal (error, G_IO_ERROR,
+ G_IO_ERROR_INVALID_FILENAME,
+ _("Invalid filename"));
+ else
{
- int errsv = errno;
-
- if (errsv == EINVAL)
- /* This must be an invalid filename, on e.g. FAT */
- g_set_error_literal (error, G_IO_ERROR,
- G_IO_ERROR_INVALID_FILENAME,
- _("Invalid filename"));
- else
- {
- char *display_name = g_filename_display_name (filename);
- g_set_error (error, G_IO_ERROR,
- g_io_error_from_errno (errsv),
- _("Error opening file '%s': %s"),
+ char *display_name = g_filename_display_name (filename);
+ g_set_error (error, G_IO_ERROR,
+ g_io_error_from_errno (errsv),
+ _("Error opening file '%s': %s"),
display_name, g_strerror (errsv));
- g_free (display_name);
- }
+ g_free (display_name);
+ }
+}
+
+static GFileOutputStream *
+output_stream_open (const char *filename,
+ gint open_flags,
+ guint mode,
+ GCancellable *cancellable,
+ GError **error)
+{
+ GLocalFileOutputStream *stream;
+ gint fd;
+
+ fd = g_open (filename, open_flags, mode);
+ if (fd == -1)
+ {
+ set_error_from_open_errno (filename, error);
return NULL;
}
}
GFileOutputStream *
+_g_local_file_output_stream_open (const char *filename,
+ gboolean readable,
+ GCancellable *cancellable,
+ GError **error)
+{
+ int open_flags;
+
+ if (g_cancellable_set_error_if_cancelled (cancellable, error))
+ return NULL;
+
+ open_flags = O_BINARY;
+ if (readable)
+ open_flags |= O_RDWR;
+ else
+ open_flags |= O_WRONLY;
+
+ return output_stream_open (filename, open_flags, 0666, cancellable, error);
+}
+
+static gint
+mode_from_flags_or_info (GFileCreateFlags flags,
+ GFileInfo *reference_info)
+{
+ if (flags & G_FILE_CREATE_PRIVATE)
+ return 0600;
+ else if (reference_info && g_file_info_has_attribute (reference_info, "unix::mode"))
+ return g_file_info_get_attribute_uint32 (reference_info, "unix::mode") & (~S_IFMT);
+ else
+ return 0666;
+}
+
+GFileOutputStream *
+_g_local_file_output_stream_create (const char *filename,
+ gboolean readable,
+ GFileCreateFlags flags,
+ GFileInfo *reference_info,
+ GCancellable *cancellable,
+ GError **error)
+{
+ int mode;
+ int open_flags;
+
+ if (g_cancellable_set_error_if_cancelled (cancellable, error))
+ return NULL;
+
+ mode = mode_from_flags_or_info (flags, reference_info);
+
+ open_flags = O_CREAT | O_EXCL | O_BINARY;
+ if (readable)
+ open_flags |= O_RDWR;
+ else
+ open_flags |= O_WRONLY;
+
+ return output_stream_open (filename, open_flags, mode, cancellable, error);
+}
+
+GFileOutputStream *
_g_local_file_output_stream_append (const char *filename,
GFileCreateFlags flags,
GCancellable *cancellable,
GError **error)
{
- GLocalFileOutputStream *stream;
int mode;
- int fd;
if (g_cancellable_set_error_if_cancelled (cancellable, error))
return NULL;
else
mode = 0666;
- fd = g_open (filename, O_CREAT | O_APPEND | O_WRONLY | O_BINARY, mode);
- if (fd == -1)
- {
- int errsv = errno;
-
- if (errsv == EINVAL)
- /* This must be an invalid filename, on e.g. FAT */
- g_set_error_literal (error, G_IO_ERROR,
- G_IO_ERROR_INVALID_FILENAME,
- _("Invalid filename"));
- else
- {
- char *display_name = g_filename_display_name (filename);
- g_set_error (error, G_IO_ERROR,
- g_io_error_from_errno (errsv),
- _("Error opening file '%s': %s"),
- display_name, g_strerror (errsv));
- g_free (display_name);
- }
- return NULL;
- }
-
- stream = g_object_new (G_TYPE_LOCAL_FILE_OUTPUT_STREAM, NULL);
- stream->priv->fd = fd;
-
- return G_FILE_OUTPUT_STREAM (stream);
+ return output_stream_open (filename, O_CREAT | O_APPEND | O_WRONLY | O_BINARY, mode,
+ cancellable, error);
}
static char *
static int
handle_overwrite_open (const char *filename,
+ gboolean readable,
const char *etag,
gboolean create_backup,
char **temp_filename,
GFileCreateFlags flags,
+ GFileInfo *reference_info,
GCancellable *cancellable,
GError **error)
{
int res;
int mode;
- if (flags & G_FILE_CREATE_PRIVATE)
- mode = 0600;
- else
- mode = 0666;
+ mode = mode_from_flags_or_info (flags, reference_info);
/* We only need read access to the original file if we are creating a backup.
* We also add O_CREATE to avoid a race if the file was just removed */
- if (create_backup)
+ if (create_backup || readable)
open_flags = O_RDWR | O_CREAT | O_BINARY;
else
open_flags = O_WRONLY | O_CREAT | O_BINARY;
char *display_name = g_filename_display_name (filename);
g_set_error (error, G_IO_ERROR,
g_io_error_from_errno (errsv),
- _("Error stating file '%s': %s"),
+ _("Error when getting information for file '%s': %s"),
display_name, g_strerror (errsv));
g_free (display_name);
goto err_out;
tmp_filename = g_build_filename (dirname, ".goutputstream-XXXXXX", NULL);
g_free (dirname);
- tmpfd = g_mkstemp (tmp_filename);
+ tmpfd = g_mkstemp_full (tmp_filename, (readable ? O_RDWR : O_WRONLY) | O_BINARY, mode);
if (tmpfd == -1)
{
g_free (tmp_filename);
)
)
{
- struct stat tmp_statbuf;
-
+ GLocalFileStat tmp_statbuf;
+ int tres;
+
+#ifdef G_OS_WIN32
+ tres = _fstati64 (tmpfd, &tmp_statbuf);
+#else
+ tres = fstat (tmpfd, &tmp_statbuf);
+#endif
/* Check that we really needed to change something */
- if (fstat (tmpfd, &tmp_statbuf) != 0 ||
+ if (tres != 0 ||
original_stat.st_uid != tmp_statbuf.st_uid ||
original_stat.st_gid != tmp_statbuf.st_gid ||
original_stat.st_mode != tmp_statbuf.st_mode)
{
- close (tmpfd);
+ (void) g_close (tmpfd, NULL);
g_unlink (tmp_filename);
g_free (tmp_filename);
goto fallback_strategy;
}
}
- close (fd);
+ (void) g_close (fd, NULL);
*temp_filename = tmp_filename;
return tmpfd;
}
G_IO_ERROR_CANT_CREATE_BACKUP,
_("Backup file creation failed"));
g_unlink (backup_filename);
- close (bfd);
+ (void) g_close (bfd, NULL);
g_free (backup_filename);
goto err_out;
}
G_IO_ERROR_CANT_CREATE_BACKUP,
_("Backup file creation failed"));
g_unlink (backup_filename);
- close (bfd);
+ (void) g_close (bfd, NULL);
g_free (backup_filename);
goto err_out;
}
- close (bfd);
+ (void) g_close (bfd, NULL);
g_free (backup_filename);
/* Seek back to the start of the file after the backup copy */
if (flags & G_FILE_CREATE_REPLACE_DESTINATION)
{
- close (fd);
+ (void) g_close (fd, NULL);
if (g_unlink (filename) != 0)
{
g_strerror (errsv));
goto err_out2;
}
-
- fd = g_open (filename, O_WRONLY | O_CREAT | O_BINARY, mode);
+
+ if (readable)
+ open_flags = O_RDWR | O_CREAT | O_BINARY;
+ else
+ open_flags = O_WRONLY | O_CREAT | O_BINARY;
+ fd = g_open (filename, open_flags, mode);
if (fd == -1)
{
int errsv = errno;
return fd;
err_out:
- close (fd);
+ (void) g_close (fd, NULL);
err_out2:
return -1;
}
GFileOutputStream *
_g_local_file_output_stream_replace (const char *filename,
+ gboolean readable,
const char *etag,
gboolean create_backup,
GFileCreateFlags flags,
+ GFileInfo *reference_info,
GCancellable *cancellable,
GError **error)
{
int fd;
char *temp_file;
gboolean sync_on_close;
+ int open_flags;
if (g_cancellable_set_error_if_cancelled (cancellable, error))
return NULL;
temp_file = NULL;
- if (flags & G_FILE_CREATE_PRIVATE)
- mode = 0600;
- else
- mode = 0666;
+ mode = mode_from_flags_or_info (flags, reference_info);
sync_on_close = FALSE;
/* If the file doesn't exist, create it */
- fd = g_open (filename, O_CREAT | O_EXCL | O_WRONLY | O_BINARY, mode);
+ open_flags = O_CREAT | O_EXCL | O_BINARY;
+ if (readable)
+ open_flags |= O_RDWR;
+ else
+ open_flags |= O_WRONLY;
+ fd = g_open (filename, open_flags, mode);
if (fd == -1 && errno == EEXIST)
{
/* The file already exists */
- fd = handle_overwrite_open (filename, etag, create_backup, &temp_file,
- flags, cancellable, error);
+ fd = handle_overwrite_open (filename, readable, etag,
+ create_backup, &temp_file,
+ flags, reference_info,
+ cancellable, error);
if (fd == -1)
return NULL;
}
else if (fd == -1)
{
- int errsv = errno;
-
- if (errsv == EINVAL)
- /* This must be an invalid filename, on e.g. FAT */
- g_set_error_literal (error, G_IO_ERROR,
- G_IO_ERROR_INVALID_FILENAME,
- _("Invalid filename"));
- else
- {
- char *display_name = g_filename_display_name (filename);
- g_set_error (error, G_IO_ERROR,
- g_io_error_from_errno (errsv),
- _("Error opening file '%s': %s"),
- display_name, g_strerror (errsv));
- g_free (display_name);
- }
+ set_error_from_open_errno (filename, error);
return NULL;
}
return G_FILE_OUTPUT_STREAM (stream);
}
+
+gint
+_g_local_file_output_stream_get_fd (GLocalFileOutputStream *stream)
+{
+ g_return_val_if_fail (G_IS_LOCAL_FILE_OUTPUT_STREAM (stream), -1);
+ return stream->priv->fd;
+}
+
+#ifdef G_OS_UNIX
+static int
+g_local_file_output_stream_get_fd (GFileDescriptorBased *fd_based)
+{
+ GLocalFileOutputStream *stream = G_LOCAL_FILE_OUTPUT_STREAM (fd_based);
+ return _g_local_file_output_stream_get_fd (stream);
+}
+#endif