af_packet raw server client program with Poll system call

  • In this program, you are going to learn

  • How to create a Socket ?

  • How to bind a socket ?

  • How to send a data ?

  • How to recv a data ?

Let us answer few basic questions in this socket

What does socket(AF_PACKET, SOCK_RAW, htons(ETH_P_ALL)) do?

How can I handle errors when creating a raw socket for packet capturing?

Are there specific privileges required to create a raw packet socket?

What kind of errors can occur when using a raw packet socket?

How do I handle errors when binding a raw packet socket?

What is the significance of error code EPERM when dealing with raw packet sockets?

How do I handle errors when using recv and send to capture packets and send packets with a raw socket?

Is it important to close the raw packet socket on error?

How do I handle timeouts when capturing packets with a raw packet socket?

Is it possible to capture packets on a specific network interface with a raw packet socket?

How do I handle errors related to socket file descriptor management?

What is the purpose of the poll system call?

How does poll differ from poll in terms of usability?

What types of file descriptors can be monitored using poll?

What is the significance of the timeout parameter in the select function?

How do you handle errors when using the select system call?

How does select handle a set of file descriptors with different states (e.g., reading, writing, exception)?

How does select Checking Ready File Descriptors?

What does it mean if select returns 0?

https://www.plantuml.com/plantuml/svg/TP5VIyCm5CNVyockBoFHADjhXx0wEXmhRdGJGP7IFpSic3KZDwlefpVT2Zh5F2JSdFnEScucv0ld3pl3BlmMTmYqBoeqHScE0el8Lc_eUJpBL_5qaMmah9VJHPx5JnAstZR4ayrzlihZD1LYpA9IDpKdT2sw_CHAuEJTeVBGrKLTkqinw4zqbGJIdsWL_x5tj58rlGnVRrGDjt3c-ArXYusdeA-MQJf_6BFthJO8_BZmaH2W2VW7ac3HtXh30oTX9E5w56xG2dXlv7gdGUya8BIZwL7eHkImQdydAmzAOKXmzpYR9Lc-dZyd4eR1-eGG_ct6lmWQ2Yzlh621qggR-pJJJPPULI42RwskHMKiuRbjmCdRFPi4A7pc5m==
  • There are many functions used in socket. We can classify those functions based on functionalities.

    • Create Socket

    • Bind Socket

    • Poll

    • Recv data_packet

    • Send data_packet

    • Close socket

  • socket() is used to create a new socket. For example,

server_socket = socket(AF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
  • bind() is used to associate the socket with a specific address and port. For example,

ret = bind(server_socket, (struct sockaddr*)&server_addr, sizeof(server_addr));
  • poll() is used for monitoring multiple file descriptors to see if I/O is possible on any of them.

ret = poll(fds, 1, -1);
  • recv is used in network programming to receive data from a connected socket. For example,

len = recv(server_socket, buffer, BUFFER_SIZE, 0);
  • send is used in network programming to send data over a connected socket. For example,

ret = send(server_socket, buffer, BUFFER_SIZE, 0);
  • close is used to close the socket To free up system resources associated with the socket. For example,

(void)close(server_socket);
  • See the full program below,

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <signal.h>
#include <netpacket/packet.h>
#include <net/ethernet.h>
#include <net/if.h>
#include <poll.h>

#define BUFFER_SIZE 1024

int server_socket;

static void sigint_handler(int signo)
{
  (void)close(server_socket);
  sleep(2);
  printf("Caught sigINT!\n");
  exit(EXIT_SUCCESS);
}

void register_signal_handler(
int signum,
void (*handler)(int))
{
  if (signal(signum, handler) ==
  SIG_ERR) {
     printf("Cannot handle signal\n");
     exit(EXIT_FAILURE);
  }
}

int main(void) 
{
  int ret, i;
  struct sockaddr_ll 
  server_addr, 
  client_addr;
  struct pollfd fds[1];
  char buffer[BUFFER_SIZE];

  register_signal_handler(SIGINT,
  sigint_handler);

  server_socket = socket(AF_PACKET, 
                  SOCK_RAW, 
                  htons(ETH_P_ALL));

  if (server_socket < 0) {
    perror("socket");
    return -1;
  }

  memset(&server_addr, 0, 
  sizeof(server_addr));
  server_addr.sll_family = AF_PACKET;
  server_addr.sll_protocol = 
  htons(ETH_P_ALL);
  server_addr.sll_ifindex = 
  if_nametoindex("lo");

  ret = bind(server_socket, 
  (struct sockaddr*)&server_addr, 
  sizeof(server_addr));
  
  if (ret < 0) {
    perror("bind");
    (void)close(server_socket);
    return -2;
  }

  printf("RAW socket is listening\n");

  socklen_t client_addr_len = 
  sizeof(client_addr);

  memset(fds, 0, sizeof(fds));
  fds[0].fd = server_socket;
  fds[0].events = POLLIN;

  while (1) {
    ret = poll(fds, 1, -1);

    if (ret < 0) {
      perror("poll");
      break;
    }
 
    if (fds[0].revents & POLLIN) {
      memset(buffer, 0, 
      sizeof(buffer));
     ret = recv(server_socket, 
     buffer, BUFFER_SIZE, 0);
     if (ret < 0) {
	perror("recvfrom error");
        break;
     } else {
        buffer[ret] = '\0';
        printf("Received: %s\n", 
        buffer);
     }
     sleep(3);

     memset(buffer, 0,
     sizeof(buffer));
     strncpy(buffer, "HELLO",
     strlen("HELLO") + 1);
     buffer[strlen(buffer)
     + 1] = '\0';
	
     ret = send(server_socket, 
     buffer, BUFFER_SIZE, 0);

     if (ret == -1) {
        perror("sendto error");
        break;
      } else {
	printf("sending message = %s\n", 
	buffer);
      }
     } 
   }

  (void)close(server_socket);

  return 0;
}
 1$ gcc -o server server.c
 2
 3$ sudo ./server
 4
 5RAW socket is listening
 6Received: hello server!
 7sending message = HELLO
 8Received: hello server!
 9sending message = HELLO
10Received: HELLO
11sending message = HELLO
12Received: hello server!
13sending message = HELLO
14Received: hello server!
15sending message = HELLO
16Received: HELLO
17sending message = HELLO
18Received: hello server!
19^CCaught sigINT!
https://www.plantuml.com/plantuml/svg/ZP71IWCn48RlUOhnKXADqbvRXAvbYwMBBMr5K2HiinCQJ3SbILVqwPsj2rgycCE4cM7-xvza75CPqd7ls4NQuHuX7agDhjooKuCDezTlc7WsLSjiCiyt4jQBoLojiYS9k-JhoFFDlLggh2Y46B7XrjOLryvYdTJth0GUKpZg16rULbMu5Bs8eS6Wsan2j9_e3VzLEqcPAhxqNszC1RTmflZJmuQgaVhBHL7C7aRiVMST0ZzTV204s0Zy0wEWJV2li-tH62G7Tu_JQRvIwzbpBg7VmW_UEKuK2GC9rmCgMGEymuQEs-ku0kZnwWu4T8O1TVDl6Bf8dcil65cqfetT7fFDghZIP9uttbP2En_nN9R6O_87DgOX-iOl
  • There are many functions used in socket. We can classify those functions based on functionalities.

    • Create Socket

    • Bind Socket

    • Poll

    • Send data_packet

    • Recv data_packet

    • Close socket

  • socket is used to create a new socket. For example,

client_socket = socket(AF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
  • bind() is used to associate the socket with a specific address and port. For example,

ret = bind(client_socket, (struct sockaddr*)&server_addr, sizeof(server_addr));
  • poll() is used for monitoring multiple file descriptors to see if I/O is possible on any of them.

ret = poll(fds, 1, -1);
  • send is used in network programming to send data over a connected socket. For example,

ret = send(client_socket, buffer, BUFFER_SIZE, 0);
  • recv is used in network programming to receive data from a connected socket. For example,

len = recv(client_socket, buffer, BUFFER_SIZE, 0);
  • close is used to close the socket To free up system resources associated with the socket. For example,

(void)close(client_socket);
  • See the full program below,

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <signal.h>
#include <netpacket/packet.h>
#include <net/ethernet.h>
#include <net/if.h>
#include <poll.h>

#define BUFFER_SIZE 1024

int client_socket;

static void sigint_handler(int signo)
{
  (void)close(client_socket);
  sleep(2);
  printf("Caught sigINT!\n");
  exit(EXIT_SUCCESS);
}

void register_signal_handler(
int signum,
void (*handler)(int))
{
  if (signal(signum, handler) ==
  SIG_ERR) {
     printf("Cannot handle signal\n");
     exit(EXIT_FAILURE);
  }
}

int main(void)
{
  int ret, i;
  struct sockaddr_ll 
  server_addr,
  client_addr;
  struct pollfd fds[1];
  char buffer[BUFFER_SIZE];

  register_signal_handler(SIGINT,
  sigint_handler);

  client_socket = socket(AF_PACKET, 
                  SOCK_RAW, 
                  htons(ETH_P_ALL));

  if (client_socket < 0) {
    perror("socket");
    return -1;
  }

  memset(&server_addr, 0, 
  sizeof(server_addr));
  server_addr.sll_family = AF_PACKET;
  server_addr.sll_protocol = 
  htons(ETH_P_ALL);
  server_addr.sll_ifindex = 
  if_nametoindex("lo");

  ret = bind(client_socket,
  (struct sockaddr*)&server_addr,
  sizeof(server_addr));

  if (ret < 0) {
    perror("bind");
    (void)close(client_socket);
    return -2;
  }

  memset(fds, 0, sizeof(fds));
  fds[0].fd = client_socket;
  fds[0].events = POLLIN;

  while (1) {
    sprintf(buffer, "%s", "hello server!");
    ret = send(client_socket, 
    buffer, BUFFER_SIZE, 0);
    if (ret < 0) {
      perror("sendto");
      break;
    } else {
       printf("Sentbuffer : %s\n",
       buffer);
    }
    sleep(2);

    ret = poll(fds, 1, -1);
    
    if (ret < 0) {
	perror("poll");
	break;
    }

    if (fds[0].revents & POLLIN) {
       memset(buffer, 0, sizeof(buffer));
       ret = recv(client_socket, 
       buffer, BUFFER_SIZE, 0);
	    
       if (ret < 0) {
	  perror("recvfrom");
          break;
       } else {
          buffer[ret] = '\0';
          printf("Received : %s\n", 
	  buffer);
       }
    }
  }

  (void)close(client_socket);

  return 0;
}

 1$ gcc -o client client.c
 2
 3$ sudo ./client
 4
 5Sentbuffer : hello server!
 6Received : hello server!
 7Sentbuffer : hello server!
 8Received : HELLO
 9Sentbuffer : hello server!
10Received : HELLO
11Sentbuffer : hello server!
12Received : HELLO
13Sentbuffer : hello server!
14Received : HELLO
15Sentbuffer : hello server!
16Received : HELLO
17Sentbuffer : hello server!
18Received : HELLO
19Sentbuffer : hello server!
20Received : HELLO
21Sentbuffer : hello server!
22Received : HELLO
23^CCaught sigINT!

Default Domain:

By default, the socket is configured to work in the AF_PACKET domain, handling all types of network data.

Additional Domain Support:

We expand the socket’s capabilities to also function in the PF_PACKET domain, allowing it to operate similarly to AF_PACKET.

Socket Creation:

We set up a network connection point known as a socket using socket(PF_PACKET, SOCK_RAW, IPPROTO_UDP).

Working Scenario:

Despite the change in domain to PF_PACKET, the socket continues to operate the same way, handling general network data.

Default Protocol Support:

By default, the socket is configured to support the capture of all Ethernet frames (ETH_P_ALL protocol).

Additional Protocol:

The socket is designed to seamlessly support an additional protocol, namely ETH_P_PAE.

Socket Creation:

A socket is created using the socket(AF_PACKET, SOCK_RAW, htons(ETH_P_PAE)) call.

Working Scenario:

Despite the change in protocol to ETH_P_PAE, the overall working scenario of the socket remains consistent.

Socket API

Learning

socket

Create a new socket

bind

Associate the socket with a specific address and port

poll

Monitor multiple file descriptors (usually sockets) for read, write, or error conditions.

recv

Receive data from a connected socket.

send

Send data over a connected socket.