C++获取扩展名匹配扩展名

最新推荐文章于 2022-12-12 10:02:21 发布

jimmylopez

最新推荐文章于 2022-12-12 10:02:21 发布

阅读量1.5k

点赞数

CC 4.0 BY-SA版权

分类专栏：专业

本文链接：https://blog.youkuaiyun.com/jimmylopez/article/details/12148391

专业专栏收录该内容

5 篇文章

订阅专栏

该篇博客介绍了如何在C++中编写两个通用函数，一个用于获取文件的扩展名，另一个用于匹配文件扩展名是否符合给定的扩展名列表。通过使用字符串处理函数如`substr`和`strtok`，实现了从文件名中提取扩展名并进行匹配的功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <iostream>
#incldue <cstring>
#include <stdlib.h>

using namespace std;

//通用函数获取扩展名
inline string getExtension(const string &fileName)
{
     string extension;
     if(fileName.rfind('.') != string::npos){
          extension = fileName.substr(fileName.rfind('.') + 1);
     }
     return extension;
}

//通用函数匹配扩展名
bool matchExtension(const string &fileName,const string &extension)
{
     char *str = (char *)malloc(extension.length() + 1);
     char *ext = (char *)malloc(getExtension(fileName).length() + 1);
     strcpy(str,extension.c_str());
     strcpy(ext,getExtension(fileName).c_str());

     bool isMatch = false;
     char *p = strtok(str,"|");

     while(p && !isMatch){
          if(strcmp(ext,p) == 0){
               isMatch = true;
          }
          p = strtok(NULL,"|");
     }
     free(str);
     free(ext);

     return isMatch;
}