From 5d8d4509dd8112b4cf2b6aada1dca9e72bac2f44 Mon Sep 17 00:00:00 2001
From: Sunzen Wang <sunzen@gmail.com>
Date: Tue, 19 May 2009 09:07:11 +0800
Subject: Enhance -s option checking to only accept valid size number

Fix: Enhance -s option checking to only accept valid size number
---
 main.cxx | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'main.cxx')

diff --git a/main.cxx b/main.cxx
index 39d835d7..ec5506f5 100644
--- a/main.cxx
+++ b/main.cxx
@@ -626,8 +626,8 @@ main (int argc, char * const argv [])
 	  break;
 
         case 's':
-          s.buffer_size = atoi (optarg);
-          if (s.buffer_size < 1 || s.buffer_size > 4095)
+          s.buffer_size = (int) strtoul (optarg, &num_endptr, 10);
+          if (*num_endptr != '\0' || s.buffer_size < 1 || s.buffer_size > 4095)
             {
               cerr << "Invalid buffer size (should be 1-4095)." << endl;
 	      usage (s, 1);
-- 
cgit