1
2#include <linux/ftrace.h>
3#include <linux/init.h>
4#include <linux/slab.h>
5#include <linux/mm_types.h>
6#include <linux/pgtable.h>
7
8#include <asm/bugs.h>
9#include <asm/cacheflush.h>
10#include <asm/idmap.h>
11#include <asm/memory.h>
12#include <asm/smp_plat.h>
13#include <asm/suspend.h>
14#include <asm/tlbflush.h>
15
16extern int __cpu_suspend(unsigned long, int (*)(unsigned long), u32 cpuid);
17extern void cpu_resume_mmu(void);
18
19#ifdef CONFIG_MMU
20int cpu_suspend(unsigned long arg, int (*fn)(unsigned long))
21{
22 struct mm_struct *mm = current->active_mm;
23 u32 __mpidr = cpu_logical_map(smp_processor_id());
24 int ret;
25
26 if (!idmap_pgd)
27 return -EINVAL;
28
29
30
31
32
33
34 pause_graph_tracing();
35
36
37
38
39
40
41
42 ret = __cpu_suspend(arg, fn, __mpidr);
43
44 unpause_graph_tracing();
45
46 if (ret == 0) {
47 cpu_switch_mm(mm->pgd, mm);
48 local_flush_bp_all();
49 local_flush_tlb_all();
50 check_other_bugs();
51 }
52
53 return ret;
54}
55#else
56int cpu_suspend(unsigned long arg, int (*fn)(unsigned long))
57{
58 u32 __mpidr = cpu_logical_map(smp_processor_id());
59 int ret;
60
61 pause_graph_tracing();
62 ret = __cpu_suspend(arg, fn, __mpidr);
63 unpause_graph_tracing();
64
65 return ret;
66}
67#define idmap_pgd NULL
68#endif
69
70
71
72
73
74
75void __cpu_suspend_save(u32 *ptr, u32 ptrsz, u32 sp, u32 *save_ptr)
76{
77 u32 *ctx = ptr;
78
79 *save_ptr = virt_to_phys(ptr);
80
81
82 *ptr++ = virt_to_phys(idmap_pgd);
83 *ptr++ = sp;
84 *ptr++ = virt_to_phys(cpu_do_resume);
85
86 cpu_do_suspend(ptr);
87
88 flush_cache_louis();
89
90
91
92
93
94
95
96
97
98
99 __cpuc_flush_dcache_area(ctx, ptrsz);
100 __cpuc_flush_dcache_area(save_ptr, sizeof(*save_ptr));
101
102 outer_clean_range(*save_ptr, *save_ptr + ptrsz);
103 outer_clean_range(virt_to_phys(save_ptr),
104 virt_to_phys(save_ptr) + sizeof(*save_ptr));
105}
106
107extern struct sleep_save_sp sleep_save_sp;
108
109static int cpu_suspend_alloc_sp(void)
110{
111 void *ctx_ptr;
112
113 ctx_ptr = kcalloc(mpidr_hash_size(), sizeof(u32), GFP_KERNEL);
114
115 if (WARN_ON(!ctx_ptr))
116 return -ENOMEM;
117 sleep_save_sp.save_ptr_stash = ctx_ptr;
118 sleep_save_sp.save_ptr_stash_phys = virt_to_phys(ctx_ptr);
119 sync_cache_w(&sleep_save_sp);
120 return 0;
121}
122early_initcall(cpu_suspend_alloc_sp);
123