atomic_get.c

#include <unistd.h>    /* for usleep() */
#include <errno.h>
#include <shm_arena.h> /* for Shared Memory Arena Library */


int main(void)
{
  int *ptr;
  int created = 0; /* bool */
  const char *SEG_NAME = "my count";
  size_t seg_size = sizeof(int);
  

  /* Get exclusive access to the shared memory arena.  Because this is
   * the first Shared Memory Arena API call, his also automatically
   * creates this local process's arena object using the default
   * shared memory arena file. */
  if(sm_arena_wrlock())
    {
      printf("Failed to get arena lock\n");
      return 1;
    }

  /* First check if the segment is already exists. */
  errno = 0;
  ptr = (int *) sm_connect(SHM_UNKNOWN_SIZE, SEG_NAME);

  if(!ptr && !errno)
  {
    errno = 0;
    /* The segment does not exist so we create it */
    ptr = (int *) sm_create(seg_size, SEG_NAME);

    if(ptr)
    {
      /* and initialize it, */
        *ptr = 0;
        created = 1;
    }
  }
  else if(!errno)
    seg_size = sm_size(ptr);

  if(!ptr)
    {
      printf("failed to create segment \"%s\"\n", SEG_NAME);
      return 1;
    }

  /* all in an atomic manner.  Release exclusive access. */
  sm_arena_unlock();

  if(ptr)
    {
      if(created)
        printf("Created shared memory segment \"%s\" of size %zu\n",
           SEG_NAME, seg_size);
      else
        printf("Shared memory segment \"%s\" of size %zu already exists\n",
           SEG_NAME, seg_size);
    }

  return 0;
}