Home  >  Article  >  Operation and Maintenance  >  How to use linux socket to send and receive messages independently using threads

How to use linux socket to send and receive messages independently using threads

WBOY
WBOYforward
2023-05-16 17:16:201143browse

代码实现

1、服务器端

/*tcpserver.c  2011.9.1 by yyg*/
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/wait.h>
#include <pthread.h>

#define MYPORT 3490  //定义端口
#define BACKLOG 10
#define MAXDATASIZE 1024

int sockfd,new_fd;
pthread_t accthread,recthread;

void recmessage(void){  //接收客户端信息函数
  while(1){
    int numbytes;
    char buf[MAXDATASIZE];
    if((numbytes = recv(new_fd,buf,MAXDATASIZE,0))==-1){
      perror("recv");
      exit(1);
    }
    buf[numbytes]='\0';
    if(strcmp(buf,"exit")==0){  //若收到的是exit字符,则代表退出通信
      printf("Client is closed\n");
      close(new_fd);
      close(sockfd);
      exit(1);
    }
    printf("client:%s\n",buf);
  }/*while*/
}

void acceptconnect(void){  //接受客户端连接请求函数
  struct sockaddr_in their_addr;
  int sin_size;
  sin_size = sizeof(struct sockaddr_in);
  if((new_fd = accept(sockfd,(struct sockaddr *)&their_addr,&sin_size))==-1){
    perror("accept");
    exit(1);
  }
  printf("server:got connection from %s\n",inet_ntoa(their_addr.sin_addr));
  /*创建子线程,用于接收信息*/
  if((pthread_create(&recthread,NULL,(void*)recmessage,NULL)) != 0){
    printf("create thread error!\r\n");
    exit(1);
  }
}

int main(void){
  struct sockaddr_in my_addr;
  /*创建套接字*/
  if((sockfd = socket(AF_INET,SOCK_STREAM,0)) == -1){
    perror("socket");
    exit(1);
  }
  /*初始化sockaddr_in结构体相关参数*/
  my_addr.sin_family = AF_INET;
  my_addr.sin_port = htons(MYPORT);
  my_addr.sin_addr.s_addr = INADDR_ANY;
  bzero(&(my_addr.sin_zero),8);

  /*绑定端口与套接字*/
  if(bind(sockfd,(struct sockaddr*)&my_addr,sizeof(struct sockaddr)) == -1){
    perror("bind");
    exit(1);
  }
  /*监听客户端套接字*/
  if(listen(sockfd,BACKLOG)== -1){
    perror("listen");
    exit(1);
  }
  printf("listening...\n");
  /*创建子线程,用于接收信息*/
  if((pthread_create(&accthread,NULL,(void*)acceptconnect,NULL))!=0){
    printf("create thread error!\n");
    exit(1);
  }
  while(1){
    char msg[MAXDATASIZE];
    scanf("%s",msg);
    if(send(new_fd,msg,strlen(msg),0) == -1){  //发送信息,与客户端交流
      perror("send");
      exit(1);
    }
    if(strcmp(msg,"exit") ==0){
      printf("byebye\n");
      close(new_fd);
      close(sockfd);
      exit(1);
    }
  }/*while*/
  return 0;
}/*main*/

2.客户端

/*tcpclient.c    2010.9.1 by yyg*/
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/wait.h>
#include <pthread.h>
#include <netdb.h>

#define PORT 3490
#define BACKLOG 10
#define MAXDATASIZE 1024

int sockfd;
pthread_t recthread;

/*接收信息函数*/
void recmessage(void){
  while(1){
    int numbytes;
    char buf[MAXDATASIZE];
    
    if((numbytes = recv(sockfd,buf,MAXDATASIZE,0))==-1){
      perror("recv");
      exit(1);
    }
    buf[numbytes]='\0';
    if(strcmp(buf,"exit")==0){
      printf("Server is closed\n");
      close(sockfd);
      exit(1);
    }
    printf("Server:%s\n",buf);
  }/*while*/
}

int main(int argc,char *argv[]){
  struct hostent *he;
  struct sockaddr_in their_addr;
  /*客户端输入方式:./client 172.31.100.236,若无输入后面IP地址,会提示错误*/
  if(argc != 2){
    fprintf(stderr,"usage:client hostname\n");
    exit(1);
  }
  /*获取主机IP地址*/
  if((he = gethostbyname(argv[1])) == NULL){
    herror("gethostbyname");
    exit(1);
  }
  /*创建套接字*/
  if((sockfd = socket(AF_INET,SOCK_STREAM,0)) == -1){
    perror("socket");
    exit(1);
  }
  /*初始化sockaddr_in结构体*/
  their_addr.sin_family = AF_INET;
  their_addr.sin_port = htons(PORT);
  their_addr.sin_addr = *((struct in_addr *)he->h_addr);
  bzero(&(their_addr.sin_zero),8);
  /*向服务器发送连接请求*/
  if(connect(sockfd,(struct sockaddr *)&their_addr,sizeof(struct sockaddr)) == -1){
    perror("connect");
    exit(1);
  }
  /*创建子线程,用于接收信息*/
  if((pthread_create(&recthread,NULL,(void*)recmessage,NULL))!= 0){
    printf("create thread error!\r\n");
    exit(1);
  }
  /*发送信息。接收发送信息用的是同一端口,都 是sockfd*/
  while(1){
    char msg[MAXDATASIZE];
    scanf("%s",msg);
    if(send(sockfd,msg,strlen(msg),0) == -1){
      perror("send");
      exit(1);
    }
    if(strcmp(msg,"exit") ==0){
      printf("byebye\n");
      close(sockfd);
      exit(1);
    }
  }/*while*/
  return 0;
}

运行结果:

终端1:

[root@localhost net]# ./tcpserver
server:got connection from 172.31.100.236
Hello,World! 
client:hello
client:xiaolian
client:iou
ok
exit
byebye

终端2:

[root@localhost net]# ./tcpclient 172.31.100.236
Server:Hello,World!
hello
xiaolian iou
Server:ok
Server is closed

The above is the detailed content of How to use linux socket to send and receive messages independently using threads. For more information, please follow other related articles on the PHP Chinese website!

Statement:
This article is reproduced at:yisu.com. If there is any infringement, please contact admin@php.cn delete