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