mm: reorder includes after introduction of linux/pgtable.h
[platform/kernel/linux-starfive.git] / arch / powerpc / mm / book3s64 / hash_native.c
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * native hashtable management.
4  *
5  * SMP scalability work:
6  *    Copyright (C) 2001 Anton Blanchard <anton@au.ibm.com>, IBM
7  */
8
9 #undef DEBUG_LOW
10
11 #include <linux/spinlock.h>
12 #include <linux/bitops.h>
13 #include <linux/of.h>
14 #include <linux/processor.h>
15 #include <linux/threads.h>
16 #include <linux/smp.h>
17 #include <linux/pgtable.h>
18
19 #include <asm/machdep.h>
20 #include <asm/mmu.h>
21 #include <asm/mmu_context.h>
22 #include <asm/trace.h>
23 #include <asm/tlb.h>
24 #include <asm/cputable.h>
25 #include <asm/udbg.h>
26 #include <asm/kexec.h>
27 #include <asm/ppc-opcode.h>
28 #include <asm/feature-fixups.h>
29
30 #include <misc/cxl-base.h>
31
32 #ifdef DEBUG_LOW
33 #define DBG_LOW(fmt...) udbg_printf(fmt)
34 #else
35 #define DBG_LOW(fmt...)
36 #endif
37
38 #ifdef __BIG_ENDIAN__
39 #define HPTE_LOCK_BIT 3
40 #else
41 #define HPTE_LOCK_BIT (56+3)
42 #endif
43
44 static DEFINE_RAW_SPINLOCK(native_tlbie_lock);
45
46 static inline void tlbiel_hash_set_isa206(unsigned int set, unsigned int is)
47 {
48         unsigned long rb;
49
50         rb = (set << PPC_BITLSHIFT(51)) | (is << PPC_BITLSHIFT(53));
51
52         asm volatile("tlbiel %0" : : "r" (rb));
53 }
54
55 /*
56  * tlbiel instruction for hash, set invalidation
57  * i.e., r=1 and is=01 or is=10 or is=11
58  */
59 static __always_inline void tlbiel_hash_set_isa300(unsigned int set, unsigned int is,
60                                         unsigned int pid,
61                                         unsigned int ric, unsigned int prs)
62 {
63         unsigned long rb;
64         unsigned long rs;
65         unsigned int r = 0; /* hash format */
66
67         rb = (set << PPC_BITLSHIFT(51)) | (is << PPC_BITLSHIFT(53));
68         rs = ((unsigned long)pid << PPC_BITLSHIFT(31));
69
70         asm volatile(PPC_TLBIEL(%0, %1, %2, %3, %4)
71                      : : "r"(rb), "r"(rs), "i"(ric), "i"(prs), "r"(r)
72                      : "memory");
73 }
74
75
76 static void tlbiel_all_isa206(unsigned int num_sets, unsigned int is)
77 {
78         unsigned int set;
79
80         asm volatile("ptesync": : :"memory");
81
82         for (set = 0; set < num_sets; set++)
83                 tlbiel_hash_set_isa206(set, is);
84
85         asm volatile("ptesync": : :"memory");
86 }
87
88 static void tlbiel_all_isa300(unsigned int num_sets, unsigned int is)
89 {
90         unsigned int set;
91
92         asm volatile("ptesync": : :"memory");
93
94         /*
95          * Flush the first set of the TLB, and any caching of partition table
96          * entries. Then flush the remaining sets of the TLB. Hash mode uses
97          * partition scoped TLB translations.
98          */
99         tlbiel_hash_set_isa300(0, is, 0, 2, 0);
100         for (set = 1; set < num_sets; set++)
101                 tlbiel_hash_set_isa300(set, is, 0, 0, 0);
102
103         /*
104          * Now invalidate the process table cache.
105          *
106          * From ISA v3.0B p. 1078:
107          *     The following forms are invalid.
108          *      * PRS=1, R=0, and RIC!=2 (The only process-scoped
109          *        HPT caching is of the Process Table.)
110          */
111         tlbiel_hash_set_isa300(0, is, 0, 2, 1);
112
113         asm volatile("ptesync": : :"memory");
114
115         asm volatile(PPC_ISA_3_0_INVALIDATE_ERAT "; isync" : : :"memory");
116 }
117
118 void hash__tlbiel_all(unsigned int action)
119 {
120         unsigned int is;
121
122         switch (action) {
123         case TLB_INVAL_SCOPE_GLOBAL:
124                 is = 3;
125                 break;
126         case TLB_INVAL_SCOPE_LPID:
127                 is = 2;
128                 break;
129         default:
130                 BUG();
131         }
132
133         if (early_cpu_has_feature(CPU_FTR_ARCH_300))
134                 tlbiel_all_isa300(POWER9_TLB_SETS_HASH, is);
135         else if (early_cpu_has_feature(CPU_FTR_ARCH_207S))
136                 tlbiel_all_isa206(POWER8_TLB_SETS, is);
137         else if (early_cpu_has_feature(CPU_FTR_ARCH_206))
138                 tlbiel_all_isa206(POWER7_TLB_SETS, is);
139         else
140                 WARN(1, "%s called on pre-POWER7 CPU\n", __func__);
141 }
142
143 static inline unsigned long  ___tlbie(unsigned long vpn, int psize,
144                                                 int apsize, int ssize)
145 {
146         unsigned long va;
147         unsigned int penc;
148         unsigned long sllp;
149
150         /*
151          * We need 14 to 65 bits of va for a tlibe of 4K page
152          * With vpn we ignore the lower VPN_SHIFT bits already.
153          * And top two bits are already ignored because we can
154          * only accomodate 76 bits in a 64 bit vpn with a VPN_SHIFT
155          * of 12.
156          */
157         va = vpn << VPN_SHIFT;
158         /*
159          * clear top 16 bits of 64bit va, non SLS segment
160          * Older versions of the architecture (2.02 and earler) require the
161          * masking of the top 16 bits.
162          */
163         if (mmu_has_feature(MMU_FTR_TLBIE_CROP_VA))
164                 va &= ~(0xffffULL << 48);
165
166         switch (psize) {
167         case MMU_PAGE_4K:
168                 /* clear out bits after (52) [0....52.....63] */
169                 va &= ~((1ul << (64 - 52)) - 1);
170                 va |= ssize << 8;
171                 sllp = get_sllp_encoding(apsize);
172                 va |= sllp << 5;
173                 asm volatile(ASM_FTR_IFCLR("tlbie %0,0", PPC_TLBIE(%1,%0), %2)
174                              : : "r" (va), "r"(0), "i" (CPU_FTR_ARCH_206)
175                              : "memory");
176                 break;
177         default:
178                 /* We need 14 to 14 + i bits of va */
179                 penc = mmu_psize_defs[psize].penc[apsize];
180                 va &= ~((1ul << mmu_psize_defs[apsize].shift) - 1);
181                 va |= penc << 12;
182                 va |= ssize << 8;
183                 /*
184                  * AVAL bits:
185                  * We don't need all the bits, but rest of the bits
186                  * must be ignored by the processor.
187                  * vpn cover upto 65 bits of va. (0...65) and we need
188                  * 58..64 bits of va.
189                  */
190                 va |= (vpn & 0xfe); /* AVAL */
191                 va |= 1; /* L */
192                 asm volatile(ASM_FTR_IFCLR("tlbie %0,1", PPC_TLBIE(%1,%0), %2)
193                              : : "r" (va), "r"(0), "i" (CPU_FTR_ARCH_206)
194                              : "memory");
195                 break;
196         }
197         return va;
198 }
199
200 static inline void fixup_tlbie_vpn(unsigned long vpn, int psize,
201                                    int apsize, int ssize)
202 {
203         if (cpu_has_feature(CPU_FTR_P9_TLBIE_ERAT_BUG)) {
204                 /* Radix flush for a hash guest */
205
206                 unsigned long rb,rs,prs,r,ric;
207
208                 rb = PPC_BIT(52); /* IS = 2 */
209                 rs = 0;  /* lpid = 0 */
210                 prs = 0; /* partition scoped */
211                 r = 1;   /* radix format */
212                 ric = 0; /* RIC_FLSUH_TLB */
213
214                 /*
215                  * Need the extra ptesync to make sure we don't
216                  * re-order the tlbie
217                  */
218                 asm volatile("ptesync": : :"memory");
219                 asm volatile(PPC_TLBIE_5(%0, %4, %3, %2, %1)
220                              : : "r"(rb), "i"(r), "i"(prs),
221                                "i"(ric), "r"(rs) : "memory");
222         }
223
224
225         if (cpu_has_feature(CPU_FTR_P9_TLBIE_STQ_BUG)) {
226                 /* Need the extra ptesync to ensure we don't reorder tlbie*/
227                 asm volatile("ptesync": : :"memory");
228                 ___tlbie(vpn, psize, apsize, ssize);
229         }
230 }
231
232 static inline void __tlbie(unsigned long vpn, int psize, int apsize, int ssize)
233 {
234         unsigned long rb;
235
236         rb = ___tlbie(vpn, psize, apsize, ssize);
237         trace_tlbie(0, 0, rb, 0, 0, 0, 0);
238 }
239
240 static inline void __tlbiel(unsigned long vpn, int psize, int apsize, int ssize)
241 {
242         unsigned long va;
243         unsigned int penc;
244         unsigned long sllp;
245
246         /* VPN_SHIFT can be atmost 12 */
247         va = vpn << VPN_SHIFT;
248         /*
249          * clear top 16 bits of 64 bit va, non SLS segment
250          * Older versions of the architecture (2.02 and earler) require the
251          * masking of the top 16 bits.
252          */
253         if (mmu_has_feature(MMU_FTR_TLBIE_CROP_VA))
254                 va &= ~(0xffffULL << 48);
255
256         switch (psize) {
257         case MMU_PAGE_4K:
258                 /* clear out bits after(52) [0....52.....63] */
259                 va &= ~((1ul << (64 - 52)) - 1);
260                 va |= ssize << 8;
261                 sllp = get_sllp_encoding(apsize);
262                 va |= sllp << 5;
263                 asm volatile(ASM_FTR_IFSET("tlbiel %0", "tlbiel %0,0", %1)
264                              : : "r" (va), "i" (CPU_FTR_ARCH_206)
265                              : "memory");
266                 break;
267         default:
268                 /* We need 14 to 14 + i bits of va */
269                 penc = mmu_psize_defs[psize].penc[apsize];
270                 va &= ~((1ul << mmu_psize_defs[apsize].shift) - 1);
271                 va |= penc << 12;
272                 va |= ssize << 8;
273                 /*
274                  * AVAL bits:
275                  * We don't need all the bits, but rest of the bits
276                  * must be ignored by the processor.
277                  * vpn cover upto 65 bits of va. (0...65) and we need
278                  * 58..64 bits of va.
279                  */
280                 va |= (vpn & 0xfe);
281                 va |= 1; /* L */
282                 asm volatile(ASM_FTR_IFSET("tlbiel %0", "tlbiel %0,1", %1)
283                              : : "r" (va), "i" (CPU_FTR_ARCH_206)
284                              : "memory");
285                 break;
286         }
287         trace_tlbie(0, 1, va, 0, 0, 0, 0);
288
289 }
290
291 static inline void tlbie(unsigned long vpn, int psize, int apsize,
292                          int ssize, int local)
293 {
294         unsigned int use_local;
295         int lock_tlbie = !mmu_has_feature(MMU_FTR_LOCKLESS_TLBIE);
296
297         use_local = local && mmu_has_feature(MMU_FTR_TLBIEL) && !cxl_ctx_in_use();
298
299         if (use_local)
300                 use_local = mmu_psize_defs[psize].tlbiel;
301         if (lock_tlbie && !use_local)
302                 raw_spin_lock(&native_tlbie_lock);
303         asm volatile("ptesync": : :"memory");
304         if (use_local) {
305                 __tlbiel(vpn, psize, apsize, ssize);
306                 asm volatile("ptesync": : :"memory");
307         } else {
308                 __tlbie(vpn, psize, apsize, ssize);
309                 fixup_tlbie_vpn(vpn, psize, apsize, ssize);
310                 asm volatile("eieio; tlbsync; ptesync": : :"memory");
311         }
312         if (lock_tlbie && !use_local)
313                 raw_spin_unlock(&native_tlbie_lock);
314 }
315
316 static inline void native_lock_hpte(struct hash_pte *hptep)
317 {
318         unsigned long *word = (unsigned long *)&hptep->v;
319
320         while (1) {
321                 if (!test_and_set_bit_lock(HPTE_LOCK_BIT, word))
322                         break;
323                 spin_begin();
324                 while(test_bit(HPTE_LOCK_BIT, word))
325                         spin_cpu_relax();
326                 spin_end();
327         }
328 }
329
330 static inline void native_unlock_hpte(struct hash_pte *hptep)
331 {
332         unsigned long *word = (unsigned long *)&hptep->v;
333
334         clear_bit_unlock(HPTE_LOCK_BIT, word);
335 }
336
337 static long native_hpte_insert(unsigned long hpte_group, unsigned long vpn,
338                         unsigned long pa, unsigned long rflags,
339                         unsigned long vflags, int psize, int apsize, int ssize)
340 {
341         struct hash_pte *hptep = htab_address + hpte_group;
342         unsigned long hpte_v, hpte_r;
343         int i;
344
345         if (!(vflags & HPTE_V_BOLTED)) {
346                 DBG_LOW("    insert(group=%lx, vpn=%016lx, pa=%016lx,"
347                         " rflags=%lx, vflags=%lx, psize=%d)\n",
348                         hpte_group, vpn, pa, rflags, vflags, psize);
349         }
350
351         for (i = 0; i < HPTES_PER_GROUP; i++) {
352                 if (! (be64_to_cpu(hptep->v) & HPTE_V_VALID)) {
353                         /* retry with lock held */
354                         native_lock_hpte(hptep);
355                         if (! (be64_to_cpu(hptep->v) & HPTE_V_VALID))
356                                 break;
357                         native_unlock_hpte(hptep);
358                 }
359
360                 hptep++;
361         }
362
363         if (i == HPTES_PER_GROUP)
364                 return -1;
365
366         hpte_v = hpte_encode_v(vpn, psize, apsize, ssize) | vflags | HPTE_V_VALID;
367         hpte_r = hpte_encode_r(pa, psize, apsize) | rflags;
368
369         if (!(vflags & HPTE_V_BOLTED)) {
370                 DBG_LOW(" i=%x hpte_v=%016lx, hpte_r=%016lx\n",
371                         i, hpte_v, hpte_r);
372         }
373
374         if (cpu_has_feature(CPU_FTR_ARCH_300)) {
375                 hpte_r = hpte_old_to_new_r(hpte_v, hpte_r);
376                 hpte_v = hpte_old_to_new_v(hpte_v);
377         }
378
379         hptep->r = cpu_to_be64(hpte_r);
380         /* Guarantee the second dword is visible before the valid bit */
381         eieio();
382         /*
383          * Now set the first dword including the valid bit
384          * NOTE: this also unlocks the hpte
385          */
386         hptep->v = cpu_to_be64(hpte_v);
387
388         __asm__ __volatile__ ("ptesync" : : : "memory");
389
390         return i | (!!(vflags & HPTE_V_SECONDARY) << 3);
391 }
392
393 static long native_hpte_remove(unsigned long hpte_group)
394 {
395         struct hash_pte *hptep;
396         int i;
397         int slot_offset;
398         unsigned long hpte_v;
399
400         DBG_LOW("    remove(group=%lx)\n", hpte_group);
401
402         /* pick a random entry to start at */
403         slot_offset = mftb() & 0x7;
404
405         for (i = 0; i < HPTES_PER_GROUP; i++) {
406                 hptep = htab_address + hpte_group + slot_offset;
407                 hpte_v = be64_to_cpu(hptep->v);
408
409                 if ((hpte_v & HPTE_V_VALID) && !(hpte_v & HPTE_V_BOLTED)) {
410                         /* retry with lock held */
411                         native_lock_hpte(hptep);
412                         hpte_v = be64_to_cpu(hptep->v);
413                         if ((hpte_v & HPTE_V_VALID)
414                             && !(hpte_v & HPTE_V_BOLTED))
415                                 break;
416                         native_unlock_hpte(hptep);
417                 }
418
419                 slot_offset++;
420                 slot_offset &= 0x7;
421         }
422
423         if (i == HPTES_PER_GROUP)
424                 return -1;
425
426         /* Invalidate the hpte. NOTE: this also unlocks it */
427         hptep->v = 0;
428
429         return i;
430 }
431
432 static long native_hpte_updatepp(unsigned long slot, unsigned long newpp,
433                                  unsigned long vpn, int bpsize,
434                                  int apsize, int ssize, unsigned long flags)
435 {
436         struct hash_pte *hptep = htab_address + slot;
437         unsigned long hpte_v, want_v;
438         int ret = 0, local = 0;
439
440         want_v = hpte_encode_avpn(vpn, bpsize, ssize);
441
442         DBG_LOW("    update(vpn=%016lx, avpnv=%016lx, group=%lx, newpp=%lx)",
443                 vpn, want_v & HPTE_V_AVPN, slot, newpp);
444
445         hpte_v = hpte_get_old_v(hptep);
446         /*
447          * We need to invalidate the TLB always because hpte_remove doesn't do
448          * a tlb invalidate. If a hash bucket gets full, we "evict" a more/less
449          * random entry from it. When we do that we don't invalidate the TLB
450          * (hpte_remove) because we assume the old translation is still
451          * technically "valid".
452          */
453         if (!HPTE_V_COMPARE(hpte_v, want_v) || !(hpte_v & HPTE_V_VALID)) {
454                 DBG_LOW(" -> miss\n");
455                 ret = -1;
456         } else {
457                 native_lock_hpte(hptep);
458                 /* recheck with locks held */
459                 hpte_v = hpte_get_old_v(hptep);
460                 if (unlikely(!HPTE_V_COMPARE(hpte_v, want_v) ||
461                              !(hpte_v & HPTE_V_VALID))) {
462                         ret = -1;
463                 } else {
464                         DBG_LOW(" -> hit\n");
465                         /* Update the HPTE */
466                         hptep->r = cpu_to_be64((be64_to_cpu(hptep->r) &
467                                                 ~(HPTE_R_PPP | HPTE_R_N)) |
468                                                (newpp & (HPTE_R_PPP | HPTE_R_N |
469                                                          HPTE_R_C)));
470                 }
471                 native_unlock_hpte(hptep);
472         }
473
474         if (flags & HPTE_LOCAL_UPDATE)
475                 local = 1;
476         /*
477          * Ensure it is out of the tlb too if it is not a nohpte fault
478          */
479         if (!(flags & HPTE_NOHPTE_UPDATE))
480                 tlbie(vpn, bpsize, apsize, ssize, local);
481
482         return ret;
483 }
484
485 static long __native_hpte_find(unsigned long want_v, unsigned long slot)
486 {
487         struct hash_pte *hptep;
488         unsigned long hpte_v;
489         unsigned long i;
490
491         for (i = 0; i < HPTES_PER_GROUP; i++) {
492
493                 hptep = htab_address + slot;
494                 hpte_v = hpte_get_old_v(hptep);
495                 if (HPTE_V_COMPARE(hpte_v, want_v) && (hpte_v & HPTE_V_VALID))
496                         /* HPTE matches */
497                         return slot;
498                 ++slot;
499         }
500
501         return -1;
502 }
503
504 static long native_hpte_find(unsigned long vpn, int psize, int ssize)
505 {
506         unsigned long hpte_group;
507         unsigned long want_v;
508         unsigned long hash;
509         long slot;
510
511         hash = hpt_hash(vpn, mmu_psize_defs[psize].shift, ssize);
512         want_v = hpte_encode_avpn(vpn, psize, ssize);
513
514         /*
515          * We try to keep bolted entries always in primary hash
516          * But in some case we can find them in secondary too.
517          */
518         hpte_group = (hash & htab_hash_mask) * HPTES_PER_GROUP;
519         slot = __native_hpte_find(want_v, hpte_group);
520         if (slot < 0) {
521                 /* Try in secondary */
522                 hpte_group = (~hash & htab_hash_mask) * HPTES_PER_GROUP;
523                 slot = __native_hpte_find(want_v, hpte_group);
524                 if (slot < 0)
525                         return -1;
526         }
527
528         return slot;
529 }
530
531 /*
532  * Update the page protection bits. Intended to be used to create
533  * guard pages for kernel data structures on pages which are bolted
534  * in the HPT. Assumes pages being operated on will not be stolen.
535  *
536  * No need to lock here because we should be the only user.
537  */
538 static void native_hpte_updateboltedpp(unsigned long newpp, unsigned long ea,
539                                        int psize, int ssize)
540 {
541         unsigned long vpn;
542         unsigned long vsid;
543         long slot;
544         struct hash_pte *hptep;
545
546         vsid = get_kernel_vsid(ea, ssize);
547         vpn = hpt_vpn(ea, vsid, ssize);
548
549         slot = native_hpte_find(vpn, psize, ssize);
550         if (slot == -1)
551                 panic("could not find page to bolt\n");
552         hptep = htab_address + slot;
553
554         /* Update the HPTE */
555         hptep->r = cpu_to_be64((be64_to_cpu(hptep->r) &
556                                 ~(HPTE_R_PPP | HPTE_R_N)) |
557                                (newpp & (HPTE_R_PPP | HPTE_R_N)));
558         /*
559          * Ensure it is out of the tlb too. Bolted entries base and
560          * actual page size will be same.
561          */
562         tlbie(vpn, psize, psize, ssize, 0);
563 }
564
565 /*
566  * Remove a bolted kernel entry. Memory hotplug uses this.
567  *
568  * No need to lock here because we should be the only user.
569  */
570 static int native_hpte_removebolted(unsigned long ea, int psize, int ssize)
571 {
572         unsigned long vpn;
573         unsigned long vsid;
574         long slot;
575         struct hash_pte *hptep;
576
577         vsid = get_kernel_vsid(ea, ssize);
578         vpn = hpt_vpn(ea, vsid, ssize);
579
580         slot = native_hpte_find(vpn, psize, ssize);
581         if (slot == -1)
582                 return -ENOENT;
583
584         hptep = htab_address + slot;
585
586         VM_WARN_ON(!(be64_to_cpu(hptep->v) & HPTE_V_BOLTED));
587
588         /* Invalidate the hpte */
589         hptep->v = 0;
590
591         /* Invalidate the TLB */
592         tlbie(vpn, psize, psize, ssize, 0);
593         return 0;
594 }
595
596
597 static void native_hpte_invalidate(unsigned long slot, unsigned long vpn,
598                                    int bpsize, int apsize, int ssize, int local)
599 {
600         struct hash_pte *hptep = htab_address + slot;
601         unsigned long hpte_v;
602         unsigned long want_v;
603         unsigned long flags;
604
605         local_irq_save(flags);
606
607         DBG_LOW("    invalidate(vpn=%016lx, hash: %lx)\n", vpn, slot);
608
609         want_v = hpte_encode_avpn(vpn, bpsize, ssize);
610         hpte_v = hpte_get_old_v(hptep);
611
612         if (HPTE_V_COMPARE(hpte_v, want_v) && (hpte_v & HPTE_V_VALID)) {
613                 native_lock_hpte(hptep);
614                 /* recheck with locks held */
615                 hpte_v = hpte_get_old_v(hptep);
616
617                 if (HPTE_V_COMPARE(hpte_v, want_v) && (hpte_v & HPTE_V_VALID))
618                         /* Invalidate the hpte. NOTE: this also unlocks it */
619                         hptep->v = 0;
620                 else
621                         native_unlock_hpte(hptep);
622         }
623         /*
624          * We need to invalidate the TLB always because hpte_remove doesn't do
625          * a tlb invalidate. If a hash bucket gets full, we "evict" a more/less
626          * random entry from it. When we do that we don't invalidate the TLB
627          * (hpte_remove) because we assume the old translation is still
628          * technically "valid".
629          */
630         tlbie(vpn, bpsize, apsize, ssize, local);
631
632         local_irq_restore(flags);
633 }
634
635 #ifdef CONFIG_TRANSPARENT_HUGEPAGE
636 static void native_hugepage_invalidate(unsigned long vsid,
637                                        unsigned long addr,
638                                        unsigned char *hpte_slot_array,
639                                        int psize, int ssize, int local)
640 {
641         int i;
642         struct hash_pte *hptep;
643         int actual_psize = MMU_PAGE_16M;
644         unsigned int max_hpte_count, valid;
645         unsigned long flags, s_addr = addr;
646         unsigned long hpte_v, want_v, shift;
647         unsigned long hidx, vpn = 0, hash, slot;
648
649         shift = mmu_psize_defs[psize].shift;
650         max_hpte_count = 1U << (PMD_SHIFT - shift);
651
652         local_irq_save(flags);
653         for (i = 0; i < max_hpte_count; i++) {
654                 valid = hpte_valid(hpte_slot_array, i);
655                 if (!valid)
656                         continue;
657                 hidx =  hpte_hash_index(hpte_slot_array, i);
658
659                 /* get the vpn */
660                 addr = s_addr + (i * (1ul << shift));
661                 vpn = hpt_vpn(addr, vsid, ssize);
662                 hash = hpt_hash(vpn, shift, ssize);
663                 if (hidx & _PTEIDX_SECONDARY)
664                         hash = ~hash;
665
666                 slot = (hash & htab_hash_mask) * HPTES_PER_GROUP;
667                 slot += hidx & _PTEIDX_GROUP_IX;
668
669                 hptep = htab_address + slot;
670                 want_v = hpte_encode_avpn(vpn, psize, ssize);
671                 hpte_v = hpte_get_old_v(hptep);
672
673                 /* Even if we miss, we need to invalidate the TLB */
674                 if (HPTE_V_COMPARE(hpte_v, want_v) && (hpte_v & HPTE_V_VALID)) {
675                         /* recheck with locks held */
676                         native_lock_hpte(hptep);
677                         hpte_v = hpte_get_old_v(hptep);
678
679                         if (HPTE_V_COMPARE(hpte_v, want_v) && (hpte_v & HPTE_V_VALID)) {
680                                 /*
681                                  * Invalidate the hpte. NOTE: this also unlocks it
682                                  */
683
684                                 hptep->v = 0;
685                         } else
686                                 native_unlock_hpte(hptep);
687                 }
688                 /*
689                  * We need to do tlb invalidate for all the address, tlbie
690                  * instruction compares entry_VA in tlb with the VA specified
691                  * here
692                  */
693                 tlbie(vpn, psize, actual_psize, ssize, local);
694         }
695         local_irq_restore(flags);
696 }
697 #else
698 static void native_hugepage_invalidate(unsigned long vsid,
699                                        unsigned long addr,
700                                        unsigned char *hpte_slot_array,
701                                        int psize, int ssize, int local)
702 {
703         WARN(1, "%s called without THP support\n", __func__);
704 }
705 #endif
706
707 static void hpte_decode(struct hash_pte *hpte, unsigned long slot,
708                         int *psize, int *apsize, int *ssize, unsigned long *vpn)
709 {
710         unsigned long avpn, pteg, vpi;
711         unsigned long hpte_v = be64_to_cpu(hpte->v);
712         unsigned long hpte_r = be64_to_cpu(hpte->r);
713         unsigned long vsid, seg_off;
714         int size, a_size, shift;
715         /* Look at the 8 bit LP value */
716         unsigned int lp = (hpte_r >> LP_SHIFT) & ((1 << LP_BITS) - 1);
717
718         if (cpu_has_feature(CPU_FTR_ARCH_300)) {
719                 hpte_v = hpte_new_to_old_v(hpte_v, hpte_r);
720                 hpte_r = hpte_new_to_old_r(hpte_r);
721         }
722         if (!(hpte_v & HPTE_V_LARGE)) {
723                 size   = MMU_PAGE_4K;
724                 a_size = MMU_PAGE_4K;
725         } else {
726                 size = hpte_page_sizes[lp] & 0xf;
727                 a_size = hpte_page_sizes[lp] >> 4;
728         }
729         /* This works for all page sizes, and for 256M and 1T segments */
730         *ssize = hpte_v >> HPTE_V_SSIZE_SHIFT;
731         shift = mmu_psize_defs[size].shift;
732
733         avpn = (HPTE_V_AVPN_VAL(hpte_v) & ~mmu_psize_defs[size].avpnm);
734         pteg = slot / HPTES_PER_GROUP;
735         if (hpte_v & HPTE_V_SECONDARY)
736                 pteg = ~pteg;
737
738         switch (*ssize) {
739         case MMU_SEGSIZE_256M:
740                 /* We only have 28 - 23 bits of seg_off in avpn */
741                 seg_off = (avpn & 0x1f) << 23;
742                 vsid    =  avpn >> 5;
743                 /* We can find more bits from the pteg value */
744                 if (shift < 23) {
745                         vpi = (vsid ^ pteg) & htab_hash_mask;
746                         seg_off |= vpi << shift;
747                 }
748                 *vpn = vsid << (SID_SHIFT - VPN_SHIFT) | seg_off >> VPN_SHIFT;
749                 break;
750         case MMU_SEGSIZE_1T:
751                 /* We only have 40 - 23 bits of seg_off in avpn */
752                 seg_off = (avpn & 0x1ffff) << 23;
753                 vsid    = avpn >> 17;
754                 if (shift < 23) {
755                         vpi = (vsid ^ (vsid << 25) ^ pteg) & htab_hash_mask;
756                         seg_off |= vpi << shift;
757                 }
758                 *vpn = vsid << (SID_SHIFT_1T - VPN_SHIFT) | seg_off >> VPN_SHIFT;
759                 break;
760         default:
761                 *vpn = size = 0;
762         }
763         *psize  = size;
764         *apsize = a_size;
765 }
766
767 /*
768  * clear all mappings on kexec.  All cpus are in real mode (or they will
769  * be when they isi), and we are the only one left.  We rely on our kernel
770  * mapping being 0xC0's and the hardware ignoring those two real bits.
771  *
772  * This must be called with interrupts disabled.
773  *
774  * Taking the native_tlbie_lock is unsafe here due to the possibility of
775  * lockdep being on. On pre POWER5 hardware, not taking the lock could
776  * cause deadlock. POWER5 and newer not taking the lock is fine. This only
777  * gets called during boot before secondary CPUs have come up and during
778  * crashdump and all bets are off anyway.
779  *
780  * TODO: add batching support when enabled.  remember, no dynamic memory here,
781  * although there is the control page available...
782  */
783 static void native_hpte_clear(void)
784 {
785         unsigned long vpn = 0;
786         unsigned long slot, slots;
787         struct hash_pte *hptep = htab_address;
788         unsigned long hpte_v;
789         unsigned long pteg_count;
790         int psize, apsize, ssize;
791
792         pteg_count = htab_hash_mask + 1;
793
794         slots = pteg_count * HPTES_PER_GROUP;
795
796         for (slot = 0; slot < slots; slot++, hptep++) {
797                 /*
798                  * we could lock the pte here, but we are the only cpu
799                  * running,  right?  and for crash dump, we probably
800                  * don't want to wait for a maybe bad cpu.
801                  */
802                 hpte_v = be64_to_cpu(hptep->v);
803
804                 /*
805                  * Call __tlbie() here rather than tlbie() since we can't take the
806                  * native_tlbie_lock.
807                  */
808                 if (hpte_v & HPTE_V_VALID) {
809                         hpte_decode(hptep, slot, &psize, &apsize, &ssize, &vpn);
810                         hptep->v = 0;
811                         ___tlbie(vpn, psize, apsize, ssize);
812                 }
813         }
814
815         asm volatile("eieio; tlbsync; ptesync":::"memory");
816 }
817
818 /*
819  * Batched hash table flush, we batch the tlbie's to avoid taking/releasing
820  * the lock all the time
821  */
822 static void native_flush_hash_range(unsigned long number, int local)
823 {
824         unsigned long vpn = 0;
825         unsigned long hash, index, hidx, shift, slot;
826         struct hash_pte *hptep;
827         unsigned long hpte_v;
828         unsigned long want_v;
829         unsigned long flags;
830         real_pte_t pte;
831         struct ppc64_tlb_batch *batch = this_cpu_ptr(&ppc64_tlb_batch);
832         unsigned long psize = batch->psize;
833         int ssize = batch->ssize;
834         int i;
835         unsigned int use_local;
836
837         use_local = local && mmu_has_feature(MMU_FTR_TLBIEL) &&
838                 mmu_psize_defs[psize].tlbiel && !cxl_ctx_in_use();
839
840         local_irq_save(flags);
841
842         for (i = 0; i < number; i++) {
843                 vpn = batch->vpn[i];
844                 pte = batch->pte[i];
845
846                 pte_iterate_hashed_subpages(pte, psize, vpn, index, shift) {
847                         hash = hpt_hash(vpn, shift, ssize);
848                         hidx = __rpte_to_hidx(pte, index);
849                         if (hidx & _PTEIDX_SECONDARY)
850                                 hash = ~hash;
851                         slot = (hash & htab_hash_mask) * HPTES_PER_GROUP;
852                         slot += hidx & _PTEIDX_GROUP_IX;
853                         hptep = htab_address + slot;
854                         want_v = hpte_encode_avpn(vpn, psize, ssize);
855                         hpte_v = hpte_get_old_v(hptep);
856
857                         if (!HPTE_V_COMPARE(hpte_v, want_v) || !(hpte_v & HPTE_V_VALID))
858                                 continue;
859                         /* lock and try again */
860                         native_lock_hpte(hptep);
861                         hpte_v = hpte_get_old_v(hptep);
862
863                         if (!HPTE_V_COMPARE(hpte_v, want_v) || !(hpte_v & HPTE_V_VALID))
864                                 native_unlock_hpte(hptep);
865                         else
866                                 hptep->v = 0;
867
868                 } pte_iterate_hashed_end();
869         }
870
871         if (use_local) {
872                 asm volatile("ptesync":::"memory");
873                 for (i = 0; i < number; i++) {
874                         vpn = batch->vpn[i];
875                         pte = batch->pte[i];
876
877                         pte_iterate_hashed_subpages(pte, psize,
878                                                     vpn, index, shift) {
879                                 __tlbiel(vpn, psize, psize, ssize);
880                         } pte_iterate_hashed_end();
881                 }
882                 asm volatile("ptesync":::"memory");
883         } else {
884                 int lock_tlbie = !mmu_has_feature(MMU_FTR_LOCKLESS_TLBIE);
885
886                 if (lock_tlbie)
887                         raw_spin_lock(&native_tlbie_lock);
888
889                 asm volatile("ptesync":::"memory");
890                 for (i = 0; i < number; i++) {
891                         vpn = batch->vpn[i];
892                         pte = batch->pte[i];
893
894                         pte_iterate_hashed_subpages(pte, psize,
895                                                     vpn, index, shift) {
896                                 __tlbie(vpn, psize, psize, ssize);
897                         } pte_iterate_hashed_end();
898                 }
899                 /*
900                  * Just do one more with the last used values.
901                  */
902                 fixup_tlbie_vpn(vpn, psize, psize, ssize);
903                 asm volatile("eieio; tlbsync; ptesync":::"memory");
904
905                 if (lock_tlbie)
906                         raw_spin_unlock(&native_tlbie_lock);
907         }
908
909         local_irq_restore(flags);
910 }
911
912 void __init hpte_init_native(void)
913 {
914         mmu_hash_ops.hpte_invalidate    = native_hpte_invalidate;
915         mmu_hash_ops.hpte_updatepp      = native_hpte_updatepp;
916         mmu_hash_ops.hpte_updateboltedpp = native_hpte_updateboltedpp;
917         mmu_hash_ops.hpte_removebolted = native_hpte_removebolted;
918         mmu_hash_ops.hpte_insert        = native_hpte_insert;
919         mmu_hash_ops.hpte_remove        = native_hpte_remove;
920         mmu_hash_ops.hpte_clear_all     = native_hpte_clear;
921         mmu_hash_ops.flush_hash_range = native_flush_hash_range;
922         mmu_hash_ops.hugepage_invalidate   = native_hugepage_invalidate;
923 }