Linux下原始套接字实现UDP客户端

通过C代码结合Linux下的socket编程，采用原始套接字技术实现UDP客户端，代码如下：

#include <stdio.h>

#include <string.h>

#include <stdint.h>

#include <unistd.h>

#include <errno.h>

#include <sys/types.h>

#include <sys/socket.h>

#include <netinet/udp.h>

#include <netinet/ip.h>

#include <linux/in.h>

#include <arpa/inet.h>

#define DEST_IP "127.0.0.1"

#define SPORT (2222)

#define DEST_PORT (12345)

#define IP_HDRLEN (20)

#define UDP_HDRLEN (8)

#define BUFFER_SIZE 1024

void statistics(char *str) {

    printf("receive: %s", str);

}

uint32_t create_udp_pack(char *buf, const char *payload, uint32_t payload_len) {

    struct udphdr *uh = (struct udphdr *)buf;

    uh->uh_sport = htons(SPORT);

    uh->uh_dport = htons(DEST_PORT);

    uh->uh_ulen = htons(UDP_HDRLEN + payload_len);

    uh->uh_sum = 0;

    memcpy(buf + sizeof(struct udphdr), payload, payload_len);

    return UDP_HDRLEN + payload_len;

}

void parse_pack(char *buf, unsigned char* valid) {

    uint16_t sport1;

    uint16_t dport;

    struct iphdr *iph = (struct iphdr *)buf;

    struct udphdr *uh = (struct udphdr *)(buf + sizeof(struct iphdr));

    sport1 = ntohs(uh->uh_sport);

    dport = ntohs(uh->uh_dport);

    uint32_t offset = sizeof(struct iphdr) + sizeof(struct udphdr);

    if (sport1 != SPORT && sport1 == DEST_PORT) {

        statistics(buf + offset);

        (*valid) = 1;

    }

}

int main(int argc , char *argv[]) {

    int sockfd;

    char message[BUFFER_SIZE] = {0};

    char send_buf[BUFFER_SIZE] = { 0 };

    char buffer[BUFFER_SIZE] = {0}; //recv buffer

    uint32_t slen = 1;

    int ret = 0;

    int on = 0;

    unsigned char valid = 0;

    sockfd = socket(AF_INET, SOCK_RAW, IPPROTO_UDP);//create socket raw

    if (sockfd == -1) {

        perror("socket error");

        return -1;

    }

    //ret = setsockopt(sockfd, IPPROTO_IP, IP_HDRINCL, &on, sizeof(on));

    struct sockaddr_in dest_addr;

    bzero(&dest_addr, sizeof(dest_addr));

    dest_addr.sin_family = AF_INET;

    dest_addr.sin_addr.s_addr = inet_addr(DEST_IP);

    dest_addr.sin_port = htons(DEST_PORT);

    while(1) {

        //read user input

        printf("input:");
        memset(message, 0, sizeof(message));

        fgets(message, BUFFER_SIZE, stdin);

        memset(send_buf, 0, sizeof(send_buf));

        slen = create_udp_pack(send_buf, message, strlen(message));

        //send msg

        ret = sendto(sockfd, send_buf, slen, 0, (struct sockaddr *)&dest_addr, sizeof(dest_addr));

        if (ret <= 0) {

            perror("sendto");

            exit(1);

        }

        //check exit

        if (strcmp(message, "exit\n") == 0) {

            printf("Bye\n");

            break;

        }

        //recv msg

Again:

        memset(buffer, 0, sizeof(buffer));

        if (recvfrom(sockfd, buffer, sizeof(buffer), 0, NULL, NULL) < 0) {

            perror("recvfrom");

            exit(1);

        }

        valid = 0;

        parse_pack(buffer, &valid);

        if (valid == 0) {

           goto Again;

        }

    }

    close(sockfd);

    return 0;

}