1 files changed, 3 insertions, 3 deletions

diff --git a/net/core/filter.c b/net/core/filter.c
index d814b8a89d0f..37f8eb06fdee 100644
--- a/net/core/filter.c
+++ b/net/core/filter.c
@@ -933,7 +933,7 @@ static struct bpf_prog *bpf_migrate_filter(struct bpf_prog *fp)
 
         /* Expand fp for appending the new filter representation. */
         old_fp = fp;
-        fp = krealloc(old_fp, bpf_prog_size(new_len), GFP_KERNEL);
+        fp = bpf_prog_realloc(old_fp, bpf_prog_size(new_len), 0);
         if (!fp) {
                 /* The old_fp is still around in case we couldn't
                  * allocate new memory, so uncharge on that one.
@@ -1013,7 +1013,7 @@ int bpf_prog_create(struct bpf_prog **pfp, struct sock_fprog_kern *fprog)
         if (fprog->filter == NULL)
                 return -EINVAL;
 
-        fp = kmalloc(bpf_prog_size(fprog->len), GFP_KERNEL);
+        fp = bpf_prog_alloc(bpf_prog_size(fprog->len), 0);
         if (!fp)
                 return -ENOMEM;
 
@@ -1069,7 +1069,7 @@ int sk_attach_filter(struct sock_fprog *fprog, struct sock *sk)
         if (fprog->filter == NULL)
                 return -EINVAL;
 
-        prog = kmalloc(bpf_fsize, GFP_KERNEL);
+        prog = bpf_prog_alloc(bpf_fsize, 0);
         if (!prog)
                 return -ENOMEM;