ldid.cpp

   1 /* ldid - (Mach-O) Link-Loader Identity Editor
   2  * Copyright (C) 2007-2015  Jay Freeman (saurik)
   3 */
   4
   5 /* GNU Affero General Public License, Version 3 {{{ */
   6 /*
   7  * This program is free software: you can redistribute it and/or modify
   8  * it under the terms of the GNU Affero General Public License as published by
   9  * the Free Software Foundation, either version 3 of the License, or
  10  * (at your option) any later version.
  11
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU Affero General Public License for more details.
  16
  17  * You should have received a copy of the GNU Affero General Public License
  18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 **/
  20 /* }}} */
  21
  22 #include <cstdio>
  23 #include <cstdlib>
  24 #include <cstring>
  25 #include <fstream>
  26 #include <iostream>
  27 #include <memory>
  28 #include <set>
  29 #include <sstream>
  30 #include <string>
  31 #include <vector>
  32
  33 #include <dirent.h>
  34 #include <errno.h>
  35 #include <fcntl.h>
  36 #include <regex.h>
  37 #include <stdbool.h>
  38 #include <stdint.h>
  39 #include <unistd.h>
  40
  41 #include <sys/mman.h>
  42 #include <sys/stat.h>
  43 #include <sys/types.h>
  44
  45 #ifndef LDID_NOSMIME
  46 #include <openssl/err.h>
  47 #include <openssl/pem.h>
  48 #include <openssl/pkcs7.h>
  49 #include <openssl/pkcs12.h>
  50 #endif
  51
  52 #ifdef __APPLE__
  53 #include <CommonCrypto/CommonDigest.h>
  54 #define LDID_SHA1_DIGEST_LENGTH CC_SHA1_DIGEST_LENGTH
  55 #define LDID_SHA1 CC_SHA1
  56 #define LDID_SHA1_CTX CC_SHA1_CTX
  57 #define LDID_SHA1_Init CC_SHA1_Init
  58 #define LDID_SHA1_Update CC_SHA1_Update
  59 #define LDID_SHA1_Final CC_SHA1_Final
  60 #else
  61 #include <openssl/sha.h>
  62 #define LDID_SHA1_DIGEST_LENGTH SHA_DIGEST_LENGTH
  63 #define LDID_SHA1 SHA1
  64 #define LDID_SHA1_CTX SHA_CTX
  65 #define LDID_SHA1_Init SHA1_Init
  66 #define LDID_SHA1_Update SHA1_Update
  67 #define LDID_SHA1_Final SHA1_Final
  68 #endif
  69
  70 #ifndef LDID_NOPLIST
  71 #include <plist/plist.h>
  72 #endif
  73
  74 #include "ldid.hpp"
  75
  76 #define _assert___(line) \
  77     #line
  78 #define _assert__(line) \
  79     _assert___(line)
  80
  81 #ifdef __EXCEPTIONS
  82 #define _assert_(expr, format, ...) \
  83     do if (!(expr)) { \
  84         fprintf(stderr, "%s(%u): _assert(): " format "\n", __FILE__, __LINE__, ## __VA_ARGS__); \
  85         throw __FILE__ "(" _assert__(__LINE__) "): _assert(" #expr ")"; \
  86     } while (false)
  87 #else
  88 // XXX: this is not acceptable
  89 #define _assert_(expr, format, ...) \
  90     do if (!(expr)) { \
  91         fprintf(stderr, "%s(%u): _assert(): " format "\n", __FILE__, __LINE__, ## __VA_ARGS__); \
  92         exit(-1); \
  93     } while (false)
  94 #endif
  95
  96 #define _assert(expr) \
  97     _assert_(expr, "%s", #expr)
  98
  99 #define _syscall(expr, ...) [&] { for (;;) { \
 100     auto _value(expr); \
 101     if ((long) _value != -1) \
 102         return _value; \
 103     int error(errno); \
 104     if (error == EINTR) \
 105         continue; \
 106     /* XXX: EINTR is included in this list to fix g++ */ \
 107     for (auto success : (long[]) {EINTR, __VA_ARGS__}) \
 108         if (error == success) \
 109             return (decltype(expr)) -success; \
 110     _assert_(false, "errno=%u", error); \
 111 } }()
 112
 113 #define _trace() \
 114     fprintf(stderr, "_trace(%s:%u): %s\n", __FILE__, __LINE__, __FUNCTION__)
 115
 116 #define _not(type) \
 117     ((type) ~ (type) 0)
 118
 119 #define _packed \
 120     __attribute__((packed))
 121
 122 template <typename Type_>
 123 struct Iterator_ {
 124     typedef typename Type_::const_iterator Result;
 125 };
 126
 127 #define _foreach(item, list) \
 128     for (bool _stop(true); _stop; ) \
 129         for (const __typeof__(list) &_list = (list); _stop; _stop = false) \
 130             for (Iterator_<__typeof__(list)>::Result _item = _list.begin(); _item != _list.end(); ++_item) \
 131                 for (bool _suck(true); _suck; _suck = false) \
 132                     for (const __typeof__(*_item) &item = *_item; _suck; _suck = false)
 133
 134 class _Scope {
 135 };
 136
 137 template <typename Function_>
 138 class Scope :
 139     public _Scope
 140 {
 141   private:
 142     Function_ function_;
 143
 144   public:
 145     Scope(const Function_ &function) :
 146         function_(function)
 147     {
 148     }
 149
 150     ~Scope() {
 151         function_();
 152     }
 153 };
 154
 155 template <typename Function_>
 156 Scope<Function_> _scope(const Function_ &function) {
 157     return Scope<Function_>(function);
 158 }
 159
 160 #define _scope__(counter, function) \
 161     __attribute__((__unused__)) \
 162     const _Scope &_scope ## counter(_scope([&]function))
 163 #define _scope_(counter, function) \
 164     _scope__(counter, function)
 165 #define _scope(function) \
 166     _scope_(__COUNTER__, function)
 167
 168 #define CPU_ARCH_MASK  uint32_t(0xff000000)
 169 #define CPU_ARCH_ABI64 uint32_t(0x01000000)
 170
 171 #define CPU_TYPE_ANY     uint32_t(-1)
 172 #define CPU_TYPE_VAX     uint32_t( 1)
 173 #define CPU_TYPE_MC680x0 uint32_t( 6)
 174 #define CPU_TYPE_X86     uint32_t( 7)
 175 #define CPU_TYPE_MC98000 uint32_t(10)
 176 #define CPU_TYPE_HPPA    uint32_t(11)
 177 #define CPU_TYPE_ARM     uint32_t(12)
 178 #define CPU_TYPE_MC88000 uint32_t(13)
 179 #define CPU_TYPE_SPARC   uint32_t(14)
 180 #define CPU_TYPE_I860    uint32_t(15)
 181 #define CPU_TYPE_POWERPC uint32_t(18)
 182
 183 #define CPU_TYPE_I386 CPU_TYPE_X86
 184
 185 #define CPU_TYPE_ARM64     (CPU_ARCH_ABI64 | CPU_TYPE_ARM)
 186 #define CPU_TYPE_POWERPC64 (CPU_ARCH_ABI64 | CPU_TYPE_POWERPC)
 187 #define CPU_TYPE_X86_64    (CPU_ARCH_ABI64 | CPU_TYPE_X86)
 188
 189 struct fat_header {
 190     uint32_t magic;
 191     uint32_t nfat_arch;
 192 } _packed;
 193
 194 #define FAT_MAGIC 0xcafebabe
 195 #define FAT_CIGAM 0xbebafeca
 196
 197 struct fat_arch {
 198     uint32_t cputype;
 199     uint32_t cpusubtype;
 200     uint32_t offset;
 201     uint32_t size;
 202     uint32_t align;
 203 } _packed;
 204
 205 struct mach_header {
 206     uint32_t magic;
 207     uint32_t cputype;
 208     uint32_t cpusubtype;
 209     uint32_t filetype;
 210     uint32_t ncmds;
 211     uint32_t sizeofcmds;
 212     uint32_t flags;
 213 } _packed;
 214
 215 #define MH_MAGIC 0xfeedface
 216 #define MH_CIGAM 0xcefaedfe
 217
 218 #define MH_MAGIC_64 0xfeedfacf
 219 #define MH_CIGAM_64 0xcffaedfe
 220
 221 #define MH_DYLDLINK   0x4
 222
 223 #define MH_OBJECT     0x1
 224 #define MH_EXECUTE    0x2
 225 #define MH_DYLIB      0x6
 226 #define MH_BUNDLE     0x8
 227 #define MH_DYLIB_STUB 0x9
 228
 229 struct load_command {
 230     uint32_t cmd;
 231     uint32_t cmdsize;
 232 } _packed;
 233
 234 #define LC_REQ_DYLD           uint32_t(0x80000000)
 235
 236 #define LC_SEGMENT            uint32_t(0x01)
 237 #define LC_SYMTAB             uint32_t(0x02)
 238 #define LC_DYSYMTAB           uint32_t(0x0b)
 239 #define LC_LOAD_DYLIB         uint32_t(0x0c)
 240 #define LC_ID_DYLIB           uint32_t(0x0d)
 241 #define LC_SEGMENT_64         uint32_t(0x19)
 242 #define LC_UUID               uint32_t(0x1b)
 243 #define LC_CODE_SIGNATURE     uint32_t(0x1d)
 244 #define LC_SEGMENT_SPLIT_INFO uint32_t(0x1e)
 245 #define LC_REEXPORT_DYLIB     uint32_t(0x1f | LC_REQ_DYLD)
 246 #define LC_ENCRYPTION_INFO    uint32_t(0x21)
 247 #define LC_DYLD_INFO          uint32_t(0x22)
 248 #define LC_DYLD_INFO_ONLY     uint32_t(0x22 | LC_REQ_DYLD)
 249 #define LC_ENCRYPTION_INFO_64 uint32_t(0x2c)
 250
 251 union Version {
 252     struct {
 253         uint8_t patch;
 254         uint8_t minor;
 255         uint16_t major;
 256     } _packed;
 257
 258     uint32_t value;
 259 };
 260
 261 struct dylib {
 262     uint32_t name;
 263     uint32_t timestamp;
 264     uint32_t current_version;
 265     uint32_t compatibility_version;
 266 } _packed;
 267
 268 struct dylib_command {
 269     uint32_t cmd;
 270     uint32_t cmdsize;
 271     struct dylib dylib;
 272 } _packed;
 273
 274 struct uuid_command {
 275     uint32_t cmd;
 276     uint32_t cmdsize;
 277     uint8_t uuid[16];
 278 } _packed;
 279
 280 struct symtab_command {
 281     uint32_t cmd;
 282     uint32_t cmdsize;
 283     uint32_t symoff;
 284     uint32_t nsyms;
 285     uint32_t stroff;
 286     uint32_t strsize;
 287 } _packed;
 288
 289 struct dyld_info_command {
 290     uint32_t cmd;
 291     uint32_t cmdsize;
 292     uint32_t rebase_off;
 293     uint32_t rebase_size;
 294     uint32_t bind_off;
 295     uint32_t bind_size;
 296     uint32_t weak_bind_off;
 297     uint32_t weak_bind_size;
 298     uint32_t lazy_bind_off;
 299     uint32_t lazy_bind_size;
 300     uint32_t export_off;
 301     uint32_t export_size;
 302 } _packed;
 303
 304 struct dysymtab_command {
 305     uint32_t cmd;
 306     uint32_t cmdsize;
 307     uint32_t ilocalsym;
 308     uint32_t nlocalsym;
 309     uint32_t iextdefsym;
 310     uint32_t nextdefsym;
 311     uint32_t iundefsym;
 312     uint32_t nundefsym;
 313     uint32_t tocoff;
 314     uint32_t ntoc;
 315     uint32_t modtaboff;
 316     uint32_t nmodtab;
 317     uint32_t extrefsymoff;
 318     uint32_t nextrefsyms;
 319     uint32_t indirectsymoff;
 320     uint32_t nindirectsyms;
 321     uint32_t extreloff;
 322     uint32_t nextrel;
 323     uint32_t locreloff;
 324     uint32_t nlocrel;
 325 } _packed;
 326
 327 struct dylib_table_of_contents {
 328     uint32_t symbol_index;
 329     uint32_t module_index;
 330 } _packed;
 331
 332 struct dylib_module {
 333     uint32_t module_name;
 334     uint32_t iextdefsym;
 335     uint32_t nextdefsym;
 336     uint32_t irefsym;
 337     uint32_t nrefsym;
 338     uint32_t ilocalsym;
 339     uint32_t nlocalsym;
 340     uint32_t iextrel;
 341     uint32_t nextrel;
 342     uint32_t iinit_iterm;
 343     uint32_t ninit_nterm;
 344     uint32_t objc_module_info_addr;
 345     uint32_t objc_module_info_size;
 346 } _packed;
 347
 348 struct dylib_reference {
 349     uint32_t isym:24;
 350     uint32_t flags:8;
 351 } _packed;
 352
 353 struct relocation_info {
 354     int32_t r_address;
 355     uint32_t r_symbolnum:24;
 356     uint32_t r_pcrel:1;
 357     uint32_t r_length:2;
 358     uint32_t r_extern:1;
 359     uint32_t r_type:4;
 360 } _packed;
 361
 362 struct nlist {
 363     union {
 364         char *n_name;
 365         int32_t n_strx;
 366     } n_un;
 367
 368     uint8_t n_type;
 369     uint8_t n_sect;
 370     uint8_t n_desc;
 371     uint32_t n_value;
 372 } _packed;
 373
 374 struct segment_command {
 375     uint32_t cmd;
 376     uint32_t cmdsize;
 377     char segname[16];
 378     uint32_t vmaddr;
 379     uint32_t vmsize;
 380     uint32_t fileoff;
 381     uint32_t filesize;
 382     uint32_t maxprot;
 383     uint32_t initprot;
 384     uint32_t nsects;
 385     uint32_t flags;
 386 } _packed;
 387
 388 struct segment_command_64 {
 389     uint32_t cmd;
 390     uint32_t cmdsize;
 391     char segname[16];
 392     uint64_t vmaddr;
 393     uint64_t vmsize;
 394     uint64_t fileoff;
 395     uint64_t filesize;
 396     uint32_t maxprot;
 397     uint32_t initprot;
 398     uint32_t nsects;
 399     uint32_t flags;
 400 } _packed;
 401
 402 struct section {
 403     char sectname[16];
 404     char segname[16];
 405     uint32_t addr;
 406     uint32_t size;
 407     uint32_t offset;
 408     uint32_t align;
 409     uint32_t reloff;
 410     uint32_t nreloc;
 411     uint32_t flags;
 412     uint32_t reserved1;
 413     uint32_t reserved2;
 414 } _packed;
 415
 416 struct section_64 {
 417     char sectname[16];
 418     char segname[16];
 419     uint64_t addr;
 420     uint64_t size;
 421     uint32_t offset;
 422     uint32_t align;
 423     uint32_t reloff;
 424     uint32_t nreloc;
 425     uint32_t flags;
 426     uint32_t reserved1;
 427     uint32_t reserved2;
 428     uint32_t reserved3;
 429 } _packed;
 430
 431 struct linkedit_data_command {
 432     uint32_t cmd;
 433     uint32_t cmdsize;
 434     uint32_t dataoff;
 435     uint32_t datasize;
 436 } _packed;
 437
 438 struct encryption_info_command {
 439     uint32_t cmd;
 440     uint32_t cmdsize;
 441     uint32_t cryptoff;
 442     uint32_t cryptsize;
 443     uint32_t cryptid;
 444 } _packed;
 445
 446 #define BIND_OPCODE_MASK                             0xf0
 447 #define BIND_IMMEDIATE_MASK                          0x0f
 448 #define BIND_OPCODE_DONE                             0x00
 449 #define BIND_OPCODE_SET_DYLIB_ORDINAL_IMM            0x10
 450 #define BIND_OPCODE_SET_DYLIB_ORDINAL_ULEB           0x20
 451 #define BIND_OPCODE_SET_DYLIB_SPECIAL_IMM            0x30
 452 #define BIND_OPCODE_SET_SYMBOL_TRAILING_FLAGS_IMM    0x40
 453 #define BIND_OPCODE_SET_TYPE_IMM                     0x50
 454 #define BIND_OPCODE_SET_ADDEND_SLEB                  0x60
 455 #define BIND_OPCODE_SET_SEGMENT_AND_OFFSET_ULEB      0x70
 456 #define BIND_OPCODE_ADD_ADDR_ULEB                    0x80
 457 #define BIND_OPCODE_DO_BIND                          0x90
 458 #define BIND_OPCODE_DO_BIND_ADD_ADDR_ULEB            0xa0
 459 #define BIND_OPCODE_DO_BIND_ADD_ADDR_IMM_SCALED      0xb0
 460 #define BIND_OPCODE_DO_BIND_ULEB_TIMES_SKIPPING_ULEB 0xc0
 461
 462 inline void get(std::streambuf &stream, void *data, size_t size) {
 463     _assert(stream.sgetn(static_cast<char *>(data), size) == size);
 464 }
 465
 466 inline void put(std::streambuf &stream, const void *data, size_t size) {
 467     _assert(stream.sputn(static_cast<const char *>(data), size) == size);
 468 }
 469
 470 inline void pad(std::streambuf &stream, size_t size) {
 471     char padding[size];
 472     memset(padding, 0, size);
 473     put(stream, padding, size);
 474 }
 475
 476 template <typename Type_>
 477 Type_ Align(Type_ value, size_t align) {
 478     value += align - 1;
 479     value /= align;
 480     value *= align;
 481     return value;
 482 }
 483
 484 static const uint8_t PageShift_(0x0c);
 485 static const uint32_t PageSize_(1 << PageShift_);
 486
 487 static inline uint16_t Swap_(uint16_t value) {
 488     return
 489         ((value >>  8) & 0x00ff) |
 490         ((value <<  8) & 0xff00);
 491 }
 492
 493 static inline uint32_t Swap_(uint32_t value) {
 494     value = ((value >>  8) & 0x00ff00ff) |
 495             ((value <<  8) & 0xff00ff00);
 496     value = ((value >> 16) & 0x0000ffff) |
 497             ((value << 16) & 0xffff0000);
 498     return value;
 499 }
 500
 501 static inline uint64_t Swap_(uint64_t value) {
 502     value = (value & 0x00000000ffffffff) << 32 | (value & 0xffffffff00000000) >> 32;
 503     value = (value & 0x0000ffff0000ffff) << 16 | (value & 0xffff0000ffff0000) >> 16;
 504     value = (value & 0x00ff00ff00ff00ff) << 8  | (value & 0xff00ff00ff00ff00) >> 8;
 505     return value;
 506 }
 507
 508 static inline int16_t Swap_(int16_t value) {
 509     return Swap_(static_cast<uint16_t>(value));
 510 }
 511
 512 static inline int32_t Swap_(int32_t value) {
 513     return Swap_(static_cast<uint32_t>(value));
 514 }
 515
 516 static inline int64_t Swap_(int64_t value) {
 517     return Swap_(static_cast<uint64_t>(value));
 518 }
 519
 520 static bool little_(true);
 521
 522 static inline uint16_t Swap(uint16_t value) {
 523     return little_ ? Swap_(value) : value;
 524 }
 525
 526 static inline uint32_t Swap(uint32_t value) {
 527     return little_ ? Swap_(value) : value;
 528 }
 529
 530 static inline uint64_t Swap(uint64_t value) {
 531     return little_ ? Swap_(value) : value;
 532 }
 533
 534 static inline int16_t Swap(int16_t value) {
 535     return Swap(static_cast<uint16_t>(value));
 536 }
 537
 538 static inline int32_t Swap(int32_t value) {
 539     return Swap(static_cast<uint32_t>(value));
 540 }
 541
 542 static inline int64_t Swap(int64_t value) {
 543     return Swap(static_cast<uint64_t>(value));
 544 }
 545
 546 class Swapped {
 547   protected:
 548     bool swapped_;
 549
 550     Swapped() :
 551         swapped_(false)
 552     {
 553     }
 554
 555   public:
 556     Swapped(bool swapped) :
 557         swapped_(swapped)
 558     {
 559     }
 560
 561     template <typename Type_>
 562     Type_ Swap(Type_ value) const {
 563         return swapped_ ? Swap_(value) : value;
 564     }
 565 };
 566
 567 class Data :
 568     public Swapped
 569 {
 570   private:
 571     void *base_;
 572     size_t size_;
 573
 574   public:
 575     Data(void *base, size_t size) :
 576         base_(base),
 577         size_(size)
 578     {
 579     }
 580
 581     void *GetBase() const {
 582         return base_;
 583     }
 584
 585     size_t GetSize() const {
 586         return size_;
 587     }
 588 };
 589
 590 class MachHeader :
 591     public Data
 592 {
 593   private:
 594     bool bits64_;
 595
 596     struct mach_header *mach_header_;
 597     struct load_command *load_command_;
 598
 599   public:
 600     MachHeader(void *base, size_t size) :
 601         Data(base, size)
 602     {
 603         mach_header_ = (mach_header *) base;
 604
 605         switch (Swap(mach_header_->magic)) {
 606             case MH_CIGAM:
 607                 swapped_ = !swapped_;
 608             case MH_MAGIC:
 609                 bits64_ = false;
 610             break;
 611
 612             case MH_CIGAM_64:
 613                 swapped_ = !swapped_;
 614             case MH_MAGIC_64:
 615                 bits64_ = true;
 616             break;
 617
 618             default:
 619                 _assert(false);
 620         }
 621
 622         void *post = mach_header_ + 1;
 623         if (bits64_)
 624             post = (uint32_t *) post + 1;
 625         load_command_ = (struct load_command *) post;
 626
 627         _assert(
 628             Swap(mach_header_->filetype) == MH_EXECUTE ||
 629             Swap(mach_header_->filetype) == MH_DYLIB ||
 630             Swap(mach_header_->filetype) == MH_BUNDLE
 631         );
 632     }
 633
 634     bool Bits64() const {
 635         return bits64_;
 636     }
 637
 638     struct mach_header *operator ->() const {
 639         return mach_header_;
 640     }
 641
 642     operator struct mach_header *() const {
 643         return mach_header_;
 644     }
 645
 646     uint32_t GetCPUType() const {
 647         return Swap(mach_header_->cputype);
 648     }
 649
 650     uint32_t GetCPUSubtype() const {
 651         return Swap(mach_header_->cpusubtype) & 0xff;
 652     }
 653
 654     struct load_command *GetLoadCommand() const {
 655         return load_command_;
 656     }
 657
 658     std::vector<struct load_command *> GetLoadCommands() const {
 659         std::vector<struct load_command *> load_commands;
 660
 661         struct load_command *load_command = load_command_;
 662         for (uint32_t cmd = 0; cmd != Swap(mach_header_->ncmds); ++cmd) {
 663             load_commands.push_back(load_command);
 664             load_command = (struct load_command *) ((uint8_t *) load_command + Swap(load_command->cmdsize));
 665         }
 666
 667         return load_commands;
 668     }
 669
 670     void ForSection(const ldid::Functor<void (const char *, const char *, void *, size_t)> &code) const {
 671         _foreach (load_command, GetLoadCommands())
 672             switch (Swap(load_command->cmd)) {
 673                 case LC_SEGMENT: {
 674                     auto segment(reinterpret_cast<struct segment_command *>(load_command));
 675                     code(segment->segname, NULL, GetOffset<void>(segment->fileoff), segment->filesize);
 676                     auto section(reinterpret_cast<struct section *>(segment + 1));
 677                     for (uint32_t i(0), e(Swap(segment->nsects)); i != e; ++i, ++section)
 678                         code(segment->segname, section->sectname, GetOffset<void>(segment->fileoff + section->offset), section->size);
 679                 } break;
 680
 681                 case LC_SEGMENT_64: {
 682                     auto segment(reinterpret_cast<struct segment_command_64 *>(load_command));
 683                     code(segment->segname, NULL, GetOffset<void>(segment->fileoff), segment->filesize);
 684                     auto section(reinterpret_cast<struct section_64 *>(segment + 1));
 685                     for (uint32_t i(0), e(Swap(segment->nsects)); i != e; ++i, ++section)
 686                         code(segment->segname, section->sectname, GetOffset<void>(segment->fileoff + section->offset), section->size);
 687                 } break;
 688             }
 689     }
 690
 691     template <typename Target_>
 692     Target_ *GetOffset(uint32_t offset) const {
 693         return reinterpret_cast<Target_ *>(offset + (uint8_t *) mach_header_);
 694     }
 695 };
 696
 697 class FatMachHeader :
 698     public MachHeader
 699 {
 700   private:
 701     fat_arch *fat_arch_;
 702
 703   public:
 704     FatMachHeader(void *base, size_t size, fat_arch *fat_arch) :
 705         MachHeader(base, size),
 706         fat_arch_(fat_arch)
 707     {
 708     }
 709
 710     fat_arch *GetFatArch() const {
 711         return fat_arch_;
 712     }
 713 };
 714
 715 class FatHeader :
 716     public Data
 717 {
 718   private:
 719     fat_header *fat_header_;
 720     std::vector<FatMachHeader> mach_headers_;
 721
 722   public:
 723     FatHeader(void *base, size_t size) :
 724         Data(base, size)
 725     {
 726         fat_header_ = reinterpret_cast<struct fat_header *>(base);
 727
 728         if (Swap(fat_header_->magic) == FAT_CIGAM) {
 729             swapped_ = !swapped_;
 730             goto fat;
 731         } else if (Swap(fat_header_->magic) != FAT_MAGIC) {
 732             fat_header_ = NULL;
 733             mach_headers_.push_back(FatMachHeader(base, size, NULL));
 734         } else fat: {
 735             size_t fat_narch = Swap(fat_header_->nfat_arch);
 736             fat_arch *fat_arch = reinterpret_cast<struct fat_arch *>(fat_header_ + 1);
 737             size_t arch;
 738             for (arch = 0; arch != fat_narch; ++arch) {
 739                 uint32_t arch_offset = Swap(fat_arch->offset);
 740                 uint32_t arch_size = Swap(fat_arch->size);
 741                 mach_headers_.push_back(FatMachHeader((uint8_t *) base + arch_offset, arch_size, fat_arch));
 742                 ++fat_arch;
 743             }
 744         }
 745     }
 746
 747     std::vector<FatMachHeader> &GetMachHeaders() {
 748         return mach_headers_;
 749     }
 750
 751     bool IsFat() const {
 752         return fat_header_ != NULL;
 753     }
 754
 755     struct fat_header *operator ->() const {
 756         return fat_header_;
 757     }
 758
 759     operator struct fat_header *() const {
 760         return fat_header_;
 761     }
 762 };
 763
 764 #define CSMAGIC_REQUIREMENT            uint32_t(0xfade0c00)
 765 #define CSMAGIC_REQUIREMENTS           uint32_t(0xfade0c01)
 766 #define CSMAGIC_CODEDIRECTORY          uint32_t(0xfade0c02)
 767 #define CSMAGIC_EMBEDDED_SIGNATURE     uint32_t(0xfade0cc0)
 768 #define CSMAGIC_EMBEDDED_SIGNATURE_OLD uint32_t(0xfade0b02)
 769 #define CSMAGIC_EMBEDDED_ENTITLEMENTS  uint32_t(0xfade7171)
 770 #define CSMAGIC_DETACHED_SIGNATURE     uint32_t(0xfade0cc1)
 771 #define CSMAGIC_BLOBWRAPPER            uint32_t(0xfade0b01)
 772
 773 #define CSSLOT_CODEDIRECTORY uint32_t(0x00000)
 774 #define CSSLOT_INFOSLOT      uint32_t(0x00001)
 775 #define CSSLOT_REQUIREMENTS  uint32_t(0x00002)
 776 #define CSSLOT_RESOURCEDIR   uint32_t(0x00003)
 777 #define CSSLOT_APPLICATION   uint32_t(0x00004)
 778 #define CSSLOT_ENTITLEMENTS  uint32_t(0x00005)
 779
 780 #define CSSLOT_SIGNATURESLOT uint32_t(0x10000)
 781
 782 #define CS_HASHTYPE_SHA1 1
 783
 784 struct BlobIndex {
 785     uint32_t type;
 786     uint32_t offset;
 787 } _packed;
 788
 789 struct Blob {
 790     uint32_t magic;
 791     uint32_t length;
 792 } _packed;
 793
 794 struct SuperBlob {
 795     struct Blob blob;
 796     uint32_t count;
 797     struct BlobIndex index[];
 798 } _packed;
 799
 800 struct CodeDirectory {
 801     uint32_t version;
 802     uint32_t flags;
 803     uint32_t hashOffset;
 804     uint32_t identOffset;
 805     uint32_t nSpecialSlots;
 806     uint32_t nCodeSlots;
 807     uint32_t codeLimit;
 808     uint8_t hashSize;
 809     uint8_t hashType;
 810     uint8_t spare1;
 811     uint8_t pageSize;
 812     uint32_t spare2;
 813 } _packed;
 814
 815 #ifndef LDID_NOFLAGT
 816 extern "C" uint32_t hash(uint8_t *k, uint32_t length, uint32_t initval);
 817 #endif
 818
 819 static void sha1(uint8_t *hash, const void *data, size_t size) {
 820     LDID_SHA1(static_cast<const uint8_t *>(data), size, hash);
 821 }
 822
 823 static void sha1(std::vector<char> &hash, const void *data, size_t size) {
 824     hash.resize(LDID_SHA1_DIGEST_LENGTH);
 825     sha1(reinterpret_cast<uint8_t *>(hash.data()), data, size);
 826 }
 827
 828 struct CodesignAllocation {
 829     FatMachHeader mach_header_;
 830     uint32_t offset_;
 831     uint32_t size_;
 832     uint32_t limit_;
 833     uint32_t alloc_;
 834     uint32_t align_;
 835
 836     CodesignAllocation(FatMachHeader mach_header, size_t offset, size_t size, size_t limit, size_t alloc, size_t align) :
 837         mach_header_(mach_header),
 838         offset_(offset),
 839         size_(size),
 840         limit_(limit),
 841         alloc_(alloc),
 842         align_(align)
 843     {
 844     }
 845 };
 846
 847 #ifndef LDID_NOTOOLS
 848 class File {
 849   private:
 850     int file_;
 851
 852   public:
 853     File() :
 854         file_(-1)
 855     {
 856     }
 857
 858     ~File() {
 859         if (file_ != -1)
 860             _syscall(close(file_));
 861     }
 862
 863     void open(const char *path, int flags) {
 864         _assert(file_ == -1);
 865         file_ = _syscall(::open(path, flags));
 866     }
 867
 868     int file() const {
 869         return file_;
 870     }
 871 };
 872
 873 class Map {
 874   private:
 875     File file_;
 876     void *data_;
 877     size_t size_;
 878
 879     void clear() {
 880         if (data_ == NULL)
 881             return;
 882         _syscall(munmap(data_, size_));
 883         data_ = NULL;
 884         size_ = 0;
 885     }
 886
 887   public:
 888     Map() :
 889         data_(NULL),
 890         size_(0)
 891     {
 892     }
 893
 894     Map(const std::string &path, int oflag, int pflag, int mflag) :
 895         Map()
 896     {
 897         open(path, oflag, pflag, mflag);
 898     }
 899
 900     Map(const std::string &path, bool edit) :
 901         Map()
 902     {
 903         open(path, edit);
 904     }
 905
 906     ~Map() {
 907         clear();
 908     }
 909
 910     bool empty() const {
 911         return data_ == NULL;
 912     }
 913
 914     void open(const std::string &path, int oflag, int pflag, int mflag) {
 915         clear();
 916
 917         file_.open(path.c_str(), oflag);
 918         int file(file_.file());
 919
 920         struct stat stat;
 921         _syscall(fstat(file, &stat));
 922         size_ = stat.st_size;
 923
 924         data_ = _syscall(mmap(NULL, size_, pflag, mflag, file, 0));
 925     }
 926
 927     void open(const std::string &path, bool edit) {
 928         if (edit)
 929             open(path, O_RDWR, PROT_READ | PROT_WRITE, MAP_SHARED);
 930         else
 931             open(path, O_RDONLY, PROT_READ, MAP_PRIVATE);
 932     }
 933
 934     void *data() const {
 935         return data_;
 936     }
 937
 938     size_t size() const {
 939         return size_;
 940     }
 941
 942     operator std::string() const {
 943         return std::string(static_cast<char *>(data_), size_);
 944     }
 945 };
 946 #endif
 947
 948 namespace ldid {
 949
 950 static void Allocate(const void *idata, size_t isize, std::streambuf &output, const Functor<size_t (const MachHeader &, size_t)> &allocate, const Functor<size_t (const MachHeader &, std::streambuf &output, size_t, const std::string &, const char *)> &save) {
 951     FatHeader source(const_cast<void *>(idata), isize);
 952
 953     size_t offset(0);
 954     if (source.IsFat())
 955         offset += sizeof(fat_header) + sizeof(fat_arch) * source.Swap(source->nfat_arch);
 956
 957     std::vector<CodesignAllocation> allocations;
 958     _foreach (mach_header, source.GetMachHeaders()) {
 959         struct linkedit_data_command *signature(NULL);
 960         struct symtab_command *symtab(NULL);
 961
 962         _foreach (load_command, mach_header.GetLoadCommands()) {
 963             uint32_t cmd(mach_header.Swap(load_command->cmd));
 964             if (false);
 965             else if (cmd == LC_CODE_SIGNATURE)
 966                 signature = reinterpret_cast<struct linkedit_data_command *>(load_command);
 967             else if (cmd == LC_SYMTAB)
 968                 symtab = reinterpret_cast<struct symtab_command *>(load_command);
 969         }
 970
 971         size_t size;
 972         if (signature == NULL)
 973             size = mach_header.GetSize();
 974         else {
 975             size = mach_header.Swap(signature->dataoff);
 976             _assert(size <= mach_header.GetSize());
 977         }
 978
 979         if (symtab != NULL) {
 980             auto end(mach_header.Swap(symtab->stroff) + mach_header.Swap(symtab->strsize));
 981             _assert(end <= size);
 982             _assert(end >= size - 0x10);
 983             size = end;
 984         }
 985
 986         size_t alloc(allocate(mach_header, size));
 987
 988         auto *fat_arch(mach_header.GetFatArch());
 989         uint32_t align;
 990
 991         if (fat_arch != NULL)
 992             align = source.Swap(fat_arch->align);
 993         else switch (mach_header.GetCPUType()) {
 994             case CPU_TYPE_POWERPC:
 995             case CPU_TYPE_POWERPC64:
 996             case CPU_TYPE_X86:
 997             case CPU_TYPE_X86_64:
 998                 align = 0xc;
 999                 break;
1000             case CPU_TYPE_ARM:
1001             case CPU_TYPE_ARM64:
1002                 align = 0xe;
1003                 break;
1004             default:
1005                 align = 0x0;
1006                 break;
1007         }
1008
1009         offset = Align(offset, 1 << align);
1010
1011         uint32_t limit(size);
1012         if (alloc != 0)
1013             limit = Align(limit, 0x10);
1014
1015         allocations.push_back(CodesignAllocation(mach_header, offset, size, limit, alloc, align));
1016         offset += size + alloc;
1017         offset = Align(offset, 0x10);
1018     }
1019
1020     size_t position(0);
1021
1022     if (source.IsFat()) {
1023         fat_header fat_header;
1024         fat_header.magic = Swap(FAT_MAGIC);
1025         fat_header.nfat_arch = Swap(uint32_t(allocations.size()));
1026         put(output, &fat_header, sizeof(fat_header));
1027         position += sizeof(fat_header);
1028
1029         _foreach (allocation, allocations) {
1030             auto &mach_header(allocation.mach_header_);
1031
1032             fat_arch fat_arch;
1033             fat_arch.cputype = Swap(mach_header->cputype);
1034             fat_arch.cpusubtype = Swap(mach_header->cpusubtype);
1035             fat_arch.offset = Swap(allocation.offset_);
1036             fat_arch.size = Swap(allocation.limit_ + allocation.alloc_);
1037             fat_arch.align = Swap(allocation.align_);
1038             put(output, &fat_arch, sizeof(fat_arch));
1039             position += sizeof(fat_arch);
1040         }
1041     }
1042
1043     _foreach (allocation, allocations) {
1044         auto &mach_header(allocation.mach_header_);
1045
1046         pad(output, allocation.offset_ - position);
1047         position = allocation.offset_;
1048
1049         std::vector<std::string> commands;
1050
1051         _foreach (load_command, mach_header.GetLoadCommands()) {
1052             std::string copy(reinterpret_cast<const char *>(load_command), load_command->cmdsize);
1053
1054             switch (mach_header.Swap(load_command->cmd)) {
1055                 case LC_CODE_SIGNATURE:
1056                     continue;
1057                 break;
1058
1059                 case LC_SEGMENT: {
1060                     auto segment_command(reinterpret_cast<struct segment_command *>(&copy[0]));
1061                     if (strncmp(segment_command->segname, "__LINKEDIT", 16) != 0)
1062                         break;
1063                     size_t size(mach_header.Swap(allocation.limit_ + allocation.alloc_ - mach_header.Swap(segment_command->fileoff)));
1064                     segment_command->filesize = size;
1065                     segment_command->vmsize = Align(size, 1 << allocation.align_);
1066                 } break;
1067
1068                 case LC_SEGMENT_64: {
1069                     auto segment_command(reinterpret_cast<struct segment_command_64 *>(&copy[0]));
1070                     if (strncmp(segment_command->segname, "__LINKEDIT", 16) != 0)
1071                         break;
1072                     size_t size(mach_header.Swap(allocation.limit_ + allocation.alloc_ - mach_header.Swap(segment_command->fileoff)));
1073                     segment_command->filesize = size;
1074                     segment_command->vmsize = Align(size, 1 << allocation.align_);
1075                 } break;
1076             }
1077
1078             commands.push_back(copy);
1079         }
1080
1081         if (allocation.alloc_ != 0) {
1082             linkedit_data_command signature;
1083             signature.cmd = mach_header.Swap(LC_CODE_SIGNATURE);
1084             signature.cmdsize = mach_header.Swap(uint32_t(sizeof(signature)));
1085             signature.dataoff = mach_header.Swap(allocation.limit_);
1086             signature.datasize = mach_header.Swap(allocation.alloc_);
1087             commands.push_back(std::string(reinterpret_cast<const char *>(&signature), sizeof(signature)));
1088         }
1089
1090         size_t begin(position);
1091
1092         uint32_t after(0);
1093         _foreach(command, commands)
1094             after += command.size();
1095
1096         std::stringbuf altern;
1097
1098         struct mach_header header(*mach_header);
1099         header.ncmds = mach_header.Swap(uint32_t(commands.size()));
1100         header.sizeofcmds = mach_header.Swap(after);
1101         put(output, &header, sizeof(header));
1102         put(altern, &header, sizeof(header));
1103         position += sizeof(header);
1104
1105         if (mach_header.Bits64()) {
1106             auto pad(mach_header.Swap(uint32_t(0)));
1107             put(output, &pad, sizeof(pad));
1108             put(altern, &pad, sizeof(pad));
1109             position += sizeof(pad);
1110         }
1111
1112         _foreach(command, commands) {
1113             put(output, command.data(), command.size());
1114             put(altern, command.data(), command.size());
1115             position += command.size();
1116         }
1117
1118         uint32_t before(mach_header.Swap(mach_header->sizeofcmds));
1119         if (before > after) {
1120             pad(output, before - after);
1121             pad(altern, before - after);
1122             position += before - after;
1123         }
1124
1125         auto top(reinterpret_cast<char *>(mach_header.GetBase()));
1126
1127         std::string overlap(altern.str());
1128         overlap.append(top + overlap.size(), Align(overlap.size(), 0x1000) - overlap.size());
1129
1130         put(output, top + (position - begin), allocation.size_ - (position - begin));
1131         position = begin + allocation.size_;
1132
1133         pad(output, allocation.limit_ - allocation.size_);
1134         position += allocation.limit_ - allocation.size_;
1135
1136         size_t saved(save(mach_header, output, allocation.limit_, overlap, top));
1137         if (allocation.alloc_ > saved)
1138             pad(output, allocation.alloc_ - saved);
1139         position += allocation.alloc_;
1140     }
1141 }
1142
1143 }
1144
1145 typedef std::map<uint32_t, std::string> Blobs;
1146
1147 static void insert(Blobs &blobs, uint32_t slot, const std::stringbuf &buffer) {
1148     auto value(buffer.str());
1149     std::swap(blobs[slot], value);
1150 }
1151
1152 static void insert(Blobs &blobs, uint32_t slot, uint32_t magic, const std::stringbuf &buffer) {
1153     auto value(buffer.str());
1154     Blob blob;
1155     blob.magic = Swap(magic);
1156     blob.length = Swap(uint32_t(sizeof(blob) + value.size()));
1157     value.insert(0, reinterpret_cast<char *>(&blob), sizeof(blob));
1158     std::swap(blobs[slot], value);
1159 }
1160
1161 static size_t put(std::streambuf &output, uint32_t magic, const Blobs &blobs) {
1162     size_t total(0);
1163     _foreach (blob, blobs)
1164         total += blob.second.size();
1165
1166     struct SuperBlob super;
1167     super.blob.magic = Swap(magic);
1168     super.blob.length = Swap(uint32_t(sizeof(SuperBlob) + blobs.size() * sizeof(BlobIndex) + total));
1169     super.count = Swap(uint32_t(blobs.size()));
1170     put(output, &super, sizeof(super));
1171
1172     size_t offset(sizeof(SuperBlob) + sizeof(BlobIndex) * blobs.size());
1173
1174     _foreach (blob, blobs) {
1175         BlobIndex index;
1176         index.type = Swap(blob.first);
1177         index.offset = Swap(uint32_t(offset));
1178         put(output, &index, sizeof(index));
1179         offset += blob.second.size();
1180     }
1181
1182     _foreach (blob, blobs)
1183         put(output, blob.second.data(), blob.second.size());
1184
1185     return offset;
1186 }
1187
1188 #ifndef LDID_NOSMIME
1189 class Buffer {
1190   private:
1191     BIO *bio_;
1192
1193   public:
1194     Buffer(BIO *bio) :
1195         bio_(bio)
1196     {
1197         _assert(bio_ != NULL);
1198     }
1199
1200     Buffer() :
1201         bio_(BIO_new(BIO_s_mem()))
1202     {
1203     }
1204
1205     Buffer(const char *data, size_t size) :
1206         Buffer(BIO_new_mem_buf(const_cast<char *>(data), size))
1207     {
1208     }
1209
1210     Buffer(const std::string &data) :
1211         Buffer(data.data(), data.size())
1212     {
1213     }
1214
1215     Buffer(PKCS7 *pkcs) :
1216         Buffer()
1217     {
1218         _assert(i2d_PKCS7_bio(bio_, pkcs) != 0);
1219     }
1220
1221     ~Buffer() {
1222         BIO_free_all(bio_);
1223     }
1224
1225     operator BIO *() const {
1226         return bio_;
1227     }
1228
1229     explicit operator std::string() const {
1230         char *data;
1231         auto size(BIO_get_mem_data(bio_, &data));
1232         return std::string(data, size);
1233     }
1234 };
1235
1236 class Stuff {
1237   private:
1238     PKCS12 *value_;
1239     EVP_PKEY *key_;
1240     X509 *cert_;
1241     STACK_OF(X509) *ca_;
1242
1243   public:
1244     Stuff(BIO *bio) :
1245         value_(d2i_PKCS12_bio(bio, NULL)),
1246         ca_(NULL)
1247     {
1248         _assert(value_ != NULL);
1249         _assert(PKCS12_parse(value_, "", &key_, &cert_, &ca_) != 0);
1250         _assert(key_ != NULL);
1251         _assert(cert_ != NULL);
1252     }
1253
1254     Stuff(const std::string &data) :
1255         Stuff(Buffer(data))
1256     {
1257     }
1258
1259     ~Stuff() {
1260         sk_X509_pop_free(ca_, X509_free);
1261         X509_free(cert_);
1262         EVP_PKEY_free(key_);
1263         PKCS12_free(value_);
1264     }
1265
1266     operator PKCS12 *() const {
1267         return value_;
1268     }
1269
1270     operator EVP_PKEY *() const {
1271         return key_;
1272     }
1273
1274     operator X509 *() const {
1275         return cert_;
1276     }
1277
1278     operator STACK_OF(X509) *() const {
1279         return ca_;
1280     }
1281 };
1282
1283 class Signature {
1284   private:
1285     PKCS7 *value_;
1286
1287   public:
1288     Signature(const Stuff &stuff, const Buffer &data) :
1289         value_(PKCS7_sign(stuff, stuff, stuff, data, PKCS7_BINARY | PKCS7_DETACHED))
1290     {
1291         _assert(value_ != NULL);
1292     }
1293
1294     ~Signature() {
1295         PKCS7_free(value_);
1296     }
1297
1298     operator PKCS7 *() const {
1299         return value_;
1300     }
1301 };
1302 #endif
1303
1304 class NullBuffer :
1305     public std::streambuf
1306 {
1307   public:
1308     virtual std::streamsize xsputn(const char_type *data, std::streamsize size) {
1309         return size;
1310     }
1311
1312     virtual int_type overflow(int_type next) {
1313         return next;
1314     }
1315 };
1316
1317 class HashBuffer :
1318     public std::streambuf
1319 {
1320   private:
1321     std::vector<char> &hash_;
1322     LDID_SHA1_CTX context_;
1323
1324   public:
1325     HashBuffer(std::vector<char> &hash) :
1326         hash_(hash)
1327     {
1328         LDID_SHA1_Init(&context_);
1329     }
1330
1331     ~HashBuffer() {
1332         hash_.resize(LDID_SHA1_DIGEST_LENGTH);
1333         LDID_SHA1_Final(reinterpret_cast<uint8_t *>(hash_.data()), &context_);
1334     }
1335
1336     virtual std::streamsize xsputn(const char_type *data, std::streamsize size) {
1337         LDID_SHA1_Update(&context_, data, size);
1338         return size;
1339     }
1340
1341     virtual int_type overflow(int_type next) {
1342         if (next == traits_type::eof())
1343             return sync();
1344         char value(next);
1345         xsputn(&value, 1);
1346         return next;
1347     }
1348 };
1349
1350 class HashProxy :
1351     public HashBuffer
1352 {
1353   private:
1354     std::streambuf &buffer_;
1355
1356   public:
1357     HashProxy(std::vector<char> &hash, std::streambuf &buffer) :
1358         HashBuffer(hash),
1359         buffer_(buffer)
1360     {
1361     }
1362
1363     virtual std::streamsize xsputn(const char_type *data, std::streamsize size) {
1364         _assert(HashBuffer::xsputn(data, size) == size);
1365         return buffer_.sputn(data, size);
1366     }
1367 };
1368
1369 #ifndef LDID_NOTOOLS
1370 static bool Starts(const std::string &lhs, const std::string &rhs) {
1371     return lhs.size() >= rhs.size() && lhs.compare(0, rhs.size(), rhs) == 0;
1372 }
1373
1374 class Split {
1375   public:
1376     std::string dir;
1377     std::string base;
1378
1379     Split(const std::string &path) {
1380         size_t slash(path.rfind('/'));
1381         if (slash == std::string::npos)
1382             base = path;
1383         else {
1384             dir = path.substr(0, slash + 1);
1385             base = path.substr(slash + 1);
1386         }
1387     }
1388 };
1389
1390 static void mkdir_p(const std::string &path) {
1391     if (path.empty())
1392         return;
1393 #ifdef __WIN32__
1394     if (_syscall(mkdir(path.c_str()), EEXIST) == -EEXIST)
1395         return;
1396 #else
1397     if (_syscall(mkdir(path.c_str(), 0755), EEXIST) == -EEXIST)
1398         return;
1399 #endif
1400     auto slash(path.rfind('/', path.size() - 1));
1401     if (slash == std::string::npos)
1402         return;
1403     mkdir_p(path.substr(0, slash));
1404 }
1405
1406 static std::string Temporary(std::filebuf &file, const Split &split) {
1407     std::string temp(split.dir + ".ldid." + split.base);
1408     mkdir_p(split.dir);
1409     _assert_(file.open(temp.c_str(), std::ios::out | std::ios::trunc | std::ios::binary) == &file, "open(): %s", temp.c_str());
1410     return temp;
1411 }
1412
1413 static void Commit(const std::string &path, const std::string &temp) {
1414     struct stat info;
1415     if (_syscall(stat(path.c_str(), &info), ENOENT) == 0) {
1416 #ifndef __WIN32__
1417         _syscall(chown(temp.c_str(), info.st_uid, info.st_gid));
1418 #endif
1419         _syscall(chmod(temp.c_str(), info.st_mode));
1420     }
1421
1422     _syscall(rename(temp.c_str(), path.c_str()));
1423 }
1424 #endif
1425
1426 namespace ldid {
1427
1428 void Sign(const void *idata, size_t isize, std::streambuf &output, const std::string &identifier, const std::string &entitlements, const std::string &key, const Slots &slots) {
1429     Allocate(idata, isize, output, fun([&](const MachHeader &mach_header, size_t size) -> size_t {
1430         size_t alloc(sizeof(struct SuperBlob));
1431
1432         uint32_t special(0);
1433
1434         special = std::max(special, CSSLOT_REQUIREMENTS);
1435         alloc += sizeof(struct BlobIndex);
1436         alloc += 0xc;
1437
1438         if (!entitlements.empty()) {
1439             special = std::max(special, CSSLOT_ENTITLEMENTS);
1440             alloc += sizeof(struct BlobIndex);
1441             alloc += sizeof(struct Blob);
1442             alloc += entitlements.size();
1443         }
1444
1445         special = std::max(special, CSSLOT_CODEDIRECTORY);
1446         alloc += sizeof(struct BlobIndex);
1447         alloc += sizeof(struct Blob);
1448         alloc += sizeof(struct CodeDirectory);
1449         alloc += identifier.size() + 1;
1450
1451         if (!key.empty()) {
1452             alloc += sizeof(struct BlobIndex);
1453             alloc += sizeof(struct Blob);
1454             // XXX: this is just a "sufficiently large number"
1455             alloc += 0x3000;
1456         }
1457
1458         _foreach (slot, slots)
1459             special = std::max(special, slot.first);
1460
1461         mach_header.ForSection(fun([&](const char *segment, const char *section, void *data, size_t size) {
1462             if (strcmp(segment, "__TEXT") == 0 && section != NULL && strcmp(section, "__info_plist") == 0)
1463                 special = std::max(special, CSSLOT_INFOSLOT);
1464         }));
1465
1466         uint32_t normal((size + PageSize_ - 1) / PageSize_);
1467         alloc = Align(alloc + (special + normal) * LDID_SHA1_DIGEST_LENGTH, 16);
1468         return alloc;
1469     }), fun([&](const MachHeader &mach_header, std::streambuf &output, size_t limit, const std::string &overlap, const char *top) -> size_t {
1470         Blobs blobs;
1471
1472         if (true) {
1473             std::stringbuf data;
1474
1475             Blobs requirements;
1476             put(data, CSMAGIC_REQUIREMENTS, requirements);
1477
1478             insert(blobs, CSSLOT_REQUIREMENTS, data);
1479         }
1480
1481         if (!entitlements.empty()) {
1482             std::stringbuf data;
1483             put(data, entitlements.data(), entitlements.size());
1484             insert(blobs, CSSLOT_ENTITLEMENTS, CSMAGIC_EMBEDDED_ENTITLEMENTS, data);
1485         }
1486
1487         if (true) {
1488             std::stringbuf data;
1489
1490             Slots posts(slots);
1491
1492             mach_header.ForSection(fun([&](const char *segment, const char *section, void *data, size_t size) {
1493                 if (strcmp(segment, "__TEXT") == 0 && section != NULL && strcmp(section, "__info_plist") == 0)
1494                     sha1(posts[CSSLOT_INFOSLOT], data, size);
1495             }));
1496
1497             uint32_t special(0);
1498             _foreach (blob, blobs)
1499                 special = std::max(special, blob.first);
1500             _foreach (slot, posts)
1501                 special = std::max(special, slot.first);
1502             uint32_t normal((limit + PageSize_ - 1) / PageSize_);
1503
1504             CodeDirectory directory;
1505             directory.version = Swap(uint32_t(0x00020001));
1506             directory.flags = Swap(uint32_t(0));
1507             directory.hashOffset = Swap(uint32_t(sizeof(Blob) + sizeof(CodeDirectory) + identifier.size() + 1 + LDID_SHA1_DIGEST_LENGTH * special));
1508             directory.identOffset = Swap(uint32_t(sizeof(Blob) + sizeof(CodeDirectory)));
1509             directory.nSpecialSlots = Swap(special);
1510             directory.codeLimit = Swap(uint32_t(limit));
1511             directory.nCodeSlots = Swap(normal);
1512             directory.hashSize = LDID_SHA1_DIGEST_LENGTH;
1513             directory.hashType = CS_HASHTYPE_SHA1;
1514             directory.spare1 = 0x00;
1515             directory.pageSize = PageShift_;
1516             directory.spare2 = Swap(uint32_t(0));
1517             put(data, &directory, sizeof(directory));
1518
1519             put(data, identifier.c_str(), identifier.size() + 1);
1520
1521             uint8_t storage[special + normal][LDID_SHA1_DIGEST_LENGTH];
1522             uint8_t (*hashes)[LDID_SHA1_DIGEST_LENGTH] = storage + special;
1523
1524             memset(storage, 0, sizeof(*storage) * special);
1525
1526             _foreach (blob, blobs) {
1527                 auto local(reinterpret_cast<const Blob *>(&blob.second[0]));
1528                 sha1((uint8_t *) (hashes - blob.first), local, Swap(local->length));
1529             }
1530
1531             _foreach (slot, posts) {
1532                 _assert(sizeof(*hashes) == slot.second.size());
1533                 memcpy(hashes - slot.first, slot.second.data(), slot.second.size());
1534             }
1535
1536             if (normal != 1)
1537                 for (size_t i = 0; i != normal - 1; ++i)
1538                     sha1(hashes[i], (PageSize_ * i < overlap.size() ? overlap.data() : top) + PageSize_ * i, PageSize_);
1539             if (normal != 0)
1540                 sha1(hashes[normal - 1], top + PageSize_ * (normal - 1), ((limit - 1) % PageSize_) + 1);
1541
1542             put(data, storage, sizeof(storage));
1543
1544             insert(blobs, CSSLOT_CODEDIRECTORY, CSMAGIC_CODEDIRECTORY, data);
1545         }
1546
1547 #ifndef LDID_NOSMIME
1548         if (!key.empty()) {
1549             std::stringbuf data;
1550             const std::string &sign(blobs[CSSLOT_CODEDIRECTORY]);
1551
1552             Stuff stuff(key);
1553             Buffer bio(sign);
1554
1555             Signature signature(stuff, sign);
1556             Buffer result(signature);
1557             std::string value(result);
1558             put(data, value.data(), value.size());
1559
1560             insert(blobs, CSSLOT_SIGNATURESLOT, CSMAGIC_BLOBWRAPPER, data);
1561         }
1562 #endif
1563
1564         return put(output, CSMAGIC_EMBEDDED_SIGNATURE, blobs);
1565     }));
1566 }
1567
1568 #ifndef LDID_NOTOOLS
1569 static void Unsign(void *idata, size_t isize, std::streambuf &output) {
1570     Allocate(idata, isize, output, fun([](const MachHeader &mach_header, size_t size) -> size_t {
1571         return 0;
1572     }), fun([](const MachHeader &mach_header, std::streambuf &output, size_t limit, const std::string &overlap, const char *top) -> size_t {
1573         return 0;
1574     }));
1575 }
1576
1577 std::string DiskFolder::Path(const std::string &path) {
1578     return path_ + "/" + path;
1579 }
1580
1581 DiskFolder::DiskFolder(const std::string &path) :
1582     path_(path)
1583 {
1584 }
1585
1586 DiskFolder::~DiskFolder() {
1587     if (!std::uncaught_exception())
1588         for (const auto &commit : commit_)
1589             Commit(commit.first, commit.second);
1590 }
1591
1592 void DiskFolder::Find(const std::string &root, const std::string &base, const Functor<void (const std::string &, const Functor<void (const Functor<void (std::streambuf &, std::streambuf &)> &)> &)>&code) {
1593     std::string path(Path(root) + base);
1594
1595     DIR *dir(opendir(path.c_str()));
1596     _assert(dir != NULL);
1597     _scope({ _syscall(closedir(dir)); });
1598
1599     while (auto child = readdir(dir)) {
1600         std::string name(child->d_name);
1601         if (name == "." || name == "..")
1602             continue;
1603         if (Starts(name, ".ldid."))
1604             continue;
1605
1606         bool directory;
1607
1608 #ifdef __WIN32__
1609         struct stat info;
1610         _syscall(stat(path.c_str(), &info));
1611         if (false);
1612         else if (S_ISDIR(info.st_mode))
1613             directory = true;
1614         else if (S_ISREG(info.st_mode))
1615             directory = false;
1616         else
1617             _assert_(false, "st_mode=%x", info.st_mode);
1618 #else
1619         switch (child->d_type) {
1620             case DT_DIR:
1621                 directory = true;
1622                 break;
1623             case DT_REG:
1624                 directory = false;
1625                 break;
1626             default:
1627                 _assert_(false, "d_type=%u", child->d_type);
1628         }
1629 #endif
1630
1631         if (directory)
1632             Find(root, base + name + "/", code);
1633         else
1634             code(base + name, fun([&](const Functor<void (std::streambuf &, std::streambuf &)> &code) {
1635                 std::string access(root + base + name);
1636                 _assert_(Open(access, fun([&](std::streambuf &data) {
1637                     NullBuffer save;
1638                     code(data, save);
1639                 })), "open(): %s", access.c_str());
1640             }));
1641     }
1642 }
1643
1644 void DiskFolder::Save(const std::string &path, const Functor<void (std::streambuf &)> &code) {
1645     std::filebuf save;
1646     auto from(Path(path));
1647     commit_[from] = Temporary(save, from);
1648     code(save);
1649 }
1650
1651 bool DiskFolder::Open(const std::string &path, const Functor<void (std::streambuf &)> &code) {
1652     std::filebuf data;
1653     auto result(data.open(Path(path).c_str(), std::ios::binary | std::ios::in));
1654     if (result == NULL)
1655         return false;
1656     _assert(result == &data);
1657     code(data);
1658     return true;
1659 }
1660
1661 void DiskFolder::Find(const std::string &path, const Functor<void (const std::string &, const Functor<void (const Functor<void (std::streambuf &, std::streambuf &)> &)> &)>&code) {
1662     Find(path, "", code);
1663 }
1664 #endif
1665
1666 SubFolder::SubFolder(Folder &parent, const std::string &path) :
1667     parent_(parent),
1668     path_(path)
1669 {
1670 }
1671
1672 void SubFolder::Save(const std::string &path, const Functor<void (std::streambuf &)> &code) {
1673     return parent_.Save(path_ + path, code);
1674 }
1675
1676 bool SubFolder::Open(const std::string &path, const Functor<void (std::streambuf &)> &code) {
1677     return parent_.Open(path_ + path, code);
1678 }
1679
1680 void SubFolder::Find(const std::string &path, const Functor<void (const std::string &, const Functor<void (const Functor<void (std::streambuf &, std::streambuf &)> &)> &)> &code) {
1681     return parent_.Find(path_ + path, code);
1682 }
1683
1684 UnionFolder::UnionFolder(Folder &parent) :
1685     parent_(parent)
1686 {
1687 }
1688
1689 void UnionFolder::Save(const std::string &path, const Functor<void (std::streambuf &)> &code) {
1690     return parent_.Save(path, code);
1691 }
1692
1693 bool UnionFolder::Open(const std::string &path, const Functor<void (std::streambuf &)> &code) {
1694     auto file(files_.find(path));
1695     if (file == files_.end())
1696         return parent_.Open(path, code);
1697
1698     auto &data(file->second);
1699     data.pubseekpos(0, std::ios::in);
1700     code(data);
1701     return true;
1702 }
1703
1704 void UnionFolder::Find(const std::string &path, const Functor<void (const std::string &, const Functor<void (const Functor<void (std::streambuf &, std::streambuf &)> &)> &)> &code) {
1705     parent_.Find(path, fun([&](const std::string &name, const Functor<void (const Functor<void (std::streambuf &, std::streambuf &)> &)> &save) {
1706         if (files_.find(path + name) == files_.end())
1707             code(name, save);
1708     }));
1709
1710     for (auto &file : files_)
1711         if (file.first.size() >= path.size() && file.first.substr(0, path.size()) == path)
1712             code(file.first.substr(path.size()), fun([&](const Functor<void (std::streambuf &, std::streambuf &)> &code) {
1713                 parent_.Save(file.first, fun([&](std::streambuf &save) {
1714                     file.second.pubseekpos(0, std::ios::in);
1715                     code(file.second, save);
1716                 }));
1717             }));
1718 }
1719
1720 #ifndef LDID_NOTOOLS
1721 static size_t copy(std::streambuf &source, std::streambuf &target) {
1722     size_t total(0);
1723     for (;;) {
1724         char data[4096];
1725         size_t writ(source.sgetn(data, sizeof(data)));
1726         if (writ == 0)
1727             break;
1728         _assert(target.sputn(data, writ) == writ);
1729         total += writ;
1730     }
1731     return total;
1732 }
1733
1734 #ifndef LDID_NOPLIST
1735 static plist_t plist(const std::string &data) {
1736     plist_t plist(NULL);
1737     if (Starts(data, "bplist00"))
1738         plist_from_bin(data.data(), data.size(), &plist);
1739     else
1740         plist_from_xml(data.data(), data.size(), &plist);
1741     _assert(plist != NULL);
1742     return plist;
1743 }
1744
1745 static void plist_d(std::streambuf &buffer, const Functor<void (plist_t)> &code) {
1746     std::stringbuf data;
1747     copy(buffer, data);
1748     auto node(plist(data.str()));
1749     _scope({ plist_free(node); });
1750     _assert(plist_get_node_type(node) == PLIST_DICT);
1751     code(node);
1752 }
1753
1754 static std::string plist_s(plist_t node) {
1755     _assert(node != NULL);
1756     _assert(plist_get_node_type(node) == PLIST_STRING);
1757     char *data;
1758     plist_get_string_val(node, &data);
1759     _scope({ free(data); });
1760     return data;
1761 }
1762 #endif
1763
1764 enum Mode {
1765     NoMode,
1766     OptionalMode,
1767     OmitMode,
1768     NestedMode,
1769     TopMode,
1770 };
1771
1772 class Expression {
1773   private:
1774     regex_t regex_;
1775
1776   public:
1777     Expression(const std::string &code) {
1778         _assert_(regcomp(&regex_, code.c_str(), REG_EXTENDED | REG_NOSUB) == 0, "regcomp()");
1779     }
1780
1781     ~Expression() {
1782         regfree(&regex_);
1783     }
1784
1785     bool operator ()(const std::string &data) const {
1786         auto value(regexec(&regex_, data.c_str(), 0, NULL, 0));
1787         if (value == REG_NOMATCH)
1788             return false;
1789         _assert_(value == 0, "regexec()");
1790         return true;
1791     }
1792 };
1793
1794 struct Rule {
1795     unsigned weight_;
1796     Mode mode_;
1797     std::string code_;
1798
1799     mutable std::auto_ptr<Expression> regex_;
1800
1801     Rule(unsigned weight, Mode mode, const std::string &code) :
1802         weight_(weight),
1803         mode_(mode),
1804         code_(code)
1805     {
1806     }
1807
1808     Rule(const Rule &rhs) :
1809         weight_(rhs.weight_),
1810         mode_(rhs.mode_),
1811         code_(rhs.code_)
1812     {
1813     }
1814
1815     void Compile() const {
1816         regex_.reset(new Expression(code_));
1817     }
1818
1819     bool operator ()(const std::string &data) const {
1820         _assert(regex_.get() != NULL);
1821         return (*regex_)(data);
1822     }
1823
1824     bool operator <(const Rule &rhs) const {
1825         if (weight_ > rhs.weight_)
1826             return true;
1827         if (weight_ < rhs.weight_)
1828             return false;
1829         return mode_ > rhs.mode_;
1830     }
1831 };
1832
1833 struct RuleCode {
1834     bool operator ()(const Rule *lhs, const Rule *rhs) const {
1835         return lhs->code_ < rhs->code_;
1836     }
1837 };
1838
1839 #ifndef LDID_NOPLIST
1840 std::string Bundle(const std::string &root, Folder &folder, const std::string &key, std::map<std::string, std::vector<char>> &remote, const std::string &entitlements) {
1841     std::string executable;
1842     std::string identifier;
1843
1844     static const std::string info("Info.plist");
1845
1846     _assert_(folder.Open(info, fun([&](std::streambuf &buffer) {
1847         plist_d(buffer, fun([&](plist_t node) {
1848             executable = plist_s(plist_dict_get_item(node, "CFBundleExecutable"));
1849             identifier = plist_s(plist_dict_get_item(node, "CFBundleIdentifier"));
1850         }));
1851     })), "open(): Info.plist");
1852
1853     static const std::string directory("_CodeSignature/");
1854     static const std::string signature(directory + "CodeResources");
1855
1856     std::map<std::string, std::multiset<Rule>> versions;
1857
1858     auto &rules1(versions[""]);
1859     auto &rules2(versions["2"]);
1860
1861     folder.Open(signature, fun([&](std::streambuf &buffer) {
1862         plist_d(buffer, fun([&](plist_t node) {
1863             // XXX: maybe attempt to preserve existing rules
1864         }));
1865     }));
1866
1867     if (true) {
1868         rules1.insert(Rule{1, NoMode, "^"});
1869         rules1.insert(Rule{10000, OmitMode, "^(Frameworks/[^/]+\\.framework/|PlugIns/[^/]+\\.appex/|PlugIns/[^/]+\\.appex/Frameworks/[^/]+\\.framework/|())SC_Info/[^/]+\\.(sinf|supf|supp)$"});
1870         rules1.insert(Rule{1000, OptionalMode, "^.*\\.lproj/"});
1871         rules1.insert(Rule{1100, OmitMode, "^.*\\.lproj/locversion.plist$"});
1872         rules1.insert(Rule{10000, OmitMode, "^Watch/[^/]+\\.app/(Frameworks/[^/]+\\.framework/|PlugIns/[^/]+\\.appex/|PlugIns/[^/]+\\.appex/Frameworks/[^/]+\\.framework/)SC_Info/[^/]+\\.(sinf|supf|supp)$"});
1873         rules1.insert(Rule{1, NoMode, "^version.plist$"});
1874     }
1875
1876     if (true) {
1877         rules2.insert(Rule{11, NoMode, ".*\\.dSYM($|/)"});
1878         rules2.insert(Rule{20, NoMode, "^"});
1879         rules2.insert(Rule{2000, OmitMode, "^(.*/)?\\.DS_Store$"});
1880         rules2.insert(Rule{10000, OmitMode, "^(Frameworks/[^/]+\\.framework/|PlugIns/[^/]+\\.appex/|PlugIns/[^/]+\\.appex/Frameworks/[^/]+\\.framework/|())SC_Info/[^/]+\\.(sinf|supf|supp)$"});
1881         rules2.insert(Rule{10, NestedMode, "^(Frameworks|SharedFrameworks|PlugIns|Plug-ins|XPCServices|Helpers|MacOS|Library/(Automator|Spotlight|LoginItems))/"});
1882         rules2.insert(Rule{1, NoMode, "^.*"});
1883         rules2.insert(Rule{1000, OptionalMode, "^.*\\.lproj/"});
1884         rules2.insert(Rule{1100, OmitMode, "^.*\\.lproj/locversion.plist$"});
1885         rules2.insert(Rule{20, OmitMode, "^Info\\.plist$"});
1886         rules2.insert(Rule{20, OmitMode, "^PkgInfo$"});
1887         rules2.insert(Rule{10000, OmitMode, "^Watch/[^/]+\\.app/(Frameworks/[^/]+\\.framework/|PlugIns/[^/]+\\.appex/|PlugIns/[^/]+\\.appex/Frameworks/[^/]+\\.framework/)SC_Info/[^/]+\\.(sinf|supf|supp)$"});
1888         rules2.insert(Rule{10, NestedMode, "^[^/]+$"});
1889         rules2.insert(Rule{20, NoMode, "^embedded\\.provisionprofile$"});
1890         rules2.insert(Rule{20, NoMode, "^version\\.plist$"});
1891     }
1892
1893     std::map<std::string, std::vector<char>> local;
1894
1895     static Expression nested("^PlugIns/[^/]*\\.appex/Info\\.plist$");
1896
1897     folder.Find("", fun([&](const std::string &name, const Functor<void (const Functor<void (std::streambuf &, std::streambuf &)> &)> &code) {
1898         if (!nested(name))
1899             return;
1900         auto bundle(root + Split(name).dir);
1901         SubFolder subfolder(folder, bundle);
1902         Bundle(bundle, subfolder, key, local, "");
1903     }));
1904
1905     folder.Find("", fun([&](const std::string &name, const Functor<void (const Functor<void (std::streambuf &, std::streambuf &)> &)> &code) {
1906         // BundleDiskRep::adjustResources -> builder.addExclusion
1907         if (name == executable || Starts(name, directory) || Starts(name, "_MASReceipt/") || name == "CodeResources")
1908             return;
1909
1910         auto &hash(local[name]);
1911         if (!hash.empty())
1912             return;
1913
1914         code(fun([&](std::streambuf &data, std::streambuf &save) {
1915             HashProxy proxy(hash, save);
1916             copy(data, proxy);
1917         }));
1918
1919         _assert(hash.size() == LDID_SHA1_DIGEST_LENGTH);
1920     }));
1921
1922     auto plist(plist_new_dict());
1923     _scope({ plist_free(plist); });
1924
1925     for (const auto &version : versions) {
1926         auto files(plist_new_dict());
1927         plist_dict_set_item(plist, ("files" + version.first).c_str(), files);
1928
1929         for (const auto &rule : version.second)
1930             rule.Compile();
1931
1932         for (const auto &hash : local)
1933             for (const auto &rule : version.second)
1934                 if (rule(hash.first)) {
1935                     if (rule.mode_ == NoMode)
1936                         plist_dict_set_item(files, hash.first.c_str(), plist_new_data(hash.second.data(), hash.second.size()));
1937                     else if (rule.mode_ == OptionalMode) {
1938                         auto entry(plist_new_dict());
1939                         plist_dict_set_item(entry, "hash", plist_new_data(hash.second.data(), hash.second.size()));
1940                         plist_dict_set_item(entry, "optional", plist_new_bool(true));
1941                         plist_dict_set_item(files, hash.first.c_str(), entry);
1942                     }
1943
1944                     break;
1945                 }
1946     }
1947
1948     for (const auto &version : versions) {
1949         auto rules(plist_new_dict());
1950         plist_dict_set_item(plist, ("rules" + version.first).c_str(), rules);
1951
1952         std::multiset<const Rule *, RuleCode> ordered;
1953         for (const auto &rule : version.second)
1954             ordered.insert(&rule);
1955
1956         for (const auto &rule : ordered)
1957             if (rule->weight_ == 1 && rule->mode_ == NoMode)
1958                 plist_dict_set_item(rules, rule->code_.c_str(), plist_new_bool(true));
1959             else {
1960                 auto entry(plist_new_dict());
1961                 plist_dict_set_item(rules, rule->code_.c_str(), entry);
1962
1963                 switch (rule->mode_) {
1964                     case NoMode:
1965                         break;
1966                     case OmitMode:
1967                         plist_dict_set_item(entry, "omit", plist_new_bool(true));
1968                         break;
1969                     case OptionalMode:
1970                         plist_dict_set_item(entry, "optional", plist_new_bool(true));
1971                         break;
1972                     case NestedMode:
1973                         plist_dict_set_item(entry, "nested", plist_new_bool(true));
1974                         break;
1975                     case TopMode:
1976                         plist_dict_set_item(entry, "top", plist_new_bool(true));
1977                         break;
1978                 }
1979
1980                 if (rule->weight_ >= 10000)
1981                     plist_dict_set_item(entry, "weight", plist_new_uint(rule->weight_));
1982                 else if (rule->weight_ != 1)
1983                     plist_dict_set_item(entry, "weight", plist_new_real(rule->weight_));
1984             }
1985     }
1986
1987     folder.Save(signature, fun([&](std::streambuf &save) {
1988         HashProxy proxy(local[signature], save);
1989         char *xml(NULL);
1990         uint32_t size;
1991         plist_to_xml(plist, &xml, &size);
1992         _scope({ free(xml); });
1993         put(proxy, xml, size);
1994     }));
1995
1996     folder.Open(executable, fun([&](std::streambuf &buffer) {
1997         // XXX: this is a miserable fail
1998         std::stringbuf temp;
1999         copy(buffer, temp);
2000         auto data(temp.str());
2001
2002         folder.Save(executable, fun([&](std::streambuf &save) {
2003             Slots slots;
2004             slots[1] = local.at(info);
2005             slots[3] = local.at(signature);
2006
2007             HashProxy proxy(local[executable], save);
2008             Sign(data.data(), data.size(), proxy, identifier, entitlements, key, slots);
2009         }));
2010     }));
2011
2012     for (const auto &hash : local)
2013         remote[root + hash.first] = hash.second;
2014
2015     return executable;
2016 }
2017 #endif
2018
2019 #endif
2020 }
2021
2022 #ifndef LDID_NOTOOLS
2023 int main(int argc, char *argv[]) {
2024 #ifndef LDID_NOSMIME
2025     OpenSSL_add_all_algorithms();
2026 #endif
2027
2028     union {
2029         uint16_t word;
2030         uint8_t byte[2];
2031     } endian = {1};
2032
2033     little_ = endian.byte[0];
2034
2035     bool flag_r(false);
2036     bool flag_e(false);
2037
2038 #ifndef LDID_NOFLAGT
2039     bool flag_T(false);
2040 #endif
2041
2042     bool flag_S(false);
2043     bool flag_s(false);
2044
2045     bool flag_D(false);
2046
2047     bool flag_A(false);
2048     bool flag_a(false);
2049
2050     bool flag_u(false);
2051
2052     uint32_t flag_CPUType(_not(uint32_t));
2053     uint32_t flag_CPUSubtype(_not(uint32_t));
2054
2055     const char *flag_I(NULL);
2056
2057 #ifndef LDID_NOFLAGT
2058     bool timeh(false);
2059     uint32_t timev(0);
2060 #endif
2061
2062     Map entitlements;
2063     Map key;
2064     ldid::Slots slots;
2065
2066     std::vector<std::string> files;
2067
2068     if (argc == 1) {
2069         fprintf(stderr, "usage: %s -S[entitlements.xml] <binary>\n", argv[0]);
2070         fprintf(stderr, "   %s -e MobileSafari\n", argv[0]);
2071         fprintf(stderr, "   %s -S cat\n", argv[0]);
2072         fprintf(stderr, "   %s -Stfp.xml gdb\n", argv[0]);
2073         exit(0);
2074     }
2075
2076     for (int argi(1); argi != argc; ++argi)
2077         if (argv[argi][0] != '-')
2078             files.push_back(argv[argi]);
2079         else switch (argv[argi][1]) {
2080             case 'r':
2081                 _assert(!flag_s);
2082                 _assert(!flag_S);
2083                 flag_r = true;
2084             break;
2085
2086             case 'e': flag_e = true; break;
2087
2088             case 'E': {
2089                 const char *slot = argv[argi] + 2;
2090                 const char *colon = strchr(slot, ':');
2091                 _assert(colon != NULL);
2092                 Map file(colon + 1, O_RDONLY, PROT_READ, MAP_PRIVATE);
2093                 char *arge;
2094                 unsigned number(strtoul(slot, &arge, 0));
2095                 _assert(arge == colon);
2096                 sha1(slots[number], file.data(), file.size());
2097             } break;
2098
2099             case 'D': flag_D = true; break;
2100
2101             case 'a': flag_a = true; break;
2102
2103             case 'A':
2104                 _assert(!flag_A);
2105                 flag_A = true;
2106                 if (argv[argi][2] != '\0') {
2107                     const char *cpu = argv[argi] + 2;
2108                     const char *colon = strchr(cpu, ':');
2109                     _assert(colon != NULL);
2110                     char *arge;
2111                     flag_CPUType = strtoul(cpu, &arge, 0);
2112                     _assert(arge == colon);
2113                     flag_CPUSubtype = strtoul(colon + 1, &arge, 0);
2114                     _assert(arge == argv[argi] + strlen(argv[argi]));
2115                 }
2116             break;
2117
2118             case 's':
2119                 _assert(!flag_r);
2120                 _assert(!flag_S);
2121                 flag_s = true;
2122             break;
2123
2124             case 'S':
2125                 _assert(!flag_r);
2126                 _assert(!flag_s);
2127                 flag_S = true;
2128                 if (argv[argi][2] != '\0') {
2129                     const char *xml = argv[argi] + 2;
2130                     entitlements.open(xml, O_RDONLY, PROT_READ, MAP_PRIVATE);
2131                 }
2132             break;
2133
2134             case 'K':
2135                 if (argv[argi][2] != '\0')
2136                     key.open(argv[argi] + 2, O_RDONLY, PROT_READ, MAP_PRIVATE);
2137             break;
2138
2139 #ifndef LDID_NOFLAGT
2140             case 'T': {
2141                 flag_T = true;
2142                 if (argv[argi][2] == '-')
2143                     timeh = true;
2144                 else {
2145                     char *arge;
2146                     timev = strtoul(argv[argi] + 2, &arge, 0);
2147                     _assert(arge == argv[argi] + strlen(argv[argi]));
2148                 }
2149             } break;
2150 #endif
2151
2152             case 'u': {
2153                 flag_u = true;
2154             } break;
2155
2156             case 'I': {
2157                 flag_I = argv[argi] + 2;
2158             } break;
2159
2160             default:
2161                 goto usage;
2162             break;
2163         }
2164
2165     _assert(flag_S || key.empty());
2166     _assert(flag_S || flag_I == NULL);
2167
2168     if (files.empty()) usage: {
2169         exit(0);
2170     }
2171
2172     size_t filei(0), filee(0);
2173     _foreach (file, files) try {
2174         std::string path(file);
2175
2176         struct stat info;
2177         _syscall(stat(path.c_str(), &info));
2178
2179         if (S_ISDIR(info.st_mode)) {
2180 #ifndef LDID_NOPLIST
2181             _assert(!flag_r);
2182             ldid::DiskFolder folder(path);
2183             std::map<std::string, std::vector<char>> hashes;
2184             path += "/" + Bundle("", folder, key, hashes, entitlements);
2185 #else
2186             _assert(false);
2187 #endif
2188         } else if (flag_S || flag_r) {
2189             Map input(path, O_RDONLY, PROT_READ, MAP_PRIVATE);
2190
2191             std::filebuf output;
2192             Split split(path);
2193             auto temp(Temporary(output, split));
2194
2195             if (flag_r)
2196                 ldid::Unsign(input.data(), input.size(), output);
2197             else {
2198                 std::string identifier(flag_I ?: split.base.c_str());
2199                 ldid::Sign(input.data(), input.size(), output, identifier, entitlements, key, slots);
2200             }
2201
2202             Commit(path, temp);
2203         }
2204
2205         bool modify(false);
2206 #ifndef LDID_NOFLAGT
2207         if (flag_T)
2208             modify = true;
2209 #endif
2210         if (flag_s)
2211             modify = true;
2212
2213         Map mapping(path, modify);
2214         FatHeader fat_header(mapping.data(), mapping.size());
2215
2216         _foreach (mach_header, fat_header.GetMachHeaders()) {
2217             struct linkedit_data_command *signature(NULL);
2218             struct encryption_info_command *encryption(NULL);
2219
2220             if (flag_A) {
2221                 if (mach_header.GetCPUType() != flag_CPUType)
2222                     continue;
2223                 if (mach_header.GetCPUSubtype() != flag_CPUSubtype)
2224                     continue;
2225             }
2226
2227             if (flag_a)
2228                 printf("cpu=0x%x:0x%x\n", mach_header.GetCPUType(), mach_header.GetCPUSubtype());
2229
2230             _foreach (load_command, mach_header.GetLoadCommands()) {
2231                 uint32_t cmd(mach_header.Swap(load_command->cmd));
2232
2233                 if (false);
2234                 else if (cmd == LC_CODE_SIGNATURE)
2235                     signature = reinterpret_cast<struct linkedit_data_command *>(load_command);
2236                 else if (cmd == LC_ENCRYPTION_INFO || cmd == LC_ENCRYPTION_INFO_64)
2237                     encryption = reinterpret_cast<struct encryption_info_command *>(load_command);
2238                 else if (cmd == LC_LOAD_DYLIB) {
2239                     volatile struct dylib_command *dylib_command(reinterpret_cast<struct dylib_command *>(load_command));
2240                     const char *name(reinterpret_cast<const char *>(load_command) + mach_header.Swap(dylib_command->dylib.name));
2241
2242                     if (strcmp(name, "/System/Library/Frameworks/UIKit.framework/UIKit") == 0) {
2243                         if (flag_u) {
2244                             Version version;
2245                             version.value = mach_header.Swap(dylib_command->dylib.current_version);
2246                             printf("uikit=%u.%u.%u\n", version.major, version.minor, version.patch);
2247                         }
2248                     }
2249                 }
2250 #ifndef LDID_NOFLAGT
2251                 else if (cmd == LC_ID_DYLIB) {
2252                     volatile struct dylib_command *dylib_command(reinterpret_cast<struct dylib_command *>(load_command));
2253
2254                     if (flag_T) {
2255                         uint32_t timed;
2256
2257                         if (!timeh)
2258                             timed = timev;
2259                         else {
2260                             dylib_command->dylib.timestamp = 0;
2261                             timed = hash(reinterpret_cast<uint8_t *>(mach_header.GetBase()), mach_header.GetSize(), timev);
2262                         }
2263
2264                         dylib_command->dylib.timestamp = mach_header.Swap(timed);
2265                     }
2266                 }
2267 #endif
2268             }
2269
2270             if (flag_D) {
2271                 _assert(encryption != NULL);
2272                 encryption->cryptid = mach_header.Swap(0);
2273             }
2274
2275             if (flag_e) {
2276                 _assert(signature != NULL);
2277
2278                 uint32_t data = mach_header.Swap(signature->dataoff);
2279
2280                 uint8_t *top = reinterpret_cast<uint8_t *>(mach_header.GetBase());
2281                 uint8_t *blob = top + data;
2282                 struct SuperBlob *super = reinterpret_cast<struct SuperBlob *>(blob);
2283
2284                 for (size_t index(0); index != Swap(super->count); ++index)
2285                     if (Swap(super->index[index].type) == CSSLOT_ENTITLEMENTS) {
2286                         uint32_t begin = Swap(super->index[index].offset);
2287                         struct Blob *entitlements = reinterpret_cast<struct Blob *>(blob + begin);
2288                         fwrite(entitlements + 1, 1, Swap(entitlements->length) - sizeof(*entitlements), stdout);
2289                     }
2290             }
2291
2292             if (flag_s) {
2293                 _assert(signature != NULL);
2294
2295                 uint32_t data = mach_header.Swap(signature->dataoff);
2296
2297                 uint8_t *top = reinterpret_cast<uint8_t *>(mach_header.GetBase());
2298                 uint8_t *blob = top + data;
2299                 struct SuperBlob *super = reinterpret_cast<struct SuperBlob *>(blob);
2300
2301                 for (size_t index(0); index != Swap(super->count); ++index)
2302                     if (Swap(super->index[index].type) == CSSLOT_CODEDIRECTORY) {
2303                         uint32_t begin = Swap(super->index[index].offset);
2304                         struct CodeDirectory *directory = reinterpret_cast<struct CodeDirectory *>(blob + begin);
2305
2306                         uint8_t (*hashes)[LDID_SHA1_DIGEST_LENGTH] = reinterpret_cast<uint8_t (*)[LDID_SHA1_DIGEST_LENGTH]>(blob + begin + Swap(directory->hashOffset));
2307                         uint32_t pages = Swap(directory->nCodeSlots);
2308
2309                         if (pages != 1)
2310                             for (size_t i = 0; i != pages - 1; ++i)
2311                                 sha1(hashes[i], top + PageSize_ * i, PageSize_);
2312                         if (pages != 0)
2313                             sha1(hashes[pages - 1], top + PageSize_ * (pages - 1), ((data - 1) % PageSize_) + 1);
2314                     }
2315             }
2316         }
2317
2318         ++filei;
2319     } catch (const char *) {
2320         ++filee;
2321         ++filei;
2322     }
2323
2324     return filee;
2325 }
2326 #endif