From e7ac67e0b5327e21fa4726c3795011a065c3b89c Mon Sep 17 00:00:00 2001
From: Hector Martin
Date: Wed, 17 Nov 2021 15:04:58 +0900
Subject: ipsw: Add a function to list the archive contents (Monterey)

This is required to enumerate the BootabilityBundle contents.

Signed-off-by: Hector Martin <marcan@marcan.st>
---
 src/ipsw.c | 120 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 src/ipsw.h |   4 +++
 2 files changed, 124 insertions(+)

(limited to 'src')

diff --git a/src/ipsw.c b/src/ipsw.c
index 844328c..7bebd56 100644
--- a/src/ipsw.c
+++ b/src/ipsw.c
@@ -31,6 +31,8 @@
 #include <errno.h>
 #include <limits.h>
 #include <sys/stat.h>
+#include <sys/types.h>
+#include <dirent.h>
 #include <zip.h>
 #ifdef HAVE_OPENSSL
 #include <openssl/sha.h>
@@ -715,6 +717,124 @@ int ipsw_extract_restore_plist(const char* ipsw, plist_t* restore_plist)
 	return -1;
 }
 
+static int ipsw_list_contents_recurse(ipsw_archive *archive, const char *path, ipsw_list_cb cb, void *ctx)
+{
+	int ret = 0;
+	char *base = build_path(archive->path, path);
+
+	DIR *dirp = opendir(base);
+
+	if (!dirp) {
+		error("ERROR: failed to open directory %s\n", base);
+		free(base);
+		return -1;
+	}
+
+	while (ret >= 0) {
+		struct dirent *dir = readdir(dirp);
+		if (!dir)
+			break;
+
+		if (!strcmp(dir->d_name, ".") || !strcmp(dir->d_name, ".."))
+			continue;
+
+		char *fpath = build_path(base, dir->d_name);
+		char *subpath;
+		if (*path)
+			subpath = build_path(path, dir->d_name);
+		else
+			subpath = strdup(dir->d_name);
+
+		struct stat st;
+		ret = lstat(fpath, &st);
+		if (ret != 0) {
+			error("ERROR: failed to stat %s\n", fpath);
+			free(fpath);
+			free(subpath);
+			break;
+		}
+
+		ret = cb(ctx, archive->path, subpath, &st);
+
+		if (ret >= 0 && S_ISDIR(st.st_mode))
+			ipsw_list_contents_recurse(archive, subpath, cb, ctx);
+
+		free(fpath);
+		free(subpath);
+	}
+
+	closedir(dirp);
+	free(base);
+	return ret;
+}
+
+int ipsw_list_contents(const char* ipsw, ipsw_list_cb cb, void *ctx)
+{
+	int ret = 0;
+
+	ipsw_archive* archive = ipsw_open(ipsw);
+	if (archive == NULL) {
+		error("ERROR: Invalid archive\n");
+		return -1;
+	}
+
+	if (archive->zip) {
+        int64_t entries = zip_get_num_entries(archive->zip, 0);
+		if (entries < 0) {
+			error("ERROR: zip_get_num_entries failed\n");
+			ipsw_close(archive);
+			return -1;
+		}
+
+		for (int64_t index = 0; index < entries; index++) {
+			zip_stat_t stat;
+
+			zip_stat_init(&stat);
+			if (zip_stat_index(archive->zip, index, 0, &stat) < 0) {
+				error("ERROR: zip_stat_index failed for %s\n", stat.name);
+				ret = -1;
+				continue;
+			}
+
+			uint8_t opsys;
+			uint32_t attributes;
+			if (zip_file_get_external_attributes(archive->zip, index, 0, &opsys, &attributes) < 0) {
+				error("ERROR: zip_file_get_external_attributes failed for %s\n", stat.name);
+				ret = -1;
+				continue;
+			}
+			if (opsys != ZIP_OPSYS_UNIX) {
+				error("ERROR: File %s does not have UNIX attributes\n", stat.name);
+				ret = -1;
+				continue;
+			}
+
+			struct stat st;
+			memset(&st, 0, sizeof(st));
+			st.st_ino = 1 + index;
+			st.st_nlink = 1;
+			st.st_mode = attributes >> 16;
+			st.st_size = stat.size;
+
+			char *name = strdup(stat.name);
+			if (name[strlen(name) - 1] == '/')
+				name[strlen(name) - 1] = '\0';
+
+			ret = cb(ctx, ipsw, name, &st);
+
+			free(name);
+
+			if (ret < 0)
+				break;
+		}
+	} else {
+		ret = ipsw_list_contents_recurse(archive, "", cb, ctx);
+	}
+
+	ipsw_close(archive);
+	return ret;
+}
+
 void ipsw_close(ipsw_archive* archive)
 {
 	if (archive != NULL) {
diff --git a/src/ipsw.h b/src/ipsw.h
index 5c75699..3b5da80 100644
--- a/src/ipsw.h
+++ b/src/ipsw.h
@@ -30,9 +30,12 @@ extern "C" {
 
 #include <stdint.h>
 #include <plist/plist.h>
+#include <sys/stat.h>
 
 int ipsw_print_info(const char* ipsw);
 
+typedef int (*ipsw_list_cb)(void *ctx, const char* ipsw, const char *name, struct stat *stat);
+
 int ipsw_is_directory(const char* ipsw);
 int ipsw_file_exists(const char* ipsw, const char* infile);
 int ipsw_get_file_size(const char* ipsw, const char* infile, uint64_t* size);
@@ -41,6 +44,7 @@ int ipsw_extract_to_file_with_progress(const char* ipsw, const char* infile, con
 int ipsw_extract_to_memory(const char* ipsw, const char* infile, unsigned char** pbuffer, unsigned int* psize);
 int ipsw_extract_build_manifest(const char* ipsw, plist_t* buildmanifest, int *tss_enabled);
 int ipsw_extract_restore_plist(const char* ipsw, plist_t* restore_plist);
+int ipsw_list_contents(const char* ipsw, ipsw_list_cb cb, void *ctx);
 
 int ipsw_get_signed_firmwares(const char* product, plist_t* firmwares);
 int ipsw_download_fw(const char *fwurl, unsigned char* isha1, const char* todir, char** ipswfile);
-- 
cgit v1.1-32-gdbae