gen/util/machOconv.c

   1 /*
   2  * Copyright (c) 1999 Apple Computer, Inc. All rights reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * Copyright (c) 1999-2003 Apple Computer, Inc.  All Rights Reserved.
   7  *
   8  * This file contains Original Code and/or Modifications of Original Code
   9  * as defined in and that are subject to the Apple Public Source License
  10  * Version 2.0 (the 'License'). You may not use this file except in
  11  * compliance with the License. Please obtain a copy of the License at
  12  * http://www.opensource.apple.com/apsl/ and read it before using this
  13  * file.
  14  *
  15  * The Original Code and all software distributed under the License are
  16  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  17  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  18  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  19  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  20  * Please see the License for the specific language governing rights and
  21  * limitations under the License.
  22  *
  23  * @APPLE_LICENSE_HEADER_END@
  24  */
  25 #include <stdio.h>
  26
  27 #include <mach/mach.h>
  28 #include <sys/file.h>
  29 #include <mach-o/loader.h>
  30 #include <libkern/OSByteOrder.h>
  31
  32 int     infile, outfile;
  33
  34 struct mach_header      mh;
  35 unsigned                cmds;
  36
  37 boolean_t               swap_ends;
  38
  39 static unsigned long swap(
  40     unsigned long x
  41 )
  42 {
  43     if (swap_ends)
  44         return OSSwapInt32(x);
  45     else
  46         return x;
  47 }
  48
  49 main(argc, argv)
  50 int     argc;
  51 char    *argv[];
  52 {
  53     kern_return_t       result;
  54     vm_address_t        data;
  55     int                 nc, ncmds;
  56     unsigned            cp;
  57
  58     if (argc == 2) {
  59         infile = open(argv[1], O_RDONLY);
  60         if (infile < 0)
  61             goto usage;
  62         outfile = fileno(stdout);
  63     }
  64     else if (argc == 3) {
  65         infile = open(argv[1], O_RDONLY);
  66         if (infile < 0)
  67             goto usage;
  68         outfile = open(argv[2], O_WRONLY|O_CREAT|O_TRUNC, 0644);
  69         if (outfile < 0)
  70             goto usage;
  71     }
  72     else {
  73 usage:
  74         fprintf(stderr, "usage: machOconv inputfile [outputfile]\n");
  75         exit(1);
  76     }
  77
  78     nc = read(infile, &mh, sizeof (mh));
  79     if (nc < 0) {
  80         perror("read mach header");
  81         exit(1);
  82     }
  83     if (nc < sizeof (mh)) {
  84         fprintf(stderr, "read mach header: premature EOF %d\n", nc);
  85         exit(1);
  86     }
  87     if (mh.magic == MH_MAGIC)
  88         swap_ends = FALSE;
  89     else if (mh.magic == MH_CIGAM)
  90         swap_ends = TRUE;
  91     else {
  92         fprintf(stderr, "bad magic number %x\n", mh.magic);
  93         exit(1);
  94     }
  95
  96     cmds = calloc(swap(mh.sizeofcmds), sizeof (char));
  97     if (cmds == 0) {
  98         fprintf(stderr, "alloc load commands: no memory\n");
  99         exit(1);
 100     }
 101     nc = read(infile, cmds, swap(mh.sizeofcmds));
 102     if (nc < 0) {
 103         perror("read load commands");
 104         exit(1);
 105     }
 106     if (nc < swap(mh.sizeofcmds)) {
 107         fprintf(stderr, "read load commands: premature EOF %d\n", nc);
 108         exit(1);
 109     }
 110
 111     for (       ncmds = swap(mh.ncmds), cp = cmds;
 112                 ncmds > 0; ncmds--) {
 113             boolean_t   isDATA;
 114             unsigned    vmsize;
 115
 116 #define lcp     ((struct load_command *)cp)
 117         switch(swap(lcp->cmd)) {
 118
 119         case LC_SEGMENT:
 120 #define scp     ((struct segment_command *)cp)
 121             isDATA = (strcmp(scp->segname, "__DATA") == 0);
 122             if (isDATA)
 123                 vmsize = swap(scp->filesize);
 124             else
 125                 vmsize = swap(scp->vmsize);
 126             result = vm_allocate(mach_task_self(), &data, vmsize, TRUE);
 127             if (result != KERN_SUCCESS) {
 128                 mach_error("vm_allocate segment data", result);
 129                 exit(1);
 130             }
 131
 132             b_lseek(infile, swap(scp->fileoff), L_SET);
 133             nc = read(infile, data, swap(scp->filesize));
 134             if (nc < 0) {
 135                 perror("read segment data");
 136                 exit(1);
 137             }
 138             if (nc < swap(scp->filesize)) {
 139                 fprintf(stderr, "read segment data: premature EOF %d\n", nc);
 140                 exit(1);
 141             }
 142
 143             nc = write(outfile, data, vmsize);
 144             if (nc < vmsize) {
 145                 perror("write segment data");
 146                 exit(1);
 147             }
 148
 149             vm_deallocate(mach_task_self(), data, vmsize);
 150             break;
 151         }
 152
 153         cp += swap(lcp->cmdsize);
 154     }
 155
 156     exit(0);
 157 }