linux c非递归遍历目录,【Linux C】遍历 指定文件夹(home)的所有文件 （非递归实现）...

为什么要写这个呢...是因为要出一些恶意的代码片段，有一个遍历文件夹的需求。按照要求，所有的恶意行为只能写在一个函数中，且不能有注释。本来不想自己写，可是google到的遍历文件夹的代码全是递归写的。虽然gcc也支持内部函数的编译，但是这样感觉不太优雅，于是稍微修改一下，就有了下面的代码。

实现思路其实很简单，其实就是链表啦，不过上次写链表还是13年吧，真太菜了。

用一个结构体保存 1. 当前文件夹的path的地址2.下一个结构体的地址(初始化为NULL)。

遍历时，创建两个结构体指针，folderfirst和folderlast。用folderfirst遍历指定文件夹(如/home)的时候，如果出现新的文件夹，就用创建新的结构体，用folderlast来增加到链表中。当前文件夹遍历完成，修改folderfirst指向下一个结构体。循环结束的条件就是，folderfirst为NULL。

PS:这么实现貌似算广度优先搜索？貌似搜索文件夹BFS比DFS要快, 和OS有关系。忘了出处了，感兴趣的可以搜索一下。

1806191523 bug：scan根目录的时候，猜测由于部分文件没有读权限，因此程序就停了。此外，有替代工具，命令"du -h"。暂时没时间改了，有空参考一下du的源码实现吧。ftp://ftp.gnu.org/gnu/coreutils/

如果有人有兴趣修改，欢迎留言讨论。

最新以及历史版本放在：https://gist.github.com/thinkycx/a6359201f79ffdf4e97a111c5fe4cf01

代码：

/*

date: 20180619

author: thinkycx

usage: gcc scanfolder.c -o scanfolder

./scanfolder [foldername]

./scscanfolderan /home

Only support to scan folder and regular file.Maybe support symbolic etc in future...

*/

#include

#include

#include

#include

#include

#include

#include

#include

#include

#include

#include

#include

void f(char *path){

#define SEARCH_PATH "/home"

struct foldernode{

char *path; // point to foldername or filename path

struct foldernode *next;

};

DIR *dir;

struct stat statbuf;

struct dirent *ptr;

char *filename;

char *filepath;

char *data;

char *foldername ;

char *folderpath ;

struct foldernode folderstart;

folderstart.path = path;

folderstart.next = NULL;

struct foldernode * folderfirst; // use to search

folderfirst = &folderstart;

struct foldernode * folderlast; // use to add foldernode

folderlast = &folderstart;

while(folderfirst != NULL){

dir = opendir(folderfirst->path);

if (dir) {

while ((ptr = readdir(dir)) != NULL) {

if(strcmp(ptr->d_name, ".")==0 || strcmp(ptr->d_name, "..")==0)

continue;

else if(ptr->d_type == 8){ //file

filename = malloc(0x100);

filepath = malloc(0x1000);

memset(filename, 0x100, 0);

memset(filepath, 0x100, 0);

strcpy(filename, ptr->d_name);

sprintf(filepath, "%s/%s", folderfirst->path, filename);

printf("[*]FILE %s\n",filepath);

free(filename);

free(filepath);

}

else if(ptr->d_type == 4){ //folder

foldername = malloc(0x100);

folderpath = malloc(0x1000);

memset(foldername, 0x100, 0);

memset(folderpath, 0x1000, 0);

strcpy(foldername, ptr->d_name);

sprintf(folderpath, "%s/%s", folderfirst->path , foldername);

printf("[*]FOLDER %s\n",folderpath);

struct foldernode *foldernew;

foldernew = malloc(sizeof(struct foldernode));

foldernew->path = malloc(strlen(folderpath)+1);

strcpy(foldernew->path, folderpath);

foldernew->next = NULL;

folderlast->next = foldernew;

folderlast = foldernew;

free(foldername);

free(folderpath);

}

}

}else{

return;

}

folderfirst = folderfirst->next; // change folderfirst point to next foldernode

closedir(dir);

}

}

int main(int argc,char **argv, char **env) {

if(argc == 2){

f(argv[1]);

}else{

printf("Usage: scanfolder [foldername]\n\tscanfolder /home\n");

}

}