1 /* JocStrap - Java/Objective-C Bootstrap
2 * Copyright (C) 2007 Jay Freeman (saurik)
6 * Redistribution and use in source and binary
7 * forms, with or without modification, are permitted
8 * provided that the following conditions are met:
10 * 1. Redistributions of source code must retain the
11 * above copyright notice, this list of conditions
12 * and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the
14 * above copyright notice, this list of conditions
15 * and the following disclaimer in the documentation
16 * and/or other materials provided with the
18 * 3. The name of the author may not be used to endorse
19 * or promote products derived from this software
20 * without specific prior written permission.
22 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS''
23 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
24 * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
25 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE
27 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
28 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
29 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
30 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
31 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
32 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
33 * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
34 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
35 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
38 #include "minimal/stdlib.h"
39 #include "minimal/mapping.h"
54 #define FAT_MAGIC 0xcafebabe
55 #define FAT_CIGAM 0xbebafeca
75 #define MH_MAGIC 0xfeedface
76 #define MH_CIGAM 0xcefaedfe
78 #define MH_EXECUTE 0x2
87 #define LC_REQ_DYLD 0x80000000
89 #define LC_LOAD_DYLIB 0x0c
90 #define LC_ID_DYLIB 0x0d
92 #define LC_CODE_SIGNATURE 0x1d
93 #define LC_REEXPORT_DYLIB (0x1f | LC_REQ_DYLD)
98 uint32_t current_version
;
99 uint32_t compatibility_version
;
102 struct dylib_command
{
108 struct uuid_command
{
114 struct linkedit_data_command
{
121 uint16_t Swap_(uint16_t value
) {
123 ((value
>> 8) & 0x00ff) |
124 ((value
<< 8) & 0xff00);
127 uint32_t Swap_(uint32_t value
) {
128 value
= ((value
>> 8) & 0x00ff00ff) |
129 ((value
<< 8) & 0xff00ff00);
130 value
= ((value
>> 16) & 0x0000ffff) |
131 ((value
<< 16) & 0xffff0000);
135 int16_t Swap_(int16_t value
) {
136 return Swap_(static_cast<uint16_t>(value
));
139 int32_t Swap_(int32_t value
) {
140 return Swap_(static_cast<uint32_t>(value
));
143 uint16_t Swap(uint16_t value
) {
144 return true ? Swap_(value
) : value
;
147 uint32_t Swap(uint32_t value
) {
148 return true ? Swap_(value
) : value
;
151 int16_t Swap(int16_t value
) {
152 return Swap(static_cast<uint16_t>(value
));
155 int32_t Swap(int32_t value
) {
156 return Swap(static_cast<uint32_t>(value
));
163 mach_header
*mach_header_
;
167 uint16_t Swap(uint16_t value
) const {
168 return swapped_
? Swap_(value
) : value
;
171 uint32_t Swap(uint32_t value
) const {
172 return swapped_
? Swap_(value
) : value
;
175 int16_t Swap(int16_t value
) const {
176 return Swap(static_cast<uint16_t>(value
));
179 int32_t Swap(int32_t value
) const {
180 return Swap(static_cast<uint32_t>(value
));
183 Framework(const char *framework_path
) :
186 base_
= map(framework_path
, 0, _not(size_t), &size_
, false);
187 fat_header
*fat_header
= reinterpret_cast<struct fat_header
*>(base_
);
189 if (Swap(fat_header
->magic
) == FAT_CIGAM
) {
190 swapped_
= !swapped_
;
192 } else if (Swap(fat_header
->magic
) != FAT_MAGIC
)
193 mach_header_
= (mach_header
*) base_
;
195 size_t fat_narch
= Swap(fat_header
->nfat_arch
);
196 fat_arch
*fat_arch
= reinterpret_cast<struct fat_arch
*>(fat_header
+ 1);
198 for (arch
= 0; arch
!= fat_narch
; ++arch
) {
199 uint32_t arch_offset
= Swap(fat_arch
->offset
);
200 mach_header_
= (mach_header
*) ((uint8_t *) base_
+ arch_offset
);
209 if (Swap(mach_header_
->magic
) == MH_CIGAM
)
210 swapped_
= !swapped_
;
211 else _assert(Swap(mach_header_
->magic
) == MH_MAGIC
);
214 Swap(mach_header_
->filetype
) == MH_EXECUTE
||
215 Swap(mach_header_
->filetype
) == MH_DYLIB
||
216 Swap(mach_header_
->filetype
) == MH_BUNDLE
220 struct mach_header
*operator ->() const {
232 std::vector
<struct load_command
*> GetLoadCommands() {
233 std::vector
<struct load_command
*> load_commands
;
235 struct load_command
*load_command
= reinterpret_cast<struct load_command
*>(mach_header_
+ 1);
236 for (uint32_t cmd
= 0; cmd
!= Swap(mach_header_
->ncmds
); ++cmd
) {
237 load_commands
.push_back(load_command
);
238 load_command
= (struct load_command
*) ((uint8_t *) load_command
+ Swap(load_command
->cmdsize
));
241 return load_commands
;
245 #define CSMAGIC_CODEDIRECTORY 0xfade0c02
246 #define CSMAGIC_EMBEDDED_SIGNATURE 0xfade0cc0
247 #define CSSLOT_CODEDIRECTORY 0
248 #define CSSLOT_REQUIREMENTS 2
259 struct BlobIndex index
[];
262 struct CodeDirectory
{
268 uint32_t identOffset
;
269 uint32_t nSpecialSlots
;
279 extern "C" uint32_t hash(uint8_t *k
, uint32_t length
, uint32_t initval
);
281 #define CODESIGN_ALLOCATE "arm-apple-darwin9-codesign_allocate"
283 void sha1(uint8_t *hash
, uint8_t *data
, size_t size
) {
286 SHA1Input(&context
, data
, size
);
287 SHA1Result(&context
, hash
);
290 int main(int argc
, const char *argv
[]) {
302 std::vector
<std::string
> files
;
305 for (int argi(1); argi
!= argc
; ++argi
)
306 if (argv
[argi
][0] != '-')
307 files
.push_back(argv
[argi
]);
308 else switch (argv
[argi
][1]) {
309 case 'R': flag_R
= true; break;
310 case 't': flag_t
= true; break;
311 case 'u': flag_u
= true; break;
312 case 'p': flag_p
= true; break;
313 case 'S': flag_S
= true; break;
317 if (argv
[argi
][2] == '-')
321 timev
= strtoul(argv
[argi
] + 2, &arge
, 0);
322 _assert(arge
== argv
[argi
] + strlen(argv
[argi
]));
331 if (files
.empty()) usage
: {
335 size_t filei(0), filee(0);
336 _foreach (file
, files
) try {
337 const char *path(file
->c_str());
338 const char *base
= strrchr(path
, '/');
339 char *temp(NULL
), *dir
;
342 dir
= strndup(path
, base
++ - path
+ 1);
349 asprintf(&temp
, "%s.%s.cs", dir
, base
);
350 const char *allocate
= getenv("CODESIGN_ALLOCATE");
351 if (allocate
== NULL
)
352 allocate
= "codesign_allocate";
354 size_t size
= _not(size_t);
356 Framework
framework(path
);
357 _foreach (load_command
, framework
.GetLoadCommands()) {
358 uint32_t cmd(framework
.Swap((*load_command
)->cmd
));
359 if (cmd
== LC_CODE_SIGNATURE
) {
360 struct linkedit_data_command
*signature
= reinterpret_cast<struct linkedit_data_command
*>(*load_command
);
361 size
= framework
.Swap(signature
->dataoff
);
362 _assert(size
< framework
.GetSize());
367 if (size
== _not(size_t))
368 size
= framework
.GetSize();
370 switch (framework
->cputype
) {
371 case 12: switch (framework
->cpusubtype
) {
372 case 0: arch
= "arm"; break;
373 case 6: arch
= "armv6"; break;
374 default: arch
= NULL
; break;
377 default: arch
= NULL
; break;
381 _assert(arch
!= NULL
);
387 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);
388 execlp(allocate
, allocate
, "-i", path
, "-a", arch
, ssize
, "-o", temp
, NULL
);
393 _syscall(waitpid(pid
, &status
, 0));
394 _assert(WIFEXITED(status
));
395 _assert(WEXITSTATUS(status
) == 0);
398 Framework
framework(temp
== NULL
? path
: temp
);
399 struct linkedit_data_command
*signature(NULL
);
402 printf("path%zu='%s'\n", filei
, file
->c_str());
404 _foreach (load_command
, framework
.GetLoadCommands()) {
405 uint32_t cmd(framework
.Swap((*load_command
)->cmd
));
407 if (flag_R
&& cmd
== LC_REEXPORT_DYLIB
)
408 (*load_command
)->cmd
= framework
.Swap(LC_LOAD_DYLIB
);
409 else if (cmd
== LC_CODE_SIGNATURE
)
410 signature
= reinterpret_cast<struct linkedit_data_command
*>(*load_command
);
411 else if (cmd
== LC_UUID
) {
412 volatile struct uuid_command
*uuid_command(reinterpret_cast<struct uuid_command
*>(*load_command
));
415 printf("uuid%zu=%.2x%.2x%.2x%.2x-%.2x%.2x-%.2x%.2x-%.2x%.2x-%.2x%.2x%.2x%.2x%.2x%.2x\n", filei
,
416 uuid_command
->uuid
[ 0], uuid_command
->uuid
[ 1], uuid_command
->uuid
[ 2], uuid_command
->uuid
[ 3],
417 uuid_command
->uuid
[ 4], uuid_command
->uuid
[ 5], uuid_command
->uuid
[ 6], uuid_command
->uuid
[ 7],
418 uuid_command
->uuid
[ 8], uuid_command
->uuid
[ 9], uuid_command
->uuid
[10], uuid_command
->uuid
[11],
419 uuid_command
->uuid
[12], uuid_command
->uuid
[13], uuid_command
->uuid
[14], uuid_command
->uuid
[15]
422 } else if (cmd
== LC_ID_DYLIB
) {
423 volatile struct dylib_command
*dylib_command(reinterpret_cast<struct dylib_command
*>(*load_command
));
426 printf("time%zu=0x%.8x\n", filei
, framework
.Swap(dylib_command
->dylib
.timestamp
));
434 dylib_command
->dylib
.timestamp
= 0;
435 timed
= hash(reinterpret_cast<uint8_t *>(framework
.GetBase()), framework
.GetSize(), timev
);
438 dylib_command
->dylib
.timestamp
= framework
.Swap(timed
);
444 _assert(signature
!= NULL
);
446 uint32_t data
= framework
.Swap(signature
->dataoff
);
447 uint32_t size
= framework
.Swap(signature
->datasize
);
449 uint8_t *top
= reinterpret_cast<uint8_t *>(framework
.GetBase());
450 uint8_t *blob
= top
+ data
;
451 struct SuperBlob
*super
= reinterpret_cast<struct SuperBlob
*>(blob
);
452 super
->magic
= Swap(CSMAGIC_EMBEDDED_SIGNATURE
);
455 uint32_t offset
= sizeof(struct SuperBlob
) + count
* sizeof(struct BlobIndex
);
457 super
->index
[0].type
= Swap(CSSLOT_CODEDIRECTORY
);
458 super
->index
[0].offset
= Swap(offset
);
460 uint32_t begin
= offset
;
461 struct CodeDirectory
*directory
= reinterpret_cast<struct CodeDirectory
*>(blob
+ begin
);
462 offset
+= sizeof(struct CodeDirectory
);
464 directory
->magic
= Swap(CSMAGIC_CODEDIRECTORY
);
465 directory
->version
= Swap(0x00020001);
466 directory
->flags
= Swap(0);
467 directory
->codeLimit
= Swap(data
);
468 directory
->hashSize
= 0x14;
469 directory
->hashType
= 0x01;
470 directory
->spare1
= 0x00;
471 directory
->pageSize
= 0x0c;
472 directory
->spare2
= Swap(0);
474 directory
->identOffset
= Swap(offset
- begin
);
475 strcpy(reinterpret_cast<char *>(blob
+ offset
), base
);
476 offset
+= strlen(base
) + 1;
478 uint8_t (*hashes
)[20] = reinterpret_cast<uint8_t (*)[20]>(blob
+ offset
);
479 uint32_t special
= 0;
481 uint32_t pages
= (data
+ 0x1000 - 1) / 0x1000;
482 directory
->nSpecialSlots
= Swap(special
);
483 directory
->nCodeSlots
= Swap(pages
);
486 for (size_t i
= 0; i
!= pages
- 1; ++i
)
487 sha1(hashes
[special
+ i
], top
+ 0x1000 * i
, 0x1000);
489 sha1(hashes
[special
+ pages
- 1], top
+ 0x1000 * (pages
- 1), data
% 0x1000);
491 directory
->hashOffset
= Swap(offset
- begin
);
492 offset
+= sizeof(*hashes
) * (special
+ pages
);
493 directory
->length
= Swap(offset
- begin
);
495 super
->index
[1].type
= Swap(CSSLOT_REQUIREMENTS
);
496 super
->index
[1].offset
= Swap(offset
);
498 memcpy(blob
+ offset
, "\xfa\xde\x0c\x01\x00\x00\x00\x0c\x00\x00\x00\x00", 0xc);
501 super
->count
= Swap(count
);
502 super
->length
= Swap(offset
);
504 _assert(offset
<= size
);
505 memset(blob
+ offset
, 0, size
- offset
);
509 _syscall(unlink(path
));
510 _syscall(rename(temp
, path
));
516 } catch (const char *) {