labcomm_fd_reader.c

#include <errno.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include "labcomm_private.h"
#include "labcomm_fd_reader.h"

#define BUFFER_SIZE 2048

struct labcomm_fd_reader {
  struct labcomm_reader reader;
  int fd;
  int close_fd_on_free;
};

static int fd_alloc(struct labcomm_reader *r, char *version)
{
  int result = 0;
  
  r->count = 0;
  r->pos = 0;
  r->data = malloc(BUFFER_SIZE);
  if (! r->data) {
    r->data_size = 0;
    result = -ENOMEM;
  } else {

    r->data_size = BUFFER_SIZE;
    result = r->data_size;
    if (version && version[0]) {
      char *tmp;
      
      tmp = labcomm_read_string(r);
      if (strcmp(tmp, version) != 0) {
	result = -EINVAL;
      } else {
	result = r->data_size;
      }
      free(tmp);
    }
  }
  return result;
}

static int fd_free(struct labcomm_reader *r)
{
  struct labcomm_fd_reader *context = r->context;

  free(r->data);
  r->data = 0;
  r->data_size = 0;
  r->count = 0;
  r->pos = 0;

  if (context->close_fd_on_free) {
    close(context->fd);
  }
  free(context);

  return 0;
}

static int fd_fill(struct labcomm_reader *r)
{
  int result = 0;
  struct labcomm_fd_reader *context = r->context;

  if (r->pos < r->count) {
    result = r->count - r->pos;
  } else {
    int err;
    
    r->pos = 0;
    err = read(context->fd, r->data, r->data_size);
    if (err <= 0) {
      r->count = 0;
      result = -EPIPE;
    } else {
      r->count = err;
      result = r->count - r->pos;
    }
  }
  return result;
}

static int fd_start(struct labcomm_reader *r)
{
  return fd_fill(r);
}

static int fd_end(struct labcomm_reader *r)
{
  return 0;
}

static const struct labcomm_reader_action action = {
  .alloc = fd_alloc,
  .free = fd_free,
  .start = fd_start,
  .fill = fd_fill,
  .end = fd_end,
  .ioctl = NULL
};

struct labcomm_reader *labcomm_fd_reader_new(int fd, int close_fd_on_free)
{
  struct labcomm_fd_reader *result;

  result = malloc(sizeof(*result));
  if (result == NULL) {
    return NULL;
  } else {
    result->fd = fd;
    result->close_fd_on_free = close_fd_on_free;
    result->reader.context = result;
    result->reader.action = &action;
    return &result->reader;
  }
}