-static ZEBRA_RES rec_release_blocks(Records p, SYSNO sysno)
-{
- struct record_index_entry entry;
- zint freeblock;
- char block_and_ref[sizeof(zint) + sizeof(short)];
- int dst_type;
- int first = 1;
-
- if (read_indx(p, sysno, &entry, sizeof(entry), 1) != 1)
- return ZEBRA_FAIL;
-
- freeblock = entry.next;
- assert(freeblock > 0);
- dst_type = (int) (freeblock & 7);
- assert(dst_type < REC_BLOCK_TYPES);
- freeblock = freeblock / 8;
- while (freeblock)
- {
- if (bf_read(p->data_BFile[dst_type], freeblock, 0,
- first ? sizeof(block_and_ref) : sizeof(zint),
- block_and_ref) != 1)
- {
- yaz_log(YLOG_FATAL|YLOG_ERRNO, "read in rec_del_single");
- return ZEBRA_FAIL;
- }
- if (first)
- {
- short ref;
- memcpy(&ref, block_and_ref + sizeof(freeblock), sizeof(ref));
- --ref;
- memcpy(block_and_ref + sizeof(freeblock), &ref, sizeof(ref));
- if (ref)
- {
- if (bf_write(p->data_BFile[dst_type], freeblock, 0,
- sizeof(block_and_ref), block_and_ref))
- {
- yaz_log(YLOG_FATAL|YLOG_ERRNO, "write in rec_del_single");
- return ZEBRA_FAIL;
- }
- return ZEBRA_OK;
- }
- first = 0;
- }
-
- if (bf_write(p->data_BFile[dst_type], freeblock, 0, sizeof(freeblock),
- &p->head.block_free[dst_type]))
- {
- yaz_log(YLOG_FATAL|YLOG_ERRNO, "write in rec_del_single");
- return ZEBRA_FAIL;
- }
- p->head.block_free[dst_type] = freeblock;
- memcpy(&freeblock, block_and_ref, sizeof(freeblock));
-
- p->head.block_used[dst_type]--;
- }
- p->head.total_bytes -= entry.size;
- return ZEBRA_OK;
-}
-
-static ZEBRA_RES rec_delete_single(Records p, Record rec)
-{
- struct record_index_entry entry;
-
- if (rec_release_blocks(p, rec_sysno_to_int(rec->sysno)) != ZEBRA_OK)
- return ZEBRA_FAIL;
-
- entry.next = p->head.index_free;
- entry.size = 0;
- p->head.index_free = rec_sysno_to_int(rec->sysno);
- write_indx(p, rec_sysno_to_int(rec->sysno), &entry, sizeof(entry));
- return ZEBRA_OK;
-}
-
-static ZEBRA_RES rec_write_tmp_buf(Records p, int size, SYSNO *sysnos)
-{
- struct record_index_entry entry;
- int no_written = 0;
- char *cptr = p->tmp_buf;
- zint block_prev = -1, block_free;
- int dst_type = 0;
- int i;
-
- for (i = 1; i<REC_BLOCK_TYPES; i++)
- if (size >= p->head.block_move[i])
- dst_type = i;
- while (no_written < size)
- {
- block_free = p->head.block_free[dst_type];
- if (block_free)
- {
- if (bf_read(p->data_BFile[dst_type],
- block_free, 0, sizeof(*p->head.block_free),
- &p->head.block_free[dst_type]) != 1)
- {
- yaz_log(YLOG_FATAL|YLOG_ERRNO, "read in %s at free block "
- ZINT_FORMAT,
- p->data_fname[dst_type], block_free);
- return ZEBRA_FAIL;
- }
- }
- else
- block_free = p->head.block_last[dst_type]++;
- if (block_prev == -1)
- {
- entry.next = block_free*8 + dst_type;
- entry.size = size;
- p->head.total_bytes += size;
- while (*sysnos > 0)
- {
- write_indx(p, *sysnos, &entry, sizeof(entry));
- sysnos++;
- }
- }
- else
- {
- memcpy(cptr, &block_free, sizeof(block_free));
- bf_write(p->data_BFile[dst_type], block_prev, 0, 0, cptr);
- cptr = p->tmp_buf + no_written;
- }
- block_prev = block_free;
- no_written += (int)(p->head.block_size[dst_type]) - sizeof(zint);
- p->head.block_used[dst_type]++;
- }
- assert(block_prev != -1);
- block_free = 0;
- memcpy(cptr, &block_free, sizeof(block_free));
- bf_write(p->data_BFile[dst_type], block_prev, 0,
- sizeof(block_free) + (p->tmp_buf+size) - cptr, cptr);
- return ZEBRA_OK;
-}
-
-Records rec_open(BFiles bfs, int rw, int compression_method)
-{
- Records p;
- int i, r;
- int version;
- ZEBRA_RES ret = ZEBRA_OK;
-
- p = (Records) xmalloc(sizeof(*p));
- p->compression_method = compression_method;
- p->rw = rw;
- p->tmp_size = 1024;
- p->index_fname = "reci";
- p->index_BFile = bf_open(bfs, p->index_fname, RIDX_CHUNK, rw);
- if (p->index_BFile == NULL)
- {
- yaz_log(YLOG_FATAL|YLOG_ERRNO, "open %s", p->index_fname);
- xfree(p);
- return 0;
- }
- p->tmp_buf = (char *) xmalloc(p->tmp_size);
- r = bf_read(p->index_BFile, 0, 0, 0, p->tmp_buf);
- switch (r)
- {
- case 0:
- memcpy(p->head.magic, REC_HEAD_MAGIC, sizeof(p->head.magic));
- sprintf(p->head.version, "%3d", REC_VERSION);
- p->head.index_free = 0;
- p->head.index_last = 1;
- p->head.no_records = 0;
- p->head.total_bytes = 0;
- for (i = 0; i<REC_BLOCK_TYPES; i++)
- {
- p->head.block_free[i] = 0;
- p->head.block_last[i] = 1;
- p->head.block_used[i] = 0;
- }
- p->head.block_size[0] = 128;
- p->head.block_move[0] = 0;
- for (i = 1; i<REC_BLOCK_TYPES; i++)
- {
- p->head.block_size[i] = p->head.block_size[i-1] * 4;
- p->head.block_move[i] = p->head.block_size[i] * 24;
- }
- if (rw)
- {
- if (rec_write_head(p) != ZEBRA_OK)
- ret = ZEBRA_FAIL;
- }
- break;
- case 1:
- memcpy(&p->head, p->tmp_buf, sizeof(p->head));
- if (memcmp(p->head.magic, REC_HEAD_MAGIC, sizeof(p->head.magic)))
- {
- yaz_log(YLOG_FATAL, "file %s has bad format", p->index_fname);
- ret = ZEBRA_FAIL;
- }
- version = atoi(p->head.version);
- if (version != REC_VERSION)
- {
- yaz_log(YLOG_FATAL, "file %s is version %d, but version"
- " %d is required", p->index_fname, version, REC_VERSION);
- ret = ZEBRA_FAIL;
- }
- break;
- }
- for (i = 0; i<REC_BLOCK_TYPES; i++)
- {
- char str[80];
- sprintf(str, "recd%c", i + 'A');
- p->data_fname[i] = (char *) xmalloc(strlen(str)+1);
- strcpy(p->data_fname[i], str);
- p->data_BFile[i] = NULL;
- }
- for (i = 0; i<REC_BLOCK_TYPES; i++)
- {
- if (!(p->data_BFile[i] = bf_open(bfs, p->data_fname[i],
- (int) (p->head.block_size[i]),
- rw)))
- {
- yaz_log(YLOG_FATAL|YLOG_ERRNO, "bf_open %s", p->data_fname[i]);
- ret = ZEBRA_FAIL;
- }
- }
- p->cache_max = 400;
- p->cache_cur = 0;
- p->record_cache = (struct record_cache_entry *)
- xmalloc(sizeof(*p->record_cache)*p->cache_max);
- zebra_mutex_init(&p->mutex);
- if (ret == ZEBRA_FAIL)
- rec_close(&p);
- return p;
-}
-
-static void rec_encode_unsigned(unsigned n, unsigned char *buf, int *len)
-{
- (*len) = 0;
- while (n > 127)
- {
- buf[*len] = 128 + (n & 127);
- n = n >> 7;
- (*len)++;
- }
- buf[*len] = n;
- (*len)++;
-}
-
-static void rec_decode_unsigned(unsigned *np, unsigned char *buf, int *len)
-{
- unsigned n = 0;
- unsigned w = 1;
- (*len) = 0;
-
- while (buf[*len] > 127)
- {
- n += w*(buf[*len] & 127);
- w = w << 7;
- (*len)++;
- }
- n += w * buf[*len];
- (*len)++;
- *np = n;
-}
-
-static void rec_encode_zint(zint n, unsigned char *buf, int *len)
-{
- (*len) = 0;
- while (n > 127)
- {
- buf[*len] = (unsigned) (128 + (n & 127));
- n = n >> 7;
- (*len)++;
- }
- buf[*len] = (unsigned) n;
- (*len)++;
-}
-
-static void rec_decode_zint(zint *np, unsigned char *buf, int *len)
-{
- zint n = 0;
- zint w = 1;
- (*len) = 0;
-
- while (buf[*len] > 127)
- {
- n += w*(buf[*len] & 127);
- w = w << 7;
- (*len)++;
- }
- n += w * buf[*len];
- (*len)++;
- *np = n;
-}
-
-static void rec_cache_flush_block1(Records p, Record rec, Record last_rec,
- char **out_buf, int *out_size,
- int *out_offset)
-{
- int i;
- int len;
-
- for (i = 0; i<REC_NO_INFO; i++)
- {
- if (*out_offset + (int) rec->size[i] + 20 > *out_size)
- {
- int new_size = *out_offset + rec->size[i] + 65536;
- char *np = (char *) xmalloc(new_size);
- if (*out_offset)
- memcpy(np, *out_buf, *out_offset);
- xfree(*out_buf);
- *out_size = new_size;
- *out_buf = np;
- }
- if (i == 0)
- {
- rec_encode_zint(rec_sysno_to_int(rec->sysno),
- (unsigned char *) *out_buf + *out_offset, &len);
- (*out_offset) += len;
- }
- if (rec->size[i] == 0)
- {
- rec_encode_unsigned(1, (unsigned char *) *out_buf + *out_offset,
- &len);
- (*out_offset) += len;
- }
- else if (last_rec && rec->size[i] == last_rec->size[i] &&
- !memcmp(rec->info[i], last_rec->info[i], rec->size[i]))
- {
- rec_encode_unsigned(0, (unsigned char *) *out_buf + *out_offset,
- &len);
- (*out_offset) += len;
- }
- else
- {
- rec_encode_unsigned(rec->size[i]+1,
- (unsigned char *) *out_buf + *out_offset,
- &len);
- (*out_offset) += len;
- memcpy(*out_buf + *out_offset, rec->info[i], rec->size[i]);
- (*out_offset) += rec->size[i];
- }
- }
-}
-
-static ZEBRA_RES rec_write_multiple(Records p, int saveCount)
-{
- int i;
- short ref_count = 0;
- char compression_method;
- Record last_rec = 0;
- int out_size = 1000;
- int out_offset = 0;
- char *out_buf = (char *) xmalloc(out_size);
- SYSNO *sysnos = (SYSNO *) xmalloc(sizeof(*sysnos) * (p->cache_cur + 1));
- SYSNO *sysnop = sysnos;
- ZEBRA_RES ret = ZEBRA_OK;
-
- for (i = 0; i<p->cache_cur - saveCount; i++)
- {
- struct record_cache_entry *e = p->record_cache + i;
- switch (e->flag)
- {
- case recordFlagNew:
- rec_cache_flush_block1(p, e->rec, last_rec, &out_buf,
- &out_size, &out_offset);
- *sysnop++ = rec_sysno_to_int(e->rec->sysno);
- ref_count++;
- e->flag = recordFlagNop;
- last_rec = e->rec;
- break;
- case recordFlagWrite:
- if (rec_release_blocks(p, rec_sysno_to_int(e->rec->sysno))
- != ZEBRA_OK)
- ret = ZEBRA_FAIL;
-
- rec_cache_flush_block1(p, e->rec, last_rec, &out_buf,
- &out_size, &out_offset);
- *sysnop++ = rec_sysno_to_int(e->rec->sysno);
- ref_count++;
- e->flag = recordFlagNop;
- last_rec = e->rec;
- break;
- case recordFlagDelete:
- if (rec_delete_single(p, e->rec) != ZEBRA_OK)
- ret = ZEBRA_FAIL;
-
- e->flag = recordFlagNop;
- break;
- default:
- break;
- }
- }
-
- *sysnop = -1;
- if (ref_count)
- {
- unsigned int csize = 0; /* indicate compression "not performed yet" */
- compression_method = p->compression_method;
- switch (compression_method)
- {
- case REC_COMPRESS_BZIP2:
-#if HAVE_BZLIB_H
- csize = out_offset + (out_offset >> 6) + 620;
- rec_tmp_expand(p, csize);
-#ifdef BZ_CONFIG_ERROR
- i = BZ2_bzBuffToBuffCompress
-#else
- i = bzBuffToBuffCompress
-#endif
- (p->tmp_buf+sizeof(zint)+sizeof(short)+
- sizeof(char),
- &csize, out_buf, out_offset, 1, 0, 30);
- if (i != BZ_OK)
- {
- yaz_log(YLOG_WARN, "bzBuffToBuffCompress error code=%d", i);
- csize = 0;
- }
- yaz_log(YLOG_LOG, "compress %4d %5d %5d", ref_count, out_offset,
- csize);
-#endif
- break;
- case REC_COMPRESS_NONE:
- break;
- }
- if (!csize)
- {
- /* either no compression or compression not supported ... */
- csize = out_offset;
- rec_tmp_expand(p, csize);
- memcpy(p->tmp_buf + sizeof(zint) + sizeof(short) + sizeof(char),
- out_buf, out_offset);
- csize = out_offset;
- compression_method = REC_COMPRESS_NONE;
- }
- memcpy(p->tmp_buf + sizeof(zint), &ref_count, sizeof(ref_count));
- memcpy(p->tmp_buf + sizeof(zint)+sizeof(short),
- &compression_method, sizeof(compression_method));
-
- /* -------- compression */
- if (rec_write_tmp_buf(p, csize + sizeof(short) + sizeof(char), sysnos)
- != ZEBRA_OK)
- ret = ZEBRA_FAIL;
- }
- xfree(out_buf);
- xfree(sysnos);
- return ret;
-}
-
-static ZEBRA_RES rec_cache_flush(Records p, int saveCount)
-{
- int i, j;
- ZEBRA_RES ret;
-
- if (saveCount >= p->cache_cur)
- saveCount = 0;
-
- ret = rec_write_multiple(p, saveCount);