From b7b98405089e69dfae0fc11e2a423860f50756e9 Mon Sep 17 00:00:00 2001 From: Robin Haberkorn Date: Wed, 18 Sep 2024 12:32:16 +0200 Subject: check that local register is not edited at the end of macro calls * This was unsafe and could easily result in crashes, since teco_qreg_current would afterwards point to an already freed Q-Register. * Since automatically editing another register or buffer is not easy to do right, we throw an error instead. --- src/error.h | 9 +++++++++ src/qreg-commands.c | 7 +++++++ src/qreg.c | 6 ++++++ 3 files changed, 22 insertions(+) (limited to 'src') diff --git a/src/error.h b/src/error.h index b12ec80..03af66c 100644 --- a/src/error.h +++ b/src/error.h @@ -47,6 +47,7 @@ typedef enum { TECO_ERROR_INVALIDQREG, TECO_ERROR_QREGOPUNSUPPORTED, TECO_ERROR_QREGCONTAINSNULL, + TECO_ERROR_EDITINGLOCALQREG, TECO_ERROR_MEMLIMIT, /** Interrupt current operation */ @@ -126,6 +127,14 @@ teco_error_qregcontainsnull_set(GError **error, const gchar *name, gsize len, gb "%sQ-Register \"%s\" contains null-byte", local ? "Local " : "", name_printable); } +static inline void +teco_error_editinglocalqreg_set(GError **error, const gchar *name, gsize len) +{ + g_autofree gchar *name_printable = teco_string_echo(name, len); + g_set_error(error, TECO_ERROR, TECO_ERROR_EDITINGLOCALQREG, + "Editing local Q-Register \"%s\" at end of macro call", name_printable); +} + static inline void teco_error_interrupted_set(GError **error) { diff --git a/src/qreg-commands.c b/src/qreg-commands.c index 8d28e7d..a96eb5f 100644 --- a/src/qreg-commands.c +++ b/src/qreg-commands.c @@ -652,8 +652,15 @@ teco_state_macro_got_register(teco_machine_main_t *ctx, teco_qreg_t *qreg, } else { g_auto(teco_qreg_table_t) table; teco_qreg_table_init(&table, FALSE); + if (!teco_qreg_execute(qreg, &table, error)) return NULL; + if (teco_qreg_current && !teco_qreg_current->must_undo) { + /* currently editing local Q-Register */ + teco_error_editinglocalqreg_set(error, teco_qreg_current->head.name.data, + teco_qreg_current->head.name.len); + return NULL; + } } return &teco_state_start; diff --git a/src/qreg.c b/src/qreg.c index cac2d12..17b4830 100644 --- a/src/qreg.c +++ b/src/qreg.c @@ -1160,6 +1160,12 @@ teco_ed_hook(teco_ed_hook_t type, GError **error) if (!teco_qreg_execute(qreg, &locals, error)) goto error_add_frame; + if (teco_qreg_current && !teco_qreg_current->must_undo) { + /* currently editing local Q-Register */ + teco_error_editinglocalqreg_set(error, teco_qreg_current->head.name.data, + teco_qreg_current->head.name.len); + goto error_add_frame; + } return teco_expressions_discard_args(error) && teco_expressions_brace_close(error); -- cgit v1.2.3