bfile/mfile.c

   1 /*
   2  * Copyright (C) 1994, Index Data I/S
   3  * All rights reserved.
   4  * Sebastian Hammer, Adam Dickmeiss
   5  *
   6  * $Log: mfile.c,v $
   7  * Revision 1.1  1994-08-23 14:41:33  quinn
   8  * First functional version.
   9  *
  10  */
  11
  12 #include <sys/types.h>
  13 #include <fcntl.h>
  14 #include <unistd.h>
  15 #include <dirent.h>
  16 #include <stdlib.h>
  17 #include <assert.h>
  18
  19 #include <util.h>
  20 #include <mfile.h>
  21
  22 static MFile_area_struct *open_areas = 0;
  23 static MFile_area_struct *default_area = 0;
  24
  25 static int scan_areadef(MFile_area ma, const char *name)
  26 {
  27     const char *ad = res_get(common_resource, name);
  28     int offset = 0, rs, size, multi, rd;
  29     char dirname[FILENAME_MAX+1], unit;
  30     mf_dir **dp = &ma->dirs, *dir = *dp;
  31
  32     if (!ad)
  33     {
  34         log(LOG_FATAL, "Could not find resource '%s'", name);
  35         return -1;
  36     }
  37     for (;;)
  38     {
  39         rs = sscanf(ad + offset, "%[^:]:%d%c %n", dirname, &size, &unit, &rd);
  40         if (rs <= 1)
  41             break;
  42         if (rs != 3)
  43         {
  44             log(LOG_FATAL, "Illegal directory description: %s", ad + offset);
  45             return -1;
  46         }
  47         switch (unit)
  48         {
  49             case 'B': case 'b': multi = 1; break;
  50             case 'K': case 'k': multi = 1024; break;
  51             case 'M': case 'm': multi = 1048576; break;
  52             default:
  53                 log(LOG_FATAL, "Illegal unit: %c in %s", unit, ad + offset);
  54                 return -1;
  55         }
  56         *dp = dir = xmalloc(sizeof(mf_dir));
  57         dir->next = 0;
  58         strcpy(dir->name, dirname);
  59         dir->max_bytes = dir->avail_bytes = size * multi;
  60         dp = &dir->next;
  61         offset += rd;
  62     }
  63     return 0;
  64 }
  65
  66 static int file_position(MFile mf, int pos)
  67 {
  68     int off = 0, c = mf->cur_file, ps;
  69
  70     if ((c > 0 && pos <= mf->files[c-1].top) ||
  71         (c < mf->no_files -1 && pos > mf->files[c].top))
  72     {
  73         c = 0;
  74         while (mf->files[c].top >= 0 && mf->files[c].top < pos)
  75         {
  76             off += mf->files[c].blocks;
  77             c++;
  78         }
  79         assert(c < mf->no_files);
  80     }
  81     else
  82         off = c ? (mf->files[c-1].top + 1) : 0;
  83     if (mf->files[c].fd < 0 && (mf->files[c].fd = open(mf->files[c].path,
  84         mf->wr ? O_RDWR|O_CREAT : O_RDONLY, 0666)) < 0)
  85     {
  86         log(LOG_FATAL|LOG_ERRNO, "Failed to open %s", mf->files[c].path);
  87         return -1;
  88     }
  89     if (lseek(mf->files[c].fd, (ps = pos - off) * mf->blocksize, SEEK_SET) < 0)
  90     {
  91         log(LOG_FATAL|LOG_ERRNO, "Failed to seek in %s", mf->files[c].path);
  92         return -1;
  93     }
  94     mf->cur_file = c;
  95     return ps;
  96 }
  97
  98 static int cmp_part_file(const void *p1, const void *p2)
  99 {
 100     return ((part_file *)p1)->number - ((part_file *)p2)->number;
 101 }
 102
 103 /*
 104  * Create a new area, cotaining metafiles in directories.
 105  * Find the part-files in each directory, and inventory the existing metafiles.
 106  */
 107 MFile_area mf_init(const char *name)
 108 {
 109     MFile_area ma = xmalloc(sizeof(MFile_area_struct)), mp;
 110     mf_dir *dirp;
 111     meta_file *meta_f;
 112     part_file *part_f;
 113     DIR *dd;
 114     struct dirent *dent;
 115     int fd, number;
 116     char metaname[FILENAME_MAX+1], tmpnam[FILENAME_MAX+1];
 117
 118     for (mp = open_areas; mp; mp = mp->next)
 119         if (!strcmp(name, mp->name))
 120             abort();
 121     strcpy(ma->name, name);
 122     ma->next = open_areas;
 123     open_areas = ma;
 124     ma->mfiles = 0;
 125     ma->dirs = 0;
 126     if (scan_areadef(ma, name) < 0)
 127     {
 128         log(LOG_FATAL, "Failed to access description of '%s'", name);
 129         return 0;
 130     }
 131     /* look at each directory */
 132     for (dirp = ma->dirs; dirp; dirp = dirp->next)
 133     {
 134         if (!(dd = opendir(dirp->name)))
 135         {
 136             log(LOG_FATAL|LOG_ERRNO, "Failed to open %s", dirp->name);
 137             return 0;
 138         }
 139         /* look at each file */
 140         while ((dent = readdir(dd)))
 141         {
 142             if (*dent->d_name == '.')
 143                 continue;
 144             if (sscanf(dent->d_name, "%[^.].%d", metaname, &number) != 2)
 145             {
 146                 log(LOG_FATAL, "Failed to resolve part-name %s", dent->d_name);
 147                 return 0;
 148             }
 149             for (meta_f = ma->mfiles; meta_f; meta_f = meta_f->next)
 150             {
 151                 /* known metafile */
 152                 if (!strcmp(meta_f->name, metaname))
 153                 {
 154                     part_f = &meta_f->files[meta_f->no_files++];
 155                     break;
 156                 }
 157             }
 158             /* new metafile */
 159             if (!meta_f)
 160             {
 161                 meta_f = xmalloc(sizeof(*meta_f));
 162                 meta_f->ma = ma;
 163                 meta_f->next = ma->mfiles;
 164                 meta_f->open = 0;
 165                 meta_f->cur_file = -1;
 166                 ma->mfiles = meta_f;
 167                 strcpy(meta_f->name, metaname);
 168                 part_f = &meta_f->files[0];
 169                 meta_f->no_files = 1;
 170             }
 171             part_f->number = number;
 172             part_f->dir = dirp;
 173             part_f->fd = -1;
 174             sprintf(tmpnam, "%s/%s", dirp->name, dent->d_name);
 175             part_f->path = xstrdup(tmpnam);
 176             /* get size */
 177             if ((fd = open(part_f->path, O_RDONLY)) < 0)
 178             {
 179                 log(LOG_FATAL|LOG_ERRNO, "Failed to access %s", dent->d_name);
 180                 return 0;
 181             }
 182             if ((part_f->bytes = lseek(fd, 0, SEEK_END)) < 0)
 183             {
 184                 log(LOG_FATAL|LOG_ERRNO, "Failed to seek in %s", dent->d_name);
 185                 return 0;
 186             }
 187             close(fd);
 188             dirp->avail_bytes -= part_f->bytes;
 189         }
 190         closedir(dd);
 191     }
 192     for (meta_f = ma->mfiles; meta_f; meta_f = meta_f->next)
 193     {
 194         log(LOG_DEBUG, "mf_init: %s consists of %d part(s)", meta_f->name,
 195             meta_f->no_files);
 196         qsort(meta_f->files, meta_f->no_files, sizeof(part_file),
 197             cmp_part_file);
 198     }
 199     return ma;
 200 }
 201
 202 /*
 203  * Open a metafile.
 204  * If !ma, Use MF_DEFAULT_AREA.
 205  */
 206 MFile mf_open(MFile_area ma, const char *name, int block_size, int wflag)
 207 {
 208     struct meta_file *new;
 209     int i;
 210     char tmp[FILENAME_MAX+1];
 211     mf_dir *dp;
 212
 213     if (!ma)
 214     {
 215         if (!default_area && !(default_area = mf_init(MF_DEFAULT_AREA)))
 216         {
 217             log(LOG_FATAL, "Failed to open default area.");
 218             return 0;
 219         }
 220         ma = default_area;
 221     }
 222     for (new = ma->mfiles; new; new = new->next)
 223         if (!strcmp(name, new->name))
 224             if (new->open)
 225                 abort();
 226             else
 227                 break;
 228     if (!new)
 229     {
 230         new = xmalloc(sizeof(*new));
 231         strcpy(new->name, name);
 232         /* allocate one, empty file */
 233         new->no_files = 1;
 234         new->files[0].bytes = new->files[0].blocks = 0;
 235         new->files[0].top = -1;
 236         new->files[0].number = 0;
 237         new->files[0].fd = -1;
 238         new->min_bytes_creat = MF_MIN_BLOCKS_CREAT * block_size;
 239         for (dp = ma->dirs; dp && dp->avail_bytes < new->min_bytes_creat;
 240             dp = dp->next);
 241         if (!dp)
 242         {
 243             log(LOG_FATAL, "Insufficient space for new mfile.");
 244             return 0;
 245         }
 246         new->files[0].dir = dp;
 247         sprintf(tmp, "%s/%s.%d", dp->name, new->name, 0);
 248         new->files[0].path = xstrdup(tmp);
 249         new->ma = ma;
 250     }
 251     new->blocksize = block_size;
 252     new->min_bytes_creat = MF_MIN_BLOCKS_CREAT * block_size;
 253     new->wr=wflag;
 254     new->cur_file = 0;
 255
 256     for (i = 0; i < new->no_files; i++)
 257     {
 258         new->files[i].blocks = new->files[i].bytes / new->blocksize;
 259         if (i == new->no_files)
 260             new->files[i].top = -1;
 261         else
 262             new->files[i].top = i ? (new->files[i-1].top + new->files[i].blocks)
 263                 : (new->files[i].blocks - 1);
 264     }
 265     return new;
 266 }
 267
 268 /*
 269  * Close a metafile.
 270  */
 271 int mf_close(MFile mf)
 272 {
 273     abort();
 274     return 0;
 275 }
 276
 277 /*
 278  * Read one block from a metafile. Interface mirrors bfile.
 279  */
 280 int mf_read(MFile mf, int no, int offset, int num, void *buf)
 281 {
 282     if (file_position(mf, no) < 0)
 283         exit(1);
 284     if (read(mf->files[mf->cur_file].fd, buf, mf->blocksize) < mf->blocksize)
 285     {
 286         log(LOG_FATAL|LOG_ERRNO, "Read failed");
 287         exit(1);
 288     }
 289     return 0;
 290 }
 291
 292 /*
 293  * Write.
 294  */
 295 int mf_write(MFile mf, int no, int offset, int num, const void *buf)
 296 {
 297     int ps;
 298     mf_dir *dp;
 299     char tmp[FILENAME_MAX+1];
 300
 301     if ((ps = file_position(mf, no)) < 0)
 302         exit(1);
 303     assert(ps <= mf->files[mf->cur_file].blocks);
 304     /* extend table */
 305     if (ps == mf->files[mf->cur_file].blocks)
 306     {
 307         assert(mf->cur_file == mf->no_files - 1);
 308         /* create new file */
 309         if (mf->files[mf->cur_file].dir->avail_bytes <
 310             mf->blocksize)
 311         {
 312             log(LOG_DEBUG, "Creating new file.");
 313             for (dp = mf->ma->dirs; dp && dp->avail_bytes < mf->min_bytes_creat;
 314                 dp = dp->next);
 315             if (!dp)
 316             {
 317                 log(LOG_FATAL, "Cannot allocate more space for %s", mf->name);
 318                 exit(1);
 319             }
 320             mf->files[mf->cur_file].top = no - 1;
 321             mf->files[++(mf->cur_file)].top = -1;
 322             mf->files[mf->cur_file].dir = dp;
 323             mf->files[mf->cur_file].number =
 324                 mf->files[mf->cur_file-1].number + 1;
 325             mf->files[mf->cur_file].blocks =
 326                 mf->files[mf->cur_file].bytes = 0;
 327             mf->files[mf->cur_file].fd = -1;
 328             sprintf(tmp, "%s/%s.%d", dp->name, mf->name,
 329                 mf->files[mf->cur_file].number);
 330             mf->files[mf->cur_file].path = xstrdup(tmp);
 331             mf->no_files++;
 332             /* open new file and position at beginning */
 333             if (file_position(mf, no) < 0)
 334                 exit(1);
 335         }
 336         mf->files[mf->cur_file].blocks++;
 337         mf->files[mf->cur_file].bytes += mf->blocksize;
 338         mf->files[mf->cur_file].dir->avail_bytes -= mf->blocksize;
 339     }
 340     if (write(mf->files[mf->cur_file].fd, buf, mf->blocksize) < mf->blocksize)
 341     {
 342         log(LOG_FATAL|LOG_ERRNO, "Write failed");
 343         exit(1);
 344     }
 345     return 0;
 346 }
 347
 348 /*
 349  * Destroy a metafile, unlinking component files. File must be open.
 350  */
 351 int mf_unlink(MFile mf)
 352 {
 353     abort();
 354     return 0;
 355 }
 356
 357 /*
 358  * Unlink the file by name, rather than MFile-handle. File should be closed.
 359  */
 360 int mf_unlink_name(MFile_area ma, const char *name)
 361 {
 362     abort();
 363     return 0;
 364 }