Files
mkrtos-real/mkrtos_user/user/app/test/net_test.c
zhangzheng 00804aaaaa optim sys
2025-02-12 17:29:30 +08:00

151 lines
3.7 KiB
C

#include "pthread.h"
#include "unistd.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#define PORT 8080
#define BUFFER_SIZE 128
static int tcp_server(void)
{
int server_socket, client_socket;
struct sockaddr_in server_addr, client_addr;
char buffer[BUFFER_SIZE];
// 创建服务器套接字
if ((server_socket = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
perror("socket creation failed");
exit(EXIT_FAILURE);
}
printf("create server socket is success:%d\n", server_socket);
// 设置服务器地址
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = INADDR_ANY;
server_addr.sin_port = htons(PORT);
// 绑定套接字到本地地址和端口
if (bind(server_socket, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0)
{
perror("bind failed");
close(server_socket);
exit(EXIT_FAILURE);
}
// 监听连接请求
if (listen(server_socket, 3) < 0)
{
perror("listen failed");
close(server_socket);
exit(EXIT_FAILURE);
}
printf("Server is listening on port %d...\n", PORT);
socklen_t acc_len;
// 接收客户端连接
client_socket = accept(server_socket, (struct sockaddr *)&client_addr, &acc_len);
if (client_socket < 0)
{
perror("accept failed");
close(server_socket);
exit(EXIT_FAILURE);
}
printf("Client connected!\n");
// 发送数据
const char *message = "Hello from server!";
send(client_socket, message, strlen(message), 0);
// 接收数据
memset(buffer, 0, BUFFER_SIZE);
ssize_t bytes_received = recv(client_socket, buffer, BUFFER_SIZE - 1, 0);
if (bytes_received < 0)
{
perror("recv failed");
close(server_socket);
exit(EXIT_FAILURE);
}
printf("Received message from client: %s\n", buffer);
// 关闭套接字
close(client_socket);
close(server_socket);
return 0;
}
static void *net_server_func(void *arg)
{
tcp_server();
return NULL;
}
static int tcp_client(void)
{
int server_socket;
struct sockaddr_in server_addr;
char buffer[BUFFER_SIZE];
// 创建服务器套接字
if ((server_socket = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
perror("socket creation failed");
exit(EXIT_FAILURE);
}
// 设置服务器地址
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = inet_addr("192.168.3.10"); // 使用本机IP地址
server_addr.sin_port = htons(PORT);
// 连接到服务器
if (connect(server_socket, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0)
{
perror("connect failed");
close(server_socket);
exit(EXIT_FAILURE);
}
printf("Connected to server!\n");
// 发送数据
const char *message = "Hello from client!";
send(server_socket, message, strlen(message), 0);
// 接收数据
memset(buffer, 0, BUFFER_SIZE);
ssize_t bytes_received = recv(server_socket, buffer, BUFFER_SIZE - 1, 0);
if (bytes_received < 0)
{
perror("recv failed");
close(server_socket);
exit(EXIT_FAILURE);
}
printf("Received message from server: %s\n", buffer);
// 关闭套接字
close(server_socket);
return 0;
}
static pthread_t net_server_pt;
int net_test(void)
{
if (pthread_create(&net_server_pt, NULL, net_server_func, NULL))
{
printf("net server pthread create failed.\n");
return -1;
}
sleep(1);
tcp_client();
pthread_join(net_server_pt, NULL);
return 0;
}