]> mj.ucw.cz Git - libucw.git/blob - lib/fb-file.c
Added a temporary hack for testing of fb-mmap. Please don't turn it on
[libucw.git] / lib / fb-file.c
1 /*
2  *      Sherlock Library -- Fast Buffered I/O on Files
3  *
4  *      (c) 1997--2002 Martin Mares <mj@ucw.cz>
5  *
6  *      This software may be freely distributed and used according to the terms
7  *      of the GNU Lesser General Public License.
8  */
9
10 #include "lib/lib.h"
11 #include "lib/fastbuf.h"
12 #include "lib/lfs.h"
13
14 #include <stdlib.h>
15 #include <string.h>
16 #include <fcntl.h>
17 #include <unistd.h>
18
19 struct fb_file {
20   struct fastbuf fb;
21   int fd;                               /* File descriptor, -1 if not a real file */
22   int is_temp_file;                     /* 0=normal file, 1=temporary file, delete on close, -1=shared FD */
23 };
24 #define FB_FILE(f) ((struct fb_file *)(f)->is_fastbuf)
25
26 static int
27 bfd_refill(struct fastbuf *f)
28 {
29   int l = read(FB_FILE(f)->fd, f->buffer, f->bufend-f->buffer);
30   if (l < 0)
31     die("Error reading %s: %m", f->name);
32   f->bptr = f->buffer;
33   f->bstop = f->buffer + l;
34   f->pos += l;
35   return l;
36 }
37
38 static void
39 bfd_spout(struct fastbuf *f)
40 {
41   int l = f->bptr - f->buffer;
42   char *c = f->buffer;
43
44   f->pos += l;
45   while (l)
46     {
47       int z = write(FB_FILE(f)->fd, c, l);
48       if (z <= 0)
49         die("Error writing %s: %m", f->name);
50       l -= z;
51       c += z;
52     }
53   f->bptr = f->buffer;
54 }
55
56 static void
57 bfd_seek(struct fastbuf *f, sh_off_t pos, int whence)
58 {
59   sh_off_t l;
60
61   if (whence == SEEK_SET && pos == f->pos)
62     return;
63
64   l = sh_seek(FB_FILE(f)->fd, pos, whence);
65   if (l < 0)
66     die("lseek on %s: %m", f->name);
67   f->pos = l;
68 }
69
70 static void
71 bfd_close(struct fastbuf *f)
72 {
73   switch (FB_FILE(f)->is_temp_file)
74     {
75     case 1:
76       if (unlink(f->name) < 0)
77         log(L_ERROR, "unlink(%s): %m", f->name);
78     case 0:
79       close(FB_FILE(f)->fd);
80     }
81   xfree(f);
82 }
83
84 static int
85 bfd_config(struct fastbuf *f, uns item, int value)
86 {
87   switch (item)
88     {
89     case BCONFIG_IS_TEMP_FILE:
90       FB_FILE(f)->is_temp_file = value;
91       return 0;
92     default:
93       return -1;
94     }
95 }
96
97 static struct fastbuf *
98 bfdopen_internal(int fd, uns buflen, byte *name)
99 {
100   int namelen = strlen(name) + 1;
101   struct fb_file *F = xmalloc(sizeof(struct fb_file) + buflen + namelen);
102   struct fastbuf *f = &F->fb;
103
104   bzero(F, sizeof(*F));
105   f->buffer = (char *)(F+1);
106   f->bptr = f->bstop = f->buffer;
107   f->bufend = f->buffer + buflen;
108   f->name = f->bufend;
109   memcpy(f->name, name, namelen);
110   F->fd = fd;
111   f->refill = bfd_refill;
112   f->spout = bfd_spout;
113   f->seek = bfd_seek;
114   f->close = bfd_close;
115   f->config = bfd_config;
116   return f;
117 }
118
119 struct fastbuf *
120 bopen(byte *name, uns mode, uns buffer)
121 {
122   struct fastbuf *b;
123
124 #if 0 /* FIXME: Dirty hack for testing fb-mmap */
125   if (buffer >= 65536 && (mode == O_RDONLY))
126     {
127       log(L_DEBUG, "bopen_mm hack: %s", name);
128       if (mode & O_WRONLY)
129         mode = (mode & ~O_WRONLY) | O_RDWR;
130       return bopen_mm(name, mode);
131     }
132 #endif
133
134   int fd = sh_open(name, mode, 0666);
135   if (fd < 0)
136     die("Unable to %s file %s: %m",
137         (mode & O_CREAT) ? "create" : "open", name);
138   b = bfdopen_internal(fd, buffer, name);
139   if (mode & O_APPEND)
140     bfd_seek(b, 0, SEEK_END);
141   return b;
142 }
143
144 struct fastbuf *
145 bfdopen(int fd, uns buffer)
146 {
147   byte x[32];
148
149   sprintf(x, "fd%d", fd);
150   return bfdopen_internal(fd, buffer, x);
151 }
152
153 struct fastbuf *
154 bfdopen_shared(int fd, uns buffer)
155 {
156   struct fastbuf *f = bfdopen(fd, buffer);
157   FB_FILE(f)->is_temp_file = -1;
158   return f;
159 }
160
161 #ifdef TEST
162
163 int main(int argc, char **argv)
164 {
165   struct fastbuf *f, *t;
166
167   f = bopen("/etc/profile", O_RDONLY, 16);
168   t = bfdopen(1, 13);
169   bbcopy(f, t, 100);
170   printf("%d %d\n", (int)btell(f), (int)btell(t));
171   bclose(f);
172   bclose(t);
173   return 0;
174 }
175
176 #endif