1 a84cbb2a 2004-04-19 devnull #include <u.h>
  2 a84cbb2a 2004-04-19 devnull #include <libc.h>
  3 a84cbb2a 2004-04-19 devnull #include <mach.h>
  4 a84cbb2a 2004-04-19 devnull #include "elf.h"
  5 a84cbb2a 2004-04-19 devnull #include "dwarf.h"
  6 a84cbb2a 2004-04-19 devnull 
  7 a84cbb2a 2004-04-19 devnull static int mapelf(Fhdr *fp, ulong base, Map *map, Regs**);
  8 dd944ec7 2005-01-23 devnull static int unpacknote(Elf *elf, uchar *a, uchar *ea, ElfNote *note, uchar **pa);
  9 a84cbb2a 2004-04-19 devnull 
 10 a84cbb2a 2004-04-19 devnull static struct
 11 a84cbb2a 2004-04-19 devnull {
 12 a84cbb2a 2004-04-19 devnull 	uint etype;
 13 a84cbb2a 2004-04-19 devnull 	uint mtype;
 14 a84cbb2a 2004-04-19 devnull 	Mach *mach;
 15 a84cbb2a 2004-04-19 devnull 	char *name;
 16 a84cbb2a 2004-04-19 devnull } mtab[] = 
 17 a84cbb2a 2004-04-19 devnull {	/* Font Tab 4 */
 18 a84cbb2a 2004-04-19 devnull 	ElfMachSparc,	MSPARC,		nil,		"sparc",
 19 a84cbb2a 2004-04-19 devnull 	ElfMach386,		M386,		&mach386,	"386",
 20 a84cbb2a 2004-04-19 devnull 	ElfMachMips,	MMIPS,		nil,		"mips",
 21 a84cbb2a 2004-04-19 devnull 	ElfMachArm,		MARM,		nil,		"arm",
 22 a84cbb2a 2004-04-19 devnull 	ElfMachPower,	MPOWER,		nil,		"powerpc",
 23 a84cbb2a 2004-04-19 devnull 	ElfMachPower64,	MNONE,		nil,		"powerpc64",
 24 a84cbb2a 2004-04-19 devnull };
 25 a84cbb2a 2004-04-19 devnull 
 26 a84cbb2a 2004-04-19 devnull static struct
 27 a84cbb2a 2004-04-19 devnull {
 28 a84cbb2a 2004-04-19 devnull 	uint etype;
 29 a84cbb2a 2004-04-19 devnull 	uint atype;
 30 a84cbb2a 2004-04-19 devnull 	char *aname;
 31 a84cbb2a 2004-04-19 devnull } atab[] = 
 32 a84cbb2a 2004-04-19 devnull {	/* Font Tab 4 */
 33 a84cbb2a 2004-04-19 devnull 	ElfAbiSystemV,	ALINUX,		"linux",	/* [sic] */
 34 a84cbb2a 2004-04-19 devnull 	ElfAbiLinux,	ALINUX,		"linux",
 35 a84cbb2a 2004-04-19 devnull 	ElfAbiFreeBSD,	AFREEBSD,	"freebsd",
 36 a84cbb2a 2004-04-19 devnull };
 37 a84cbb2a 2004-04-19 devnull 
 38 a84cbb2a 2004-04-19 devnull static struct
 39 a84cbb2a 2004-04-19 devnull {
 40 a84cbb2a 2004-04-19 devnull 	uint mtype;
 41 a84cbb2a 2004-04-19 devnull 	uint atype;
 42 dd944ec7 2005-01-23 devnull 	void (*elfcore)(Fhdr*, Elf*, ElfNote*);
 43 a84cbb2a 2004-04-19 devnull } ctab[] = 
 44 a84cbb2a 2004-04-19 devnull {	/* Font Tab 4 */
 45 dd944ec7 2005-01-23 devnull 	M386,		ALINUX,		elfcorelinux386,
 46 dd944ec7 2005-01-23 devnull 	M386,		ANONE,		elfcorelinux386,	/* [sic] */
 47 dd944ec7 2005-01-23 devnull /*	M386,		AFREEBSD,	elfcorefreebsd386, */
 48 a84cbb2a 2004-04-19 devnull };
 49 a84cbb2a 2004-04-19 devnull 
 50 a84cbb2a 2004-04-19 devnull int
 51 a84cbb2a 2004-04-19 devnull crackelf(int fd, Fhdr *fp)
 52 a84cbb2a 2004-04-19 devnull {
 53 dd944ec7 2005-01-23 devnull 	uchar *a, *sa, *ea;
 54 dd944ec7 2005-01-23 devnull 	int i, havetext, havedata, n;
 55 a84cbb2a 2004-04-19 devnull 	Elf *elf;
 56 dd944ec7 2005-01-23 devnull 	ElfNote note;
 57 a84cbb2a 2004-04-19 devnull 	ElfProg *p;
 58 a84cbb2a 2004-04-19 devnull 	ElfSect *s1, *s2;
 59 dd944ec7 2005-01-23 devnull 	void (*elfcore)(Fhdr*, Elf*, ElfNote*);
 60 a84cbb2a 2004-04-19 devnull 
 61 a84cbb2a 2004-04-19 devnull 	if((elf = elfinit(fd)) == nil)
 62 a84cbb2a 2004-04-19 devnull 		return -1;
 63 a84cbb2a 2004-04-19 devnull 
 64 a84cbb2a 2004-04-19 devnull 	fp->fd = fd;
 65 a84cbb2a 2004-04-19 devnull 	fp->elf = elf;
 66 a84cbb2a 2004-04-19 devnull 	fp->dwarf = dwarfopen(elf);	/* okay to fail */
 67 a84cbb2a 2004-04-19 devnull 	fp->syminit = symelf;
 68 a84cbb2a 2004-04-19 devnull 
 69 a84cbb2a 2004-04-19 devnull 	if((s1 = elfsection(elf, ".stab")) != nil && s1->link!=0 && s1->link < elf->nsect){
 70 a84cbb2a 2004-04-19 devnull 		s2 = &elf->sect[s1->link];
 71 a84cbb2a 2004-04-19 devnull 		if(elfmap(elf, s1) >= 0 && elfmap(elf, s2) >= 0){
 72 a84cbb2a 2004-04-19 devnull 			fp->stabs.stabbase = s1->base;
 73 a84cbb2a 2004-04-19 devnull 			fp->stabs.stabsize = s1->size;
 74 929ba9b9 2004-04-20 devnull 			fp->stabs.strbase = (char*)s2->base;
 75 a84cbb2a 2004-04-19 devnull 			fp->stabs.strsize = s2->size;
 76 a84cbb2a 2004-04-19 devnull 			fp->stabs.e2 = elf->hdr.e2;
 77 a84cbb2a 2004-04-19 devnull 			fp->stabs.e4 = elf->hdr.e4;
 78 a84cbb2a 2004-04-19 devnull 		}
 79 a84cbb2a 2004-04-19 devnull 	}
 80 a84cbb2a 2004-04-19 devnull 
 81 a84cbb2a 2004-04-19 devnull 	for(i=0; i<nelem(mtab); i++){
 82 a84cbb2a 2004-04-19 devnull 		if(elf->hdr.machine != mtab[i].etype)
 83 a84cbb2a 2004-04-19 devnull 			continue;
 84 a84cbb2a 2004-04-19 devnull 		fp->mach = mtab[i].mach;
 85 a84cbb2a 2004-04-19 devnull 		fp->mname = mtab[i].name;
 86 a84cbb2a 2004-04-19 devnull 		fp->mtype = mtab[i].mtype;
 87 a84cbb2a 2004-04-19 devnull 		break;
 88 a84cbb2a 2004-04-19 devnull 	}
 89 a84cbb2a 2004-04-19 devnull 	if(i == nelem(mtab)){
 90 a84cbb2a 2004-04-19 devnull 		werrstr("unsupported machine type %d", elf->hdr.machine);
 91 a84cbb2a 2004-04-19 devnull 		goto err;
 92 a84cbb2a 2004-04-19 devnull 	}
 93 a84cbb2a 2004-04-19 devnull 
 94 a84cbb2a 2004-04-19 devnull 	if(mach == nil)
 95 a84cbb2a 2004-04-19 devnull 		mach = fp->mach;
 96 a84cbb2a 2004-04-19 devnull 
 97 a84cbb2a 2004-04-19 devnull 	fp->aname = "unknown";
 98 a84cbb2a 2004-04-19 devnull 	for(i=0; i<nelem(atab); i++){
 99 a84cbb2a 2004-04-19 devnull 		if(elf->hdr.abi != atab[i].etype)
100 a84cbb2a 2004-04-19 devnull 			continue;
101 a84cbb2a 2004-04-19 devnull 		fp->atype = atab[i].atype;
102 a84cbb2a 2004-04-19 devnull 		fp->aname = atab[i].aname;
103 a84cbb2a 2004-04-19 devnull 		break;
104 a84cbb2a 2004-04-19 devnull 	}
105 a84cbb2a 2004-04-19 devnull 
106 a84cbb2a 2004-04-19 devnull 	switch(elf->hdr.type){
107 a84cbb2a 2004-04-19 devnull 	default:
108 a84cbb2a 2004-04-19 devnull 		werrstr("unknown file type %d", elf->hdr.type);
109 a84cbb2a 2004-04-19 devnull 		goto err;
110 a84cbb2a 2004-04-19 devnull 	case ElfTypeExecutable:
111 a84cbb2a 2004-04-19 devnull 		fp->ftype = FEXEC;
112 a84cbb2a 2004-04-19 devnull 		fp->fname = "executable";
113 a84cbb2a 2004-04-19 devnull 		break;
114 a84cbb2a 2004-04-19 devnull 	case ElfTypeRelocatable:
115 a84cbb2a 2004-04-19 devnull 		fp->ftype = FRELOC;
116 a84cbb2a 2004-04-19 devnull 		fp->fname = "relocatable";
117 a84cbb2a 2004-04-19 devnull 		break;
118 a84cbb2a 2004-04-19 devnull 	case ElfTypeSharedObject:
119 a84cbb2a 2004-04-19 devnull 		fp->ftype = FSHOBJ;
120 a84cbb2a 2004-04-19 devnull 		fp->fname = "shared object";
121 a84cbb2a 2004-04-19 devnull 		break;
122 a84cbb2a 2004-04-19 devnull 	case ElfTypeCore:
123 a84cbb2a 2004-04-19 devnull 		fp->ftype = FCORE;
124 a84cbb2a 2004-04-19 devnull 		fp->fname = "core dump";
125 a84cbb2a 2004-04-19 devnull 		break;
126 a84cbb2a 2004-04-19 devnull 	}
127 a84cbb2a 2004-04-19 devnull 
128 a84cbb2a 2004-04-19 devnull 	fp->map = mapelf;
129 a84cbb2a 2004-04-19 devnull 
130 a84cbb2a 2004-04-19 devnull 	if(fp->ftype == FCORE){
131 dd944ec7 2005-01-23 devnull 		elfcore = nil;
132 a84cbb2a 2004-04-19 devnull 		for(i=0; i<nelem(ctab); i++){
133 a84cbb2a 2004-04-19 devnull 			if(ctab[i].atype != fp->atype
134 a84cbb2a 2004-04-19 devnull 			|| ctab[i].mtype != fp->mtype)
135 a84cbb2a 2004-04-19 devnull 				continue;
136 dd944ec7 2005-01-23 devnull 			elfcore = ctab[i].elfcore;
137 a84cbb2a 2004-04-19 devnull 			break;
138 a84cbb2a 2004-04-19 devnull 		}
139 dd944ec7 2005-01-23 devnull 		if(elfcore)
140 dd944ec7 2005-01-23 devnull 		for(i=0; i<elf->nprog; i++){
141 dd944ec7 2005-01-23 devnull 			p = &elf->prog[i];
142 dd944ec7 2005-01-23 devnull 			if(p->type != ElfProgNote)
143 dd944ec7 2005-01-23 devnull 				continue;
144 dd944ec7 2005-01-23 devnull 			n = p->filesz;
145 dd944ec7 2005-01-23 devnull 			a = malloc(n);
146 dd944ec7 2005-01-23 devnull 			if(a == nil)
147 dd944ec7 2005-01-23 devnull 				goto err;
148 dd944ec7 2005-01-23 devnull 			if(seek(fp->fd, p->offset, 0) < 0 || readn(fp->fd, a, n) != n){
149 dd944ec7 2005-01-23 devnull 				free(a);
150 dd944ec7 2005-01-23 devnull 				continue;
151 dd944ec7 2005-01-23 devnull 			}
152 dd944ec7 2005-01-23 devnull 			sa = a;
153 dd944ec7 2005-01-23 devnull 			ea = a+n;
154 dd944ec7 2005-01-23 devnull 			while(a < ea){
155 dd944ec7 2005-01-23 devnull 				note.offset = (a-sa) + p->offset;
156 dd944ec7 2005-01-23 devnull 				if(unpacknote(elf, a, ea, &note, &a) < 0)
157 dd944ec7 2005-01-23 devnull 					break;
158 dd944ec7 2005-01-23 devnull 				elfcore(fp, elf, &note);
159 dd944ec7 2005-01-23 devnull 			}
160 dd944ec7 2005-01-23 devnull 			free(sa);
161 dd944ec7 2005-01-23 devnull 		}
162 a84cbb2a 2004-04-19 devnull 		return 0;
163 a84cbb2a 2004-04-19 devnull 	}
164 a84cbb2a 2004-04-19 devnull 
165 a84cbb2a 2004-04-19 devnull 	fp->entry = elf->hdr.entry;
166 a84cbb2a 2004-04-19 devnull 
167 a84cbb2a 2004-04-19 devnull 	/* First r-x section we find is the text and initialized data */
168 a84cbb2a 2004-04-19 devnull 	/* First rw- section we find is the r/w data */
169 a84cbb2a 2004-04-19 devnull 	havetext = 0;
170 a84cbb2a 2004-04-19 devnull 	havedata = 0;
171 a84cbb2a 2004-04-19 devnull 	for(i=0; i<elf->nprog; i++){
172 a84cbb2a 2004-04-19 devnull 		p = &elf->prog[i];
173 a84cbb2a 2004-04-19 devnull 		if(p->type != ElfProgLoad)
174 a84cbb2a 2004-04-19 devnull 			continue;
175 a84cbb2a 2004-04-19 devnull 		if(!havetext && p->flags == (ElfProgFlagRead|ElfProgFlagExec) && p->align >= mach->pgsize){
176 a84cbb2a 2004-04-19 devnull 			havetext = 1;
177 a84cbb2a 2004-04-19 devnull 			fp->txtaddr = p->vaddr;
178 a84cbb2a 2004-04-19 devnull 			fp->txtsz = p->memsz;
179 a84cbb2a 2004-04-19 devnull 			fp->txtoff = p->offset;
180 a84cbb2a 2004-04-19 devnull 		}
181 a84cbb2a 2004-04-19 devnull 		if(!havedata && p->flags == (ElfProgFlagRead|ElfProgFlagWrite) && p->align >= mach->pgsize){
182 a84cbb2a 2004-04-19 devnull 			havedata = 1;
183 a84cbb2a 2004-04-19 devnull 			fp->dataddr = p->vaddr;
184 a84cbb2a 2004-04-19 devnull 			fp->datsz = p->filesz;
185 a84cbb2a 2004-04-19 devnull 			fp->datoff = p->offset;
186 a84cbb2a 2004-04-19 devnull 			fp->bsssz = p->memsz - p->filesz;
187 a84cbb2a 2004-04-19 devnull 		}
188 a84cbb2a 2004-04-19 devnull 	}
189 a84cbb2a 2004-04-19 devnull 	if(!havetext){
190 a84cbb2a 2004-04-19 devnull 		werrstr("did not find text segment in elf binary");
191 a84cbb2a 2004-04-19 devnull 		goto err;
192 a84cbb2a 2004-04-19 devnull 	}
193 a84cbb2a 2004-04-19 devnull 	if(!havedata){
194 a84cbb2a 2004-04-19 devnull 		werrstr("did not find data segment in elf binary");
195 a84cbb2a 2004-04-19 devnull 		goto err;
196 a84cbb2a 2004-04-19 devnull 	}
197 a84cbb2a 2004-04-19 devnull 	return 0;
198 a84cbb2a 2004-04-19 devnull 
199 a84cbb2a 2004-04-19 devnull err:
200 a84cbb2a 2004-04-19 devnull 	elfclose(elf);
201 a84cbb2a 2004-04-19 devnull 	return -1;
202 a84cbb2a 2004-04-19 devnull }
203 a84cbb2a 2004-04-19 devnull 
204 a84cbb2a 2004-04-19 devnull static int
205 a84cbb2a 2004-04-19 devnull mapelf(Fhdr *fp, ulong base, Map *map, Regs **regs)
206 a84cbb2a 2004-04-19 devnull {
207 a84cbb2a 2004-04-19 devnull 	int i;
208 a84cbb2a 2004-04-19 devnull 	Elf *elf;
209 a84cbb2a 2004-04-19 devnull 	ElfProg *p;
210 a84cbb2a 2004-04-19 devnull 	ulong sz;
211 a84cbb2a 2004-04-19 devnull 	ulong lim;
212 a84cbb2a 2004-04-19 devnull 	Seg s;
213 a84cbb2a 2004-04-19 devnull 
214 a84cbb2a 2004-04-19 devnull 	elf = fp->elf;
215 a84cbb2a 2004-04-19 devnull 	if(elf == nil){
216 a84cbb2a 2004-04-19 devnull 		werrstr("not an elf file");
217 a84cbb2a 2004-04-19 devnull 		return -1;
218 a84cbb2a 2004-04-19 devnull 	}
219 a84cbb2a 2004-04-19 devnull 
220 a84cbb2a 2004-04-19 devnull 	for(i=0; i<elf->nprog; i++){
221 a84cbb2a 2004-04-19 devnull 		p = &elf->prog[i];
222 a84cbb2a 2004-04-19 devnull 		if(p->type != ElfProgLoad)
223 a84cbb2a 2004-04-19 devnull 			continue;
224 a84cbb2a 2004-04-19 devnull 		if(p->align < mach->pgsize)
225 a84cbb2a 2004-04-19 devnull 			continue;
226 a84cbb2a 2004-04-19 devnull 		if(p->filesz){
227 a84cbb2a 2004-04-19 devnull 			memset(&s, 0, sizeof s);
228 a84cbb2a 2004-04-19 devnull 			s.file = fp->filename;
229 a84cbb2a 2004-04-19 devnull 			s.fd = fp->fd;
230 a84cbb2a 2004-04-19 devnull 			if(fp->ftype == FCORE)
231 a84cbb2a 2004-04-19 devnull 				s.name = "core";
232 a84cbb2a 2004-04-19 devnull 			else if(p->flags == 5)
233 a84cbb2a 2004-04-19 devnull 				s.name = "text";
234 a84cbb2a 2004-04-19 devnull 			else
235 a84cbb2a 2004-04-19 devnull 				s.name = "data";
236 a84cbb2a 2004-04-19 devnull 			s.base = base+p->vaddr;
237 a84cbb2a 2004-04-19 devnull 			s.size = p->filesz;
238 a84cbb2a 2004-04-19 devnull 			s.offset = p->offset;
239 a84cbb2a 2004-04-19 devnull 			if(addseg(map, s) < 0)
240 a84cbb2a 2004-04-19 devnull 				return -1;
241 a84cbb2a 2004-04-19 devnull 		}
242 a84cbb2a 2004-04-19 devnull 		/*
243 a84cbb2a 2004-04-19 devnull 		 * If memsz > filesz, we're supposed to zero fill.
244 a84cbb2a 2004-04-19 devnull 		 * Core files have zeroed sections where the pages
245 a84cbb2a 2004-04-19 devnull 		 * can be filled in from the text file, so if this is a core
246 a84cbb2a 2004-04-19 devnull 		 * we only fill in that which isn't yet mapped.
247 a84cbb2a 2004-04-19 devnull 		 */
248 a84cbb2a 2004-04-19 devnull 		if(fp->ftype == FCORE){
249 a84cbb2a 2004-04-19 devnull 			sz = p->filesz;
250 a84cbb2a 2004-04-19 devnull 			while(sz < p->memsz){
251 a84cbb2a 2004-04-19 devnull 				if(addrtoseg(map, base+p->vaddr+sz, &s) < 0){
252 a84cbb2a 2004-04-19 devnull 					lim = base + p->vaddr + p->memsz;
253 a84cbb2a 2004-04-19 devnull 					if(addrtosegafter(map, base+p->vaddr+sz, &s) >= 0 && s.base < lim)
254 a84cbb2a 2004-04-19 devnull 						lim = s.base;
255 a84cbb2a 2004-04-19 devnull 					memset(&s, 0, sizeof s);
256 a84cbb2a 2004-04-19 devnull 					s.name = "zero";
257 a84cbb2a 2004-04-19 devnull 					s.base = base + p->vaddr + sz;
258 a84cbb2a 2004-04-19 devnull 					s.size = lim - s.base;
259 a84cbb2a 2004-04-19 devnull 					s.offset = p->offset;
260 a84cbb2a 2004-04-19 devnull 					if(addseg(map, s) < 0)
261 a84cbb2a 2004-04-19 devnull 						return -1;
262 a84cbb2a 2004-04-19 devnull 				}else
263 a84cbb2a 2004-04-19 devnull 					sz = (s.base+s.size) - (base + p->vaddr);
264 a84cbb2a 2004-04-19 devnull 			}
265 a84cbb2a 2004-04-19 devnull 		}else{
266 a84cbb2a 2004-04-19 devnull 			if(p->filesz < p->memsz){
267 a84cbb2a 2004-04-19 devnull 				memset(&s, 0, sizeof s);
268 a84cbb2a 2004-04-19 devnull 				s.name = "zero";
269 a84cbb2a 2004-04-19 devnull 				s.base = base + p->vaddr + p->filesz;
270 a84cbb2a 2004-04-19 devnull 				s.size = p->memsz - p->filesz;
271 a84cbb2a 2004-04-19 devnull 				if(addseg(map, s) < 0)
272 a84cbb2a 2004-04-19 devnull 					return -1;
273 a84cbb2a 2004-04-19 devnull 			}
274 a84cbb2a 2004-04-19 devnull 		}			
275 a84cbb2a 2004-04-19 devnull 	}
276 a84cbb2a 2004-04-19 devnull 
277 dd944ec7 2005-01-23 devnull 	if(fp->nthread && regs)
278 dd944ec7 2005-01-23 devnull 		*regs = coreregs(fp, fp->thread[0].id);
279 a84cbb2a 2004-04-19 devnull 
280 a84cbb2a 2004-04-19 devnull 	return 0;	
281 a84cbb2a 2004-04-19 devnull }
282 a84cbb2a 2004-04-19 devnull 
283 a84cbb2a 2004-04-19 devnull static int
284 a84cbb2a 2004-04-19 devnull unpacknote(Elf *elf, uchar *a, uchar *ea, ElfNote *note, uchar **pa)
285 a84cbb2a 2004-04-19 devnull {
286 a84cbb2a 2004-04-19 devnull 	if(a+12 > ea)
287 a84cbb2a 2004-04-19 devnull 		return -1;
288 a84cbb2a 2004-04-19 devnull 	note->namesz = elf->hdr.e4(a);
289 a84cbb2a 2004-04-19 devnull 	note->descsz = elf->hdr.e4(a+4);
290 a84cbb2a 2004-04-19 devnull 	note->type = elf->hdr.e4(a+8);
291 a84cbb2a 2004-04-19 devnull 	a += 12;
292 a84cbb2a 2004-04-19 devnull 	note->name = (char*)a;
293 a84cbb2a 2004-04-19 devnull /* XXX fetch alignment constants from elsewhere */
294 a84cbb2a 2004-04-19 devnull 	a += (note->namesz+3)&~3;
295 a84cbb2a 2004-04-19 devnull 	note->desc = (uchar*)a;
296 a84cbb2a 2004-04-19 devnull 	a += (note->descsz+3)&~3;
297 a84cbb2a 2004-04-19 devnull 	if(a > ea)
298 a84cbb2a 2004-04-19 devnull 		return -1;
299 a84cbb2a 2004-04-19 devnull 	*pa = a;
300 a84cbb2a 2004-04-19 devnull 	return 0;
301 a84cbb2a 2004-04-19 devnull }
302 a84cbb2a 2004-04-19 devnull