IPV6 TCP server client program with Select system call

  • In this program, you are going to learn

  • How to create a Socket ?

  • How to bind a socket ?

  • How to listen a socket ?

  • How to connect a socket ?

  • How to accept 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_STREAM, IPPROTO_TCP) do?

When is it appropriate to use SOCK_STREAM with IPv6?

What privileges are required to create an IPv6 TCP socket?

Can socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP) be used for regular UDP communication?

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

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

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

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

How do I handle connection establishment with IPv6 TCP sockets?

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

How do I handle IPv6 DNS resolution in my application?

How should you handle errors when using accept() in socket programming?

Why is it important to check the return value of send() and recv() in socket programming?

Can you use a TCP socket (SOCK_STREAM) for sending and receiving data concurrently between a client and server?

What is the purpose of the select system call in network programming?

How does select help in handling multiple sockets efficiently?

What types of file descriptors can be monitored using select?

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/XPBFJy8m5CVl-okU5zBAJEIW1um914T2H4QsUV6obEujD9PjQGl--EjzXQ636EsXVNtjv_jjNpkoJXYttUZWmgrnWs0R8L6BLR1V28Qsbg_es7YQpnPHTXj26ayUypHBel5J2BFbCecpECycItuN35UgAfYJJMxHxD3aPH42iyvifGCl98h2NFAUt-734ApwnBfanmJt8bfPXzMvp8rVyZcf5LQESd0FGafitF-EqB7y3dEDbHUTFkGlKHApda5Ha81jascKiHD3OZjRtjPA8x3zm0QSWxB0Fj1ogXbgb8wLnKQyGny67J24nVDytkr9IfN0o72M_cd9WLweQZt0jw51kJi7Ljkon9_I7gOShlmnhidiG5fiN-etqXagpP4yKAWjNRQgUK2WAdtV5c2I9VE-f5kpNQqABdLjyLIPSElg9XWHH3_j2m==
  • There are many functions used in socket. We can classify those functions based on functionalities.

    • Create Socket

    • Bind Socket

    • Listen Socket

    • Select

    • Accept Socket

    • Recv data_packet

    • Send data_packet

    • Close socket

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

tcp_server_fd = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP);
  • bind() is used to associate the socket with a specific address and port. For example,

bind_ret = bind(tcp_server_fd, (struct sockaddr*)&tcp_addr, sizeof(tcp_addr));
  • listen() is used to set up a socket to accept incoming connections. For example,

listen_ret = listen(tcp_server_fd, 5);
  • select() is used in network programming to monitor multiple file descriptors (usually sockets) for read, write, or error conditions. For example,

select_ret = select(fdmax + 1, &read_fds, NULL, NULL, NULL);
  • accept() is used in network programming on the server side to accept a connection request from a client. For example,

tcp_client_fd = accept(tcp_server_fd, (struct sockaddr*) &tcp_addr, &tcp_addr_len);
  • recv is used in network programming to receive data from a connected socket. For example,

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

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

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

  1#include <stdio.h>
  2#include <stdlib.h>
  3#include <string.h>
  4#include <unistd.h>
  5#include <sys/socket.h>
  6#include <netinet/in.h>
  7#include <sys/types.h>
  8#include <arpa/inet.h>
  9#include <sys/un.h>
 10#include <signal.h>
 11
 12int tcp_server_fd = -1;
 13int tcp_client_fd = -1;
 14
 15static void sigint_handler(int signo)
 16{
 17  (void)close(tcp_server_fd);
 18  (void)close(tcp_client_fd);
 19  sleep(2);
 20  printf("Caught sigINT!\n");
 21  exit(EXIT_SUCCESS);
 22}
 23
 24void register_signal_handler(
 25int signum,
 26void (*handler)(int))
 27{
 28  if (signal(signum, handler) == SIG_ERR) {
 29     printf("Cannot handle signal\n");
 30     exit(EXIT_FAILURE);
 31  }
 32}
 33
 34void validate_convert_port(
 35char *port_str,
 36struct sockaddr_in6 *sock_addr)
 37{
 38 int port;
 39
 40 if (port_str == NULL) {
 41   perror("Invalid port_str\n");
 42   exit(EXIT_FAILURE);
 43 }
 44 
 45 if (sock_addr == NULL) {
 46   perror("Invalid sock_addr\n");
 47   exit(EXIT_FAILURE);
 48 }
 49
 50 port = atoi(port_str);
 51
 52 if (port == 0) {
 53     perror("Invalid port\n");
 54     exit(EXIT_FAILURE);
 55 }
 56
 57 sock_addr->sin6_port = htons(
 58 (uint16_t)port);
 59 printf("Port: %d\n",
 60 ntohs(sock_addr->sin6_port));
 61}
 62
 63void validate_convert_addr(
 64char *ip_str,
 65struct sockaddr_in6 *sock_addr)
 66{
 67  if (ip_str == NULL) {
 68    perror("Invalid ip_str\n");
 69    exit(EXIT_FAILURE);
 70  }
 71
 72  if (sock_addr == NULL) {
 73    perror("Invalid sock_addr\n");
 74    exit(EXIT_FAILURE);
 75  }
 76
 77  printf("IP Address: %s\n", ip_str);
 78
 79  if (inet_pton(AF_INET6, ip_str,
 80  &(sock_addr->sin6_addr)) <= 0) {
 81    perror("Invalid address\n");
 82    exit(EXIT_FAILURE);
 83  }
 84}
 85
 86void recv_send(char *buffer)
 87{
 88  int len, ret;  
 89
 90  memset(buffer, 0, sizeof(buffer));
 91  len = recv(tcp_client_fd, buffer,
 92  sizeof(buffer) - 1, 0);
 93
 94  if (len > 0) {
 95    buffer[len] = '\0';
 96    printf("Received: %s\n",
 97    buffer);
 98
 99    memset(buffer, 0, 
100    sizeof(buffer));
101    strncpy(buffer, "HELLO",
102    strlen("HELLO") + 1);
103    buffer[strlen(buffer) + 1] = '\0';
104    printf("Sentbuffer = %s\n", 
105    buffer);
106
107    ret = send(tcp_client_fd, buffer,
108    strlen(buffer), 0);
109
110    if (ret < 0) {
111       perror("send error\n");
112       (void)close(tcp_client_fd);
113       (void)close(tcp_server_fd);
114       exit(EXIT_FAILURE);
115    }
116
117   } else if (len < 0) {
118	  perror("recv");
119	  (void)close(tcp_client_fd);
120	  (void)close(tcp_server_fd);
121          exit(EXIT_FAILURE);
122    }
123}
124
125int main(int argc, char *argv[]) 
126{
127  int fdmax = 0; 
128  int flag = 0;
129  int ret;
130  struct sockaddr_in6 
131  tcp_addr;
132  fd_set read_fds;
133  char buffer[1024];
134  socklen_t tcp_addr_len = sizeof(
135  tcp_addr);
136
137  register_signal_handler(SIGINT, 
138  sigint_handler);
139
140  if (argc != 3) {
141    printf("%s<port-number><ip-addr>", 
142    argv[0]);
143    exit(EXIT_FAILURE);
144  }
145
146  memset(&tcp_addr, 0, 
147  sizeof(tcp_addr));
148  tcp_addr.sin6_family = AF_INET6;
149  validate_convert_port(argv[1], 
150  &tcp_addr);
151  validate_convert_addr(argv[2], 
152  &tcp_addr);
153
154  tcp_server_fd = socket(AF_INET6, 
155                  SOCK_STREAM, 
156                  IPPROTO_TCP);
157  
158  if (tcp_server_fd < 0) {
159    perror("socket");
160    return -1;
161  }
162
163  ret = bind(tcp_server_fd, 
164  (struct sockaddr *)&tcp_addr, 
165  sizeof(tcp_addr));
166  
167  if (ret < 0)
168  {
169    perror("bind");
170    (void)close(tcp_server_fd);
171    return -2;
172  }
173
174  ret = listen(tcp_server_fd, 5);
175  
176  if (ret < 0)
177  {
178    perror("listen");
179    (void)close(tcp_server_fd);
180    return -3;
181  }
182
183  tcp_client_fd = accept(tcp_server_fd,
184  (struct sockaddr *) &tcp_addr,
185  &tcp_addr_len);
186
187  if (tcp_client_fd < 0) {
188   perror("accept");
189   (void)close(tcp_server_fd);
190   return -4;
191  }
192
193  (void)printf("Server lsiten...\n");
194
195  fdmax = tcp_client_fd;
196  
197  while (1) {
198    FD_ZERO(&read_fds);
199    FD_SET(tcp_client_fd, &read_fds);
200
201    ret = select(fdmax + 1, &read_fds, 
202    NULL, NULL, NULL);
203    
204    if (ret < 0) {
205      perror("select");
206      break;
207    }
208
209    if (FD_ISSET(tcp_client_fd,
210        &read_fds)) {
211        recv_send(buffer);
212    }
213  }
214
215  (void)close(tcp_client_fd);
216  (void)close(tcp_server_fd);
217
218  return 0;
219}
 1$ gcc -o server server.c
 2
 3$ sudo ./server 8080 ::1
 4
 5IP Address: ::1
 6Server lsiten...
 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/TL6nJiCm4Dtp5STIsJH8T64eIwkGIX6bYPAmi5YfSr4jN3kAdIBuUYub5Qa47ivdtxtdUyyhvwlEzqSTtFW37X5SMqdKrJuu5uAbi_8TFLjlHBABouSGYZHw5aMPn-kN498iozCo5MMKySTWAQqnA3sJMg7negb3OCvtlVGmy5HrtTtocPUj6D8GBfdGQ0Z_SL0QWPqtjk0Sb0FsYOxJ66ZgAUs-Rnea1c8dBFivyX3kXp4sJ-8jpbCswx2ggTsDbqLSJZckoWxrrUGmXyMa9OJTwtOxZGHI3J2YJOf_YJcGgsOK0RIM7ShJdqBK5zhc8WJkXjS7CQWTMM8i3yW1rGnnj2agy-rSaZViP5NDfROEVva9whnjWnK1w7k_0G==
  • There are many functions used in socket. We can classify those functions based on functionalities.

    • Create Socket

    • Connect Socket

    • Select

    • Recv data_packet

    • Send data_packet

    • Close socket

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

client_fd = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP);
  • connect is used in network programming to establish a connection from a client to a server. For example,

cli_connect = connect(client_fd, (struct sockaddr*)&tcp_addr, tcp_addr_len);
  • select is used in network programming to monitor multiple file descriptors (usually sockets) for read, write, or error conditions. For example,

ret = select(client_fd + 1, &read_fds, NULL, NULL, NULL);
  • send is used in network programming to send data over a connected socket. For example,

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

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

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

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <sys/un.h>
#include <signal.h>

int client_fd = -1;

static void sigint_handler(int signo)
{
  (void)close(client_fd);
  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 = recv(client_fd, buffer,
  sizeof(buffer) - 1, 0);

  if (len > 0) {
    buffer[len] = '\0';
    (void)printf("Received: %s\n",
    buffer);
    
  } else if (len == 0) {
     printf("Connection closed\n");
     exit(EXIT_FAILURE);
   }
}

void send_data(char *buffer)
{
  int ret;
  
  memset(buffer, 0, sizeof(buffer));
	
  strncpy(buffer, "HI",
  strlen("HI") + 1);
  buffer[strlen(buffer) + 1] = '\0';

  ret = send(client_fd, buffer,
  strlen(buffer), 0);

  if (ret < 0) {
    perror("send error\n");
    (void)close(client_fd);
    exit(EXIT_FAILURE);
  }
  printf("sentbuffer = %s\n", 
  buffer);
}

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 cli_connect; 
  int ret;
  struct sockaddr_in6 
  tcp_addr;
  fd_set read_fds;
  char buffer[1024];

  socklen_t tcp_addr_len = sizeof(
  tcp_addr);

  register_signal_handler(SIGINT,
  sigint_handler);

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

  memset(&tcp_addr, 0, 
  sizeof(tcp_addr));
  tcp_addr.sin6_family = AF_INET6;
  validate_convert_port(argv[1], 
  &tcp_addr);
  validate_convert_addr(argv[2], 
  &tcp_addr);
  
  client_fd = socket(AF_INET6, 
              SOCK_STREAM, 
              IPPROTO_TCP);

  if (client_fd < 0) {
    perror("socket");
    return -1;
  }
  
  cli_connect = connect(client_fd, 
  (struct sockaddr *)&tcp_addr, 
  tcp_addr_len);

  if (cli_connect < 0) {
	  perror("connect");
	  return -2;
  } else {
  	printf("connected\n");
  }

  while (1) {	
   send_data(buffer);
   
   FD_ZERO(&read_fds);
   FD_SET(client_fd, &read_fds);

   ret = select(client_fd + 1, 
   &read_fds, NULL, NULL, NULL);

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

   if (FD_ISSET(client_fd, 
      &read_fds)) {
      recv_data(buffer);
    }
  }

  (void)close(client_fd);

  return 0;
}
 1$ gcc -o client client.c
 2
 3$ sudo ./client 8080 ::1
 4
 5Port: 8080
 6IP Address: ::1
 7connected
 8sentbuffer = HI
 9Received: HELLO
10sentbuffer = HI
11Received: HELLO
12sentbuffer = HI
13Received: HELLO
14sentbuffer = HI
15Received: HELLO
16sentbuffer = HI
17Received: HELLO
18sentbuffer = HI
19Received: HELLO
20sentbuffer = HI
21Received: HELLO
22sentbuffer = HI
23^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_STREAM, IPPROTO_TCP).

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

listen

Set up a socket to accept incoming connections.

connect

Establish a connection from a client to a server.

accept

Server side to accept a connection request from a client.

select

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.