man: --nbi, not --NBI
[wraplinux.git] / mapfile.c
1 /* ----------------------------------------------------------------------- *
2  *
3  *   Copyright 2008 rPath, Inc. - All Rights Reserved
4  *
5  *   This program is free software; you can redistribute it and/or modify
6  *   it under the terms of the GNU General Public License as published by
7  *   the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
8  *   Boston MA 02110-1301, USA; either version 2 of the License, or
9  *   (at your option) any later version; incorporated herein by reference.
10  *
11  * ----------------------------------------------------------------------- */
12
13 /*
14  * mapfile.c
15  *
16  * Memory-map a file, and/or read it into a memory buffer.
17  * If the "writable" flag is set, return a writable memory
18  * buffer, *not* one which writes back to the file!
19  *
20  * Note: if this is modified to use a buffer, always round the buffer
21  * size up to at least a dword boundary.
22  */
23
24 #include "wraplinux.h"
25
26 #include <unistd.h>
27 #include <fcntl.h>
28 #include <sys/mman.h>
29 #include <sys/stat.h>
30
31 void *mapfile(int fd, size_t *len, bool writable)
32 {
33         struct stat st;
34         void *ptr;
35         int prot, flags;
36
37         if (fstat(fd, &st))
38                 return NULL;
39
40         *len = st.st_size;
41
42         if (writable) {
43                 prot  = PROT_READ|PROT_WRITE;
44                 flags = MAP_PRIVATE;
45         } else {
46                 prot  = PROT_READ;
47                 flags = MAP_SHARED;
48         }
49
50         ptr = mmap(NULL, st.st_size, prot, flags, fd, 0);
51
52         return (ptr == MAP_FAILED) ? NULL : ptr;
53 }
54
55 void unmapfile(int fd, void *ptr, size_t len)
56 {
57         if (ptr)
58                 munmap(ptr, len);
59         if (fd >= 0)
60                 close(fd);
61 }