#include <mach/vm_param.h>
#include <sys/param.h>
#include <sys/systm.h>
#include <sys/signalvar.h>
#include <sys/resourcevar.h>
#include <sys/namei.h>
#include <sys/vnode.h>
#include <sys/proc.h>
#include <sys/timeb.h>
#include <sys/times.h>
#include <sys/buf.h>
#include <sys/acct.h>
#include <sys/file.h>
#include <sys/uio.h>
#include <sys/kernel.h>
#include <sys/stat.h>
#include <mach-o/loader.h>
#include <mach-o/nlist.h>
#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 int IODTGetLoaderInfo( char *key, void **infoAddr, int *infoSize );
extern void IODTFreeLoaderInfo( char *key, void *infoAddr, int infoSize );
struct segment_command *findSegmentByName( struct mach_header *mh, const char *section_name );
int get_kernel_symfile( struct proc *p, char **symfile )
{
if ( kernel_symfile_opened == 0 )
{
kernel_symfile_opened = 1;
error_code = output_kernel_symbols( p );
}
if ( error_code == 0 ) *symfile = "\\mach.sym";
return error_code;
}
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;
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 );
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);
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);
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;
}
}
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;
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;
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);
}
struct segment_command *findSegmentByName( struct mach_header *mh, const char *section_name )
{
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);
}
return NULL;
}