From 02b77f7b2b8d8d5337ef3d8b2cb232cae27c5cfc Mon Sep 17 00:00:00 2001 From: Sami Kerola Date: Sun, 1 May 2011 19:24:47 +0200 Subject: column.c: validate numeric user inputs Use strtol_or_err from strutils.h to check numeric user input is sane. Signed-off-by: Sami Kerola --- text-utils/column.c | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) (limited to 'text-utils/column.c') diff --git a/text-utils/column.c b/text-utils/column.c index 6b797d340..7604dc905 100644 --- a/text-utils/column.c +++ b/text-utils/column.c @@ -54,6 +54,7 @@ #include "widechar.h" #include "c.h" #include "xalloc.h" +#include "strutils.h" #ifdef HAVE_WIDECHAR #define wcs_width(s) wcswidth(s,wcslen(s)) @@ -81,7 +82,7 @@ typedef struct _tbl { int cols, *len; } TBL; -int termwidth = 80; /* default terminal width */ +long termwidth; int entries; /* number of records */ int eval; /* exit value */ @@ -134,7 +135,8 @@ int main(int argc, char **argv) if (ioctl(STDIN_FILENO, TIOCGWINSZ, &win) == -1 || !win.ws_col) { if ((p = getenv("COLUMNS")) != NULL) - termwidth = atoi(p); + termwidth = strtol_or_err(p, + _("terminal environment COLUMNS failed")); } else termwidth = win.ws_col; @@ -149,7 +151,11 @@ int main(int argc, char **argv) PACKAGE_STRING); return(EXIT_SUCCESS); case 'c': - termwidth = atoi(optarg); + termwidth = strtol_or_err(optarg, + _("bad columns width value")); + if (termwidth < 1) + errx(EXIT_FAILURE, + _("-%c positive integer expected as an argument"), ch); break; case 's': separator = mbs_to_wcs(optarg); -- cgit v1.2.3-55-g7522