《本文源代码下载》
目录
源代码
icmpping.c
/**
* 实现 ICMP ping功能，检测 目的 IP地址通路，并计算时延
*
* 作者： RToax
* 日期： 2020年10月12日
*/
#include <stdio.h>
#include <signal.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/timerfd.h>
#include <sys/time.h>
#include <unistd.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <netinet/ip_icmp.h>
#include <netdb.h>
#include <errno.h>
#include <malloc.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#include <pthread.h>
#include "icmpping.h"
#ifndef offsetof
#define offsetof(type, member) ( (int) & ((type*)0) -> member )
#endif
#define ICMP_BUF_SIZE 256
#define ICMP_RECVBUF_SIZE (50 * 1024)
#define ICMP_MAGIC_ID 0xf1f2
#define ICMP_PROTO_NAME "icmp"
#define ICMP_DATA "Rong Tao's ICMP Ping"
#define ICMP_PING_SUCC __ICMP_PING_SUCC
#define ICMP_PING_FAIL __ICMP_PING_FAIL
#define ICMP_LOG(fmt...) //printf(fmt)
#define ICMP_LOG_ERR(fmt...) fprintf(stderr, fmt);
typedef enum {
__ICMP_PING_FAIL = ICMPPING_FAIL,
__ICMP_PING_SUCC = ICMPPING_SUCC,
}icmp_ping_rlst_t;
/* type of icmp ping */
typedef struct __icmp_ping_s {
int sockfd;
struct sockaddr_in dst_addr, from_addr;
int ntx, nrx, ntry; //number of send pkg and recv pkg
int tx, rx;
int timeout_s; //time out second
int (*log)(const char *fmt, ...);
int rslt; //__ICMP_PING_XXXX
}icmp_ping_t;
/* functions */
static icmp_ping_t* icmp_ping_init(const char *ipv4, int timeout_sec, int ntry, int (*log)(const char *fmt, ...));
int icmp_ping(const char *ipv4, unsigned int timeout, int ntry, int (*log)(const char *fmt, ...), long int *latency);
static void icmp_ping_finish (icmp_ping_t *ping);
static int icmp_ping_destroy(icmp_ping_t* ping);
static int icmp_socket();
static int icmp_dst_addr(const char *addrHost, struct sockaddr_in * dst_addr);
static unsigned short icmp_gen_chksum(unsigned short * data, int len);
static int icmp_pkg_pack(void *buffer, int pack_no, const void *data, int data_size);
static int icmp_send_pkg(icmp_ping_t *ping, const void *data, int size);
static int icmp_recv_pkg(icmp_ping_t *ping, void *recvbuf, int size);
static int icmp_pkg_unpack(icmp_ping_t *ping, char * buf, int len);
static void __icmp_timeval_sub(struct timeval * out, struct timeval * in);
/* alogrithm of checksum */
static unsigned short icmp_gen_chksum(unsigned short * data, int len)
{
int nleft = len;
int sum = 0;
unsigned short *w = data;
unsigned short answer = 0;
while (nleft > 1)
{
sum += *w++;
nleft -= 2;
}
if (nleft == 1)
{
* (unsigned char *) (&answer) = *(unsigned char *)w;
sum += answer;
}
sum = (sum >> 16) + (sum & 0xffff);
sum += (sum >> 16);
answer = ~sum;
return answer;
}
/* create a socke to icmp */
static int icmp_socket()
{
int sockfd;
int size = ICMP_RECVBUF_SIZE;
struct protoent * protocol = NULL;
if ((protocol = getprotobyname(ICMP_PROTO_NAME)) == NULL)
{
ICMP_LOG_ERR("getprotobyname error.\n\r");
return -1;
}
if ((sockfd = socket(AF_INET, SOCK_RAW, protocol->p_proto)) < 0)
{
ICMP_LOG_ERR("socket error.\n\r");
return -1;
}
if(setsockopt(sockfd, SOL_SOCKET, SO_RCVBUF, &size, sizeof(size))!=0)
{
ICMP_LOG_ERR("setsockopt SO_RCVBUF error.\n\r");
close(sockfd);
return -1;
}
#if 1
struct timeval timeout = {1,0}; //sec
if(setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, (char*)&timeout, sizeof(struct timeval))!=0)
{
ICMP_LOG_ERR("setsockopt SO_RCVTIMEO error.\n\r");
close(sockfd);
return -1;
}
#endif
return sockfd;
}
/* get dst address */
static int icmp_dst_addr(const char *addrHost, struct sockaddr_in * dst_addr)
{
struct hostent * host = NULL;
unsigned long inaddr = 0;
bzero(dst_addr, sizeof(struct sockaddr_in));
dst_addr->sin_family = AF_INET;
if ((inaddr = inet_addr(addrHost)) == INADDR_NONE)
{
if ((host = gethostbyname(addrHost)) == NULL)
{
ICMP_LOG_ERR("gethostbyname error.\n\r");
return (-1);
}
memcpy((char *) &dst_addr->sin_addr, host->h_addr, host->h_length);
}
else
{
memcpy((char *) &dst_addr->sin_addr, (char *) &inaddr, sizeof(dst_addr->sin_addr));
}
return 0;
}
/* set icmp hdr */
static int icmp_pkg_pack(void *buffer, int pack_no, const void *data, int data_size)
{
int i, packsize = 0;
struct icmp * icmp = malloc(sizeof(struct icmp));
icmp->icmp_type = ICMP_ECHO;
icmp->icmp_code = 0;
icmp->icmp_cksum = 0;
icmp->icmp_seq = htons(pack_no);
icmp->icmp_id = htons(ICMP_MAGIC_ID);
gettimeofday((struct timeval *) &icmp->icmp_data, NULL);
memcpy(buffer, icmp, sizeof(struct icmp));
packsize += sizeof(struct icmp);
if(data && data_size)
{
memcpy(buffer+packsize, data, data_size);
packsize += data_size;
}
return packsize;
}
/* send icmp package */
static int icmp_send_pkg(icmp_ping_t *ping, const void *data, int size)
{
int packetsize;
unsigned short checksum = 0;
int n = 0;
char pkg_buffer[ICMP_BUF_SIZE];
packetsize = icmp_pkg_pack(pkg_buffer, ping->ntx, data, size);
checksum = icmp_gen_chksum((unsigned short *)pkg_buffer, packetsize);
#define ICMP_PKG_CHKSUM_OFFSET offsetof(struct icmp, icmp_cksum)
#define ICMP_PKG_CHKSUM_SIZE 2
memcpy(pkg_buffer + ICMP_PKG_CHKSUM_OFFSET, &checksum, ICMP_PKG_CHKSUM_SIZE);
if ((n = sendto(ping->sockfd, pkg_buffer, packetsize, 0,
(struct sockaddr *) &ping->dst_addr, sizeof(struct sockaddr_in)))< 0)
{
ICMP_LOG_ERR("sendto error. n = %d\n\r", n);
return 0;
}
ping->ntx++;
return n;
}
/* send icmp package */
static int icmp_recv_pkg(icmp_ping_t *ping, void *recvbuf, int size)
{
int n, fromlen;
fromlen = sizeof(struct sockaddr_in);
if ((n = recvfrom(ping->sockfd, recvbuf, size, 0,
(struct sockaddr *) &ping->from_addr, &fromlen)) < 0)
{
ICMP_LOG_ERR("recvfrom error.n = %d\n\r", n);
return 0;
}
ping->nrx++;
return n;
}
/* unpack icmp pkg */
static int icmp_pkg_unpack(icmp_ping_t *ping, char * buf, int len)
{
int i, iphdrlen;
struct ip * ip = NULL;
struct icmp * icmp = NULL;
struct timeval * tvsend = NULL;
double rtt;
ip = (struct ip *)buf;
iphdrlen = ip->ip_hl << 2;
icmp = (struct icmp *) (buf + iphdrlen);
len -= iphdrlen;
if (len < 8)
{
ICMP_LOG_ERR("ICMP packet\'s length is less than 8\n\r");
return - 1;
}
/* ensure icmp reply is mine response msg */
if ((icmp->icmp_type == ICMP_ECHOREPLY) && (icmp->icmp_id = ntohs(ICMP_MAGIC_ID)))
{
tvsend = (struct timeval *)icmp->icmp_data;
struct timeval tvrecv;
gettimeofday(&tvrecv, NULL);
__icmp_timeval_sub(&tvrecv, tvsend);
rtt = tvrecv.tv_sec * 1000 + tvrecv.tv_usec / 1000;
if(ping->log)
ping->log("%d byte from %s: icmp_req=%u ttl=%d rtt=%.3f ms\n\r",
len, inet_ntoa(ping->from_addr.sin_addr), ntohs(icmp->icmp_seq), ip->ip_ttl, rtt);
}
else
{
return - 1;
}
return 0;
}
static void __icmp_timeval_sub(struct timeval * out, struct timeval * in)
{
if ((out->tv_usec -= in->tv_usec) < 0)
{
--out->tv_sec;
out->tv_usec += 1000000;
}
out->tv_sec -= in->tv_sec;
}
/**********************************************************************************************************************/
/* icmp ping with timeout */
static void icmp_ping_finish (icmp_ping_t *ping)
{
if(ping->log)
{
ping->log("\n--- ping statistics ---\n\r");
ping->log("%d packets transmitted, %d received, %d%% lost\n\r",
ping->ntx, ping->nrx,
(ping->ntx - ping->nrx) / ping->ntx * 100);
}
if(ping->ntx == ping->ntry)
ping->rslt = __ICMP_PING_SUCC;
return;
}
static icmp_ping_t* icmp_ping_init(const char *ipv4, int timeout_sec, int ntry, int (*log)(const char *fmt, ...))
{
icmp_ping_t *ping = (icmp_ping_t *)malloc(sizeof(icmp_ping_t));
if(!ping) return NULL;
memset(ping, 0x00, sizeof(icmp_ping_t));
ping->sockfd = icmp_socket();
ping->timeout_s = timeout_sec;
ping->ntry = ntry;
ping->log = log;
icmp_dst_addr(ipv4, &ping->dst_addr);
if(ping->log)
ping->log("PING %s(%s): %d bytes data in ICMP packets.\n\r",
ipv4, inet_ntoa(ping->dst_addr.sin_addr), sizeof(ICMP_DATA));
return ping;
}
static int icmp_ping_destroy(icmp_ping_t* ping)
{
if(ping)
{
close(ping->sockfd);
free(ping);
}
ping = NULL;
}
static long int timeval_sub_usec(const struct timeval *before, const struct timeval *after)
{
long int usec_diff = 0;
usec_diff = (after->tv_sec-before->tv_sec)*1000000 + after->tv_usec-before->tv_usec;
return usec_diff;
}
/* PING */
int icmp_ping(const char *ipv4, unsigned int timeout, int ntry, int (*log)(const char *fmt, ...), long int *latency)
{
icmp_ping_t *ping = icmp_ping_init(ipv4, timeout, ntry, log);
struct timeval starttime, endtime;
long int usec_diff = 0;
char pkg_buffer[ICMP_BUF_SIZE];
ping->rslt = __ICMP_PING_FAIL;
while(ping->ntx < ping->ntry)
{
gettimeofday(&starttime, NULL);
ping->tx = icmp_send_pkg(ping, ICMP_DATA, sizeof(ICMP_DATA));
if(ping->tx < 8)
{
ping->rslt = __ICMP_PING_FAIL;
break;
}
ping->rx = icmp_recv_pkg(ping, pkg_buffer, ping->tx);
if(ping->rx<8)
{
ping->rslt = __ICMP_PING_FAIL;
break;
}
gettimeofday(&endtime, NULL);
usec_diff += timeval_sub_usec(&starttime, &endtime);
// printf("usec_diff = %ld\n", usec_diff);
if (icmp_pkg_unpack(ping, pkg_buffer, ping->rx) == -1)
continue;
//sleep(1);
}
if(latency) *latency = (long int)(usec_diff / ping->ntry);
icmp_ping_finish(ping);
icmp_ping_destroy(ping);
return ping->rslt;
}
icmpping.h
/**
* 实现 ICMP ping功能，检测 目的 IP地址通路，并计算时延
*
* 作者： RToax
* 日期： 2020年10月12日
*/
#ifndef __ICMP_PING_H
#define __ICMP_PING_H 1
/* icmp_ping 返回值 */
#define ICMPPING_SUCC 0 /* 成功 */
#define ICMPPING_FAIL 1 /* 失败 */
/**
* icmp_ping - icmp ping
*
* param[in] ipv4: 目的IPv4地址
* param[in] timeout: 超时设置，若超出设定值，且ping不通情况下，函数自动返回
* param[in] ntry: ping的次数，发 req包数量
* param[in] log: 打印调试信息，若为 NULL则不打印
* param[in] latency: 计算平均时延，单位 微妙 micro second（参见结构 struct timeval）
* 该参数 填 NULL 时，不计算时延
*
* return 成功返回 ICMPPING_SUCC， 失败返回 ICMPPING_FAIL
*/
int icmp_ping(const char *ipv4, unsigned int timeout, int ntry,
int (*log)(const char *fmt, ...), long int *latency);
#endif /*<__ICMP_PING_H>*/
main.c
#include <stdio.h>
#include "icmpping.h"
int main(int argc, char * argv[])
{
if (argc < 2)
{
printf("Usage: %s hostname/IP address\n\r", argv[0]);
return (-1);
}
long int latency;
int rslt = icmp_ping(argv[1], 1, 3, NULL, &latency);
if(rslt == ICMPPING_SUCC)
printf("ping \033[1;32m%s\033[m is OK. latency = %ld microseconds\n", argv[1], latency);
else
printf("ping \033[1;31m%s\033[m is not OK.\n", argv[1]);
return 0;
}
makefile
# 编译makefile
# RToax 2020年10月12日
ICMP_INC := -I ./
ICMP_SRC := icmpping.c
ICMP_DEP := icmpping.o
ICMP_TARGET=icmpping
all:$(ICMP_TARGET)
$(ICMP_DEP): $(ICMP_SRC)
gcc -c $(ICMP_SRC) $(ICMP_INC) -o $(ICMP_DEP)
$(ICMP_TARGET): $(ICMP_DEP)
gcc $(ICMP_DEP) main.c $(ICMP_INC) -o $(ICMP_TARGET)
clean:
rm -f $(ICMP_DEP) $(ICMP_TARGET)
测试结果
[root@localhost icmpping]# ./icmpping 0
ping 0 is OK. latency = 18 microseconds
[root@localhost icmpping]# ./icmpping 127.0.0.1
ping 127.0.0.1 is OK. latency = 15 microseconds
[root@localhost icmpping]# ./icmpping 10.170.6.24
ping 10.170.6.24 is OK. latency = 231 microseconds
[root@localhost icmpping]# ./icmpping 10.37.8.37
ping 10.37.8.37 is OK. latency = 20744 microseconds
1511