starnakin/42_ft_nmap
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add: multi scan

Browse Source
This commit is contained in:
starnakin
2025-07-02 07:09:54 -05:00
parent a5d663fa9a
commit afde93c7cf
8 changed files with 133 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/parsing.h
View File

@ -28,6 +28,6 @@ struct option_lst {
struct option_lst *parse_options(int ac, char *const *av); struct option_lst *parse_options(int ac, char *const *av);
char *get_option_arg(const struct option_lst *options, e_flag flag); char *get_option_arg(const struct option_lst *options, e_flag flag);
bool option_isset(const struct option_lst *options, e_flag flag); bool option_isset(const struct option_lst *options, e_flag flag);
e_scantype parse_type(const char *arg); uint8_t parse_type(char *arg);
void free_options(struct option_lst *options); void free_options(struct option_lst *options);
int parsing(struct scan *general, const struct option_lst *options); int parsing(struct scan *general, const struct option_lst *options);

2
include/response.h
View File

@ -22,7 +22,7 @@ typedef enum {
}; };
struct response { struct response {
e_state states[SCAN_ALL]; e_state states[NB_SCAN];
char *service; char *service;
}; };

20
include/scan.h
View File

@ -4,26 +4,26 @@
#include "host.h" #include "host.h"
typedef enum { #define SCAN_NULL (1 << 0)
SCAN_NULL, #define SCAN_SYN (1 << 1)
SCAN_SYN, #define SCAN_ACK (1 << 2)
SCAN_ACK, #define SCAN_FIN (1 << 3)
SCAN_FIN, #define SCAN_XMAS (1 << 4)
SCAN_XMAS, #define SCAN_UDP (1 << 5)
SCAN_UDP, #define SCAN_ALL (SCAN_NULL | SCAN_SYN | SCAN_ACK | SCAN_FIN | SCAN_XMAS | SCAN_UDP)
SCAN_ALL,
} e_scantype;
[[__maybe_unused__]] static const char *types_str[] = { [[__maybe_unused__]] static const char *types_str[] = {
"NULL", "SYN", "ACK", "FIN", "XMAS", "UDP", "NULL", "SYN", "ACK", "FIN", "XMAS", "UDP",
}; };
#define NB_SCAN (sizeof(types_str) / sizeof(*types_str))
struct scan { struct scan {
struct host host; struct host host;
char *dest_addr; char *dest_addr;
uint16_t port_start; uint16_t port_start;
uint16_t port_end; uint16_t port_end;
e_scantype type; uint8_t type;
uint8_t ttl; uint8_t ttl;
uint8_t max_retries; uint8_t max_retries;
struct response *responses; struct response *responses;

14
src/packet.c
View File

@ -23,11 +23,11 @@ int create_tcp_packet(struct tcphdr *tcphdr, const struct scan *data)
tcphdr->source = htons(1234); tcphdr->source = htons(1234);
tcphdr->dest = htons(data->port_start); tcphdr->dest = htons(data->port_start);
tcphdr->doff = sizeof(struct tcphdr) / sizeof(int); tcphdr->doff = sizeof(struct tcphdr) / sizeof(int);
tcphdr->fin = data->type == SCAN_XMAS || data->type == SCAN_FIN; tcphdr->fin = ((data->type & SCAN_XMAS) != 0) || ((data->type & SCAN_FIN) != 0);
tcphdr->syn = data->type == SCAN_SYN; tcphdr->syn = ((data->type & SCAN_SYN) != 0);
tcphdr->psh = data->type == SCAN_XMAS; tcphdr->psh = ((data->type & SCAN_XMAS) != 0);
tcphdr->ack = data->type == SCAN_ACK; tcphdr->ack = ((data->type & SCAN_ACK) != 0);
tcphdr->urg = data->type == SCAN_XMAS; tcphdr->urg = ((data->type & SCAN_XMAS) != 0);
tcphdr->window = htons(5840); tcphdr->window = htons(5840);
tcphdr->urg_ptr = 0; tcphdr->urg_ptr = 0;
@ -52,7 +52,7 @@ int create_tcp_packet(struct tcphdr *tcphdr, const struct scan *data)
static void *create_packet(const struct scan *data, size_t packet_size) static void *create_packet(const struct scan *data, size_t packet_size)
{ {
const bool isudp = data->type == SCAN_UDP; const bool isudp = ((data->type & SCAN_UDP) != 0);
void *packet = calloc(packet_size, 1); void *packet = calloc(packet_size, 1);
if (!packet) { if (!packet) {
dprintf(2, dprintf(2,
@ -90,7 +90,7 @@ int send_packet(const struct scan *data, int sockfd)
conn_addr.sin_addr.s_addr = inet_addr(data->dest_addr); conn_addr.sin_addr.s_addr = inet_addr(data->dest_addr);
size_t packet_size = sizeof(struct iphdr) + size_t packet_size = sizeof(struct iphdr) +
(data->type == SCAN_UDP ? sizeof(struct udphdr) (((data->type & SCAN_UDP) != 0) ? sizeof(struct udphdr)
: sizeof(struct tcphdr)); : sizeof(struct tcphdr));
void *packet = create_packet(data, packet_size); void *packet = create_packet(data, packet_size);

49
src/parsing.c
View File

@ -40,15 +40,36 @@ char *get_option_arg(const struct option_lst *options, e_flag flag)
return NULL; return NULL;
} }
e_scantype parse_type(const char *arg) uint8_t parse_type(char *arg)
{ {
if (!arg) if (!arg)
return SCAN_ALL; return SCAN_ALL;
for (size_t i = 0; i < 6; i++) uint8_t type = 0;
if (!strcmp(arg, types_str[i])) char *current_arg = arg;
return i; while (current_arg[0] != '\0')
dprintf(2, "ft_nmap: invalid argument to --scan: '%s'\n", arg); {
return -1; if (strncmp(current_arg, "NULL", 4) == 0)
type |= SCAN_NULL;
else if (strncmp(current_arg, "SYN", 3) == 0)
type |= SCAN_SYN;
else if (strncmp(current_arg, "ACK", 3) == 0)
type |= SCAN_ACK;
else if (strncmp(current_arg, "FIN", 3) == 0)
type |= SCAN_FIN;
else if (strncmp(current_arg, "XMAS", 4) == 0)
type |= SCAN_XMAS;
else if (strncmp(current_arg, "UDP", 5) == 0)
type |= SCAN_UDP;
else
{
dprintf(2, "ft_nmap: invalid argument to --scan: '%s'\n", arg);
return -1;
}
current_arg += strcspn(current_arg, ",");
if (current_arg[0] == ',')
current_arg++;
}
return type;
} }
int parse_ports(const char *arg, uint16_t *p_start, uint16_t *p_end) int parse_ports(const char *arg, uint16_t *p_start, uint16_t *p_end)
@ -114,20 +135,6 @@ static int check_arg(e_flag flag, const char *arg)
} }
break; break;
} }
case FL_TYPE:
if (!strcmp(arg, "SYN"))
break;
if (!strcmp(arg, "NULL"))
break;
if (!strcmp(arg, "ACK"))
break;
if (!strcmp(arg, "FIN"))
break;
if (!strcmp(arg, "XMAS"))
break;
if (!strcmp(arg, "UDP"))
break;
return -1;
case FL_TTL: { case FL_TTL: {
int nb = atoi(arg); int nb = atoi(arg);
if (nb == 0) { if (nb == 0) {
@ -248,7 +255,7 @@ int parsing(struct scan *general, const struct option_lst *options)
return -1; return -1;
} }
general->type = parse_type(get_option_arg(options, FL_TYPE)); general->type = parse_type(get_option_arg(options, FL_TYPE));
if ((int)general->type == -1) if (general->type == (uint8_t) -1)
return -1; return -1;
const char *max_retries = get_option_arg(options, FL_MAXRETRIES); const char *max_retries = get_option_arg(options, FL_MAXRETRIES);
general->max_retries = max_retries ? atoi(max_retries) : 1; general->max_retries = max_retries ? atoi(max_retries) : 1;

23
src/print.c
View File

@ -31,20 +31,18 @@ void print_config(const struct scan *general, const char *hosts_path,
printf("Number of scans to be performed: %d\n", printf("Number of scans to be performed: %d\n",
general->port_end - general->port_start + 1); general->port_end - general->port_start + 1);
printf("Scans to be performed: "); printf("Scans to be performed: ");
if (general->type == SCAN_ALL) for (uint8_t type = 0; type < NB_SCAN; type++)
for (e_scantype type = SCAN_NULL; type < SCAN_ALL; type++) if (general->type & (1 << type))
printf("%s ", types_str[type]); printf("%s ", types_str[type]);
else
printf("%s", types_str[general->type]);
printf("\n"); printf("\n");
if (nb_threads > 1) if (nb_threads > 1)
printf("No of threads: %d\n", nb_threads); printf("No of threads: %d\n", nb_threads);
} }
bool is_port_opened(const e_state states[6], e_scantype type) bool is_port_opened(const e_state states[6], uint8_t type)
{ {
if (type == SCAN_ALL) { if (type == SCAN_ALL) {
for (e_scantype i = SCAN_NULL; i < SCAN_ALL; i++) for (uint8_t i = 0; i < NB_SCAN; i++)
if (states[i] == OPENED) if (states[i] == OPENED)
return true; return true;
return false; return false;
@ -52,20 +50,19 @@ bool is_port_opened(const e_state states[6], e_scantype type)
return states[type] == OPENED; return states[type] == OPENED;
} }
static void print_port_state(uint16_t port, e_scantype type, static void print_port_state(uint16_t port, uint8_t type,
const struct response *response) const struct response *response)
{ {
printf("%-5d %-12s ", port, printf("%-5d %-12s ", port,
response->service ? response->service : "Unassigned"); response->service ? response->service : "Unassigned");
if (response->service) if (response->service)
free(response->service); free(response->service);
if (type == SCAN_ALL) for (uint8_t i = 0; i < NB_SCAN; i++)
for (e_scantype i = SCAN_NULL; i < SCAN_ALL; i++) {
if ((1 << i) & type)
printf("%s(%s) ", types_str[i], printf("%s(%s) ", types_str[i],
states_str[response->states[i]]); states_str[response->states[i]]);
else }
printf("%s(%s) ", types_str[type],
states_str[response->states[type]]);
printf("\n"); printf("\n");
} }

108
src/response.c
View File

@ -23,29 +23,39 @@ static char *get_service_name(int port, char *proto)
void tcp_response(const struct tcphdr *tcphdr, const struct scan *data) void tcp_response(const struct tcphdr *tcphdr, const struct scan *data)
{ {
const e_scantype type = data->type; if (data->type & SCAN_UDP) {
if (type == SCAN_UDP) {
dprintf(2, dprintf(2,
"ft_nmap: error: received a TCP responses for an UDP " "ft_nmap: error: received a TCP responses for an UDP "
"scan\n"); "scan\n");
return; return;
} }
uint8_t type_index;
for (uint8_t i = 0; i < NB_SCAN; i++)
{
if (data->type & (1 << i))
{
type_index = i;
break;
}
}
if (data->responses->service == NULL) if (data->responses->service == NULL)
data->responses->service = data->responses->service =
get_service_name(data->port_start, "tcp"); get_service_name(data->port_start, "tcp");
if (type == SCAN_SYN) { if (data->type & SCAN_SYN) {
if (tcphdr->ack == 1 && tcphdr->syn == 1) if (tcphdr->ack == 1 && tcphdr->syn == 1)
data->responses->states[type] = OPENED; data->responses->states[type_index] = OPENED;
else if (tcphdr->ack == 1 && tcphdr->rst == 1) else if (tcphdr->ack == 1 && tcphdr->rst == 1)
data->responses->states[type] = CLOSED; data->responses->states[type_index] = CLOSED;
} else if (type == SCAN_ACK && tcphdr->rst == 1) } else if (data->type & SCAN_ACK && tcphdr->rst == 1)
data->responses->states[type] = UNFILTERED; data->responses->states[type_index] = UNFILTERED;
else if (type == SCAN_NULL && tcphdr->rst == 1) else if (data->type & SCAN_NULL && tcphdr->rst == 1)
data->responses->states[type] = CLOSED; data->responses->states[type_index] = CLOSED;
else if (type == SCAN_FIN && tcphdr->rst == 1) else if (data->type & SCAN_FIN && tcphdr->rst == 1)
data->responses->states[type] = CLOSED; data->responses->states[type_index] = CLOSED;
else if (type == SCAN_XMAS && tcphdr->rst == 1) else if (data->type & SCAN_XMAS && tcphdr->rst == 1)
data->responses->states[type] = CLOSED; data->responses->states[type_index] = CLOSED;
} }
void udp_response(const struct udphdr *udphdr, const struct scan *data) void udp_response(const struct udphdr *udphdr, const struct scan *data)
@ -65,46 +75,62 @@ void udp_response(const struct udphdr *udphdr, const struct scan *data)
void icmp_response(const struct icmphdr *icmphdr, const struct scan *data) void icmp_response(const struct icmphdr *icmphdr, const struct scan *data)
{ {
const e_scantype type = data->type; uint8_t type_index;
for (uint8_t i = 0; i < NB_SCAN; i++)
{
if (data->type & (1 << i))
{
type_index = i;
break;
}
}
data->responses->service = get_service_name(data->port_start, "udp"); data->responses->service = get_service_name(data->port_start, "udp");
if (data->responses->service == NULL) if (data->responses->service == NULL)
data->responses->service = data->responses->service =
get_service_name(data->port_start, "tcp"); get_service_name(data->port_start, "tcp");
if (type == SCAN_SYN && icmphdr->type == 3) if (data->type & SCAN_SYN && icmphdr->type == 3)
data->responses->states[type] = FILTERED; data->responses->states[type_index] = FILTERED;
else if (type == SCAN_ACK && icmphdr->type == 3) else if (data->type & SCAN_ACK && icmphdr->type == 3)
data->responses->states[type] = FILTERED; data->responses->states[type_index] = FILTERED;
else if (type == SCAN_NULL && icmphdr->type == 3) else if (data->type & SCAN_NULL && icmphdr->type == 3)
data->responses->states[type] = FILTERED; data->responses->states[type_index] = FILTERED;
else if (type == SCAN_FIN && icmphdr->type == 3) else if (data->type & SCAN_FIN && icmphdr->type == 3)
data->responses->states[type] = FILTERED; data->responses->states[type_index] = FILTERED;
else if (type == SCAN_XMAS && icmphdr->type == 3) else if (data->type & SCAN_XMAS && icmphdr->type == 3)
data->responses->states[type] = FILTERED; data->responses->states[type_index] = FILTERED;
else if (type == SCAN_UDP && icmphdr->type == 3 && icmphdr->code == 3) else if (data->type & SCAN_UDP && icmphdr->type == 3 && icmphdr->code == 3)
data->responses->states[type] = CLOSED; data->responses->states[type_index] = CLOSED;
else if (type == SCAN_UDP && icmphdr->type == 3 && icmphdr->code != 3) else if (data->type & SCAN_UDP && icmphdr->type == 3 && icmphdr->code != 3)
data->responses->states[type] = FILTERED; data->responses->states[type_index] = FILTERED;
} }
void no_response(const struct scan *data) void no_response(const struct scan *data)
{ {
const e_scantype type = data->type; uint8_t type_index;
for (uint8_t i = 0; i < NB_SCAN; i++)
{
if (data->type & (1 << i))
{
type_index = i;
break;
}
}
data->responses->service = get_service_name(data->port_start, "udp"); data->responses->service = get_service_name(data->port_start, "udp");
if (data->responses->service == NULL) if (data->responses->service == NULL)
data->responses->service = data->responses->service =
get_service_name(data->port_start, "tcp"); get_service_name(data->port_start, "tcp");
if (type == SCAN_SYN) if (data->type & SCAN_SYN)
data->responses->states[type] = FILTERED; data->responses->states[type_index] = FILTERED;
else if (type == SCAN_ACK) else if (data->type & SCAN_ACK)
data->responses->states[type] = FILTERED; data->responses->states[type_index] = FILTERED;
else if (type == SCAN_NULL) else if (data->type & SCAN_NULL)
data->responses->states[type] = OPENFILTERED; data->responses->states[type_index] = OPENFILTERED;
else if (type == SCAN_FIN) else if (data->type & SCAN_FIN)
data->responses->states[type] = OPENFILTERED; data->responses->states[type_index] = OPENFILTERED;
else if (type == SCAN_XMAS) else if (data->type & SCAN_XMAS)
data->responses->states[type] = OPENFILTERED; data->responses->states[type_index] = OPENFILTERED;
else if (type == SCAN_UDP) else if (data->type & SCAN_UDP)
data->responses->states[type] = OPENFILTERED; data->responses->states[type_index] = OPENFILTERED;
} }

19
src/scan.c
View File

@ -107,20 +107,19 @@ int scan(struct scan *data)
pcap_setnonblock(handle, 1, errbuf); pcap_setnonblock(handle, 1, errbuf);
if (data->type == SCAN_ALL) { for (uint8_t type = 0; type < NB_SCAN; type++) {
for (e_scantype type = SCAN_NULL; type < SCAN_ALL; type++) { uint8_t type_bak = data->type;
data->type = type; data->type = data->type & (1 << type);
if (data->type)
{
for (uint8_t retries = 0; retries < data->max_retries; for (uint8_t retries = 0; retries < data->max_retries;
retries++) retries++)
{
if (send_recv_packet(data, sockfd, handle)) if (send_recv_packet(data, sockfd, handle))
break; break;
}
} }
data->type = SCAN_ALL; data->type = type_bak;
} else {
for (uint8_t retries = 0; retries < data->max_retries;
retries++)
if (send_recv_packet(data, sockfd, handle))
break;
} }
pcap_close(handle); pcap_close(handle);