]> mj.ucw.cz Git - libucw.git/blob - lib/partmap.c
Merge with git+ssh://git.ucw.cz/projects/sherlock/GIT/sherlock.git
[libucw.git] / lib / partmap.c
1 /*
2  *      UCW Library -- Mapping of File Parts
3  *
4  *      (c) 2003--2006 Martin Mares <mj@ucw.cz>
5  *      (c) 2003--2005 Robert Spalek <robert@ucw.cz>
6  *
7  *      This software may be freely distributed and used according to the terms
8  *      of the GNU Lesser General Public License.
9  */
10
11 #include "lib/lib.h"
12 #include "lib/lfs.h"
13 #include "lib/partmap.h"
14
15 #include <stdio.h>
16 #include <stdlib.h>
17 #include <fcntl.h>
18 #include <unistd.h>
19 #include <sys/stat.h>
20 #include <sys/mman.h>
21
22 #ifdef CONFIG_PARTMAP_IS_MMAP
23 #define PARTMAP_WINDOW ~(size_t)0
24 #else
25 #ifdef TEST
26 #define PARTMAP_WINDOW 4096
27 #else
28 #define PARTMAP_WINDOW 16777216
29 #endif
30 #endif
31
32 struct partmap *
33 partmap_open(char *name, int writeable)
34 {
35   struct partmap *p = xmalloc_zero(sizeof(struct partmap));
36
37   p->fd = sh_open(name, writeable ? O_RDWR : O_RDONLY);
38   if (p->fd < 0)
39     die("open(%s): %m", name);
40   if ((p->file_size = sh_seek(p->fd, 0, SEEK_END)) < 0)
41     die("lseek(%s): %m", name);
42   p->writeable = writeable;
43 #ifdef CONFIG_PARTMAP_IS_MMAP
44   partmap_load(p, 0, p->file_size);
45 #endif
46   return p;
47 }
48
49 sh_off_t
50 partmap_size(struct partmap *p)
51 {
52   return p->file_size;
53 }
54
55 void
56 partmap_close(struct partmap *p)
57 {
58   if (p->start_map)
59     munmap(p->start_map, p->end_off - p->start_off);
60   close(p->fd);
61   xfree(p);
62 }
63
64 void
65 partmap_load(struct partmap *p, sh_off_t start, uns size)
66 {
67   if (p->start_map)
68     munmap(p->start_map, p->end_off - p->start_off);
69   sh_off_t end = start + size;
70   sh_off_t win_start = start/CPU_PAGE_SIZE * CPU_PAGE_SIZE;
71   size_t win_len = PARTMAP_WINDOW;
72   if ((sh_off_t) (win_start+win_len) > p->file_size)
73     win_len = ALIGN_TO(p->file_size - win_start, CPU_PAGE_SIZE);
74   if ((sh_off_t) (win_start+win_len) < end)
75     die("partmap_map: Window is too small for mapping %d bytes", size);
76   p->start_map = sh_mmap(NULL, win_len, p->writeable ? (PROT_READ | PROT_WRITE) : PROT_READ, MAP_SHARED, p->fd, win_start);
77   if (p->start_map == MAP_FAILED)
78     die("mmap failed at position %lld: %m", (long long)win_start);
79   p->start_off = win_start;
80   p->end_off = win_start+win_len;
81   madvise(p->start_map, win_len, MADV_SEQUENTIAL);
82 }
83
84 #ifdef TEST
85 int main(int argc, char **argv)
86 {
87   struct partmap *p = partmap_open(argv[1], 0);
88   uns l = partmap_size(p);
89   uns i;
90   for (i=0; i<l; i++)
91     putchar(*(char *)partmap_map(p, i, 1));
92   partmap_close(p);
93   return 0;
94 }
95 #endif