Avoid address overflow (HostImage)

fixstars · Aug 10, 2023 · bd8966b · bd8966b
1 parent 0a1cd3e
commit bd8966b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/host_image.h b/test/host_image.h
@@ -94,8 +94,8 @@ class HostImage
 		memcpy(rhs.data, data, elemSize(type) * rows * step);
 	}
 
-	template <typename T> T* ptr(int y = 0) { return (T*)data + y * step; }
-	template <typename T> const T* ptr(int y = 0) const { return (T*)data + y * step; }
+	template <typename T> T* ptr(int y = 0) { return (T*)data + y * (size_t)step; }
+	template <typename T> const T* ptr(int y = 0) const { return (T*)data + y * (size_t)step; }
 
 	void* data;
 	int rows, cols, step;