ordma_lwt.c

#include <caml/mlvalues.h>
#include <caml/alloc.h>
#include <caml/unixsupport.h>

#include <caml/bigarray.h>

#include <rdma/rsocket.h>
#include "ordma_debug.h"

static int msg_flag_table[] = {
  MSG_OOB, MSG_DONTROUTE, MSG_PEEK
};


value ordma_lwt_unix_recv(value fd, value buf, value ofs, value len, value flags)
{
  int ret;
  ret = rrecv(Int_val(fd),
              &Byte(String_val(buf), Long_val(ofs)),
              Long_val(len),
              caml_convert_flag_list(flags, msg_flag_table)
              );
  if (ret == -1) {
    uerror("Lwt_rsocket.recv", Nothing);
  }
  return Val_int(ret);
}

value ordma_lwt_unix_bytes_recv(value fd, value buf, value ofs, value len, value flags)
{
  int ret;
  ret = rrecv(Int_val(fd),
              (char*)Caml_ba_array_val(buf)->data + Long_val(ofs),
              Long_val(len),
              caml_convert_flag_list(flags, msg_flag_table)
              );
  if (ret == -1) {
    uerror("Lwt_socket.Bytes.recv", Nothing);
  }
  return Val_int(ret);
}


value ordma_lwt_unix_send(value fd, value buf, value ofs, value len, value flags)
{
  int ret;
  ret = rsend(Int_val(fd), &Byte(String_val(buf), Long_val(ofs)), Long_val(len),
             caml_convert_flag_list(flags, msg_flag_table));
  if (ret == -1) {
    uerror("Lwt_socket.send", Nothing);
  }
  return Val_int(ret);
}

value ordma_lwt_unix_bytes_send(value fd, value buf, value ofs, value len, value flags)
{
  int ret;
  ret = rsend(Int_val(fd), (char*)Caml_ba_array_val(buf)->data + Long_val(ofs), Long_val(len),
             caml_convert_flag_list(flags, msg_flag_table));
  if (ret == -1) {
    uerror("Lwt_socket.Bytes.send", Nothing);
  }
  return Val_int(ret);
}

CAMLprim value ordma_lwt_unix_readable(value fd)
{
  
  struct pollfd pollfd;
  pollfd.fd = Int_val(fd);
  ORDMA_LOG("ordma_lwt_unix_readable(%i)", pollfd.fd);
  pollfd.events = POLLIN;
  pollfd.revents = 0;
  if (rpoll(&pollfd, 1, 0) < 0)
    uerror("readable", Nothing);
  return (Val_bool(pollfd.revents & POLLIN));
}

CAMLprim value ordma_lwt_unix_writable(value fd)
{
  struct pollfd pollfd;
  pollfd.fd = Int_val(fd);
  ORDMA_LOG("ordma_lwt_unix_writable(%i)", pollfd.fd);
  pollfd.events = POLLOUT;
  pollfd.revents = 0;
  if (rpoll(&pollfd, 1, 0) < 0)
    uerror("writable", Nothing);
  return (Val_bool(pollfd.revents & POLLOUT));
}

//TODO: ordma_close....