draft support retrieve Exif from heif file

Signed-off-by: benstone <[email protected]>

Author: benstonezhang

ext/exif/exif.c

@@ -1284,6 +1284,18 @@ typedef struct {
 	mn_offset_mode_t offset_mode;
 } maker_note_type;
 
+#define FOURCC(id) (((uint32_t)(id[0])<<24) | (id[1]<<16) | (id[2]<<8) | (id[3]))
+
+typedef struct {
+	uint64_t	size;
+	uint32_t	type;
+} isobmff_box_type;
+
+typedef struct {
+	uint32_t	offset;
+	uint32_t	size;
+} isobmff_item_pos_type;
+
 /* Some maker notes (e.g. DJI info tag) require custom parsing */
 #define REQUIRES_CUSTOM_PARSING NULL
 
@@ -4285,11 +4297,125 @@ static bool exif_process_IFD_in_TIFF(image_info_type *ImageInfo, size_t dir_offs
 	return result;
 }
 
+static int exif_isobmff_parse_box(unsigned char *buf, isobmff_box_type *box)
+{
+	box->size = php_ifd_get32u(buf, 1);
+	buf += 4;
+	box->type = php_ifd_get32u(buf, 1);
+	if (box->size != 1) {
+		return 8;
+	}
+	buf += 4;
+	box->size = php_ifd_get64u(buf, 1);
+	return 16;
+}
+
+static void exif_isobmff_parse_meta(unsigned char *data, unsigned char *end, isobmff_item_pos_type *pos)
+{
+	isobmff_box_type box, item;
+	unsigned char *box_offset, *p, *p2;
+	int header_size, exif_id = -1, version, item_count, i;
+
+	for (box_offset = data + 4; box_offset < end; box_offset += box.size) {
+		header_size = exif_isobmff_parse_box(box_offset, &box);
+		if (box.type == FOURCC("iinf")) {
+			p = box_offset + header_size;
+			version = p[0];
+			p += 4;
+			if (version < 2) {
+				item_count = php_ifd_get16u(p, 1);
+				p += 2;
+			} else {
+				item_count = php_ifd_get32u(p, 1);
+				p += 4;
+			}
+			for (i=0; i<item_count; i++) {
+				header_size = exif_isobmff_parse_box(p, &item);
+				if (!memcmp(p + header_size + 8, "Exif", 4)) {
+					exif_id = php_ifd_get16u(p + header_size + 4, 1);
+					break;
+				}
+				p += item.size;
+			}
+			if (exif_id < 0) {
+				break;
+			}
+		}
+		else if (box.type == FOURCC("iloc")) {
+			p = box_offset + header_size;
+			version = p[0];
+			p += 6;
+			if (version < 2) {
+				item_count = php_ifd_get16u(p, 1);
+				p += 2;
+			} else {
+				item_count = php_ifd_get32u(p, 1);
+				p += 4;
+			}
+			for (i=0, p2=p; i<item_count; i++, p2 += 16) {
+				fflush(stdout);
+				if (php_ifd_get16u(p2, 1) == exif_id) {
+					pos->offset = php_ifd_get32u(p2 + 8, 1);
+					pos->size = php_ifd_get32u(p2 + 12, 1);
+					break;
+				}
+			}
+			break;
+		}
+	}
+}
+
+static bool exif_scan_HEIF_header(image_info_type *ImageInfo, unsigned char *buf)
+{
+	isobmff_box_type box;
+	isobmff_item_pos_type pos;
+	unsigned char *data;
+	off_t offset;
+	uint64_t limit;
+	int box_header_size, remain;
+	bool ret = false;
+
+	pos.size = 0;
+	for (offset = php_ifd_get32u(buf, 1); ImageInfo->FileSize > offset + 16; offset += box.size) {
+		if ((php_stream_seek(ImageInfo->infile, offset, SEEK_SET) < 0) ||
+			(exif_read_from_stream_file_looped(ImageInfo->infile, (char*)buf, 16) != 16)) {
+			break;
+		}
+		box_header_size = exif_isobmff_parse_box(buf, &box);
+		if (box.type == FOURCC("meta")) {
+			limit = box.size - box_header_size;
+			data = (unsigned char *)safe_emalloc(1, limit, 0);
+			remain = 16 - box_header_size;
+			if (remain) {
+				memcpy(data, buf + box_header_size, remain);
+			}
+			if (exif_read_from_stream_file_looped(ImageInfo->infile, data + remain, limit - remain) == limit - remain) {
+				exif_isobmff_parse_meta(data, data + limit, &pos);
+			}
+			efree(data);
+			if ((pos.size) &&
+				(ImageInfo->FileSize >= pos.offset + pos.size) &&
+				(php_stream_seek(ImageInfo->infile, pos.offset + 2, SEEK_SET) >= 0)) {
+				limit = pos.size - 2;
+				data = (unsigned char *)safe_emalloc(1, limit, 0);
+				if (exif_read_from_stream_file_looped(ImageInfo->infile, data, limit) == limit) {
+					exif_process_APP1(ImageInfo, data, limit, pos.offset + 2);
+					ret = true;
+				}
+				efree(data);
+			}
+			break;
+		}
+	}
+
+	return ret;
+}
+
 /* {{{ exif_scan_FILE_header
  * Parse the marker stream until SOS or EOI is seen; */
 static bool exif_scan_FILE_header(image_info_type *ImageInfo)
 {
-	unsigned char file_header[8];
+	unsigned char file_header[16];
 	bool ret = false;
 
 	ImageInfo->FileType = IMAGE_FILETYPE_UNKNOWN;
@@ -4338,6 +4464,16 @@ static bool exif_scan_FILE_header(image_info_type *ImageInfo)
 				} else {
 					exif_error_docref(NULL EXIFERR_CC, ImageInfo, E_WARNING, "Invalid TIFF file");
 				}
+			} else if ((ImageInfo->FileSize > 12) &&
+				(!memcmp(file_header + 4, "ftyp", 4)) &&
+				(exif_read_from_stream_file_looped(ImageInfo->infile, (char*)(file_header + 8), 4) == 4) &&
+				((!memcmp(file_header + 8, "heic", 4)) || (!memcmp(file_header + 8, "heix", 4)) || (!memcmp(file_header + 8, "mif1", 4)))) {
+					if (exif_scan_HEIF_header(ImageInfo, file_header)) {
+						ImageInfo->FileType = IMAGE_FILETYPE_HEIF;
+						ret = true;
+					} else {
+						exif_error_docref(NULL EXIFERR_CC, ImageInfo, E_WARNING, "Invalid HEIF file");
+					}
 			} else {
 				exif_error_docref(NULL EXIFERR_CC, ImageInfo, E_WARNING, "File not supported");
 				return false;

ext/standard/php_image.h

@@ -44,6 +44,7 @@ typedef enum
   IMAGE_FILETYPE_ICO,
   IMAGE_FILETYPE_WEBP,
   IMAGE_FILETYPE_AVIF,
+  IMAGE_FILETYPE_HEIF,
 /* WHEN EXTENDING: PLEASE ALSO REGISTER IN basic_function.stub.php */
   IMAGE_FILETYPE_COUNT
 } image_filetype;