src/fops.c

   1 /*
   2  *  2ooM: The Master of Orion II Reverse Engineering Project
   3  *  Default file operations structures for liblbx.
   4  *  Copyright © 2010, 2014 Nick Bowler
   5  *
   6  *  This program is free software: you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation, either version 3 of the License, or
   9  *  (at your option) any later version.
  10  *
  11  *  This program is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19 #include <stdio.h>
  20 #include <errno.h>
  21 #include <assert.h>
  22
  23 #include "misc.h"
  24 #include "error.h"
  25 #include "lbx.h"
  26
  27 /* Default I/O operations for ordinary files. */
  28 static size_t file_read(void *buf, size_t size, void *handle)
  29 {
  30         size_t rc = fread(buf, 1, size, (FILE *)handle);
  31
  32         if (rc < size && ferror((FILE *)handle))
  33                 lbx_error_raise(-errno);
  34         return rc;
  35 }
  36
  37 static int file_seek(void *handle, long offset, int whence)
  38 {
  39         if (fseek((FILE *)handle, offset, whence) == -1) {
  40                 lbx_error_raise(-errno);
  41                 return -1;
  42         }
  43
  44         return 0;
  45 }
  46
  47 static int file_eof(void *handle)
  48 {
  49         return feof((FILE *)handle);
  50 }
  51
  52 const struct lbx_file_ops lbx_default_fops = {
  53         .read = file_read,
  54         .seek = file_seek,
  55         .eof  = file_eof,
  56 };
  57
  58 /* Read function for pipes that tracks the number of bytes read. */
  59 static size_t pipe_read(void *buf, size_t size, void *handle)
  60 {
  61         struct lbx_pipe_state *state = handle;
  62         size_t rc;
  63
  64         rc = file_read(buf, size, state->f);
  65         state->offset += rc;
  66         return rc;
  67 }
  68
  69 /* Seek function for pipes that reads data into the void. */
  70 static int pipe_seek(void *handle, long offset, int whence)
  71 {
  72         struct lbx_pipe_state *state = handle;
  73         long distance;
  74
  75         switch (whence) {
  76         case SEEK_SET:
  77                 distance = offset - state->offset;
  78                 break;
  79         case SEEK_CUR:
  80                 distance = offset;
  81                 break;
  82         case SEEK_END:
  83                 distance = -1;
  84                 break;
  85         default:
  86                 assert(0);
  87         }
  88
  89         if (distance < 0)
  90                 return -1;
  91
  92         while (distance > 0) {
  93                 static unsigned char oblivion[1024];
  94                 size_t rc, amount = MIN(sizeof oblivion, distance);
  95
  96                 rc = pipe_read(oblivion, amount, handle);
  97                 distance -= rc;
  98
  99                 if (rc < amount)
 100                         return -1;
 101         }
 102
 103         clearerr(state->f);
 104         return 0;
 105 }
 106
 107 static int pipe_eof(void *handle)
 108 {
 109         struct lbx_pipe_state *state = handle;
 110
 111         return feof(state->f);
 112 }
 113
 114 const struct lbx_file_ops lbx_pipe_fops = {
 115         .read = pipe_read,
 116         .seek = pipe_seek,
 117         .eof  = pipe_eof,
 118 };
 119
 120 static size_t lbx_read(void *buf, size_t size, void *handle)
 121 {
 122         return lbx_file_read(handle, buf, size);
 123 }
 124
 125 static int lbx_seek(void *handle, long offset, int whence)
 126 {
 127         return lbx_file_seek(handle, offset, whence);
 128 }
 129
 130 static int lbx_eof(void *handle)
 131 {
 132         return lbx_file_eof(handle);
 133 }
 134
 135 const struct lbx_file_ops lbx_arch_fops = {
 136         .read = lbx_read,
 137         .seek = lbx_seek,
 138         .eof  = lbx_eof,
 139 };