首頁 > 運維 > linux運維 > linux socket怎麼實作使用執行緒獨立收發訊息

linux socket怎麼實作使用執行緒獨立收發訊息

WBOY
發布: 2023-05-16 17:16:20
轉載
1170 人瀏覽過

代码实现

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
登入後複製

以上是linux socket怎麼實作使用執行緒獨立收發訊息的詳細內容。更多資訊請關注PHP中文網其他相關文章!

相關標籤:
來源:yisu.com
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板