add app0 loading

src/loader.c

@@ -1,39 +1,61 @@
 #include "loader.h"
 
-#include <stdio.h>
-#include <stdint.h>
+#include <stdlib.h>
+#include <memory.h>
+#include <assert.h>
 
-#include "util.h"
+#define memzero(buffer, size) memset(buffer, 0, size)
 
-static int load_segment(FILE* fp);
+static int load_segment(JPEG* jpeg, FILE* fp);
 
-static int load_rst_segment(FILE* fp, uint8_t n);
-static int load_app_segment(FILE* fp, uint8_t n);
+static int load_rst_segment(JPEG* jpeg, FILE* fp, uint8_t n);
+static int load_app_segment(JPEG* jpeg, FILE* fp, uint8_t n);
 
-static int load_app0_segment(FILE* fp);
+static int load_app0_segment(JPEG* jpeg, FILE* fp);
 
-int load_jpeg(const char* filename)
+JPEG* load_jpeg(const char* filename)
 {
 	FILE* fp = fopen(filename, "r");
 	if (fp == NULL)
 	{
-		return 1;
+		return NULL;
 	}
 
+	JPEG* jpeg = (JPEG*)malloc(sizeof(jpeg));
+	memzero(jpeg, sizeof(JPEG));
+
 	while (!feof(fp))
 	{
-		if (load_segment(fp) != 0)
+		if (load_segment(jpeg, fp) != 0)
 		{
 			fprintf(stderr, "Segment loading failed\n");
-			return 1;
+			free_jpeg(jpeg);
+
+			return NULL;
 		}
 	}
 
 	fclose(fp);
-	return 0;
+	return jpeg;
 }
 
-int load_segment(FILE* fp)
+void free_jpeg(JPEG* jpeg)
+{
+	if (jpeg == NULL)
+		return;
+
+	if (jpeg->app0)
+	{
+		if (jpeg->app0->thumbnail_data)
+		{
+			free(jpeg->app0->thumbnail_data);
+		}
+
+		free(jpeg->app0);
+	}
+}
+
+int load_segment(JPEG* jpeg, FILE* fp)
 {
 	uint8_t segment_marker[2];
 	size_t segment_marker_size = sizeof(segment_marker);
@@ -53,14 +75,14 @@ int load_segment(FILE* fp)
 	// Handle special APPn/RSTn markers
 	if (segment_marker[1] >= 0xD0 && segment_marker[1] <= 0xD7)
 	{
-		if (load_rst_segment(fp, segment_marker[1] | 0x0F) != 0)
+		if (load_rst_segment(jpeg, fp, segment_marker[1] | 0x0F) != 0)
 		{
 			return 1;
 		}
 	}
 	else if ((segment_marker[1] & 0xF0) == 0xE0)
 	{
-		if (load_app_segment(fp, segment_marker[1] & 0x0F) != 0)
+		if (load_app_segment(jpeg, fp, segment_marker[1] & 0x0F) != 0)
 		{
 			return 1;
 		}
@@ -82,19 +104,19 @@ int load_segment(FILE* fp)
 	return 0;
 }
 
-int load_rst_segment(FILE* fp, uint8_t n)
+int load_rst_segment(JPEG* jpeg, FILE* fp, uint8_t n)
 {
 	DEBUG_LOG("RST%d marker encountered", n);
 	return 1;
 }
 
-int load_app_segment(FILE* fp, uint8_t n)
+int load_app_segment(JPEG* jpeg, FILE* fp, uint8_t n)
 {
 	DEBUG_LOG("APP%d marker encountered", n);
 
 	switch (n)
 	{
-	case 0:	return load_app0_segment();
+	case 0:	return load_app0_segment(jpeg, fp);
 
 	default: 
 		fprintf(stderr, "Unknown APP segment ID %d\n", n);
@@ -104,7 +126,68 @@ int load_app_segment(FILE* fp, uint8_t n)
 	return 0;
 }
 
-int load_app0_segment(FILE* fp)
+int load_app0_segment(JPEG* jpeg, FILE* fp)
 {
+	if (jpeg->app0 != NULL)
+	{
+		fprintf(stderr, "Found more than one APP0 marker\n");
+		return 1;
+	}
 
+	assert(fp);
+
+	jpeg->app0 = (JFIFAPP0Segment*)malloc(sizeof(JFIFAPP0Segment));
+	memzero(jpeg->app0, sizeof(JFIFAPP0Segment));
+
+	if (jpeg->app0 == NULL)
+	{
+		fprintf(stderr, "Failed to allocate memory for APP0 header\n");
+		return 1;
+	}
+
+	// Extract header without thumbnail data
+	size_t jfif_header_size = sizeof(JFIFAPP0Segment) - sizeof(uint8_t*);
+	if (fread(jpeg->app0, sizeof(uint8_t), jfif_header_size, fp) != jfif_header_size)
+	{
+		fprintf(stderr, "Incomplete APP0 header\n");
+		return 1;
+	}
+
+	jpeg->app0->length = bswap_16(jpeg->app0->length);
+	jpeg->app0->density_x = bswap_16(jpeg->app0->density_x);
+	jpeg->app0->density_y = bswap_16(jpeg->app0->density_y);
+
+	size_t thumbnail_data_size = jpeg->app0->thumbnail_x * jpeg->app0->thumbnail_y;
+	if (thumbnail_data_size > 0)
+	{
+		jpeg->app0->thumbnail_data = (uint8_t*)malloc(thumbnail_data_size * sizeof(uint8_t));
+		if (jpeg->app0->thumbnail_data == NULL)
+		{
+			fprintf(stderr, "Failed to allocate memory for thumbnail data\n");
+			return 1;
+		}
+
+		if (fread(jpeg->app0->thumbnail_data, sizeof(uint8_t), thumbnail_data_size, fp) != thumbnail_data_size)
+		{
+			fprintf(stderr, "Incomplete thumbnail data\n");
+			return 1;
+		}
+	}
+
+	DEBUG_LOG(
+		"JFIFAPP0Segment\n"
+		"length = %u\n"
+		"identifier = %s\n"
+		"version = %u.%02u\n"
+		"density units = %u\n"
+		"density x, y = %u, %u\n"
+		"thumbnail x, y = %u, %u",
+		jpeg->app0->length,
+		jpeg->app0->identifier,
+		jpeg->app0->version.major, jpeg->app0->version.minor,
+		jpeg->app0->density_units,
+		jpeg->app0->density_x, jpeg->app0->density_y,
+		jpeg->app0->thumbnail_x, jpeg->app0->thumbnail_y
+	);
+	return 0;
 }