Linux网络通信(二)Socket编写TCP/UDP

来源:互联网 发布:河南自考网络助学平台 编辑:程序博客网 时间:2024/05/16 09:59

、通信流程

       TCP:

  

 

   UDP:

 

   部分函数知识可以参考之前写的 Linux 网络编程——TCP

  TCP通信:

 

  客户端:

[cpp] view plain copy
 print?
  1. #include<stdio.h>  
  2. #include<sys/socket.h>  
  3. #include<netinet/in.h>  
  4. #include<stdlib.h>  
  5. #include<memory.h>  
  6. #include<string.h>  
  7.   
  8.   
  9. #define PORT 8888  
  10. int main(int argc,char **argv){  
  11.     int sock;  
  12.     struct sockaddr_in my_addr;  
  13.     int len;  
  14.     char buf[100];  
  15.     if(argc <3){  
  16.         printf("Usage: %s <ip> <port> <message>",argv[0]);  
  17.         exit(1);  
  18.     }  
  19.   
  20.     if((sock=socket(AF_INET,SOCK_STREAM,0))<0){  
  21.         printf("socket create error!\n");  
  22.         exit(1);  
  23.     }  
  24.     my_addr.sin_family=AF_INET;  
  25.     my_addr.sin_port=htons(8888);  
  26.     if(inet_aton(argv[1],(struct in_addr *)&my_addr.sin_addr.s_addr) == 0){  
  27.         printf("%s chage error!\n",argv[1]);  
  28.         exit(1);  
  29.     }  
  30.     if(connect(sock,(struct sockaddr*)&my_addr,sizeof(struct sockaddr))<0){  
  31.         printf("connect error!\n");  
  32.         exit(1);  
  33.     }  
  34.     printf("connected!\n");  
  35.   
  36.     while(1){  
  37.         memset(buf,0,100);  
  38.         len=recv(sock,buf,100,0);  
  39.         if(len<0){  
  40.             printf("recv error!\n");  
  41.             exit(1);  
  42.         }else if(len == 0){  
  43.             printf("the client quit!\n");  
  44.             exit(1);  
  45.         }else{  
  46.             printf("receive message: %s \n",buf);  
  47.         }  
  48.         printf("-------------------------\n");  
  49.         memset(buf,0,100);  
  50.         printf("Input message to send: ");  
  51.         fgets(buf,100,stdin);  
  52.         len=send(sock,buf,strlen(buf)-1,0);  
  53.         if(len<0){  
  54.             printf("send error!\n");  
  55.             exit(1);  
  56.         }else{  
  57.             printf("send succeed! send : %s\n",buf);  
  58.         }  
  59.     }  
  60.     close(sock);  
  61. }  


  服务端:

[cpp] view plain copy
 print?
  1. #include<stdio.h>  
  2. #include<sys/socket.h>  
  3. #include<netinet/in.h>  
  4. #include<stdlib.h>  
  5. #include<memory.h>  
  6. #include<string.h>  
  7. #include<sys/socket.h>  
  8.   
  9. #define PORT 8888  
  10. int main(){  
  11.     int sock,new_sock;  
  12.     struct sockaddr_in my_addr,client_addr;  
  13.     int len;  
  14.     char buf[100];  
  15.   
  16.     if((sock=socket(AF_INET,SOCK_STREAM,0))<0){  
  17.         printf("socket create error!\n");  
  18.         exit(1);  
  19.     }  
  20.     memset(&my_addr,0,sizeof(my_addr));  
  21.     my_addr.sin_family=AF_INET;  
  22.     my_addr.sin_port=htons(PORT);  
  23.     my_addr.sin_addr.s_addr=INADDR_ANY;  
  24.   
  25.     if(bind(sock,(struct sockaddr*)&my_addr,sizeof(my_addr))==-1){  
  26.         printf("bind error!\n");  
  27.         exit(1);  
  28.     }  
  29.     if(listen(sock,5)<0){  
  30.         printf("listen error!\n");  
  31.         exit(1);  
  32.     }  
  33.     len=sizeof(struct sockaddr);  
  34.     if((new_sock=accept(sock,(struct sockaddr*)&client_addr,&len))<0){  
  35.         printf("accept error!\n");  
  36.         exit(1);  
  37.     }else{  
  38.         printf("server: get connection from %s,port %d socket %d \n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port),new_sock);  
  39.   
  40.     }  
  41.     while(1){  
  42.         memset(buf,0,100);  
  43.         printf("Input message to send: ");  
  44.         fgets(buf,100,stdin);  
  45.         if(!strncasecmp(buf,"quit",4)){  
  46.             printf("server is closing...\n");  
  47.             break;  
  48.         }  
  49.         len=send(new_sock,buf,strlen(buf)-1,0);  
  50.         if(len<0){  
  51.             printf("send error!\n");  
  52.             exit(1);  
  53.         }else{  
  54.             printf("send succeed! send : %s\n",buf);  
  55.         }  
  56.         printf("------------------------------\n");  
  57.         memset(buf,0,100);  
  58.         len=recv(new_sock,buf,100,0);  
  59.         if(len<0){  
  60.             printf("recv error!\n");  
  61.             exit(1);  
  62.         }else if(len == 0){  
  63.             printf("the client quit!\n");  
  64.             break;  
  65.         }else{  
  66.             printf("receive message: %s \n",buf);  
  67.         }  
  68.     }  
  69.     close(sock);  
  70.     close(new_sock);  
  71. }  
  72.            


 

         相对于TCP来说UDP就要简单写,毕竟步骤少嘛:

 

     服务端 :

[cpp] view plain copy
 print?
  1. #include <stdio.h>  
  2. #include <string.h>  
  3. #include <sys/types.h>  
  4. #include <netinet/in.h>  
  5. #include <sys/socket.h>  
  6. #include <errno.h>  
  7. #include <stdlib.h>  
  8. #include <arpa/inet.h>  
  9.   
  10. int main(int argc, char **argv){  
  11.     struct sockaddr_in s_addr,c_addr;  
  12.     int sock;  
  13.     socklen_t addr_len;  
  14.     int len;  
  15.     char buff[128];  
  16.   
  17.     if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == -1) {  
  18.         perror("socket");  
  19.         exit(errno);  
  20.     } else  
  21.         printf("create socket.\n\r");  
  22.   
  23.     memset(&s_addr, 0, sizeof(struct sockaddr_in));  
  24.   
  25.     s_addr.sin_family = AF_INET;  
  26.     s_addr.sin_port = htons(7838);  
  27.     s_addr.sin_addr.s_addr = INADDR_ANY;  
  28.   
  29.     if ((bind(sock, (struct sockaddr *) &s_addr, sizeof(s_addr))) == -1) {  
  30.         perror("bind");  
  31.         exit(errno);  
  32.     } else  
  33.         printf("bind address to socket.\n\r");  
  34.   
  35.     addr_len = sizeof(c_addr);  
  36.     while (1) {  
  37.                   memset(buff,0,sizeof(buff));  
  38.                   //第三个参数标识为0,第四个参数为信息来源主机地址信息  
  39.         len = recvfrom(sock, buff, sizeof(buff) - 1, 0, (struct sockaddr *) &c_addr, &addr_len);  
  40.         if (len < 0) {  
  41.             perror("recvfrom");  
  42.             exit(errno);  
  43.         }  
  44.   
  45.         buff[len] = '\0';  
  46.         printf("recive come from %s:%d message:%s\n\r",inet_ntoa(c_addr.sin_addr), ntohs(c_addr.sin_port), buff);  
  47.     }  
  48.     return 0;  
  49. }  

 

     客户端:

[cpp] view plain copy
 print?
  1. #include <stdio.h>  
  2. #include <string.h>  
  3. #include <sys/types.h>  
  4. #include <netinet/in.h>  
  5. #include <sys/socket.h>  
  6. #include <errno.h>  
  7. #include <stdlib.h>  
  8. #include <arpa/inet.h>  
  9.   
  10. int main(int argc, char **argv){  
  11.     struct sockaddr_in s_addr;  
  12.     int sock;  
  13.     int addr_len;  
  14.     int len;  
  15.     char buff[128];  
  16.   
  17.     if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == -1) {  
  18.         perror("socket");  
  19.         exit(errno);  
  20.     } else  
  21.         printf("create socket.\n\r");  
  22.   
  23.     s_addr.sin_family = AF_INET;  
  24.     s_addr.sin_port = htons(7838);  
  25.     if (argc>2){  
  26.         s_addr.sin_addr.s_addr = inet_addr(argv[1]);  
  27.         strcpy(buff,argv[2]);  
  28.     }  
  29.     else {  
  30.         printf("input sever ip!\n");  
  31.         exit(0);  
  32.     }  
  33.   
  34.     addr_len = sizeof(s_addr);  
  35.     len = sendto(sock, buff, strlen(buff), 0, (struct sockaddr *) &s_addr, addr_len);  
  36.     if (len < 0) {  
  37.         printf("\n\rsend error.\n\r");  
  38.         return 3;  
  39.     }  
  40.   
  41.     printf("send success.\n\r");  
  42.     return 0;  
  43. }  


0 0
原创粉丝点击