tests/cdt/tshare.c

/***********************************************************************
*                                                                      *
*               This software is part of the ast package               *
*          Copyright (c) 1999-2011 AT&T Intellectual Property          *
*                      and is licensed under the                       *
*                 Eclipse Public License, Version 1.0                  *
*                    by AT&T Intellectual Property                     *
*                                                                      *
*                A copy of the License is available at                 *
*          http://www.eclipse.org/org/documents/epl-v10.html           *
*         (with md5 checksum b35adb5213ca9657e911e9befb180842)         *
*                                                                      *
*              Information and Software Systems Research               *
*                            AT&T Research                             *
*                           Florham Park NJ                            *
*                                                                      *
*                 Glenn Fowler <gsf@research.att.com>                  *
*                                                                      *
***********************************************************************/
#include    "dttest.h"

static char Space[16*1024*1024];
static char *Current = &Space[0];
static int  Close = 0;
static int  Free = 0;

#if __STD_C
static int event(Dt_t* dt, int type, Void_t* obj, Dtdisc_t* disc)
#else
static int event(dt, type, obj, disc)
Dt_t*   dt;
int type;
Void_t* obj;
Dtdisc_t* disc;
#endif
{   if(type == DT_OPEN)
    {   /* opening first dictionary */
        if(obj)
        {   if(Current == &Space[0])
                return 0;
            else /* opening a dictionary sharing with some previous one */
            {   *((Void_t**)obj) = (Void_t*)(&Space[0]);
                return 1;
            }
        }
        else    return 0;
    }
    else if(type == DT_CLOSE)
    {   if(Close == 0 ) /* do not free objects */
            return 1;
        else    return 0;
    }
    else    return 0;
}

#if __STD_C
static Void_t* memory(Dt_t* dt, Void_t* buf, size_t size, Dtdisc_t* disc)
#else
static Void_t* memory(dt,buf,size,disc)
Dt_t*   dt;
Void_t* buf;
size_t  size;
Dtdisc_t* disc;
#endif
{
    if(!buf)
    {   size = ((size + sizeof(Void_t*)-1)/sizeof(Void_t*))*sizeof(Void_t*);
        buf = (Void_t*)Current;
        Current += size;
    }
    else
    {   if(Close > 0)
            Free += 1;
    }
    return buf;
}

Dtdisc_t Disc =
    { 0, sizeof(long), -1,
      newint, NIL(Dtfree_f), compare, hashint,
      memory, event
    };

tmain()
{
    Dt_t        *dt1, *dt2;
    long        i, k;

    if(!(dt1 = dtopen(&Disc,Dtoset)) )
        terror("Opening Dtoset1");
    if((long)dtinsert(dt1,1L) != 1)
        terror("Inserting 1");
    if((long)dtinsert(dt1,3L) != 3)
        terror("Inserting 3");
    if((long)dtinsert(dt1,5L) != 5)
        terror("Inserting 5");

    if(!(dt2 = dtopen(&Disc,Dtoset)) )
        terror("Opening Dtoset2");
    if((long)dtinsert(dt2,2L) != 2)
        terror("Inserting 2");
    if((long)dtinsert(dt2,4L) != 4)
        terror("Inserting 4");
    if((long)dtinsert(dt2,6L) != 6)
        terror("Inserting 6");

    for(i = 1; i <= 6; ++i)
        if((long)dtsearch(dt1,i) != i)
            terror("Didn't find a long");

    for(i = (long)dtlast(dt2), k = 6; i != 0; i = (long)dtprev(dt2,i), k -= 1)
        if(i != k)
            terror("Didn't walk a long");

    /* this test makes sure that dtclose() does not free objects */
    Close = 0;
    dtclose(dt1);
    if(Free > 0 )
        terror("Memory should not have been freed");

    /* this test makes sure that all objects are freed */
    Close = 1;
    dtclose(dt2);
    if(Free <= 0 )
        terror("Memory should have been freed");

    /* test to make sure that shared dictionaries use the same method */
    Current = &Space[0];
    if(!(dt1 = dtopen(&Disc, Dtrhset)) )
        terror("Opening first dictionary");
    if((dt2 = dtopen(&Disc, Dtset)) )
        terror("This open should have failed");

    texit(0);
}