photon_buffer_internal.c

Go to the documentation of this file.

// =============================================================================
//  Photon RDMA Library (libphoton)
//
//  Copyright (c) 2016, Trustees of Indiana University,
//  All rights reserved.
//
//  This software may be modified and distributed under the terms of the BSD
//  license.  See the COPYING file for details.
//
//  This software was created at the Indiana University Center for Research in
//  Extreme Scale Technologies (CREST).
// =============================================================================

#include <stdlib.h>
#include <string.h>
#include <inttypes.h>

#include "libphoton.h"
#include "photon_backend.h"
#include "photon_buffer_internal.h"

static photonBufferInterface ifaces[BUFFER_IFACES_MAX];
static int iface_count = 0;

int photon_buffer_init(photonBufferInterface bi) {
  if (iface_count >= BUFFER_IFACES_MAX) {
    log_err("Exceeded max buffer interfaces: %d", BUFFER_IFACES_MAX);
    return PHOTON_ERROR;
  }
  ifaces[iface_count++] = bi;
  return PHOTON_OK;
}

photonBufferHandle _photon_buffer_create(void *addr, uint64_t size, int flags) {
  photonBufferHandle new_buf;

  dbg_trace("%llu", size);

  new_buf = malloc(sizeof(struct photon_buffer_handle_t));
  if (!new_buf) {
    log_err("malloc failed");
    return NULL;
  }

  memset(new_buf, 0, sizeof(*new_buf));
    
  new_buf->bint.buf.addr = (uintptr_t)addr;
  new_buf->bint.buf.size = size;
  new_buf->bint.buf.priv = (struct photon_buffer_priv_t){0,0};
  new_buf->bint.priv_ptr = NULL;
  new_buf->bint.priv_size = 0;

  // TODO: setup only the appropriate buffer interfaces for this handle
  new_buf->ifaces = ifaces;
  new_buf->iface_count = iface_count;
  
  new_buf->ref_count = 1;
  new_buf->is_registered = 0;

  dbg_trace("allocated buffer handle: %p", new_buf);
  dbg_trace("contains ptr to addr: %p of size %" PRIu64, addr, size);
  
  return new_buf;
}

photonBufferHandle photon_buffer_create(void *addr, uint64_t size, int flags) {
  int i;
  
  if (iface_count <= 0) {
    log_err("No buffer interfaces set!");
    return NULL;
  }

  photonBufferHandle buf = _photon_buffer_create(addr, size, flags);
  if (buf) {
    for (i=0; i<buf->iface_count; i++) {
      if (buf->ifaces[i]->buffer_create)
    buf->ifaces[i]->buffer_create(&(buf->bint), flags);
    }
  }

  return buf;
}

void photon_buffer_free(photonBufferHandle buf) {
  int i;
  
  if (iface_count <= 0) {
    log_err("No buffer interfaces set!");
    return;
  }

  for (i=0; i<buf->iface_count; i++) {
    if (buf->ifaces[i]->buffer_free)
      buf->ifaces[i]->buffer_free(&(buf->bint));
  }
}

int photon_buffer_register(photonBufferHandle buf, int flags) {
  int i, err = PHOTON_OK;
  
  if (iface_count <= 0) {
    log_err("No buffer interfaces  set!");
    return PHOTON_ERROR;
  }

  for (i=0; i<buf->iface_count; i++) {
    if (buf->ifaces[i]->buffer_register)
      err += buf->ifaces[i]->buffer_register(&(buf->bint), flags);
  }

  if (err == PHOTON_OK) {
    buf->is_registered = 1;
  }

  return err;
}

int photon_buffer_unregister(photonBufferHandle buf) {
  int i, err = PHOTON_OK;
  
  if (iface_count <= 0) {
    log_err("No buffer interfaces set!");
    return PHOTON_ERROR;
  }

  for (i=0; i<buf->iface_count; i++) {
    if (buf->ifaces[i]->buffer_unregister)
      err += buf->ifaces[i]->buffer_unregister(&(buf->bint));
  }

  if (err == PHOTON_OK) {
    buf->is_registered = 0;
  }
  
  return err;
}