2 * Copyright (c) 2004 Teodor Sigaev <teodor@sigaev.ru>
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. Neither the name of the author nor the names of any co-contributors
14 * may be used to endorse or promote products derived from this software
15 * without specific prior written permission.
17 * THIS SOFTWARE IS PROVIDED BY CONTRIBUTORS ``AS IS'' AND ANY EXPRESS
18 * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL CONTRIBUTORS BE LIABLE FOR ANY
21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
23 * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
25 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
26 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
27 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
42 findFreeSpace(FDB *db, size_t length) {
43 FDBFreeSpace *ptr = db->space;
45 while(ptr && ptr - db->space < db->listcur) {
46 if ( ptr->length >= length )
55 addFreeSpace(FDB *db, off_t offset, size_t length) {
56 if ( db->listcur >= db->listlen ) {
58 db->space = (FDBFreeSpace*) trealloc( db->space, db->listlen * sizeof(FDBFreeSpace) );
61 db->space[ db->listcur ].offset=offset;
62 db->space[ db->listcur ].length=length;
68 cmpFS(const void* a, const void* b) {
69 if ( ((FDBFreeSpace*)a)->offset == ((FDBFreeSpace*)b)->offset )
71 return ( ((FDBFreeSpace*)a)->offset > ((FDBFreeSpace*)b)->offset ) ? 1 : -1;
75 FDBVacuumFreeSpace(FDB *db) {
76 FDBFreeSpace *ptr=db->space+1, *ok=db->space;
77 if ( db->listcur < 2 )
80 qsort( db->space, db->listcur, sizeof(FDBFreeSpace), cmpFS);
82 while( ptr - db->space < db->listcur ) {
83 tassert( ok->offset + ok->length <= ptr->offset );
84 if ( ok->offset + ok->length == ptr->offset || ptr->length==0 ) {
85 ok->length += ptr->length;
89 memcpy(ok, ptr, sizeof(FDBFreeSpace));
94 db->listcur = ok - db->space + 1;
98 FDBOpen(FDB *db, char *file, int readonly) {
102 memset(db, 0, sizeof(FDB));
106 db->fd = open(file, O_RDONLY | O_SHLOCK);
108 db->fd = open(file, O_CREAT | O_RDWR | O_EXLOCK, 0666);
112 memset(db, 0, sizeof(FDB));
113 tlog(TL_CRIT,"FDBOpen: open failed: %s", strerror(errno));
117 rc = read(db->fd, &header, sizeof(FDBHeader));
121 tlog(TL_CRIT,"FDBOpen: read failed: %s", strerror(errno));
123 } else if ( rc==0 ) {
124 memset(&header, 0, sizeof(FDBHeader));
125 } else if ( rc != sizeof(FDBHeader) ) {
127 tlog(TL_CRIT,"FDBOpen: header fault: %d bytes only", rc);
129 } else if ( header.isopened ) {
131 tlog(TL_CRIT,"FDBOpen: file wasn't closed correctly");
136 if ( !db->readonly ) {
137 if ( header.freespace ) {
138 db->listlen = db->listcur = (header.lenfreespace / sizeof(FDBFreeSpace));
139 db->space = (FDBFreeSpace*)tmalloc( header.lenfreespace + sizeof(FDBFreeSpace) );
141 if ( lseek(db->fd, header.freespace, SEEK_SET)!=header.freespace ||
142 read( db->fd, db->space, header.lenfreespace ) != header.lenfreespace ) {
144 tlog(TL_CRIT,"FDBOpen: free space read failed: %s", strerror(errno));
147 FDBVacuumFreeSpace(db);
150 db->space = (FDBFreeSpace*)tmalloc( db->listlen*sizeof(FDBFreeSpace) );
153 header.freespace = 0;
154 header.lenfreespace = 0;
157 if ( lseek(db->fd, 0, SEEK_SET)!=0 ||
158 write(db->fd, &header, sizeof(FDBHeader)) != sizeof(FDBHeader) ||
161 if ( db->space ) tfree( db->space );
162 tlog(TL_CRIT,"FDBOpen: can't modify header: %s", strerror(errno));
173 if ( !db->readonly) {
176 memset(&header,0,sizeof(FDBHeader));
181 FDBVacuumFreeSpace(db);
183 header.lenfreespace = sizeof(FDBFreeSpace)*db->listcur;
184 ptr = findFreeSpace( db, header.lenfreespace );
187 header.freespace = ptr->offset;
188 if ( lseek(db->fd, ptr->offset, SEEK_SET) != ptr->offset )
189 tlog(TL_CRIT|TL_EXIT,"FDBClose: lseek failed: %s", strerror(errno));
191 if ( (header.freespace = lseek(db->fd, 0, SEEK_END)) < 0 )
192 tlog(TL_CRIT|TL_EXIT,"FDBClose: lseek failed: %s", strerror(errno));
193 header.lenfreespace += sizeof(FDBFreeSpace);
194 addFreeSpace(db, header.freespace, header.lenfreespace);
197 if ( write(db->fd, db->space, header.lenfreespace) != header.lenfreespace )
198 tlog(TL_CRIT|TL_EXIT,"FDBClose: write failed: %s", strerror(errno));
203 if ( lseek(db->fd,0,SEEK_SET)!=0 ||
204 write(db->fd, &header, sizeof(FDBHeader)) != sizeof(FDBHeader) ||
206 tlog(TL_CRIT|TL_EXIT,"FDBClose: header write failed: %s", strerror(errno));
218 readLen(FDB *db, off_t offset, size_t *size) {
219 if ( lseek(db->fd,offset,SEEK_SET)!=offset)
222 if ( read(db->fd,size,sizeof(size_t)) != sizeof(size_t) )
230 FDBDelete(FDB *db, off_t offset, size_t length) {
235 if ( readLen(db, offset, &length) != FDB_OK )
238 addFreeSpace(db, offset, length);
245 FDBGet(FDB *db, off_t offset, size_t length, FDBRecord **record) {
254 if ( readLen(db, offset, &length) != FDB_OK )
257 *record = (FDBRecord*)tmalloc( length );
259 if ( lseek(db->fd,offset,SEEK_SET)!=offset)
262 if ( (rc=read(db->fd,*record,length)) != length ) {
263 (*record)->length = rc;
264 tlog(TL_CRIT,"FDBGet: read (%d bytes) less than needed (%d bytes): %s", rc, length, strerror(errno));
265 return FDB_INCORRECT;
268 if ( (*record)->length != length ) {
269 tlog(TL_ALARM, "FDBGet: wrong length in opts: %d bytes and %d bytes really", length, (*record)->length);
270 if ( (*record)->length > length ) {
271 rc = (*record)->length;
273 return FDBGet(db, offset, rc, record);
282 FDBPut(FDB *db, FDBRecord *record, off_t *offset ) {
288 ptr = findFreeSpace( db, record->length );
290 *offset = ptr->offset;
291 ptr->offset += record->length;
292 ptr->length -= record->length;
293 if ( ptr->length == 0 )
294 FDBVacuumFreeSpace(db);
295 if ( lseek(db->fd, *offset, SEEK_SET) != *offset )
296 tlog(TL_CRIT|TL_EXIT,"FDBPut: lseek failed: %s", strerror(errno));
298 if ( (*offset = lseek(db->fd, 0, SEEK_END)) < 0 )
299 tlog(TL_CRIT|TL_EXIT,"FDBPut: lseek failed: %s", strerror(errno));
302 if ( write(db->fd, record, record->length) != record->length )
303 tlog(TL_CRIT|TL_EXIT,"FDBPut: write failed: %s", strerror(errno));