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"
48 #include <sys/types.h>
56 #define FAT_MAGIC 0xcafebabe
57 #define FAT_CIGAM 0xbebafeca
77 #define MH_MAGIC 0xfeedface
78 #define MH_CIGAM 0xcefaedfe
80 #define MH_EXECUTE 0x2
89 #define LC_REQ_DYLD 0x80000000
91 #define LC_LOAD_DYLIB 0x0c
92 #define LC_ID_DYLIB 0x0d
94 #define LC_CODE_SIGNATURE 0x1d
95 #define LC_REEXPORT_DYLIB (0x1f | LC_REQ_DYLD)
100 uint32_t current_version
;
101 uint32_t compatibility_version
;
104 struct dylib_command
{
110 struct uuid_command
{
116 struct linkedit_data_command
{
123 uint16_t Swap_(uint16_t value
) {
125 ((value
>> 8) & 0x00ff) |
126 ((value
<< 8) & 0xff00);
129 uint32_t Swap_(uint32_t value
) {
130 value
= ((value
>> 8) & 0x00ff00ff) |
131 ((value
<< 8) & 0xff00ff00);
132 value
= ((value
>> 16) & 0x0000ffff) |
133 ((value
<< 16) & 0xffff0000);
137 int16_t Swap_(int16_t value
) {
138 return Swap_(static_cast<uint16_t>(value
));
141 int32_t Swap_(int32_t value
) {
142 return Swap_(static_cast<uint32_t>(value
));
145 uint16_t Swap(uint16_t value
) {
146 return true ? Swap_(value
) : value
;
149 uint32_t Swap(uint32_t value
) {
150 return true ? Swap_(value
) : value
;
153 int16_t Swap(int16_t value
) {
154 return Swap(static_cast<uint16_t>(value
));
157 int32_t Swap(int32_t value
) {
158 return Swap(static_cast<uint32_t>(value
));
165 mach_header
*mach_header_
;
169 uint16_t Swap(uint16_t value
) const {
170 return swapped_
? Swap_(value
) : value
;
173 uint32_t Swap(uint32_t value
) const {
174 return swapped_
? Swap_(value
) : value
;
177 int16_t Swap(int16_t value
) const {
178 return Swap(static_cast<uint16_t>(value
));
181 int32_t Swap(int32_t value
) const {
182 return Swap(static_cast<uint32_t>(value
));
185 Framework(const char *framework_path
) :
188 base_
= map(framework_path
, 0, _not(size_t), &size_
, false);
189 fat_header
*fat_header
= reinterpret_cast<struct fat_header
*>(base_
);
191 if (Swap(fat_header
->magic
) == FAT_CIGAM
) {
192 swapped_
= !swapped_
;
194 } else if (Swap(fat_header
->magic
) != FAT_MAGIC
)
195 mach_header_
= (mach_header
*) base_
;
197 size_t fat_narch
= Swap(fat_header
->nfat_arch
);
198 fat_arch
*fat_arch
= reinterpret_cast<struct fat_arch
*>(fat_header
+ 1);
200 for (arch
= 0; arch
!= fat_narch
; ++arch
) {
201 uint32_t arch_offset
= Swap(fat_arch
->offset
);
202 mach_header_
= (mach_header
*) ((uint8_t *) base_
+ arch_offset
);
211 if (Swap(mach_header_
->magic
) == MH_CIGAM
)
212 swapped_
= !swapped_
;
213 else _assert(Swap(mach_header_
->magic
) == MH_MAGIC
);
216 Swap(mach_header_
->filetype
) == MH_EXECUTE
||
217 Swap(mach_header_
->filetype
) == MH_DYLIB
||
218 Swap(mach_header_
->filetype
) == MH_BUNDLE
222 struct mach_header
*operator ->() const {
234 std::vector
<struct load_command
*> GetLoadCommands() {
235 std::vector
<struct load_command
*> load_commands
;
237 struct load_command
*load_command
= reinterpret_cast<struct load_command
*>(mach_header_
+ 1);
238 for (uint32_t cmd
= 0; cmd
!= Swap(mach_header_
->ncmds
); ++cmd
) {
239 load_commands
.push_back(load_command
);
240 load_command
= (struct load_command
*) ((uint8_t *) load_command
+ Swap(load_command
->cmdsize
));
243 return load_commands
;
247 #define CSMAGIC_CODEDIRECTORY 0xfade0c02
248 #define CSMAGIC_EMBEDDED_SIGNATURE 0xfade0cc0
249 #define CSSLOT_CODEDIRECTORY 0
250 #define CSSLOT_REQUIREMENTS 2
261 struct BlobIndex index
[];
264 struct CodeDirectory
{
270 uint32_t identOffset
;
271 uint32_t nSpecialSlots
;
281 extern "C" uint32_t hash(uint8_t *k
, uint32_t length
, uint32_t initval
);
283 #define CODESIGN_ALLOCATE "arm-apple-darwin9-codesign_allocate"
285 void sha1(uint8_t *hash
, uint8_t *data
, size_t size
) {
288 SHA1Input(&context
, data
, size
);
289 SHA1Result(&context
, hash
);
292 int main(int argc
, const char *argv
[]) {
304 std::vector
<std::string
> files
;
307 for (int argi(1); argi
!= argc
; ++argi
)
308 if (argv
[argi
][0] != '-')
309 files
.push_back(argv
[argi
]);
310 else switch (argv
[argi
][1]) {
311 case 'R': flag_R
= true; break;
312 case 't': flag_t
= true; break;
313 case 'u': flag_u
= true; break;
314 case 'p': flag_p
= true; break;
315 case 'S': flag_S
= true; break;
319 if (argv
[argi
][2] == '-')
323 timev
= strtoul(argv
[argi
] + 2, &arge
, 0);
324 _assert(arge
== argv
[argi
] + strlen(argv
[argi
]));
333 if (files
.empty()) usage
: {
337 size_t filei(0), filee(0);
338 _foreach (file
, files
) try {
339 const char *path(file
->c_str());
340 const char *base
= strrchr(path
, '/');
341 char *temp(NULL
), *dir
;
345 dir
= strndup(path
, base
++ - path
+ 1);
352 asprintf(&temp
, "%s.%s.cs", dir
, base
);
353 const char *allocate
= getenv("CODESIGN_ALLOCATE");
354 if (allocate
== NULL
)
355 allocate
= "codesign_allocate";
357 size_t size
= _not(size_t);
359 Framework
framework(path
);
360 _foreach (load_command
, framework
.GetLoadCommands()) {
361 uint32_t cmd(framework
.Swap((*load_command
)->cmd
));
362 if (cmd
== LC_CODE_SIGNATURE
) {
363 struct linkedit_data_command
*signature
= reinterpret_cast<struct linkedit_data_command
*>(*load_command
);
364 size
= framework
.Swap(signature
->dataoff
);
365 _assert(size
< framework
.GetSize());
370 if (size
== _not(size_t))
371 size
= framework
.GetSize();
373 switch (framework
->cputype
) {
374 case 12: switch (framework
->cpusubtype
) {
375 case 0: arch
= "arm"; break;
376 case 6: arch
= "armv6"; break;
377 default: arch
= NULL
; break;
380 default: arch
= NULL
; break;
384 _assert(arch
!= NULL
);
390 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);
391 execlp(allocate
, allocate
, "-i", path
, "-a", arch
, ssize
, "-o", temp
, NULL
);
396 _syscall(waitpid(pid
, &status
, 0));
397 _assert(WIFEXITED(status
));
398 _assert(WEXITSTATUS(status
) == 0);
401 Framework
framework(temp
== NULL
? path
: temp
);
402 struct linkedit_data_command
*signature(NULL
);
405 printf("path%zu='%s'\n", filei
, file
->c_str());
407 _foreach (load_command
, framework
.GetLoadCommands()) {
408 uint32_t cmd(framework
.Swap((*load_command
)->cmd
));
410 if (flag_R
&& cmd
== LC_REEXPORT_DYLIB
)
411 (*load_command
)->cmd
= framework
.Swap(LC_LOAD_DYLIB
);
412 else if (cmd
== LC_CODE_SIGNATURE
)
413 signature
= reinterpret_cast<struct linkedit_data_command
*>(*load_command
);
414 else if (cmd
== LC_UUID
) {
415 volatile struct uuid_command
*uuid_command(reinterpret_cast<struct uuid_command
*>(*load_command
));
418 printf("uuid%zu=%.2x%.2x%.2x%.2x-%.2x%.2x-%.2x%.2x-%.2x%.2x-%.2x%.2x%.2x%.2x%.2x%.2x\n", filei
,
419 uuid_command
->uuid
[ 0], uuid_command
->uuid
[ 1], uuid_command
->uuid
[ 2], uuid_command
->uuid
[ 3],
420 uuid_command
->uuid
[ 4], uuid_command
->uuid
[ 5], uuid_command
->uuid
[ 6], uuid_command
->uuid
[ 7],
421 uuid_command
->uuid
[ 8], uuid_command
->uuid
[ 9], uuid_command
->uuid
[10], uuid_command
->uuid
[11],
422 uuid_command
->uuid
[12], uuid_command
->uuid
[13], uuid_command
->uuid
[14], uuid_command
->uuid
[15]
425 } else if (cmd
== LC_ID_DYLIB
) {
426 volatile struct dylib_command
*dylib_command(reinterpret_cast<struct dylib_command
*>(*load_command
));
429 printf("time%zu=0x%.8x\n", filei
, framework
.Swap(dylib_command
->dylib
.timestamp
));
437 dylib_command
->dylib
.timestamp
= 0;
438 timed
= hash(reinterpret_cast<uint8_t *>(framework
.GetBase()), framework
.GetSize(), timev
);
441 dylib_command
->dylib
.timestamp
= framework
.Swap(timed
);
447 _assert(signature
!= NULL
);
449 uint32_t data
= framework
.Swap(signature
->dataoff
);
450 uint32_t size
= framework
.Swap(signature
->datasize
);
452 uint8_t *top
= reinterpret_cast<uint8_t *>(framework
.GetBase());
453 uint8_t *blob
= top
+ data
;
454 struct SuperBlob
*super
= reinterpret_cast<struct SuperBlob
*>(blob
);
455 super
->magic
= Swap(CSMAGIC_EMBEDDED_SIGNATURE
);
458 uint32_t offset
= sizeof(struct SuperBlob
) + count
* sizeof(struct BlobIndex
);
460 super
->index
[0].type
= Swap(CSSLOT_CODEDIRECTORY
);
461 super
->index
[0].offset
= Swap(offset
);
463 uint32_t begin
= offset
;
464 struct CodeDirectory
*directory
= reinterpret_cast<struct CodeDirectory
*>(blob
+ begin
);
465 offset
+= sizeof(struct CodeDirectory
);
467 directory
->magic
= Swap(CSMAGIC_CODEDIRECTORY
);
468 directory
->version
= Swap(0x00020001);
469 directory
->flags
= Swap(0);
470 directory
->codeLimit
= Swap(data
);
471 directory
->hashSize
= 0x14;
472 directory
->hashType
= 0x01;
473 directory
->spare1
= 0x00;
474 directory
->pageSize
= 0x0c;
475 directory
->spare2
= Swap(0);
477 directory
->identOffset
= Swap(offset
- begin
);
478 strcpy(reinterpret_cast<char *>(blob
+ offset
), base
);
479 offset
+= strlen(base
) + 1;
481 uint8_t (*hashes
)[20] = reinterpret_cast<uint8_t (*)[20]>(blob
+ offset
);
482 uint32_t special
= 0;
484 uint32_t pages
= (data
+ 0x1000 - 1) / 0x1000;
485 directory
->nSpecialSlots
= Swap(special
);
486 directory
->nCodeSlots
= Swap(pages
);
489 for (size_t i
= 0; i
!= pages
- 1; ++i
)
490 sha1(hashes
[special
+ i
], top
+ 0x1000 * i
, 0x1000);
492 sha1(hashes
[special
+ pages
- 1], top
+ 0x1000 * (pages
- 1), data
% 0x1000);
494 directory
->hashOffset
= Swap(offset
- begin
);
495 offset
+= sizeof(*hashes
) * (special
+ pages
);
496 directory
->length
= Swap(offset
- begin
);
498 super
->index
[1].type
= Swap(CSSLOT_REQUIREMENTS
);
499 super
->index
[1].offset
= Swap(offset
);
501 memcpy(blob
+ offset
, "\xfa\xde\x0c\x01\x00\x00\x00\x0c\x00\x00\x00\x00", 0xc);
504 super
->count
= Swap(count
);
505 super
->length
= Swap(offset
);
507 _assert(offset
<= size
);
508 memset(blob
+ offset
, 0, size
- offset
);
513 _syscall(stat(path
, &info
));
514 _syscall(chown(temp
, info
.st_uid
, info
.st_gid
));
515 _syscall(chmod(temp
, info
.st_mode
));
516 _syscall(unlink(path
));
517 _syscall(rename(temp
, path
));
523 } catch (const char *) {