From: Artem Kuzin <[email protected]>

Co-developed-by: Nikita Panov <[email protected]>
Signed-off-by: Nikita Panov <[email protected]>
Co-developed-by: Alexander Grubnikov <[email protected]>
Signed-off-by: Alexander Grubnikov <[email protected]>
Signed-off-by: Artem Kuzin <[email protected]>
---
 kernel/bpf/bpf_struct_ops.c    | 8 ++++----
 kernel/bpf/core.c              | 4 ++--
 kernel/bpf/trampoline.c        | 6 +++---
 net/bpf/bpf_dummy_struct_ops.c | 2 +-
 4 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/kernel/bpf/bpf_struct_ops.c b/kernel/bpf/bpf_struct_ops.c
index 116a0ce378ec..9fb1dc5fbd5c 100644
--- a/kernel/bpf/bpf_struct_ops.c
+++ b/kernel/bpf/bpf_struct_ops.c
@@ -512,7 +512,7 @@ static long bpf_struct_ops_map_update_elem(struct bpf_map 
*map, void *key,
                err = st_ops->validate(kdata);
                if (err)
                        goto reset_unlock;
-               set_memory_rox((long)st_map->image, 1);
+               numa_set_memory_rox((long)st_map->image, 1);
                /* Let bpf_link handle registration & unregistration.
                 *
                 * Pair with smp_load_acquire() during lookup_elem().
@@ -521,7 +521,7 @@ static long bpf_struct_ops_map_update_elem(struct bpf_map 
*map, void *key,
                goto unlock;
        }
 
-       set_memory_rox((long)st_map->image, 1);
+       numa_set_memory_rox((long)st_map->image, 1);
        err = st_ops->reg(kdata);
        if (likely(!err)) {
                /* This refcnt increment on the map here after
@@ -544,8 +544,8 @@ static long bpf_struct_ops_map_update_elem(struct bpf_map 
*map, void *key,
         * there was a race in registering the struct_ops (under the same name) 
to
         * a sub-system through different struct_ops's maps.
         */
-       set_memory_nx((long)st_map->image, 1);
-       set_memory_rw((long)st_map->image, 1);
+       numa_set_memory_nx((long)st_map->image, 1);
+       numa_set_memory_rw((long)st_map->image, 1);
 
 reset_unlock:
        bpf_struct_ops_map_put_progs(st_map);
diff --git a/kernel/bpf/core.c b/kernel/bpf/core.c
index e3e45b651cd4..73ebda57c0f5 100644
--- a/kernel/bpf/core.c
+++ b/kernel/bpf/core.c
@@ -870,7 +870,7 @@ static struct bpf_prog_pack 
*alloc_new_pack(bpf_jit_fill_hole_t bpf_fill_ill_ins
        list_add_tail(&pack->list, &pack_list);
 
        set_vm_flush_reset_perms(pack->ptr);
-       set_memory_rox((unsigned long)pack->ptr, BPF_PROG_PACK_SIZE / 
PAGE_SIZE);
+       numa_set_memory_rox((unsigned long)pack->ptr, BPF_PROG_PACK_SIZE / 
PAGE_SIZE);
        return pack;
 }
 
@@ -888,7 +888,7 @@ void *bpf_prog_pack_alloc(u32 size, bpf_jit_fill_hole_t 
bpf_fill_ill_insns)
                if (ptr) {
                        bpf_fill_ill_insns(ptr, size);
                        set_vm_flush_reset_perms(ptr);
-                       set_memory_rox((unsigned long)ptr, size / PAGE_SIZE);
+                       numa_set_memory_rox((unsigned long)ptr, size / 
PAGE_SIZE);
                }
                goto out;
        }
diff --git a/kernel/bpf/trampoline.c b/kernel/bpf/trampoline.c
index 53ff50cac61e..964ae6128ef7 100644
--- a/kernel/bpf/trampoline.c
+++ b/kernel/bpf/trampoline.c
@@ -444,7 +444,7 @@ static int bpf_trampoline_update(struct bpf_trampoline *tr, 
bool lock_direct_mut
        if (err < 0)
                goto out_free;
 
-       set_memory_rox((long)im->image, 1);
+       numa_set_memory_rox((long)im->image, 1);
 
        WARN_ON(tr->cur_image && total == 0);
        if (tr->cur_image)
@@ -465,8 +465,8 @@ static int bpf_trampoline_update(struct bpf_trampoline *tr, 
bool lock_direct_mut
                tr->fops->trampoline = 0;
 
                /* reset im->image memory attr for arch_prepare_bpf_trampoline 
*/
-               set_memory_nx((long)im->image, 1);
-               set_memory_rw((long)im->image, 1);
+               numa_set_memory_nx((long)im->image, 1);
+               numa_set_memory_rw((long)im->image, 1);
                goto again;
        }
 #endif
diff --git a/net/bpf/bpf_dummy_struct_ops.c b/net/bpf/bpf_dummy_struct_ops.c
index 5918d1b32e19..45a5dbd379ac 100644
--- a/net/bpf/bpf_dummy_struct_ops.c
+++ b/net/bpf/bpf_dummy_struct_ops.c
@@ -124,7 +124,7 @@ int bpf_struct_ops_test_run(struct bpf_prog *prog, const 
union bpf_attr *kattr,
        if (err < 0)
                goto out;
 
-       set_memory_rox((long)image, 1);
+       numa_set_memory_rox((long)image, 1);
        prog_ret = dummy_ops_call_op(image, args);
 
        err = dummy_ops_copy_args(args);
-- 
2.34.1


Reply via email to