Annotation of researchv10no/libc/gen/omalloc.c, revision 1.1

1.1     ! root        1: #ifdef debug
        !             2: #define ASSERT(p) if(!(p))botch("p");else
        !             3: botch(s)
        !             4: char *s;
        !             5: {
        !             6:        printf("assertion botched: %s\n",s);
        !             7:        abort();
        !             8: }
        !             9: #else
        !            10: #define ASSERT(p)
        !            11: #endif
        !            12: 
        !            13: /*     C storage allocator
        !            14:  *     circular first-fit strategy
        !            15:  *     works with noncontiguous, but monotonically linked, arena
        !            16:  *     each block is preceded by a ptr to the (pointer of) 
        !            17:  *     the next following block
        !            18:  *     blocks are exact number of words long 
        !            19:  *     aligned to the data type requirements of ALIGN
        !            20:  *     pointers to blocks must have BUSY bit 0
        !            21:  *     bit in ptr is 1 for busy, 0 for idle
        !            22:  *     gaps in arena are merely noted as busy blocks
        !            23:  *     last block of arena is empty and
        !            24:  *     has a pointer to first
        !            25:  *     idle blocks are coalesced during space search
        !            26:  *
        !            27:  *     a different implementation may need to redefine
        !            28:  *     ALIGN, NALIGN, BLOCK, BUSY, INT
        !            29:  *     where INT is integer type to which a pointer can be cast
        !            30: */
        !            31: #define INT int
        !            32: #define ALIGN int
        !            33: #define NALIGN 1
        !            34: #define WORD sizeof(union store)
        !            35: #define BLOCK 1024
        !            36: #define BUSY 1
        !            37: #define NULL 0
        !            38: #define testbusy(p) ((INT)(p)&BUSY)
        !            39: #define setbusy(p) (union store *)((INT)(p)|BUSY)
        !            40: #define clearbusy(p) (union store *)((INT)(p)&~BUSY)
        !            41: 
        !            42: union store {
        !            43:              union store *ptr;
        !            44:              ALIGN dummy[NALIGN];
        !            45:              int calloc;       /*calloc clears an array of integers*/
        !            46: };
        !            47: 
        !            48: static union store alloca;     /* initial arena */
        !            49: static union store *allocb = &alloca;  /*arena base*/
        !            50: static union store *allocp = &alloca;  /*search ptr*/
        !            51: static union store *allocx;    /*for benefit of realloc*/
        !            52: extern char *sbrk();
        !            53: 
        !            54: char *
        !            55: malloc(nbytes)
        !            56: unsigned nbytes;
        !            57: {
        !            58:        register union store *p, *q;
        !            59:        register nw;
        !            60:        register temp;
        !            61:        register union store *r = 0;
        !            62: 
        !            63:        nw = (nbytes+WORD+WORD-1)/WORD;
        !            64:        ASSERT(allock(allocp));
        !            65:        for(; ; ) {     /* done at most twice */
        !            66:                p = allocp;
        !            67:                if(alloca.ptr!=0)               /*C can't initialize union*/
        !            68:                for(temp=0; ; ) {
        !            69:                        if(!testbusy(p->ptr)) {
        !            70:                                while(!testbusy((q=p->ptr)->ptr)) {
        !            71:                                        ASSERT(q>p);
        !            72:                                        p->ptr = q->ptr;
        !            73:                                        allocp = p;
        !            74:                                }
        !            75:                                if(q>=p+nw && p+nw>=p)
        !            76:                                        goto found;
        !            77:                                r = p;
        !            78:                        }
        !            79:                        q = p;
        !            80:                        p = clearbusy(p->ptr);
        !            81:                        if(p <= q) {
        !            82:                                ASSERT(p==allocb);
        !            83:                                if(p != allocb)
        !            84:                                        return(NULL);
        !            85:                                if(++temp>1)
        !            86:                                        break;
        !            87:                        }
        !            88:                }
        !            89:                temp = nw;
        !            90:                p = (union store *)sbrk(0);
        !            91:                if (r && !testbusy(r->ptr) && r->ptr + 1 == p)
        !            92:                        temp -= p - r - 1;
        !            93:                temp = ((temp+BLOCK/WORD)/(BLOCK/WORD))*(BLOCK/WORD);
        !            94:                if(p+temp <= p)
        !            95:                        return(NULL);
        !            96:                for(; ; ) {
        !            97:                        q = (union store *)sbrk(temp*WORD);
        !            98:                        if((INT)q != -1)
        !            99:                                break;
        !           100:                        temp -= (temp-nw)/2;
        !           101:                        if(temp-nw <= 1)
        !           102:                                return(NULL);
        !           103:                }
        !           104:                ialloc((char *)q, (unsigned)temp*WORD);
        !           105:        }
        !           106: found:
        !           107:        allocp = p + nw;
        !           108:        if(q>allocp) {
        !           109:                allocx = allocp->ptr;
        !           110:                allocp->ptr = p->ptr;
        !           111:        }
        !           112:        p->ptr = setbusy(allocp);
        !           113:        return((char *)(p+1));
        !           114: }
        !           115: 
        !           116: /*     freeing strategy tuned for LIFO allocation
        !           117: */
        !           118: free(ap)
        !           119: char *ap;
        !           120: {
        !           121:        register union store *p = (union store *)ap;
        !           122: 
        !           123:        allocp = --p;
        !           124:        ASSERT(allock(allocp));
        !           125:        ASSERT(testbusy(p->ptr));
        !           126:        p->ptr = clearbusy(p->ptr);
        !           127:        ASSERT(p->ptr > allocp);
        !           128: }
        !           129: 
        !           130: /* ialloc(q, nbytes) inserts a block that did not come
        !           131:  * from malloc into the arena
        !           132:  *
        !           133:  * q points to new block
        !           134:  * r points to last of new block
        !           135:  * p points to last cell of arena before new block
        !           136:  * s points to first cell of arena after new block
        !           137: */
        !           138: ialloc(qq, nbytes)
        !           139: char *qq;
        !           140: unsigned nbytes;
        !           141: {
        !           142:        register union store *p, *q, *s;
        !           143:        union store *r;
        !           144: 
        !           145:        q = (union store *)qq;
        !           146:        r = q + (nbytes/WORD) - 1;
        !           147:        q->ptr = r;
        !           148:        if(alloca.ptr==0)               /*C can't initialize union*/
        !           149:                alloca.ptr = &alloca;
        !           150:        for(p=allocb; ; p=s) {
        !           151:                s = clearbusy(p->ptr);
        !           152:                if(s==allocb)
        !           153:                        break;
        !           154:                ASSERT(s>p);
        !           155:                if(s>r) {
        !           156:                        if(p<q)
        !           157:                                break;
        !           158:                        else
        !           159:                                ASSERT(p>r);
        !           160:                }
        !           161:        }
        !           162:        p->ptr = q==p+1? q: setbusy(q);
        !           163:        r->ptr = s==r+1? s: setbusy(s);
        !           164:        if(allocb > q)
        !           165:                allocb = q;
        !           166:        allocp = allocb;
        !           167: }
        !           168: 
        !           169: /*     realloc(p, nbytes) reallocates a block obtained from malloc()
        !           170:  *     and freed since last call of malloc()
        !           171:  *     to have new size nbytes, and old content
        !           172:  *     returns new location, or 0 on failure
        !           173: */
        !           174: 
        !           175: char *
        !           176: realloc(pp, nbytes)
        !           177: char *pp;
        !           178: unsigned nbytes;
        !           179: {
        !           180:        register union store *q;
        !           181:        register union store *p = (union store *)pp;
        !           182:        union store *s, *t;
        !           183:        register unsigned nw;
        !           184:        unsigned onw;
        !           185: 
        !           186:        ASSERT(allock(p-1));
        !           187:        if(testbusy(p[-1].ptr))
        !           188:                free((char *)p);
        !           189:        onw = p[-1].ptr - p;
        !           190:        q = (union store *)malloc(nbytes);
        !           191:        if(q==NULL || q==p)
        !           192:                return((char *)q);
        !           193:        ASSERT(q<p||q>p[-1].ptr);
        !           194:        s = p;
        !           195:        t = q;
        !           196:        nw = (nbytes+WORD-1)/WORD;
        !           197:        if(nw<onw)
        !           198:                onw = nw;
        !           199:        while(onw--!=0)
        !           200:                *t++ = *s++;
        !           201:        ASSERT(clearbusy(q[-1].ptr)-q==nw);
        !           202:        if(q<p && q+nw>=p)
        !           203:                (q+(q+nw-p))->ptr = allocx;
        !           204:        ASSERT(allock(q-1));
        !           205:        return((char *)q);
        !           206: }
        !           207: 
        !           208: #ifdef debug
        !           209: allock(q)
        !           210: union store *q;
        !           211: {
        !           212: #ifdef longdebug
        !           213:        register union store *p, *r;
        !           214:        int x;
        !           215:        x = 0;
        !           216:        p = allocb;
        !           217:        if(alloca.ptr==0)
        !           218:                return(1);
        !           219:        for( ; (r=clearbusy(p->ptr)) > p; p=r) {
        !           220:                if(p==q)
        !           221:                        x++;
        !           222:        }
        !           223:        return(r==allocb&(x==1|p==q));
        !           224: #else
        !           225:        return(q>=allocb);
        !           226: #endif
        !           227: }
        !           228: #endif
        !           229: 

unix.superglobalmegacorp.com

This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.