src/binlibs/pluto/lzio.c

   1 /*
   2 ** $Id: lzio.c,v 1.31.1.1 2007/12/27 13:02:25 roberto Exp $
   3 ** a generic input stream interface
   4 ** See Copyright Notice in lua.h
   5 */
   6
   7
   8 #include <string.h>
   9
  10 #define lzio_c
  11 #define LUA_CORE
  12
  13 #include "pdep/pdep.h"
  14
  15 int pdep_fill (ZIO *z) {
  16   size_t size;
  17   lua_State *L = z->L;
  18   const char *buff;
  19   lua_unlock(L);
  20   buff = z->reader(L, z->data, &size);
  21   lua_lock(L);
  22   if (buff == NULL || size == 0) return EOZ;
  23   z->n = size - 1;
  24   z->p = buff;
  25   return char2int(*(z->p++));
  26 }
  27
  28
  29 int pdep_lookahead (ZIO *z) {
  30   if (z->n == 0) {
  31     if (pdep_fill(z) == EOZ)
  32       return EOZ;
  33     else {
  34       z->n++;  /* pdep_fill removed first byte; put back it */
  35       z->p--;
  36     }
  37   }
  38   return char2int(*z->p);
  39 }
  40
  41
  42 void pdep_init (lua_State *L, ZIO *z, lua_Reader reader, void *data) {
  43   z->L = L;
  44   z->reader = reader;
  45   z->data = data;
  46   z->n = 0;
  47   z->p = NULL;
  48 }
  49
  50
  51 /* --------------------------------------------------------------- read --- */
  52 size_t pdep_read (ZIO *z, void *b, size_t n) {
  53   while (n) {
  54     size_t m;
  55     if (pdep_lookahead(z) == EOZ)
  56       return n;  /* return number of missing bytes */
  57     m = (n <= z->n) ? n : z->n;  /* min. between n and z->n */
  58     memcpy(b, z->p, m);
  59     z->n -= m;
  60     z->p += m;
  61     b = (char *)b + m;
  62     n -= m;
  63   }
  64   return 0;
  65 }
  66
  67 /* ------------------------------------------------------------------------ */
  68 char *pdep_openspace (lua_State *L, Mbuffer *buff, size_t n) {
  69   if (n > buff->buffsize) {
  70     if (n < LUA_MINBUFFER) n = LUA_MINBUFFER;
  71     pdep_resizebuffer(L, buff, n);
  72   }
  73   return buff->buffer;
  74 }
  75
  76