87 lines
2.2 KiB
C
87 lines
2.2 KiB
C
// tcp_c1 的服务端
|
||
#include <stdio.h>
|
||
#include <sys/socket.h>
|
||
#include <netinet/in.h>
|
||
#include <arpa/inet.h>
|
||
#include <unistd.h>
|
||
#include <string.h> // bzero
|
||
#include <stdlib.h> // atoi
|
||
|
||
int main(int argc, char const *argv[])
|
||
{
|
||
if (argc != 2)
|
||
{
|
||
printf("usage: %s port\n", argv[0]);
|
||
return 1;
|
||
}
|
||
|
||
// TCP 的 socket 套接字类型 SOCK_STREAM
|
||
int sock_fd = socket(AF_INET, SOCK_STREAM, 0);
|
||
if (sock_fd < 0)
|
||
{
|
||
perror("socket");
|
||
return 1;
|
||
}
|
||
|
||
// 设置绑定：绑定服务器的IP和端口号
|
||
struct sockaddr_in server_addr;
|
||
bzero(&server_addr, sizeof(server_addr));
|
||
server_addr.sin_family = AF_INET;
|
||
server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
|
||
server_addr.sin_port = htons(atoi(argv[1]));
|
||
|
||
bind(sock_fd, (struct sockaddr *)&server_addr, sizeof(server_addr));
|
||
|
||
// 创建监听队列
|
||
if (listen(sock_fd, 50) == -1)
|
||
{
|
||
perror("listen");
|
||
}
|
||
|
||
// 接收客户端的连接
|
||
struct sockaddr_in client_addr;
|
||
bzero(&client_addr, sizeof(client_addr));
|
||
socklen_t client_addr_len = sizeof(client_addr);
|
||
|
||
// 阻塞到有客户端连接为止
|
||
int client_fd = accept(sock_fd, (struct sockaddr *)&client_addr, &client_addr_len);
|
||
|
||
// 客户端 ip
|
||
char client_ip[16] = "";
|
||
inet_ntop(AF_INET, &client_addr.sin_addr.s_addr, client_ip, 16);
|
||
|
||
printf("%s(%d) 连接成功\n", client_ip, ntohs(client_addr.sin_port));
|
||
|
||
while (1)
|
||
{
|
||
// 接受（读取客户端数据）
|
||
char buf[128] = "";
|
||
ssize_t recv_len = read(client_fd, buf, 128); // 用 recv 也可以
|
||
// ssize_t recv_len = recv(client_fd, buf, 128, 0);
|
||
|
||
if (recv_len > 0)
|
||
{
|
||
printf("%s say: %s\n", client_ip, buf);
|
||
}
|
||
|
||
if (strncmp(buf, "bye", 3) == 0)
|
||
{
|
||
break;
|
||
}
|
||
|
||
memset(buf, 0, 128);
|
||
scanf("%s", buf);
|
||
// if (send(client_fd, buf, strlen(buf) + 1, 0) != -1) // 用 send 也可以
|
||
if (write(client_fd, buf, strlen(buf) + 1) != -1) // 用 send 也可以
|
||
{
|
||
printf("发送数据 OK\n");
|
||
}
|
||
}
|
||
|
||
close(client_fd);
|
||
close(sock_fd);
|
||
printf("----over----\n");
|
||
|
||
return 0;
|
||
}
|