使用libmagic确定文件MIME类型【示例】

柔情只为你懂 2022-08-05 01:15 182阅读 0赞

**引用：   《http://blog.163.com/lixiangqiu\_9202/blog/static/53575037201481310276358/》**

没事写一个上传图片到imagebin.org的程序，但发现imagebin.org限制了上传的文件类型，这当然是可以理解的，imagebin.org为检查你上传的图片文件，如果与上传的文件MIME类型与Content-Type不一致的话会无法上传，我们当然可以手动根据自己上传的文件而设置Content-Type，但显然作为一个懒人，我更希望能够根据图片文件本身自动设置这个参数。

这个看起来根据文件的后缀名来确定Content-Type是一个不错的选择，毕竟是自己使用，但如果能有个库根据文件识别出MIME文件类型的话，那么我一定会很乐意的，没错，就是抱着这种心态我找到了libmagic，这是一个c语言库，[Linux][]上file命令便是使用了这个库。

在linux下安装使用这个库非常简单，直接使用包管理器就可以安装了，我们安装后man libmagic可以查看到它的文档，libmagic可以做的不仅仅是检查文件的MIME类型，但这里我只说获取文件MIME类型，至于其它的，请自行查看libmagic文档。

想要查看文件MIME类型只需要知道4个函数就可以了，  
magic\_t magic\_open(int flags);  
int magic\_load(magic\_t cookie, const char \*filename);  
void magic\_close(magic\_t cookie);  
const char \*magic\_file(magic\_t cookie, const char \*filename);

头文件

1.  \#include <magic.h>

gcc编译时需要加上-lmagic参数

看函数名称就一目了然，先使用magic\_open创建一个magic\_t数据结构  
再magic\_load加载数据库  
此时便可以使用magic\_file进行检测了，使用完成后使用magic\_close释放资源  
可以说这是libmagic的使用方式。

想要获取文件MIME类型在调用magic\_open时我们的flags要设置为MAGIC\_MIME\_TYPE，flags的具体参数可以参考libmagic文档  
加载数据库时我们设置NULL来使用默认的数据库，默认的数据库linux一般是/usr/share/misc/magic.mgc，这取决于MAGIC这个系统变量。

下面是一个简单的示例

1.  \#include <stdio.h>
2.  \#include <magic.h>
3.   
4.  int main(int argc,char \*\*argv)
5.  \{ 
6.   magic\_t cookie;
7.   
8.   cookie=magic\_open(MAGIC\_MIME\_TYPE);
9.   magic\_load(cookie,NULL);
10.  puts(magic\_file(cookie,argv\[1\]));
11.  magic\_close(cookie);
12.  
13.  return 0;
14. \}

这切就是这么简单

下面干脆把上传图片到imagebin.org的程序也放上来吧，以免日后不小心丢掉了

1.  \#include <stdio.h>
2.  \#include <magic.h>
3.  \#include <curl/curl.h>
4.   
5.  size\_t filtration\_data(void \*ptr,size\_t size,size\_t nmemb,void \*data)
6.  \{ 
7.   return nmemb;
8.  \}
9.   
10. int display\_progress(void \*clientp,double dltotal,double dlnow,
11.   double ultotal,double ulnow)
12. \{ 
13.  static int onece=0;
14.  
15.  if(ulnow/ultotal > 0)
16.  \{ 
17.   if(ulnow >= ultotal)
18.   \{ 
19.    if(onece) return 0;
20.  
21.    onece=1;
22.   \}
23.  
24.   printf("…%.0f%%",ulnow/ultotal\*100);
25.   fflush(stdout);
26.  \}
27.  
28.  return 0;
29. \}
30.  
31. int main(int argc,char \*\*argv)
32. \{ 
33.  CURL \*curl;
34.  char \*url="http://imagebin.org/index.php";
35.  struct curl\_httppost \*post=NULL,\*last=NULL;
36.  char \*res;
37.  const char \*content\_type;
38.  magic\_t cookie;
39.  
40.  cookie=magic\_open(MAGIC\_MIME\_TYPE);
41.  if(!cookie)
42.  \{ 
43.   perror("magic\_open");
44.   return -1;
45.  \}
46.  
47.  if(magic\_load(cookie,NULL) == -1)
48.  \{ 
49.   printf("%s\\n",magic\_error(cookie));
50.   magic\_close(cookie);
51.   return -1;
52.  \}
53.  
54.  content\_type=magic\_file(cookie,argv\[1\]);
55.  if(!content\_type)
56.  \{ 
57.   puts(magic\_error(cookie));
58.   magic\_close(cookie);
59.   return -1;
60.  \}
61.  
62.  curl\_formadd(&post,&last,CURLFORM\_COPYNAME,"nickname",
63.    CURLFORM\_COPYCONTENTS,"test",CURLFORM\_END);
64.  curl\_formadd(&post,&last,CURLFORM\_COPYNAME,"remember\_nickname",
65.    CURLFORM\_COPYCONTENTS,"N",CURLFORM\_END);
66.  curl\_formadd(&post,&last,CURLFORM\_COPYNAME,"image",
67.    CURLFORM\_FILE,argv\[1\],
68.    CURLFORM\_CONTENTTYPE,content\_type,CURLFORM\_END);
69.  curl\_formadd(&post,&last,CURLFORM\_COPYNAME,"title",
70.    CURLFORM\_COPYCONTENTS,"test",CURLFORM\_END);
71.  curl\_formadd(&post,&last,CURLFORM\_COPYNAME,"description",
72.    CURLFORM\_COPYCONTENTS,"test",CURLFORM\_END);
73.  curl\_formadd(&post,&last,CURLFORM\_COPYNAME,"disclaimer\_agree",
74.    CURLFORM\_COPYCONTENTS,"Y",CURLFORM\_END);
75.  curl\_formadd(&post,&last,CURLFORM\_COPYNAME,"Submit",
76.    CURLFORM\_COPYCONTENTS,"Submit",CURLFORM\_END);
77.  curl\_formadd(&post,&last,CURLFORM\_COPYNAME,"mode",
78.    CURLFORM\_COPYCONTENTS,"add",CURLFORM\_END);
79.  
80.  curl=curl\_easy\_init();
81.  curl\_easy\_setopt(curl,CURLOPT\_URL,url);
82.  curl\_easy\_setopt(curl,CURLOPT\_HTTPPOST,post);
83.  curl\_easy\_setopt(curl,CURLOPT\_FOLLOWLOCATION,1);
84.  curl\_easy\_setopt(curl,CURLOPT\_WRITEFUNCTION,filtration\_data);
85.  curl\_easy\_setopt(curl,CURLOPT\_USERAGENT,"Mozilla/5.0 (X11; Linux i686; rv:31.0) Gecko/20100101 Firefox/31.0");
86.  curl\_easy\_setopt(curl,CURLOPT\_NOPROGRESS,0);
87.  curl\_easy\_setopt(curl,CURLOPT\_PROGRESSFUNCTION,display\_progress);
88.  
89.  curl\_easy\_perform(curl);
90.  curl\_easy\_getinfo(curl,CURLINFO\_EFFECTIVE\_URL,&res);
91.  printf("\\n%s\\n",res);
92.  curl\_easy\_cleanup(curl);
93.  curl\_formfree(post);
94.  magic\_close(cookie);
95.  
96.  return 0;
97. \}

**依葫芦 画瓢** ：

#include <stdio.h>
    #include <magic.h>
    
    int main(int argc,char **argv)
    {
     struct magic_set   *cookie = NULL;
    
     cookie=magic_open(MAGIC_MIME);
     magic_load(cookie, NULL);
     puts(magic_file(cookie,argv[1]));
     magic_close(cookie);
    
     return 0;
    }

[Linux]: http://briskblog.eu.org/?s=Linux