Check the argument for -t if it is number.

Signed-off-by: Adham Masarwah <adham@mellanox.com>

diff --git a/small_utils/vpd.c b/small_utils/vpd.c
index 796bc40e68d85a11edb124c93071a9d0df6acd0b..bb76a9878ddb5d9e95adabeeff24ed88e3d23f72 100644 (file)
--- a/small_utils/vpd.c
+++ b/small_utils/vpd.c
@@ -531,6 +531,7 @@ int vpd_open(const char *name, int* vpd_path_exists)
 int main(int argc, char **argv)
 {
        const char *name;
+       char* endptr;
        int fd;
        int i;
        int rc = 0;
@@ -568,7 +569,11 @@ int main(int argc, char **argv)
                                ignore_w=1;
                                break;
                        case 't':
-                               timeout_t = strtol(optarg, NULL, 0);
+                               timeout_t = strtol(optarg, &endptr, 0);
+                               if (*endptr != '\0') {
+                                   fprintf(stderr, "-E- Invalid timeout argument: %s.\n",  optarg);
+                                   return 1;
+                               }
                                if ( timeout_t <= 0 ) {
                                    fprintf(stderr, "-E- Wrong timeout, it should be > 0 !\n");
                                    return 1;