X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;f=lib%2Ffb-file.c;h=915c1b40f78018ff4478eae894af59d487ca9fd7;hb=d5713848229da7765d89ecc2db81c5f99552f8b4;hp=7d9de003abf6a7fb9f3cfc95b2253af71b308b67;hpb=e38455f76749ff06f64649871ed5e33302389948;p=libucw.git diff --git a/lib/fb-file.c b/lib/fb-file.c index 7d9de003..915c1b40 100644 --- a/lib/fb-file.c +++ b/lib/fb-file.c @@ -1,7 +1,7 @@ /* * Sherlock Library -- Fast Buffered I/O on Files * - * (c) 1997--2002 Martin Mares + * (c) 1997--2004 Martin Mares * * This software may be freely distributed and used according to the terms * of the GNU Lesser General Public License. @@ -22,14 +22,15 @@ struct fb_file { int is_temp_file; /* 0=normal file, 1=temporary file, delete on close, -1=shared FD */ }; #define FB_FILE(f) ((struct fb_file *)(f)->is_fastbuf) +#define FB_BUFFER(f) (byte *)(FB_FILE(f) + 1) static int bfd_refill(struct fastbuf *f) { + f->bptr = f->buffer = FB_BUFFER(f); int l = read(FB_FILE(f)->fd, f->buffer, f->bufend-f->buffer); if (l < 0) die("Error reading %s: %m", f->name); - f->bptr = f->buffer; f->bstop = f->buffer + l; f->pos += l; return l; @@ -39,7 +40,7 @@ static void bfd_spout(struct fastbuf *f) { int l = f->bptr - f->buffer; - char *c = f->buffer; + byte *c = f->buffer; f->pos += l; while (l) @@ -50,7 +51,7 @@ bfd_spout(struct fastbuf *f) l -= z; c += z; } - f->bptr = f->buffer; + f->bptr = f->buffer = FB_BUFFER(f); } static void @@ -102,7 +103,7 @@ bfdopen_internal(int fd, uns buflen, byte *name) struct fastbuf *f = &F->fb; bzero(F, sizeof(*F)); - f->buffer = (char *)(F+1); + f->buffer = (byte *)(F+1); f->bptr = f->bstop = f->buffer; f->bufend = f->buffer + buflen; f->name = f->bufend; @@ -113,36 +114,41 @@ bfdopen_internal(int fd, uns buflen, byte *name) f->seek = bfd_seek; f->close = bfd_close; f->config = bfd_config; + f->can_overwrite_buffer = 2; return f; } struct fastbuf * -bopen(byte *name, uns mode, uns buffer) +bopen(byte *name, uns mode, uns buflen) { struct fastbuf *b; - int fd = sh_open(name, mode, 0666); + int fd; + + if (!buflen) + return bopen_mm(name, mode); + fd = sh_open(name, mode, 0666); if (fd < 0) die("Unable to %s file %s: %m", (mode & O_CREAT) ? "create" : "open", name); - b = bfdopen_internal(fd, buffer, name); + b = bfdopen_internal(fd, buflen, name); if (mode & O_APPEND) bfd_seek(b, 0, SEEK_END); return b; } struct fastbuf * -bfdopen(int fd, uns buffer) +bfdopen(int fd, uns buflen) { byte x[32]; sprintf(x, "fd%d", fd); - return bfdopen_internal(fd, buffer, x); + return bfdopen_internal(fd, buflen, x); } struct fastbuf * -bfdopen_shared(int fd, uns buffer) +bfdopen_shared(int fd, uns buflen) { - struct fastbuf *f = bfdopen(fd, buffer); + struct fastbuf *f = bfdopen(fd, buflen); FB_FILE(f)->is_temp_file = -1; return f; }