检查文件是否在C特定类型(Check if a file is a specific type in

我在写我的第一个C程序，虽然我来自一个C ++背景。

我需要通过文件的目录遍历和检查，看看是否该文件是一个头文件，然后返回计数。

我的代码如下，它很简陋，我认为：

static int CountHeaders( const char* dirname ) {

    int header_count = 0;
    DIR* dir_ptr;
    struct dirent* entry;

    dir_ptr = opendir( dirname );

    while( ( entry = readdir( dir_ptr ) ) )
    {
        if ( entry->d_type == DT_REG )
        {
            //second if statement to verify the file is a header file should be???
            ++header_count;
        }
    }

    closedir( dir_ptr );

    return header_count;
}

这将是一个很好的if语句来检查，看看是否该文件是一个头？

Answer 1:

简单地检查文件的扩展名是.h ，是这样的：

const char *ext = strrchr (entry->d_name, '.');     
if ((ext != NULL) && (!strcmp (ext+1, "h"))) {
     // header file
}

Ofcourse，请注意，这是假定所有的头文件有一个.h扩展，这可能会或可能不会是真实的，C标准并不要求头文件必须有一个.h扩展。

Answer 2:

每个dirent结构具有d_name包含文件的文件名，所以我会看看是否跟随一些模式，比如在结束.h或.hpp 。

这将是沿着线的代码：

int len = strlen (entry->d_name);
if ((len >= 2) && strcmp (&(entry->d_name[len - 2]), ".h") == 0))
    header_count++;
if ((len >= 4) && strcmp (&(entry->d_name[len - 4]), ".hpp") == 0))
    header_count++;

当然，这不会赶上真正邪恶的人把他们的可执行文件ha_ha_fooled_you.hpp但thanfkfully他们是少数。

你甚至可能要考虑的endsWith()函数，使您的生活更轻松：

int endsWith (char *str, char *end) {
    size_t slen = strlen (str);
    size_t elen = strlen (end);
    if (slen < elen)
        return 0;
    return (strcmp (&(str[slen-elen]), end) == 0);
}
:
if (endsWith (entry->d_name, ".h"))   header_count++;
if (endsWith (entry->d_name, ".hpp")) header_count++;