Unix Socket - ตัวอย่างเซิร์ฟเวอร์

ในการสร้างเซิร์ฟเวอร์ TCP คุณต้องทำตามขั้นตอนด้านล่าง -

  • สร้างซ็อกเก็ตด้วยการเรียกระบบซ็อกเก็ต ()

  • ผูกซ็อกเก็ตกับแอดเดรสโดยใช้การเรียกระบบbind () สำหรับซ็อกเก็ตเซิร์ฟเวอร์บนอินเทอร์เน็ตแอดเดรสประกอบด้วยหมายเลขพอร์ตบนเครื่องโฮสต์

  • ฟังการเชื่อมต่อกับการโทรของระบบlisten ()

  • ยอมรับการเชื่อมต่อกับการเรียกระบบaccept () โดยทั่วไปสายนี้จะบล็อกจนกว่าไคลเอ็นต์จะเชื่อมต่อกับเซิร์ฟเวอร์

  • ส่งและรับข้อมูลโดยใช้การเรียกระบบread ()และwrite ()

ตอนนี้ให้เราใส่ขั้นตอนเหล่านี้ในรูปแบบของซอร์สโค้ด ใส่รหัสนี้ลงในไฟล์server.cและคอมไพล์ด้วยคอมไพเลอร์gcc

#include <stdio.h>
#include <stdlib.h>

#include <netdb.h>
#include <netinet/in.h>

#include <string.h>

int main( int argc, char *argv[] ) {
   int sockfd, newsockfd, portno, clilen;
   char buffer[256];
   struct sockaddr_in serv_addr, cli_addr;
   int  n;
   
   /* First call to socket() function */
   sockfd = socket(AF_INET, SOCK_STREAM, 0);
   
   if (sockfd < 0) {
      perror("ERROR opening socket");
      exit(1);
   }
   
   /* Initialize socket structure */
   bzero((char *) &serv_addr, sizeof(serv_addr));
   portno = 5001;
   
   serv_addr.sin_family = AF_INET;
   serv_addr.sin_addr.s_addr = INADDR_ANY;
   serv_addr.sin_port = htons(portno);
   
   /* Now bind the host address using bind() call.*/
   if (bind(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0) {
      perror("ERROR on binding");
      exit(1);
   }
      
   /* Now start listening for the clients, here process will
      * go in sleep mode and will wait for the incoming connection
   */
   
   listen(sockfd,5);
   clilen = sizeof(cli_addr);
   
   /* Accept actual connection from the client */
   newsockfd = accept(sockfd, (struct sockaddr *)&cli_addr, &clilen);
	
   if (newsockfd < 0) {
      perror("ERROR on accept");
      exit(1);
   }
   
   /* If connection is established then start communicating */
   bzero(buffer,256);
   n = read( newsockfd,buffer,255 );
   
   if (n < 0) {
      perror("ERROR reading from socket");
      exit(1);
   }
   
   printf("Here is the message: %s\n",buffer);
   
   /* Write a response to the client */
   n = write(newsockfd,"I got your message",18);
   
   if (n < 0) {
      perror("ERROR writing to socket");
      exit(1);
   }
      
   return 0;
}

จัดการการเชื่อมต่อหลายรายการ

เพื่อให้เซิร์ฟเวอร์สามารถจัดการการเชื่อมต่อพร้อมกันหลายรายการเราได้ทำการเปลี่ยนแปลงต่อไปนี้ในโค้ดด้านบน -

  • ใส่คำสั่งยอมรับและรหัสต่อไปนี้ในลูปไม่มีที่สิ้นสุด

  • หลังจากสร้างการเชื่อมต่อแล้วให้เรียกใช้fork ()เพื่อสร้างกระบวนการใหม่

  • กระบวนการลูกจะปิดsockfdและเรียกใช้ฟังก์ชันdoprocessingโดยส่งตัวบอกไฟล์ซ็อกเก็ตใหม่เป็นอาร์กิวเมนต์ เมื่อกระบวนการทั้งสองเสร็จสิ้นการสนทนาตามที่ระบุโดยการกลับมาของกระบวนการdoprocessing ()กระบวนการนี้ก็จะออกไป

  • ปิดการปกครองnewsockfd เนื่องจากโค้ดทั้งหมดนี้อยู่ในวงวนไม่สิ้นสุดโค้ดจะกลับไปที่คำสั่ง accept เพื่อรอการเชื่อมต่อครั้งต่อไป

#include <stdio.h>
#include <stdlib.h>

#include <netdb.h>
#include <netinet/in.h>

#include <string.h>

void doprocessing (int sock);

int main( int argc, char *argv[] ) {
   int sockfd, newsockfd, portno, clilen;
   char buffer[256];
   struct sockaddr_in serv_addr, cli_addr;
   int n, pid;
   
   /* First call to socket() function */
   sockfd = socket(AF_INET, SOCK_STREAM, 0);
   
   if (sockfd < 0) {
      perror("ERROR opening socket");
      exit(1);
   }
   
   /* Initialize socket structure */
   bzero((char *) &serv_addr, sizeof(serv_addr));
   portno = 5001;
   
   serv_addr.sin_family = AF_INET;
   serv_addr.sin_addr.s_addr = INADDR_ANY;
   serv_addr.sin_port = htons(portno);
   
   /* Now bind the host address using bind() call.*/
   if (bind(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0) {
      perror("ERROR on binding");
      exit(1);
   }
   
   /* Now start listening for the clients, here
      * process will go in sleep mode and will wait
      * for the incoming connection
   */
   
   listen(sockfd,5);
   clilen = sizeof(cli_addr);
   
   while (1) {
      newsockfd = accept(sockfd, (struct sockaddr *) &cli_addr, &clilen);
		
      if (newsockfd < 0) {
         perror("ERROR on accept");
         exit(1);
      }
      
      /* Create child process */
      pid = fork();
		
      if (pid < 0) {
         perror("ERROR on fork");
         exit(1);
      }
      
      if (pid == 0) {
         /* This is the client process */
         close(sockfd);
         doprocessing(newsockfd);
         exit(0);
      }
      else {
         close(newsockfd);
      }
		
   } /* end of while */
}

การแบ่งรหัสต่อไปนี้แสดงการใช้ฟังก์ชันการประมวลผลแบบง่าย

void doprocessing (int sock) {
   int n;
   char buffer[256];
   bzero(buffer,256);
   n = read(sock,buffer,255);
   
   if (n < 0) {
      perror("ERROR reading from socket");
      exit(1);
   }
   
   printf("Here is the message: %s\n",buffer);
   n = write(sock,"I got your message",18);
   
   if (n < 0) {
      perror("ERROR writing to socket");
      exit(1);
   }
	
}