]> git.saurik.com Git - apple/xnu.git/blobdiff - bsd/kern/kern_symfile.c
xnu-344.49.tar.gz
[apple/xnu.git] / bsd / kern / kern_symfile.c
index d90edce209d30964466459605916c15e235ca60f..e667644912e8b05c89aad85f5d47c2e38af15e36 100644 (file)
@@ -3,19 +3,22 @@
  *
  * @APPLE_LICENSE_HEADER_START@
  * 
- * The contents of this file constitute Original Code as defined in and
- * are subject to the Apple Public Source License Version 1.1 (the
- * "License").  You may not use this file except in compliance with the
- * License.  Please obtain a copy of the License at
- * http://www.apple.com/publicsource and read it before using this file.
+ * Copyright (c) 1999-2003 Apple Computer, Inc.  All Rights Reserved.
  * 
- * This Original Code and all software distributed under the License are
- * distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this
+ * file.
+ * 
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
- * License for the specific language governing rights and limitations
- * under the License.
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
  * 
  * @APPLE_LICENSE_HEADER_END@
  */
 #include <vm/vm_kern.h>
 
 extern unsigned char   rootdevice[];
-extern vm_size_t       page_size;
-
-
-int kernel_symfile_opened = 0;
-int error_code = 0;
+extern struct mach_header _mh_execute_header;
 
-extern int  IODTGetLoaderInfo(  char *key, void **infoAddr, int *infoSize );
-extern void IODTFreeLoaderInfo( char *key, void *infoAddr, int infoSize );
+static int kernel_symfile_opened = 0;
+static int error_code = 0;
 
-struct segment_command *findSegmentByName( struct mach_header *mh, const char *section_name );
+extern int  IODTGetLoaderInfo(char *key, void **infoAddr, int *infoSize);
+extern void IODTFreeLoaderInfo(char *key, void *infoAddr, int infoSize);
 
 /*
  * 
  */
-int get_kernel_symfile( struct proc *p, char **symfile )
+static int output_kernel_symbols(struct proc *p)
 {
-    if ( kernel_symfile_opened == 0 )
-    {
-        kernel_symfile_opened = 1;
-        error_code = output_kernel_symbols( p );
+    struct vnode               *vp;
+    struct pcred               *pcred = p->p_cred;
+    struct ucred               *cred = pcred->pc_ucred;
+    struct nameidata           nd;
+    struct vattr               vattr;
+    struct load_command                *cmd;
+    struct mach_header         *orig_mh, *mh;
+    struct segment_command     *orig_ds, *orig_ts, *orig_le, *sg;
+    struct section             *se, *const_text;
+    struct symtab_command      *st, *orig_st;
+    struct nlist               *sym;
+    vm_size_t                  orig_mhsize, orig_st_size;
+    vm_offset_t                        header;
+    vm_size_t                  header_size;
+    int                                error, error1;
+    int                                i, j;
+    caddr_t                    addr;
+    vm_offset_t                        offset;
+    int                                rc_mh, rc_sc;
+
+    error = EFAULT;
+
+    vp         = NULL;
+    header     = NULL;
+    orig_mh    = NULL;
+    orig_st    = NULL;
+    
+    // Dispose of unnecessary gumf, the booter doesn't need to load these
+    rc_mh = IODTGetLoaderInfo("Kernel-__HEADER",
+                               (void **)&orig_mh, &orig_mhsize);
+    if (rc_mh && orig_mh)
+       IODTFreeLoaderInfo("Kernel-__HEADER",
+                           (void *)orig_mh, round_page(orig_mhsize));
+
+    rc_sc = IODTGetLoaderInfo("Kernel-__SYMTAB",
+                               (void **) &orig_st, &orig_st_size);
+    if (rc_sc && orig_st)
+       IODTFreeLoaderInfo("Kernel-__SYMTAB",
+                           (void *)orig_st, round_page(orig_st_size));
+
+    if (pcred->p_svuid != pcred->p_ruid || pcred->p_svgid != pcred->p_rgid)
+       goto out;
+
+    // Check to see if the root is 'e' or 'n', is this a test for network?
+    if (rootdevice[0] == 'e' && rootdevice[1] == 'n')
+       goto out;
+
+    NDINIT(&nd, LOOKUP, FOLLOW, UIO_SYSSPACE, "mach.sym", p);
+    if((error = vn_open(&nd, O_CREAT | FWRITE, S_IRUSR | S_IRGRP | S_IROTH))) goto out;
+
+    vp = nd.ni_vp;
+    
+    /* Don't dump to non-regular files or files with links. */
+    error = EFAULT;
+    if (vp->v_type != VREG || VOP_GETATTR(vp, &vattr, cred, p)
+    ||  vattr.va_nlink != 1)
+       goto out;
+
+    VATTR_NULL(&vattr);
+    vattr.va_size = 0;
+    VOP_LEASE(vp, p, cred, LEASE_WRITE);
+    VOP_SETATTR(vp, &vattr, cred, p);
+    p->p_acflag |= ACORE;
+
+    // If the file type is MH_EXECUTE then this must be a kernel
+    // as all Kernel extensions must be of type MH_OBJECT
+    orig_ds = orig_ts = orig_le = NULL;
+    orig_st = NULL;
+    orig_mh = &_mh_execute_header;
+    cmd = (struct load_command *) &orig_mh[1];
+    for (i = 0; i < orig_mh->ncmds; i++) {
+       if (cmd->cmd == LC_SEGMENT) {
+           struct segment_command *sg = (struct segment_command *) cmd;
+    
+           if (!strcmp(SEG_TEXT, sg->segname))
+               orig_ts = sg;
+           else if (!strcmp(SEG_DATA, sg->segname))
+               orig_ds = sg;
+           else if (!strcmp(SEG_LINKEDIT, sg->segname))
+               orig_le = sg;
+       }
+       else if (cmd->cmd == LC_SYMTAB)
+           orig_st = (struct symtab_command *) cmd;
+    
+       cmd = (struct load_command *) ((caddr_t) cmd + cmd->cmdsize);
     }
-    if ( error_code == 0 ) *symfile = "\\mach.sym";
 
-    return error_code;
-}
+    if (!orig_ts || !orig_ds || !orig_le || !orig_st) 
+       goto out;
 
-/*
- * 
- */
-int output_kernel_symbols( register struct proc *p )
-{
-       register struct vnode           *vp;
-       register struct pcred           *pcred = p->p_cred;
-       register struct ucred           *cred = pcred->pc_ucred;
-       struct nameidata                nd;
-       struct vattr                    vattr;
-       struct mach_header              *orig_mh, *mh;
-        struct load_command            *lc;
-        struct segment_command         *orig_ds, *orig_ts, *sg;
-        struct section                 *se;
-       struct symtab_command           *sc, *sc0;
-        struct nlist                    *nl;
-       vm_size_t                       orig_mhsize, sc0_size;
-        vm_offset_t                    header;
-        vm_size_t                      header_size;
-       int                             error, error1;
-        int                            i, j;
-        int                            symfoffset, symsize;
-        int                             rc_mh, rc_sc;
-
-        error = EFAULT;
-
-        vp       = NULL;
-        header   = NULL;
-        orig_mh  = NULL;
-        sc0      = NULL;
-        
-        rc_mh = IODTGetLoaderInfo( "Kernel-__HEADER", (void **)&orig_mh, &orig_mhsize );
-        rc_sc = IODTGetLoaderInfo( "Kernel-__SYMTAB", (void **)&sc0, &sc0_size );
-        
-        if ( rc_mh != 0 || orig_mh == 0 || orig_mhsize < sizeof(struct mach_header) )    goto out;
-        if ( rc_sc != 0 || sc0 == 0     || sc0_size    < sizeof(struct symtab_command) ) goto out;
-
-       if ( pcred->p_svuid != pcred->p_ruid || pcred->p_svgid != pcred->p_rgid ) goto out;
-
-        if ( rootdevice[0] == 'e' && rootdevice[1] == 'n' ) goto out;
-
-       NDINIT(&nd, LOOKUP, FOLLOW, UIO_SYSSPACE, "mach.sym", p);
-       if( (error = vn_open(&nd, O_CREAT | FWRITE, S_IRUSR | S_IRGRP | S_IROTH )) != 0 ) goto out;
-
-       vp = nd.ni_vp;
-       
-       /* Don't dump to non-regular files or files with links. */
-        error = EFAULT;
-       if (vp->v_type != VREG || VOP_GETATTR(vp, &vattr, cred, p) || vattr.va_nlink != 1) goto out;
-
-       VATTR_NULL(&vattr);
-       vattr.va_size = 0;
-       VOP_LEASE(vp, p, cred, LEASE_WRITE);
-       VOP_SETATTR(vp, &vattr, cred, p);
-       p->p_acflag |= ACORE;
-
-        orig_ts = findSegmentByName(orig_mh, "__TEXT");
-        orig_ds = findSegmentByName(orig_mh, "__DATA");
-
-        if ( orig_ts == NULL || orig_ds == NULL ) goto out;
-
-       header_size =   sizeof(struct mach_header) 
-                      + orig_ts->cmdsize
-                      + orig_ds->cmdsize
-                      + sizeof(struct symtab_command);
-
-       (void) kmem_alloc_wired( kernel_map,
-                               (vm_offset_t *)&header,
-                               (vm_size_t)header_size);
-
-        if ( header == NULL ) goto out;
-
-        bzero( (void *)header, header_size );
-
-       /*
-        *      Set up Mach-O header.
-        */
-       mh = (struct mach_header *) header;
-       mh->magic = orig_mh->magic;
-       mh->cputype = orig_mh->cputype;
-       mh->cpusubtype = orig_mh->cpusubtype;
-       mh->filetype = orig_mh->filetype;
-       mh->ncmds = 3;
-       mh->sizeofcmds = header_size - sizeof(struct mach_header);
-
-       /*
-        *      Copy __DATA and __TEXT segment commands from mach_kernel so loadable drivers
-         *     get correct section alignment hints.
-        */
-        sg = (struct segment_command *)(mh+1);
-        bcopy( orig_ts, sg, orig_ts->cmdsize );
-
-        sg = (struct segment_command *)((int)sg + sg->cmdsize);
-        bcopy( orig_ds, sg, orig_ds->cmdsize );
-
-        sg = (struct segment_command *)(mh+1);
-        
-        for ( i = 0; i < 2; i++ )
-        {
-            sg->vmaddr   = 0;
-            sg->vmsize   = 0x1000;
-            sg->fileoff  = 0;
-            sg->filesize = 0;
-            sg->maxprot  = 0;
-            sg->initprot = 0;
-            sg->flags    = 0;
-
-            se = (struct section *)(sg+1);
-            for ( j = 0; j < sg->nsects; j++, se++ )
-            {
-                se->addr  = 0;
-                se->size  = 0;
-                se->offset = 0;
-                se->nreloc = 0;
-            }
-
-            sg = (struct segment_command *)((int)sg + sg->cmdsize);
-        }
-
-        symfoffset = round_page(header_size);
-
-       /*
-        *      Set up LC_SYMTAB command
-        */
-       sc          = (struct symtab_command *)sg;
-       sc->cmd     = LC_SYMTAB;
-       sc->cmdsize = sizeof(struct symtab_command);
-        sc->symoff  = symfoffset;
-        sc->nsyms   = sc0->nsyms;
-        sc->strsize = sc0->strsize;
-        sc->stroff =  symfoffset + sc->nsyms * sizeof(struct nlist);    
-
-        symsize = sc->nsyms * sizeof(struct nlist) + sc->strsize;
-
-        nl = (struct nlist *)(sc0+1);
-        for (i = 0; i < sc->nsyms; i++, nl++ )
-        {
-            if ( (nl->n_type & N_TYPE) == N_SECT )
-            {
-                nl->n_sect = NO_SECT;
-                nl->n_type = (nl->n_type & ~N_TYPE) | N_ABS;
-            }
-        }
-
-       /*
-        *      Write out the load commands at the beginning of the
-        *      file.
-        */
-       error = vn_rdwr(UIO_WRITE, vp, (caddr_t)mh, header_size, (off_t)0,
-                       UIO_SYSSPACE, IO_NODELOCKED|IO_UNIT, cred, (int *) 0, p);
-        if ( error != 0 ) goto out;
-        
-        /*
-         *     Write out kernel symbols
-         */
-       error = vn_rdwr(UIO_WRITE, vp, (caddr_t)(sc0+1), symsize, symfoffset,
-                       UIO_SYSSPACE, IO_NODELOCKED|IO_UNIT, cred, (int *) 0, p);
-        if ( error != 0 ) goto out;
+    const_text = NULL;
+    se = (struct section *) &orig_ts[1];
+    for (i = 0; i < orig_ts->nsects; i++, se++) {
+       if (!strcmp("__const", se->sectname)) {
+           const_text = se;
+           break;
+       }
+    }
+    if (!const_text)
+       goto out;
+
+    header_size =   sizeof(struct mach_header) 
+                   + orig_ts->cmdsize
+                   + orig_ds->cmdsize
+                   + sizeof(struct symtab_command);
+
+    (void) kmem_alloc_wired(kernel_map,
+                           (vm_offset_t *) &header,
+                           (vm_size_t) header_size);
+    if (header)
+       bzero((void *) header, header_size);
+    else
+       goto out;
+
+    /*
+     * Set up Mach-O header.
+     */
+    mh = (struct mach_header *) header;
+    mh->magic      = orig_mh->magic;
+    mh->cputype    = orig_mh->cputype;
+    mh->cpusubtype = orig_mh->cpusubtype;
+    mh->filetype   = orig_mh->filetype;
+    mh->ncmds      = 3;
+    mh->sizeofcmds = header_size - sizeof(struct mach_header);
+    mh->flags      = orig_mh->flags;
+
+    // Initialise the current file offset and addr
+    offset = round_page(header_size);
+    addr = (caddr_t) const_text->addr; // Load address of __TEXT,__const
+
+    /*
+     * Construct a TEXT segment load command
+     * the only part of the TEXT segment we keep is the __TEXT,__const
+     * which contains the kernel vtables. 
+     */
+    sg = (struct segment_command *) &mh[1];
+    bcopy(orig_ts, sg, orig_ts->cmdsize);
+    sg->vmaddr   = (unsigned long) addr;
+    sg->vmsize   = const_text->size;
+    sg->fileoff  = 0;
+    sg->filesize = const_text->size + round_page(header_size);
+    sg->maxprot  = 0;
+    sg->initprot = 0;
+    sg->flags    = 0;
+    se = (struct section *)(sg+1);
+    for ( j = 0; j < sg->nsects; j++, se++ ) {
+       se->addr  = (unsigned long) addr;
+       se->size  = 0;
+       se->offset = offset;
+       se->nreloc = 0;
+       if (!strcmp("__const", se->sectname)) {
+           se->size = const_text->size;
+           addr    += const_text->size;
+           offset  += const_text->size;
+           const_text = se;
+       }
+    }
+    offset = round_page((vm_address_t) offset);
+
+    // Now copy of the __DATA segment load command, the image need
+    // not be stored to disk nobody needs it, yet!
+    sg = (struct segment_command *)((int)sg + sg->cmdsize);
+    bcopy(orig_ds, sg, orig_ds->cmdsize);
+
+    sg->vmaddr   = (unsigned long) addr;
+    sg->vmsize   = 0x1000;     // One page for some reason?
+    sg->fileoff  = offset;
+    sg->filesize = 0;
+    sg->maxprot  = 0;
+    sg->initprot = 0;
+    sg->flags    = 0;
+    se = (struct section *)(sg+1);
+    for ( j = 0; j < sg->nsects; j++, se++ ) {
+       se->addr  = (unsigned long) addr;
+       se->size  = 0;
+       se->offset = offset;
+       se->nreloc = 0;
+    }
+    offset = round_page(offset);
+
+
+    /*
+     * Set up LC_SYMTAB command
+     */
+    st          = (struct symtab_command *)((int)sg + sg->cmdsize);
+    st->cmd     = LC_SYMTAB;
+    st->cmdsize = sizeof(struct symtab_command);
+    st->symoff  = offset;
+    st->nsyms   = orig_st->nsyms;
+    st->strsize = orig_st->strsize;
+    st->stroff =  offset + st->nsyms * sizeof(struct nlist);    
+
+    /*
+     * Convert the symbol table in place from section references
+     * to absolute references.
+     */
+    sym = (struct nlist *) orig_le->vmaddr;
+    for (i = 0; i < st->nsyms; i++, sym++ ) {
+       if ( (sym->n_type & N_TYPE) == N_SECT) {
+           sym->n_sect = NO_SECT;
+           sym->n_type = (sym->n_type & ~N_TYPE) | N_ABS;
+       }
+    }
+
+    /*
+     * Write out the load commands at the beginning of the file.
+     */
+    error = vn_rdwr(UIO_WRITE, vp, (caddr_t) mh, header_size, (off_t) 0,
+                   UIO_SYSSPACE, IO_NODELOCKED|IO_UNIT, cred, (int *) 0, p);
+    if (error)
+       goto out;
+
+    /*
+     * Write out the __TEXT,__const data segment.
+     */
+    error = vn_rdwr(UIO_WRITE, vp, (caddr_t) const_text->addr,
+                   const_text->size, const_text->offset,
+                   UIO_SYSSPACE, IO_NODELOCKED|IO_UNIT, cred, (int *) 0, p);
+    if (error)
+       goto out;
+
+    /*
+     *         Write out kernel symbols
+     */
+    offset = st->nsyms * sizeof(struct nlist) + st->strsize;   // symtab size
+    error = vn_rdwr(UIO_WRITE, vp,
+                  (caddr_t) orig_le->vmaddr, offset, st->symoff,
+                   UIO_SYSSPACE, IO_NODELOCKED|IO_UNIT, cred, (int *) 0, p);
+    if (error)
+       goto out;
 
 out:
-        if ( header != 0 ) kmem_free(kernel_map, header, header_size);
-        if ( orig_mh != 0 ) IODTFreeLoaderInfo( "Kernel-__HEADER", (void *)orig_mh, round_page(orig_mhsize) );
-        if ( sc0     != 0 ) IODTFreeLoaderInfo( "Kernel-__SYMTAB", (void *)sc0,     round_page(sc0_size) );        
-  
-        if ( vp != 0 )
-        {
-           VOP_UNLOCK(vp, 0, p);
-           error1 = vn_close(vp, FWRITE, cred, p);
-           if (error == 0) error = error1;
-        }
-
-       return(error);
-}
+    if (header)
+       kmem_free(kernel_map, header, header_size);
+
+    if (vp) {
+       VOP_UNLOCK(vp, 0, p);
+       error1 = vn_close(vp, FWRITE, cred, p);
+       if (!error) error = error1;
+    }
 
+    return(error);
+}
 /*
  * 
  */
-struct segment_command *findSegmentByName( struct mach_header *mh, const char *section_name )
+int get_kernel_symfile(struct proc *p, char **symfile)
 {
-    struct segment_command     *sg;
-    int                                i;
-     
-    sg = (struct segment_command *)(mh+1);
-
-    for ( i=0; i < mh->ncmds; i++ )
-    {
-        if ( (sg->cmd == LC_SEGMENT) && (strcmp(sg->segname, section_name) == 0) )
-        {
-            return sg;
-        }
-        
-        sg = (struct segment_command *)((int)sg + sg->cmdsize);
+    if (!kernel_symfile_opened) {
+        kernel_symfile_opened = 1;
+        error_code = output_kernel_symbols(p);
     }
+    if (!error_code)
+       *symfile = "\\mach.sym";
 
-    return NULL;
-}    
-            
-     
-       
-            
-
+    return error_code;
+}