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.
35 #include <sys/types.h>
45 findFreeSpace(FDB *db, size_t length) {
46 FDBFreeSpace *ptr = db->space;
48 while(ptr && ptr - db->space < db->listcur) {
49 if ( ptr->length >= length )
58 addFreeSpace(FDB *db, off_t offset, size_t length) {
59 if ( db->listcur >= db->listlen ) {
61 db->space = (FDBFreeSpace*) trealloc( db->space, db->listlen * sizeof(FDBFreeSpace) );
64 db->space[ db->listcur ].offset=offset;
65 db->space[ db->listcur ].length=length;
71 cmpFS(const void* a, const void* b) {
72 if ( ((FDBFreeSpace*)a)->offset == ((FDBFreeSpace*)b)->offset )
74 return ( ((FDBFreeSpace*)a)->offset > ((FDBFreeSpace*)b)->offset ) ? 1 : -1;
78 FDBVacuumFreeSpace(FDB *db) {
79 FDBFreeSpace *ptr=db->space+1, *ok=db->space;
81 if ( db->listcur < 2 )
84 qsort( db->space, db->listcur, sizeof(FDBFreeSpace), cmpFS);
87 while( ptr - db->space < db->listcur ) {
88 if ( ok->offset + ok->length == ptr->offset || ptr->length==0 ) {
89 ok->length += ptr->length;
96 /* remove void spaces */
98 while( ptr - db->space < db->listcur ) {
99 if ( ptr->length != 0 ) {
101 memcpy(ok,ptr,sizeof(FDBFreeSpace));
106 db->listcur = ok - db->space;
110 FDBOpen(FDB *db, char *file, int readonly) {
114 memset(db, 0, sizeof(FDB));
118 db->fd = open(file, O_RDONLY);
119 if ( flock( db->fd, LOCK_SH ) < 0 ) {
120 tlog(TL_CRIT,"FDBOpen: flock failed: %s",strerror(errno));
125 db->fd = open(file, O_CREAT | O_RDWR, 0666);
126 if ( flock( db->fd, LOCK_EX ) < 0 ) {
127 tlog(TL_CRIT,"FDBOpen: flock failed: %s",strerror(errno));
134 memset(db, 0, sizeof(FDB));
135 tlog(TL_CRIT,"FDBOpen: open failed: %s", strerror(errno));
139 rc = read(db->fd, &header, sizeof(FDBHeader));
142 flock( db->fd, LOCK_UN );
144 tlog(TL_CRIT,"FDBOpen: read failed: %s", strerror(errno));
146 } else if ( rc==0 ) {
147 memset(&header, 0, sizeof(FDBHeader));
148 } else if ( rc != sizeof(FDBHeader) ) {
149 flock( db->fd, LOCK_UN );
151 tlog(TL_CRIT,"FDBOpen: header fault: %d bytes only", rc);
153 } else if ( header.isopened ) {
154 flock( db->fd, LOCK_UN );
156 tlog(TL_CRIT,"FDBOpen: file wasn't closed correctly");
161 if ( !db->readonly ) {
162 if ( header.freespace ) {
163 db->listlen = db->listcur = (header.lenfreespace / sizeof(FDBFreeSpace));
164 db->space = (FDBFreeSpace*)tmalloc( header.lenfreespace + sizeof(FDBFreeSpace) );
166 if ( lseek(db->fd, header.freespace, SEEK_SET)!=header.freespace ||
167 read( db->fd, db->space, header.lenfreespace ) != header.lenfreespace ) {
168 flock( db->fd, LOCK_UN );
170 tlog(TL_CRIT,"FDBOpen: free space read failed: %s", strerror(errno));
173 FDBVacuumFreeSpace(db);
176 db->space = (FDBFreeSpace*)tmalloc( db->listlen*sizeof(FDBFreeSpace) );
179 header.freespace = 0;
180 header.lenfreespace = 0;
183 if ( lseek(db->fd, 0, SEEK_SET)!=0 ||
184 write(db->fd, &header, sizeof(FDBHeader)) != sizeof(FDBHeader) ||
186 flock( db->fd, LOCK_UN );
188 if ( db->space ) tfree( db->space );
189 tlog(TL_CRIT,"FDBOpen: can't modify header: %s", strerror(errno));
200 if ( !db->readonly) {
203 memset(&header,0,sizeof(FDBHeader));
208 FDBVacuumFreeSpace(db);
210 header.lenfreespace = sizeof(FDBFreeSpace)*db->listcur;
211 ptr = findFreeSpace( db, header.lenfreespace );
214 header.freespace = ptr->offset;
215 if ( lseek(db->fd, ptr->offset, SEEK_SET) != ptr->offset )
216 tlog(TL_CRIT|TL_EXIT,"FDBClose: lseek failed: %s", strerror(errno));
218 if ( (header.freespace = lseek(db->fd, 0, SEEK_END)) < 0 )
219 tlog(TL_CRIT|TL_EXIT,"FDBClose: lseek failed: %s", strerror(errno));
220 header.lenfreespace += sizeof(FDBFreeSpace);
221 addFreeSpace(db, header.freespace, header.lenfreespace);
224 if ( write(db->fd, db->space, header.lenfreespace) != header.lenfreespace )
225 tlog(TL_CRIT|TL_EXIT,"FDBClose: write failed: %s", strerror(errno));
230 if ( lseek(db->fd,0,SEEK_SET)!=0 ||
231 write(db->fd, &header, sizeof(FDBHeader)) != sizeof(FDBHeader) ||
233 tlog(TL_CRIT|TL_EXIT,"FDBClose: header write failed: %s", strerror(errno));
236 flock( db->fd, LOCK_UN );
246 readLen(FDB *db, off_t offset, size_t *size) {
247 if ( lseek(db->fd,offset,SEEK_SET)!=offset)
250 if ( read(db->fd,size,sizeof(size_t)) != sizeof(size_t) )
258 FDBDelete(FDB *db, off_t offset, size_t length) {
263 if ( readLen(db, offset, &length) != FDB_OK )
266 addFreeSpace(db, offset, length);
273 FDBGet(FDB *db, off_t offset, size_t length, FDBRecord **record) {
278 if ( offset < sizeof(FDBHeader) )
282 if ( readLen(db, offset, &length) != FDB_OK )
285 *record = (FDBRecord*)tmalloc( length );
287 if ( lseek(db->fd,offset,SEEK_SET)!=offset)
290 if ( (rc=read(db->fd,*record,length)) != length ) {
291 (*record)->length = rc;
292 tlog(TL_CRIT,"FDBGet: read (%d bytes) less than needed (%d bytes): %s", rc, length, strerror(errno));
293 return FDB_INCORRECT;
296 if ( (*record)->length != length ) {
297 tlog(TL_ALARM, "FDBGet: wrong length in opts: %d bytes and %d bytes really", length, (*record)->length);
298 if ( (*record)->length > length ) {
299 rc = (*record)->length;
301 return FDBGet(db, offset, rc, record);
310 FDBPut(FDB *db, FDBRecord *record, off_t *offset ) {
316 ptr = findFreeSpace( db, record->length );
318 *offset = ptr->offset;
319 ptr->length -= record->length;
320 if ( ptr->length == 0 ) {
321 if ( (ptr - db->space) + 1 != db->listcur )
322 memmove(ptr, ptr+1, (db->listcur - (ptr - db->space) + 1) * sizeof(FDBFreeSpace));
325 ptr->offset += record->length;
326 if ( lseek(db->fd, *offset, SEEK_SET) != *offset )
327 tlog(TL_CRIT|TL_EXIT,"FDBPut: lseek failed: %s", strerror(errno));
329 if ( (*offset = lseek(db->fd, 0, SEEK_END)) < 0 )
330 tlog(TL_CRIT|TL_EXIT,"FDBPut: lseek failed: %s", strerror(errno));
333 if ( write(db->fd, record, record->length) != record->length )
334 tlog(TL_CRIT|TL_EXIT,"FDBPut: write failed: %s", strerror(errno));