/* kc3
* Copyright 2022,2023,2024 kmx.io <contact@kmx.io>
*
* Permission is hereby granted to use this software granted the above
* copyright notice and this permission paragraph are included in all
* copies and substantial portions of this software.
*
* THIS SOFTWARE IS PROVIDED "AS-IS" WITHOUT ANY GUARANTEE OF
* PURPOSE AND PERFORMANCE. IN NO EVENT WHATSOEVER SHALL THE
* AUTHOR BE CONSIDERED LIABLE FOR THE USE AND PERFORMANCE OF
* THIS SOFTWARE.
*/
#include "alloc.h"
#include "assert.h"
#include <errno.h>
#include <string.h>
#include <sys/ioctl.h>
#include <unistd.h>
#include "buf.h"
#include "buf_fd.h"
#include "buf_save.h"
sw buf_fd_open_r_refill (s_buf *buf);
sw buf_fd_open_w_flush (s_buf *buf);
sw buf_fd_open_w_seek (s_buf *buf, sw offset, u8 whence);
void buf_fd_close (s_buf *buf)
{
assert(buf);
buf_flush(buf);
buf->flush = NULL;
buf->refill = NULL;
free(buf->user_ptr);
buf->user_ptr = NULL;
}
s_buf * buf_fd_open_r (s_buf *buf, s32 fd)
{
s_buf_fd *buf_fd;
assert(buf);
assert(fd);
buf_fd = alloc(sizeof(s_buf_fd));
if (! buf_fd)
return NULL;
buf_fd->fd = fd;
buf->line = 0;
buf->refill = buf_fd_open_r_refill;
buf->user_ptr = buf_fd;
return buf;
}
sw buf_fd_open_r_refill (s_buf *buf)
{
int avail;
s32 fd;
uw r;
uw size;
assert(buf);
assert(buf->user_ptr);
if (buf->rpos > buf->wpos ||
buf->wpos > buf->size)
return -1;
size = buf->size - buf->wpos;
fd = ((s_buf_fd *) (buf->user_ptr))->fd;
//r = read(fd, buf->ptr.pchar + buf->wpos, size);
if (ioctl(fd, FIONREAD, &avail) == -1 ||
avail <= 0)
return -1;
if ((uw) avail > size)
avail = size;
r = read(fd, buf->ptr.pchar + buf->wpos, avail);
if (buf->wpos + r > buf->size) {
err_puts("buf_fd_open_r_refill: buffer overflow");
assert(! "buf_fd_open_r_refill: buffer overflow");
return -1;
}
buf->wpos += r;
return r;
}
s_buf * buf_fd_open_w (s_buf *buf, s32 fd)
{
s_buf_fd *buf_fd;
assert(buf);
assert(fd);
buf_fd = alloc(sizeof(s_buf_fd));
if (! buf_fd)
return NULL;
buf_fd->fd = fd;
buf->flush = buf_fd_open_w_flush;
buf->seek = buf_fd_open_w_seek;
buf->user_ptr = buf_fd;
return buf;
}
sw buf_fd_open_w_flush (s_buf *buf)
{
sw bytes;
s_buf_fd *buf_fd;
sw e;
uw min_wpos;
s_buf_save *save;
sw size;
sw w;
assert(buf);
assert(buf->user_ptr);
if (buf->rpos)
return -1;
if (buf->rpos > buf->wpos)
return -1;
if (buf->wpos > buf->size)
return -1;
min_wpos = buf_save_min_wpos(buf);
size = min_wpos;
if (size == 0)
return buf->size - buf->wpos;
buf_fd = buf->user_ptr;
bytes = 0;
while (bytes < size) {
if ((w = write(buf_fd->fd, buf->ptr.pchar + bytes,
size - bytes)) < 0) {
e = errno;
err_write_1("buf_fd_open_w_flush: write: ");
err_puts(strerror(e));
return -1;
}
bytes += w;
}
buf->wpos -= size;
save = buf->save;
while (save) {
save->wpos -= size;
save = save->next;
}
return size;
}
sw buf_fd_open_w_seek (s_buf *buf, sw offset, u8 whence)
{
s_buf_fd *buf_fd;
sw r;
assert(buf);
assert(! buf->save);
assert(buf->user_ptr);
buf_fd = buf->user_ptr;
if ((r = buf_flush(buf)) < 0)
return r;
if (lseek(buf_fd->fd, offset, whence)) {
err_puts("buf_fd_open_w_seek: fseek");
return -1;
}
return 0;
}
