Check for nullptr in tsm_vte_get_flags

aheck · Sep 20, 2022 · d66dd16 · d66dd16
1 parent 2c8ba7b
commit d66dd16
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/tsm/tsm-vte.c b/src/tsm/tsm-vte.c
@@ -593,6 +593,10 @@ void tsm_vte_get_def_attr(struct tsm_vte *vte, struct tsm_screen_attr *out)
 SHL_EXPORT
 unsigned int tsm_vte_get_flags(struct tsm_vte *vte)
 {
+	if (!vte) {
+		return 0;
+	}
+
 	return vte->flags;
 }
 

diff --git a/test/test_vte.c b/test/test_vte.c
@@ -91,6 +91,7 @@ START_TEST(test_vte_null)
 	ck_assert_int_eq(r, -EINVAL);
 
 	tsm_vte_get_def_attr(NULL, NULL);
+	tsm_vte_get_flags(NULL);
 
 	tsm_vte_get_mouse_mode(NULL);
 	tsm_vte_get_mouse_event(NULL);