#include <stdio.h>

#include "mem/FixMemPool.h"
#include <sys/ipc.h>
#include "test/ShmTableTest.h"

void printFixMemPoolInfo(LPFIXMEMPOLL pstFixMemPool)
{
    int i;
    int iFree;
    char* pszMem;

    if (NULL == pstFixMemPool) return;

    printf("pool freeNum:%d, block num:%d, block size:%d\n", pstFixMemPool->dwFreeNum, pstFixMemPool->dwBlockNum, pstFixMemPool->dwBlockSize);

    for (i = 0; i < pstFixMemPool->dwBlockNum; i++)
    {
        if (pstFixMemPool->piBlock[i] & 0x80000000)
        {
            printf("0x%x ", pstFixMemPool->piBlock[i]);
        }
        else
        {
            printf("%d ", pstFixMemPool->piBlock[i]);
        }
    }
    printf("\n");

    for (i = 0; i < pstFixMemPool->dwBlockNum; i++)
    {
        iFree = 0;
        pszMem = fix_mem_pool_at(pstFixMemPool, i, &iFree);

        if (pszMem == NULL || iFree == 1) continue;
        printf("%s ", pszMem);
    }
    printf("\n");
}

void printMemChar(char* aszMem[], int iNum)
{
    int i;

    for (i = 0; i < iNum; i++)
    {
        printf("%s ", aszMem[i]);
    }

    printf("\n");
}

void testFixMemPool()
{
    char szMemData[1024];
    FIXMEMPOLL stFixMemPool;
    int iRet;
    int aiIndex[5];
    char* aszMem[5];
    int i;

    iRet = init_fix_mem_pool(&stFixMemPool, szMemData, sizeof(szMemData), 32);

    if (iRet != 0)
    {
        printf("error to init fix mem pool:%d\n", iRet);
        return;
    }

    printFixMemPoolInfo(&stFixMemPool);

    for (i = 0; i < 5; i++)
    {
        aszMem[i] = fix_mem_pool_malloc(&stFixMemPool, &aiIndex[i]);
        if (aszMem[i] == NULL)
        {
            printf("error to allocate mem for:%d\n", i);
            return;
        }

        sprintf(aszMem[i], "Hello_%d", i);

        printf("i=%d, memIndex=%d\n", i, aiIndex[i]);
    }

    printFixMemPoolInfo(&stFixMemPool);
    printMemChar(aszMem, 5);

    iRet = fix_mem_pool_free(&stFixMemPool, aiIndex[3]);

    if (iRet != 0)
    {
        printf("1.error to free mem:%d\n", iRet);
    }

    printFixMemPoolInfo(&stFixMemPool);

    iRet = fix_mem_pool_free(&stFixMemPool, aiIndex[3]);

    if (iRet != 0)
    {
        printf("2.error to free mem:%d\n", iRet);
    }

    printFixMemPoolInfo(&stFixMemPool);

    aszMem[3] = fix_mem_pool_malloc(&stFixMemPool, &aiIndex[3]);
    if (aszMem[3] == NULL)
    {
        printf("error to allocate mem for %d", 3);
        return;
    }

    sprintf(aszMem[3], "New_%d", 3);

    printFixMemPoolInfo(&stFixMemPool);
    printMemChar(aszMem, 5);

    for (i = 0; i < 5; i++)
    {
        iRet = fix_mem_pool_free(&stFixMemPool, aiIndex[i]);
        if (iRet != 0)
        {
            printf("error to free mem for:%d, cause:%d\n", i, iRet);
            return;
        }
    }

    printFixMemPoolInfo(&stFixMemPool);
}

int main(int argc, const char* argv[])
{
    char chType;
    //int chFlag = 0664 | IPC_CREAT;
    //int iFlag = chFlag & (~IPC_CREAT);
    //size_t t = 0xFFFFFFFF;

    //printf("0x%x,0x%x, 0x%x, %d, 0x%x, 0x%zd, %d\n", IPC_CREAT, iFlag, chFlag, iFlag & IPC_CREAT, 0664, t, sizeof(size_t));

    if (argc <= 1)
    {
        printf("not enough parameter!need command type!\n");
        return -1;
    }

    chType = *argv[1];

    switch (chType)
    {
    case 'm':   /* 内存池测试 */
        testFixMemPool();
        break;
    case 't':
        {
            int iType = argc > 2 ? *argv[2] : 'p';
            int iID = argc > 3 ? atoi(argv[3]) : 1;
            test_shm_table(iType, iID);
        }
        break;
    }

    return 0;
}