Loading...
1/*
2 * relocate_kernel.S - put the kernel image in place to boot
3 * Copyright (C) 2002-2004 Eric Biederman <ebiederm@xmission.com>
4 *
5 * This source code is licensed under the GNU General Public License,
6 * Version 2. See the file COPYING for more details.
7 */
8
9#include <linux/linkage.h>
10#include <asm/page_types.h>
11#include <asm/kexec.h>
12#include <asm/processor-flags.h>
13
14/*
15 * Must be relocatable PIC code callable as a C function
16 */
17
18#define PTR(x) (x << 2)
19
20/*
21 * control_page + KEXEC_CONTROL_CODE_MAX_SIZE
22 * ~ control_page + PAGE_SIZE are used as data storage and stack for
23 * jumping back
24 */
25#define DATA(offset) (KEXEC_CONTROL_CODE_MAX_SIZE+(offset))
26
27/* Minimal CPU state */
28#define ESP DATA(0x0)
29#define CR0 DATA(0x4)
30#define CR3 DATA(0x8)
31#define CR4 DATA(0xc)
32
33/* other data */
34#define CP_VA_CONTROL_PAGE DATA(0x10)
35#define CP_PA_PGD DATA(0x14)
36#define CP_PA_SWAP_PAGE DATA(0x18)
37#define CP_PA_BACKUP_PAGES_MAP DATA(0x1c)
38
39 .text
40 .globl relocate_kernel
41relocate_kernel:
42 /* Save the CPU context, used for jumping back */
43
44 pushl %ebx
45 pushl %esi
46 pushl %edi
47 pushl %ebp
48 pushf
49
50 movl 20+8(%esp), %ebp /* list of pages */
51 movl PTR(VA_CONTROL_PAGE)(%ebp), %edi
52 movl %esp, ESP(%edi)
53 movl %cr0, %eax
54 movl %eax, CR0(%edi)
55 movl %cr3, %eax
56 movl %eax, CR3(%edi)
57 movl %cr4, %eax
58 movl %eax, CR4(%edi)
59
60 /* read the arguments and say goodbye to the stack */
61 movl 20+4(%esp), %ebx /* page_list */
62 movl 20+8(%esp), %ebp /* list of pages */
63 movl 20+12(%esp), %edx /* start address */
64 movl 20+16(%esp), %ecx /* cpu_has_pae */
65 movl 20+20(%esp), %esi /* preserve_context */
66
67 /* zero out flags, and disable interrupts */
68 pushl $0
69 popfl
70
71 /* save some information for jumping back */
72 movl PTR(VA_CONTROL_PAGE)(%ebp), %edi
73 movl %edi, CP_VA_CONTROL_PAGE(%edi)
74 movl PTR(PA_PGD)(%ebp), %eax
75 movl %eax, CP_PA_PGD(%edi)
76 movl PTR(PA_SWAP_PAGE)(%ebp), %eax
77 movl %eax, CP_PA_SWAP_PAGE(%edi)
78 movl %ebx, CP_PA_BACKUP_PAGES_MAP(%edi)
79
80 /*
81 * get physical address of control page now
82 * this is impossible after page table switch
83 */
84 movl PTR(PA_CONTROL_PAGE)(%ebp), %edi
85
86 /* switch to new set of page tables */
87 movl PTR(PA_PGD)(%ebp), %eax
88 movl %eax, %cr3
89
90 /* setup a new stack at the end of the physical control page */
91 lea PAGE_SIZE(%edi), %esp
92
93 /* jump to identity mapped page */
94 movl %edi, %eax
95 addl $(identity_mapped - relocate_kernel), %eax
96 pushl %eax
97 ret
98
99identity_mapped:
100 /* set return address to 0 if not preserving context */
101 pushl $0
102 /* store the start address on the stack */
103 pushl %edx
104
105 /*
106 * Set cr0 to a known state:
107 * - Paging disabled
108 * - Alignment check disabled
109 * - Write protect disabled
110 * - No task switch
111 * - Don't do FP software emulation.
112 * - Proctected mode enabled
113 */
114 movl %cr0, %eax
115 andl $~(X86_CR0_PG | X86_CR0_AM | X86_CR0_WP | X86_CR0_TS | X86_CR0_EM), %eax
116 orl $(X86_CR0_PE), %eax
117 movl %eax, %cr0
118
119 /* clear cr4 if applicable */
120 testl %ecx, %ecx
121 jz 1f
122 /*
123 * Set cr4 to a known state:
124 * Setting everything to zero seems safe.
125 */
126 xorl %eax, %eax
127 movl %eax, %cr4
128
129 jmp 1f
1301:
131
132 /* Flush the TLB (needed?) */
133 xorl %eax, %eax
134 movl %eax, %cr3
135
136 movl CP_PA_SWAP_PAGE(%edi), %eax
137 pushl %eax
138 pushl %ebx
139 call swap_pages
140 addl $8, %esp
141
142 /*
143 * To be certain of avoiding problems with self-modifying code
144 * I need to execute a serializing instruction here.
145 * So I flush the TLB, it's handy, and not processor dependent.
146 */
147 xorl %eax, %eax
148 movl %eax, %cr3
149
150 /*
151 * set all of the registers to known values
152 * leave %esp alone
153 */
154
155 testl %esi, %esi
156 jnz 1f
157 xorl %edi, %edi
158 xorl %eax, %eax
159 xorl %ebx, %ebx
160 xorl %ecx, %ecx
161 xorl %edx, %edx
162 xorl %esi, %esi
163 xorl %ebp, %ebp
164 ret
1651:
166 popl %edx
167 movl CP_PA_SWAP_PAGE(%edi), %esp
168 addl $PAGE_SIZE, %esp
1692:
170 call *%edx
171
172 /* get the re-entry point of the peer system */
173 movl 0(%esp), %ebp
174 call 1f
1751:
176 popl %ebx
177 subl $(1b - relocate_kernel), %ebx
178 movl CP_VA_CONTROL_PAGE(%ebx), %edi
179 lea PAGE_SIZE(%ebx), %esp
180 movl CP_PA_SWAP_PAGE(%ebx), %eax
181 movl CP_PA_BACKUP_PAGES_MAP(%ebx), %edx
182 pushl %eax
183 pushl %edx
184 call swap_pages
185 addl $8, %esp
186 movl CP_PA_PGD(%ebx), %eax
187 movl %eax, %cr3
188 movl %cr0, %eax
189 orl $X86_CR0_PG, %eax
190 movl %eax, %cr0
191 lea PAGE_SIZE(%edi), %esp
192 movl %edi, %eax
193 addl $(virtual_mapped - relocate_kernel), %eax
194 pushl %eax
195 ret
196
197virtual_mapped:
198 movl CR4(%edi), %eax
199 movl %eax, %cr4
200 movl CR3(%edi), %eax
201 movl %eax, %cr3
202 movl CR0(%edi), %eax
203 movl %eax, %cr0
204 movl ESP(%edi), %esp
205 movl %ebp, %eax
206
207 popf
208 popl %ebp
209 popl %edi
210 popl %esi
211 popl %ebx
212 ret
213
214 /* Do the copies */
215swap_pages:
216 movl 8(%esp), %edx
217 movl 4(%esp), %ecx
218 pushl %ebp
219 pushl %ebx
220 pushl %edi
221 pushl %esi
222 movl %ecx, %ebx
223 jmp 1f
224
2250: /* top, read another word from the indirection page */
226 movl (%ebx), %ecx
227 addl $4, %ebx
2281:
229 testb $0x1, %cl /* is it a destination page */
230 jz 2f
231 movl %ecx, %edi
232 andl $0xfffff000, %edi
233 jmp 0b
2342:
235 testb $0x2, %cl /* is it an indirection page */
236 jz 2f
237 movl %ecx, %ebx
238 andl $0xfffff000, %ebx
239 jmp 0b
2402:
241 testb $0x4, %cl /* is it the done indicator */
242 jz 2f
243 jmp 3f
2442:
245 testb $0x8, %cl /* is it the source indicator */
246 jz 0b /* Ignore it otherwise */
247 movl %ecx, %esi /* For every source page do a copy */
248 andl $0xfffff000, %esi
249
250 movl %edi, %eax
251 movl %esi, %ebp
252
253 movl %edx, %edi
254 movl $1024, %ecx
255 rep ; movsl
256
257 movl %ebp, %edi
258 movl %eax, %esi
259 movl $1024, %ecx
260 rep ; movsl
261
262 movl %eax, %edi
263 movl %edx, %esi
264 movl $1024, %ecx
265 rep ; movsl
266
267 lea PAGE_SIZE(%ebp), %esi
268 jmp 0b
2693:
270 popl %esi
271 popl %edi
272 popl %ebx
273 popl %ebp
274 ret
275
276 .globl kexec_control_code_size
277.set kexec_control_code_size, . - relocate_kernel
1/* SPDX-License-Identifier: GPL-2.0-only */
2/*
3 * relocate_kernel.S - put the kernel image in place to boot
4 * Copyright (C) 2002-2004 Eric Biederman <ebiederm@xmission.com>
5 */
6
7#include <linux/linkage.h>
8#include <asm/page_types.h>
9#include <asm/kexec.h>
10#include <asm/nospec-branch.h>
11#include <asm/processor-flags.h>
12
13/*
14 * Must be relocatable PIC code callable as a C function, in particular
15 * there must be a plain RET and not jump to return thunk.
16 */
17
18#define PTR(x) (x << 2)
19
20/*
21 * control_page + KEXEC_CONTROL_CODE_MAX_SIZE
22 * ~ control_page + PAGE_SIZE are used as data storage and stack for
23 * jumping back
24 */
25#define DATA(offset) (KEXEC_CONTROL_CODE_MAX_SIZE+(offset))
26
27/* Minimal CPU state */
28#define ESP DATA(0x0)
29#define CR0 DATA(0x4)
30#define CR3 DATA(0x8)
31#define CR4 DATA(0xc)
32
33/* other data */
34#define CP_VA_CONTROL_PAGE DATA(0x10)
35#define CP_PA_PGD DATA(0x14)
36#define CP_PA_SWAP_PAGE DATA(0x18)
37#define CP_PA_BACKUP_PAGES_MAP DATA(0x1c)
38
39 .text
40SYM_CODE_START_NOALIGN(relocate_kernel)
41 /* Save the CPU context, used for jumping back */
42
43 pushl %ebx
44 pushl %esi
45 pushl %edi
46 pushl %ebp
47 pushf
48
49 movl 20+8(%esp), %ebp /* list of pages */
50 movl PTR(VA_CONTROL_PAGE)(%ebp), %edi
51 movl %esp, ESP(%edi)
52 movl %cr0, %eax
53 movl %eax, CR0(%edi)
54 movl %cr3, %eax
55 movl %eax, CR3(%edi)
56 movl %cr4, %eax
57 movl %eax, CR4(%edi)
58
59 /* read the arguments and say goodbye to the stack */
60 movl 20+4(%esp), %ebx /* page_list */
61 movl 20+8(%esp), %ebp /* list of pages */
62 movl 20+12(%esp), %edx /* start address */
63 movl 20+16(%esp), %ecx /* cpu_has_pae */
64 movl 20+20(%esp), %esi /* preserve_context */
65
66 /* zero out flags, and disable interrupts */
67 pushl $0
68 popfl
69
70 /* save some information for jumping back */
71 movl PTR(VA_CONTROL_PAGE)(%ebp), %edi
72 movl %edi, CP_VA_CONTROL_PAGE(%edi)
73 movl PTR(PA_PGD)(%ebp), %eax
74 movl %eax, CP_PA_PGD(%edi)
75 movl PTR(PA_SWAP_PAGE)(%ebp), %eax
76 movl %eax, CP_PA_SWAP_PAGE(%edi)
77 movl %ebx, CP_PA_BACKUP_PAGES_MAP(%edi)
78
79 /*
80 * get physical address of control page now
81 * this is impossible after page table switch
82 */
83 movl PTR(PA_CONTROL_PAGE)(%ebp), %edi
84
85 /* switch to new set of page tables */
86 movl PTR(PA_PGD)(%ebp), %eax
87 movl %eax, %cr3
88
89 /* setup a new stack at the end of the physical control page */
90 lea PAGE_SIZE(%edi), %esp
91
92 /* jump to identity mapped page */
93 movl %edi, %eax
94 addl $(identity_mapped - relocate_kernel), %eax
95 pushl %eax
96 ANNOTATE_UNRET_SAFE
97 ret
98 int3
99SYM_CODE_END(relocate_kernel)
100
101SYM_CODE_START_LOCAL_NOALIGN(identity_mapped)
102 /* set return address to 0 if not preserving context */
103 pushl $0
104 /* store the start address on the stack */
105 pushl %edx
106
107 /*
108 * Set cr0 to a known state:
109 * - Paging disabled
110 * - Alignment check disabled
111 * - Write protect disabled
112 * - No task switch
113 * - Don't do FP software emulation.
114 * - Protected mode enabled
115 */
116 movl %cr0, %eax
117 andl $~(X86_CR0_PG | X86_CR0_AM | X86_CR0_WP | X86_CR0_TS | X86_CR0_EM), %eax
118 orl $(X86_CR0_PE), %eax
119 movl %eax, %cr0
120
121 /* clear cr4 if applicable */
122 testl %ecx, %ecx
123 jz 1f
124 /*
125 * Set cr4 to a known state:
126 * Setting everything to zero seems safe.
127 */
128 xorl %eax, %eax
129 movl %eax, %cr4
130
131 jmp 1f
1321:
133
134 /* Flush the TLB (needed?) */
135 xorl %eax, %eax
136 movl %eax, %cr3
137
138 movl CP_PA_SWAP_PAGE(%edi), %eax
139 pushl %eax
140 pushl %ebx
141 call swap_pages
142 addl $8, %esp
143
144 /*
145 * To be certain of avoiding problems with self-modifying code
146 * I need to execute a serializing instruction here.
147 * So I flush the TLB, it's handy, and not processor dependent.
148 */
149 xorl %eax, %eax
150 movl %eax, %cr3
151
152 /*
153 * set all of the registers to known values
154 * leave %esp alone
155 */
156
157 testl %esi, %esi
158 jnz 1f
159 xorl %edi, %edi
160 xorl %eax, %eax
161 xorl %ebx, %ebx
162 xorl %ecx, %ecx
163 xorl %edx, %edx
164 xorl %esi, %esi
165 xorl %ebp, %ebp
166 ANNOTATE_UNRET_SAFE
167 ret
168 int3
1691:
170 popl %edx
171 movl CP_PA_SWAP_PAGE(%edi), %esp
172 addl $PAGE_SIZE, %esp
1732:
174 ANNOTATE_RETPOLINE_SAFE
175 call *%edx
176
177 /* get the re-entry point of the peer system */
178 movl 0(%esp), %ebp
179 call 1f
1801:
181 popl %ebx
182 subl $(1b - relocate_kernel), %ebx
183 movl CP_VA_CONTROL_PAGE(%ebx), %edi
184 lea PAGE_SIZE(%ebx), %esp
185 movl CP_PA_SWAP_PAGE(%ebx), %eax
186 movl CP_PA_BACKUP_PAGES_MAP(%ebx), %edx
187 pushl %eax
188 pushl %edx
189 call swap_pages
190 addl $8, %esp
191 movl CP_PA_PGD(%ebx), %eax
192 movl %eax, %cr3
193 movl %cr0, %eax
194 orl $X86_CR0_PG, %eax
195 movl %eax, %cr0
196 lea PAGE_SIZE(%edi), %esp
197 movl %edi, %eax
198 addl $(virtual_mapped - relocate_kernel), %eax
199 pushl %eax
200 ANNOTATE_UNRET_SAFE
201 ret
202 int3
203SYM_CODE_END(identity_mapped)
204
205SYM_CODE_START_LOCAL_NOALIGN(virtual_mapped)
206 movl CR4(%edi), %eax
207 movl %eax, %cr4
208 movl CR3(%edi), %eax
209 movl %eax, %cr3
210 movl CR0(%edi), %eax
211 movl %eax, %cr0
212 movl ESP(%edi), %esp
213 movl %ebp, %eax
214
215 popf
216 popl %ebp
217 popl %edi
218 popl %esi
219 popl %ebx
220 ANNOTATE_UNRET_SAFE
221 ret
222 int3
223SYM_CODE_END(virtual_mapped)
224
225 /* Do the copies */
226SYM_CODE_START_LOCAL_NOALIGN(swap_pages)
227 movl 8(%esp), %edx
228 movl 4(%esp), %ecx
229 pushl %ebp
230 pushl %ebx
231 pushl %edi
232 pushl %esi
233 movl %ecx, %ebx
234 jmp 1f
235
2360: /* top, read another word from the indirection page */
237 movl (%ebx), %ecx
238 addl $4, %ebx
2391:
240 testb $0x1, %cl /* is it a destination page */
241 jz 2f
242 movl %ecx, %edi
243 andl $0xfffff000, %edi
244 jmp 0b
2452:
246 testb $0x2, %cl /* is it an indirection page */
247 jz 2f
248 movl %ecx, %ebx
249 andl $0xfffff000, %ebx
250 jmp 0b
2512:
252 testb $0x4, %cl /* is it the done indicator */
253 jz 2f
254 jmp 3f
2552:
256 testb $0x8, %cl /* is it the source indicator */
257 jz 0b /* Ignore it otherwise */
258 movl %ecx, %esi /* For every source page do a copy */
259 andl $0xfffff000, %esi
260
261 movl %edi, %eax
262 movl %esi, %ebp
263
264 movl %edx, %edi
265 movl $1024, %ecx
266 rep ; movsl
267
268 movl %ebp, %edi
269 movl %eax, %esi
270 movl $1024, %ecx
271 rep ; movsl
272
273 movl %eax, %edi
274 movl %edx, %esi
275 movl $1024, %ecx
276 rep ; movsl
277
278 lea PAGE_SIZE(%ebp), %esi
279 jmp 0b
2803:
281 popl %esi
282 popl %edi
283 popl %ebx
284 popl %ebp
285 ANNOTATE_UNRET_SAFE
286 ret
287 int3
288SYM_CODE_END(swap_pages)
289
290 .globl kexec_control_code_size
291.set kexec_control_code_size, . - relocate_kernel