客服端

  1. // 
  2.  
  3. // main.cpp 
  4.  
  5. // hello_client 
  6.  
  7. // 
  8.  
  9. // Created by app05 on 15-10-22. 
  10.  
  11. // Copyright (c) 2015年 app05. All rights reserved. 
  12.  
  13. // 
  14.  
  15. // 
  16.  
  17. #include 
  18.  
  19. #include 
  20.  
  21. #include 
  22.  
  23. #include 
  24.  
  25. #include 
  26.  
  27. #include 
  28.  
  29. #include 
  30.  
  31. #define BUF_SIZE 100 
  32.  
  33. #define NAME_SIZE 20 
  34.  
  35. void * send_msg(void * arg); 
  36.  
  37. void * recv_msg(void * arg); 
  38.  
  39. void error_handling(char *message); 
  40.  
  41. char name[NAME_SIZE] = "[DEFAULT]"; 
  42.  
  43. char msg[BUF_SIZE]; 
  44.  
  45. int main(int argc, const char * argv[]) { 
  46.  
  47. int sock; 
  48.  
  49. struct sockaddr_in serv_addr; 
  50.  
  51. pthread_t snd_thread, rcv_thread; 
  52.  
  53. void * thread_return; 
  54.  
  55. if(argc != 4) 
  56.  
  57. { 
  58.  
  59. printf("Usage: %s \n", argv[0]); 
  60.  
  61. exit(1); 
  62.  
  63. } 
  64.  
  65. sprintf(name, "[%s]", argv[3]); //聊天人名字，配置到编译器参数里 
  66.  
  67. sock = socket(PF_INET, SOCK_STREAM, 0); 
  68.  
  69. if(sock == -1) 
  70.  
  71. error_handling("socket() error"); 
  72.  
  73. memset(&serv_addr, 0, sizeof(serv_addr)); 
  74.  
  75. serv_addr.sin_family = AF_INET; 
  76.  
  77. serv_addr.sin_addr.s_addr = inet_addr(argv[1]); 
  78.  
  79. serv_addr.sin_port = htons(atoi(argv[2])); 
  80.  
  81. if (connect(sock, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) == -1) 
  82.  
  83. error_handling("connect() error"); 
  84.  
  85. //多线程分离输入和输出 
  86.  
  87. pthread_create(&snd_thread, NULL, send_msg, (void *)&sock); 
  88.  
  89. pthread_create(&rcv_thread, NULL, recv_msg, (void *)&sock); 
  90.  
  91. //阻塞，等待返回 
  92.  
  93. pthread_join(snd_thread, &thread_return); 
  94.  
  95. pthread_join(rcv_thread, &thread_return); 
  96.  
  97. close(sock); 
  98.  
  99. return 0; 
  100.  
  101. } 
  102.  
  103. //发送消息 
  104.  
  105. void * send_msg(void * arg) 
  106.  
  107. { 
  108.  
  109. int sock = *((int *)arg); 
  110.  
  111. char name_msg[NAME_SIZE + BUF_SIZE]; 
  112.  
  113. while (1) { 
  114.  
  115. fgets(msg, BUF_SIZE, stdin); 
  116.  
  117. if (!strcmp(msg, "q\n") || !strcmp(msg, "Q \n")) { 
  118.  
  119. close(sock); 
  120.  
  121. exit(0); 
  122.  
  123. } 
  124.  
  125. sprintf(name_msg, "%s %s", name, msg); 
  126.  
  127. write(sock, name_msg, strlen(name_msg)); 
  128.  
  129. } 
  130.  
  131. return NULL; 
  132.  
  133. } 
  134.  
  135. //接收消息 
  136.  
  137. void * recv_msg(void * arg) 
  138.  
  139. { 
  140.  
  141. int sock = *((int *)arg); 
  142.  
  143. char name_msg[NAME_SIZE + BUF_SIZE]; 
  144.  
  145. int str_len; 
  146.  
  147. while (1) { 
  148.  
  149. str_len = read(sock, name_msg, NAME_SIZE + BUF_SIZE - 1); 
  150.  
  151. if(str_len == -1) 
  152.  
  153. return (void *)-1; 
  154.  
  155. name_msg[str_len] = 0; 
  156.  
  157. fputs(name_msg, stdout); 
  158.  
  159. } 
  160.  
  161. return NULL; 
  162.  
  163. } 
  164.  
  165. void error_handling(char *message) 
  166.  
  167. { 
  168.  
  169. fputs(message, stderr); 
  170.  
  171. fputc('\n', stderr); 
  172.  
  173. exit(1); 
  174.  
  175. }

 TCP/IP网络编程 多线程服务器端的实现

 TCP/IP网络编程 多线程服务器端的实现

 TCP/IP网络编程 多线程服务器端的实现


共4页: 上一页1234下一页

相关内容