175 lines
2.8 KiB
C
175 lines
2.8 KiB
C
#include "fitz.h"
|
|
|
|
fz_stream *
|
|
fz_new_stream(void *state,
|
|
int(*read)(fz_stream *stm, unsigned char *buf, int len),
|
|
void(*close)(fz_stream *stm))
|
|
{
|
|
fz_stream *stm;
|
|
|
|
stm = fz_malloc(sizeof(fz_stream));
|
|
|
|
stm->refs = 1;
|
|
stm->error = 0;
|
|
stm->eof = 0;
|
|
stm->pos = 0;
|
|
|
|
stm->bits = 0;
|
|
stm->avail = 0;
|
|
|
|
stm->bp = stm->buf;
|
|
stm->rp = stm->bp;
|
|
stm->wp = stm->bp;
|
|
stm->ep = stm->buf + sizeof stm->buf;
|
|
|
|
stm->state = state;
|
|
stm->read = read;
|
|
stm->close = close;
|
|
stm->seek = NULL;
|
|
|
|
return stm;
|
|
}
|
|
|
|
fz_stream *
|
|
fz_keep_stream(fz_stream *stm)
|
|
{
|
|
stm->refs ++;
|
|
return stm;
|
|
}
|
|
|
|
void
|
|
fz_close(fz_stream *stm)
|
|
{
|
|
stm->refs --;
|
|
if (stm->refs == 0)
|
|
{
|
|
if (stm->close)
|
|
stm->close(stm);
|
|
fz_free(stm);
|
|
}
|
|
}
|
|
|
|
/* File stream */
|
|
|
|
static int read_file(fz_stream *stm, unsigned char *buf, int len)
|
|
{
|
|
int n = read(*(int*)stm->state, buf, len);
|
|
if (n < 0)
|
|
return fz_throw("read error: %s", strerror(errno));
|
|
return n;
|
|
}
|
|
|
|
static void seek_file(fz_stream *stm, int offset, int whence)
|
|
{
|
|
int n = lseek(*(int*)stm->state, offset, whence);
|
|
if (n < 0)
|
|
fz_warn("cannot lseek: %s", strerror(errno));
|
|
stm->pos = n;
|
|
stm->rp = stm->bp;
|
|
stm->wp = stm->bp;
|
|
}
|
|
|
|
static void close_file(fz_stream *stm)
|
|
{
|
|
int n = close(*(int*)stm->state);
|
|
if (n < 0)
|
|
fz_warn("close error: %s", strerror(errno));
|
|
fz_free(stm->state);
|
|
}
|
|
|
|
fz_stream *
|
|
fz_open_fd(int fd)
|
|
{
|
|
fz_stream *stm;
|
|
int *state;
|
|
|
|
state = fz_malloc(sizeof(int));
|
|
*state = fd;
|
|
|
|
stm = fz_new_stream(state, read_file, close_file);
|
|
stm->seek = seek_file;
|
|
|
|
return stm;
|
|
}
|
|
|
|
fz_stream *
|
|
fz_open_file(const char *name)
|
|
{
|
|
int fd = open(name, O_BINARY | O_RDONLY, 0);
|
|
if (fd == -1)
|
|
return NULL;
|
|
return fz_open_fd(fd);
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
fz_stream *
|
|
fz_open_file_w(const wchar_t *name)
|
|
{
|
|
int fd = _wopen(name, O_BINARY | O_RDONLY, 0);
|
|
if (fd == -1)
|
|
return NULL;
|
|
return fz_open_fd(fd);
|
|
}
|
|
#endif
|
|
|
|
/* Memory stream */
|
|
|
|
static int read_buffer(fz_stream *stm, unsigned char *buf, int len)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
static void seek_buffer(fz_stream *stm, int offset, int whence)
|
|
{
|
|
if (whence == 0)
|
|
stm->rp = stm->bp + offset;
|
|
if (whence == 1)
|
|
stm->rp += offset;
|
|
if (whence == 2)
|
|
stm->rp = stm->ep - offset;
|
|
stm->rp = CLAMP(stm->rp, stm->bp, stm->ep);
|
|
stm->wp = stm->ep;
|
|
}
|
|
|
|
static void close_buffer(fz_stream *stm)
|
|
{
|
|
if (stm->state)
|
|
fz_drop_buffer(stm->state);
|
|
}
|
|
|
|
fz_stream *
|
|
fz_open_buffer(fz_buffer *buf)
|
|
{
|
|
fz_stream *stm;
|
|
|
|
stm = fz_new_stream(fz_keep_buffer(buf), read_buffer, close_buffer);
|
|
stm->seek = seek_buffer;
|
|
|
|
stm->bp = buf->data;
|
|
stm->rp = buf->data;
|
|
stm->wp = buf->data + buf->len;
|
|
stm->ep = buf->data + buf->len;
|
|
|
|
stm->pos = buf->len;
|
|
|
|
return stm;
|
|
}
|
|
|
|
fz_stream *
|
|
fz_open_memory(unsigned char *data, int len)
|
|
{
|
|
fz_stream *stm;
|
|
|
|
stm = fz_new_stream(NULL, read_buffer, close_buffer);
|
|
stm->seek = seek_buffer;
|
|
|
|
stm->bp = data;
|
|
stm->rp = data;
|
|
stm->wp = data + len;
|
|
stm->ep = data + len;
|
|
|
|
stm->pos = len;
|
|
|
|
return stm;
|
|
}
|