Linux網絡編程之套接字

套接字可以讓Linux在網絡上通信，用起來就像管道一樣，當然管道都是單向的，套接字既能寫也能收！

以下是多客戶服務器代碼：

1. #include <stdio.h>
2. #include <stdlib.h>
3. #include <unistd.h>
4. #include <sys/types.h>
5. #include <sys/socket.h>
6. #include <signal.h>
7. #include <arpa/inet.h>
8. #include <string.h>
9. #define PORT 6000
10. #define SIZE 256
12. static void handler()//回收子進程函數
13. {
14. wait(NULL);
15. }
17. int main(int argc, char **argv)
18. {
19. int server_socketfd;
20. int client_socketfd;
21. int ret;
22. pid_t pid;
23. size_t address_len;
24. struct sockaddr_in server_address;
25. struct sockaddr_in client_address;
27. server_socketfd = socket(AF_INET, SOCK_STREAM,0);//創建一個套接字
29. memset(&server_address,0,sizeof(struct sockaddr_in));
31. server_address.sin_family = AF_INET;//協議族設置為ARPA因特網協議
32. server_address.sin_port = htons(PORT);//端口號設為PORT（6000）
33. server_address.sin_addr.s_addr = htonl(INADDR_ANY);//允許任何客戶端可以連接
34. address_len = sizeof(struct sockaddr_in);
36. ret = bind(server_socketfd, (struct sockaddr*)&server_address, address_len);//將創建的套接字和address中的地址綁定在一起
37. if( ret == -1 )
38. {
39. printf("bind failed!\n");
40. exit(0);
41. }
43. ret = listen(server_socketfd, 5);//創建一個監聽5個請求的隊列，也就是說這個服務器允許5個人同時登入
44. if ( ret == -1 )
45. {
46. printf("listen failed!\n");
47. exit(0);
48. }
50. while (1)
51. {
52. client_socketfd = accept(server_socketfd, (struct sockaddr*)&client_address, &address_len);//客戶連接服務器時建立客戶套接字
54. signal(SIGCHLD,handler);//為了避免子進程結束時造成僵屍進程，需要對結束子進程作處理，handler函數負責回收子進程
56. pid = fork();
57. if (pid == 0)//為客戶分配的子進程
58. {
59. char buffer[256];
60. int nread;
61. nread = read(client_socketfd,buffer,sizeof(buffer));
62. printf("%s",buffer);
64. }
65. else if (pid > 0)//父進程把分配的客戶套接字關閉，用來供給下個客戶
66. {
67. close(client_socketfd);//關閉客戶套接字
68. }
69. else
70. {
71. printf("fork faild!\n");
72. exit(0);
73. }
74. }
75. return 0;
76. }

客戶端代碼：

1. #include <stdio.h>
2. #include <stdlib.h>
3. #include <unistd.h>
4. #include <sys/types.h>
5. #include <sys/socket.h>
6. #include <netinet/in.h>
7. #include <arpa/inet.h>
8. #include <string.h>
10. #define SIZE 256
11. #define PORT 6000
13. int main()
14. {
15. int socketfd;
16. int ret;
17. char buffer[SIZE];
18. struct sockaddr_in address;
20. socketfd = socket(AF_INET, SOCK_STREAM, 0);//創建一個套接字
22. address.sin_family = AF_INET;//設置為因特網協議
23. address.sin_port = htons(PORT);//設置端口號
24. inet_pton(AF_INET, "127.0.0.1", &address.sin_addr.s_addr);//設置ip地址
26. ret = connect(socketfd, (struct sockaddr*) &address, sizeof(address));//連接服務器
28. if (ret == -1)
29. {
30. printf("can't connect!\n");
31. exit(0);
32. }
34. printf("send:");
35. fgets(buffer,sizeof(buffer),stdin);
36. write(socketfd,buffer,sizeof(buffer));//把數據傳給服務器
38. close(socketfd);//關閉套接字
39. return 0;
40. }