Blame gio/glocalfileoutputstream.c

Packit ae235b
/* GIO - GLib Input, Output and Streaming Library
Packit ae235b
 * 
Packit ae235b
 * Copyright (C) 2006-2007 Red Hat, Inc.
Packit ae235b
 *
Packit ae235b
 * This library is free software; you can redistribute it and/or
Packit ae235b
 * modify it under the terms of the GNU Lesser General Public
Packit ae235b
 * License as published by the Free Software Foundation; either
Packit ae235b
 * version 2.1 of the License, or (at your option) any later version.
Packit ae235b
 *
Packit ae235b
 * This library is distributed in the hope that it will be useful,
Packit ae235b
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
Packit ae235b
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
Packit ae235b
 * Lesser General Public License for more details.
Packit ae235b
 *
Packit ae235b
 * You should have received a copy of the GNU Lesser General
Packit ae235b
 * Public License along with this library; if not, see <http://www.gnu.org/licenses/>.
Packit ae235b
 *
Packit ae235b
 * Author: Alexander Larsson <alexl@redhat.com>
Packit ae235b
 */
Packit ae235b
Packit ae235b
#include "config.h"
Packit ae235b
Packit ae235b
#include <sys/types.h>
Packit ae235b
#include <sys/stat.h>
Packit ae235b
#include <fcntl.h>
Packit ae235b
#include <errno.h>
Packit ae235b
#include <string.h>
Packit ae235b
Packit ae235b
#include <glib.h>
Packit ae235b
#include <glib/gstdio.h>
Packit ae235b
#include "glibintl.h"
Packit ae235b
#include "gioerror.h"
Packit ae235b
#include "gcancellable.h"
Packit ae235b
#include "glocalfileoutputstream.h"
Packit ae235b
#include "gfileinfo.h"
Packit ae235b
#include "glocalfileinfo.h"
Packit ae235b
Packit ae235b
#ifdef G_OS_UNIX
Packit ae235b
#include <unistd.h>
Packit ae235b
#include "gfiledescriptorbased.h"
Packit ae235b
#endif
Packit ae235b
Packit ae235b
#include "glib-private.h"
Packit ae235b
Packit ae235b
#ifdef G_OS_WIN32
Packit ae235b
#include <io.h>
Packit ae235b
#ifndef S_ISDIR
Packit ae235b
#define S_ISDIR(m) (((m) & _S_IFMT) == _S_IFDIR)
Packit ae235b
#endif
Packit ae235b
#ifndef S_ISREG
Packit ae235b
#define S_ISREG(m) (((m) & _S_IFMT) == _S_IFREG)
Packit ae235b
#endif
Packit ae235b
#endif
Packit ae235b
Packit ae235b
#ifndef O_BINARY
Packit ae235b
#define O_BINARY 0
Packit ae235b
#endif
Packit ae235b
Packit ae235b
struct _GLocalFileOutputStreamPrivate {
Packit ae235b
  char *tmp_filename;
Packit ae235b
  char *original_filename;
Packit ae235b
  char *backup_filename;
Packit ae235b
  char *etag;
Packit ae235b
  guint sync_on_close : 1;
Packit ae235b
  guint do_close : 1;
Packit ae235b
  int fd;
Packit ae235b
};
Packit ae235b
Packit ae235b
#ifdef G_OS_UNIX
Packit ae235b
static void       g_file_descriptor_based_iface_init   (GFileDescriptorBasedIface *iface);
Packit ae235b
#endif
Packit ae235b
Packit ae235b
#define g_local_file_output_stream_get_type _g_local_file_output_stream_get_type
Packit ae235b
#ifdef G_OS_UNIX
Packit ae235b
G_DEFINE_TYPE_WITH_CODE (GLocalFileOutputStream, g_local_file_output_stream, G_TYPE_FILE_OUTPUT_STREAM,
Packit ae235b
                         G_ADD_PRIVATE (GLocalFileOutputStream)
Packit ae235b
			 G_IMPLEMENT_INTERFACE (G_TYPE_FILE_DESCRIPTOR_BASED,
Packit ae235b
						g_file_descriptor_based_iface_init))
Packit ae235b
#else
Packit ae235b
G_DEFINE_TYPE_WITH_CODE (GLocalFileOutputStream, g_local_file_output_stream, G_TYPE_FILE_OUTPUT_STREAM,
Packit ae235b
                         G_ADD_PRIVATE (GLocalFileOutputStream))
Packit ae235b
#endif
Packit ae235b
Packit ae235b
Packit ae235b
/* Some of the file replacement code was based on the code from gedit,
Packit ae235b
 * relicenced to LGPL with permissions from the authors.
Packit ae235b
 */
Packit ae235b
Packit ae235b
#define BACKUP_EXTENSION "~"
Packit ae235b
Packit ae235b
static gssize     g_local_file_output_stream_write        (GOutputStream      *stream,
Packit ae235b
							   const void         *buffer,
Packit ae235b
							   gsize               count,
Packit ae235b
							   GCancellable       *cancellable,
Packit ae235b
							   GError            **error);
Packit ae235b
static gboolean   g_local_file_output_stream_close        (GOutputStream      *stream,
Packit ae235b
							   GCancellable       *cancellable,
Packit ae235b
							   GError            **error);
Packit ae235b
static GFileInfo *g_local_file_output_stream_query_info   (GFileOutputStream  *stream,
Packit ae235b
							   const char         *attributes,
Packit ae235b
							   GCancellable       *cancellable,
Packit ae235b
							   GError            **error);
Packit ae235b
static char *     g_local_file_output_stream_get_etag     (GFileOutputStream  *stream);
Packit ae235b
static goffset    g_local_file_output_stream_tell         (GFileOutputStream  *stream);
Packit ae235b
static gboolean   g_local_file_output_stream_can_seek     (GFileOutputStream  *stream);
Packit ae235b
static gboolean   g_local_file_output_stream_seek         (GFileOutputStream  *stream,
Packit ae235b
							   goffset             offset,
Packit ae235b
							   GSeekType           type,
Packit ae235b
							   GCancellable       *cancellable,
Packit ae235b
							   GError            **error);
Packit ae235b
static gboolean   g_local_file_output_stream_can_truncate (GFileOutputStream  *stream);
Packit ae235b
static gboolean   g_local_file_output_stream_truncate     (GFileOutputStream  *stream,
Packit ae235b
							   goffset             size,
Packit ae235b
							   GCancellable       *cancellable,
Packit ae235b
							   GError            **error);
Packit ae235b
#ifdef G_OS_UNIX
Packit ae235b
static int        g_local_file_output_stream_get_fd       (GFileDescriptorBased *stream);
Packit ae235b
#endif
Packit ae235b
Packit ae235b
static void
Packit ae235b
g_local_file_output_stream_finalize (GObject *object)
Packit ae235b
{
Packit ae235b
  GLocalFileOutputStream *file;
Packit ae235b
  
Packit ae235b
  file = G_LOCAL_FILE_OUTPUT_STREAM (object);
Packit ae235b
  
Packit ae235b
  g_free (file->priv->tmp_filename);
Packit ae235b
  g_free (file->priv->original_filename);
Packit ae235b
  g_free (file->priv->backup_filename);
Packit ae235b
  g_free (file->priv->etag);
Packit ae235b
Packit ae235b
  G_OBJECT_CLASS (g_local_file_output_stream_parent_class)->finalize (object);
Packit ae235b
}
Packit ae235b
Packit ae235b
static void
Packit ae235b
g_local_file_output_stream_class_init (GLocalFileOutputStreamClass *klass)
Packit ae235b
{
Packit ae235b
  GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
Packit ae235b
  GOutputStreamClass *stream_class = G_OUTPUT_STREAM_CLASS (klass);
Packit ae235b
  GFileOutputStreamClass *file_stream_class = G_FILE_OUTPUT_STREAM_CLASS (klass);
Packit ae235b
Packit ae235b
  gobject_class->finalize = g_local_file_output_stream_finalize;
Packit ae235b
Packit ae235b
  stream_class->write_fn = g_local_file_output_stream_write;
Packit ae235b
  stream_class->close_fn = g_local_file_output_stream_close;
Packit ae235b
  file_stream_class->query_info = g_local_file_output_stream_query_info;
Packit ae235b
  file_stream_class->get_etag = g_local_file_output_stream_get_etag;
Packit ae235b
  file_stream_class->tell = g_local_file_output_stream_tell;
Packit ae235b
  file_stream_class->can_seek = g_local_file_output_stream_can_seek;
Packit ae235b
  file_stream_class->seek = g_local_file_output_stream_seek;
Packit ae235b
  file_stream_class->can_truncate = g_local_file_output_stream_can_truncate;
Packit ae235b
  file_stream_class->truncate_fn = g_local_file_output_stream_truncate;
Packit ae235b
}
Packit ae235b
Packit ae235b
#ifdef G_OS_UNIX
Packit ae235b
static void
Packit ae235b
g_file_descriptor_based_iface_init (GFileDescriptorBasedIface *iface)
Packit ae235b
{
Packit ae235b
  iface->get_fd = g_local_file_output_stream_get_fd;
Packit ae235b
}
Packit ae235b
#endif
Packit ae235b
Packit ae235b
static void
Packit ae235b
g_local_file_output_stream_init (GLocalFileOutputStream *stream)
Packit ae235b
{
Packit ae235b
  stream->priv = g_local_file_output_stream_get_instance_private (stream);
Packit ae235b
  stream->priv->do_close = TRUE;
Packit ae235b
}
Packit ae235b
Packit ae235b
static gssize
Packit ae235b
g_local_file_output_stream_write (GOutputStream  *stream,
Packit ae235b
				  const void     *buffer,
Packit ae235b
				  gsize           count,
Packit ae235b
				  GCancellable   *cancellable,
Packit ae235b
				  GError        **error)
Packit ae235b
{
Packit ae235b
  GLocalFileOutputStream *file;
Packit ae235b
  gssize res;
Packit ae235b
Packit ae235b
  file = G_LOCAL_FILE_OUTPUT_STREAM (stream);
Packit ae235b
Packit ae235b
  while (1)
Packit ae235b
    {
Packit ae235b
      if (g_cancellable_set_error_if_cancelled (cancellable, error))
Packit ae235b
	return -1;
Packit ae235b
      res = write (file->priv->fd, buffer, count);
Packit ae235b
      if (res == -1)
Packit ae235b
	{
Packit ae235b
          int errsv = errno;
Packit ae235b
Packit ae235b
	  if (errsv == EINTR)
Packit ae235b
	    continue;
Packit ae235b
	  
Packit ae235b
	  g_set_error (error, G_IO_ERROR,
Packit ae235b
		       g_io_error_from_errno (errsv),
Packit ae235b
		       _("Error writing to file: %s"),
Packit ae235b
		       g_strerror (errsv));
Packit ae235b
	}
Packit ae235b
      
Packit ae235b
      break;
Packit ae235b
    }
Packit ae235b
  
Packit ae235b
  return res;
Packit ae235b
}
Packit ae235b
Packit ae235b
void
Packit ae235b
_g_local_file_output_stream_set_do_close (GLocalFileOutputStream *out,
Packit ae235b
					  gboolean do_close)
Packit ae235b
{
Packit ae235b
  out->priv->do_close = do_close;
Packit ae235b
}
Packit ae235b
Packit ae235b
gboolean
Packit ae235b
_g_local_file_output_stream_really_close (GLocalFileOutputStream *file,
Packit ae235b
					  GCancellable   *cancellable,
Packit ae235b
					  GError        **error)
Packit ae235b
{
Packit ae235b
  GLocalFileStat final_stat;
Packit ae235b
Packit ae235b
#ifdef HAVE_FSYNC
Packit ae235b
  if (file->priv->sync_on_close &&
Packit ae235b
      fsync (file->priv->fd) != 0)
Packit ae235b
    {
Packit ae235b
      int errsv = errno;
Packit ae235b
      
Packit ae235b
      g_set_error (error, G_IO_ERROR,
Packit ae235b
		   g_io_error_from_errno (errsv),
Packit ae235b
		   _("Error writing to file: %s"),
Packit ae235b
		   g_strerror (errsv));
Packit ae235b
      goto err_out;
Packit ae235b
    }
Packit ae235b
#endif
Packit ae235b
 
Packit ae235b
#ifdef G_OS_WIN32
Packit ae235b
Packit ae235b
  /* Must close before renaming on Windows, so just do the close first
Packit ae235b
   * in all cases for now.
Packit ae235b
   */
Packit ae235b
  if (GLIB_PRIVATE_CALL (g_win32_fstat) (file->priv->fd, &final_stat) == 0)
Packit ae235b
    file->priv->etag = _g_local_file_info_create_etag (&final_stat);
Packit ae235b
Packit ae235b
  if (!g_close (file->priv->fd, NULL))
Packit ae235b
    {
Packit ae235b
      int errsv = errno;
Packit ae235b
      
Packit ae235b
      g_set_error (error, G_IO_ERROR,
Packit ae235b
		   g_io_error_from_errno (errsv),
Packit ae235b
		   _("Error closing file: %s"),
Packit ae235b
		   g_strerror (errsv));
Packit ae235b
      return FALSE;
Packit ae235b
    }
Packit ae235b
Packit ae235b
#endif
Packit ae235b
Packit ae235b
  if (file->priv->tmp_filename)
Packit ae235b
    {
Packit ae235b
      /* We need to move the temp file to its final place,
Packit ae235b
       * and possibly create the backup file
Packit ae235b
       */
Packit ae235b
Packit ae235b
      if (file->priv->backup_filename)
Packit ae235b
	{
Packit ae235b
	  if (g_cancellable_set_error_if_cancelled (cancellable, error))
Packit ae235b
	    goto err_out;
Packit ae235b
	  
Packit ae235b
#ifdef G_OS_UNIX
Packit ae235b
	  /* create original -> backup link, the original is then renamed over */
Packit ae235b
	  if (g_unlink (file->priv->backup_filename) != 0 &&
Packit ae235b
	      errno != ENOENT)
Packit ae235b
	    {
Packit ae235b
              int errsv = errno;
Packit ae235b
Packit ae235b
	      g_set_error (error, G_IO_ERROR,
Packit ae235b
			   G_IO_ERROR_CANT_CREATE_BACKUP,
Packit ae235b
			   _("Error removing old backup link: %s"),
Packit ae235b
			   g_strerror (errsv));
Packit ae235b
	      goto err_out;
Packit ae235b
	    }
Packit ae235b
Packit ae235b
	  if (link (file->priv->original_filename, file->priv->backup_filename) != 0)
Packit ae235b
	    {
Packit ae235b
	      /*  link failed or is not supported, try rename  */
Packit ae235b
	      if (g_rename (file->priv->original_filename, file->priv->backup_filename) != 0)
Packit ae235b
		{
Packit ae235b
                  int errsv = errno;
Packit ae235b
Packit ae235b
	    	  g_set_error (error, G_IO_ERROR,
Packit ae235b
		    	       G_IO_ERROR_CANT_CREATE_BACKUP,
Packit ae235b
			       _("Error creating backup copy: %s"),
Packit ae235b
			       g_strerror (errsv));
Packit ae235b
	          goto err_out;
Packit ae235b
		}
Packit ae235b
	    }
Packit ae235b
#else
Packit ae235b
	    /* If link not supported, just rename... */
Packit ae235b
	  if (g_rename (file->priv->original_filename, file->priv->backup_filename) != 0)
Packit ae235b
	    {
Packit ae235b
              int errsv = errno;
Packit ae235b
Packit ae235b
	      g_set_error (error, G_IO_ERROR,
Packit ae235b
			   G_IO_ERROR_CANT_CREATE_BACKUP,
Packit ae235b
			   _("Error creating backup copy: %s"),
Packit ae235b
			   g_strerror (errsv));
Packit ae235b
	      goto err_out;
Packit ae235b
	    }
Packit ae235b
#endif
Packit ae235b
	}
Packit ae235b
      
Packit ae235b
Packit ae235b
      if (g_cancellable_set_error_if_cancelled (cancellable, error))
Packit ae235b
	goto err_out;
Packit ae235b
Packit ae235b
      /* tmp -> original */
Packit ae235b
      if (g_rename (file->priv->tmp_filename, file->priv->original_filename) != 0)
Packit ae235b
	{
Packit ae235b
          int errsv = errno;
Packit ae235b
Packit ae235b
	  g_set_error (error, G_IO_ERROR,
Packit ae235b
		       g_io_error_from_errno (errsv),
Packit ae235b
		       _("Error renaming temporary file: %s"),
Packit ae235b
		       g_strerror (errsv));
Packit ae235b
	  goto err_out;
Packit ae235b
	}
Packit ae235b
Packit ae235b
      g_clear_pointer (&file->priv->tmp_filename, g_free);
Packit ae235b
    }
Packit ae235b
  
Packit ae235b
  if (g_cancellable_set_error_if_cancelled (cancellable, error))
Packit ae235b
    goto err_out;
Packit ae235b
      
Packit ae235b
#ifndef G_OS_WIN32		/* Already did the fstat() and close() above on Win32 */
Packit ae235b
Packit ae235b
  if (fstat (file->priv->fd, &final_stat) == 0)
Packit ae235b
    file->priv->etag = _g_local_file_info_create_etag (&final_stat);
Packit ae235b
Packit ae235b
  if (!g_close (file->priv->fd, NULL))
Packit ae235b
    {
Packit ae235b
      int errsv = errno;
Packit ae235b
      
Packit ae235b
      g_set_error (error, G_IO_ERROR,
Packit ae235b
                   g_io_error_from_errno (errsv),
Packit ae235b
                   _("Error closing file: %s"),
Packit ae235b
                   g_strerror (errsv));
Packit ae235b
      goto err_out;
Packit ae235b
    }
Packit ae235b
Packit ae235b
#endif
Packit ae235b
  
Packit ae235b
  return TRUE;
Packit ae235b
 err_out:
Packit ae235b
Packit ae235b
#ifndef G_OS_WIN32
Packit ae235b
  /* A simple try to close the fd in case we fail before the actual close */
Packit ae235b
  g_close (file->priv->fd, NULL);
Packit ae235b
#endif
Packit ae235b
  if (file->priv->tmp_filename)
Packit ae235b
    g_unlink (file->priv->tmp_filename);
Packit ae235b
Packit ae235b
  return FALSE;
Packit ae235b
}
Packit ae235b
Packit ae235b
Packit ae235b
static gboolean
Packit ae235b
g_local_file_output_stream_close (GOutputStream  *stream,
Packit ae235b
				  GCancellable   *cancellable,
Packit ae235b
				  GError        **error)
Packit ae235b
{
Packit ae235b
  GLocalFileOutputStream *file;
Packit ae235b
Packit ae235b
  file = G_LOCAL_FILE_OUTPUT_STREAM (stream);
Packit ae235b
Packit ae235b
  if (file->priv->do_close)
Packit ae235b
    return _g_local_file_output_stream_really_close (file,
Packit ae235b
						     cancellable,
Packit ae235b
						     error);
Packit ae235b
  return TRUE;
Packit ae235b
}
Packit ae235b
Packit ae235b
static char *
Packit ae235b
g_local_file_output_stream_get_etag (GFileOutputStream *stream)
Packit ae235b
{
Packit ae235b
  GLocalFileOutputStream *file;
Packit ae235b
Packit ae235b
  file = G_LOCAL_FILE_OUTPUT_STREAM (stream);
Packit ae235b
  
Packit ae235b
  return g_strdup (file->priv->etag);
Packit ae235b
}
Packit ae235b
Packit ae235b
static goffset
Packit ae235b
g_local_file_output_stream_tell (GFileOutputStream *stream)
Packit ae235b
{
Packit ae235b
  GLocalFileOutputStream *file;
Packit ae235b
  off_t pos;
Packit ae235b
Packit ae235b
  file = G_LOCAL_FILE_OUTPUT_STREAM (stream);
Packit ae235b
  
Packit ae235b
  pos = lseek (file->priv->fd, 0, SEEK_CUR);
Packit ae235b
Packit ae235b
  if (pos == (off_t)-1)
Packit ae235b
    return 0;
Packit ae235b
  
Packit ae235b
  return pos;
Packit ae235b
}
Packit ae235b
Packit ae235b
static gboolean
Packit ae235b
g_local_file_output_stream_can_seek (GFileOutputStream *stream)
Packit ae235b
{
Packit ae235b
  GLocalFileOutputStream *file;
Packit ae235b
  off_t pos;
Packit ae235b
Packit ae235b
  file = G_LOCAL_FILE_OUTPUT_STREAM (stream);
Packit ae235b
  
Packit ae235b
  pos = lseek (file->priv->fd, 0, SEEK_CUR);
Packit ae235b
Packit ae235b
  if (pos == (off_t)-1 && errno == ESPIPE)
Packit ae235b
    return FALSE;
Packit ae235b
  
Packit ae235b
  return TRUE;
Packit ae235b
}
Packit ae235b
Packit ae235b
static int
Packit ae235b
seek_type_to_lseek (GSeekType type)
Packit ae235b
{
Packit ae235b
  switch (type)
Packit ae235b
    {
Packit ae235b
    default:
Packit ae235b
    case G_SEEK_CUR:
Packit ae235b
      return SEEK_CUR;
Packit ae235b
      
Packit ae235b
    case G_SEEK_SET:
Packit ae235b
      return SEEK_SET;
Packit ae235b
      
Packit ae235b
    case G_SEEK_END:
Packit ae235b
      return SEEK_END;
Packit ae235b
    }
Packit ae235b
}
Packit ae235b
Packit ae235b
static gboolean
Packit ae235b
g_local_file_output_stream_seek (GFileOutputStream  *stream,
Packit ae235b
				 goffset             offset,
Packit ae235b
				 GSeekType           type,
Packit ae235b
				 GCancellable       *cancellable,
Packit ae235b
				 GError            **error)
Packit ae235b
{
Packit ae235b
  GLocalFileOutputStream *file;
Packit ae235b
  off_t pos;
Packit ae235b
Packit ae235b
  file = G_LOCAL_FILE_OUTPUT_STREAM (stream);
Packit ae235b
Packit ae235b
  pos = lseek (file->priv->fd, offset, seek_type_to_lseek (type));
Packit ae235b
Packit ae235b
  if (pos == (off_t)-1)
Packit ae235b
    {
Packit ae235b
      int errsv = errno;
Packit ae235b
Packit ae235b
      g_set_error (error, G_IO_ERROR,
Packit ae235b
		   g_io_error_from_errno (errsv),
Packit ae235b
		   _("Error seeking in file: %s"),
Packit ae235b
		   g_strerror (errsv));
Packit ae235b
      return FALSE;
Packit ae235b
    }
Packit ae235b
  
Packit ae235b
  return TRUE;
Packit ae235b
}
Packit ae235b
Packit ae235b
static gboolean
Packit ae235b
g_local_file_output_stream_can_truncate (GFileOutputStream *stream)
Packit ae235b
{
Packit ae235b
  /* We can't truncate pipes and stuff where we can't seek */
Packit ae235b
  return g_local_file_output_stream_can_seek (stream);
Packit ae235b
}
Packit ae235b
Packit ae235b
static gboolean
Packit ae235b
g_local_file_output_stream_truncate (GFileOutputStream  *stream,
Packit ae235b
				     goffset             size,
Packit ae235b
				     GCancellable       *cancellable,
Packit ae235b
				     GError            **error)
Packit ae235b
{
Packit ae235b
  GLocalFileOutputStream *file;
Packit ae235b
  int res;
Packit ae235b
Packit ae235b
  file = G_LOCAL_FILE_OUTPUT_STREAM (stream);
Packit ae235b
Packit ae235b
 restart:
Packit ae235b
#ifdef G_OS_WIN32
Packit ae235b
  res = g_win32_ftruncate (file->priv->fd, size);
Packit ae235b
#else
Packit ae235b
  res = ftruncate (file->priv->fd, size);
Packit ae235b
#endif
Packit ae235b
  
Packit ae235b
  if (res == -1)
Packit ae235b
    {
Packit ae235b
      int errsv = errno;
Packit ae235b
Packit ae235b
      if (errsv == EINTR)
Packit ae235b
	{
Packit ae235b
	  if (g_cancellable_set_error_if_cancelled (cancellable, error))
Packit ae235b
	    return FALSE;
Packit ae235b
	  goto restart;
Packit ae235b
	}
Packit ae235b
Packit ae235b
      g_set_error (error, G_IO_ERROR,
Packit ae235b
		   g_io_error_from_errno (errsv),
Packit ae235b
		   _("Error truncating file: %s"),
Packit ae235b
		   g_strerror (errsv));
Packit ae235b
      return FALSE;
Packit ae235b
    }
Packit ae235b
  
Packit ae235b
  return TRUE;
Packit ae235b
}
Packit ae235b
Packit ae235b
Packit ae235b
static GFileInfo *
Packit ae235b
g_local_file_output_stream_query_info (GFileOutputStream  *stream,
Packit ae235b
				       const char         *attributes,
Packit ae235b
				       GCancellable       *cancellable,
Packit ae235b
				       GError            **error)
Packit ae235b
{
Packit ae235b
  GLocalFileOutputStream *file;
Packit ae235b
Packit ae235b
  file = G_LOCAL_FILE_OUTPUT_STREAM (stream);
Packit ae235b
Packit ae235b
  if (g_cancellable_set_error_if_cancelled (cancellable, error))
Packit ae235b
    return NULL;
Packit ae235b
  
Packit ae235b
  return _g_local_file_info_get_from_fd (file->priv->fd,
Packit ae235b
					 attributes,
Packit ae235b
					 error);
Packit ae235b
}
Packit ae235b
Packit ae235b
GFileOutputStream *
Packit ae235b
_g_local_file_output_stream_new (int fd)
Packit ae235b
{
Packit ae235b
  GLocalFileOutputStream *stream;
Packit ae235b
Packit ae235b
  stream = g_object_new (G_TYPE_LOCAL_FILE_OUTPUT_STREAM, NULL);
Packit ae235b
  stream->priv->fd = fd;
Packit ae235b
  return G_FILE_OUTPUT_STREAM (stream);
Packit ae235b
}
Packit ae235b
Packit ae235b
static void
Packit ae235b
set_error_from_open_errno (const char *filename,
Packit ae235b
                           GError    **error)
Packit ae235b
{
Packit ae235b
  int errsv = errno;
Packit ae235b
  
Packit ae235b
  if (errsv == EINVAL)
Packit ae235b
    /* This must be an invalid filename, on e.g. FAT */
Packit ae235b
    g_set_error_literal (error, G_IO_ERROR,
Packit ae235b
                         G_IO_ERROR_INVALID_FILENAME,
Packit ae235b
                         _("Invalid filename"));
Packit ae235b
  else
Packit ae235b
    {
Packit ae235b
      char *display_name = g_filename_display_name (filename);
Packit ae235b
      g_set_error (error, G_IO_ERROR,
Packit ae235b
                   g_io_error_from_errno (errsv),
Packit ae235b
                   _("Error opening file ā€œ%sā€: %s"),
Packit ae235b
		       display_name, g_strerror (errsv));
Packit ae235b
      g_free (display_name);
Packit ae235b
    }
Packit ae235b
}
Packit ae235b
Packit ae235b
static GFileOutputStream *
Packit ae235b
output_stream_open (const char    *filename,
Packit ae235b
                    gint           open_flags,
Packit ae235b
                    guint          mode,
Packit ae235b
                    GCancellable  *cancellable,
Packit ae235b
                    GError       **error)
Packit ae235b
{
Packit ae235b
  GLocalFileOutputStream *stream;
Packit ae235b
  gint fd;
Packit ae235b
Packit ae235b
  fd = g_open (filename, open_flags, mode);
Packit ae235b
  if (fd == -1)
Packit ae235b
    {
Packit ae235b
      set_error_from_open_errno (filename, error);
Packit ae235b
      return NULL;
Packit ae235b
    }
Packit ae235b
  
Packit ae235b
  stream = g_object_new (G_TYPE_LOCAL_FILE_OUTPUT_STREAM, NULL);
Packit ae235b
  stream->priv->fd = fd;
Packit ae235b
  return G_FILE_OUTPUT_STREAM (stream);
Packit ae235b
}
Packit ae235b
Packit ae235b
GFileOutputStream *
Packit ae235b
_g_local_file_output_stream_open  (const char        *filename,
Packit ae235b
				   gboolean          readable,
Packit ae235b
				   GCancellable      *cancellable,
Packit ae235b
				   GError           **error)
Packit ae235b
{
Packit ae235b
  int open_flags;
Packit ae235b
Packit ae235b
  if (g_cancellable_set_error_if_cancelled (cancellable, error))
Packit ae235b
    return NULL;
Packit ae235b
Packit ae235b
  open_flags = O_BINARY;
Packit ae235b
  if (readable)
Packit ae235b
    open_flags |= O_RDWR;
Packit ae235b
  else
Packit ae235b
    open_flags |= O_WRONLY;
Packit ae235b
Packit ae235b
  return output_stream_open (filename, open_flags, 0666, cancellable, error);
Packit ae235b
}
Packit ae235b
Packit ae235b
static gint
Packit ae235b
mode_from_flags_or_info (GFileCreateFlags   flags,
Packit ae235b
                         GFileInfo         *reference_info)
Packit ae235b
{
Packit ae235b
  if (flags & G_FILE_CREATE_PRIVATE)
Packit ae235b
    return 0600;
Packit ae235b
  else if (reference_info && g_file_info_has_attribute (reference_info, "unix::mode"))
Packit ae235b
    return g_file_info_get_attribute_uint32 (reference_info, "unix::mode") & (~S_IFMT);
Packit ae235b
  else
Packit ae235b
    return 0666;
Packit ae235b
}
Packit ae235b
Packit ae235b
GFileOutputStream *
Packit ae235b
_g_local_file_output_stream_create  (const char        *filename,
Packit ae235b
				     gboolean          readable,
Packit ae235b
				     GFileCreateFlags   flags,
Packit ae235b
                                     GFileInfo         *reference_info,
Packit ae235b
				     GCancellable      *cancellable,
Packit ae235b
				     GError           **error)
Packit ae235b
{
Packit ae235b
  int mode;
Packit ae235b
  int open_flags;
Packit ae235b
Packit ae235b
  if (g_cancellable_set_error_if_cancelled (cancellable, error))
Packit ae235b
    return NULL;
Packit ae235b
Packit ae235b
  mode = mode_from_flags_or_info (flags, reference_info);
Packit ae235b
Packit ae235b
  open_flags = O_CREAT | O_EXCL | O_BINARY;
Packit ae235b
  if (readable)
Packit ae235b
    open_flags |= O_RDWR;
Packit ae235b
  else
Packit ae235b
    open_flags |= O_WRONLY;
Packit ae235b
Packit ae235b
  return output_stream_open (filename, open_flags, mode, cancellable, error);
Packit ae235b
}
Packit ae235b
Packit ae235b
GFileOutputStream *
Packit ae235b
_g_local_file_output_stream_append  (const char        *filename,
Packit ae235b
				     GFileCreateFlags   flags,
Packit ae235b
				     GCancellable      *cancellable,
Packit ae235b
				     GError           **error)
Packit ae235b
{
Packit ae235b
  int mode;
Packit ae235b
Packit ae235b
  if (g_cancellable_set_error_if_cancelled (cancellable, error))
Packit ae235b
    return NULL;
Packit ae235b
Packit ae235b
  if (flags & G_FILE_CREATE_PRIVATE)
Packit ae235b
    mode = 0600;
Packit ae235b
  else
Packit ae235b
    mode = 0666;
Packit ae235b
Packit ae235b
  return output_stream_open (filename, O_CREAT | O_APPEND | O_WRONLY | O_BINARY, mode,
Packit ae235b
                             cancellable, error);
Packit ae235b
}
Packit ae235b
Packit ae235b
static char *
Packit ae235b
create_backup_filename (const char *filename)
Packit ae235b
{
Packit ae235b
  return g_strconcat (filename, BACKUP_EXTENSION, NULL);
Packit ae235b
}
Packit ae235b
Packit ae235b
#define BUFSIZE	8192 /* size of normal write buffer */
Packit ae235b
Packit ae235b
static gboolean
Packit ae235b
copy_file_data (gint     sfd,
Packit ae235b
		gint     dfd,
Packit ae235b
		GError **error)
Packit ae235b
{
Packit ae235b
  gboolean ret = TRUE;
Packit ae235b
  gpointer buffer;
Packit ae235b
  const gchar *write_buffer;
Packit ae235b
  gssize bytes_read;
Packit ae235b
  gssize bytes_to_write;
Packit ae235b
  gssize bytes_written;
Packit ae235b
  
Packit ae235b
  buffer = g_malloc (BUFSIZE);
Packit ae235b
  
Packit ae235b
  do
Packit ae235b
    {
Packit ae235b
      bytes_read = read (sfd, buffer, BUFSIZE);
Packit ae235b
      if (bytes_read == -1)
Packit ae235b
	{
Packit ae235b
          int errsv = errno;
Packit ae235b
Packit ae235b
	  if (errsv == EINTR)
Packit ae235b
	    continue;
Packit ae235b
	  
Packit ae235b
	  g_set_error (error, G_IO_ERROR,
Packit ae235b
		       g_io_error_from_errno (errsv),
Packit ae235b
		       _("Error reading from file: %s"),
Packit ae235b
		       g_strerror (errsv));
Packit ae235b
	  ret = FALSE;
Packit ae235b
	  break;
Packit ae235b
	}
Packit ae235b
      
Packit ae235b
      bytes_to_write = bytes_read;
Packit ae235b
      write_buffer = buffer;
Packit ae235b
      
Packit ae235b
      do
Packit ae235b
	{
Packit ae235b
	  bytes_written = write (dfd, write_buffer, bytes_to_write);
Packit ae235b
	  if (bytes_written == -1)
Packit ae235b
	    {
Packit ae235b
              int errsv = errno;
Packit ae235b
Packit ae235b
	      if (errsv == EINTR)
Packit ae235b
		continue;
Packit ae235b
	      
Packit ae235b
	      g_set_error (error, G_IO_ERROR,
Packit ae235b
			   g_io_error_from_errno (errsv),
Packit ae235b
			   _("Error writing to file: %s"),
Packit ae235b
			   g_strerror (errsv));
Packit ae235b
	      ret = FALSE;
Packit ae235b
	      break;
Packit ae235b
	    }
Packit ae235b
	  
Packit ae235b
	  bytes_to_write -= bytes_written;
Packit ae235b
	  write_buffer += bytes_written;
Packit ae235b
	}
Packit ae235b
      while (bytes_to_write > 0);
Packit ae235b
      
Packit ae235b
    } while ((bytes_read != 0) && (ret == TRUE));
Packit ae235b
Packit ae235b
  g_free (buffer);
Packit ae235b
  
Packit ae235b
  return ret;
Packit ae235b
}
Packit ae235b
Packit ae235b
static int
Packit ae235b
handle_overwrite_open (const char    *filename,
Packit ae235b
		       gboolean       readable,
Packit ae235b
		       const char    *etag,
Packit ae235b
		       gboolean       create_backup,
Packit ae235b
		       char         **temp_filename,
Packit ae235b
		       GFileCreateFlags flags,
Packit ae235b
                       GFileInfo       *reference_info,
Packit ae235b
		       GCancellable  *cancellable,
Packit ae235b
		       GError       **error)
Packit ae235b
{
Packit ae235b
  int fd = -1;
Packit ae235b
  GLocalFileStat original_stat;
Packit ae235b
  char *current_etag;
Packit ae235b
  gboolean is_symlink;
Packit ae235b
  int open_flags;
Packit ae235b
  int res;
Packit ae235b
  int mode;
Packit ae235b
  int errsv;
Packit ae235b
Packit ae235b
  mode = mode_from_flags_or_info (flags, reference_info);
Packit ae235b
Packit ae235b
  /* We only need read access to the original file if we are creating a backup.
Packit ae235b
   * We also add O_CREATE to avoid a race if the file was just removed */
Packit ae235b
  if (create_backup || readable)
Packit ae235b
    open_flags = O_RDWR | O_CREAT | O_BINARY;
Packit ae235b
  else
Packit ae235b
    open_flags = O_WRONLY | O_CREAT | O_BINARY;
Packit ae235b
  
Packit ae235b
  /* Some systems have O_NOFOLLOW, which lets us avoid some races
Packit ae235b
   * when finding out if the file we opened was a symlink */
Packit ae235b
#ifdef O_NOFOLLOW
Packit ae235b
  is_symlink = FALSE;
Packit ae235b
  fd = g_open (filename, open_flags | O_NOFOLLOW, mode);
Packit ae235b
  errsv = errno;
Packit ae235b
#if defined(__FreeBSD__) || defined(__FreeBSD_kernel__) || defined(__DragonFly__)
Packit ae235b
  if (fd == -1 && errsv == EMLINK)
Packit ae235b
#elif defined(__NetBSD__)
Packit ae235b
  if (fd == -1 && errsv == EFTYPE)
Packit ae235b
#else
Packit ae235b
  if (fd == -1 && errsv == ELOOP)
Packit ae235b
#endif
Packit ae235b
    {
Packit ae235b
      /* Could be a symlink, or it could be a regular ELOOP error,
Packit ae235b
       * but then the next open will fail too. */
Packit ae235b
      is_symlink = TRUE;
Packit ae235b
      fd = g_open (filename, open_flags, mode);
Packit ae235b
    }
Packit ae235b
#else
Packit ae235b
  fd = g_open (filename, open_flags, mode);
Packit ae235b
  errsv = errno;
Packit ae235b
  /* This is racy, but we do it as soon as possible to minimize the race */
Packit ae235b
  is_symlink = g_file_test (filename, G_FILE_TEST_IS_SYMLINK);
Packit ae235b
#endif
Packit ae235b
Packit ae235b
  if (fd == -1)
Packit ae235b
    {
Packit ae235b
      char *display_name = g_filename_display_name (filename);
Packit ae235b
      g_set_error (error, G_IO_ERROR,
Packit ae235b
		   g_io_error_from_errno (errsv),
Packit ae235b
		   _("Error opening file ā€œ%sā€: %s"),
Packit ae235b
		   display_name, g_strerror (errsv));
Packit ae235b
      g_free (display_name);
Packit ae235b
      return -1;
Packit ae235b
    }
Packit ae235b
  
Packit ae235b
#ifdef G_OS_WIN32
Packit ae235b
  res = GLIB_PRIVATE_CALL (g_win32_fstat) (fd, &original_stat);
Packit ae235b
#else
Packit ae235b
  res = fstat (fd, &original_stat);
Packit ae235b
#endif
Packit ae235b
  errsv = errno;
Packit ae235b
Packit ae235b
  if (res != 0)
Packit ae235b
    {
Packit ae235b
      char *display_name = g_filename_display_name (filename);
Packit ae235b
      g_set_error (error, G_IO_ERROR,
Packit ae235b
		   g_io_error_from_errno (errsv),
Packit ae235b
		   _("Error when getting information for file ā€œ%sā€: %s"),
Packit ae235b
		   display_name, g_strerror (errsv));
Packit ae235b
      g_free (display_name);
Packit ae235b
      goto err_out;
Packit ae235b
    }
Packit ae235b
  
Packit ae235b
  /* not a regular file */
Packit ae235b
  if (!S_ISREG (original_stat.st_mode))
Packit ae235b
    {
Packit ae235b
      if (S_ISDIR (original_stat.st_mode))
Packit ae235b
	g_set_error_literal (error,
Packit ae235b
                             G_IO_ERROR,
Packit ae235b
                             G_IO_ERROR_IS_DIRECTORY,
Packit ae235b
                             _("Target file is a directory"));
Packit ae235b
      else
Packit ae235b
	g_set_error_literal (error,
Packit ae235b
                             G_IO_ERROR,
Packit ae235b
                             G_IO_ERROR_NOT_REGULAR_FILE,
Packit ae235b
                             _("Target file is not a regular file"));
Packit ae235b
      goto err_out;
Packit ae235b
    }
Packit ae235b
  
Packit ae235b
  if (etag != NULL)
Packit ae235b
    {
Packit ae235b
      current_etag = _g_local_file_info_create_etag (&original_stat);
Packit ae235b
      if (strcmp (etag, current_etag) != 0)
Packit ae235b
	{
Packit ae235b
	  g_set_error_literal (error,
Packit ae235b
                               G_IO_ERROR,
Packit ae235b
                               G_IO_ERROR_WRONG_ETAG,
Packit ae235b
                               _("The file was externally modified"));
Packit ae235b
	  g_free (current_etag);
Packit ae235b
	  goto err_out;
Packit ae235b
	}
Packit ae235b
      g_free (current_etag);
Packit ae235b
    }
Packit ae235b
Packit ae235b
  /* We use two backup strategies.
Packit ae235b
   * The first one (which is faster) consist in saving to a
Packit ae235b
   * tmp file then rename the original file to the backup and the
Packit ae235b
   * tmp file to the original name. This is fast but doesn't work
Packit ae235b
   * when the file is a link (hard or symbolic) or when we can't
Packit ae235b
   * write to the current dir or can't set the permissions on the
Packit ae235b
   * new file. 
Packit ae235b
   * The second strategy consist simply in copying the old file
Packit ae235b
   * to a backup file and rewrite the contents of the file.
Packit ae235b
   */
Packit ae235b
  
Packit ae235b
  if ((flags & G_FILE_CREATE_REPLACE_DESTINATION) ||
Packit ae235b
      (!(original_stat.st_nlink > 1) && !is_symlink))
Packit ae235b
    {
Packit ae235b
      char *dirname, *tmp_filename;
Packit ae235b
      int tmpfd;
Packit ae235b
      
Packit ae235b
      dirname = g_path_get_dirname (filename);
Packit ae235b
      tmp_filename = g_build_filename (dirname, ".goutputstream-XXXXXX", NULL);
Packit ae235b
      g_free (dirname);
Packit ae235b
Packit ae235b
      tmpfd = g_mkstemp_full (tmp_filename, (readable ? O_RDWR : O_WRONLY) | O_BINARY, mode);
Packit ae235b
      if (tmpfd == -1)
Packit ae235b
	{
Packit ae235b
	  g_free (tmp_filename);
Packit ae235b
	  goto fallback_strategy;
Packit ae235b
	}
Packit ae235b
      
Packit ae235b
      /* try to keep permissions (unless replacing) */
Packit ae235b
Packit ae235b
      if ( ! (flags & G_FILE_CREATE_REPLACE_DESTINATION) &&
Packit ae235b
	   (
Packit ae235b
#ifdef HAVE_FCHOWN
Packit ae235b
	    fchown (tmpfd, original_stat.st_uid, original_stat.st_gid) == -1 ||
Packit ae235b
#endif
Packit ae235b
#ifdef HAVE_FCHMOD
Packit ae235b
	    fchmod (tmpfd, original_stat.st_mode) == -1 ||
Packit ae235b
#endif
Packit ae235b
	    0
Packit ae235b
	    )
Packit ae235b
	  )
Packit ae235b
	{
Packit ae235b
          GLocalFileStat tmp_statbuf;
Packit ae235b
          int tres;
Packit ae235b
Packit ae235b
#ifdef G_OS_WIN32
Packit ae235b
          tres = GLIB_PRIVATE_CALL (g_win32_fstat) (tmpfd, &tmp_statbuf);
Packit ae235b
#else
Packit ae235b
          tres = fstat (tmpfd, &tmp_statbuf);
Packit ae235b
#endif
Packit ae235b
	  /* Check that we really needed to change something */
Packit ae235b
	  if (tres != 0 ||
Packit ae235b
	      original_stat.st_uid != tmp_statbuf.st_uid ||
Packit ae235b
	      original_stat.st_gid != tmp_statbuf.st_gid ||
Packit ae235b
	      original_stat.st_mode != tmp_statbuf.st_mode)
Packit ae235b
	    {
Packit ae235b
	      g_close (tmpfd, NULL);
Packit ae235b
	      g_unlink (tmp_filename);
Packit ae235b
	      g_free (tmp_filename);
Packit ae235b
	      goto fallback_strategy;
Packit ae235b
	    }
Packit ae235b
	}
Packit ae235b
Packit ae235b
      g_close (fd, NULL);
Packit ae235b
      *temp_filename = tmp_filename;
Packit ae235b
      return tmpfd;
Packit ae235b
    }
Packit ae235b
Packit ae235b
 fallback_strategy:
Packit ae235b
Packit ae235b
  if (create_backup)
Packit ae235b
    {
Packit ae235b
#if defined(HAVE_FCHOWN) && defined(HAVE_FCHMOD)
Packit ae235b
      struct stat tmp_statbuf;      
Packit ae235b
#endif
Packit ae235b
      char *backup_filename;
Packit ae235b
      int bfd;
Packit ae235b
      
Packit ae235b
      backup_filename = create_backup_filename (filename);
Packit ae235b
Packit ae235b
      if (g_unlink (backup_filename) == -1 && errno != ENOENT)
Packit ae235b
	{
Packit ae235b
	  g_set_error_literal (error,
Packit ae235b
                               G_IO_ERROR,
Packit ae235b
                               G_IO_ERROR_CANT_CREATE_BACKUP,
Packit ae235b
                               _("Backup file creation failed"));
Packit ae235b
	  g_free (backup_filename);
Packit ae235b
	  goto err_out;
Packit ae235b
	}
Packit ae235b
Packit ae235b
      bfd = g_open (backup_filename,
Packit ae235b
		    O_WRONLY | O_CREAT | O_EXCL | O_BINARY,
Packit ae235b
		    original_stat.st_mode & 0777);
Packit ae235b
Packit ae235b
      if (bfd == -1)
Packit ae235b
	{
Packit ae235b
	  g_set_error_literal (error,
Packit ae235b
                               G_IO_ERROR,
Packit ae235b
                               G_IO_ERROR_CANT_CREATE_BACKUP,
Packit ae235b
                               _("Backup file creation failed"));
Packit ae235b
	  g_free (backup_filename);
Packit ae235b
	  goto err_out;
Packit ae235b
	}
Packit ae235b
Packit ae235b
      /* If needed, Try to set the group of the backup same as the
Packit ae235b
       * original file. If this fails, set the protection
Packit ae235b
       * bits for the group same as the protection bits for
Packit ae235b
       * others. */
Packit ae235b
#if defined(HAVE_FCHOWN) && defined(HAVE_FCHMOD)
Packit ae235b
      if (fstat (bfd, &tmp_statbuf) != 0)
Packit ae235b
	{
Packit ae235b
	  g_set_error_literal (error,
Packit ae235b
                               G_IO_ERROR,
Packit ae235b
                               G_IO_ERROR_CANT_CREATE_BACKUP,
Packit ae235b
                               _("Backup file creation failed"));
Packit ae235b
	  g_unlink (backup_filename);
Packit ae235b
	  g_close (bfd, NULL);
Packit ae235b
	  g_free (backup_filename);
Packit ae235b
	  goto err_out;
Packit ae235b
	}
Packit ae235b
      
Packit ae235b
      if ((original_stat.st_gid != tmp_statbuf.st_gid)  &&
Packit ae235b
	  fchown (bfd, (uid_t) -1, original_stat.st_gid) != 0)
Packit ae235b
	{
Packit ae235b
	  if (fchmod (bfd,
Packit ae235b
		      (original_stat.st_mode & 0707) |
Packit ae235b
		      ((original_stat.st_mode & 07) << 3)) != 0)
Packit ae235b
	    {
Packit ae235b
	      g_set_error_literal (error,
Packit ae235b
                                   G_IO_ERROR,
Packit ae235b
                                   G_IO_ERROR_CANT_CREATE_BACKUP,
Packit ae235b
                                   _("Backup file creation failed"));
Packit ae235b
	      g_unlink (backup_filename);
Packit ae235b
	      g_close (bfd, NULL);
Packit ae235b
	      g_free (backup_filename);
Packit ae235b
	      goto err_out;
Packit ae235b
	    }
Packit ae235b
	}
Packit ae235b
#endif
Packit ae235b
Packit ae235b
      if (!copy_file_data (fd, bfd, NULL))
Packit ae235b
	{
Packit ae235b
	  g_set_error_literal (error,
Packit ae235b
                               G_IO_ERROR,
Packit ae235b
                               G_IO_ERROR_CANT_CREATE_BACKUP,
Packit ae235b
                               _("Backup file creation failed"));
Packit ae235b
	  g_unlink (backup_filename);
Packit ae235b
          g_close (bfd, NULL);
Packit ae235b
	  g_free (backup_filename);
Packit ae235b
	  
Packit ae235b
	  goto err_out;
Packit ae235b
	}
Packit ae235b
      
Packit ae235b
      g_close (bfd, NULL);
Packit ae235b
      g_free (backup_filename);
Packit ae235b
Packit ae235b
      /* Seek back to the start of the file after the backup copy */
Packit ae235b
      if (lseek (fd, 0, SEEK_SET) == -1)
Packit ae235b
	{
Packit ae235b
          int errsv = errno;
Packit ae235b
Packit ae235b
	  g_set_error (error, G_IO_ERROR,
Packit ae235b
		       g_io_error_from_errno (errsv),
Packit ae235b
		       _("Error seeking in file: %s"),
Packit ae235b
		       g_strerror (errsv));
Packit ae235b
	  goto err_out;
Packit ae235b
	}
Packit ae235b
    }
Packit ae235b
Packit ae235b
  if (flags & G_FILE_CREATE_REPLACE_DESTINATION)
Packit ae235b
    {
Packit ae235b
      g_close (fd, NULL);
Packit ae235b
      
Packit ae235b
      if (g_unlink (filename) != 0)
Packit ae235b
	{
Packit ae235b
	  int errsv = errno;
Packit ae235b
	  
Packit ae235b
	  g_set_error (error, G_IO_ERROR,
Packit ae235b
		       g_io_error_from_errno (errsv),
Packit ae235b
		       _("Error removing old file: %s"),
Packit ae235b
		       g_strerror (errsv));
Packit ae235b
	  goto err_out2;
Packit ae235b
	}
Packit ae235b
Packit ae235b
      if (readable)
Packit ae235b
	open_flags = O_RDWR | O_CREAT | O_BINARY;
Packit ae235b
      else
Packit ae235b
	open_flags = O_WRONLY | O_CREAT | O_BINARY;
Packit ae235b
      fd = g_open (filename, open_flags, mode);
Packit ae235b
      if (fd == -1)
Packit ae235b
	{
Packit ae235b
	  int errsv = errno;
Packit ae235b
	  char *display_name = g_filename_display_name (filename);
Packit ae235b
	  g_set_error (error, G_IO_ERROR,
Packit ae235b
		       g_io_error_from_errno (errsv),
Packit ae235b
		       _("Error opening file ā€œ%sā€: %s"),
Packit ae235b
		       display_name, g_strerror (errsv));
Packit ae235b
	  g_free (display_name);
Packit ae235b
	  goto err_out2;
Packit ae235b
	}
Packit ae235b
    }
Packit ae235b
  else
Packit ae235b
    {
Packit ae235b
      /* Truncate the file at the start */
Packit ae235b
#ifdef G_OS_WIN32
Packit ae235b
      if (g_win32_ftruncate (fd, 0) == -1)
Packit ae235b
#else
Packit ae235b
	if (ftruncate (fd, 0) == -1)
Packit ae235b
#endif
Packit ae235b
	  {
Packit ae235b
	    int errsv = errno;
Packit ae235b
	    
Packit ae235b
	    g_set_error (error, G_IO_ERROR,
Packit ae235b
			 g_io_error_from_errno (errsv),
Packit ae235b
			 _("Error truncating file: %s"),
Packit ae235b
			 g_strerror (errsv));
Packit ae235b
	    goto err_out;
Packit ae235b
	  }
Packit ae235b
    }
Packit ae235b
    
Packit ae235b
  return fd;
Packit ae235b
Packit ae235b
 err_out:
Packit ae235b
  g_close (fd, NULL);
Packit ae235b
 err_out2:
Packit ae235b
  return -1;
Packit ae235b
}
Packit ae235b
Packit ae235b
GFileOutputStream *
Packit ae235b
_g_local_file_output_stream_replace (const char        *filename,
Packit ae235b
				     gboolean          readable,
Packit ae235b
				     const char        *etag,
Packit ae235b
				     gboolean           create_backup,
Packit ae235b
				     GFileCreateFlags   flags,
Packit ae235b
                                     GFileInfo         *reference_info,
Packit ae235b
				     GCancellable      *cancellable,
Packit ae235b
				     GError           **error)
Packit ae235b
{
Packit ae235b
  GLocalFileOutputStream *stream;
Packit ae235b
  int mode;
Packit ae235b
  int fd;
Packit ae235b
  char *temp_file;
Packit ae235b
  gboolean sync_on_close;
Packit ae235b
  int open_flags;
Packit ae235b
Packit ae235b
  if (g_cancellable_set_error_if_cancelled (cancellable, error))
Packit ae235b
    return NULL;
Packit ae235b
Packit ae235b
  temp_file = NULL;
Packit ae235b
Packit ae235b
  mode = mode_from_flags_or_info (flags, reference_info);
Packit ae235b
  sync_on_close = FALSE;
Packit ae235b
Packit ae235b
  /* If the file doesn't exist, create it */
Packit ae235b
  open_flags = O_CREAT | O_EXCL | O_BINARY;
Packit ae235b
  if (readable)
Packit ae235b
    open_flags |= O_RDWR;
Packit ae235b
  else
Packit ae235b
    open_flags |= O_WRONLY;
Packit ae235b
  fd = g_open (filename, open_flags, mode);
Packit ae235b
Packit ae235b
  if (fd == -1 && errno == EEXIST)
Packit ae235b
    {
Packit ae235b
      /* The file already exists */
Packit ae235b
      fd = handle_overwrite_open (filename, readable, etag,
Packit ae235b
                                  create_backup, &temp_file,
Packit ae235b
                                  flags, reference_info,
Packit ae235b
                                  cancellable, error);
Packit ae235b
      if (fd == -1)
Packit ae235b
	return NULL;
Packit ae235b
Packit ae235b
      /* If the final destination exists, we want to sync the newly written
Packit ae235b
       * file to ensure the data is on disk when we rename over the destination.
Packit ae235b
       * otherwise if we get a system crash we can lose both the new and the
Packit ae235b
       * old file on some filesystems. (I.E. those that don't guarantee the
Packit ae235b
       * data is written to the disk before the metadata.)
Packit ae235b
       */
Packit ae235b
      sync_on_close = TRUE;
Packit ae235b
    }
Packit ae235b
  else if (fd == -1)
Packit ae235b
    {
Packit ae235b
      set_error_from_open_errno (filename, error);
Packit ae235b
      return NULL;
Packit ae235b
    }
Packit ae235b
  
Packit ae235b
 
Packit ae235b
  stream = g_object_new (G_TYPE_LOCAL_FILE_OUTPUT_STREAM, NULL);
Packit ae235b
  stream->priv->fd = fd;
Packit ae235b
  stream->priv->sync_on_close = sync_on_close;
Packit ae235b
  stream->priv->tmp_filename = temp_file;
Packit ae235b
  if (create_backup)
Packit ae235b
    stream->priv->backup_filename = create_backup_filename (filename);
Packit ae235b
  stream->priv->original_filename =  g_strdup (filename);
Packit ae235b
  
Packit ae235b
  return G_FILE_OUTPUT_STREAM (stream);
Packit ae235b
}
Packit ae235b
Packit ae235b
gint
Packit ae235b
_g_local_file_output_stream_get_fd (GLocalFileOutputStream *stream)
Packit ae235b
{
Packit ae235b
  g_return_val_if_fail (G_IS_LOCAL_FILE_OUTPUT_STREAM (stream), -1);
Packit ae235b
  return stream->priv->fd;
Packit ae235b
}
Packit ae235b
Packit ae235b
#ifdef G_OS_UNIX
Packit ae235b
static int
Packit ae235b
g_local_file_output_stream_get_fd (GFileDescriptorBased *fd_based)
Packit ae235b
{
Packit ae235b
  GLocalFileOutputStream *stream = G_LOCAL_FILE_OUTPUT_STREAM (fd_based);
Packit ae235b
  return _g_local_file_output_stream_get_fd (stream);
Packit ae235b
}
Packit ae235b
#endif