[PATCH v4 2/3] trace-cmd: Remove the die() call from read_proc()

From: Vladislav Valtchev (VMware)
Date: Tue Jan 16 2018 - 14:55:02 EST


As trace-stack.c's read_proc() function is going to be used by trace-cmd stat,
we don't want it to make the program die in case something went wrong.
Therefore, this simple patch makes read_proc() to just return -1 in case the
proc file was empty or read() failed with an error, instead of using die().

Signed-off-by: Vladislav Valtchev (VMware) <vladislav.valtchev@xxxxxxxxx>
---
trace-stack.c | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/trace-stack.c b/trace-stack.c
index fada62d..3e1e41b 100644
--- a/trace-stack.c
+++ b/trace-stack.c
@@ -79,7 +79,7 @@ static int read_proc(int *status)
close(fd);

if (n <= 0)
- die("error reading %s", PROC_FILE);
+ return -1;

if (n >= sizeof(buf))
return -1;
@@ -101,6 +101,7 @@ static void change_stack_tracer_status(unsigned new_status)
{
char buf[1];
int status;
+ int ret;
int fd;
int n;

@@ -109,7 +110,11 @@ static void change_stack_tracer_status(unsigned new_status)
return;
}

- if (read_proc(&status) > 0 && status == new_status)
+ ret = read_proc(&status);
+ if (ret < 0)
+ die("error reading %s", PROC_FILE);
+
+ if (ret > 0 && status == new_status)
return; /* nothing to do */

fd = open(PROC_FILE, O_WRONLY);
--
2.14.1