Changeset 353 for trunk/util/ldid.cpp

Timestamp:

07/06/08 20:56:57 (6 months ago)

Author:

saurik

Message:

Added support for entitlement XML files.

Files:

• trunk/util/ldid.cpp (modified) (17 diffs)

trunk/util/ldid.cpp

@@ -52 +52 @@
     uint32_t magic;
     uint32_t nfat_arch;
-};
+} _packed;
 
 #define FAT_MAGIC 0xcafebabe
@@ -63 +63 @@
     uint32_t size;
     uint32_t align;
-};
+} _packed;
 
 struct mach_header {
@@ -73 +73 @@
     uint32_t sizeofcmds;
     uint32_t flags;
-};
+} _packed;
 
 #define MH_MAGIC 0xfeedface
@@ -86 +86 @@
     uint32_t cmd;
     uint32_t cmdsize;
-};
+} _packed;
 
 #define LC_REQ_DYLD  0x80000000
@@ -101 +101 @@
     uint32_t current_version;
     uint32_t compatibility_version;
-};
+} _packed;
 
 struct dylib_command {
@@ -107 +107 @@
     uint32_t cmdsize;
     struct dylib dylib;
-};
+} _packed;
 
 struct uuid_command {
@@ -113 +113 @@
     uint32_t cmdsize;
     uint8_t uuid[16];
-};
+} _packed;
 
 struct linkedit_data_command {
@@ -120 +120 @@
     uint32_t dataoff;
     uint32_t datasize;
-};
+} _packed;
 
 uint16_t Swap_(uint16_t value) {
@@ -246 +246 @@
 };
 
-#define CSMAGIC_CODEDIRECTORY 0xfade0c02
+#define CSMAGIC_CODEDIRECTORY      0xfade0c02
 #define CSMAGIC_EMBEDDED_SIGNATURE 0xfade0cc0
+#define CSMAGIC_ENTITLEMENTS       0xfade7171
+
 #define CSSLOT_CODEDIRECTORY 0
-#define CSSLOT_REQUIREMENTS 2
+#define CSSLOT_REQUIREMENTS  2
+#define CSSLOT_ENTITLEMENTS  5
 
 struct BlobIndex {
     uint32_t type;
     uint32_t offset;
-};
-
-struct SuperBlob {
+} _packed;
+
+struct Blob {
     uint32_t magic;
     uint32_t length;
+} _packed;
+
+struct SuperBlob {
+    struct Blob blob;
     uint32_t count;
     struct BlobIndex index[];
-};
+} _packed;
 
 struct CodeDirectory {
-    uint32_t magic;
-    uint32_t length;
+    struct Blob blob;
     uint32_t version;
     uint32_t flags;
@@ -278 +284 @@
     uint8_t pageSize;
     uint32_t spare2;
-};
+} _packed;
 
 extern "C" uint32_t hash(uint8_t *k, uint32_t length, uint32_t initval);
@@ -302 +308 @@
     bool timeh(false);
     uint32_t timev(0);
+
+    const void *xmld(NULL);
+    size_t xmls(0);
 
     std::vector<std::string> files;
@@ -314 +323 @@
             case 'u': flag_u = true; break;
             case 'p': flag_p = true; break;
-            case 'S': flag_S = true; break;
+
+            case 'S':
+                flag_S = true;
+                if (argv[argi][2] != '\0') {
+                    const char *xml = argv[argi] + 2;
+                    xmld = map(xml, 0, _not(size_t), &xmls, true);
+                }
+            break;
 
             case 'T': {
@@ -389 +405 @@
             if (pid == 0) {
                 char *ssize;
-                asprintf(&ssize, "%u", (sizeof(struct SuperBlob) + 2 * sizeof(struct BlobIndex) + sizeof(struct CodeDirectory) + strlen(base) + 1 + (size + 0x1000 - 1) / 0x1000 * 0x14 + 0xc + 15) / 16 * 16);
+                asprintf(&ssize, "%u", (sizeof(struct SuperBlob) + 2 * sizeof(struct BlobIndex) + sizeof(struct CodeDirectory) + strlen(base) + 1 + (size + 0x1000 - 1) / 0x1000 * 0x14 + 0xc + (xmld == NULL ? 0 : 0x10 + xmls) + 15) / 16 * 16);
+                //printf("codesign_allocate -i %s -a %s %s -o %s\n", path, arch, ssize, temp);
                 execlp(allocate, allocate, "-i", path, "-a", arch, ssize, "-o", temp, NULL);
                 _assert(false);
@@ -454 +471 @@
             uint8_t *blob = top + data;
             struct SuperBlob *super = reinterpret_cast<struct SuperBlob *>(blob);
-            super->magic = Swap(CSMAGIC_EMBEDDED_SIGNATURE);
-
-            uint32_t count = 2;
+            super->blob.magic = Swap(CSMAGIC_EMBEDDED_SIGNATURE);
+
+            uint32_t count = xmld == NULL ? 2 : 3;
             uint32_t offset = sizeof(struct SuperBlob) + count * sizeof(struct BlobIndex);
 
@@ -466 +483 @@
             offset += sizeof(struct CodeDirectory);
 
-            directory->magic = Swap(CSMAGIC_CODEDIRECTORY);
+            directory->blob.magic = Swap(CSMAGIC_CODEDIRECTORY);
             directory->version = Swap(0x00020001);
             directory->flags = Swap(0);
@@ -495 +512 @@
             directory->hashOffset = Swap(offset - begin);
             offset += sizeof(*hashes) * (special + pages);
-            directory->length = Swap(offset - begin);
+            directory->blob.length = Swap(offset - begin);
 
             super->index[1].type = Swap(CSSLOT_REQUIREMENTS);
@@ -503 +520 @@
             offset += 0xc;
 
+            if (xmld != NULL) {
+                super->index[2].type = Swap(CSSLOT_ENTITLEMENTS);
+                super->index[2].offset = Swap(offset);
+
+                uint32_t begin = offset;
+                struct Blob *entitlements = reinterpret_cast<struct Blob *>(blob + begin);
+                offset += sizeof(struct Blob);
+
+                memcpy(blob + offset, xmld, xmls);
+                offset += xmls;
+
+                entitlements->magic = Swap(CSMAGIC_ENTITLEMENTS);
+                entitlements->length = Swap(offset - begin);
+            }
+
             super->count = Swap(count);
-            super->length = Swap(offset);
-
-            _assert(offset <= size);
+            super->blob.length = Swap(offset);
+
+            if (offset > size) {
+                fprintf(stderr, "offset (%zu) > size (%zu)\n", offset, size);
+                _assert(false);
+            } //else fprintf(stderr, "offset (%zu) <= size (%zu)\n", offset, size);
+
             memset(blob + offset, 0, size - offset);
         }