X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;f=ucw%2Fpartmap.c;h=fd9d91249ea8e715942b1e070ada30a141ecd8d3;hb=315d65b74168b23e820b9dde6166ec19bd048228;hp=ed74df363219608c6c58b60e0da0f047aaa85c28;hpb=e8060fa4647c5c4ccf07efb91a35f65c543dfc4f;p=libucw.git

diff --git a/ucw/partmap.c b/ucw/partmap.c
index ed74df36..fd9d9124 100644
--- a/ucw/partmap.c
+++ b/ucw/partmap.c
@@ -2,14 +2,14 @@
  *	UCW Library -- Mapping of File Parts
  *
  *	(c) 2003--2006 Martin Mares <mj@ucw.cz>
- *	(c) 2003--2005 Robert Spalek <robert@ucw.cz>
+ *	(c) 2003--2009 Robert Spalek <robert@ucw.cz>
  *
  *	This software may be freely distributed and used according to the terms
  *	of the GNU Lesser General Public License.
  */
 
 #include "ucw/lib.h"
-#include "ucw/lfs.h"
+#include "ucw/io.h"
 #include "ucw/partmap.h"
 
 #include <stdio.h>
@@ -19,7 +19,7 @@
 #include <sys/stat.h>
 #include <sys/mman.h>
 
-#ifdef CONFIG_PARTMAP_IS_MMAP
+#ifdef CONFIG_UCW_PARTMAP_IS_MMAP
 #define PARTMAP_WINDOW ~(size_t)0
 #else
 #ifdef TEST
@@ -40,7 +40,7 @@ partmap_open(char *name, int writeable)
   if ((p->file_size = ucw_seek(p->fd, 0, SEEK_END)) < 0)
     die("lseek(%s): %m", name);
   p->writeable = writeable;
-#ifdef CONFIG_PARTMAP_IS_MMAP
+#ifdef CONFIG_UCW_PARTMAP_IS_MMAP
   partmap_load(p, 0, p->file_size);
 #endif
   return p;
@@ -69,13 +69,18 @@ partmap_load(struct partmap *p, ucw_off_t start, uns size)
   ucw_off_t end = start + size;
   ucw_off_t win_start = start/CPU_PAGE_SIZE * CPU_PAGE_SIZE;
   size_t win_len = PARTMAP_WINDOW;
-  if ((ucw_off_t) (win_start+win_len) > p->file_size)
+  if (win_len > (size_t) (p->file_size - win_start))
     win_len = ALIGN_TO(p->file_size - win_start, CPU_PAGE_SIZE);
   if ((ucw_off_t) (win_start+win_len) < end)
     die("partmap_map: Window is too small for mapping %d bytes", size);
-  p->start_map = ucw_mmap(NULL, win_len, p->writeable ? (PROT_READ | PROT_WRITE) : PROT_READ, MAP_SHARED, p->fd, win_start);
-  if (p->start_map == MAP_FAILED)
-    die("mmap failed at position %lld: %m", (long long)win_start);
+  if (win_len)
+    {
+      p->start_map = ucw_mmap(NULL, win_len, p->writeable ? (PROT_READ | PROT_WRITE) : PROT_READ, MAP_SHARED, p->fd, win_start);
+      if (p->start_map == MAP_FAILED)
+	die("mmap failed at position %lld: %m", (long long)win_start);
+    }
+  else
+    p->start_map = NULL;
   p->start_off = win_start;
   p->end_off = win_start+win_len;
   madvise(p->start_map, win_len, MADV_SEQUENTIAL);