X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;f=ucw%2Ffb-mmap.c;h=8f9f0bb28b7d59ac4605d9552d3093012e809a7c;hb=46a64e45c1d3fa42376cffcf90cfa0b66d781772;hp=afd0bb12621352f854ae1819914b0dde15a8df8a;hpb=e8060fa4647c5c4ccf07efb91a35f65c543dfc4f;p=libucw.git diff --git a/ucw/fb-mmap.c b/ucw/fb-mmap.c index afd0bb12..8f9f0bb2 100644 --- a/ucw/fb-mmap.c +++ b/ucw/fb-mmap.c @@ -22,6 +22,7 @@ static uns mmap_window_size = 16*CPU_PAGE_SIZE; static uns mmap_extend_size = 4*CPU_PAGE_SIZE; +#ifndef TEST static struct cf_section fbmm_config = { CF_ITEMS { CF_UNS("WindowSize", &mmap_window_size), @@ -34,6 +35,7 @@ static void CONSTRUCTOR fbmm_init_config(void) { cf_declare_section("FBMMap", &fbmm_config, 0); } +#endif struct fb_mmap { struct fastbuf fb; @@ -68,7 +70,10 @@ bfmm_map_window(struct fastbuf *f) else f->buffer = ucw_mmap(f->buffer, ll, prot, MAP_SHARED | MAP_FIXED, F->fd, pos0); if (f->buffer == (byte *) MAP_FAILED) - die("mmap(%s): %m", f->name); + { + f->buffer = NULL; + bthrow(f, "mmap", "mmap(%s): %m", f->name); + } #ifdef MADV_SEQUENTIAL if (ll > CPU_PAGE_SIZE) madvise(f->buffer, ll, MADV_SEQUENTIAL); @@ -113,7 +118,7 @@ bfmm_spout(struct fastbuf *f) { F->file_extend = ALIGN_TO(F->file_extend + mmap_extend_size, (ucw_off_t)CPU_PAGE_SIZE); if (ucw_ftruncate(F->fd, F->file_extend)) - die("ftruncate(%s): %m", f->name); + bthrow(f, "write", "ftruncate(%s): %m", f->name); } bfmm_map_window(f); f->bstop = f->bptr; @@ -138,12 +143,12 @@ static void bfmm_close(struct fastbuf *f) { struct fb_mmap *F = FB_MMAP(f); - if (f->buffer) munmap(f->buffer, F->window_size); - if (F->file_extend > F->file_size && + if (!(f->flags & FB_DEAD) && + F->file_extend > F->file_size && ucw_ftruncate(F->fd, F->file_size)) - die("ftruncate(%s): %m", f->name); + bthrow(f, "write", "ftruncate(%s): %m", f->name); bclose_file_helper(f, F->fd, F->is_temp_file); xfree(f); } @@ -177,7 +182,7 @@ bfmmopen_internal(int fd, const char *name, uns mode) F->fd = fd; F->file_extend = F->file_size = ucw_seek(fd, 0, SEEK_END); if (F->file_size < 0) - die("seek(%s): %m", name); + bthrow(f, "open", "fb-mmap: Cannot detect size of %s -- is it seekable?", name); if (mode & O_APPEND) f->pos = F->file_size; F->mode = mode; @@ -192,7 +197,7 @@ bfmmopen_internal(int fd, const char *name, uns mode) #ifdef TEST -int main(int argc, char **argv) +int main(int UNUSED argc, char **argv) { struct fb_params par = { .type = FB_MMAP }; struct fastbuf *f = bopen_file(argv[1], O_RDONLY, &par);