Windows C++ 判斷文件是否是圖片格式的方法。

一、通過后綴名去判斷。

bool IsImageByTail(const std::wstring &path)
{
    std::wstring file_exten;
    size_t pos = path.rfind(L'.');
    if (pos == std::wstring::npos)
        return false;
    file_exten = path.substr(pos, std::wstring::npos);
    //把file_exten轉小寫
    for (size_t u = 0; u < file_exten.length();u++)
    {
        if (file_exten[u] >= L'A' && file_exten[u]<='Z')
        {
            file_exten[u] += L'a' - L'A';
        }
    }
    if (file_exten == L".jpg" || file_exten == L".tif"
        || file_exten == L".png" || file_exten == L".bmp"
        || file_exten == L".gif" || file_exten == L".ico")
        return true;
    return false;
}

            優點：效率快，不用讀取整個文件，無依賴，理解簡單。

            缺點：不准確，用戶可以通過修改后綴名蒙混過關。 

             

二、通過文件頭判斷

           我們知道，圖片文件的文件頭帶有圖片標記信息，常見的如下：

           JPEG (jpg)，文件頭：FFD8FF

           PNG (png)，文件頭：89504E47

           GIF (gif)，文件頭：47494638

           TIFF (tif)，文件頭：49492A00

           Windows Bitmap (bmp)，文件頭：424D

bool IsImageByHead(const std::wstring &path)
{
    //讀取文件首部4個字節
    HANDLE hFile = CreateFile(path.c_str(), FILE_GENERIC_READ,      //  打開文件，獲得文件讀句柄
        FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE,     //  共享方式打開，避免其他地方需要讀寫此文件
        NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);

    if (INVALID_HANDLE_VALUE == hFile)                              //  文件打開失敗，返回錯誤值
        return false;
    BYTE data[4] = { 0 };
    DWORD readSize;
    bool ok=false;
    if(ReadFile(hFile, data, 4, &readSize, NULL))
    {
        if (readSize == 4)
        {
            if (data[0] == 0xFF && data[1]==0xD8 && data[2]==0xFF)
            {
                ok = true;
            }
            else if (data[0] == 0x89 && data[1] == 0x50 && data[2] == 0x4E && data[3] == 0x47)
            {
                ok = true;
            }
            else if (data[0] == 0x47 && data[1] == 0x49 && data[2] == 0x46 && data[3] == 0x38)
            {
                ok = true;
            }
            else if (data[0] == 0x49 && data[1] == 0x49 && data[2] == 0x2A && data[3] == 0x00)
            {
                ok = true;
            }
            else if (data[0] == 0x42 && data[1] == 0x4D)
            {
                ok = true;
            }
        }
    }
    CloseHandle(hFile);                                             //  關閉文件句柄，避免句柄泄露
    return ok;
}

            優點：只需要文件讀取函數，實現邏輯較簡單。

            缺點：不准確，圖片可能不完整，頭部可能被偽造。

         

三、嚴格的讀取文件頭，匹配內部的長度、寬度與文件大小，校驗值等。

          這個算法偏復雜，需要對各類圖片格式了如指掌，解析到位。

          這里不實現了。

            

          優點：准確無誤。

          缺點：實現難度復雜，需要對各類圖片了如指掌。

 

四、通過GDI+來解析圖片，判斷圖片的有效性

          感謝GDI+，幫助我們做了解析圖片的格式與內容。並提供Image類統一管理。

          

#include <gdiplus.h>
#pragma comment(lib,"gdiplus.lib")
bool IsImageByGDI(const std::wstring &path)
{
    Gdiplus::Image image_src(path.c_str());
    Gdiplus::Status status = image_src.GetLastStatus();
    if (status != Gdiplus::Ok)
    {
        return false;
    }
    GUID guid;
    if (image_src.GetRawFormat(&guid) != Gdiplus::Ok)
    {
        return false;
    }
    if (guid == Gdiplus::ImageFormatGIF || guid == Gdiplus::ImageFormatJPEG || guid == Gdiplus::ImageFormatPNG
        || guid == Gdiplus::ImageFormatBMP || guid == Gdiplus::ImageFormatIcon || guid == Gdiplus::ImageFormatTIFF)
    {
        return true;
    }
    return false;
}

         優點：准確無誤、實現簡單。

         缺點：需要依賴GDI+庫，效率比較低。

 

         當然，如果我們不需要准確的判斷的話。上述函數可以結合使用。

         如：bool isPic  =  IsImageByTail  ||   IsImageByHead。可以解決無后綴的圖片的判斷，並且依賴較低。

                bool isPic  =  IsImageByTail  ||   IsImageByGDI 。可以通過后綴名提前過濾一遍，加快效率。

 

        另外：上述函數稍微修改下，也可以返回具體的圖片類型（到底是jpg呢還是png呢）。

 

        另外：GDI+真的是不錯的東西！

        本來想更新一個GDI+系列的，感覺關注的人不多，動力不足。

        從零開始學習GDI+ （一）我的第一個GDI+程序