blob: dd8e9e6f0fd69db314ec34d1c549d9c7cefe0017 [file] [log] [blame] [raw]
/*
* Copyright (C) 2000-2005 SWsoft. All rights reserved.
*
* This code is public domain.
*
* This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
* WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
*/
#include <stdio.h>
#include <ctype.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
#include <netdb.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <linux/if_ether.h>
#include <linux/if_arp.h>
#include <linux/if_packet.h>
#include <linux/if.h>
#include <netinet/in.h>
#include <sys/ioctl.h>
#include <arpa/inet.h>
#include <getopt.h>
#include <unistd.h>
#include <signal.h>
#define EXC_OK 0
#define EXC_USAGE 1
#define EXC_SYS 2
#define EXC_RECV 3
#define EXC_NORECV 4
/* log facility */
#define LOG_ERROR 0x01
#define LOG_WARNING 0x02
#define LOG_INFO 0x03
#define LOG_DEBUG 0x04
#define LOGGER_NAME "arpsend: "
int debug_level = LOG_INFO;
#define logger(level, fmt, args...) \
(debug_level < level ? (void)0 : fprintf(stderr, LOGGER_NAME fmt "\n", ##args))
#define IP_ADDR_LEN 4
#define REQUEST ARPOP_REQUEST
#define REPLY ARPOP_REPLY
struct arp_packet {
u_char targ_hw_addr[ETH_ALEN]; /* ethernet destination MAC address */
u_char src_hw_addr[ETH_ALEN]; /* ethernet source MAC address */
u_short frame_type; /* ethernet packet type */
/* ARP packet */
u_short hw_type; /* hardware adress type (ethernet) */
u_short prot_type; /* resolve protocol address type (ip) */
u_char hw_addr_size;
u_char prot_addr_size;
u_short op; /* subtype */
u_char sndr_hw_addr[ETH_ALEN];
u_char sndr_ip_addr[IP_ADDR_LEN];
u_char rcpt_hw_addr[ETH_ALEN];
u_char rcpt_ip_addr[IP_ADDR_LEN];
/* ethernet padding */
u_char padding[18];
};
enum {
AR_NOTHING = 0,
AR_UPDATE,
AR_DETECT,
AR_REQUEST,
AR_REPLY
} cmd = AR_NOTHING;
char* iface = NULL;
int timeout = 1;
int count = -1;
/* source MAC address in Ethernet header */
unsigned char src_hwaddr[ETH_ALEN];
int src_hwaddr_flag = 0;
/* target MAC address in Ethernet header */
unsigned char trg_hwaddr[ETH_ALEN];
int trg_hwaddr_flag = 0;
/* source MAC address in ARP header */
unsigned char src_arp_hwaddr[ETH_ALEN];
int src_arp_hwaddr_flag = 0;
/* target MAC address in ARP header */
unsigned char trg_arp_hwaddr[ETH_ALEN];
int trg_arp_hwaddr_flag = 0;
struct in_addr src_ipaddr;
int src_ipaddr_flag = 0;
int at_once = 0;
#define MAX_IPADDR_NUMBER 1024
struct in_addr trg_ipaddr[MAX_IPADDR_NUMBER];
int trg_ipaddr_count = 0;
int trg_ipaddr_flag = 0;
const unsigned char broadcast[ETH_ALEN] = {0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF};
struct in_addr zero = {0x00000000};
struct sockaddr_ll iaddr;
char* print_hw_addr(const char* addr);
char* print_ip_addr(const char* addr);
char* print_arp_packet(struct arp_packet* pkt);
int read_hw_addr(char* buf, const char* str);
int read_ip_addr(struct in_addr* in_addr, const char* str);
char* programm_name = NULL;
void usage()
{
fprintf(stderr, "Usage: %s <-U -i <src_ip_addr> | -D -e <trg_ip_addr> "
"[-e <trg_ip_addr>] ...> [-c <count>] [-w <timeout>] "
"interface_name\n", programm_name);
exit(EXC_USAGE);
}
void parse_options (int argc, char **argv)
{
#define read_hw(addr) \
({ \
if (!strcmp(optarg, "broadcast")) \
memcpy(addr, broadcast, ETH_ALEN); \
else if (read_hw_addr(addr, optarg) < 0) \
usage(); \
addr##_flag = 1; \
})
#define read_ip(addr) \
({ \
if (read_ip_addr(&addr, optarg) < 0) \
usage(); \
addr##_flag = 1; \
})
char c;
static char short_options[] = "UDQPc:w:s:t:S:T:i:e:ov";
static struct option long_options[] =
{
{"update", 0, NULL, 'U'},
{"detect", 0, NULL, 'D'},
{"request", 0, NULL, 'Q'},
{"reply", 0, NULL, 'P'},
{"count", 1, NULL, 'c'},
{"timeout", 1, NULL, 'w'},
{"src-hw", 1, NULL, 's'},
{"trg-hw", 1, NULL, 't'},
{"src-arp", 1, NULL, 'S'},
{"trg-arp", 1, NULL, 'T'},
{"src-ip", 1, NULL, 'i'},
{"trg-ip", 1, NULL, 'e'},
{"at-once", 0, NULL, 'o'}
};
while ((c = getopt_long(argc, argv, short_options, long_options, NULL)) != -1)
{
switch (c)
{
case 'U':
if (cmd)
usage();
cmd = AR_UPDATE;
break;
case 'D':
if (cmd)
usage();
cmd = AR_DETECT;
break;
case 'Q':
if (cmd)
usage();
cmd = AR_REQUEST;
break;
case 'P':
if (cmd)
usage();
cmd = AR_REPLY;
break;
case 'c':
count = atoi(optarg);
if (!count)
usage();
break;
case 'w':
timeout = atoi(optarg);
break;
case 's':
read_hw(src_hwaddr);
break;
case 't':
read_hw(trg_hwaddr);
break;
case 'S':
read_hw(src_arp_hwaddr);
break;
case 'T':
read_hw(trg_arp_hwaddr);
break;
case 'i':
read_ip(src_ipaddr);
break;
case 'e':
if (trg_ipaddr_count >= MAX_IPADDR_NUMBER)
usage();
if (read_ip_addr(&trg_ipaddr[trg_ipaddr_count++], optarg) < 0)
usage();
trg_ipaddr_flag = 1;
break;
case 'o':
at_once = 1;
break;
case 'v':
debug_level ++ ;
break;
default:
usage();
}
}
argc -= optind;
argv += optind;
if (cmd == AR_NOTHING || argc != 1
|| (cmd == AR_DETECT && !trg_ipaddr_flag)
|| (cmd == AR_UPDATE && !src_ipaddr_flag))
usage();
iface = argv[0];
/* user have to choose something */
if (!cmd)
usage();
if (cmd == AR_DETECT
&& trg_ipaddr_count <= 1) {
/* for detection with no more then one target
we exit on first reply */
at_once = 1;
}
}
#define COMMON_ARP \
frame_type : htons(ETH_P_ARP), \
hw_type : htons(ARPHRD_ETHER), \
prot_type : htons(ETH_P_IP), \
hw_addr_size : ETH_ALEN, \
prot_addr_size : IP_ADDR_LEN
unsigned char real_hwaddr[ETH_ALEN];
struct in_addr real_ipaddr;
int init_device_addresses(int sock, const char* device)
{
struct ifreq ifr;
int ifindex;
short int ifflags;
memset(&ifr, 0, sizeof(ifr));
strncpy(ifr.ifr_name, device, IFNAMSIZ-1);
if (ioctl(sock, SIOCGIFINDEX, &ifr) != 0) {
logger(LOG_ERROR, "unknown iface %s : %m", device);
return -1;
}
ifindex = ifr.ifr_ifindex;
if (ioctl(sock, SIOCGIFFLAGS, &ifr) != 0) {
logger(LOG_ERROR, "ioctl(SIOCGIFFLAGS) : %m");
return -1;
}
ifflags = ifr.ifr_flags;
if (!(ifflags & IFF_UP)) {
logger(LOG_ERROR, "iface '%s' is down", device);
return -1;
}
if (ifflags & (IFF_NOARP|IFF_LOOPBACK)) {
logger(LOG_ERROR, "iface '%s' is not ARPable", device);
return -1;
}
if (ifflags & IFF_SLAVE) {
logger(LOG_ERROR, "iface '%s' is slave", device);
return -1;
}
/* get interface HW address */
if (ioctl(sock, SIOCGIFHWADDR, &ifr) != 0) {
logger(LOG_ERROR, "can't get iface '%s' HW address : %m", device);
return -1;
}
memcpy(real_hwaddr, ifr.ifr_hwaddr.sa_data, ETH_ALEN);
if (ioctl(sock, SIOCGIFADDR, &ifr)) {
logger(LOG_ERROR, "can't get iface '%s' address : %m", device);
return -1;
}
memcpy(&real_ipaddr, ifr.ifr_addr.sa_data + 2, IP_ADDR_LEN);
logger(LOG_DEBUG, "got addresses hw='%s', ip='%s'",
print_hw_addr(real_hwaddr),
print_ip_addr((char*) &real_ipaddr));
/* fill interface description struct */
iaddr.sll_ifindex = ifindex;
iaddr.sll_family = AF_PACKET;
iaddr.sll_protocol = htons(ETH_P_ARP);
memcpy(iaddr.sll_addr, real_hwaddr, iaddr.sll_halen = ETH_ALEN);
return 0;
}
int sock;
/* sent packet */
struct arp_packet pkt;
void create_arp_packet(struct arp_packet* pkt)
{
#define set_ip(to, from) (memcpy(to, from, IP_ADDR_LEN))
#define set_hw(to, from) (memcpy(to, from, ETH_ALEN))
#define check(check, two) (check##_flag ? check : two)
*pkt = ((struct arp_packet)
{COMMON_ARP, op : htons(cmd == AR_REPLY ? REPLY : REQUEST)});
set_ip(pkt->sndr_ip_addr,
(src_ipaddr_flag ? &src_ipaddr : &real_ipaddr));
set_hw(pkt->targ_hw_addr, check(trg_hwaddr, broadcast));
set_hw(pkt->src_hw_addr, check(src_hwaddr, real_hwaddr));
set_hw(pkt->rcpt_hw_addr, check(trg_arp_hwaddr, pkt->targ_hw_addr));
set_hw(pkt->sndr_hw_addr, check(src_arp_hwaddr, pkt->src_hw_addr));
/* special case 'cause we support multiple receipient ip addresses
we 'll setup 'pkt.rcpt_ip_addr' separately for each specified
(-e option) 'trg_ipaddr'
*/
if (trg_ipaddr_flag) {
/* at least one trg_ipaddr is specified */
set_ip(pkt->rcpt_ip_addr, &trg_ipaddr[0]);
} else {
set_ip(pkt->rcpt_ip_addr, &real_ipaddr);
/* set 'trg_ipaddr' for checking incomming packets */
set_ip(&trg_ipaddr[0], &real_ipaddr);
trg_ipaddr_count = 1;
}
#undef set_ip
}
void set_trg_ipaddr(struct arp_packet* pkt, const struct in_addr ipaddr)
{
#define set_ip(to, from) (memcpy(to, &from, IP_ADDR_LEN))
set_ip(pkt->rcpt_ip_addr, ipaddr);
#undef set_ip
}
int recv_response = 0;
void finish()
{
switch (cmd)
{
case AR_DETECT:
case AR_UPDATE:
exit((recv_response) ? EXC_RECV : EXC_OK);
case AR_REQUEST:
exit((recv_response) ? EXC_OK : EXC_NORECV);
case AR_REPLY:
exit(EXC_OK);
default:
exit(EXC_SYS);
}
}
int recv_pack(unsigned char *buf, int len, struct sockaddr_ll *from)
{
int rc = -1;
int i;
struct arp_packet * recv_pkt = (struct arp_packet*) buf;
if (recv_pkt->frame_type != htons(ETH_P_ARP)) {
logger(LOG_ERROR, "unknown eth frame type : %#x", recv_pkt->frame_type);
goto out;
}
if (recv_pkt->op != htons(REQUEST)
&& recv_pkt->op != htons(REPLY)) {
logger(LOG_ERROR, "unknown arp packet type : %#x", recv_pkt->op);
goto out;
}
for (i = 0; i < trg_ipaddr_count; i ++) {
// check for all sent packets addresses
if (memcmp(&trg_ipaddr[i], recv_pkt->sndr_ip_addr, IP_ADDR_LEN))
continue;
logger(LOG_DEBUG, "recv packet %s", print_arp_packet(recv_pkt));
logger(LOG_INFO, "%s is detected on another computer : %s",
print_ip_addr((char*) &trg_ipaddr[i]),
print_hw_addr(recv_pkt->sndr_hw_addr));
recv_response++;
/* finish on first response */
if (at_once)
finish();
}
/* unknown packet */
logger(LOG_DEBUG, "recv unknown packet %s", print_arp_packet(recv_pkt));
rc = 0;
out:
return rc;
}
void sender(void)
{
int i;
if (count-- == 0)
finish();
/* send multiple packets */
for (i = 0; i < trg_ipaddr_count; i ++) {
set_trg_ipaddr(&pkt, trg_ipaddr[i]);
logger(LOG_DEBUG, "send packet: %s", print_arp_packet(&pkt));
if(sendto(sock, &pkt,sizeof(pkt), 0,
(struct sockaddr*) &iaddr, sizeof(iaddr)) < 0) {
logger(LOG_ERROR, "sendto : %m");
exit(EXC_SYS);
}
}
if (count == 0
&& cmd != AR_DETECT
&& cmd != AR_REQUEST)
finish();
alarm(timeout);
}
void set_signal(int signo, void (*handler)(void))
{
struct sigaction sa;
memset(&sa, 0, sizeof(sa));
sa.sa_handler = (void (*)(int))handler;
sa.sa_flags = SA_RESTART;
sigaction(signo, &sa, NULL);
}
int main(int argc,char** argv)
{
programm_name = argv[0];
parse_options (argc, argv);
sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ARP));
if (sock < 0)
{
logger(LOG_ERROR, "socket : %m");
exit(EXC_SYS);
}
if (init_device_addresses(sock, iface) < 0)
exit(EXC_SYS);
create_arp_packet(&pkt);
set_signal(SIGALRM, sender);
sender();
while(1)
{
sigset_t sset, osset;
char packet[4096];
struct sockaddr_ll from;
int alen = sizeof(from);
int cc;
cc = recvfrom(sock, packet, sizeof(packet), 0, (struct sockaddr *)&from, &alen);
if (cc < 0)
{
logger(LOG_ERROR, "recvfrom : %m");
continue;
}
sigemptyset(&sset);
sigaddset(&sset, SIGALRM);
//sigaddset(&sset, SIGINT);
sigprocmask(SIG_BLOCK, &sset, &osset);
recv_pack(packet, cc, &from);
sigprocmask(SIG_SETMASK, &osset, NULL);
}
exit(EXC_OK);
}
static char* get_buf()
{
static int num = 0;
static char buf[10][1024];
return buf[num = (num+1) % 10];
}
char* print_arp_packet(struct arp_packet* pkt)
{
char* point = get_buf();
sprintf(point, "eth '%s' -> eth '%s'; arp sndr '%s' '%s'; %s; arp recipient '%s' '%s'",
print_hw_addr(pkt->src_hw_addr),
print_hw_addr(pkt->targ_hw_addr),
print_hw_addr(pkt->sndr_hw_addr),
print_ip_addr(pkt->sndr_ip_addr),
(pkt->op == htons(REQUEST)) ? "request" : (pkt->op == htons(REPLY) ? "reply" : "unknown"),
print_hw_addr(pkt->rcpt_hw_addr),
print_ip_addr(pkt->rcpt_ip_addr));
return point;
}
/* get MAC address in user form */
char* print_hw_addr(const char* addr)
{
int i;
char* point = get_buf();
char* current = point;
for (i = 0; i < ETH_ALEN; i++)
point += sprintf(point, i == (ETH_ALEN-1) ? "%02x" : "%02x:", (unsigned char) addr[i]);
return current;
}
/* get IP address in user form */
char* print_ip_addr(const char* addr)
{
int i;
char* point = get_buf();
char* current = point;
for (i = 0; i < IP_ADDR_LEN; i++)
point += sprintf(point, i == (IP_ADDR_LEN-1) ? "%u" : "%u.", (unsigned char) addr[i]);
return current;
}
/* trasform user -> computer freindly MAC address */
int read_hw_addr(char* buf, const char* str)
{
int rc = -1;
int i;
for(i = 0; i < 2*ETH_ALEN; i++)
{
char c, val;
c = tolower(*str++);
if(!c)
goto out;
if(isdigit(c))
val = c - '0';
else if (c >= 'a' && c <= 'f')
val = c - 'a' + 10;
else
goto out;
if (i % 2)
{
*buf++ |= val;
if(*str == ':')
str++;
}
else
*buf = val << 4;
}
rc = 0;
out:
return rc;
}
/* trasform user -> computer freindly IP address */
int read_ip_addr(struct in_addr* in_addr, const char* str)
{
in_addr->s_addr=inet_addr(str);
if(in_addr->s_addr == -1)
return -1;
return 0;
}