tcp_server.cpp
2.48 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
#include <iostream>
using namespace std;
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>
#define SIZE 1024
void connectToClient(int &clientfd) {
char recvBuf[SIZE]; // 接收缓冲区
char sendBuf[SIZE]; // 发送缓冲区
int recvRet, sendRet; // 接受反馈及发送反馈
while (clientfd != -1) {
// receive data
recvRet = recv(clientfd, recvBuf, sizeof(recvBuf), 0);
while(recvRet > 0) {
recvBuf[recvRet] = '\0';
if(strcmp(recvBuf, "$") == 0)
break;
cout << "recv data from client " << clientfd << ", data : " << recvBuf << endl;
recvRet = recv(clientfd, recvBuf, sizeof(recvBuf), 0);
}
// // send data
// cout << clientfd << ">> ";
// cin >> sendBuf;
// while(sendBuf) {
// sendRet = send(clientfd, sendBuf, sizeof(sendBuf), 0);
// if (strcmp(sendBuf, "$") == 0)
// break;
// if (sendRet == -1)
// cout <<"send data error." << endl;
// else
// cout << "Sent data to client: " << sendBuf << endl;
// cout << clientfd << ">> ";
// cin >> sendBuf;
// }
// // 是否关闭连接
// recvRet = recv(clientfd, recvBuf, sizeof(recvBuf), 0);
// if (recvRet == -1)
// cout << "Error receiving data\n";
// else if(strcmp(recvBuf, "y") == 0) {
// cout << "Have closed connection " << clientfd << '.' << endl;
// close(clientfd);
// break;
// }
}
}
int main() {
// create socket
int serverfd = socket(AF_INET, SOCK_STREAM, 0); // IPV4 + TCP
if(-1 == serverfd) {
printf("create socket error");
return -1;
}
// bind port
struct sockaddr_in bindaddr;
bindaddr.sin_family = AF_INET;
bindaddr.sin_addr.s_addr = htonl(INADDR_ANY);
bindaddr.sin_port = htons(30030);
if(-1 == bind(serverfd, (struct sockaddr *)&bindaddr, sizeof(bindaddr))) {
cout << "bind error" << endl;
return -1;
}
// start listen
if (listen(serverfd, 1024) == -1) {
printf("listem error");
return -1;
}
while (1) {
struct sockaddr_in clientaddr;
socklen_t clientaddrlen = sizeof(clientaddr);
// accept connection
int clientfd = accept(serverfd, (struct sockaddr *)&clientaddr, &clientaddrlen);
if(clientfd!= -1) {
// 获取客户端的信息
char cliIp[16];
inet_ntop(AF_INET, &clientaddr.sin_addr.s_addr, cliIp, sizeof(cliIp));
unsigned short cliPort = ntohs(clientaddr.sin_port);
cout << "client ip is : " << cliIp << ", port is : " << cliPort << endl;
connectToClient(clientfd);
}
}
//close socket
close(serverfd);
return 0;
}