|
|
@@ -6593,8 +6593,10 @@ int tg_set_cfs_quota(struct task_group *tg, long cfs_quota_us)
|
|
|
period = ktime_to_ns(tg->cfs_bandwidth.period);
|
|
|
if (cfs_quota_us < 0)
|
|
|
quota = RUNTIME_INF;
|
|
|
- else
|
|
|
+ else if ((u64)cfs_quota_us <= U64_MAX / NSEC_PER_USEC)
|
|
|
quota = (u64)cfs_quota_us * NSEC_PER_USEC;
|
|
|
+ else
|
|
|
+ return -EINVAL;
|
|
|
|
|
|
return tg_set_cfs_bandwidth(tg, period, quota);
|
|
|
}
|
|
|
@@ -6616,6 +6618,9 @@ int tg_set_cfs_period(struct task_group *tg, long cfs_period_us)
|
|
|
{
|
|
|
u64 quota, period;
|
|
|
|
|
|
+ if ((u64)cfs_period_us > U64_MAX / NSEC_PER_USEC)
|
|
|
+ return -EINVAL;
|
|
|
+
|
|
|
period = (u64)cfs_period_us * NSEC_PER_USEC;
|
|
|
quota = tg->cfs_bandwidth.quota;
|
|
|
|