Эх сурвалжийг харах

tools:iio:generic_buffer: catch errors for arguments conversion

Add handler to catch errors on conversion of numerical arguments.

Signed-off-by: Hartmut Knaack <knaack.h@gmx.de>
Signed-off-by: Jonathan Cameron <jic23@kernel.org>
Hartmut Knaack 10 жил өмнө
parent
commit
c8ce9903cb

+ 9 - 0
tools/iio/generic_buffer.c

@@ -206,13 +206,22 @@ int main(int argc, char **argv)
 			noevents = 1;
 			noevents = 1;
 			break;
 			break;
 		case 'c':
 		case 'c':
+			errno = 0;
 			num_loops = strtoul(optarg, &dummy, 10);
 			num_loops = strtoul(optarg, &dummy, 10);
+			if (errno)
+				return -errno;
 			break;
 			break;
 		case 'w':
 		case 'w':
+			errno = 0;
 			timedelay = strtoul(optarg, &dummy, 10);
 			timedelay = strtoul(optarg, &dummy, 10);
+			if (errno)
+				return -errno;
 			break;
 			break;
 		case 'l':
 		case 'l':
+			errno = 0;
 			buf_len = strtoul(optarg, &dummy, 10);
 			buf_len = strtoul(optarg, &dummy, 10);
+			if (errno)
+				return -errno;
 			break;
 			break;
 		case 'g':
 		case 'g':
 			notrigger = 1;
 			notrigger = 1;