traverse_dir.h
#ifndef TRAVERSE_DIR_H
#define TRAVERSE_DIR_H
#include <vector>
#include <string>
using namespace std;
#ifdef _WIN32
#include <io.h>
#include <sys/types.h>
#else
#include <dirent.h>
#endif
#define _MAX_FNAME 256
#ifdef _WIN32
struct dirent
{
long d_ino;
off_t d_off;
unsigned short d_reclen;
char d_name[_MAX_FNAME+1];
};
typedef struct
{
long handle;
short offset;
short finished;
struct _finddata_t fileinfo;
char *dir;
struct dirent dent;
} DIR;
DIR* opendir(const char *);
struct dirent * readdir(DIR *);
int closedir(DIR *);
#endif
int traverse_dir(string path,vector<string>& files,bool recursive = false,bool includedir = false);
#endif
traverse_dir.cpp
#include <stdio.h>
#include <fcntl.h>
#include <sys/types.h>
#include <ctype.h>
#include <stdlib.h>
#include <errno.h>
#include <sys/stat.h>
#include "traverse_dir.h"
#ifdef _WIN32
DIR* opendir(const char *dir)
{
DIR *dp;
char filespec[_MAX_FNAME];
long handle;
int index;
strncpy(filespec, dir ,_MAX_FNAME);
index = strlen(filespec) - 1;
if (index >= 0 && (filespec[index] == '/' || filespec[index] == '\\'))
filespec[index] = '\0';
strcat(filespec, "/*");
dp = (DIR *)malloc(sizeof(DIR));
dp->offset = 0;
dp->finished = 0;
dp->dir = strdup(dir);
if ((handle = _findfirst(filespec, &(dp->fileinfo))) < 0)
{
if (errno == ENOENT)
dp->finished = 1;
else
return NULL;
}
dp->handle = handle;
return dp;
}
struct dirent* readdir(DIR *dp)
{
if (!dp || dp->finished)
return NULL;
if (dp->offset != 0)
{
if (_findnext(dp->handle, &(dp->fileinfo)) < 0)
{
dp->finished = 1;
return NULL;
}
}
dp->offset++;
strncpy(dp->dent.d_name, dp->fileinfo.name, _MAX_FNAME);
dp->dent.d_ino = 1;
dp->dent.d_reclen = strlen(dp->dent.d_name);
dp->dent.d_off = dp->offset;
return &(dp->dent);
}
int closedir(DIR *dp)
{
if (!dp)
return 0;
_findclose(dp->handle);
if(dp->dir)
{
free(dp->dir);
dp->dir=NULL;
}
if(dp)
{
free(dp);
dp=NULL;
}
return 0;
}
#endif
int traverse_dir(string path,vector<string>& files,bool recursive,bool includedir)
{
if (path.empty())
return -1;
DIR* dir;
dir = opendir(path.c_str());
if (dir == NULL)
return -2;
struct dirent *direntry;
struct stat st;
char temp[256] = {0};
while (direntry = readdir(dir))
{
if (!strcmp(direntry->d_name, ".") || !strcmp(direntry->d_name, ".."))
continue;
if(path[path.size()-1] != '/')
sprintf(temp, "%s/%s", path.c_str(), direntry->d_name);
else
sprintf(temp, "%s%s", path.c_str(), direntry->d_name);
if (stat(temp, &st) == 0)
{
#ifdef WIN32
if((st.st_mode&_S_IFDIR) == _S_IFDIR)
{
if(recursive)
{
traverse_dir(temp,files,recursive, includedir);
}
}
#else
if (S_ISDIR(st.st_mode))
{
if (recursive)
{
traverse_dir(temp, files, recursive);
}
}
#endif
else
{
if(includedir)
files.push_back(temp);
else
{
string file = temp;
files.push_back(file.substr(file.find_last_of('/') + 1));//only need file name
}
}
}
}
closedir(dir);
return (int)files.size();
}
main.cpp
int main()
{
std::string strDir = "F:\\HGUI\\HGUI";
std::vector<std::string> vec;
int ret = traverse_dir(strDir, vec, true, true);
for (int i=0; i<vec.size(); ++i)
{
cout<<vec[i]<<endl;
}
return 0;
}
1 . int _access( const char *path, int mode );
功 能 : 测定文件/目录存取权限.
头文件 : #include <io.h>
参 数 : path:文件或者目录
mode:权限设定,其值如下:
00 Existence only
02 Write permission
04 Read permission
06 Read and write permission
返回值 : 拥有该权限返回0
没有权限返回-1，且设置errno为如下值
ENOENT 路径/文件不存在
EACCES 没有相应权限
2 . int _chdir( const char *dirname );
功 能 : 更改当前工作目录.
头文件 : #include <direct.h>
返回值 : 成功返回0
失败返回-1，且设置errno如下:
ENOENT 该路径不存在
3 . int _chdrive( int drive );
功 能 : 更改当前工作驱动器.
头文件 : #include <direct.h>
返回值 : 成功返回0
失败返回-1
注 释 : 参数说明
drive =1<==> A盘
drive =2<==> B盘
drive =3<==> C盘
如此等等,该函数可以由_chdir代替
4 . int _findclose( long handle );
功 能 : 关闭搜寻句柄并释放相应资源
头文件 : #include <io.h>
参 数 : long handle 搜索句柄(通常由紧靠其前的_findfirst()返回,_findfirst()见下)
fileinfo 文件信息buffer
返回值 : 成功返回0
出错返回-1,且设置errno为如下值
ENOENT 没有更多的符合该泛式的文件
5 . long _findfirst( char *filespec, struct _finddata_t *fileinfo );
功 能 : 提供与filespec指定入口泛式匹配的第一个文件.通常后继用_findnext函数后续使用来完成某泛式下的文件遍历.
头文件 : #include <io.h>
参 数 : filespec 目标文件规范,可以包含通配符
fileinfo 文件信息buffer
返回值 : 成功返回唯一的搜索句柄
出错返回-1,且设置errno为如下值
ENOENT 该泛式无法匹配
EINVAL 无效文件名
6 . int _findnext( long handle, struct _finddata_t *fileinfo );
功 能 : 按照前面_findfirst中的泛式规则，查找下一个符合该泛式的文件，并以 此为依据修改fileinfo中的值
头文件 : #include <io.h>
参 数 : long handle 搜索句柄(通常由紧靠其前的_findfirst()返回)
fileinfo 文件信息buffer
返回值 : 成功返回0
出错返回-1,且设置errno为如下值
ENOENT 没有更多的符合该泛式的文件
7 . char *_getcwd( char *buffer, int maxlen );
功 能 : 获得当前工作目录.
头文件 : #include <direct.h>
返回值 : 成功返回指向buffer的pointer
失败返回NULL，且设置errno为以下三个值之一:
ENODEV 无该设备
ENOMEM 内存不够
ERANGE 结果超出范围
注 意 : 当第一个参数为 NULL 时,第二个参数 maxlen 长度设置无效,且函数使用malloc分配足够内存,需要将函数返回值传递给free()函数来释放内存.
当第一个参数不为 NULL 时,maxlen 指定长度不够函数返回错,设置errno为ERANGE
8 . char *_getdcwd( int drive, char *buffer, int maxlen );
功 能 : 获得指定驱动器的当前工作路径.
头文件 : #include <direct.h>
返回值 : 成功返回指向buffer的pointer
失败返回NULL，且设置errno为以下三个值之一:
ENODEV 无该设备
ENOMEM 内存不够
ERANGE 结果超出范围
注 意 : 当第一个参数为 NULL 时,该函数设置errno为ERANGE
9 . int _getdrive( void );
功 能 : 获得当前磁盘驱动器.
头文件 : #include <direct.h>
返回值 : 返回驱动器值,1<==>A 2<==>B 如此等等;函数不会出错!
10 . unsigned long _getdrives(void);
功 能 : 获得当前所有驱动器.
头文件 : #include <direct.h>
返回值 : 各个位代表对应驱动器,
bit 0 <==> A
bit 1 <==> B
bit 2 <==> C
... ...
注：bit x 表示unsigned long的第x位
11 . int _mkdir( const char *dirname );
功 能 : 创建一个新目录,目录名为dirname.
头文件 : #include <direct.h>
返回值 : 成功返回0
失败返回-1，且设置errno为以下三个值之一
EACCESS 权限不允许
EEXIST 该目录已存在
ENOENT 无该文件或目录
12 . int _rmdir( const char *dirname );
功 能 : 删除名为dirname的目录.
头文件 : #include <direct.h>
返回值 : 成功返回0
失败返回-1，且设置errno为以下三个值之一
EACCESS 权限不允许
ENOTEMPTY dirname不是文件夹;
或者该文件夹不空;
或者dirname为当前工作文件夹;
或者dirname为当根文件夹;
ENOENT 无该文件或目录