/****************************************************************************** * zstr.cpp - code for class 'zStr'- a module that reads compressed text * files and provides lookup and parsing functions based on * class StrKey * * * Copyright 2009 CrossWire Bible Society (http://www.crosswire.org) * CrossWire Bible Society * P. O. Box 2528 * Tempe, AZ 85280-2528 * * This program is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License as published by the * Free Software Foundation version 2. * * This program is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * General Public License for more details. * */ #include #include #include #include #include #include #include #include #include #include #include #include #include SWORD_NAMESPACE_START /****************************************************************************** * zStr Statics */ int zStr::instance = 0; const int zStr::IDXENTRYSIZE = 8; const int zStr::ZDXENTRYSIZE = 8; /****************************************************************************** * zStr Constructor - Initializes data for instance of zStr * * ENT: ipath - path of the directory where data and index files are located. */ zStr::zStr(const char *ipath, int fileMode, long blockCount, SWCompress *icomp) { SWBuf buf; nl = '\n'; lastoff = -1; path = 0; stdstr(&path, ipath); compressor = (icomp) ? icomp : new SWCompress(); this->blockCount = blockCount; if (fileMode == -1) { // try read/write if possible fileMode = FileMgr::RDWR; } buf.setFormatted("%s.idx", path); idxfd = FileMgr::getSystemFileMgr()->open(buf, fileMode, true); buf.setFormatted("%s.dat", path); datfd = FileMgr::getSystemFileMgr()->open(buf, fileMode, true); buf.setFormatted("%s.zdx", path); zdxfd = FileMgr::getSystemFileMgr()->open(buf, fileMode, true); buf.setFormatted("%s.zdt", path); zdtfd = FileMgr::getSystemFileMgr()->open(buf, fileMode, true); if (datfd <= 0) { SWLog::getSystemLog()->logError("%d", errno); } cacheBlock = 0; cacheBlockIndex = -1; cacheDirty = false; instance++; } /****************************************************************************** * zStr Destructor - Cleans up instance of zStr */ zStr::~zStr() { flushCache(); if (path) delete [] path; --instance; FileMgr::getSystemFileMgr()->close(idxfd); FileMgr::getSystemFileMgr()->close(datfd); FileMgr::getSystemFileMgr()->close(zdxfd); FileMgr::getSystemFileMgr()->close(zdtfd); if (compressor) delete compressor; } /****************************************************************************** * zStr::getidxbufdat - Gets the index string at the given dat offset * NOTE: buf is calloc'd, or if not null, realloc'd and must * be free'd by calling function * * ENT: ioffset - offset in dat file to lookup * buf - address of pointer to allocate for storage of string */ void zStr::getKeyFromDatOffset(long ioffset, char **buf) const { int size; char ch; if (datfd > 0) { datfd->seek(ioffset, SEEK_SET); for (size = 0; datfd->read(&ch, 1) == 1; size++) { if ((ch == '\\') || (ch == 10) || (ch == 13)) break; } *buf = (*buf) ? (char *)realloc(*buf, size*2 + 1) : (char *)malloc(size*2 + 1); if (size) { datfd->seek(ioffset, SEEK_SET); datfd->read(*buf, size); } (*buf)[size] = 0; toupperstr_utf8(*buf, size*2); } else { *buf = (*buf) ? (char *)realloc(*buf, 1) : (char *)malloc(1); **buf = 0; } } /****************************************************************************** * zStr::getidxbuf - Gets the index string at the given idx offset * NOTE: buf is calloc'd, or if not null, realloc'd * and must be freed by calling function * * ENT: ioffset - offset in idx file to lookup * buf - address of pointer to allocate for storage of string */ void zStr::getKeyFromIdxOffset(long ioffset, char **buf) const { __u32 offset; if (idxfd > 0) { idxfd->seek(ioffset, SEEK_SET); idxfd->read(&offset, 4); offset = swordtoarch32(offset); getKeyFromDatOffset(offset, buf); } } /****************************************************************************** * zStr::findoffset - Finds the offset of the key string from the indexes * * ENT: key - key string to lookup * offset - address to store the starting offset * size - address to store the size of the entry * away - number of entries before of after to jump * (default = 0) * * RET: error status */ signed char zStr::findKeyIndex(const char *ikey, long *idxoff, long away) const { char *maxbuf = 0, *trybuf = 0, *key = 0, quitflag = 0; signed char retval = 0; __s32 headoff, tailoff, tryoff = 0, maxoff = 0; __u32 start, size; int diff = 0; bool awayFromSubstrCheck = false; if (idxfd->getFd() >= 0) { tailoff = maxoff = idxfd->seek(0, SEEK_END) - IDXENTRYSIZE; if (*ikey) { headoff = 0; stdstr(&key, ikey, 3); toupperstr_utf8(key, strlen(key)*3); int keylen = strlen(key); bool substr = false; getKeyFromIdxOffset(maxoff, &maxbuf); while (headoff < tailoff) { tryoff = (lastoff == -1) ? headoff + (((((tailoff / IDXENTRYSIZE) - (headoff / IDXENTRYSIZE))) / 2) * IDXENTRYSIZE) : lastoff; lastoff = -1; getKeyFromIdxOffset(tryoff, &trybuf); if (!*trybuf && tryoff) { // In case of extra entry at end of idx (not first entry) tryoff += (tryoff > (maxoff / 2))?-IDXENTRYSIZE:IDXENTRYSIZE; retval = -1; break; } diff = strcmp(key, trybuf); if (!diff) break; if (!strncmp(trybuf, key, keylen)) substr = true; if (diff < 0) tailoff = (tryoff == headoff) ? headoff : tryoff; else headoff = tryoff; if (tailoff == headoff + IDXENTRYSIZE) { if (quitflag++) headoff = tailoff; } } // didn't find exact match if (headoff >= tailoff) { tryoff = headoff; if (!substr && ((tryoff != maxoff)||(strncmp(key, maxbuf, keylen)<0))) { awayFromSubstrCheck = true; away--; // if our entry doesn't startwith our key, prefer the previous entry over the next } } if (trybuf) free(trybuf); delete [] key; if (maxbuf) free(maxbuf); } else { tryoff = 0; } idxfd->seek(tryoff, SEEK_SET); start = size = 0; retval = (idxfd->read(&start, 4) == 4) ? retval : -1; retval = (idxfd->read(&size, 4) == 4) ? retval : -1; start = swordtoarch32(start); size = swordtoarch32(size); if (idxoff) *idxoff = tryoff; while (away) { __u32 laststart = start; __u32 lastsize = size; __s32 lasttry = tryoff; tryoff += (away > 0) ? IDXENTRYSIZE : -IDXENTRYSIZE; bool bad = false; if (((long)(tryoff + (away*IDXENTRYSIZE)) < -IDXENTRYSIZE) || (tryoff + (away*IDXENTRYSIZE) > (maxoff+IDXENTRYSIZE))) bad = true; else if (idxfd->seek(tryoff, SEEK_SET) < 0) bad = true; if (bad) { if(!awayFromSubstrCheck) retval = -1; start = laststart; size = lastsize; tryoff = lasttry; if (idxoff) *idxoff = tryoff; break; } idxfd->read(&start, 4); idxfd->read(&size, 4); start = swordtoarch32(start); size = swordtoarch32(size); if (idxoff) *idxoff = tryoff; if (((laststart != start) || (lastsize != size)) && (start >= 0) && (size)) away += (away < 0) ? 1 : -1; } lastoff = tryoff; } else { if (idxoff) *idxoff = 0; retval = -1; } return retval; } /****************************************************************************** * zStr::getText - gets text at a given offset * * ENT: * offset - idxoffset where the key is located. * buf - buffer to store text * idxbuf - buffer to store index key * NOTE: buffer will be alloc'd / realloc'd and * should be free'd by the client * */ void zStr::getText(long offset, char **idxbuf, char **buf) { char *ch; char *idxbuflocal = 0; getKeyFromIdxOffset(offset, &idxbuflocal); __u32 start; __u32 size; do { idxfd->seek(offset, SEEK_SET); idxfd->read(&start, 4); idxfd->read(&size, 4); start = swordtoarch32(start); size = swordtoarch32(size); *buf = (*buf) ? (char *)realloc(*buf, size*2 + 1) : (char *)malloc(size*2 + 1); *idxbuf = (*idxbuf) ? (char *)realloc(*idxbuf, size*2 + 1) : (char *)malloc(size*2 + 1); memset(*buf, 0, size + 1); memset(*idxbuf, 0, size + 1); datfd->seek(start, SEEK_SET); datfd->read(*buf, (int)(size)); for (ch = *buf; *ch; ch++) { // skip over index string if (*ch == 10) { ch++; break; } } memmove(*buf, ch, size - (unsigned long)(ch-*buf)); // resolve link if (!strncmp(*buf, "@LINK", 5)) { for (ch = *buf; *ch; ch++) { // null before nl if (*ch == 10) { *ch = 0; break; } } findKeyIndex(*buf + 6, &offset); } else break; } while (true); // while we're resolving links if (idxbuflocal) { __u32 localsize = strlen(idxbuflocal); localsize = (localsize < (size - 1)) ? localsize : (size - 1); strncpy(*idxbuf, idxbuflocal, localsize); (*idxbuf)[localsize] = 0; free(idxbuflocal); } __u32 block = 0; __u32 entry = 0; memmove(&block, *buf, sizeof(__u32)); memmove(&entry, *buf + sizeof(__u32), sizeof(__u32)); block = swordtoarch32(block); entry = swordtoarch32(entry); getCompressedText(block, entry, buf); } /****************************************************************************** * zStr::getCompressedText - Get text entry from a compressed index / zdata * file. */ void zStr::getCompressedText(long block, long entry, char **buf) { __u32 size = 0; if (cacheBlockIndex != block) { __u32 start = 0; zdxfd->seek(block * ZDXENTRYSIZE, SEEK_SET); zdxfd->read(&start, 4); zdxfd->read(&size, 4); start = swordtoarch32(start); size = swordtoarch32(size); SWBuf buf; buf.setSize(size + 5); zdtfd->seek(start, SEEK_SET); zdtfd->read(buf.getRawData(), size); flushCache(); unsigned long len = size; buf.setSize(size); rawZFilter(buf, 0); // 0 = decipher compressor->zBuf(&len, buf.getRawData()); char *rawBuf = compressor->Buf(0, &len); cacheBlock = new EntriesBlock(rawBuf, len); cacheBlockIndex = block; } size = cacheBlock->getEntrySize(entry); *buf = (*buf) ? (char *)realloc(*buf, size*2 + 1) : (char *)malloc(size*2 + 1); strcpy(*buf, cacheBlock->getEntry(entry)); } /****************************************************************************** * zLD::settext - Sets text for current offset * * ENT: key - key for this entry * buf - buffer to store * len - length of buffer (0 - null terminated) */ void zStr::setText(const char *ikey, const char *buf, long len) { __u32 start, outstart; __u32 size, outsize; __s32 endoff; long idxoff = 0; __s32 shiftSize; static const char nl[] = {13, 10}; char *tmpbuf = 0; char *key = 0; char *dbKey = 0; char *idxBytes = 0; char *outbuf = 0; char *ch = 0; len = (len < 0) ? strlen(buf) : len; stdstr(&key, ikey, 3); toupperstr_utf8(key, strlen(key)*3); char notFound = findKeyIndex(ikey, &idxoff, 0); if (!notFound) { getKeyFromIdxOffset(idxoff, &dbKey); int diff = strcmp(key, dbKey); if (diff < 0) { } else if (diff > 0) { idxoff += IDXENTRYSIZE; } else if ((!diff) && (len > 0 /*we're not deleting*/)) { // got absolute entry do { idxfd->seek(idxoff, SEEK_SET); idxfd->read(&start, 4); idxfd->read(&size, 4); start = swordtoarch32(start); size = swordtoarch32(size); tmpbuf = new char [ size + 2 ]; memset(tmpbuf, 0, size + 2); datfd->seek(start, SEEK_SET); datfd->read(tmpbuf, size); for (ch = tmpbuf; *ch; ch++) { // skip over index string if (*ch == 10) { ch++; break; } } memmove(tmpbuf, ch, size - (unsigned long)(ch-tmpbuf)); // resolve link if (!strncmp(tmpbuf, "@LINK", 5) && (len)) { for (ch = tmpbuf; *ch; ch++) { // null before nl if (*ch == 10) { *ch = 0; break; } } findKeyIndex(tmpbuf + IDXENTRYSIZE, &idxoff); delete [] tmpbuf; } else break; } while (true); // while we're resolving links } } endoff = idxfd->seek(0, SEEK_END); shiftSize = endoff - idxoff; if (shiftSize > 0) { idxBytes = new char [ shiftSize ]; idxfd->seek(idxoff, SEEK_SET); idxfd->read(idxBytes, shiftSize); } outbuf = new char [ len + strlen(key) + 5 ]; sprintf(outbuf, "%s%c%c", key, 13, 10); size = strlen(outbuf); if (len > 0) { // NOT a link if (!cacheBlock) { flushCache(); cacheBlock = new EntriesBlock(); cacheBlockIndex = (zdxfd->seek(0, SEEK_END) / ZDXENTRYSIZE); } else if (cacheBlock->getCount() >= blockCount) { flushCache(); cacheBlock = new EntriesBlock(); cacheBlockIndex = (zdxfd->seek(0, SEEK_END) / ZDXENTRYSIZE); } __u32 entry = cacheBlock->addEntry(buf); cacheDirty = true; outstart = archtosword32(cacheBlockIndex); outsize = archtosword32(entry); memcpy (outbuf + size, &outstart, sizeof(__u32)); memcpy (outbuf + size + sizeof(__u32), &outsize, sizeof(__u32)); size += (sizeof(__u32) * 2); } else { // link memcpy(outbuf + size, buf, len); size += len; } start = datfd->seek(0, SEEK_END); outstart = archtosword32(start); outsize = archtosword32(size); idxfd->seek(idxoff, SEEK_SET); if (len > 0) { datfd->seek(start, SEEK_SET); datfd->write(outbuf, size); // add a new line to make data file easier to read in an editor datfd->write(&nl, 2); idxfd->write(&outstart, 4); idxfd->write(&outsize, 4); if (idxBytes) { idxfd->write(idxBytes, shiftSize); } } else { // delete entry if (idxBytes) { idxfd->write(idxBytes+IDXENTRYSIZE, shiftSize-IDXENTRYSIZE); idxfd->seek(-1, SEEK_CUR); // last valid byte FileMgr::getSystemFileMgr()->trunc(idxfd); // truncate index } } if (idxBytes) delete [] idxBytes; delete [] key; delete [] outbuf; free(dbKey); } /****************************************************************************** * zLD::linkentry - links one entry to another * * ENT: testmt - testament to find (0 - Bible/module introduction) * destidxoff - dest offset into .vss * srcidxoff - source offset into .vss */ void zStr::linkEntry(const char *destkey, const char *srckey) { char *text = new char [ strlen(destkey) + 7 ]; sprintf(text, "@LINK %s", destkey); setText(srckey, text); delete [] text; } void zStr::flushCache() { if (cacheBlock) { if (cacheDirty) { __u32 start = 0; unsigned long size = 0; __u32 outstart = 0, outsize = 0; const char *rawBuf = cacheBlock->getRawData(&size); compressor->Buf(rawBuf, &size); compressor->zBuf(&size); SWBuf buf; buf.setSize(size + 5); memcpy(buf.getRawData(), compressor->zBuf(&size), size); // 1 = encipher buf.setSize(size); rawZFilter(buf, 1); // 1 = encipher long zdxSize = zdxfd->seek(0, SEEK_END); unsigned long zdtSize = zdtfd->seek(0, SEEK_END); if ((cacheBlockIndex * ZDXENTRYSIZE) > (zdxSize - ZDXENTRYSIZE)) { // New Block start = zdtSize; } else { zdxfd->seek(cacheBlockIndex * ZDXENTRYSIZE, SEEK_SET); zdxfd->read(&start, 4); zdxfd->read(&outsize, 4); start = swordtoarch32(start); outsize = swordtoarch32(outsize); if (start + outsize >= zdtSize) { // last entry, just overwrite // start is already set } else if (size < outsize) { // middle entry, but smaller, that's fine and let's preserve bigger size size = outsize; } else { // middle and bigger-- we have serious problems, for now let's put it at the end = lots of wasted space start = zdtSize; } } outstart = archtosword32(start); outsize = archtosword32((__u32)size); zdxfd->seek(cacheBlockIndex * ZDXENTRYSIZE, SEEK_SET); zdtfd->seek(start, SEEK_SET); zdtfd->write(buf, size); // add a new line to make data file easier to read in an editor zdtfd->write(&nl, 2); zdxfd->write(&outstart, 4); zdxfd->write(&outsize, 4); } delete cacheBlock; cacheBlock = 0; } cacheBlockIndex = -1; cacheDirty = false; } /****************************************************************************** * zLD::CreateModule - Creates new module files * * ENT: path - directory to store module files * RET: error status */ signed char zStr::createModule(const char *ipath) { char *path = 0; char *buf = new char [ strlen (ipath) + 20 ]; FileDesc *fd, *fd2; stdstr(&path, ipath); if ((path[strlen(path)-1] == '/') || (path[strlen(path)-1] == '\\')) path[strlen(path)-1] = 0; sprintf(buf, "%s.dat", path); FileMgr::removeFile(buf); fd = FileMgr::getSystemFileMgr()->open(buf, FileMgr::CREAT|FileMgr::WRONLY, FileMgr::IREAD|FileMgr::IWRITE); fd->getFd(); FileMgr::getSystemFileMgr()->close(fd); sprintf(buf, "%s.idx", path); FileMgr::removeFile(buf); fd2 = FileMgr::getSystemFileMgr()->open(buf, FileMgr::CREAT|FileMgr::WRONLY, FileMgr::IREAD|FileMgr::IWRITE); fd2->getFd(); FileMgr::getSystemFileMgr()->close(fd2); sprintf(buf, "%s.zdt", path); FileMgr::removeFile(buf); fd2 = FileMgr::getSystemFileMgr()->open(buf, FileMgr::CREAT|FileMgr::WRONLY, FileMgr::IREAD|FileMgr::IWRITE); fd2->getFd(); FileMgr::getSystemFileMgr()->close(fd2); sprintf(buf, "%s.zdx", path); FileMgr::removeFile(buf); fd2 = FileMgr::getSystemFileMgr()->open(buf, FileMgr::CREAT|FileMgr::WRONLY, FileMgr::IREAD|FileMgr::IWRITE); fd2->getFd(); FileMgr::getSystemFileMgr()->close(fd2); delete [] path; return 0; } SWORD_NAMESPACE_END