Форум программистов [Powered by Invision Power Board]

Код

#include <stdlib.h>
#include <stdio.h>
#include <fcntl.h>
#include <unistd.h>
#include <netinet/in.h>
#include <errno.h>
#include <string.h>
#include <linux/if_ether.h>
#include <linux/ip.h>
#include <linux/tcp.h>


int main(void)
{
    extern int errno;
    int            foutput; // дескриптор файла дампа
    int            rec; // получено из сокета
    int            size_data; // объем полезных данных
    struct iphdr ip; // IP заголовок
    struct tcphdr tcp; // TCP заголовок
    int    raw_socket; // дескриптор сокета
    
    __u8 buff[ETH_FRAME_LEN]; // буфер ETH пакета
    __u8  buff_tcp[1460]; // буфер данных

    foutput = open("dump-rtmp", O_WRONLY | O_TRUNC | O_CREAT  , 0644);
    if (foutput < 0)
    {
        fprintf(stderr, "Cannot open file 'dump-rtmp' -%s\n", strerror(errno));
        exit(1);
    }

    if ((raw_socket = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL))) <0)
    {
        fprintf(stderr,"Cannot open socket\n");
        exit(1);
    }

    for (;;)
    {
        rec = recvfrom(raw_socket, (char *)buff, ETH_FRAME_LEN, 0, NULL, NULL);
        
        if (rec < 0 || rec > ETH_FRAME_LEN)
        {
            fprintf(stderr,"recvfrom - error\n");
            return -1;
        }

        memcpy((void *)&ip, buff + ETH_HLEN, sizeof(struct iphdr));

        if ((ip.version) == 4)
        {
            memcpy((void *)&tcp, buff + ETH_HLEN + ip.ihl * 4, sizeof(struct tcphdr));

            if ((ip.protocol == IPPROTO_TCP) & (ntohs(tcp.dest) == 1935))
            {
                size_data = (rec-tcp.doff*4-ip.ihl*4-ETH_HLEN);
                memcpy((void *)&buff_tcp, (buff + ETH_HLEN + ip.ihl*4 + tcp.doff*4), size_data);

                if (size_data > 0)
                {
                    printf("TCP пакет - %u\n", size_data);
                    write(foutput, buff_tcp, size_data);
                }
            }
        }
    }
    exit(0);
}