X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Flanguage%2Fdata-io%2Fdata-writer.c;h=b5df59d4605f2535a690aea77e0934eb46ee6a65;hb=8ea97bfb8e21cac5964212c175782c48159764b7;hp=6199e77264b1d87cdaef52c3ccabefc9fa051208;hpb=480a0746507ce73d26f528b56dc3ed80195096e0;p=pspp
diff --git a/src/language/data-io/data-writer.c b/src/language/data-io/data-writer.c
index 6199e77264..b5df59d460 100644
--- a/src/language/data-io/data-writer.c
+++ b/src/language/data-io/data-writer.c
@@ -1,20 +1,18 @@
-/* PSPP - computes sample statistics.
+/* PSPP - a program for statistical analysis.
Copyright (C) 1997-2004, 2006 Free Software Foundation, Inc.
- This program is free software; you can redistribute it and/or
- modify it under the terms of the GNU General Public License as
- published by the Free Software Foundation; either version 2 of the
- License, or (at your option) any later version.
+ This program is free software: you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation, either version 3 of the License, or
+ (at your option) any later version.
- This program 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
- General Public License for more details.
+ This program 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 General Public License for more details.
You should have received a copy of the GNU General Public License
- along with this program; if not, write to the Free Software
- Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
- 02110-1301, USA. */
+ along with this program. If not, see . */
#include
@@ -22,25 +20,32 @@
#include
#include
+#include
#include
+#include
#include
+#include
#include
-#include
#include
+#include
#include
#include
#include "minmax.h"
+#include "xalloc.h"
#include "gettext.h"
#define _(msgid) gettext (msgid)
+#define N_(msgid) (msgid)
/* Data file writer. */
struct dfm_writer
{
struct file_handle *fh; /* File handle. */
+ struct fh_lock *lock; /* Exclusive access to file. */
FILE *file; /* Associated file. */
+ struct replace_file *rf; /* Atomic file replacement support. */
};
/* Opens a file handle for writing as a data file. */
@@ -48,36 +53,38 @@ struct dfm_writer *
dfm_open_writer (struct file_handle *fh)
{
struct dfm_writer *w;
- void **aux;
-
- aux = fh_open (fh, FH_REF_FILE, "data file", "ws");
- if (aux == NULL)
- return NULL;
- if (*aux != NULL)
- return *aux;
+ struct fh_lock *lock;
- w = *aux = xmalloc (sizeof *w);
- w->fh = fh;
- w->file = fn_open (fh_get_file_name (w->fh), "wb");
+ lock = fh_lock (fh, FH_REF_FILE, N_("data file"), FH_ACC_WRITE, false);
+ if (lock == NULL)
+ return NULL;
- if (w->file == NULL)
+ w = fh_lock_get_aux (lock);
+ if (w != NULL)
+ return w;
+
+ w = xmalloc (sizeof *w);
+ w->fh = fh_ref (fh);
+ w->lock = lock;
+ w->rf = replace_file_start (fh_get_file_name (w->fh), "wb",
+ (S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP
+ | S_IROTH | S_IWOTH), &w->file, NULL);
+ if (w->rf == NULL)
{
msg (ME, _("An error occurred while opening \"%s\" for writing "
"as a data file: %s."),
fh_get_file_name (w->fh), strerror (errno));
- goto error;
+ dfm_close_writer (w);
+ return NULL;
}
+ fh_lock_set_aux (lock, w);
return w;
-
- error:
- dfm_close_writer (w);
- return NULL;
}
/* Returns false if an I/O error occurred on WRITER, true otherwise. */
bool
-dfm_write_error (const struct dfm_writer *writer)
+dfm_write_error (const struct dfm_writer *writer)
{
return ferror (writer->file);
}
@@ -94,20 +101,20 @@ dfm_put_record (struct dfm_writer *w, const char *rec, size_t len)
if (dfm_write_error (w))
return false;
- switch (fh_get_mode (w->fh))
+ switch (fh_get_mode (w->fh))
{
case FH_MODE_TEXT:
fwrite (rec, len, 1, w->file);
putc ('\n', w->file);
break;
- case FH_MODE_BINARY:
+ case FH_MODE_FIXED:
{
size_t record_width = fh_get_record_width (w->fh);
size_t write_bytes = MIN (len, record_width);
size_t pad_bytes = record_width - write_bytes;
fwrite (rec, write_bytes, 1, w->file);
- while (pad_bytes > 0)
+ while (pad_bytes > 0)
{
static const char spaces[32] = " ";
size_t chunk = MIN (pad_bytes, sizeof spaces);
@@ -117,6 +124,45 @@ dfm_put_record (struct dfm_writer *w, const char *rec, size_t len)
}
break;
+ case FH_MODE_VARIABLE:
+ {
+ uint32_t size = len;
+ integer_convert (INTEGER_NATIVE, &size, INTEGER_LSB_FIRST, &size,
+ sizeof size);
+ fwrite (&size, sizeof size, 1, w->file);
+ fwrite (rec, len, 1, w->file);
+ fwrite (&size, sizeof size, 1, w->file);
+ }
+ break;
+
+ case FH_MODE_360_VARIABLE:
+ case FH_MODE_360_SPANNED:
+ {
+ size_t ofs = 0;
+ if (fh_get_mode (w->fh) == FH_MODE_360_VARIABLE)
+ len = MIN (65527, len);
+ while (ofs < len)
+ {
+ size_t chunk = MIN (65527, len - ofs);
+ uint32_t bdw = (chunk + 8) << 16;
+ int scc = (ofs == 0 && chunk == len ? 0
+ : ofs == 0 ? 1
+ : ofs + chunk == len ? 2
+ : 3);
+ uint32_t rdw = ((chunk + 4) << 16) | (scc << 8);
+
+ integer_convert (INTEGER_NATIVE, &bdw, INTEGER_MSB_FIRST, &bdw,
+ sizeof bdw);
+ integer_convert (INTEGER_NATIVE, &rdw, INTEGER_MSB_FIRST, &rdw,
+ sizeof rdw);
+ fwrite (&bdw, 1, sizeof bdw, w->file);
+ fwrite (&rdw, 1, sizeof rdw, w->file);
+ fwrite (rec + ofs, 1, chunk, w->file);
+ ofs += chunk;
+ }
+ }
+ break;
+
default:
NOT_REACHED ();
}
@@ -128,28 +174,34 @@ dfm_put_record (struct dfm_writer *w, const char *rec, size_t len)
bool
dfm_close_writer (struct dfm_writer *w)
{
- char *file_name;
bool ok;
if (w == NULL)
return true;
- file_name = xstrdup (fh_get_name (w->fh));
- if (fh_close (w->fh, "data file", "ws"))
- {
- free (file_name);
- return true;
- }
+ if (fh_unlock (w->lock))
+ return true;
ok = true;
if (w->file != NULL)
{
+ const char *file_name = fh_get_file_name (w->fh);
ok = !dfm_write_error (w) && !fn_close (file_name, w->file);
if (!ok)
msg (ME, _("I/O error occurred writing data file \"%s\"."), file_name);
+
+ if (ok ? !replace_file_commit (w->rf) : !replace_file_abort (w->rf))
+ ok = false;
}
+ fh_unref (w->fh);
free (w);
- free (file_name);
return ok;
}
+
+/* Returns the legacy character encoding of data written to WRITER. */
+enum legacy_encoding
+dfm_writer_get_legacy_encoding (const struct dfm_writer *writer)
+{
+ return fh_get_legacy_encoding (writer->fh);
+}