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

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

原创最新推荐文章于 2022-12-12 10:02:21 发布 · 1.5k 阅读

0 ·

CC 4.0 BY-SA版权

专业专栏收录该内容

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;
}