IPV6 UDP 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 ?

Before executing the program that relies on IPv6, enable the loopback interface for IPv6 using the following command:

sudo sysctl -w net.ipv6.conf.lo.disable_ipv6=0
  • This command sets the disable_ipv6 parameter to 0 for the loopback interface (lo), allowing IPv6 functionality.

  • Ensure to use this command cautiously and consider the implications, especially on production systems.

  • After enabling IPv6, proceed to execute your program that relies on IPv6 functionality.

Let us answer few basic questions in this socket

What does socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP) do?

When is it appropriate to use SOCK_DGRAM with IPv6?

What privileges are required to create an IPv6 UDP socket?

Can this socket be used for TCP communication?

How does an IPv6 UDP socket differ from an IPv4 UDP socket?

Can I use IPv6 UDP sockets in a mixed IPv4/IPv6 environment?

How do I specify a port number for an IPv6 UDP socket?

Can I use IPv6 UDP sockets for non-blocking I/O?

How do I handle connection establishment with IPv6 UDP sockets?

Are there any compatibility issues with older systems when using IPv6 UDP sockets?

How do I handle socket errors related to network communication?

When handling socket errors, is it important to close the socket?

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?

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

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

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

How does poll Checking Ready File Descriptors?

What does it mean if poll returns 0?

https://www.plantuml.com/plantuml/svg/ZL9TQy8m57tFhxXxaMGhexxieJBG-J7Ad9MgBnkZrFO6ms8ZIUpOVlrkjJ3T1mn2midDEUVUG_hEvzRljvfT-0rk4TmkBr3dQtOeiCYPuXKz7qooU3PUtWImI8OFsUW-7Jm64C_dQR9CijLeBdeiMgkgv0vjZJOxyWBWpjjzuQ71ULdQIz5fNpGe0AS-q4X-SYSQALcwv_3bMfPm2sUANpMiiVAEwlDaEerdFVQsKHg17mxU5GAK0_wEJh1ePxJcn0kWIoiCG_9G4dWhPLkjJgibWCAeMZ80H1QBMbgp_JxTUYybqXHtgybad6QB-6aSGFZhr8LMvD5EVOeoZHMrqzWuh4fl_Z8XJNhAZr3yo-Ud3Mf76LL6CF9IijdRt8RBT7fLK5YyDgeKXJOEpriXklDcn_f4eY_p2G==
  • There are many functions used in socket. We can classify those functions based on functionalities.

    • Create Socket

    • Bind Socket

    • Poll

    • Recvfrom data_packet

    • Sendto data_packet

    • Close socket

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

server_socket = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP);
  • 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, 1000);
  • recvfrom is commonly used with UDP sockets, where communication is connectionless. it provides information about the source (sender) of the data, including the sender’s IP address and port number. For example,

len = recvfrom(server_socket, buffer, BUFFER_SIZE, 0, (struct sockaddr*)&client_addr, &client_addr_len);
  • sendto is used to send the encoded message to the specified server address and port using a UDP socket. For example,

ret = sendto(server_socket, buffer, strlen(buffer), 0, (struct sockaddr*)&client_addr, client_addr_len);
  • 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 <poll.h>

#define BUFFER_SIZE 1024

int server_socket = -1;

static void sigint_handler(int signo)
{
  (void)close(server_socket);
  sleep(2);
  (void)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);
  }
}

void validate_convert_port(
char *port_str,
struct sockaddr_in6 *sock_addr)
{
 int port;

 if (port_str == NULL) {
   perror("Invalid port_str\n");
   exit(EXIT_FAILURE);
 }

 if (sock_addr == NULL) {
   perror("Invalid sock_addr\n");
   exit(EXIT_FAILURE);
 }

 port = atoi(port_str);

 if (port == 0) {
     perror("Invalid port\n");
     exit(EXIT_FAILURE);
 }

 sock_addr->sin6_port = htons(
 (uint16_t)port);
 printf("Port: %d\n",
 ntohs(sock_addr->sin6_port));
}

int main(int argc, char *argv[])
{
  int ret;
  struct sockaddr_in6
  server_addr,
  client_addr;
  char buffer[BUFFER_SIZE];
  struct pollfd fds[1];

  socklen_t client_addr_len = sizeof(
  client_addr);
  
  register_signal_handler(SIGINT,
  sigint_handler);

  if (argc != 2) {
   printf("%s <port-number>",
   argv[0]);
   exit(EXIT_FAILURE);
  }

  memset(&server_addr, 0,
  sizeof(server_addr));
  server_addr.sin6_family = AF_INET6;
  server_addr.sin6_addr = in6addr_any;
  validate_convert_port(argv[1],
  &server_addr);

  server_socket = socket(AF_INET6,
                  SOCK_DGRAM,
                  IPPROTO_UDP);

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

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

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

  printf("UDP listining\n");

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

  while (1) {

    ret = poll(fds, 1, 1000);

    if (ret < 0) {
	perror("poll");
    	break; 
    }

    if (fds[0].revents & POLLIN) {
      ret = recvfrom(server_socket,
      buffer, BUFFER_SIZE, 0,
      (struct sockaddr*)&client_addr,
      &client_addr_len);

      if (ret > 0) {
        buffer[ret] = '\0';
        printf("Received: %s\n",
        buffer);

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

        ret = sendto(server_socket,
        buffer,
        strlen(buffer), 0,
        (struct sockaddr*)&client_addr,
        client_addr_len);

        if (ret < 0) {
           perror("sendto");
           break;
        }
      } else if (ret < 0) {
         perror("recvfrom");
         break;
      }
      printf("Sentbuffer = %s\n",
      buffer);
    }
  }

  (void)close(server_socket);

  return 0;
}


 1$ gcc -o server server.c
 2
 3$ sudo ./server 8080
 4
 5Port: 8080
 6UDP listining
 7Received: HI
 8Sentbuffer = HELLO
 9Received: HI
10Sentbuffer = HELLO
11Received: HI
12Sentbuffer = HELLO
13Received: HI
14Sentbuffer = HELLO
15Received: HI
16Sentbuffer = HELLO
17Received: HI
18Sentbuffer = HELLO
19Received: HI
20Sentbuffer = HELLO
21Received: HI
22Sentbuffer = HELLO
23Received: HI
24^CCaught sigINT!
https://www.plantuml.com/plantuml/svg/TL5DJyCm3BttLvZBb424EWuSXf0sze4gobfrsmM4gYvrj8YicPAi27uzxbO9E9238_lbFTilOnyg5uvx4rs47UuH_A6IQAfjT0AYaRVo7GERBCfaEL_V2bXbqwTozbXCdWKaULvawwpSp79-5ttij45WfuiDEGVjWNsYvwI2JHqiaqPZ4ygpfe3jKIbq0dnm1XjsJhc0M02Zsb46w9vMTUqk-S2ZQz6LNKOK_OLMiLyrJXECLErVuxThLSCz_6dsWs5BLKzudgLfiYJWO8rXX0gu4J2CuvYKj0BM4rpF6FGC3kHKqsy6tHavbAroTl_VWW-RnM9UbAlaPNxQRhb9qtEaLcWykTPO7f59MdMnTtAwBj8hIVQnrkgQIsCz_cr1T1_i8HeJYJxn6m==
  • There are many functions used in socket. We can classify those functions based on functionalities.

    • Create Socket

    • Poll

    • Sendto data_packet

    • Recvfrom data_packet

    • Close socket

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

client_socket = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP);
  • poll() is used for monitoring multiple file descriptors to see if I/O is possible on any of them.

ret = poll(fds, 2, 1000);
  • sendto is used to send the encoded message to the specified server address and port using a UDP socket. For example,

ret = sendto(client_socket, buffer, strlen(buffer), 0, (struct sockaddr*)&server_addr, sizeof(server_addr));
  • recvfrom is commonly used with UDP sockets, where communication is connectionless. it provides information about the source (sender) of the data, including the sender’s IP address and port number. For example,

len = recvfrom(client_socket, buffer, BUFFER_SIZE, 0, NULL, NULL);
  • 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 <poll.h>

#define BUFFER_SIZE 1024

int client_socket = -1;

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

void validate_convert_port(
char *port_str,
struct sockaddr_in6 *sock_addr)
{
 int port;

 if (port_str == NULL) {
   perror("Invalid port_str\n");
   exit(EXIT_FAILURE);
 }

 if (sock_addr == NULL) {
   perror("Invalid sock_addr\n");
   exit(EXIT_FAILURE);
 }

 port = atoi(port_str);

 if (port == 0) {
   perror("Invalid port\n");
   exit(EXIT_FAILURE);
 }

 sock_addr->sin6_port = htons(
 (uint16_t)port);
 printf("Port: %d\n",
 ntohs(sock_addr->sin6_port));
}

void validate_convert_addr(
char *ip_str,
struct sockaddr_in6 *sock_addr)
{
  if (ip_str == NULL) {
   perror("Invalid ip_str\n");
   exit(EXIT_FAILURE);
 }

 if (sock_addr == NULL) {
   perror("Invalid sock_addr\n");
   exit(EXIT_FAILURE);
 }

 printf("IP Address: %s\n", ip_str);

 if (inet_pton(AF_INET6, ip_str,
 &(sock_addr->sin6_addr)) <= 0) {
    perror("Invalid address\n");
    exit(EXIT_FAILURE);
  }
}

void recv_data(char *buffer)
{
  int ret, len;

  len = recvfrom(client_socket,
  buffer, BUFFER_SIZE, 0, NULL, NULL);

  if (len > 0) {
    buffer[len] = '\0';
    (void)printf("Received: %s\n",
    buffer);

  } else if (len == 0) {
     printf("Connection closed\n");
     exit(EXIT_FAILURE);
   }
}

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(int argc, char *argv[])
{
  int ret, len;
  struct sockaddr_in6
  server_addr;
  char buffer[BUFFER_SIZE];
  struct pollfd fds[1];
  char *str = "HI";

  register_signal_handler(SIGINT,
  sigint_handler);

  if (argc != 3) {
    printf("%s<port-number><ip-addr>\n",
    argv[0]);
    exit(EXIT_FAILURE);
  }

  memset(&server_addr, 0,
  sizeof(server_addr));
  server_addr.sin6_family = AF_INET6;
  validate_convert_port(argv[1],
  &server_addr);
  validate_convert_addr(argv[2],
  &server_addr);

  client_socket = socket(AF_INET6,
                  SOCK_DGRAM,
                  IPPROTO_UDP);

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

  while (1) {
    ret = sendto(client_socket, str,
    strlen(str), 0,
    (struct sockaddr*)&server_addr,
    sizeof(server_addr));
    printf("sendbuffer = %s\n", str);

    if (ret < 0) {
       perror("send error\n");
       (void)close(client_socket);
       break;
    }

    fds[0].fd = client_socket;
    fds[0].events = POLLIN;

    ret = poll(fds, 2, 1000);

    if (ret < 0) {
       perror("poll");
       (void)close(client_socket);
       break;
    }

    if (fds[0].revents & POLLIN) {
       recv_data(buffer);
    }
  }

  (void)close(client_socket);

  return 0;
}


 1$ gcc -o client client.c
 2
 3$ sudo ./client 8080 ::1
 4
 5Port: 8080
 6IP Address: ::1
 7sendbuffer = HI
 8Received: HELLO
 9sendbuffer = HI
10Received: HELLO
11sendbuffer = HI
12Received: HELLO
13sendbuffer = HI
14Received: HELLO
15sendbuffer = HI
16Received: HELLO
17sendbuffer = HI
18Received: HELLO
19sendbuffer = HI
20Received: HELLO
21sendbuffer = HI
22^CCaught sigINT!

$ sudo ./server 8080 ::1

$ sudo ./client 8080 ::1

program to run with elevated privileges, listen on port 8080, and bind to the loopback address ::1.

<port_number> <ip_address> decided by the user based on the connection.

Default Domain:

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

Additional Domain Support:

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

Socket Creation:

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

Working Scenario:

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

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.

recvfrom

It provides information about the source (sender) of the data, including the sender’s IP address and port number.

sendto

Send the encoded message to the specified server address and port using a UDP socket.