On Thu, 31 Aug 2023, Reinette Chatre wrote: > Hi Ilpo, > > On 8/31/2023 4:08 AM, Ilpo Järvinen wrote: > > Benchmark command is copied into an array in the stack. The array is > > BENCHMARK_ARGS items long but the command line could try to provide a > > longer command. Argument size is also fixed by BENCHMARK_ARG_SIZE (63 > > bytes of space after fitting the terminating \0 character) and user > > could have inputted argument longer than that. > > > > Return error in case the benchmark command does not fit to the space > > allocated for it. > > > > Fixes: ecdbb911f22d ("selftests/resctrl: Add MBM test") > > Signed-off-by: Ilpo Järvinen <ilpo.jarvinen@xxxxxxxxxxxxxxx> > > Tested-by: Shaopeng Tan <tan.shaopeng@xxxxxxxxxxxxxx> > > Reviewed-by: Shaopeng Tan <tan.shaopeng@xxxxxxxxxxxxxx> > > --- > > tools/testing/selftests/resctrl/resctrl_tests.c | 5 +++++ > > 1 file changed, 5 insertions(+) > > > > diff --git a/tools/testing/selftests/resctrl/resctrl_tests.c b/tools/testing/selftests/resctrl/resctrl_tests.c > > index d511daeb6851..a9331b31c32d 100644 > > --- a/tools/testing/selftests/resctrl/resctrl_tests.c > > +++ b/tools/testing/selftests/resctrl/resctrl_tests.c > > @@ -255,9 +255,14 @@ int main(int argc, char **argv) > > return ksft_exit_skip("Not running as root. Skipping...\n"); > > > > if (has_ben) { > > + if (argc - ben_ind >= BENCHMARK_ARGS) > > + ksft_exit_fail_msg("Too long benchmark command.\n"); > > + > > /* Extract benchmark command from command line. */ > > for (i = ben_ind; i < argc; i++) { > > benchmark_cmd[i - ben_ind] = benchmark_cmd_area[i]; > > + if (strlen(argv[i]) >= BENCHMARK_ARG_SIZE - 1) > > Should this perhaps be: > if (strlen(argv[i]) >= BENCHMARK_ARG_SIZE) > > As you note in the longest string that can be fitted should be 63 to account for > the \0. If I understand correctly comparing with "BENCHMARK_ARG_SIZE - 1" would > would consider a 63 byte string as invalid. Of course, I don't know why I added that - 1 there. -- i.