93 lines
2.8 KiB
Diff
93 lines
2.8 KiB
Diff
From 426f57383d647406ae9c628c472059c27cd6e040 Mon Sep 17 00:00:00 2001
|
|
From: Chris Coulson <chris.coulson@canonical.com>
|
|
Date: Fri, 10 Jul 2020 14:41:45 +0100
|
|
Subject: [PATCH 5/7] script: Avoid a use-after-free when redefining a function
|
|
during execution
|
|
|
|
---
|
|
grub-core/script/execute.c | 2 ++
|
|
grub-core/script/function.c | 16 +++++++++++++---
|
|
grub-core/script/parser.y | 3 ++-
|
|
include/grub/script_sh.h | 2 ++
|
|
4 files changed, 19 insertions(+), 4 deletions(-)
|
|
|
|
diff --git a/grub-core/script/execute.c b/grub-core/script/execute.c
|
|
index eccb48e..58ac52e 100644
|
|
--- a/grub-core/script/execute.c
|
|
+++ b/grub-core/script/execute.c
|
|
@@ -872,7 +872,9 @@ grub_script_function_call (grub_script_function_t func, int argc, char **args)
|
|
old_scope = scope;
|
|
scope = &new_scope;
|
|
|
|
+ func->executing++;
|
|
ret = grub_script_execute (func->func);
|
|
+ func->executing--;
|
|
|
|
function_return = 0;
|
|
active_loops = loops;
|
|
diff --git a/grub-core/script/function.c b/grub-core/script/function.c
|
|
index d36655e..3aad04b 100644
|
|
--- a/grub-core/script/function.c
|
|
+++ b/grub-core/script/function.c
|
|
@@ -34,6 +34,7 @@ grub_script_function_create (struct grub_script_arg *functionname_arg,
|
|
func = (grub_script_function_t) grub_malloc (sizeof (*func));
|
|
if (! func)
|
|
return 0;
|
|
+ func->executing = 0;
|
|
|
|
func->name = grub_strdup (functionname_arg->str);
|
|
if (! func->name)
|
|
@@ -60,10 +61,19 @@ grub_script_function_create (struct grub_script_arg *functionname_arg,
|
|
grub_script_function_t q;
|
|
|
|
q = *p;
|
|
- grub_script_free (q->func);
|
|
- q->func = cmd;
|
|
grub_free (func);
|
|
- func = q;
|
|
+ if (q->executing > 0)
|
|
+ {
|
|
+ grub_error (GRUB_ERR_BAD_ARGUMENT,
|
|
+ N_("attempt to redefine a function being executed"));
|
|
+ func = NULL;
|
|
+ }
|
|
+ else
|
|
+ {
|
|
+ grub_script_free (q->func);
|
|
+ q->func = cmd;
|
|
+ func = q;
|
|
+ }
|
|
}
|
|
else
|
|
{
|
|
diff --git a/grub-core/script/parser.y b/grub-core/script/parser.y
|
|
index 4f0ab83..f80b86b 100644
|
|
--- a/grub-core/script/parser.y
|
|
+++ b/grub-core/script/parser.y
|
|
@@ -289,7 +289,8 @@ function: "function" "name"
|
|
grub_script_mem_free (state->func_mem);
|
|
else {
|
|
script->children = state->scripts;
|
|
- grub_script_function_create ($2, script);
|
|
+ if (!grub_script_function_create ($2, script))
|
|
+ grub_script_free (script);
|
|
}
|
|
|
|
state->scripts = $<scripts>3;
|
|
diff --git a/include/grub/script_sh.h b/include/grub/script_sh.h
|
|
index 360c2be..328e600 100644
|
|
--- a/include/grub/script_sh.h
|
|
+++ b/include/grub/script_sh.h
|
|
@@ -365,6 +365,8 @@ struct grub_script_function
|
|
/* The next element. */
|
|
struct grub_script_function *next;
|
|
|
|
+ unsigned executing;
|
|
+
|
|
int references;
|
|
};
|
|
typedef struct grub_script_function *grub_script_function_t;
|
|
--
|
|
2.23.0
|
|
|