pintos-os.org Git - pspp/blob - src/dfm-write.c

   1 /* PSPP - computes sample statistics.
   2    Copyright (C) 1997-2004 Free Software Foundation, Inc.
   3    Written by Ben Pfaff <blp@gnu.org>.
   4
   5    This program is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU General Public License as
   7    published by the Free Software Foundation; either version 2 of the
   8    License, or (at your option) any later version.
   9
  10    This program is distributed in the hope that it will be useful, but
  11    WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13    General Public License for more details.
  14
  15    You should have received a copy of the GNU General Public License
  16    along with this program; if not, write to the Free Software
  17    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  18    02110-1301, USA. */
  19
  20 #include <config.h>
  21 #include "dfm-write.h"
  22 #include <assert.h>
  23 #include <errno.h>
  24 #include <stdlib.h>
  25 #include "alloc.h"
  26 #include "error.h"
  27 #include "file-handle.h"
  28 #include "filename.h"
  29 #include "str.h"
  30
  31 #include "gettext.h"
  32 #define _(msgid) gettext (msgid)
  33
  34 /* Data file writer. */
  35 struct dfm_writer
  36   {
  37     struct file_handle *fh;     /* File handle. */
  38     struct file_ext file;       /* Associated file. */
  39     char *bounce;               /* Bounce buffer for fixed-size fields. */
  40   };
  41
  42 /* Opens a file handle for writing as a data file. */
  43 struct dfm_writer *
  44 dfm_open_writer (struct file_handle *fh)
  45 {
  46   struct dfm_writer *w;
  47   void **aux;
  48
  49   aux = fh_open (fh, "data file", "ws");
  50   if (aux == NULL)
  51     return NULL;
  52   if (*aux != NULL)
  53     return *aux;
  54
  55   w = *aux = xmalloc (sizeof *w);
  56   w->fh = fh;
  57   w->file.file = NULL;
  58   w->bounce = NULL;
  59
  60   w->file.filename = xstrdup (handle_get_filename (w->fh));
  61   w->file.mode = "wb";
  62   w->file.file = NULL;
  63   w->file.sequence_no = NULL;
  64   w->file.param = NULL;
  65   w->file.postopen = NULL;
  66   w->file.preclose = NULL;
  67
  68   if (!fn_open_ext (&w->file))
  69     {
  70       msg (ME, _("An error occurred while opening \"%s\" for writing "
  71                  "as a data file: %s."),
  72            handle_get_filename (w->fh), strerror (errno));
  73       goto error;
  74     }
  75
  76   return w;
  77
  78  error:
  79   err_cond_fail ();
  80   dfm_close_writer (w);
  81   return NULL;
  82 }
  83
  84 /* Writes record REC having length LEN to the file corresponding to
  85    HANDLE.  REC is not null-terminated.  Returns nonzero on success,
  86    zero on failure. */
  87 int
  88 dfm_put_record (struct dfm_writer *w, const char *rec, size_t len)
  89 {
  90   assert (w != NULL);
  91
  92   if (handle_get_mode (w->fh) == MODE_BINARY
  93       && len < handle_get_record_width (w->fh))
  94     {
  95       size_t rec_width = handle_get_record_width (w->fh);
  96       if (w->bounce == NULL)
  97         w->bounce = xmalloc (rec_width);
  98       memcpy (w->bounce, rec, len);
  99       memset (&w->bounce[len], 0, rec_width - len);
 100       rec = w->bounce;
 101       len = rec_width;
 102     }
 103
 104   if (fwrite (rec, len, 1, w->file.file) != 1)
 105     {
 106       msg (ME, _("Error writing file %s: %s."),
 107            handle_get_name (w->fh), strerror (errno));
 108       err_cond_fail ();
 109       return 0;
 110     }
 111
 112   return 1;
 113 }
 114
 115 /* Closes data file writer W. */
 116 void
 117 dfm_close_writer (struct dfm_writer *w)
 118 {
 119   if (fh_close (w->fh, "data file", "ws"))
 120     return;
 121
 122   if (w->file.file)
 123     {
 124       fn_close_ext (&w->file);
 125       free (w->file.filename);
 126       w->file.filename = NULL;
 127     }
 128   free (w->bounce);
 129   free (w);
 130 }