Linux 发送 ICMP 数据

发送ICMP数据包:

  以下程序必须在root权限下编译和运行

#include <stdio.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <strings.h>
#include <netinet/ip_icmp.h>
#include <stdlib.h>
#include <arpa/inet.h>

void send_echo_req(int sockfd, struct sockaddr_in *dstaddr);  
uint16_t in_cksum(uint16_t *addr, int len);  
void recv_echo_reply(int sockfd);  

void err_sys(const char *errmsg)  
{
    perror(errmsg);
    exit(1);  
}

int main(int argc, char **argv)  
{
    int sockfd;
    struct sockaddr_in dstaddr;  

    if ((sockfd = socket(PF_INET, SOCK_RAW, IPPROTO_ICMP)) == -1)
        err_sys("socket");

    bzero(&dstaddr, sizeof(dstaddr));
    dstaddr.sin_family = AF_INET;
    dstaddr.sin_port = htons(0);

    if (inet_pton(AF_INET, argv[1], &dstaddr.sin_addr) <= 0)
        err_sys("inet_pton");

    send_echo_req(sockfd, &dstaddr);
    recv_echo_reply(sockfd);
    exit(0);  
}  

//发送icmp请求，指定请求地址。请求[struct icmp]数据包
void send_echo_req(int sockfd, struct sockaddr_in *dstaddr)  
{
    char buf[100];
    size_t len = sizeof(struct icmp);
    struct icmp *icmp; //icmp数据包
    socklen_t dstlen = sizeof(struct sockaddr_in);

    bzero(buf, sizeof(buf));
    icmp = (struct icmp *)buf;
    icmp->icmp_type = ICMP_ECHO;//8
    icmp->icmp_code = 0;
    icmp->icmp_id = getpid();
    icmp->icmp_seq = 1;
    icmp->icmp_cksum = in_cksum((uint16_t *) icmp, sizeof(struct icmp));
    if (sendto(sockfd, buf, len, 0, (struct sockaddr *)dstaddr, dstlen) == -1)
        err_sys("sendto");  
}

//接受icmp回应，返回IP数据包,[struct ip],通过返回源地址，可获取路由信息
void recv_echo_reply(int sockfd)  
{
    char buf[100];
    ssize_t n;
    struct ip *ip;
    struct icmp *icmp;
    while (1) {
        alarm(5); /* set timeout */
        if ((n = read(sockfd, buf, sizeof(buf))) == -1)
            err_sys("read");
        ip = (struct ip *)buf;          //转化成ip数据报

        if (ip->ip_p != IPPROTO_ICMP) { //如果是icmp协议
            fprintf(stderr, "protocol error.");
            exit(1);
        }
        //定位到icmp报文头
        icmp = (struct icmp *)(buf + sizeof(struct ip));

        if (icmp->icmp_type == ICMP_ECHOREPLY) {
            if (icmp->icmp_id != getpid()) {
                fprintf(stderr, "not this process.");
                exit(1);
            } else {
                printf("destination host is alive.\n");
                break;
            }
        }
    }  
}

//
uint16_t in_cksum(uint16_t *addr, int len)  
{
    int nleft = len;
    uint32_t sum = 0;
    uint16_t *w = addr;
    uint16_t 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);  
}