Changeset 310 for trunk/util/ldid.cpp

Timestamp:

06/14/08 09:29:39 (5 months ago)

Author:

saurik

Message:

Added requirement slots and armv6 support to ldid/codesign.

Files:

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

trunk/util/ldid.cpp

@@ -218 +218 @@
     }
 
+    struct mach_header *operator ->() const {
+        return mach_header_;
+    }
+
     void *GetBase() {
         return base_;
@@ -242 +246 @@
 #define CSMAGIC_EMBEDDED_SIGNATURE 0xfade0cc0
 #define CSSLOT_CODEDIRECTORY 0
+#define CSSLOT_REQUIREMENTS 2
 
 struct BlobIndex {
@@ -347 +352 @@
                 allocate = "codesign_allocate";
 
-            size_t size; {
+            size_t size = _not(size_t);
+            const char *arch; {
                 Framework framework(path);
-                size = framework.GetSize();
+                _foreach (load_command, framework.GetLoadCommands()) {
+                    uint32_t cmd(framework.Swap((*load_command)->cmd));
+                    if (cmd == LC_CODE_SIGNATURE) {
+                        struct linkedit_data_command *signature = reinterpret_cast<struct linkedit_data_command *>(*load_command);
+                        size = framework.Swap(signature->dataoff);
+                        _assert(size < framework.GetSize());
+                        break;
+                    }
+                }
+
+                if (size == _not(size_t))
+                    size = framework.GetSize();
+
+                switch (framework->cputype) {
+                    case 12: switch (framework->cpusubtype) {
+                        case 0: arch = "arm"; break;
+                        case 6: arch = "armv6"; break;
+                        default: arch = NULL; break;
+                    } break;
+
+                    default: arch = NULL; break;
+                }
             }
+
+            _assert(arch != NULL);
 
             pid_t pid = fork();
@@ -356 +385 @@
             if (pid == 0) {
                 char *ssize;
-                asprintf(&ssize, "%u", (sizeof(struct SuperBlob) + sizeof(struct BlobIndex) + sizeof(struct CodeDirectory) + strlen(base) + 1 + (size + 0x1000 - 1) / 0x1000 * 0x14 + 15) / 16 * 16);
-                printf("%s\n", ssize);
-                execlp(allocate, allocate, "-i", path, "-a", "arm", ssize, "-o", temp, NULL);
+                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);
+                execlp(allocate, allocate, "-i", path, "-a", arch, ssize, "-o", temp, NULL);
                 _assert(false);
             }
@@ -424 +452 @@
             super->magic = Swap(CSMAGIC_EMBEDDED_SIGNATURE);
 
-            uint32_t count = 1;
+            uint32_t count = 2;
             uint32_t offset = sizeof(struct SuperBlob) + count * sizeof(struct BlobIndex);
 
@@ -465 +493 @@
             directory->length = Swap(offset - begin);
 
+            super->index[1].type = Swap(CSSLOT_REQUIREMENTS);
+            super->index[1].offset = Swap(offset);
+
+            memcpy(blob + offset, "\xfa\xde\x0c\x01\x00\x00\x00\x0c\x00\x00\x00\x00", 0xc);
+            offset += 0xc;
+
             super->count = Swap(count);
             super->length = Swap(offset);
 
-            _assert(offset < size);
+            _assert(offset <= size);
             memset(blob + offset, 0, size - offset);
         }