blob: 4142813e38c06cc9b108a64cbc218c3f13ddd686 (
plain)
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
|
/* vi: set sw=4 ts=4: */
/*
* Utility routines.
*
* Copyright (C) many different people.
* If you wrote this, please acknowledge your work.
*
* Licensed under GPLv2 or later, see file LICENSE in this source tree.
*/
#include "libbb.h"
typedef struct ino_dev_hash_bucket_struct {
ino_t ino;
dev_t dev;
/*
* Above fields can be 64-bit, while pointer may be 32-bit.
* Putting "next" field here may reduce size of this struct:
*/
struct ino_dev_hash_bucket_struct *next;
/*
* Reportedly, on cramfs a file and a dir can have same ino.
* Need to also remember "file/dir" bit:
*/
char isdir; /* bool */
char name[1];
} ino_dev_hashtable_bucket_t;
#define HASH_SIZE 311u /* Should be prime */
#define hash_inode(i) ((unsigned)(i) % HASH_SIZE)
/* array of [HASH_SIZE] elements */
static ino_dev_hashtable_bucket_t **ino_dev_hashtable;
/*
* Return name if statbuf->st_ino && statbuf->st_dev are recorded in
* ino_dev_hashtable, else return NULL
*/
char* FAST_FUNC is_in_ino_dev_hashtable(const struct stat *statbuf)
{
ino_dev_hashtable_bucket_t *bucket;
if (!ino_dev_hashtable)
return NULL;
bucket = ino_dev_hashtable[hash_inode(statbuf->st_ino)];
while (bucket != NULL) {
if ((bucket->ino == statbuf->st_ino)
&& (bucket->dev == statbuf->st_dev)
&& (bucket->isdir == !!S_ISDIR(statbuf->st_mode))
) {
return bucket->name;
}
bucket = bucket->next;
}
return NULL;
}
/* Add statbuf to statbuf hash table */
void FAST_FUNC add_to_ino_dev_hashtable(const struct stat *statbuf, const char *name)
{
int i;
ino_dev_hashtable_bucket_t *bucket;
if (!name)
name = "";
bucket = xmalloc(sizeof(ino_dev_hashtable_bucket_t) + strlen(name));
bucket->ino = statbuf->st_ino;
bucket->dev = statbuf->st_dev;
bucket->isdir = !!S_ISDIR(statbuf->st_mode);
strcpy(bucket->name, name);
if (!ino_dev_hashtable)
ino_dev_hashtable = xzalloc(HASH_SIZE * sizeof(*ino_dev_hashtable));
i = hash_inode(statbuf->st_ino);
bucket->next = ino_dev_hashtable[i];
ino_dev_hashtable[i] = bucket;
}
#if ENABLE_DU || ENABLE_FEATURE_CLEAN_UP
/* Clear statbuf hash table */
void FAST_FUNC reset_ino_dev_hashtable(void)
{
int i;
ino_dev_hashtable_bucket_t *bucket, *next;
if (!ino_dev_hashtable)
return;
for (i = 0; i < HASH_SIZE; i++) {
bucket = ino_dev_hashtable[i];
while (bucket != NULL) {
next = bucket->next;
free(bucket);
bucket = next;
}
}
free(ino_dev_hashtable);
ino_dev_hashtable = NULL;
}
#endif
|