replace parsing to use getopt (safer), change count opt behavior, close socket when needed
This commit is contained in:
2
Makefile
2
Makefile
@ -21,6 +21,8 @@ OBJS = $(addprefix $(OBJS_DIR)/,$(SRCS:.c=.o))\
|
|||||||
|
|
||||||
all: $(NAME)
|
all: $(NAME)
|
||||||
|
|
||||||
|
bonus: all
|
||||||
|
|
||||||
$(NAME): $(OBJS_DIR) $(OBJS)
|
$(NAME): $(OBJS_DIR) $(OBJS)
|
||||||
$(CC) $(FLAGS) $(OBJS) -o $@ -lm
|
$(CC) $(FLAGS) $(OBJS) -o $@ -lm
|
||||||
|
|
||||||
|
@ -6,7 +6,7 @@
|
|||||||
/* By: tomoron <tomoron@student.42angouleme.fr> +#+ +:+ +#+ */
|
/* By: tomoron <tomoron@student.42angouleme.fr> +#+ +:+ +#+ */
|
||||||
/* +#+#+#+#+#+ +#+ */
|
/* +#+#+#+#+#+ +#+ */
|
||||||
/* Created: 2025/04/26 19:54:25 by tomoron #+# #+# */
|
/* Created: 2025/04/26 19:54:25 by tomoron #+# #+# */
|
||||||
/* Updated: 2025/06/02 22:10:42 by tomoron ### ########.fr */
|
/* Updated: 2025/08/13 16:59:55 by tomoron ### ########.fr */
|
||||||
/* */
|
/* */
|
||||||
/* ************************************************************************** */
|
/* ************************************************************************** */
|
||||||
|
|
||||||
@ -83,7 +83,7 @@ int ping_host(t_settings *set, char *host)
|
|||||||
wl = 0;
|
wl = 0;
|
||||||
inet_ntop(info->ai_family, (void *)&((struct sockaddr_in *)(info->ai_addr))->sin_addr, set->current_ip, sizeof(set->current_ip));
|
inet_ntop(info->ai_family, (void *)&((struct sockaddr_in *)(info->ai_addr))->sin_addr, set->current_ip, sizeof(set->current_ip));
|
||||||
ping_start_print(set, host);
|
ping_start_print(set, host);
|
||||||
while((seq < set->count || set->count == -1) && !g_stop)
|
while((seq < set->count || set->count == 0) && !g_stop)
|
||||||
{
|
{
|
||||||
tmp = send_icmp(set, info, &seq, &last_sent, &stats);
|
tmp = send_icmp(set, info, &seq, &last_sent, &stats);
|
||||||
if(tmp)
|
if(tmp)
|
||||||
|
@ -6,7 +6,7 @@
|
|||||||
/* By: tomoron <tomoron@student.42angouleme.fr> +#+ +:+ +#+ */
|
/* By: tomoron <tomoron@student.42angouleme.fr> +#+ +:+ +#+ */
|
||||||
/* +#+#+#+#+#+ +#+ */
|
/* +#+#+#+#+#+ +#+ */
|
||||||
/* Created: 2025/04/24 22:49:22 by tomoron #+# #+# */
|
/* Created: 2025/04/24 22:49:22 by tomoron #+# #+# */
|
||||||
/* Updated: 2025/08/08 22:19:52 by tomoron ### ########.fr */
|
/* Updated: 2025/08/13 16:35:29 by tomoron ### ########.fr */
|
||||||
/* */
|
/* */
|
||||||
/* ************************************************************************** */
|
/* ************************************************************************** */
|
||||||
#include "includes/ft_ping.h"
|
#include "includes/ft_ping.h"
|
||||||
@ -125,7 +125,7 @@ void ttl_exceeded(t_icmp_ip_reply *res,t_settings *settings, size_t len)
|
|||||||
sent->header.ttl, sent->header.protocol, htons(sent->header.check),\
|
sent->header.ttl, sent->header.protocol, htons(sent->header.check),\
|
||||||
convert_ip(sent->header.saddr));
|
convert_ip(sent->header.saddr));
|
||||||
printf(" %s\nICMP: ", convert_ip(sent->header.daddr));
|
printf(" %s\nICMP: ", convert_ip(sent->header.daddr));
|
||||||
printf("type %d, code %d, size %zu, id 0x%04x, seq 0x%04x\n", sent->icmp.type, sent->icmp.code, sent->header.tot_len - sizeof(struct iphdr), sent->icmp.identifier, sent->icmp.sequence);
|
printf("type %d, code %d, size %zu, id 0x%04x, seq 0x%04x\n", sent->icmp.type, sent->icmp.code, htons(sent->header.tot_len) - sizeof(struct iphdr), sent->icmp.identifier, sent->icmp.sequence);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -16,6 +16,7 @@
|
|||||||
#include <sys/param.h>
|
#include <sys/param.h>
|
||||||
#include <math.h>
|
#include <math.h>
|
||||||
#include <netinet/ip.h>
|
#include <netinet/ip.h>
|
||||||
|
#include <getopt.h>
|
||||||
|
|
||||||
#include <fcntl.h>
|
#include <fcntl.h>
|
||||||
|
|
||||||
@ -29,8 +30,9 @@ Send ICMP ECHO_REQUEST packets to network hosts.\n\
|
|||||||
-i, --interval=N Wait N seconds between sending each packet\n\
|
-i, --interval=N Wait N seconds between sending each packet\n\
|
||||||
-w, --timeout=N Stop after N seconds\n\
|
-w, --timeout=N Stop after N seconds\n\
|
||||||
-W, --linger=N Number of seconds to wait for response\n\
|
-W, --linger=N Number of seconds to wait for response\n\
|
||||||
|
-v, --verbose Enable verbose mode\n\
|
||||||
\n\
|
\n\
|
||||||
-?, --help Give this help list\n"
|
-h, --help Print this usage info\n"
|
||||||
|
|
||||||
extern int g_stop;
|
extern int g_stop;
|
||||||
|
|
||||||
@ -56,7 +58,7 @@ typedef struct s_settings
|
|||||||
int linger; //-W (--linger)
|
int linger; //-W (--linger)
|
||||||
int ttl; // --ttl
|
int ttl; // --ttl
|
||||||
short setTtl; // is ttl set
|
short setTtl; // is ttl set
|
||||||
short verbose; // -v (--verbose)
|
int verbose; // -v (--verbose)
|
||||||
|
|
||||||
int socket;
|
int socket;
|
||||||
uint16_t id;
|
uint16_t id;
|
||||||
@ -102,8 +104,8 @@ typedef struct s_stats
|
|||||||
t_icmp_echo prepare_icmp_echo(uint16_t sequence, uint16_t identifier);
|
t_icmp_echo prepare_icmp_echo(uint16_t sequence, uint16_t identifier);
|
||||||
int add_host(t_settings *res, char *host);
|
int add_host(t_settings *res, char *host);
|
||||||
void free_hosts(t_host_list *list);
|
void free_hosts(t_host_list *list);
|
||||||
t_settings parse_args(int argc, char **argv);
|
int parse_args(int argc, char **argv, t_settings *set);
|
||||||
void show_help(t_settings *set, char *name);
|
void show_help(t_settings *set);
|
||||||
void send_pings(t_settings *set);
|
void send_pings(t_settings *set);
|
||||||
uint16_t calc_checksum(void *ptr, size_t len);
|
uint16_t calc_checksum(void *ptr, size_t len);
|
||||||
double timediff(struct timeval *from);
|
double timediff(struct timeval *from);
|
||||||
|
28
srcs/main.c
28
srcs/main.c
@ -6,7 +6,7 @@
|
|||||||
/* By: tomoron <tomoron@student.42angouleme.fr> +#+ +:+ +#+ */
|
/* By: tomoron <tomoron@student.42angouleme.fr> +#+ +:+ +#+ */
|
||||||
/* +#+#+#+#+#+ +#+ */
|
/* +#+#+#+#+#+ +#+ */
|
||||||
/* Created: 2025/04/24 00:03:56 by tomoron #+# #+# */
|
/* Created: 2025/04/24 00:03:56 by tomoron #+# #+# */
|
||||||
/* Updated: 2025/08/08 22:30:18 by tomoron ### ########.fr */
|
/* Updated: 2025/08/15 02:14:40 by tomoron ### ########.fr */
|
||||||
/* */
|
/* */
|
||||||
/* ************************************************************************** */
|
/* ************************************************************************** */
|
||||||
|
|
||||||
@ -14,10 +14,10 @@
|
|||||||
|
|
||||||
int g_stop = 0;
|
int g_stop = 0;
|
||||||
|
|
||||||
void show_help(t_settings *set, char *name)
|
void show_help(t_settings *set)
|
||||||
{
|
{
|
||||||
set->stop = 1;
|
set->stop = 1;
|
||||||
printf("Usage: %s %s", name, HELP_MESSAGE);
|
printf("Usage: %s %s", set->name, HELP_MESSAGE);
|
||||||
}
|
}
|
||||||
|
|
||||||
int init_socket(char *name)
|
int init_socket(char *name)
|
||||||
@ -58,25 +58,23 @@ int main(int argc, char **argv)
|
|||||||
{
|
{
|
||||||
t_settings settings;
|
t_settings settings;
|
||||||
|
|
||||||
settings = parse_args(argc, argv);
|
|
||||||
if(settings.stop)
|
|
||||||
goto early_stop;
|
|
||||||
settings.socket = init_socket(argv[0]);
|
|
||||||
settings.id = get_id();
|
|
||||||
settings.name = argv[0];
|
|
||||||
signal(SIGINT, signal_handler);
|
signal(SIGINT, signal_handler);
|
||||||
|
settings.name = argv[0];
|
||||||
|
if(parse_args(argc, argv, &settings))
|
||||||
|
return(1);
|
||||||
|
settings.id = get_id();
|
||||||
|
if(!settings.err && !settings.stop)
|
||||||
|
settings.socket = init_socket(argv[0]);
|
||||||
if(settings.stop || settings.err || settings.socket == -1)
|
if(settings.stop || settings.err || settings.socket == -1)
|
||||||
{
|
{
|
||||||
early_stop:
|
if(settings.socket != -1)
|
||||||
|
close(settings.socket);
|
||||||
free_hosts(settings.hosts);
|
free_hosts(settings.hosts);
|
||||||
return((settings.err != 0) * 64);
|
return((settings.err != 0) * 64);
|
||||||
}
|
}
|
||||||
if(!settings.hosts)
|
|
||||||
{
|
|
||||||
fprintf(stderr, "%s: missing host operand\n", argv[0]);
|
|
||||||
return(64);
|
|
||||||
}
|
|
||||||
send_pings(&settings);
|
send_pings(&settings);
|
||||||
|
if(settings.socket != -1)
|
||||||
|
close(settings.socket);
|
||||||
free_hosts(settings.hosts);
|
free_hosts(settings.hosts);
|
||||||
return(0);
|
return(0);
|
||||||
}
|
}
|
||||||
|
187
srcs/parsing.c
187
srcs/parsing.c
@ -6,80 +6,40 @@
|
|||||||
/* By: tomoron <tomoron@student.42angouleme.fr> +#+ +:+ +#+ */
|
/* By: tomoron <tomoron@student.42angouleme.fr> +#+ +:+ +#+ */
|
||||||
/* +#+#+#+#+#+ +#+ */
|
/* +#+#+#+#+#+ +#+ */
|
||||||
/* Created: 2025/04/26 17:01:53 by tomoron #+# #+# */
|
/* Created: 2025/04/26 17:01:53 by tomoron #+# #+# */
|
||||||
/* Updated: 2025/06/01 21:37:16 by tomoron ### ########.fr */
|
/* Updated: 2025/08/15 02:16:12 by tomoron ### ########.fr */
|
||||||
/* */
|
/* */
|
||||||
/* ************************************************************************** */
|
/* ************************************************************************** */
|
||||||
#include "includes/ft_ping.h"
|
#include "includes/ft_ping.h"
|
||||||
|
|
||||||
char *get_value_in_arg(char *arg, int end)
|
int get_set_int(char *arg, t_settings *set)
|
||||||
{
|
{
|
||||||
char *res;
|
|
||||||
|
|
||||||
res = strstr(arg, "=");
|
|
||||||
if(res - arg != end)
|
|
||||||
return (0);
|
|
||||||
if (!res)
|
|
||||||
return (0);
|
|
||||||
res++;
|
|
||||||
return(res);
|
|
||||||
}
|
|
||||||
|
|
||||||
char *get_setting(int *i, int argc, char **argv, int end, t_settings *set)
|
|
||||||
{
|
|
||||||
char *value;
|
|
||||||
|
|
||||||
value = 0;
|
|
||||||
if(argv[*i][0] == '-' && argv[*i][1] == '-')
|
|
||||||
value = get_value_in_arg(argv[*i], end);
|
|
||||||
else if((*i) + 1 < argc)
|
|
||||||
value = argv[++(*i)];
|
|
||||||
if(!value)
|
|
||||||
{
|
|
||||||
fprintf(stderr, "%s: option '%s' requires an option\n", argv[0], argv[*i]);
|
|
||||||
set->err = 1;
|
|
||||||
return(0);
|
|
||||||
}
|
|
||||||
return(value);
|
|
||||||
}
|
|
||||||
|
|
||||||
int get_set_int(int *i, int argc, char **argv, int end, t_settings *set)
|
|
||||||
{
|
|
||||||
char *value;
|
|
||||||
char *tmp;
|
char *tmp;
|
||||||
|
|
||||||
value = get_setting(i, argc, argv, end, set);
|
tmp = arg;
|
||||||
if(!value)
|
|
||||||
return(0);
|
|
||||||
tmp = value;
|
|
||||||
while(*tmp)
|
while(*tmp)
|
||||||
{
|
{
|
||||||
if(!isdigit(*tmp))
|
if(!isdigit(*tmp))
|
||||||
{
|
{
|
||||||
fprintf(stderr, "%s: invalid value (`%s' near `%s')\n", argv[0], value, tmp);
|
fprintf(stderr, "%s: invalid value (`%s' near `%s')\n", set->name, arg, tmp);
|
||||||
set->err = 1;
|
set->err = 1;
|
||||||
return(0);
|
return(0);
|
||||||
}
|
}
|
||||||
tmp++;
|
tmp++;
|
||||||
}
|
}
|
||||||
return(atoi(value));
|
return(atoi(arg));
|
||||||
}
|
}
|
||||||
|
|
||||||
double get_set_float(int *i, int argc, char **argv, int end, t_settings *set)
|
double get_set_float(char *arg, t_settings *set)
|
||||||
{
|
{
|
||||||
char *value;
|
|
||||||
char *tmp;
|
char *tmp;
|
||||||
int point;
|
int point;
|
||||||
|
|
||||||
value = get_setting(i, argc, argv, end, set);
|
tmp = arg;
|
||||||
if(!value)
|
|
||||||
return(0);
|
|
||||||
tmp = value;
|
|
||||||
point = 0;
|
|
||||||
while(*tmp)
|
while(*tmp)
|
||||||
{
|
{
|
||||||
if((!isdigit(*tmp) && *tmp != '.') || (*tmp == '.' && point))
|
if((!isdigit(*tmp) && *tmp != '.') || (*tmp == '.' && point))
|
||||||
{
|
{
|
||||||
fprintf(stderr, "%s: invalid value (`%s' near `%s')\n", argv[0], value, tmp);
|
fprintf(stderr, "%s: invalid value (`%s' near `%s')\n", set->name, arg, tmp);
|
||||||
set->err = 1;
|
set->err = 1;
|
||||||
return(0);
|
return(0);
|
||||||
}
|
}
|
||||||
@ -87,33 +47,7 @@ double get_set_float(int *i, int argc, char **argv, int end, t_settings *set)
|
|||||||
point = 1;
|
point = 1;
|
||||||
tmp++;
|
tmp++;
|
||||||
}
|
}
|
||||||
return(atof(value));
|
return(atof(arg));
|
||||||
}
|
|
||||||
|
|
||||||
void parse_opt(int *i, t_settings *set, int argc, char **argv)
|
|
||||||
{
|
|
||||||
if(!strcmp(argv[*i], "-?") || !strcmp(argv[*i], "--help"))
|
|
||||||
show_help(set, *argv);
|
|
||||||
else if(!strcmp(argv[*i], "-c") || !strncmp(argv[*i], "--count", 7))
|
|
||||||
set->count = get_set_int(i, argc, argv, 7, set);
|
|
||||||
else if(!strcmp(argv[*i], "-v") || !strncmp(argv[*i], "--verbose", 7))
|
|
||||||
set->verbose = 1;
|
|
||||||
else if(!strcmp(argv[*i], "-w") || !strncmp(argv[*i], "--timeout", 9))
|
|
||||||
set->timeout = get_set_int(i, argc, argv, 9, set);
|
|
||||||
else if(!strcmp(argv[*i], "-W") || !strncmp(argv[*i], "--linger", 8))
|
|
||||||
set->linger = get_set_int(i, argc, argv, 8, set);
|
|
||||||
else if(!strcmp(argv[*i], "-i") || !strncmp(argv[*i], "--interval", 10))
|
|
||||||
set->interval = get_set_float(i, argc, argv, 10, set);
|
|
||||||
else if(!strncmp(argv[*i], "--ttl", 5))
|
|
||||||
{
|
|
||||||
set->ttl = get_set_int(i, argc, argv, 5, set);
|
|
||||||
set->setTtl = 1;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
fprintf(stderr, "%s: unrecognized option '%s'\n", argv[0], argv[*i]);
|
|
||||||
set->err = 1;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
int check_values(t_settings *set, char *name)
|
int check_values(t_settings *set, char *name)
|
||||||
@ -139,32 +73,81 @@ int check_values(t_settings *set, char *name)
|
|||||||
return(1);
|
return(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
t_settings parse_args(int argc, char **argv)
|
void init_settings(t_settings *set)
|
||||||
{
|
{
|
||||||
int i;
|
char *tmp;
|
||||||
t_settings res;
|
tmp = set->name;
|
||||||
|
bzero(set, sizeof(t_settings));
|
||||||
i = 1;
|
set->name = tmp;
|
||||||
bzero(&res, sizeof(t_settings));
|
set->timeout = -1;
|
||||||
res.count = -1;
|
set->interval = 1;
|
||||||
res.timeout = -1;
|
set->socket = -1;
|
||||||
res.interval = 1;
|
set->linger = 10;
|
||||||
res.linger = 10;
|
}
|
||||||
while(i < argc)
|
|
||||||
{
|
int parse_args(int argc, char **argv, t_settings *set)
|
||||||
if(*argv[i] == '-')
|
{
|
||||||
parse_opt(&i, &res, argc, argv);
|
int arg;
|
||||||
else
|
struct option opts[] =
|
||||||
{
|
{
|
||||||
if(!add_host(&res, argv[i]))
|
{"count", required_argument, 0, 'c'},
|
||||||
{
|
{"interval", required_argument, 0, 'i'},
|
||||||
res.err = 1;
|
{"timeout", required_argument, 0, 'w'},
|
||||||
return(res);
|
{"linger", required_argument, 0, 'W'},
|
||||||
}
|
{"ttl", required_argument , 0, 1000},
|
||||||
}
|
{"verbose", no_argument, &set->verbose, 1},
|
||||||
i++;
|
{"help", no_argument, 0, 'h'},
|
||||||
}
|
{0, 0, 0, 0}
|
||||||
if(!res.err)
|
};
|
||||||
res.err = !check_values(&res, argv[0]);
|
|
||||||
return(res);
|
|
||||||
|
unsetenv("POSIXLY_CORRECT");
|
||||||
|
init_settings(set);
|
||||||
|
arg = 0;
|
||||||
|
while (arg != -1)
|
||||||
|
{
|
||||||
|
arg = getopt_long(argc, argv, "c:i:w:W:vh", opts, NULL);
|
||||||
|
switch (arg) {
|
||||||
|
case 'c':
|
||||||
|
set->count = get_set_int(optarg, set);
|
||||||
|
break;
|
||||||
|
case 'i':
|
||||||
|
set->interval = get_set_float(optarg, set);
|
||||||
|
break;
|
||||||
|
case 'w':
|
||||||
|
set->timeout = get_set_int(optarg, set);
|
||||||
|
break;
|
||||||
|
case 'W':
|
||||||
|
set->linger = get_set_int(optarg, set);
|
||||||
|
break;
|
||||||
|
case 1000:
|
||||||
|
set->ttl = get_set_int(optarg, set);
|
||||||
|
break;
|
||||||
|
case 'h':
|
||||||
|
case '?':
|
||||||
|
show_help(set);
|
||||||
|
return(1);
|
||||||
|
default :
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
while(optind < argc)
|
||||||
|
{
|
||||||
|
if(!add_host(set, argv[optind]))
|
||||||
|
{
|
||||||
|
set->err = 1;
|
||||||
|
return (0);
|
||||||
|
}
|
||||||
|
optind++;
|
||||||
|
}
|
||||||
|
if(!set->hosts)
|
||||||
|
{
|
||||||
|
fprintf(stderr, "%s: missing host operand\n", set->name);
|
||||||
|
set->err = 1;
|
||||||
|
return(1);
|
||||||
|
}
|
||||||
|
if (!set->err)
|
||||||
|
set->err = !check_values(set, argv[0]);
|
||||||
|
return (0);
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user