tree: handle a pointer returned by malloc()

It is wrong to assume that malloc() always returns allocated memory.
Unfortunately, it can return NULL.

The rest of the code that uses malloc() should be refactored. That's
the reason why we can't do it right now.

1 files changed, 1 insertions, 0 deletions

diff --git a/src/tree.c b/src/tree.c
index 293a86a..632921f 100644
--- a/src/tree.c
+++ b/src/tree.c
@@ -26,6 +26,7 @@
 #include "tree.h"
 #include "xstd.h"
 #include "easydir.h"
+#include "output.h"
 
 #define ANSIC_RST	"\x1B[0m"
 #define ANSIC_BBLU	"\x1B[34;1m"