af unix tcp server client program with Epoll 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 ?

Let us answer few basic questions in this socket

What does AF_UNIX represent in the socket call?

Why use SOCK_STREAM as the socket type?

What does the third parameter (0) indicate in socket(AF_UNIX, SOCK_STREAM, 0)?

How can I set up a server and client using AF_UNIX sockets?

How should errors in the socket call be handled?

What is the role of the file system path in AF_UNIX sockets?

What is the role of the file system path in AF_UNIX sockets?

Why might bind() or listen() fail in socket programming?

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?

What is the primary purpose of the epoll system call?

What types of file descriptors can be monitored using epoll?

What data structure is used by epoll to store events?

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

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

How does epoll Checking Ready File Descriptors?

What does it mean if epoll returns 0?

https://www.plantuml.com/plantuml/svg/RL9VQy8m47_lfxYzIB9LiQyEmQ8TZ5KTMeSmHedf5SDYMvAqidtw9RREClPokV3xSxcxpBHXojGdwTsP8vuGTCKuIdRmBe0tqIN_H4F2ftItUjxxi5tFNz9jied2fGzZ-kXD51gOma4K6T6e6bHfdlb0j54rD-3qBClKFHrqgBlveCKtblbVLbeNr7bAeGqMVRzbk4-ZjsYLR1tEfS32MC1MPfnZPVhSrIwEs-YuM9LIjjGsvGgPmO2CRoWtabnvFaIlwpXEvqcSXek53x_5V1XWOrEhEn-5H22NWmIKWj10lb3Jg-EP2TEplEXqlmqVXe4r4ZcG5dmVVumoPjZ8FNLwgqh1RgRet0703fmtfFUeGvtdU1jfUwKmXA3TaDDejElvJsEKj8FkD2qVfRQj5INrh4ZaBdRjpfDD_C1jZwGfHKQvB3NUNArKcxBoPbPWVzGF
  • There are many functions used in socket. We can classify those functions based on functionalities.

    • Create Socket

    • Bind Socket

    • Listen Socket

    • Accept Socket

    • Epoll create1

    • Epoll_ctl

    • Epoll_wait

    • Recv data_packet

    • Send data_packet

    • Close socket

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

server_fd = socket(AF_UNIX, SOCK_STREAM, 0);
  • bind() is used to associate the socket with a specific address and port. For example,

ret = bind(server_fd, (struct sockaddr*)&server_addr, sizeof(struct sockaddr_un));
  • listen() is used to set up a socket to accept incoming connections. For example,

ret = listen(server_fd, MAX_CLIENTS);
  • accept() is used in network programming on the server side to accept a connection request from a client. For example,

client_fd = accept(server_fd, NULL, NULL);
  • epoll_create1() creating an epoll instance using epoll_create1, The size parameter is an advisory hint for the kernel regarding the number of file descriptors expected to be monitored, For example,

epoll_fd = epoll_create1(0));
  • epoll_ctl() After creating an epoll instance, file descriptors are added to it using epoll_ctl, For example,

ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, client_fd, &event);
  • epoll_wait() The application then enters a loop where it waits for events using epoll_wait, For example,

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

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

ret = send(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(client_fd);
  • See the full program below,

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

#define MAX_EVENTS 5
#define SOCKET_PATH "/tmp/my_socket"

int server_fd = -1;
int client_fd = -1;
int epoll_fd = -1;

static void sigint_handler(int signo)
{
  unlink(SOCKET_PATH);
  (void)close(epoll_fd);
  (void)close(server_fd);
  (void)close(client_fd);
  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);
  }
}

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

  memset(buffer, 0,
  sizeof(buffer));
  len = recv(client_fd, buffer,
  sizeof(buffer) - 1, 0);

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

    memset(buffer, 0,
    sizeof(buffer));
    strncpy(buffer, "HELLO",
    strlen("HELLO") + 1);
    buffer[strlen(buffer) + 1] = '\0';
    printf("Sentbuffer = %s\n",
    buffer);

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

    if (ret < 0) {
       perror("send error\n");
       (void)close(client_fd);
       (void)close(server_fd);
       exit(EXIT_FAILURE);
    }

  } else if (len < 0) {
      perror("recv");
      (void)close(client_fd);
      (void)close(server_fd);
      exit(EXIT_FAILURE);
   }
}

int main(void) 
{
  int ready_fds; 
  int ret;
  struct sockaddr_un 
  server_addr, 
  client_addr;
  char buffer[1024];
  struct epoll_event 
  events[MAX_EVENTS];
  struct epoll_event event;

  register_signal_handler(SIGINT,
  sigint_handler);
  
  memset(&server_addr, 0, 
  sizeof(struct sockaddr_un));
  server_addr.sun_family = AF_UNIX;
  strncpy(server_addr.sun_path, 
  SOCKET_PATH, 
  sizeof(server_addr.sun_path) - 1);
  unlink(SOCKET_PATH);

  server_fd = socket(AF_UNIX, 
              SOCK_STREAM, 0);

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

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

  ret = listen(server_fd, 
  MAX_EVENTS);
  
  if (ret < 0)
  {
    perror("listen");
    (void)close(server_fd);
    return -3;
  }

  printf("server listening\n");

  client_fd = accept(server_fd, 
  NULL, NULL);

  if (client_fd < 0) {
    perror("accept");
    (void)close(server_fd);
    return -4;
  }

  printf("Connection accepted\n");
  
  epoll_fd = epoll_create1(0);
 
  if (epoll_fd < 0) {
    perror("Epoll failed");
    exit(EXIT_FAILURE);
  }

  event.events = EPOLLIN;
  event.data.fd = client_fd;
  
  ret = epoll_ctl(epoll_fd, 
  EPOLL_CTL_ADD, client_fd, 
  &event);

  if (ret < 0) {
    perror("Epoll_ctl failed");
    exit(EXIT_FAILURE);
  }

  while (1) {
   ready_fds = epoll_wait(epoll_fd, 
               events, 
               MAX_EVENTS, -1);

    if (ready_fds < 0) {
      perror("Epoll wait failed");
      break;
    }

    if (events[0].data.fd == 
        client_fd) {
      recv_send(buffer);
    }
  }
  
  (void)close(client_fd);
  (void)close(server_fd);
  return 0;
}

 1$ gcc -o server server.c
 2
 3$ sudo ./server
 4
 5server listening
 6Connection accepted
 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/RP5HQy8m4CVV-obkBv9iKUohGx1e1sDLXtPZC4Q9wHN3OY99Mja-_LBReOoz93dk_dU__sNcFBU-FgZenk_nWE2ENA3Yk-YSY2REY4_q97aeNbQFRmosw_bJiSqtQR9aCARtqKGOhL5u8fH4xOkgP42SjxNmq8fvMTfREd1e6xH56p5myXjDzRUgg3Lj6-BHA1Nwm1Iwfx38FSPaVCaAhyXl7OFqUPrbnJpFYcInO715Ci0cF8Fkj9SAWPml4bCAqW7vGaU3HTJbDVskhYfiARrLg4aNqjvjD_J4fR-QVnxY62oJjo9zJLVvbi4m3kMo0j8btySVev9xFcfzJI-85CBUTOy201EBelcNfjjPJmD3Y3iYL2xuqeP6mOQishFtEiytsPq81abZP4c5CWuldOFKUNECPa4G_ly7
  • There are many functions used in socket. We can classify those functions based on functionalities.

    • Create Socket

    • Connect Socket

    • Epoll create1

    • Epoll_ctl

    • Epoll_wait

    • Send data_packet

    • Recv data_packet

    • Close socket

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

client_fd = socket(AF_UNIX, SOCK_STREAM, 0);
  • 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*)&server_addr, sizeof(struct sockaddr_un));
  • epoll_create1() creating an epoll instance using epoll_create1, The size parameter is an advisory hint for the kernel regarding the number of file descriptors expected to be monitored, For example,

epoll_fd = epoll_create1(0));
  • epoll_ctl() After creating an epoll instance, file descriptors are added to it using epoll_ctl, For example,

ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, client_fd, &event);
  • epoll_wait() The application then enters a loop where it waits for events using epoll_wait, For example,

ret = epoll_wait(epoll_fd, events, MAX_EVENTS, -1);
  • 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 <sys/un.h>
#include <signal.h>
#include <sys/epoll.h>
#include <netinet/in.h>

#define MAX_EVENTS 3
#define SOCKET_PATH "/tmp/my_socket"

int client_fd = -1;
int epoll_fd = -1;

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

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(void) 
{
  int ready_fds; 
  int cli_connect;
  int ret;
  struct sockaddr_un 
  server_addr;
  char buffer[1024];
  struct epoll_event 
  events[MAX_EVENTS];
  
  register_signal_handler(SIGINT,
  sigint_handler);

  memset(&server_addr, 0, 
  sizeof(struct sockaddr_un));
  server_addr.sun_family = AF_UNIX;
  strncpy(server_addr.sun_path, 
  SOCKET_PATH, 
  sizeof(server_addr.sun_path) - 1);

  client_fd = socket(AF_UNIX, 
              SOCK_STREAM, 0);

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

  cli_connect = connect(client_fd, 
  (struct sockaddr*)&server_addr, 
  sizeof(struct sockaddr_un));

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

  ret = epoll_fd = epoll_create1(0);

  if (ret < 0) {
   perror("Epoll creation failed");
   exit(EXIT_FAILURE);
  }

  struct epoll_event event;

  event.events = EPOLLIN | EPOLLET;
  event.data.fd = client_fd;
  ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD,
  client_fd, &event);

  if (ret  == -1) {
       perror("Epoll_ctl failed");
       exit(EXIT_FAILURE);
  }

  while (1) {
    send_data(buffer);

    ready_fds = epoll_wait(epoll_fd, 
                events, 
                MAX_EVENTS, -1);

    if (ready_fds < 0) {
      perror("Epoll wait failed");
      break;
    }

    if (events[0].data.fd == 
    client_fd) {
      recv_data(buffer);
    }
  }

  (void)close(client_fd);
  
  return 0;
}
 1$ gcc -o client client.c
 2
 3$ sudo ./client
 4
 5connected
 6sentbuffer = HI
 7Received: HELLO
 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!

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.

epoll

handles a set of file descriptors with different states, such as reading, writing, and exceptions, by using the struct epoll_event structure and the associated event flags..

recv

Receive data from a connected socket.

send

Send data over a connected socket.