storm/w3m
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Determine the format of an image file by its header data not by its file name suffix.

Browse Source
This commit is contained in:
Araki Ken
2014-03-01 16:16:48 +09:00
committed by Tatsuya Kinoshita
parent 0c0d4b5f38
commit e6856f2329
1 changed files with 56 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

126
image.c
View File

@@ -628,86 +628,72 @@ getImage(Image * image, ParsedURL *current, int flag)
static int static int
parseImageHeader(char *path, u_int *width, u_int *height) parseImageHeader(char *path, u_int *width, u_int *height)
{ {
char *suffix; FILE *fp;
u_char buf[8];
suffix = path + strlen(path); if (!(fp = fopen(path, "r"))) return FALSE;
if (strcasecmp(suffix - 5, ".jpeg") == 0 || strcasecmp(suffix - 4 , ".jpg") == 0) {
FILE *fp = fopen(path, "r");
if (fp) {
u_char buf[2];
if (fread(buf, 1, 2, fp) != 2 || memcmp(buf, "\xff\xd8", 2) != 0 ||
fseek(fp, 2, SEEK_CUR) < 0) /* 0xffe0 */
goto jpg_error;
while (fread(buf, 1, 2, fp) == 2) { if (fread(buf, 1, 2, fp) != 2) goto error;
size_t len = ((buf[0] << 8) | buf[1]) - 2;
if (fseek(fp, len, SEEK_CUR) < 0) goto jpg_error;
if (fread(buf, 1, 2, fp) == 2 && memcmp(buf, "\xff\xc0", 2) == 0) {
fseek(fp, 3, SEEK_CUR);
if (fread(buf, 1, 2, fp) == 2) {
*height = (buf[0] << 8) | buf[1];
if (fread(buf, 1, 2, fp) == 2) {
*width = (buf[0] << 8) | buf[1];
fclose(fp);
return TRUE;
}
}
break;
}
}
jpg_error: if (memcmp(buf, "\xff\xd8", 2) == 0) {
fclose(fp); /* JPEG */
return FALSE; if (fseek(fp, 2, SEEK_CUR) < 0) goto error; /* 0xffe0 */
} while (fread(buf, 1, 2, fp) == 2) {
} size_t len = ((buf[0] << 8) | buf[1]) - 2;
else if (strcasecmp(suffix - 4, ".png") == 0) { if (fseek(fp, len, SEEK_CUR) < 0) goto error;
FILE *fp = fopen(path, "r"); if (fread(buf, 1, 2, fp) == 2 &&
if (fp) { /* SOF0 or SOF2 */
u_char buf[8]; (memcmp(buf, "\xff\xc0", 2) == 0 || memcmp(buf, "\xff\xc2", 2) == 0)) {
if (fread(buf, 1, 8, fp) != 8 || fseek(fp, 3, SEEK_CUR);
memcmp(buf, "\x89\x50\x4e\x47\x0d\x0a\x1a\x0a", 8) != 0 ||
fseek(fp, 8, SEEK_CUR) < 0)
goto png_error;
if (fread(buf, 1, 4, fp) == 4) {
*width = (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) | buf[3];
if (fread(buf, 1, 4, fp) == 4) {
*height = (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) | buf[3];
fclose(fp);
return TRUE;
}
}
png_error:
fclose(fp);
return FALSE;
}
}
else if (strcasecmp(suffix - 4, ".gif") == 0) {
FILE *fp = fopen(path, "r");
if (fp) {
u_char buf[3];
if (fread(buf, 1, 3, fp) != 3 || memcmp(buf, "GIF", 3) != 0 ||
fseek(fp, 3, SEEK_CUR) < 0) {
goto png_error;
}
if (fread(buf, 1, 2, fp) == 2) {
*width = (buf[1] << 8) | buf[0];
if (fread(buf, 1, 2, fp) == 2) { if (fread(buf, 1, 2, fp) == 2) {
*height = (buf[1] << 8) | buf[0]; *height = (buf[0] << 8) | buf[1];
fclose(fp); if (fread(buf, 1, 2, fp) == 2) {
return TRUE; *width = (buf[0] << 8) | buf[1];
goto success;
}
} }
break;
} }
} }
goto error;
gif_error:
fclose(fp);
return FALSE;
} }
if (fread(buf + 2, 1, 1, fp) != 1) goto error;
if (memcmp(buf, "GIF", 3) == 0) {
/* GIF */
if (fseek(fp, 3, SEEK_CUR) < 0) goto error;
if (fread(buf, 1, 2, fp) == 2) {
*width = (buf[1] << 8) | buf[0];
if (fread(buf, 1, 2, fp) == 2) {
*height = (buf[1] << 8) | buf[0];
goto success;
}
}
}
if (fread(buf + 3, 1, 5, fp) != 5) goto error;
if (memcmp(buf, "\x89\x50\x4e\x47\x0d\x0a\x1a\x0a", 8) == 0) {
/* PNG */
if (fseek(fp, 8, SEEK_CUR) < 0) goto error;
if (fread(buf, 1, 4, fp) == 4) {
*width = (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) | buf[3];
if (fread(buf, 1, 4, fp) == 4) {
*height = (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) | buf[3];
goto success;
}
}
}
error:
fprintf(stderr,"%s\n",path);
fclose(fp);
return FALSE; return FALSE;
success:
fclose(fp);
return TRUE;
} }
int int