void rm_session_clear(RmSession *session) {
RmCfg *cfg = session->cfg;
/* Free mem */
if(cfg->paths) {
g_strfreev(cfg->paths);
}
g_free(cfg->sort_criteria);
g_timer_destroy(session->timer);
rm_file_tables_destroy(session->tables);
rm_fmt_close(session->formats);
g_ptr_array_free(session->pattern_cache, TRUE);
if(session->mounts) {
rm_mounts_table_destroy(session->mounts);
}
if(session->dir_merger) {
rm_tm_destroy(session->dir_merger);
}
g_free(cfg->joined_argv);
g_free(cfg->is_prefd);
g_free(cfg->iwd);
rm_trie_destroy(&cfg->file_trie);
}
int main(void) {
RmTrie trie;
rm_trie_init(&trie);
GTimer *timer = g_timer_new();
g_timer_start(timer);
char buf[1024];
int i = 0;
while(fgets(buf, sizeof(buf), stdin)) {
buf[strlen(buf) - 1] = 0;
rm_trie_insert(&trie, buf, GUINT_TO_POINTER(++i));
memset(buf, 0, sizeof(buf));
}
g_printerr("Took %2.5f to insert %d items\n", g_timer_elapsed(timer, NULL), i);
rm_trie_print(&trie);
memset(buf, 0, sizeof(buf));
rm_trie_build_path(&trie, rm_trie_search_node(&trie, "/usr/bin/rmlint"), buf,
sizeof(buf));
g_printerr("=> %s\n", buf);
g_timer_start(timer);
const int N = 10000000;
for(int x = 0; x < N; x++) {
rm_trie_search(&trie, "/usr/bin/rmlint");
}
g_printerr("Took %2.5f to search\n", g_timer_elapsed(timer, NULL));
g_printerr("%u\n", GPOINTER_TO_UINT(rm_trie_search(&trie, "/usr/bin/rmlint")));
g_printerr("%u\n", GPOINTER_TO_UINT(rm_trie_search(&trie, "/a/b/c")));
rm_trie_destroy(&trie);
g_timer_destroy(timer);
return 0;
}
void rm_tm_destroy(RmTreeMerger *self) {
g_hash_table_unref(self->result_table);
g_hash_table_unref(self->file_groups);
g_hash_table_unref(self->file_checks);
GList *digest_keys = g_hash_table_get_keys(self->known_hashs);
g_list_free_full(digest_keys, (GDestroyNotify)rm_digest_free);
g_hash_table_unref(self->known_hashs);
g_queue_clear(&self->valid_dirs);
/* Kill all RmDirectories stored in the tree */
rm_trie_iter(&self->dir_tree, NULL, true, false, rm_tm_destroy_iter, NULL);
rm_trie_destroy(&self->dir_tree);
rm_trie_destroy(&self->count_tree);
g_slice_free(RmTreeMerger, self);
}
static bool rm_tm_count_files(RmTrie *count_tree, char **paths, RmSession *session) {
if(*paths == NULL) {
rm_log_error("No paths passed to rm_tm_count_files\n");
return false;
}
int fts_flags = FTS_COMFOLLOW;
if(session->cfg->follow_symlinks) {
fts_flags |= FTS_LOGICAL;
} else {
fts_flags |= FTS_PHYSICAL;
}
/* This tree stores the full file paths.
It is joined into a full directory tree later.
*/
RmTrie file_tree;
rm_trie_init(&file_tree);
FTS *fts = fts_open(paths, fts_flags, NULL);
if(fts == NULL) {
rm_log_perror("fts_open failed");
return false;
}
FTSENT *ent = NULL;
while((ent = fts_read(fts))) {
/* Handle large files (where fts fails with FTS_NS) */
if(ent->fts_info == FTS_NS) {
RmStat stat_buf;
if(rm_sys_stat(ent->fts_path, &stat_buf) == -1) {
rm_log_perror("stat(2) failed");
continue;
} else {
/* Must be a large file (or followed link to it) */
ent->fts_info = FTS_F;
}
}
switch(ent->fts_info) {
case FTS_ERR:
case FTS_DC:
/* Save this path as an error */
rm_trie_insert(&file_tree, ent->fts_path, GINT_TO_POINTER(true));
break;
case FTS_F:
case FTS_SL:
case FTS_NS:
case FTS_SLNONE:
case FTS_DEFAULT:
/* Save this path as countable file */
if(ent->fts_statp->st_size > 0) {
rm_trie_insert(&file_tree, ent->fts_path, GINT_TO_POINTER(false));
}
case FTS_D:
case FTS_DNR:
case FTS_DOT:
case FTS_DP:
case FTS_NSOK:
default:
/* other fts states, that do not count as errors or files */
break;
}
}
if(fts_close(fts) != 0) {
rm_log_perror("fts_close failed");
return false;
}
rm_trie_iter(&file_tree, NULL, true, false, rm_tm_count_art_callback, count_tree);
/* Now flag everything as a no-go over the given paths,
* otherwise we would continue merging till / with fatal consequences,
* since / does not have more files as paths[0]
*/
for(int i = 0; paths[i]; ++i) {
/* Just call the callback directly */
RmNode *node = rm_trie_search_node(&file_tree, paths[i]);
if(node != NULL) {
node->data = GINT_TO_POINTER(true);
rm_tm_count_art_callback(&file_tree, node, 0, count_tree);
}
}
#ifdef _RM_TREEMERGE_DEBUG
rm_trie_print(count_tree);
#endif
rm_trie_destroy(&file_tree);
return true;
}
