Annotation of embedaddon/php/ext/dba/libcdb/cdb_make.c, revision 1.1.1.1

1.1       misho       1: /*
                      2:    +----------------------------------------------------------------------+
                      3:    | PHP Version 5                                                        |
                      4:    +----------------------------------------------------------------------+
                      5:    | Copyright (c) 1997-2010 The PHP Group                                |
                      6:    +----------------------------------------------------------------------+
                      7:    | This source file is subject to version 3.01 of the PHP license,      |
                      8:    | that is bundled with this package in the file LICENSE, and is        |
                      9:    | available through the world-wide-web at the following url:           |
                     10:    | http://www.php.net/license/3_01.txt                                  |
                     11:    | If you did not receive a copy of the PHP license and are unable to   |
                     12:    | obtain it through the world-wide-web, please send a note to          |
                     13:    | license@php.net so we can mail you a copy immediately.               |
                     14:    +----------------------------------------------------------------------+
                     15:    | Author: Marcus Boerger <helly@php.net>                               |
                     16:    +----------------------------------------------------------------------+
                     17:  */
                     18: 
                     19: /* $Id: cdb_make.c 293036 2010-01-03 09:23:27Z sebastian $ */
                     20: 
                     21: /* incorporated from D.J.Bernstein's cdb-0.75 (http://cr.yp.to/cdb.html)*/
                     22: 
                     23: #ifdef HAVE_CONFIG_H
                     24: #include "config.h"
                     25: #endif
                     26: 
                     27: #include "php.h"
                     28: 
                     29: #include <sys/types.h>
                     30: #ifdef HAVE_UNISTD_H
                     31: #include <unistd.h>
                     32: #endif
                     33: #include <stdlib.h>
                     34: #include <stdio.h>
                     35: #include <errno.h>
                     36: #include "cdb.h"
                     37: #include "cdb_make.h"
                     38: #include "uint32.h"
                     39: 
                     40: /* {{{ cdb_make_write */
                     41: static int cdb_make_write(struct cdb_make *c, char *buf, uint32 sz TSRMLS_DC) {
                     42:        return php_stream_write(c->fp, buf, sz) == sz ? 0 : -1;
                     43: }
                     44: 
                     45: /* {{{ cdb_posplus */
                     46: static int cdb_posplus(struct cdb_make *c, uint32 len)
                     47: {
                     48:        uint32 newpos = c->pos + len;
                     49:        if (newpos < len) {
                     50:                errno = ENOMEM;
                     51:                return -1;
                     52:        }
                     53:        c->pos = newpos;
                     54:        return 0;
                     55: }
                     56: /* }}} */
                     57: 
                     58: /* {{{ cdb_make_start */
                     59: int cdb_make_start(struct cdb_make *c, php_stream * f TSRMLS_DC)
                     60: {
                     61:        c->head = 0;
                     62:        c->split = 0;
                     63:        c->hash = 0;
                     64:        c->numentries = 0;
                     65:        c->fp = f;
                     66:        c->pos = sizeof(c->final);
                     67:        if (php_stream_seek(f, c->pos, SEEK_SET) == -1) {
                     68:                php_error_docref(NULL TSRMLS_CC, E_NOTICE, "Fseek failed");
                     69:                return -1;
                     70:        }
                     71:        return php_stream_tell(c->fp);
                     72: }
                     73: /* }}} */
                     74: 
                     75: /* {{{ cdb_make_addend */
                     76: int cdb_make_addend(struct cdb_make *c, unsigned int keylen, unsigned int datalen, uint32 h TSRMLS_DC)
                     77: {
                     78:        struct cdb_hplist *head;
                     79: 
                     80:        head = c->head;
                     81:        if (!head || (head->num >= CDB_HPLIST)) {
                     82:                head = (struct cdb_hplist *) emalloc(sizeof(struct cdb_hplist));
                     83:                if (!head)
                     84:                        return -1;
                     85:                head->num = 0;
                     86:                head->next = c->head;
                     87:                c->head = head;
                     88:        }
                     89:        head->hp[head->num].h = h;
                     90:        head->hp[head->num].p = c->pos;
                     91:        ++head->num;
                     92:        ++c->numentries;
                     93:        if (cdb_posplus(c,8) == -1)
                     94:                return -1;
                     95:        if (cdb_posplus(c, keylen) == -1)
                     96:                return -1;
                     97:        if (cdb_posplus(c, datalen) == -1)
                     98:                return -1;
                     99:        return 0;
                    100: }
                    101: /* }}} */
                    102: 
                    103: /* {{{ cdb_make_addbegin */
                    104: int cdb_make_addbegin(struct cdb_make *c, unsigned int keylen, unsigned int datalen TSRMLS_DC)
                    105: {
                    106:        char buf[8];
                    107: 
                    108:        if (keylen > 0xffffffff) {
                    109:                errno = ENOMEM;
                    110:                return -1;
                    111:        }
                    112:        if (datalen > 0xffffffff) {
                    113:                errno = ENOMEM;
                    114:                return -1;
                    115:        }
                    116: 
                    117:        uint32_pack(buf, keylen);
                    118:        uint32_pack(buf + 4, datalen);
                    119:        if (cdb_make_write(c, buf, 8 TSRMLS_CC) != 0)
                    120:                return -1;
                    121:        return 0;
                    122: }
                    123: 
                    124: /* {{{ cdb_make_add */
                    125: int cdb_make_add(struct cdb_make *c,char *key,unsigned int keylen,char *data,unsigned int datalen TSRMLS_DC)
                    126: {
                    127:        if (cdb_make_addbegin(c, keylen, datalen TSRMLS_CC) == -1)
                    128:                return -1;
                    129:        if (cdb_make_write(c, key, keylen TSRMLS_CC) != 0)
                    130:                return -1;
                    131:        if (cdb_make_write(c, data, datalen TSRMLS_CC) != 0)
                    132:                return -1;
                    133:        return cdb_make_addend(c, keylen, datalen, cdb_hash(key, keylen) TSRMLS_CC);
                    134: }
                    135: /* }}} */
                    136: 
                    137: /* {{{ cdb_make_finish */
                    138: int cdb_make_finish(struct cdb_make *c TSRMLS_DC)
                    139: {
                    140:        char buf[8];
                    141:        int i;
                    142:        uint32 len;
                    143:        uint32 u;
                    144:        uint32 memsize;
                    145:        uint32 count;
                    146:        uint32 where;
                    147:        struct cdb_hplist *x;
                    148:        struct cdb_hp *hp;
                    149: 
                    150:        for (i = 0;i < 256;++i)
                    151:        c->count[i] = 0;
                    152: 
                    153:        for (x = c->head; x; x = x->next) {
                    154:                i = x->num;
                    155:                while (i--)
                    156:                ++c->count[255 & x->hp[i].h];
                    157:        }
                    158: 
                    159:        memsize = 1;
                    160:        for (i = 0;i < 256;++i) {
                    161:                u = c->count[i] * 2;
                    162:                if (u > memsize)
                    163:                memsize = u;
                    164:        }
                    165: 
                    166:        memsize += c->numentries; /* no overflow possible up to now */
                    167:        u = (uint32) 0 - (uint32) 1;
                    168:        u /= sizeof(struct cdb_hp);
                    169:        if (memsize > u) {
                    170:                errno = ENOMEM;
                    171:                return -1;
                    172:        }
                    173: 
                    174:        c->split = (struct cdb_hp *) safe_emalloc(memsize, sizeof(struct cdb_hp), 0);
                    175:        if (!c->split)
                    176:                return -1;
                    177: 
                    178:        c->hash = c->split + c->numentries;
                    179: 
                    180:        u = 0;
                    181:        for (i = 0;i < 256;++i) {
                    182:                u += c->count[i]; /* bounded by numentries, so no overflow */
                    183:                c->start[i] = u;
                    184:        }
                    185: 
                    186:        for (x = c->head; x; x = x->next) {
                    187:                i = x->num;
                    188:                while (i--)
                    189:                c->split[--c->start[255 & x->hp[i].h]] = x->hp[i];
                    190:        }
                    191: 
                    192:        for (i = 0;i < 256;++i) {
                    193:                count = c->count[i];
                    194: 
                    195:                len = count + count; /* no overflow possible */
                    196:                uint32_pack(c->final + 8 * i,c->pos);
                    197:                uint32_pack(c->final + 8 * i + 4,len);
                    198: 
                    199:                for (u = 0;u < len;++u)
                    200:                        c->hash[u].h = c->hash[u].p = 0;
                    201: 
                    202:                hp = c->split + c->start[i];
                    203:                for (u = 0;u < count;++u) {
                    204:                        where = (hp->h >> 8) % len;
                    205:                        while (c->hash[where].p)
                    206:                                if (++where == len)
                    207:                                        where = 0;
                    208:                        c->hash[where] = *hp++;
                    209:                }
                    210: 
                    211:                for (u = 0;u < len;++u) {
                    212:                        uint32_pack(buf, c->hash[u].h);
                    213:                        uint32_pack(buf + 4, c->hash[u].p);
                    214:                        if (cdb_make_write(c, buf, 8 TSRMLS_CC) != 0)
                    215:                                return -1;
                    216:                        if (cdb_posplus(c, 8) == -1)
                    217:                                return -1;
                    218:                }
                    219:        }
                    220: 
                    221:        if (c->split)
                    222:                efree(c->split);
                    223: 
                    224:        for (x = c->head; x; c->head = x) {
                    225:                x = x->next;
                    226:                efree(c->head);
                    227:        }
                    228: 
                    229:        if (php_stream_flush(c->fp) != 0)
                    230:                return -1;
                    231:        php_stream_rewind(c->fp);
                    232:        if (php_stream_tell(c->fp) != 0)
                    233:                return -1;
                    234:        if (cdb_make_write(c, c->final, sizeof(c->final) TSRMLS_CC) != 0)
                    235:                return -1;
                    236:        return php_stream_flush(c->fp);
                    237: }
                    238: /* }}} */
                    239: 
                    240: /* {{{ cdb_make_version */
                    241: char *cdb_make_version() 
                    242: {
                    243:        return "0.75, $Revision: 293036 $";
                    244: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>