> fix vgpu-monitor panic due to cache file size incorrect

Signed-off-by: Kebe <[email protected]>
Project-HAMi · May 24, 2024 · 0cc987e · 0cc987e
1 parent 3c0c045
commit 0cc987e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cmd/vGPUmonitor/cudevshr.go b/cmd/vGPUmonitor/cudevshr.go
@@ -138,6 +138,10 @@ func mmapcachefile(filename string, nc *nvidiaCollector) error {
 		fmt.Println("openfile error=", err.Error())
 		return err
 	}
+	ss, _ := f.Stat()
+	if ss.Size() < int64(unsafe.Sizeof(*m)) {
+		return fmt.Errorf("cache file %s size %d is less than %d", filename, ss.Size(), unsafe.Sizeof(*m))
+	}
 	data, err := syscall.Mmap(int(f.Fd()), 0, int(unsafe.Sizeof(*m)), syscall.PROT_WRITE|syscall.PROT_READ, syscall.MAP_SHARED)
 	if err != nil {
 		return err