Correctly call vfree

Author: 0x5459

kernel-open/common/inc/nv-linux.h

@@ -506,7 +506,7 @@ static inline void *nv_vmalloc(unsigned long size)
 static inline void nv_vfree(void *ptr, NvU64 size)
 {
     NV_MEMDBG_REMOVE(ptr, size);
-    vfree(ptr);
+    vfree(ptr, size);
 }
 
 static inline void *nv_ioremap(NvU64 phys, NvU64 size)

kernel-open/nvidia-modeset/nvidia-modeset-linux.c

@@ -260,7 +260,7 @@ void nvkms_free(void *ptr, size_t size)
     if (size <= KMALLOC_LIMIT) {
         kfree(ptr);
     } else {
-        vfree(ptr);
+        vfree(ptr, size);
     }
 }
 

kernel-open/nvidia-uvm/nv-kthread-q-selftest.c

@@ -241,7 +241,7 @@ static int _multithreaded_q_kthread_function(void *args)
 
 done:
     if (q_items)
-        vfree(q_items);
+        vfree(q_items, alloc_size);
 
     while (!kthread_should_stop())
         schedule();

kernel-open/nvidia-uvm/uvm_kvmalloc.c

@@ -297,16 +297,20 @@ void *__uvm_kvmalloc_zero(size_t size, const char *file, int line, const char *f
 
 void uvm_kvfree(void *p)
 {
+    uvm_vmalloc_hdr_t *hdr = NULL;
+
     if (!p)
         return;
 
     if (uvm_leak_checker)
         alloc_tracking_remove(p);
 
-    if (is_vmalloc_addr(p))
-        vfree(get_hdr(p));
-    else
+    if (is_vmalloc_addr(p)){
+        hdr = get_hdr(p);
+        vfree(hdr, hdr->alloc_size);
+    } else {
         kfree(p);
+    }
 }
 
 // Handle reallocs of kmalloc-based allocations
@@ -334,7 +338,7 @@ static void *realloc_from_vmalloc(void *p, size_t new_size)
     void *new_p;
 
     if (new_size == 0) {
-        vfree(old_hdr);
+        vfree(old_hdr, old_hdr->alloc_size);
         return ZERO_SIZE_PTR; // What krealloc returns for this case
     }
 
@@ -348,7 +352,7 @@ static void *realloc_from_vmalloc(void *p, size_t new_size)
         return NULL;
 
     memcpy(new_p, p, min(new_size, old_hdr->alloc_size));
-    vfree(old_hdr);
+    vfree(old_hdr, old_hdr->alloc_size);
     return new_p;
 }
 

kernel-open/nvidia/linux_nvswitch.c

@@ -711,7 +711,7 @@ _nvswitch_os_free
 
     if (is_vmalloc_addr(ptr))
     {
-        vfree(ptr);
+        vfree(ptr, -1ull);
     }
     else
     {