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
|
#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 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;
}
}
void
rebootcmd(int argc, char *argv[])
{
struct {
Exec;
uvlong hdr[1];
} ehdr;
Chan *c;
ulong magic, text, rtext, entry, data, size, align;
uchar *p;
if(argc == 0)
exit(0);
c = namec(argv[0], Aopen, OEXEC, 0);
if(waserror()){
cclose(c);
nexterror();
}
readn(c, &ehdr, sizeof(Exec));
magic = beswal(ehdr.magic);
entry = beswal(ehdr.entry);
text = beswal(ehdr.text);
data = beswal(ehdr.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, ehdr.hdr, sizeof(ehdr.hdr));
switch(magic){
case R_MAGIC:
align = 0x10000; /* 64k segment alignment for arm64 */
break;
default:
align = BY2PG;
break;
}
/* round text out to page boundary */
rtext = ROUND(entry+text, align)-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);
reboot((void*)entry, p, size);
error(Egreg);
}
|
