c++判断图片格式

<strong>转自：http://blog.csdn.net/zxlstudio/article/details/10362187</strong>

#include <stdio.h> #include <stdlib.h> #include <string.h> #include <assert.h> #include <errno.h> #define GETBYTELEN   2 #define IMAGE_YES    1 #define IMAGE_NO     0 /*---JPEG/JPG(1)---*/ #define IMAGE_JPEG_TYPE  0x01 #define IMAGE_JPEG_JPG "JPEG/JPG" #define JPEG_FIRSTBIT  "FF" #define JPEG_SECONDBIT "D8" /*---BMP(2)---*/ #define IMAGE_BMP_TYPE  0x02 #define IMAGE_BMP     "BMP" #define BMP_FIRSTBIT  "42" #define BMP_SECONDBIT "4D" /*---PNG(3)---*/ #define IMAGE_PNG_TYPE  0x04 #define IMAGE_PNG     "PNG" #define PNG_FIRSTBIT  "89" #define PNG_SECONDBIT "50" /*---GIF(4)---*/ #define IMAGE_GIF_TYPE 0x08 #define IMAGE_GIF     "GIF" #define GIF_FIRSTBIT  "47" #define GIF_SECONDBIT "49" #define GIF_THIRDBIT  "46" #define GIF_FIFTHBIT1 "39"   // 可能存在是37 #define GIF_FIFTHBIT2 "37" /*---TIFF(5)---*/ #define IMAGE_TIFF_TYPE  0x10 #define IMAGE_TIFF      "TIFF" #define TIFF_FIRSTBIT   "4D" #define TIFF_SECONDBIT  "4D" #define TIFF_FIRSTBIT2  "49"  // 或者  #define TIFF_SECONDBIT2 "49" /*---ICO(6)---*/ #define IMAGE_ICO    "ICO" #define ICO_THIRDBIT "1"  // 01 #define ICO_FIFTHBIT "1"  // 01 /*---TGA(7)---*/ #define IMAGE_TGA    "TGA" #define TGA_THIRDBIT "2"  // 02 #define TGA_FIFTHBIT "0"  // 00 /*---CUR(8)---*/ #define IMAGE_CUR    "CUR" #define CUR_THIRDBIT "2"  // 02 #define CUR_FIFTHBIT "1"  // 01 /*---PCX(9)---*/ #define IMAGE_PCX    "PCX" #define PCX_FIRSTBIT "A"  // 0A /*---IFF(10)---*/ #define IMAGE_IFF     "IFF" #define IFF_FIRSTBIT  "46" #define IFF_SECONDBIT "4F" #define IFF_THIRDBIT  "52" #define IFF_FORTHBIT  "4D" /*---ANI(11)---*/ #define IMAGE_ANI     "ANI" #define ANI_FIRSTBIT  "52" #define ANI_SECONDBIT "49" #define ANI_THIRDBIT  "46" #define ANI_FORTHBIT  "46" /* 根据图片文件头部数据判断*/int Check_Image_Header(void *FileData) { unsigned char *imagebuf = (unsigned char *)FileData; // unsigned char firstBit[3]; // unsigned char secondBit[3]; // unsigned char thirdBit[3]; // unsigned char forthBit[3]; // unsigned char fifthBit[3]; unsigned char checkBit[5][3]; if(NULL == imagebuf){ return IMAGE_NO; } snprintf(checkBit[0], sizeof(checkBit[0]), "%X", imagebuf[0]); snprintf(checkBit[1], sizeof(checkBit[1]), "%X", imagebuf[1]); snprintf(checkBit[2], sizeof(checkBit[2]), "%X", imagebuf[2]); snprintf(checkBit[3], sizeof(checkBit[3]), "%X", imagebuf[3]); snprintf(checkBit[4], sizeof(checkBit[4]), "%X", imagebuf[4]);  // snprintf(firstBit,  sizeof(firstBit),  "%X", imagebuf[0]); // snprintf(secondBit, sizeof(secondBit), "%X", imagebuf[1]); // snprintf(thirdBit,  sizeof(thirdBit),  "%X", imagebuf[2]); // snprintf(forthBit,  sizeof(forthBit),  "%X", imagebuf[3]); // snprintf(fifthBit,  sizeof(fifthBit),  "%X", imagebuf[4]); if( (0 == strncmp(checkBit[0], JPEG_FIRSTBIT,  GETBYTELEN)) &&  (0 == strncmp(checkBit[1], JPEG_SECONDBIT, GETBYTELEN))) { return IMAGE_YES; } if( (0 == strncmp(checkBit[0], BMP_FIRSTBIT,  GETBYTELEN)) &&  (0 == strncmp(checkBit[1], BMP_SECONDBIT, GETBYTELEN))) { return IMAGE_YES; } if( (0 == strncmp(checkBit[0], PNG_FIRSTBIT,   GETBYTELEN)) &&  (0 == strncmp(checkBit[1], PNG_SECONDBIT, GETBYTELEN))) { return IMAGE_YES; } if( (0 == strncmp(checkBit[0], GIF_FIRSTBIT,   GETBYTELEN))   && (0 == strncmp(checkBit[1], GIF_SECONDBIT, GETBYTELEN))   && (0 == strncmp(checkBit[2], GIF_THIRDBIT,   GETBYTELEN))   && ((0 == strncmp(checkBit[4], GIF_FIFTHBIT1, GETBYTELEN))   ||  (0 == strncmp(checkBit[4], GIF_FIFTHBIT2, GETBYTELEN)))) { return IMAGE_YES; } if( ((0 == strncmp(checkBit[0], TIFF_FIRSTBIT,    GETBYTELEN))  &&  (0 == strncmp(checkBit[1], TIFF_SECONDBIT,  GETBYTELEN))) ||    ((0 == strncmp(checkBit[0], TIFF_FIRSTBIT2,   GETBYTELEN))  &&     (0 == strncmp(checkBit[1], TIFF_SECONDBIT2, GETBYTELEN)))) { return IMAGE_YES; } if( (0 == strncmp(checkBit[2], ICO_THIRDBIT, GETBYTELEN)) &&  (0 == strncmp(checkBit[4], ICO_FIFTHBIT, GETBYTELEN))){ return IMAGE_YES; } if( (0 == strncmp(checkBit[2], TGA_THIRDBIT, GETBYTELEN)) &&  (0 == strncmp(checkBit[4], TGA_FIFTHBIT, GETBYTELEN))){ return IMAGE_YES; } if( (0 == strncmp(checkBit[2], CUR_THIRDBIT, GETBYTELEN)) &&  (0 == strncmp(checkBit[4], CUR_FIFTHBIT, GETBYTELEN))){ return IMAGE_YES; } if(0 == strncmp(checkBit[0], PCX_FIRSTBIT, GETBYTELEN)){ return IMAGE_YES; } if( (0 == strncmp(checkBit[0], IFF_FIRSTBIT,  GETBYTELEN))   && (0 == strncmp(checkBit[1], IFF_SECONDBIT, GETBYTELEN))   && (0 == strncmp(checkBit[2], IFF_THIRDBIT,   GETBYTELEN))   && (0 == strncmp(checkBit[3], IFF_FORTHBIT,   GETBYTELEN))) { return IMAGE_YES; } if( (0 == strncmp(checkBit[0], ANI_FIRSTBIT,   GETBYTELEN))   && (0 == strncmp(checkBit[1], ANI_SECONDBIT, GETBYTELEN))   && (0 == strncmp(checkBit[2], ANI_THIRDBIT,   GETBYTELEN))   && (0 == strncmp(checkBit[3], ANI_FORTHBIT,   GETBYTELEN))) { return IMAGE_YES; } return IMAGE_NO; } int main(int argc, char *argv[]) { int res = -1; char image_data[10]; FILE *fd = NULL; if(argc < 2){ printf("usage: %s <imagefilename>\n", argv[0]); exit(1); } assert(argv[1] != NULL); if( (fd = fopen(argv[1], "rb")) == NULL){ perror("fopen error\n"); exit(1); } fread(image_data, sizeof(unsigned char), sizeof(image_data), fd); printf("Image_head = %d\n", Check_Image_Header(image_data)); printf("name = %d\n", res); fclose(fd); return 0; }