Mercurial > notdcc
diff srvrlib/read_rcd.c @ 0:c7f6b056b673
First import of vendor version
author | Peter Gervai <grin@grin.hu> |
---|---|
date | Tue, 10 Mar 2009 13:49:58 +0100 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/srvrlib/read_rcd.c Tue Mar 10 13:49:58 2009 +0100 @@ -0,0 +1,190 @@ +/* Distributed Checksum Clearinghouse + * + * Copyright (c) 2008 by Rhyolite Software, LLC + * + * This agreement is not applicable to any entity which sells anti-spam + * solutions to others or provides an anti-spam solution as part of a + * security solution sold to other entities, or to a private network + * which employs the DCC or uses data provided by operation of the DCC + * but does not provide corresponding data to other users. + * + * Permission to use, copy, modify, and distribute this software without + * changes for any purpose with or without fee is hereby granted, provided + * that the above copyright notice and this permission notice appear in all + * copies and any distributed versions or copies are either unchanged + * or not called anything similar to "DCC" or "Distributed Checksum + * Clearinghouse". + * + * Parties not eligible to receive a license under this agreement can + * obtain a commercial license to use DCC by contacting Rhyolite Software + * at sales@rhyolite.com. + * + * A commercial license would be for Distributed Checksum and Reputation + * Clearinghouse software. That software includes additional features. This + * free license for Distributed ChecksumClearinghouse Software does not in any + * way grant permision to use Distributed Checksum and Reputation Clearinghouse + * software + * + * THE SOFTWARE IS PROVIDED "AS IS" AND RHYOLITE SOFTWARE, LLC DISCLAIMS ALL + * WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES + * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL RHYOLITE SOFTWARE, LLC + * BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES + * OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, + * WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, + * ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS + * SOFTWARE. + * + * Rhyolite Software DCC 1.3.103-1.24 $Revision$ + */ + +#include "srvr_defs.h" + +static int read_rcd_fd = -1; +#define BUF_SIZE (64*1024) +static u_char read_rcd_buf[BUF_SIZE]; +static u_int read_rcd_bufsize = BUF_SIZE; +static off_t read_rcd_base; +static u_int read_rcd_buflen; + +static int system_pagesize; + + + +int +read_db(DCC_EMSG emsg, void *buf, u_int buf_len, + int fd, off_t pos, const char *file_nm) +{ + int i; + + if (-1 == lseek(fd, pos, SEEK_SET)) { + dcc_pemsg(EX_IOERR, emsg, "lseek(%s, 0): %s", + DB_NM2PATH_ERR(file_nm), ERROR_STR()); + return -1; + } + + i = read(fd, buf, buf_len); + if (i >= 0) + return i; + + dcc_pemsg(EX_IOERR, emsg, "read(%s): %s", + DB_NM2PATH_ERR(file_nm), ERROR_STR()); + return -1; +} + + + +u_char +read_db_hdr(DCC_EMSG emsg, DB_HDR *hdrp, + int fd, const char *file_nm) +{ + return (sizeof(*hdrp) == read_db(emsg, hdrp, sizeof(*hdrp), + fd, 0, file_nm)); +} + + + +/* invalidate the read_rcd() read-ahead cache */ +void +read_rcd_invalidate(u_int bufsize) +{ + read_rcd_fd = -1; + + if (system_pagesize == 0) + system_pagesize = getpagesize(); + bufsize = ((bufsize + system_pagesize-1) / system_pagesize + * system_pagesize); + if (bufsize == 0 || bufsize > BUF_SIZE) + bufsize = BUF_SIZE; + read_rcd_bufsize = bufsize; +} + + + +static int +read_rcd_sub(DCC_EMSG emsg, void *buf, u_int buflen, + int fd, off_t pos, const char *file_nm) +{ + int i; + + if (fd == read_rcd_fd + && pos >= read_rcd_base + && pos+buflen <= read_rcd_base+read_rcd_buflen) { + memcpy(buf, &read_rcd_buf[pos-read_rcd_base], buflen); + return buflen; + } + + if (system_pagesize == 0) + system_pagesize = getpagesize(); + read_rcd_base = pos - (pos % system_pagesize); + + i = read_db(emsg, read_rcd_buf, read_rcd_bufsize, + read_rcd_fd = fd, + read_rcd_base, file_nm); + if (i <= 0) { + read_rcd_fd = -1; + return i; + } + read_rcd_buflen = i; + if (buflen > read_rcd_buflen) + buflen = read_rcd_buflen; + memcpy(buf, &read_rcd_buf[pos-read_rcd_base], buflen); + return buflen; +} + + + +int /* -1=error 0=eof >0=record length */ +read_rcd(DCC_EMSG emsg, DB_RCD *rcd, + int fd, off_t pos, const char *file_nm) +{ + int i, cks_len, num_cks; + + /* read a record, starting with its beginning */ + i = read_rcd_sub(emsg, rcd, DB_RCD_HDR_LEN, + fd, pos, file_nm); + if (DB_RCD_HDR_LEN != i) { + if (i == 0) + return 0; + if (i > 0) { + dcc_pemsg(EX_DATAERR, emsg, + "header of record at "OFF_HPAT" of %s" + " truncated by %d bytes", + lseek(fd, 0, SEEK_CUR) - i, + DB_NM2PATH_ERR(file_nm), DB_RCD_HDR_LEN-i); + } + return -1; + } + + num_cks = DB_NUM_CKS(rcd); + if (num_cks > DIM(rcd->cks)) { + dcc_pemsg(EX_DATAERR, emsg, + "bogus checksum count %#x at "OFF_HPAT" of %s", + rcd->fgs_num_cks, lseek(fd, 0, SEEK_CUR), + DB_NM2PATH_ERR(file_nm)); + return -1; + } + cks_len = num_cks * sizeof(rcd->cks[0]); + if (cks_len != 0) { + i = read_rcd_sub(emsg, rcd->cks, cks_len, + fd, pos+DB_RCD_HDR_LEN, file_nm); + if (i != cks_len) { + if (i < 0) + return -1; + if (!i) + dcc_pemsg(EX_DATAERR, emsg, + "record at "OFF_HPAT" of %s" + " truncated after header", + lseek(fd, 0, SEEK_CUR), + DB_NM2PATH_ERR(file_nm)); + else + dcc_pemsg(EX_DATAERR, emsg, + "record at "OFF_HPAT" of %s" + " truncated by %d bytes", + lseek(fd, 0, SEEK_CUR), + DB_NM2PATH_ERR(file_nm), cks_len-i); + return 0; + } + } + + return DB_RCD_HDR_LEN+cks_len; +}