APUE统计目录各类文件个数(c++版)

4.20.cpp

#include <iostream>#include <sys/stat.h>#include <string.h>#include <stdexcept>#include <memory>#include <dirent.h>using namespace std;class Dir{public:    string dirname;    int nreg;    int nblk;    int nchr;    int ndir;    int nslink;    int nfifo;    int nsock;    int ntotal;private:    struct stat* statbuf;    DIR *dp;    struct dirent *dirp;private:    void isFile()    {    switch (statbuf->st_mode & S_IFMT)    {        case S_IFREG:        ++nreg; break;        case S_IFBLK:        ++nblk; break;        case S_IFCHR:        ++nchr; break;        case S_IFLNK:        ++nslink; break;        case S_IFIFO:        ++nfifo; break;        case S_IFSOCK:        ++nsock; break;        default:        throw out_of_range("未知类型文件");    }    }    void isDir()    {    ++ndir;    if ((dp = opendir(dirname.c_str())) == NULL)        throw out_of_range("打开目录失败");    while ((dirp = readdir(dp)) != NULL)    {        if (strcmp(dirp->d_name,".") == 0 || strcmp(dirp->d_name,"..") ==0)        continue;       string ndirname = dirname + "/" + dirp->d_name;        cout << ndirname  << endl;        unique_ptr<Dir> dir(new Dir(ndirname));        nreg += dir->nreg;        nchr += dir->nchr;        nblk += dir->nblk;        nslink += dir->nslink;        ndir += dir->ndir;        nfifo += dir->nfifo;        nsock += dir->nsock;        ntotal += dir->ntotal;    }    }public:    Dir(const string& s):dirname(s),nreg(0),nblk(0),ndir(0),nfifo(0),nchr(0)             ,nsock(0),nslink(0),ntotal(0),statbuf(new struct stat)    {    if (lstat(dirname.c_str(),statbuf) < 0)        throw out_of_range("lstat error\n");    if (S_ISDIR(statbuf->st_mode) == 0)        isFile();    else if (S_ISLNK(statbuf->st_mode) != 0)        ++nslink;    else        isDir();    ntotal = nreg + nblk + ndir + nfifo + nchr + nsock + nslink;    }    void display()    {    cout << "********************************************" << endl;    cout << "普通文件总量:         " << nreg << endl;    cout << "字符特殊文件总量:     " << nchr << endl;    cout << "块特殊文件总量:       " << nblk << endl;    cout << "符号链接总量:           " << nslink << endl;    cout << "套接字文件总量:       " << nsock << endl;    cout << "目录文件总量:           " << ndir << endl;    cout << "管道文件总量:         " << nfifo << endl;    cout << "文件总量:         " << ntotal << endl;    cout << "********************************************" << endl;    }    ~Dir()    {    if (S_ISDIR(statbuf->st_mode) != 0)       closedir(dp);    if (statbuf != NULL)        delete statbuf;    }};int main(int argc,char* argv[]){    try    {    if (argc != 2)        throw domain_error("add <filename>");    Dir dir(argv[1]);    dir.display();    }    catch(const exception& e)    {    cout << e.what() << endl;    }}

运行: