shared_memory.cpp

#include <shm_base.hpp>
 
namespace irlab
{
 
namespace shm
{
 
int
disconnectMemory(std::string name)
{
  if (name[0] == '/')
  {
    name = name.erase(0, 1);
  }
  std::string str_buf = "/shm_" + regex_replace(name, std::regex("/"), "_");
  return shm_unlink(str_buf.c_str());
}
 
 
SharedMemory::SharedMemory(int oflag, PERM perm)
: shm_fd(-1)
, shm_oflag(oflag)
, shm_perm(perm)
, shm_size(0)
, shm_ptr(nullptr)
{
}
 
 
size_t
SharedMemory::getSize() const
{
  return shm_size;
}
 
 
unsigned char*
SharedMemory::getPtr()
{
  return shm_ptr;
}
 
SharedMemoryPosix::SharedMemoryPosix(std::string name, int oflag, PERM perm)
: SharedMemory(oflag, perm)
, shm_name(name)
{
  if (shm_name[0] == '/')
  {
    shm_name = shm_name.erase(0, 1);
  }
}
 
 
SharedMemoryPosix::~SharedMemoryPosix()
{
  if (shm_fd >= 0)
  {
    close(shm_fd);
  }
}
 
 
bool
SharedMemoryPosix::connect(size_t size)
{
  std::string str_buf = "/shm_" + regex_replace(shm_name, std::regex("/"), "_");
 
  shm_fd = shm_open(str_buf.c_str(), shm_oflag, static_cast<mode_t>(shm_perm));  
  if (shm_fd < 0)
  {
    return false;
  }
  struct stat stat;
  fstat(shm_fd, &stat);
  if (size == 0)
  {
    shm_size = stat.st_size;
  } else {
    shm_size = size;
    if (stat.st_size < shm_size)
    {
      if (ftruncate(shm_fd, shm_size) < 0)
      {
        throw std::runtime_error("shm::getMemory(): Could not change shared memory size!");
      }
      // To Update stat.st_size
      fstat(shm_fd, &stat);
    }
  }
  shm_ptr = reinterpret_cast<unsigned char *>(mmap(NULL,
          stat.st_size,
          PROT_READ|PROT_WRITE,
          MAP_SHARED,
          shm_fd,
            0));
 
  if (shm_ptr == MAP_FAILED) {
    close(shm_fd);
    shm_fd = -1;
    return false;
  }
 
  return true;
}
 
 
int
SharedMemoryPosix::disconnect()
{
  return disconnectMemory(shm_name);
}
 
 
bool
SharedMemoryPosix::isDisconnected() const
{
  struct stat stat;
  if (shm_fd < 0)
  {
    return true;
  }
 
  fstat(shm_fd, &stat);
  if (stat.st_nlink <= 0)
  {
    return true;
  }
  return false;
}
 
 
}
 
}