106 lines
3.5 KiB
Diff
106 lines
3.5 KiB
Diff
From 0000000000000000000000000000000000000000 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] script: Avoid a use-after-free when redefining a function
|
|
during execution
|
|
|
|
Defining a new function with the same name as a previously defined
|
|
function causes the grub_script and associated resources for the
|
|
previous function to be freed. If the previous function is currently
|
|
executing when a function with the same name is defined, this results
|
|
in use-after-frees when processing subsequent commands in the original
|
|
function.
|
|
|
|
Instead, reject a new function definition if it has the same name as
|
|
a previously defined function, and that function is currently being
|
|
executed. Although a behavioural change, this should be backwards
|
|
compatible with existing configurations because they can't be
|
|
dependent on the current behaviour without being broken.
|
|
|
|
Signed-off-by: Chris Coulson <chris.coulson@canonical.com>
|
|
Reviewed-by: Daniel Kiper <daniel.kiper@oracle.com>
|
|
Upstream-commit-id: f6253a1f540
|
|
---
|
|
grub-core/script/execute.c | 2 ++
|
|
grub-core/script/function.c | 16 +++++++++++++---
|
|
include/grub/script_sh.h | 2 ++
|
|
grub-core/script/parser.y | 3 ++-
|
|
4 files changed, 19 insertions(+), 4 deletions(-)
|
|
|
|
diff --git a/grub-core/script/execute.c b/grub-core/script/execute.c
|
|
index 528ddfd36..a1aadb9ee 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 d36655e51..3aad04bf9 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/include/grub/script_sh.h b/include/grub/script_sh.h
|
|
index b382bcf09..6c48e0751 100644
|
|
--- a/include/grub/script_sh.h
|
|
+++ b/include/grub/script_sh.h
|
|
@@ -361,6 +361,8 @@ struct grub_script_function
|
|
|
|
/* The next element. */
|
|
struct grub_script_function *next;
|
|
+
|
|
+ unsigned executing;
|
|
};
|
|
typedef struct grub_script_function *grub_script_function_t;
|
|
|
|
diff --git a/grub-core/script/parser.y b/grub-core/script/parser.y
|
|
index 4f0ab8319..f80b86b6f 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;
|