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 vfgetc(VFILE* vfp)
 102 {
 103         int c;
 104         if ((c = fgetc(vfp->fp)) == EOF)
 105         {
 106                 PERR("vfgetc: fgetc == EOF!");
 107                 if (vfp->lastvol) return EOF; /* último volumen. */
 108                 if (vfvol_close(vfp)) return EOF; /* error al cerrar. */
 109                 if (vfvol_open_next(vfp)) return EOF; /* error al abrir. */
 110                 c = fgetc(vfp->fp); /* obtengo caracter */
 111         }
 112         /* vfp->currpos++; */
 113         return c;
 114 }
 115
 116 int vfputc(int c, VFILE* vfp)
 117 {
 118         /* Si no es multivolumen o hay lugar, agrego y salgo. */
 119         if (!vfp->volsize || vfp->room--) return fputc(c, vfp->fp);
 120         PERR("vfputc: Necesito otro volumen!\n");
 121         /* Si no hay lugar, abro otro volumen. */
 122         if (vfvol_close(vfp)) return EOF; /* error al cerrar. */
 123         if (vfvol_open_next(vfp)) return EOF; /* error al abrir. */
 124         vfp->room--; /* resto de nuevo el espacio porque al abrirlo lo resetea. */
 125         return fputc(c, vfp->fp);
 126 }
 127
 128 size_t vfread(void* ptr, size_t size, size_t nmemb, VFILE* vfp)
 129 {
 130         int i = 0;
 131         size_t total = size * nmemb;
 132         /* leo uno a uno y si hay error salgo. */
 133         while (i < total && (((char*)ptr)[i++] = vfgetc(vfp)) != EOF);
 134         return i % size;
 135 }
 136
 137 size_t vfwrite(const void *ptr, size_t size, size_t nmemb, VFILE* vfp)
 138 {
 139         int i = 0;
 140         size_t total = size * nmemb;
 141         /* escribo uno a uno y si hay error salgo. */
 142         while (i < total && (vfputc(((char*)ptr)[i++], vfp)) != EOF);
 143         return i % size;
 144 }
 145
 146 int vfvol_close(VFILE* vfp)
 147 {
 148         /* Si es de escritura y el último guardo la cabecera. */
 149         if (vfp->mode == VFWRITE && vfp->lastvol)
 150         {
 151                 int ret;
 152                 PERR("vfvol_close: modo == VFWRITE");
 153                 /* Me posiciono al principio del archivo. */
 154                 if ((ret = fseek(vfp->fp, 0l, SEEK_SET)))
 155                 {
 156                         PERR("vfvol_close: fseek error");
 157                         return ret; /* fseek error. */
 158                 }
 159                 /* Guardo cabecera para indicar si es el último volumen o no. */
 160                 if ((ret = fputc(vfp->lastvol, vfp->fp)) == EOF)
 161                 {
 162                         PERR("vfvol_close: fputc error");
 163                         return ret; /* fputc error. */
 164                 }
 165         }
 166         return fclose(vfp->fp);
 167 }
 168
 169 int vfvol_open_next(VFILE* vfp)
 170 {
 171         char* volname = 0;
 172         char* name    = vfp->name; /* si es el archivo principal uso el nombre. */
 173         PERR("vfvol_open_next: ping");
 174         if (vfp->currvol >= 0) /* si no es el archivo principal. */
 175         {
 176                 PERR("vfvol_open_next: No es el archivo principal");
 177                 /* 1 para el \0 y sizeof(int) * 3 para el número de volumen. */
 178                 volname = malloc(strlen(vfp->name) + 1 + sizeof(int) * 3);
 179                 if (!volname) return 1; /* no hay más memoria */
 180                 /* Construyo el nombre del archivo. */
 181                 sprintf(volname, VFNAMETEMPLATE, vfp->name, vfp->currvol + 1);
 182                 name = volname; /* uso este nuevo nombre. */
 183         }
 184         /* Abro dependiendo del modo. */
 185         if (!(vfp->fp = fopen(name, (vfp->mode == VFREAD) ? "r" : "w")))
 186         {
 187                 PERR("vfvol_open_next: error al abrir!");
 188                 if (volname) free(volname);
 189                 return 2; /* error al abrir. */
 190         }
 191         vfp->currvol++;
 192         if (volname) free(volname);
 193         /* Si es para lectura, me fijo si es el últio a leer. */
 194         if (vfp->mode == VFREAD) vfp->lastvol = fgetc(vfp->fp);
 195         /* Si es para escritura, guardo header por default (hay más volúmenes). */
 196         if (vfp->mode == VFWRITE)
 197         {
 198                 vfp->room = vfp->volsize; /* tengo todo el espacio disponible. */
 199                 return fputc(vfp->lastvol, vfp->fp) == EOF;
 200         }
 201         return 0;
 202 }
 203