1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
#include "u.h"
#include "../port/lib.h"
#include "mem.h"
#include "dat.h"
#include "fns.h"
#include "../port/error.h"
#include "a.out.h"
static ulong
l2be(long l)
{
uchar *cp;
cp = (uchar*)&l;
return (cp[0]<<24) | (cp[1]<<16) | (cp[2]<<8) | cp[3];
}
static void
readn(Chan *c, void *vp, long n)
{
char *p = vp;
long nn;
while(n > 0) {
nn = devtab[c->type]->read(c, p, n, c->offset);
if(nn == 0)
error(Eshort);
c->offset += nn;
p += nn;
n -= nn;
}
}
static void
setbootcmd(int argc, char *argv[])
{
char *buf, *p, *ep;
int i;
p = buf = smalloc(1024);
ep = buf + 1024;
for(i=0; i<argc; i++)
p = seprint(p, ep, "%q ", argv[i]);
*p = 0;
ksetenv("bootcmd", buf, 1);
free(buf);
}
void
rebootcmd(int argc, char *argv[])
{
Chan *c;
Exec exec;
ulong magic, text, rtext, entry, data, size;
uchar *p;
if(argc == 0)
exit(0);
c = namec(argv[0], Aopen, OEXEC, 0);
if(waserror()){
cclose(c);
nexterror();
}
readn(c, &exec, sizeof(Exec));
magic = l2be(exec.magic);
entry = l2be(exec.entry);
text = l2be(exec.text);
data = l2be(exec.data);
if(!(magic == AOUT_MAGIC)){
switch(magic){
case I_MAGIC:
case S_MAGIC:
if((I_MAGIC == AOUT_MAGIC) || (S_MAGIC == AOUT_MAGIC))
break;
default:
error(Ebadexec);
}
}
if(magic & HDR_MAGIC)
readn(c, &exec, 8);
/* round text out to page boundary */
rtext = PGROUND(entry+text)-entry;
size = rtext + data;
p = malloc(size);
if(p == nil)
error(Enomem);
if(waserror()){
free(p);
nexterror();
}
memset(p, 0, size);
readn(c, p, text);
readn(c, p + rtext, data);
ksetenv("bootfile", argv[0], 1);
setbootcmd(argc-1, argv+1);
reboot((void*)entry, p, size);
error(Egreg);
}
|
