Happy new year.
[idzebra-moved-to-github.git] / dict / open.c
1 /* This file is part of the Zebra server.
2    Copyright (C) 1994-2011 Index Data
3
4    Zebra is free software; you can redistribute it and/or modify it under
5    the terms of the GNU General Public License as published by the Free
6    Software Foundation; either version 2, or (at your option) any later
7    version.
8
9    Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
10    WARRANTY; without even the implied warranty of MERCHANTABILITY or
11    FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12    for more details.
13
14    You should have received a copy of the GNU General Public License
15    along with this program; if not, write to the Free Software
16    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17
18 */
19
20
21
22 #include <stdlib.h>
23 #include <string.h>
24 #include <stdio.h>
25
26 #include "dict-p.h"
27
28 void dict_clean(Dict dict)
29 {
30     int page_size = dict->head.page_size;
31     void *head_buf;
32     int compact_flag = dict->head.compact_flag;
33
34     memset(dict->head.magic_str, 0, sizeof(dict->head.magic_str));
35     strcpy(dict->head.magic_str, DICT_MAGIC);
36     dict->head.last = 1;
37     dict->head.root = 0;
38     dict->head.freelist = 0;
39     dict->head.page_size = page_size;
40     dict->head.compact_flag = compact_flag;
41     
42     /* create header with information (page 0) */
43     if (dict->rw) 
44         dict_bf_newp(dict->dbf, 0, &head_buf, page_size);
45 }
46
47 Dict dict_open(BFiles bfs, const char *name, int cache, int rw,
48                int compact_flag, int page_size)
49 {
50     Dict dict;
51     void *head_buf;
52
53     dict = (Dict) xmalloc(sizeof(*dict));
54
55     if (cache < 5)
56         cache = 5;
57
58     dict->grep_cmap = NULL;
59     page_size = DICT_DEFAULT_PAGESIZE;
60     if (page_size < 2048)
61     {
62         yaz_log(YLOG_WARN, "Page size for dict %s %d<2048. Set to 2048",
63                 name, page_size);
64         page_size = 2048;
65     }
66     dict->dbf = dict_bf_open(bfs, name, page_size, cache, rw);
67     dict->rw = rw;
68     dict->no_split = 0;
69     dict->no_insert = 0;
70     dict->no_lookup = 0;
71
72     if(!dict->dbf)
73     {
74         yaz_log(YLOG_WARN, "Cannot open `%s'", name);
75         xfree(dict);
76         return NULL;
77     }
78     if (dict_bf_readp(dict->dbf, 0, &head_buf) <= 0)
79     {
80         dict->head.page_size = page_size;
81         dict->head.compact_flag = compact_flag;
82         dict_clean(dict);
83     }
84     else /* header was there, check magic and page size */
85     {
86         memcpy(&dict->head, head_buf, sizeof(dict->head));
87         if (strcmp(dict->head.magic_str, DICT_MAGIC))
88         {
89             yaz_log(YLOG_WARN, "Bad magic of `%s'", name);
90             dict_bf_close(dict->dbf);
91             xfree(dict);
92             return 0;
93         }
94         if (dict->head.page_size != page_size)
95         {
96             yaz_log(YLOG_WARN, "Page size for existing dict %s is %d. Current is %d",
97                     name, dict->head.page_size, page_size);
98         }
99     }
100     if (dict->head.compact_flag)
101         dict_bf_compact(dict->dbf);
102     return dict;
103 }
104
105 int dict_strcmp(const Dict_char *s1, const Dict_char *s2)
106 {
107     return strcmp((const char *) s1, (const char *) s2);
108 }
109
110 int dict_strncmp(const Dict_char *s1, const Dict_char *s2, size_t n)
111 {
112     return strncmp((const char *) s1, (const char *) s2, n);
113 }
114
115 int dict_strlen(const Dict_char *s)
116 {
117     return strlen((const char *) s);
118 }
119
120 zint dict_get_no_lookup(Dict dict)
121 {
122     return dict->no_lookup;
123 }
124
125 zint dict_get_no_insert(Dict dict)
126 {
127     return dict->no_insert;
128 }
129
130 zint dict_get_no_split(Dict dict)
131 {
132     return dict->no_split;
133 }
134
135 /*
136  * Local variables:
137  * c-basic-offset: 4
138  * c-file-style: "Stroustrup"
139  * indent-tabs-mode: nil
140  * End:
141  * vim: shiftwidth=4 tabstop=8 expandtab
142  */
143