| /* 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 { |
| struct ino_dev_hash_bucket_struct *next; |
| ino_t ino; |
| dev_t dev; |
| char name[1]; |
| } ino_dev_hashtable_bucket_t; |
| |
| #define HASH_SIZE 311 /* Should be prime */ |
| #define hash_inode(i) ((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) |
| ) { |
| 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; |
| |
| i = hash_inode(statbuf->st_ino); |
| if (!name) |
| name = ""; |
| bucket = xmalloc(sizeof(ino_dev_hashtable_bucket_t) + strlen(name)); |
| bucket->ino = statbuf->st_ino; |
| bucket->dev = statbuf->st_dev; |
| strcpy(bucket->name, name); |
| |
| if (!ino_dev_hashtable) |
| ino_dev_hashtable = xzalloc(HASH_SIZE * sizeof(*ino_dev_hashtable)); |
| |
| 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; |
| |
| for (i = 0; ino_dev_hashtable && i < HASH_SIZE; i++) { |
| while (ino_dev_hashtable[i] != NULL) { |
| bucket = ino_dev_hashtable[i]->next; |
| free(ino_dev_hashtable[i]); |
| ino_dev_hashtable[i] = bucket; |
| } |
| } |
| free(ino_dev_hashtable); |
| ino_dev_hashtable = NULL; |
| } |
| #endif |