又一个OOC的示例---内存池

// pool

struct vsfpool_t

{

        uint32_t *flags;

        void *buffer;

        uint32_t size;

        uint32_t num;

};



#define VSFPOOL_DEFINE(name, type, num)                        \

        struct vsfpool_##name##_t\

        {\

                struct vsfpool_t pool;\

        uint32_t mskarr[((num) + 31) >> 5];\

        type buffer[(num)];\

        } name



#define VSFPOOL_INIT(p, type, n)                        \

        do {\

                (p)->pool.flags = (p)->mskarr;\

                (p)->pool.buffer = (p)->buffer;\

                (p)->pool.size = sizeof(type);\

                (p)->pool.num = (n);\

                vsfpool_init((struct vsfpool_t *)(p));\

        } while (0)



#define VSFPOOL_ALLOC(p, type)                                \

    (type *)vsfpool_alloc((struct vsfpool_t *)(p))



#define VSFPOOL_FREE(p, buf)                                        \

        vsfpool_free((struct vsfpool_t *)(p), (buf))



void vsfpool_init(struct vsfpool_t *pool);

void* vsfpool_alloc(struct vsfpool_t *pool);

void vsfpool_free(struct vsfpool_t *pool, void *buffer);

// pool

void vsfpool_init(struct vsfpool_t *pool)

{

        memset(pool->flags, 0, (pool->num + 31) >> 3);

}



void* vsfpool_alloc(struct vsfpool_t *pool)

{

        int index = mskarr_ffz(pool->flags, (pool->num + 31) >> 5);



        if (index >= pool->num)

        {

                return NULL;

        }

        mskarr_set(pool->flags, index);

        return (uint8_t *)pool->buffer + index * pool->size;

}



void vsfpool_free(struct vsfpool_t *pool, void *buffer)

{

        int index = ((uint8_t *)buffer - (uint8_t *)pool->buffer) / pool->size;



        if (index < pool->num)

        {

                mskarr_clr(pool->flags, index);

        }

}

VSFPOOL_DEFINE(vsfip_buffer_pool, struct vsfip_buffer_t, VSFIP_CFG_IPBUF_NUM);



VSFPOOL_INIT(&vsfip_buffer_pool, struct vsfip_buffer_t, VSFIP_CFG_IPBUF_NUM);

buffer = VSFPOOL_ALLOC(&vsfip_buffer_pool, struct vsfip_buffer_t);

VSFPOOL_FREE(&vsfip_buffer_pool, buffer);

void mskarr_set(uint32_t *arr, int bit)

{

        arr[bit >> 5] |= (1 << (bit & 0x1F));

}



void mskarr_clr(uint32_t *arr, int bit)

{

        arr[bit >> 5] &= ~(1 << (bit & 0x1F));

}



int mskarr_ffz(uint32_t *arr, int arrlen)

{

        int i, tmp;



        for (i = 0; i < arrlen; i++)

        {

                tmp = ffz(arr[i]);

                if (tmp >= 0)

                {

                        return (i << 5) + tmp;

                }

        }

        return -1;

}