src/vfile/vfile.c

   1 /* vim: set noexpandtab tabstop=4 shiftwidth=4 wrap:
   2  *----------------------------------------------------------------------------
   3  *                                  jacu
   4  *----------------------------------------------------------------------------
   5  * This file is part of jacu.
   6  *
   7  * jacu is free software; you can redistribute it and/or modify it under the
   8  * terms of the GNU General Public License as published by the Free Software
   9  * Foundation; either version 2 of the License, or (at your option) any later
  10  * version.
  11  *
  12  * jacu is distributed in the hope that it will be useful, but WITHOUT ANY
  13  * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  14  * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
  15  * details.
  16  *
  17  * You should have received a copy of the GNU General Public License along
  18  * with jacu; if not, write to the Free Software Foundation, Inc., 59 Temple
  19  * Place, Suite 330, Boston, MA  02111-1307  USA
  20  *----------------------------------------------------------------------------
  21  * Creado:  mié jun 16 14:04:55 ART 2004
  22  * Autores: Leandro Lucarella <llucare@fi.uba.ar>
  23  *----------------------------------------------------------------------------
  24  *
  25  * $Id$
  26  *
  27  */
  28
  29 #include "common.h"
  30 #include "vfile.h"
  31 #include <malloc.h>
  32 #include <string.h>
  33
  34 /** \file
  35  *
  36  * Archivo virtual divisible en volúmenes.
  37  *
  38  * Implementación de un archivo virtual divisible en volúmenes.
  39  * Provee algunas funciones compatibles con la stdio de ANSI C para utilizar
  40  * de forma transparente archivos virtuales divididos en varios volúmenes.
  41  *
  42  */
  43
  44 /** Abre un volumen. */
  45 int vfvol_open_next(VFILE* vfp);
  46
  47 /** Cierra un volumen. */
  48 int vfvol_close(VFILE* vfp);
  49
  50 VFILE* vfopen(const char* path, const char* mode, long volsize)
  51 {
  52         VFILE* vfp = malloc(sizeof(VFILE));
  53         if (!vfp) return 0; /* no hay más memoria */
  54         if (volsize < VFMINVOLSIZE) /* tamaño de volumen inválido. */
  55         {
  56                 PERR("vfopen: volsize = 0");
  57                 vfp->volsize = 0;
  58         }
  59         else /* tamaño de volumen válido. */
  60         {
  61                 vfp->volsize = --volsize; /* Uno menos para guardar cabecera. */
  62         }
  63          /* Chequeo que sea un modo correcto. */
  64         if (mode[0] == 'r' && mode[1] != '+')      vfp->mode = VFREAD;
  65         else if (mode[0] == 'w' && mode[1] != '+') vfp->mode = VFWRITE;
  66         else /* no es un modo válido. */
  67         {
  68                 PERR("vfopen: modo inválido!");
  69                 free(vfp);
  70                 return 0;
  71         }
  72         if (!(vfp->name = str_dup(path))) /* no hay más memoria */
  73         {
  74                 PERR("vfopen: no se pudo hacer str_dup!");
  75                 free(vfp);
  76                 return 0;
  77         }
  78         /* Abrimos primer archivo de los volumenes. */
  79         vfp->currvol = -1;
  80         vfp->lastvol = 0;
  81         if (vfvol_open_next(vfp)) /* no se pudo abrir el primer volumen. */
  82         {
  83                 PERR("vfopen: no se pudo abrir archivo inicial!");
  84                 free(vfp->name);
  85                 free(vfp);
  86                 return 0;
  87         }
  88         return vfp; /* todo ok. */
  89 }
  90
  91 int vfclose(VFILE* vfp)
  92 {
  93         int ret;
  94         vfp->lastvol = 1; /* es el último volumen. */
  95         ret = vfvol_close(vfp);
  96         free(vfp->name);
  97         free(vfp);
  98         return ret;
  99 }
 100
 101 int vfeof(VFILE* vfp)
 102 {
 103         return vfp->lastvol && feof(vfp->fp);
 104 }
 105
 106 int vfgetc(VFILE* vfp)
 107 {
 108         int c;
 109         if ((c = fgetc(vfp->fp)) == EOF)
 110         {
 111                 PERR("vfgetc: fgetc == EOF!");
 112                 if (vfp->lastvol) return EOF; /* último volumen. */
 113                 if (vfvol_close(vfp)) return EOF; /* error al cerrar. */
 114                 if (vfvol_open_next(vfp)) return EOF; /* error al abrir. */
 115                 c = fgetc(vfp->fp); /* obtengo caracter */
 116         }
 117         /* vfp->currpos++; */
 118         return c;
 119 }
 120
 121 int vfputc(int c, VFILE* vfp)
 122 {
 123         /* Si no es multivolumen o hay lugar, agrego y salgo. */
 124         if (!vfp->volsize || vfp->room--) return fputc(c, vfp->fp);
 125         PERR("vfputc: Necesito otro volumen!\n");
 126         /* Si no hay lugar, abro otro volumen. */
 127         if (vfvol_close(vfp)) return EOF; /* error al cerrar. */
 128         if (vfvol_open_next(vfp)) return EOF; /* error al abrir. */
 129         vfp->room--; /* resto de nuevo el espacio porque al abrirlo lo resetea. */
 130         return fputc(c, vfp->fp);
 131 }
 132
 133 size_t vfread(void* ptr, size_t size, size_t nmemb, VFILE* vfp)
 134 {
 135         int c;
 136         size_t i = 0;
 137         size_t total = size * nmemb;
 138         /* leo uno a uno y si hay error salgo. */
 139         while (i < total)
 140         {
 141                 if ((c = vfgetc(vfp)) == EOF)
 142                 {
 143                         PERR("vfread: EOF");
 144                         break;
 145                 }
 146                 else ((char*)ptr)[i++] = c;
 147         }
 148         return i / size;
 149 }
 150
 151 size_t vfwrite(const void *ptr, size_t size, size_t nmemb, VFILE* vfp)
 152 {
 153         size_t i = 0;
 154         size_t total = size * nmemb;
 155         /* escribo uno a uno y si hay error salgo. */
 156         while (i < total && (vfputc(((char*)ptr)[i++], vfp)) != EOF);
 157         return i / size;
 158 }
 159
 160 int vfvol_close(VFILE* vfp)
 161 {
 162         /* Si es de escritura y el último guardo la cabecera. */
 163         if (vfp->mode == VFWRITE && vfp->lastvol)
 164         {
 165                 int ret;
 166                 PERR("vfvol_close: modo == VFWRITE");
 167                 /* Me posiciono al principio del archivo. */
 168                 if ((ret = fseek(vfp->fp, 0l, SEEK_SET)))
 169                 {
 170                         PERR("vfvol_close: fseek error");
 171                         return ret; /* fseek error. */
 172                 }
 173                 /* Guardo cabecera para indicar si es el último volumen o no. */
 174                 if ((ret = fputc(vfp->lastvol, vfp->fp)) == EOF)
 175                 {
 176                         PERR("vfvol_close: fputc error");
 177                         return ret; /* fputc error. */
 178                 }
 179         }
 180         return fclose(vfp->fp);
 181 }
 182
 183 int vfvol_open_next(VFILE* vfp)
 184 {
 185         char* volname = 0;
 186         char* name    = vfp->name; /* si es el archivo principal uso el nombre. */
 187         PERR("vfvol_open_next: ping");
 188         if (vfp->currvol >= 0) /* si no es el archivo principal. */
 189         {
 190                 PERR("vfvol_open_next: No es el archivo principal");
 191                 /* 1 para el \0 y sizeof(int) * 3 para el número de volumen. */
 192                 volname = malloc(strlen(vfp->name) + 1 + sizeof(int) * 3);
 193                 if (!volname) return 1; /* no hay más memoria */
 194                 /* Construyo el nombre del archivo. */
 195                 sprintf(volname, VFNAMETEMPLATE, vfp->name, vfp->currvol + 1);
 196                 name = volname; /* uso este nuevo nombre. */
 197         }
 198         /* Abro dependiendo del modo. */
 199         if (!(vfp->fp = fopen(name, (vfp->mode == VFREAD) ? "r" : "w")))
 200         {
 201                 PERR("vfvol_open_next: error al abrir!");
 202                 if (volname) free(volname);
 203                 return 2; /* error al abrir. */
 204         }
 205         vfp->currvol++;
 206         if (volname) free(volname);
 207         /* Si es para lectura, me fijo si es el últio a leer. */
 208         if (vfp->mode == VFREAD) vfp->lastvol = fgetc(vfp->fp);
 209         /* Si es para escritura, guardo header por default (hay más volúmenes). */
 210         if (vfp->mode == VFWRITE)
 211         {
 212                 vfp->room = vfp->volsize; /* tengo todo el espacio disponible. */
 213                 return fputc(vfp->lastvol, vfp->fp) == EOF;
 214         }
 215         return 0;
 216 }
 217