linux C UDP单线程运行没问题,但是服务端改成多线程,就输出乱码,写入的文件大小也不对,可能是逻辑上有问题。
服务端代码:
#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <string.h>
#include <netinet/in.h>
#include <errno.h>
#include <memory.h>
#include <stdlib.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <pthread.h>
#define SPORT 5050 // 服务器端口号
void *Handle_file_recv(void *arg);
struct ARG{
int sockfd;
struct sockaddr_in client;
char filename[50];
};
int main()
{
int sockfd;
struct ARG arg;
pthread_t thread; // 声明线程ID
char filename[50] = {};
int addrlen = sizeof(struct sockaddr);
struct sockaddr_in serAddr;
memset(&serAddr, 0, sizeof(serAddr));
serAddr.sin_family = AF_INET;
// 端口自行修改
serAddr.sin_port = htons(SPORT);
serAddr.sin_addr.s_addr = htonl(INADDR_ANY);
// 创建socket
printf("服务器创建socket...\n");
sockfd = socket(AF_INET, SOCK_DGRAM, 0);
if (0 > sockfd)
{
perror("socket");
return -1;
}
// bind
printf("绑定socket与地址...\n");
if (bind(sockfd, (struct sockaddr*)&serAddr, sizeof(serAddr)) == -1)
{
perror("bind error");
return -1;
}
else
{
printf("bind 成功!\n");
}
while(1)
{
printf("等待中...\n");
struct sockaddr_in client;
memset(filename, 0, sizeof(filename)); // 初始化filename
int f_size = recvfrom(sockfd, filename, sizeof(filename), 0, (struct sockaddr*)&client, &addrlen);
// 显示client端的网络地址
printf("收到来自 %s 的信息\n", inet_ntoa(client.sin_addr));
if(f_size < 0)
{
printf("接收函数出错!\n");
return -1;
}
arg.sockfd = sockfd;
memcpy((void*)&arg.client, &client, sizeof(client));
strcpy(arg.filename, filename);
if(pthread_create(&thread, NULL, Handle_file_recv, (void *)&arg))
{
perror("thread create error!");
return -1;
}
}
close(sockfd);
return 0;
}
// 处理文件接收
void *Handle_file_recv(void *arg)
{
struct ARG *info;
info = (struct ARG*)arg;
printf("收到文件名:%s\n", info->filename);
char buf[1024] = {};
int r_size = 0;
int w_size = 0;
int addrlen = sizeof(struct sockaddr);
int fd = open(info->filename, O_CREAT | O_RDWR, 0777);
int times = 0;
do
{
times++;
printf("times = %d\t", times);
memset(buf, 0, sizeof(buf)); // 清空buf
r_size = recvfrom(info->sockfd, buf, sizeof(buf), 0, (struct sockaddr*)&info->client, &addrlen);
printf("[收到字符数:%d ", r_size);
w_size = write(fd, buf, r_size);
printf("写入文件字节数:%d]\n", w_size);
} while (r_size == 1024);
usleep(10000); // 延迟一下再输出
printf("*********** 文件:%s 接收完成!***********\n\n", info->filename);
close(fd);
pthread_exit(NULL);
}
客户端代码:
#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <string.h>
#include <netinet/in.h>
#include <errno.h>
#include <memory.h>
#include <stdlib.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <dirent.h>
#include <termios.h>
#include <fcntl.h>
#define SPORT 5050 // 服务器端口号
void put(struct sockaddr_in server);
char* get_str(char* str, size_t len);
int sockfd = 0;
int main()
{
struct sockaddr_in server;
if((sockfd = socket(AF_INET,SOCK_DGRAM,0))<0)
{
printf("socket build error!\n");
return -1;
}
memset(&server, 0, sizeof(server));
server.sin_family= AF_INET;
server.sin_port = htons(SPORT);
if(inet_pton(AF_INET, "192.168.50.1", &server.sin_addr)<0)
{
printf("inet_pton error!\n");
return -1;
}
put(server);
return 0;
}
// 上传
void put(struct sockaddr_in server)
{
int r_size = 0;
int w_size = 0;
char buf[1024] = {};
char pathname[100] = {};
char *filename = malloc(50);
memset(filename, 0, 50);
int fd;
while(1)
{
printf("请输入要上传的文件名:");
get_str(pathname, 100);
fd = open(pathname, O_RDONLY);
if (fd == -1)
{
printf("文件不存在,请重新输入!\n");
continue;
}
break;
}
usleep(100000);
if (strrchr(pathname, '/') == NULL) // 搜索最后一次出现 / 的位置
{
strncpy(filename, pathname, 50);
}
else
{
filename = strrchr(pathname, '/');
filename += 1;
}
int addrlen = sizeof(struct sockaddr);
printf("发送文件名:%s 至服务端\n", filename);
int flag = sendto(sockfd, filename, strlen(filename) + 1, 0, (struct sockaddr *)&server, addrlen);
if(flag < 0)
{
printf("发送文件名%s 失败!", filename);
}
else
{
printf("发送文件名%s 成功!", filename);
}
usleep(100000);
memset(buf, 0, sizeof(buf));
sleep(1);
lseek(fd, 0, SEEK_END);
off_t end_pos = lseek(fd, 0, SEEK_CUR);
lseek(fd, 0, SEEK_SET);
printf("开始发送数据...\n");
int times = 0;
do
{
usleep(10000);
times++;
printf("times = %d\t", times);
r_size = read(fd, buf, 1024);
printf("[读取文件字节数:%d ", r_size);
w_size = sendto(sockfd, buf, r_size, 0, (struct sockaddr *)&server, addrlen);
if(w_size < 0)
{
printf("发送失败!\n");
break;
}
else
{
printf("发送字符数:%d]\n", w_size);
off_t cur_pos = lseek(fd, 0, SEEK_CUR);
if(cur_pos == end_pos && w_size == 1024)
{
times++;
printf("times = %d\t", times);
char end[1] = "\0";
printf("[读取文件字节数:1 ");
w_size = sendto(sockfd, end, sizeof(end), 0, (struct sockaddr *)&server, addrlen);
printf("发送字节数:%d]\n", w_size);
break;
}
}
} while (r_size == 1024);
close(fd);
printf("发送完成!\n");
return;
}
char* get_str(char* str, size_t len)
{
if(NULL == str)
{
puts("空指针异常!");
return NULL;
}
scanf("%99s", str);
return str;
}