From 049aaa4c96552bc4a8966bccd3d65c1c4b4da496 Mon Sep 17 00:00:00 2001 From: dulibo1 Date: Wed, 13 Sep 2023 15:44:59 +0800 Subject: [PATCH] pmconfig: check domain if invaild return error avoid to crash Signed-off-by: dulibo1 --- nshlib/nsh_syscmds.c | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/nshlib/nsh_syscmds.c b/nshlib/nsh_syscmds.c index e052e0b38..eb81fbb1b 100644 --- a/nshlib/nsh_syscmds.c +++ b/nshlib/nsh_syscmds.c @@ -253,6 +253,11 @@ int cmd_pmconfig(FAR struct nsh_vtbl_s *vtbl, int argc, FAR char **argv) if (argc == 2) { ctrl.domain = atoi(argv[1]); + if (ctrl.domain < 0 || ctrl.domain >= CONFIG_PM_NDOMAINS) + { + nsh_error(vtbl, g_fmtargrange, argv[1]); + return ERROR; + } } ctrl.action = BOARDIOC_PM_QUERYSTATE; @@ -275,6 +280,11 @@ int cmd_pmconfig(FAR struct nsh_vtbl_s *vtbl, int argc, FAR char **argv) if (argc == 4) { ctrl.domain = atoi(argv[3]); + if (ctrl.domain < 0 || ctrl.domain >= CONFIG_PM_NDOMAINS) + { + nsh_error(vtbl, g_fmtargrange, argv[3]); + return ERROR; + } } if (strcmp(argv[1], "stay") == 0)