fix: check if number contain only digit

This commit is contained in:
2025-11-18 08:16:02 -06:00
parent 0bb5365fdf
commit 6dcc4f39cc

View File

@ -24,6 +24,7 @@
#include <stdbool.h> #include <stdbool.h>
#include <sys/socket.h> #include <sys/socket.h>
#include <netdb.h> #include <netdb.h>
#include <ctype.h>
#include <net/ethernet.h> #include <net/ethernet.h>
bool loop = true; bool loop = true;
@ -41,23 +42,29 @@ static int parsing_number(const char *str, size_t min, size_t max, size_t *out)
while (*start != '\0') while (*start != '\0')
{ {
if (!isdigit(*start))
goto value_error;
size_t tmp = number * 10 + *start - '0'; size_t tmp = number * 10 + *start - '0';
if (tmp < number) if (tmp < number)
goto value_error; goto range_error;
number = tmp; number = tmp;
start++; start++;
} }
if (min > number || number > max) if (min > number || number > max)
goto value_error; goto range_error;
*out = number; *out = number;
return 0; return 0;
value_error: range_error:
print_err("invalid argument: '%s': out of range: %zu <= value <= %zu", str, min, max); print_err("invalid argument: '%s': out of range: %zu <= value <= %zu", str, min, max);
return 1; return 1;
value_error:
print_err("invalid argument: in '%s' '%c' is not a digit", str, *start);
return 1;
} }
static int get_setting(char * const *av, struct setting *setting) static int get_setting(char * const *av, struct setting *setting)
@ -95,7 +102,7 @@ int main(int ac, char **av)
if (get_setting(av + 1, &settings)) if (get_setting(av + 1, &settings))
return 1; return 1;
size_t payload_size = 20; size_t payload_size = 56;
size_t packet_size = sizeof(struct icmphdr) + payload_size; size_t packet_size = sizeof(struct icmphdr) + payload_size;
signal(SIGINT, signal_handler); signal(SIGINT, signal_handler);
@ -166,7 +173,7 @@ int main(int ac, char **av)
double time_interval = ((stop.tv_sec - start.tv_sec) * 1000 + ((double)stop.tv_usec - (double) start.tv_usec) / 1000); double time_interval = ((stop.tv_sec - start.tv_sec) * 1000 + ((double)stop.tv_usec - (double) start.tv_usec) / 1000);
struct icmphdr *hdr = (struct icmphdr *) packet; struct icmphdr *hdr = (struct icmphdr *) packet;
printf("%zu bytes from %s (%s): icmp_seq=%d, ttl=%u, time=%fms\n", payload_size + sizeof(struct icmphdr) + sizeof(struct iphdr) + sizeof(struct ether_header), settings.dest.reverse_dns, settings.dest.ipstr, htons(hdr->un.echo.sequence), 116, time_interval); printf("%zu bytes from %s (%s): icmp_seq=%d, ttl=%u, time=%fms\n", payload_size + sizeof(struct icmphdr), settings.dest.reverse_dns, settings.dest.ipstr, htons(hdr->un.echo.sequence), 116, time_interval);
sleep(1); sleep(1);