基于TCP协议的文件传输(套接字)实例:
客户端:只用于收文件;
执行命令:./执行文件名 IP Port
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <fcntl.h>
int main(int argc, char** argv)
{
int sock = socket(AF_INET, SOCK_STREAM, 0);
if(-1 == sock)
{
perror("socket");
return 1;
}
struct sockaddr_in srv_addr;
srv_addr.sin_family = AF_INET;
srv_addr.sin_addr.s_addr = inet_addr(argv[1]);
srv_addr.sin_port = htons(atoi(argv[2]));
if(-1 == connect(sock, (struct sockaddr*)&srv_addr, sizeof(srv_addr)))
{
perror("connect");
return 1;
}
char msg[300];
int ret;
ret= recv(sock,msg,sizeof(msg), 0);
int fd = open("msg",O_WRONLY | O_CREAT);
while(1)
{
ret = recv(sock, msg, sizeof(msg), 0);
if(ret <= 0) break;
write(fd,msg,sizeof(ret));
}
close(fd);
close(sock);
return 0;
}
服务器端一:面向所有人,利用while循环发送文件,不能同时向多人发送;
执行命令:./执行文件名 传输文件名
#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <fcntl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <time.h>
int main(int argc, char** argv)
{
int sock_listen = socket(AF_INET, SOCK_STREAM, 0);
if(-1 == sock_listen)
{
perror("socket");
return 1;
}
struct sockaddr_in myaddr;
myaddr.sin_family = AF_INET;
myaddr.sin_addr.s_addr = INADDR_ANY;
myaddr.sin_port = htons(8888);
if(-1 == bind(sock_listen, (struct sockaddr*)&myaddr, sizeof(myaddr)))
{
perror("bind");
return 1;
}
if(-1 == listen(sock_listen, 5))
{
perror("listen");
return 1;
}
while(1)
{
struct sockaddr_in client_addr;
socklen_t len = sizeof(client_addr);
int sock_conn = accept(sock_listen, (struct sockaddr*)&client_addr, &len);
if(-1 == sock_conn)
{
perror("accept");
continue;
}
printf("\n%s:%hu 已连接...\n", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
char file_name[300];
char msg[1024];
int ret;
unsigned long cnt1 = 0, cnt2 = 0;
time_t start, end;
char* p = NULL;
p = strrchr(argv[1], '/');
if(NULL == p)
{
strcpy(file_name, argv[1]);
}
else
{
strcpy(file_name, p + 1);
}
send(sock_conn, file_name, sizeof(file_name), 0);
int fd = open(argv[1], O_RDONLY);
printf("正在努力发送文件...\n");
time(&start);
while((ret = read(fd, msg, sizeof(msg))) > 0)
{
cnt1 += ret;
ret = send(sock_conn, msg, ret, 0);
if(ret == -1)
{
printf("发送文件失败!\n");
break;
}
cnt2 += ret;
}
if(cnt1 != 0 && cnt1 == cnt2)
{
time(&end);
printf("发送文件成功!(耗时 %d 秒)\n", end - start);
}
close(sock_conn);
}
close(sock_listen);
return 0;
}
方法二:利用fork,使用多进程方式提高效率
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <time.h>
#include <sys/types.h>
#include <fcntl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
int main(int argc, char** argv)
{
int sock_listen = socket(AF_INET, SOCK_STREAM, 0);
if(-1 == sock_listen)
{
perror("socket");
return 1;
}
struct sockaddr_in myaddr;
myaddr.sin_family = AF_INET;
myaddr.sin_addr.s_addr = INADDR_ANY;
myaddr.sin_port = htons(8888);
if(-1 == bind(sock_listen, (struct sockaddr*)&myaddr, sizeof(myaddr)))
{
perror("bind");
return 1;
}
if(-1 == listen(sock_listen, 5))
{
perror("listen");
return 1;
}
while(1)
{
struct sockaddr_in client_addr;
socklen_t len = sizeof(client_addr);
int sock_conn = accept(sock_listen, (struct sockaddr*)&client_addr, &len);
if(-1 == sock_conn)
{
perror("accept");
continue;
}
pid_t pid = fork();
if(-1 == pid)
{
perror("fork");
close(sock_conn);
continue;
}
if(pid > 0)
{
close(sock_conn);
continue;
}
close(sock_listen);
printf("\n%s:%hu 已连接...\n", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
char file_name[300];
char msg[1024];
int ret;
unsigned long cnt1 = 0, cnt2 = 0;
time_t start, end;
char* p = NULL;
p = strrchr(argv[1], '/');
if(NULL == p)
{
strcpy(file_name, argv[1]);
}
else
{
strcpy(file_name, p + 1);
}
send(sock_conn, file_name, sizeof(file_name), 0);
int fd = open(argv[1], O_RDONLY);
printf("正在努力发送文件...\n");
time(&start);
while((ret = read(fd, msg, sizeof(msg))) > 0)
{
cnt1 += ret;
ret = send(sock_conn, msg, ret, 0);
if(ret == -1)
{
printf("发送文件失败!\n");
break;
}
cnt2 += ret;
}
if(cnt1 != 0 && cnt1 == cnt2)
{
time(&end);
printf("发送文件成功!(耗时 %lu 秒)\n", end - start);
}
close(sock_conn);
exit(0);
}
close(sock_listen);
return 0;
}
方法三:利用多线程方式
#include <stdio.h>
#include <string.h>
#include <time.h>
#include <sys/types.h>
#include <fcntl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <pthread.h>
static const char* file_path = NULL;
static char file_name[300];
static void* send_file(void* arg);
int main(int argc, char** argv)
{
file_path = argv[1];
char* p = NULL;
p = strrchr(argv[1], '/');
if(NULL == p)
{
strcpy(file_name, argv[1]);
}
else
{
strcpy(file_name, p + 1);
}
int sock_listen = socket(AF_INET, SOCK_STREAM, 0);
if(-1 == sock_listen)
{
perror("socket");
return 1;
}
struct sockaddr_in myaddr;
myaddr.sin_family = AF_INET;
myaddr.sin_addr.s_addr = INADDR_ANY;
myaddr.sin_port = htons(8888);
if(-1 == bind(sock_listen, (struct sockaddr*)&myaddr, sizeof(myaddr)))
{
perror("bind");
return 1;
}
if(-1 == listen(sock_listen, 5))
{
perror("listen");
return 1;
}
while(1)
{
struct sockaddr_in client_addr;
socklen_t len = sizeof(client_addr);
int sock_conn = accept(sock_listen, (struct sockaddr*)&client_addr, &len);
if(-1 == sock_conn)
{
perror("accept");
continue;
}
printf("\n%s:%hu 已连接...\n", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
pthread_t tid;
if(pthread_create(&tid, NULL, send_file, (void*)(long)sock_conn))
{
perror("pthread_create");
close(sock_conn);
continue;
}
}
close(sock_listen);
return 0;
}
void* send_file(void* arg)
{
pthread_detach(pthread_self());
int sock_conn = (int)(long)arg;
char msg[1024];
int ret;
unsigned long cnt1 = 0, cnt2 = 0;
time_t start, end;
send(sock_conn, file_name, sizeof(file_name), 0);
int fd = open(file_path, O_RDONLY);
printf("正在努力发送文件...\n");
time(&start);
while((ret = read(fd, msg, sizeof(msg))) > 0)
{
cnt1 += ret;
ret = send(sock_conn, msg, ret, 0);
if(ret == -1)
{
printf("发送文件失败!\n");
break;
}
cnt2 += ret;
}
if(cnt1 != 0 && cnt1 == cnt2)
{
time(&end);
printf("发送文件成功!(耗时 %lu 秒)\n", end - start);
}
close(sock_conn);
pthread_exit(NULL);
}
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)