SWDEV-342013,SWDEV-342014 - Update validation on hipDeviceGet and hipDeviceGetName
Change-Id: I719da6169ce9f505449e07775cb94a5718f79ca3
[ROCm/clr commit: 3415f0a56e]
This commit is contained in:
@@ -121,9 +121,14 @@ void ihipDestroyDevice() {
|
||||
}
|
||||
|
||||
hipError_t ihipDeviceGet(hipDevice_t* device, int deviceId) {
|
||||
if (deviceId < 0 || static_cast<size_t>(deviceId) >= g_devices.size() || device == nullptr) {
|
||||
if (device == nullptr) {
|
||||
return hipErrorInvalidValue;
|
||||
}
|
||||
|
||||
if (deviceId < 0 || static_cast<size_t>(deviceId) >= g_devices.size()) {
|
||||
return hipErrorInvalidDevice;
|
||||
}
|
||||
|
||||
*device = deviceId;
|
||||
return hipSuccess;
|
||||
}
|
||||
@@ -213,7 +218,7 @@ hipError_t hipDeviceGetName(char *name, int len, hipDevice_t device) {
|
||||
|
||||
// Make sure that the size of `dest` is big enough to hold `src` including
|
||||
// trailing zero byte
|
||||
if (nameLen > (cl_uint)(len - 1)) {
|
||||
if (nameLen <= 0) {
|
||||
HIP_RETURN(hipErrorInvalidValue);
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user