test-example-projects/c_projects/19wnuster/include/nuster/shmem.h

/*
 * include/nuster/shmem.h
 * This file defines everything related to nuster shmem.
 *
 * Copyright (C) Jiang Wenyuan, < koubunen AT gmail DOT com >
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation, version 2.1
 * exclusively.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
 */

#ifndef _NUSTER_SHMEM_H
#define _NUSTER_SHMEM_H

#include <nuster/common.h>


#define NST_SHMEM_BLOCK_MIN_SIZE      4096ULL
#define NST_SHMEM_BLOCK_MIN_SHIFT     12
#define NST_SHMEM_CHUNK_MIN_SIZE      8ULL
#define NST_SHMEM_CHUNK_MIN_SHIFT     3
#define NST_SHMEM_BLOCK_MAX_SIZE      1024 * 1024 * 2
#define NST_SHMEM_BLOCK_MAX_SHIFT     21
#define NST_SHMEM_INFO_BITMAP_BITS    32


/* start                                 alignment                   stop
 * |                                     |   |                       |
 * |_______|_0_|_...._|_M_|_0_|_..._|_N__|_*_|__0__|__...__|__N__|_*_|
 *         |              |                  |             |     |
 *         chunk        block              begin         end   <bitmap>
 *
 *
 */

/*
 * info:
 * | bitmap: 32 | reserved: 16 | 5 | full: 1 | bitmap: 1 | inited: 1 | type: 8 |
 * bitmap: points to bitmap area, doesn't change once set
 * chunk size[n]: 1<<(NST_SHMEM_CHUNK_MIN_SHIFT + n)
 */
typedef struct nst_shmem_ctrl {
    uint64_t                     info;
    uint8_t                     *bitmap;

    struct nst_shmem_ctrl       *prev;
    struct nst_shmem_ctrl       *next;
} nst_shmem_ctrl_t;

typedef struct nst_shmem {
    uint8_t                     *start;
    uint8_t                     *stop;
    uint8_t                     *bitmap;
    char                         name[16];

#if defined NUSTER_USE_PTHREAD || defined USE_PTHREAD_PSHARED
    pthread_mutex_t              mutex;
#else
    unsigned int                 waiters;
#endif

    uint64_t                     size;
    uint64_t                     used;

    uint32_t                     block_size;  /* max shmem can be allocated */
    uint32_t                     chunk_size;  /* min shmem can be allocated */
    int                          chunk_shift;
    int                          block_shift;

    int                          chunks;
    int                          blocks;

    nst_shmem_ctrl_t           **chunk;
    nst_shmem_ctrl_t            *block;
    nst_shmem_ctrl_t            *empty;
    nst_shmem_ctrl_t            *full;

    struct {
        uint8_t                 *begin;
        uint8_t                 *free;
        uint8_t                 *end;
    } data;
} nst_shmem_t;


#define bit_set(bit, i)         (bit |= 1 << i)
#define bit_clear(bit, i)       (bit &= ~(1 << i))
#define bit_used(bit, i)        (((bit) >> (i)) & 1)
#define bit_unused(bit, i)      ((((bit) >> (i)) & 1) == 0)

static inline void
_nst_shmem_block_set_type(nst_shmem_ctrl_t *block, uint8_t type) {
    *(uint8_t *)(&block->info) = type;
}

static inline void
_nst_shmem_block_set_inited(nst_shmem_ctrl_t *block) {
    bit_set(block->info, 9);
}

static inline int
_nst_shmem_block_is_inited(nst_shmem_ctrl_t *block) {
    return bit_used(block->info, 9);
}

static inline void
_nst_shmem_block_set_bitmap(nst_shmem_ctrl_t *block) {
    bit_set(block->info, 10);
}

static inline void
_nst_shmem_block_set_full(nst_shmem_ctrl_t *block) {
    bit_set(block->info, 11);
}

static inline int
_nst_shmem_block_is_full(nst_shmem_ctrl_t *block) {
    return bit_used(block->info, 11);
}

static inline void
_nst_shmem_block_clear_full(nst_shmem_ctrl_t *block) {
    bit_clear(block->info, 11);
}

nst_shmem_t *
nst_shmem_create(char *name, uint64_t size, uint32_t block_size, uint32_t chunk_size);

void *nst_shmem_alloc(nst_shmem_t *shmem, int size);
void nst_shmem_free(nst_shmem_t *shmem, void *p);

#endif /* _NUSTER_SHMEM_H */