pintos-os.org Git - pspp/blob - src/libpspp/zip-writer.c

   1 /* PSPP - a program for statistical analysis.
   2    Copyright (C) 2010, 2012 Free Software Foundation, Inc.
   3
   4    This program is free software: you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation, either version 3 of the License, or
   7    (at your option) any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program.  If not, see <http://www.gnu.org/licenses/>. */
  16
  17 #include <config.h>
  18
  19 #include "libpspp/zip-writer.h"
  20 #include "libpspp/zip-private.h"
  21
  22 #include <byteswap.h>
  23 #include <errno.h>
  24 #include <stdlib.h>
  25 #include <time.h>
  26
  27 #include "gl/crc.h"
  28 #include "gl/error.h"
  29 #include "gl/fwriteerror.h"
  30 #include "gl/xalloc.h"
  31
  32 #include "gettext.h"
  33 #define _(msgid) gettext (msgid)
  34
  35 struct zip_writer
  36   {
  37     char *file_name;            /* File name, for use in error mesages. */
  38     FILE *file;                 /* Output stream. */
  39
  40     uint16_t date, time;        /* Date and time in MS-DOS format. */
  41
  42     /* Members already added to the file, so that we can summarize them to the
  43        central directory at the end of the ZIP file. */
  44     struct zip_member *members;
  45     size_t n_members, allocated_members;
  46   };
  47
  48 struct zip_member
  49   {
  50     uint32_t offset;            /* Starting offset in file. */
  51     uint32_t size;              /* Length of member file data, in bytes. */
  52     uint32_t crc;               /* CRC-32 of member file data.. */
  53     char *name;                 /* Name of member file. */
  54   };
  55
  56 static void
  57 put_bytes (struct zip_writer *zw, const void *p, size_t n)
  58 {
  59   fwrite (p, 1, n, zw->file);
  60 }
  61
  62 static void
  63 put_u16 (struct zip_writer *zw, uint16_t x)
  64 {
  65 #ifdef WORDS_BIGENDIAN
  66   x = bswap_16 (x);
  67 #endif
  68   put_bytes (zw, &x, sizeof x);
  69 }
  70
  71 static void
  72 put_u32 (struct zip_writer *zw, uint32_t x)
  73 {
  74 #ifdef WORDS_BIGENDIAN
  75   x = bswap_32 (x);
  76 #endif
  77   put_bytes (zw, &x, sizeof x);
  78 }
  79
  80 /* Starts writing a new ZIP file named FILE_NAME.  Returns a new zip_writer if
  81    successful, otherwise a null pointer. */
  82 struct zip_writer *
  83 zip_writer_create (const char *file_name)
  84 {
  85   struct zip_writer *zw;
  86   struct tm *tm;
  87   time_t now;
  88   FILE *file;
  89
  90   file = fopen (file_name, "wb");
  91   if (file == NULL)
  92     {
  93       error (0, errno, _("%s: error opening output file"), file_name);
  94       return NULL;
  95     }
  96
  97   zw = xmalloc (sizeof *zw);
  98   zw->file_name = xstrdup (file_name);
  99   zw->file = file;
 100
 101   now = time (NULL);
 102   tm = localtime (&now);
 103   zw->date = tm->tm_mday + ((tm->tm_mon + 1) << 5) + ((tm->tm_year - 80) << 9);
 104   zw->time = tm->tm_sec / 2 + (tm->tm_min << 5) + (tm->tm_hour << 11);
 105
 106   zw->members = NULL;
 107   zw->n_members = 0;
 108   zw->allocated_members = 0;
 109
 110   return zw;
 111 }
 112
 113 /* Adds the contents of FILE, with name MEMBER_NAME, to ZW. */
 114 void
 115 zip_writer_add (struct zip_writer *zw, FILE *file, const char *member_name)
 116 {
 117   struct zip_member *member;
 118   uint32_t offset, size;
 119   size_t bytes_read;
 120   uint32_t crc;
 121   char buf[4096];
 122
 123   /* Local file header. */
 124   offset = ftello (zw->file);
 125   put_u32 (zw, MAGIC_LHDR);     /* local file header signature */
 126   put_u16 (zw, 10);             /* version needed to extract */
 127   put_u16 (zw, 1 << 3);         /* general purpose bit flag */
 128   put_u16 (zw, 0);              /* compression method */
 129   put_u16 (zw, zw->time);       /* last mod file time */
 130   put_u16 (zw, zw->date);       /* last mod file date */
 131   put_u32 (zw, 0);              /* crc-32 */
 132   put_u32 (zw, 0);              /* compressed size */
 133   put_u32 (zw, 0);              /* uncompressed size */
 134   put_u16 (zw, strlen (member_name)); /* file name length */
 135   put_u16 (zw, 0);                    /* extra field length */
 136   put_bytes (zw, member_name, strlen (member_name));
 137
 138   /* File data. */
 139   size = crc = 0;
 140   fseeko (file, 0, SEEK_SET);
 141   while ((bytes_read = fread (buf, 1, sizeof buf, file)) > 0)
 142     {
 143       put_bytes (zw, buf, bytes_read);
 144       size += bytes_read;
 145       crc = crc32_update (crc, buf, bytes_read);
 146     }
 147
 148   /* Data descriptor. */
 149   put_u32 (zw, MAGIC_DDHD);
 150   put_u32 (zw, crc);
 151   put_u32 (zw, size);
 152   put_u32 (zw, size);
 153
 154   /* Add to set of members. */
 155   if (zw->n_members >= zw->allocated_members)
 156     zw->members = x2nrealloc (zw->members, &zw->allocated_members,
 157                               sizeof *zw->members);
 158   member = &zw->members[zw->n_members++];
 159   member->offset = offset;
 160   member->size = size;
 161   member->crc = crc;
 162   member->name = xstrdup (member_name);
 163 }
 164
 165 /* Finalizes the contents of ZW and closes it.  Returns true if successful,
 166    false if a write error occurred while finalizing the file or at any earlier
 167    time. */
 168 bool
 169 zip_writer_close (struct zip_writer *zw)
 170 {
 171   uint32_t dir_start, dir_end;
 172   size_t i;
 173   bool ok;
 174
 175   if (zw == NULL)
 176     return true;
 177
 178   dir_start = ftello (zw->file);
 179   for (i = 0; i < zw->n_members; i++)
 180     {
 181       struct zip_member *m = &zw->members[i];
 182
 183       /* Central directory file header. */
 184       put_u32 (zw, MAGIC_SOCD);       /* central file header signature */
 185       put_u16 (zw, 63);               /* version made by */
 186       put_u16 (zw, 10);               /* version needed to extract */
 187       put_u16 (zw, 1 << 3);           /* general purpose bit flag */
 188       put_u16 (zw, 0);                /* compression method */
 189       put_u16 (zw, zw->time);         /* last mod file time */
 190       put_u16 (zw, zw->date);         /* last mod file date */
 191       put_u32 (zw, m->crc);           /* crc-32 */
 192       put_u32 (zw, m->size);          /* compressed size */
 193       put_u32 (zw, m->size);          /* uncompressed size */
 194       put_u16 (zw, strlen (m->name)); /* file name length */
 195       put_u16 (zw, 0);                /* extra field length */
 196       put_u16 (zw, 0);                /* file comment length */
 197       put_u16 (zw, 0);                /* disk number start */
 198       put_u16 (zw, 0);                /* internal file attributes */
 199       put_u32 (zw, 0);                /* external file attributes */
 200       put_u32 (zw, m->offset);        /* relative offset of local header */
 201       put_bytes (zw, m->name, strlen (m->name));
 202       free (m->name);
 203     }
 204   free (zw->members);
 205   dir_end = ftello (zw->file);
 206
 207   /* End of central directory record. */
 208   put_u32 (zw, MAGIC_EOCD);     /* end of central dir signature */
 209   put_u16 (zw, 0);              /* number of this disk */
 210   put_u16 (zw, 0);              /* number of the disk with the
 211                                    start of the central directory */
 212   put_u16 (zw, zw->n_members);  /* total number of entries in the
 213                                    central directory on this disk */
 214   put_u16 (zw, zw->n_members);  /* total number of entries in
 215                                    the central directory */
 216   put_u32 (zw, dir_end - dir_start); /* size of the central directory */
 217   put_u32 (zw, dir_start);      /* offset of start of central
 218                                    directory with respect to
 219                                    the starting disk number */
 220   put_u16 (zw, 0);              /* .ZIP file comment length */
 221
 222   if (!fwriteerror (zw->file))
 223     ok = true;
 224   else
 225     {
 226       error (0, errno, _("%s: write failed"), zw->file_name);
 227       ok = false;
 228     }
 229
 230   free (zw->file_name);
 231   free (zw);
 232
 233   return ok;
 234 }