linux 下查找圖片文件方法

　　通常是通過文件后綴名查找圖片文件，如果沒有文件后綴的圖片或者偽造的圖片文件，則這種判定方法將達不到要求。我們可以根據讀取文件頭進行圖片文件類型的判定。

　　比較流行的圖片文件類型有：jpg png bmp gif 這幾種，下面將介紹區分這幾種圖片的方式：

 

　　BMP (bmp)  文件頭(2 byte)： 　　0x42,0x4D

　　JPEG (jpg)  文件頭(3 byte)：　　 0xFF,0xD8,0xFF

　　PNG (png)  文件頭(4 byte)：　　0x89,0x50,0x4E,0x47

　　GIF (gif)     文件頭(4byte)：　　  0x47,0x49,0x46,0x38

　　應用程序通過可以讀取文件頭進行匹配。linux 下測試例子：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <dirent.h>

int main(int argc, const char *argv[])
{
    int ret = 0;
    DIR * dir;
    struct dirent * info;
//    int i=0;

    if(argc != 2){
        printf("Usage:  %s /path/to/file",argv[0]);
        return -1;
    }

    dir = opendir(argv[1]);
    if(!dir){
        perror("opendir");
        return -1;
    }

    while((info = readdir(dir)) != NULL){
        struct stat file_sta;
        ret = stat(info->d_name,&file_sta);
        if(ret < 0){
            perror("stat");
            return -1;
        }

        if(S_ISREG(file_sta.st_mode)){
            printf("%s \n",info->d_name);
            FILE *fp = fopen(info->d_name,"rb");
            if(!fp){
                perror("fopen");
                continue;
            }
            // 此處的類型定義必須定義為 unsigned 類型 （不然編譯器會當做有符號類型處理）
            unsigned  char buf[4] = {0};
            ret = fread(buf,4,1,fp);

            if((buf[0] == 0x42) && (buf[1] == 0x4D)){
                printf("file type(bmp) : %s \n",info->d_name);
            }
            else if((buf[0] == 0xFF) && (buf[1] == 0xD8) && (buf[2] == 0xFF)){
                printf("file type(jpg) : %s \n",info->d_name);
            }
            else if((buf[0] == 0x89) && (buf[1] == 0x50) && (buf[2] == 0x4e) && (buf[3] == 0x47)){
                printf("file type(png) : %s \n",info->d_name);
            }
            else if((buf[0] == 0x47) && (buf[1] == 0x49) && (buf[2] == 0x46) && (buf[3] == 0x38)){
                printf("file type(gif) : %s \n",info->d_name);
            }
            else{


            }
            fclose(fp);
        }
    }

    closedir(dir);

    return 0;
}

　　

　　運行結果：