From 838d14520267769648fb2fc2a637107a1d102590 Mon Sep 17 00:00:00 2001 From: Jiri Olsa Date: Thu, 28 Nov 2013 11:30:15 +0100 Subject: [PATCH] perf tools: Fine tune readn function Added a 'left' variable to make the flow clearer, and added a debug check for the return value - returning 'n' is more obvious. Added small comment for readn. Signed-off-by: Jiri Olsa Original-patch-by: Ingo Molnar Cc: Adrian Hunter Cc: David Ahern Cc: Frederic Weisbecker Cc: Ingo Molnar Cc: Mike Galbraith Cc: Namhyung Kim Cc: Peter Zijlstra Link: http://lkml.kernel.org/r/1385634619-8129-4-git-send-email-jolsa@redhat.com Signed-off-by: Arnaldo Carvalho de Melo --- tools/perf/util/util.c | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/tools/perf/util/util.c b/tools/perf/util/util.c index 9440481e9092..6ea0b4ae9569 100644 --- a/tools/perf/util/util.c +++ b/tools/perf/util/util.c @@ -6,6 +6,7 @@ #endif #include #include +#include /* * XXX We need to find a better place for these things... @@ -151,21 +152,26 @@ unsigned long convert_unit(unsigned long value, char *unit) return value; } +/* + * Read exactly 'n' bytes or return an error. + */ ssize_t readn(int fd, void *buf, size_t n) { void *buf_start = buf; + size_t left = n; - while (n) { - ssize_t ret = read(fd, buf, n); + while (left) { + ssize_t ret = read(fd, buf, left); if (ret <= 0) return ret; - n -= ret; - buf += ret; + left -= ret; + buf += ret; } - return buf - buf_start; + BUG_ON((size_t)(buf - buf_start) != n); + return n; } size_t hex_width(u64 v) -- 2.30.2