1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
/* Copyright (C) 1994, 1995, 1996 Free Software Foundation, Inc.
This file is part of the GNU C Library.
The GNU C Library is free software; you can redistribute it and/or
modify it under the terms of the GNU Library General Public License as
published by the Free Software Foundation; either version 2 of the
License, or (at your option) any later version.
The GNU C Library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Library General Public License for more details.
You should have received a copy of the GNU Library General Public
License along with the GNU C Library; see the file COPYING.LIB. If
not, write to the Free Software Foundation, Inc., 675 Mass Ave,
Cambridge, MA 02139, USA. */
#include <ansidecl.h>
#include <errno.h>
#include <stddef.h>
#include <dirent.h>
#include <unistd.h>
#include <sys/types.h>
#include <stdlib.h>
#include "dirstream.h"
/* Internal data structure for telldir and seekdir. */
struct record
{
struct record *next; /* Link in chain. */
off_t cookie; /* Value returned by `telldir'. */
off_t pos;
size_t offset;
};
#define NBUCKETS 32
static struct record *records[32];
static off_t lastpos;
__libc_lock_define_initialized(static, lock); /* Locks above data. */
/* Return the current position of DIRP. */
off_t
DEFUN(telldir, (dirp), DIR *dirp)
{
struct record *new;
off_t pos;
__libc_lock_lock (lock);
new = malloc (sizeof *new);
if (new == NULL)
return (off_t) -1;
new->pos = dirp->filepos;
new->offset = dirp->offset;
new->cookie = ++lastpos;
new->next = records[new->cookie % NBUCKETS];
records[new->cookie % NBUCKETS] = new;
pos = new->cookie;
__libc_lock_unlock (lock);
return pos;
}
/* Seek to position POS in DIRP. */
void
DEFUN(seekdir, (dirp, pos), DIR *dirp AND __off_t pos)
{
struct record *r, **prevr;
__libc_lock_lock (lock);
for (prevr = &records[pos % NBUCKETS], r = *prevr;
r != NULL;
prevr = &r->next, r = r->next)
if (r->cookie == pos)
{
__libc_lock_lock (dirp->__lock);
if (dirp->filepos != r->pos || dirp->offset != r->offset)
{
dirp->size = 0; /* Must read a fresh buffer. */
/* Move to the saved position. */
__lseek (dirp->fd, r->pos, SEEK_SET);
dirp->filepos = r->pos;
dirp->offset = 0;
/* Read entries until we reach the saved offset. */
while (dirp->offset < r->offset)
{
struct dirent *scan;
__libc_lock_unlock (dirp->__lock);
scan = readdir (dirp);
__libc_lock_lock (dirp->__lock);
if (! scan)
break;
}
}
__libc_lock_unlock (dirp->__lock);
/* To prevent leaking memory, cookies returned from telldir
can only be used once. So free this one's record now. */
*prevr = r->next;
free (r);
break;
}
__libc_lock_unlock (lock);
/* If we lost there is no way to indicate it. Oh well. */
}
|