include/linux/thread_info.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 /* thread_info.h: common low-level thread information accessors
   3  *
   4  * Copyright (C) 2002  David Howells (dhowells@redhat.com)
   5  * - Incorporating suggestions made by Linus Torvalds
   6  */
   7
   8 #ifndef _LINUX_THREAD_INFO_H
   9 #define _LINUX_THREAD_INFO_H
  10
  11 #include <linux/types.h>
  12 #include <linux/bug.h>
  13 #include <linux/restart_block.h>
  14
  15 #ifdef CONFIG_THREAD_INFO_IN_TASK
  16 /*
  17  * For CONFIG_THREAD_INFO_IN_TASK kernels we need <asm/current.h> for the
  18  * definition of current, but for !CONFIG_THREAD_INFO_IN_TASK kernels,
  19  * including <asm/current.h> can cause a circular dependency on some platforms.
  20  */
  21 #include <asm/current.h>
  22 #define current_thread_info() ((struct thread_info *)current)
  23 #endif
  24
  25 #include <linux/bitops.h>
  26
  27 /*
  28  * For per-arch arch_within_stack_frames() implementations, defined in
  29  * asm/thread_info.h.
  30  */
  31 enum {
  32         BAD_STACK = -1,
  33         NOT_STACK = 0,
  34         GOOD_FRAME,
  35         GOOD_STACK,
  36 };
  37
  38 #include <asm/thread_info.h>
  39
  40 #ifdef __KERNEL__
  41
  42 #ifndef THREAD_ALIGN
  43 #define THREAD_ALIGN    THREAD_SIZE
  44 #endif
  45
  46 #define THREADINFO_GFP          (GFP_KERNEL_ACCOUNT | __GFP_ZERO)
  47
  48 /*
  49  * flag set/clear/test wrappers
  50  * - pass TIF_xxxx constants to these functions
  51  */
  52
  53 static inline void set_ti_thread_flag(struct thread_info *ti, int flag)
  54 {
  55         set_bit(flag, (unsigned long *)&ti->flags);
  56 }
  57
  58 static inline void clear_ti_thread_flag(struct thread_info *ti, int flag)
  59 {
  60         clear_bit(flag, (unsigned long *)&ti->flags);
  61 }
  62
  63 static inline int test_and_set_ti_thread_flag(struct thread_info *ti, int flag)
  64 {
  65         return test_and_set_bit(flag, (unsigned long *)&ti->flags);
  66 }
  67
  68 static inline int test_and_clear_ti_thread_flag(struct thread_info *ti, int flag)
  69 {
  70         return test_and_clear_bit(flag, (unsigned long *)&ti->flags);
  71 }
  72
  73 static inline int test_ti_thread_flag(struct thread_info *ti, int flag)
  74 {
  75         return test_bit(flag, (unsigned long *)&ti->flags);
  76 }
  77
  78 #define set_thread_flag(flag) \
  79         set_ti_thread_flag(current_thread_info(), flag)
  80 #define clear_thread_flag(flag) \
  81         clear_ti_thread_flag(current_thread_info(), flag)
  82 #define test_and_set_thread_flag(flag) \
  83         test_and_set_ti_thread_flag(current_thread_info(), flag)
  84 #define test_and_clear_thread_flag(flag) \
  85         test_and_clear_ti_thread_flag(current_thread_info(), flag)
  86 #define test_thread_flag(flag) \
  87         test_ti_thread_flag(current_thread_info(), flag)
  88
  89 #define tif_need_resched() test_thread_flag(TIF_NEED_RESCHED)
  90
  91 #ifndef CONFIG_HAVE_ARCH_WITHIN_STACK_FRAMES
  92 static inline int arch_within_stack_frames(const void * const stack,
  93                                            const void * const stackend,
  94                                            const void *obj, unsigned long len)
  95 {
  96         return 0;
  97 }
  98 #endif
  99
 100 #ifdef CONFIG_HARDENED_USERCOPY
 101 extern void __check_object_size(const void *ptr, unsigned long n,
 102                                         bool to_user);
 103
 104 static __always_inline void check_object_size(const void *ptr, unsigned long n,
 105                                               bool to_user)
 106 {
 107         if (!__builtin_constant_p(n))
 108                 __check_object_size(ptr, n, to_user);
 109 }
 110 #else
 111 static inline void check_object_size(const void *ptr, unsigned long n,
 112                                      bool to_user)
 113 { }
 114 #endif /* CONFIG_HARDENED_USERCOPY */
 115
 116 extern void __compiletime_error("copy source size is too small")
 117 __bad_copy_from(void);
 118 extern void __compiletime_error("copy destination size is too small")
 119 __bad_copy_to(void);
 120
 121 static inline void copy_overflow(int size, unsigned long count)
 122 {
 123         WARN(1, "Buffer overflow detected (%d < %lu)!\n", size, count);
 124 }
 125
 126 static __always_inline bool
 127 check_copy_size(const void *addr, size_t bytes, bool is_source)
 128 {
 129         int sz = __compiletime_object_size(addr);
 130         if (unlikely(sz >= 0 && sz < bytes)) {
 131                 if (!__builtin_constant_p(bytes))
 132                         copy_overflow(sz, bytes);
 133                 else if (is_source)
 134                         __bad_copy_from();
 135                 else
 136                         __bad_copy_to();
 137                 return false;
 138         }
 139         check_object_size(addr, bytes, is_source);
 140         return true;
 141 }
 142
 143 #ifndef arch_setup_new_exec
 144 static inline void arch_setup_new_exec(void) { }
 145 #endif
 146
 147 #endif  /* __KERNEL__ */
 148
 149 #endif /* _LINUX_THREAD_INFO_H */