test_ncbi_heapmgr.c

ncbi源码

/*
 * ===========================================================================
 * PRODUCTION $Log: test_ncbi_heapmgr.c,v $
 * PRODUCTION Revision 1000.0  2003/10/29 17:05:07  gouriano
 * PRODUCTION PRODUCTION: IMPORTED [ORIGINAL] Dev-tree R6.12
 * PRODUCTION
 * ===========================================================================
 */

/*  $Id: test_ncbi_heapmgr.c,v 1000.0 2003/10/29 17:05:07 gouriano Exp $
 * ===========================================================================
 *
 *                            PUBLIC DOMAIN NOTICE
 *               National Center for Biotechnology Information
 *
 *  This software/database is a "United States Government Work" under the
 *  terms of the United States Copyright Act.  It was written as part of
 *  the author's official duties as a United States Government employee and
 *  thus cannot be copyrighted.  This software/database is freely available
 *  to the public for use. The National Library of Medicine and the U.S.
 *  Government have not placed any restriction on its use or reproduction.
 *
 *  Although all reasonable efforts have been taken to ensure the accuracy
 *  and reliability of the software and data, the NLM and the U.S.
 *  Government do not and cannot warrant the performance or results that
 *  may be obtained by using this software or data. The NLM and the U.S.
 *  Government disclaim all warranties, express or implied, including
 *  warranties of performance, merchantability or fitness for any particular
 *  purpose.
 *
 *  Please cite the author in any work or product based on this material.
 *
 * ===========================================================================
 *
 * Author:  Anton Lavrentiev
 *
 * File Description:
 *   Simple random test of heap manager API. Not an interface example!
 *
 */

#include "../ncbi_priv.h"
#include <connect/ncbi_heapmgr.h>
#include <stdlib.h>
#include <time.h>
#if 0
#  define eLOG_Warning eLOG_Fatal
#  define eLOG_Error   eLOG_Fatal
#  include "../ncbi_heapmgr.c"
#endif
/* This header must go last */
#include "test_assert.h"


#ifdef __cplusplus
extern "C" {
#endif

/*ARGSUSED*/
static void* s_Expand(void* base, TNCBI_Size size, void* arg)
{
    if (base && size)
        return realloc(base, size);
    if (size)
        return malloc(size);
    if (base)
        free(base);
    return 0;
}

#ifdef __cplusplus
} /* extern "C" */
#endif


int main(void)
{
    SHEAP_Block* blk;
    int r, j, i;
    HEAP heap;
    char* c;

    CORE_SetLOGFILE(stderr, 0/*false*/);
    for (j = 1; j <= 3; j++) {
        srand((int)time(0));
        CORE_LOGF(eLOG_Note, ("Creating heap %d", j));
        if (!(heap = HEAP_Create(0, 0, 4096, s_Expand, 0)))
            CORE_LOG(eLOG_Error, "Cannot create heap");
        while (rand() != 12345) {
            r = rand() & 7;
            if (r == 1 || r == 3) {
                i = rand() & 0xFF;
                if (i) {
                    CORE_LOGF(eLOG_Note, ("Allocating %d bytes", i));
                    if (!(blk = HEAP_Alloc(heap, i)))
                        CORE_LOG(eLOG_Error, "Allocation failed");
                    else
                        CORE_LOG(eLOG_Note, "Done");
                    c = (char*) blk + sizeof(*blk);
                    while (i--)
                        *c++ = rand();
                }
            } else if (r == 2 || r == 4) {
                blk = 0;
                i = 0;
                do {
                    blk = HEAP_Walk(heap, blk);
                    if (!blk)
                        break;
                    i++;
                } while (rand() & 0x7);
                if (blk && (short) blk->flag) {
                    CORE_LOGF(eLOG_Note,
                              ("Deleting block #%d, size %u", i,
                               (unsigned) (blk->size - sizeof(*blk))));
                    HEAP_Free(heap, blk);
                    CORE_LOG(eLOG_Note, "Done");
                }
            } else if (r == 5) {
                blk = 0;
                i = 0;
                CORE_LOG(eLOG_Note, "Walking the heap");
                do {
                    blk = HEAP_Walk(heap, blk);
                    if (blk)
                        CORE_LOGF(eLOG_Note,
                                  ("Block #%d (%s), size %u", ++i,
                                   (short) blk->flag ? "used" : "free",
                                   (unsigned) (blk->size - sizeof(*blk))));
                } while (blk);
                CORE_LOGF(eLOG_Note,
                          ("Total of %d block%s", i, i == 1 ? "" : "s"));
            } else if (r == 6 || r == 7) {
                HEAP newheap;

                if (r == 6)
                    newheap = HEAP_Attach(/* HACK! */*((char**)heap));
                else
                    newheap = HEAP_Copy(heap);

                if (!newheap)
                    CORE_LOGF(eLOG_Error, ("%s failed",
                                           r == 6 ? "Attach" : "Copy"));
                blk = 0;
                i = 0;
                CORE_LOGF(eLOG_Note, ("Walking %s heap",
                                      r == 6 ? "attached" : "copied"));
                do {
                    blk = HEAP_Walk(newheap, blk);
                    if (blk)
                        CORE_LOGF(eLOG_Note,
                                  ("Block #%d (%s), size %u", ++i,
                                   (short) blk->flag ? "used" : "free",
                                   (unsigned) (blk->size - sizeof(*blk))));
                } while (blk);
                CORE_LOGF(eLOG_Note,
                          ("Total of %d block%s", i, i == 1 ? "" : "s"));
                HEAP_Detach(newheap);
            } else {
                TNCBI_Size size = HEAP_Size(heap);
                CORE_LOGF(eLOG_Note, ("Heap trimmed: %u -> %u",
                                      (unsigned) size,
                                      (unsigned) HEAP_Trim(heap)));
            }
        }
        HEAP_Destroy(heap);
        CORE_LOGF(eLOG_Note, ("Heap %d done", j));
    }
    CORE_LOG(eLOG_Note, "Test completed");
    return 0;
}


/*
 * --------------------------------------------------------------------------
 * $Log: test_ncbi_heapmgr.c,v $
 * Revision 1000.0  2003/10/29 17:05:07  gouriano
 * PRODUCTION: IMPORTED [ORIGINAL] Dev-tree R6.12
 *
 * Revision 6.12  2003/08/25 14:58:10  lavr
 * Adjust test to take advantage of modified API
 *
 * Revision 6.11  2003/07/31 17:54:16  lavr
 * +HEAP_Trim() test
 *
 * Revision 6.10  2003/02/27 15:34:35  lavr
 * Log moved to end
 *
 * Revision 6.9  2002/04/15 19:21:44  lavr
 * +#include "../test/test_assert.h"
 *
 * Revision 6.8  2001/07/03 20:53:38  lavr
 * HEAP_Copy() test added
 *
 * Revision 6.7  2001/06/19 19:12:04  lavr
 * Type change: size_t -> TNCBI_Size; time_t -> TNCBI_Time
 *
 * Revision 6.6  2001/01/23 23:22:05  lavr
 * Patched logging (in a few places)
 *
 * Revision 6.5  2001/01/12 23:59:53  lavr
 * Message logging modified for use LOG facility only
 *
 * Revision 6.4  2000/12/29 18:23:42  lavr
 * getpagesize() replaced by a constant 4096, which is "more portable".
 *
 * Revision 6.3  2000/05/31 23:12:32  lavr
 * First try to assemble things together to get working service mapper
 *
 * Revision 6.2  2000/05/16 15:21:03  lavr
 * Cleaned up with format - argument correspondence; #include <time.h> added
 *
 * Revision 6.1  2000/05/12 19:35:13  lavr
 * First working revision
 *
 * ==========================================================================
 */