C/C++判斷某張jpg圖片是否完整

本文轉自：https://blog.csdn.net/10km/article/details/82263274

關於jpg圖片的編碼格式可以在網上找資料，本文主要用一段程序實現如何識別一張jpg圖片是否完整。

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <stdbool.h>
#include <iostream>

using namespace std;


static inline uint16_t archswap16(uint16_t D) {
    return((D << 8) | (D >> 8));
}
// gcc 預定義宏判斷大小端
#if __BYTE_ORDER__==__ORDER_LITTLE_ENDIAN__
#define arch_swap_be16(X)        archswap16(X)
#else
#define arch_swap_be16(X)        (X)
#endif

/**
* 從文件流中讀取一個大端的16位整數,
* 讀取成功返回true,否則返回false
*/
static bool read_be16(FILE *stream, /*uint16_t *value*/uint8_t *value)
{
    if (sizeof(*value) != fread(value, sizeof(*value), 1, stream))
        return false;
    *value = (arch_swap_be16(*value));
    return true;
}

/*
* Common JPEG markers(JPEG標記定義)
* 參見 https://en.wikipedia.org/wiki/JPEG
* */
#define JMK_SOI         0xFFD8     /* Start Of Image */
#define JMK_SOF0        0xFFC0     /* Start Of Frame (baseline DCT) */
#define JMK_SOF2        0xFFC2     /* Start Of Frame (progressive DCT) */
#define JMK_DHT         0xFFC4     /* Define Huffman Table(s) */
#define JMK_DQT         0xFFDB     /* Define Quantization Table(s) */
#define JMK_DRI         0xFFDD     /* Define Restart Interval */
#define JMK_SOS         0xFFDA     /* Start Of Scan */
#define JMK_RST_mask    0xFFD0     /* mask of Restart */
#define JMK_APP_mask    0xFFE0     /* mask of Application-specific */
#define JMK_COM         0xFFFE     /* Comment */
#define JMK_EOI         0xFFD9     /* End Of Image */

/**
* 判斷文件流是否為JPEG格式圖像.
* 邏輯說明:循環從文件流中讀取　JPEG 標記,直到遇到SOF0,SOF2標記，就返回true,否則返回false.
*/
bool check_jpg(FILE* stream)
{
    //uint16_t jpeg_marker;
    uint8_t jpeg_marker;
    if (!stream)
        return false;
    for (; read_be16(stream, &jpeg_marker) /* 讀取一個JPEG標記 */;)
    {
        /* 當前標記的數據長度(不含標記本身) */
        //uint16_t payload = 1; /* 設置為0或1用於指定當前JPEG 標記是否有附加數據*/
        uint8_t payload = 1; /* 設置為0或1用於指定當前JPEG 標記是否有附加數據*/
        switch (jpeg_marker)
        {
        case JMK_SOI:
            payload = 0; /* no payload,沒有附加數據 */
            break;
        case JMK_SOF0:
        case JMK_SOF2:
            return true; /* 找到SOF0,SOF2 marker,確認為JPEG 格式*/
        case JMK_DHT:
        case JMK_DQT:
        case JMK_DRI:
        case JMK_SOS:
        case JMK_COM:
            break;
        case JMK_EOI:
            return false; /* not JPEG image*/
        default:
            if ((0XFFF8 & jpeg_marker) == JMK_RST_mask) {
                payload = 0; /* RST0~7(FFD0~FFD7),no payload */
            }
            else if ((0XFFF0 & jpeg_marker) == JMK_APP_mask) {
                /* APP0~APP15,do nothing */
            }
            else
                return false; /* not JPEG image*/
        }
        if (payload) {
            /*讀取 JPEG 標記之后的附加數據長度字段，根據這個字段的值移動文件游標位置跳到下一個 JPEG 標記 */
            if (!read_be16(stream, &payload))
                return false; /* not JPEG image*/
            fseek(stream, payload - sizeof(payload), SEEK_CUR);
        }
    }
    return false; /* not JPEG image*/
}

int main(void)
{
    //保存輸入圖像文件名和輸出圖像文件名
    char InImgName[10];
    //圖像數據長度
    int length;
    //文件指針
    FILE* fp;
    //輸入要讀取的圖像名
    cout << "Enter Image name:";
    cin >> InImgName;
    //以二進制方式打開圖像
    if ((fp = fopen(InImgName, "rb")) == NULL)
    {
        cout << "Open image failed!" << endl;
        exit(0);
    }
    rewind(fp);
    if (check_jpg(fp)) {
        cout << "jpg file" << endl;
    }
    else {
        cout << "no jpg file" << endl;
    }

    fclose(fp);

    system("pause");

    return 0;
}