您的位置:首页 > 理论基础 > 计算机网络

linux下文件传送程序代码两例(分别基于tcp和udp)

2012-06-07 21:05 567 查看
linux下文件传送程序代码两例(分别基于tcp和udp)

时间:2010-10-27 10:34:37来源:网络 作者:未知 点击:12次

基于TCP的文件传输程序:

基于TCP的文件传输程序:

服务器端代码:

#include <netinet/in.h> // for sockaddr_in

#include <sys/types.h> // for socket

#include <sys/socket.h> // for socket

#include <stdio.h> // for printf

#include <stdlib.h> // for exit

#include <string.h> // for bzero

/*

#include <sys/types.h>

#include <sys/stat.h>

#include <fcntl.h>

#include <unistd.h>

*/

#define HELLO_WORLD_SERVER_PORT 6666

#define LENGTH_OF_LISTEN_QUEUE 20

#define BUFFER_SIZE 1024

#define FILE_NAME_MAX_SIZE 512

int main(int argc, char **argv)

{

//设置一个socket地址结构server_addr,代表服务器internet地址, 端口

struct sockaddr_in server_addr;

bzero(&server_addr,sizeof(server_addr)); //把一段内存区的内容全部设置为0

server_addr.sin_family = AF_INET;

server_addr.sin_addr.s_addr = htons(INADDR_ANY);

server_addr.sin_port = htons(HELLO_WORLD_SERVER_PORT);

//创建用于internet的流协议(TCP)socket,用server_socket代表服务器socket

int server_socket = socket(PF_INET,SOCK_STREAM,0);

if( server_socket < 0)

{

printf("Create Socket Failed!");

exit(1);

}

//把socket和socket地址结构联系起来

if( bind(server_socket,(struct sockaddr*)&server_addr,sizeof(server_addr)))

{

printf("Server Bind Port : %d Failed!", HELLO_WORLD_SERVER_PORT);

exit(1);

}

//server_socket用于监听

if ( listen(server_socket, LENGTH_OF_LISTEN_QUEUE) )

{

printf("Server Listen Failed!");

exit(1);

}

while (1) //服务器端要一直运行

{

//定义客户端的socket地址结构client_addr

struct sockaddr_in client_addr;

socklen_t length = sizeof(client_addr);

//接受一个到server_socket代表的socket的一个连接

//如果没有连接请求,就等待到有连接请求--这是accept函数的特性

//accept函数返回一个新的socket,这个socket(new_server_socket)用于同连接到的客户的通信

//new_server_socket代表了服务器和客户端之间的一个通信通道

//accept函数把连接到的客户端信息填写到客户端的socket地址结构client_addr中

int new_server_socket = accept(server_socket,(struct sockaddr*)&client_addr,&length);

if ( new_server_socket < 0)

{

printf("Server Accept Failed! ");

break;

}

char buffer[BUFFER_SIZE];

bzero(buffer, BUFFER_SIZE);

length = recv(new_server_socket,buffer,BUFFER_SIZE,0);

if (length < 0)

{

printf("Server Recieve Data Failed! ");

break;

}

char file_name[FILE_NAME_MAX_SIZE+1];

bzero(file_name, FILE_NAME_MAX_SIZE+1);

strncpy(file_name, buffer, strlen(buffer)>FILE_NAME_MAX_SIZE?FILE_NAME_MAX_SIZE:strlen(buffer));

// int fp = open(file_name, O_RDONLY);

// if( fp < 0 )

FILE * fp = fopen(file_name,"a");

if(NULL == fp )

{

printf("File:\t%s Can Not creat file! ", file_name);

}

else

{

bzero(buffer, BUFFER_SIZE);

int file_block_length = 0;

// while( (file_block_length = read(fp,buffer,BUFFER_SIZE))>0)

int length = 0;

while( length = recv(new_server_socket,buffer,BUFFER_SIZE,0))

{

if(length < 0)

{

printf("Recieve Data From Server %s Failed! ", argv[1]);

break;

}

// int write_length = write(fp, buffer,length);

int write_length = fwrite(buffer,sizeof(char),length,fp);

if (write_length<length)

{

printf("File:\t%s Write Failed ", file_name);

break;

}

bzero(buffer,BUFFER_SIZE);

}

// close(fp);

fclose(fp);

printf("Recieve File:\t %s From Server Finished ",file_name);

}

//关闭与客户端的连接

close(new_server_socket);

}

//关闭监听用的socket

close(server_socket);

return 0;

}

客户端代码:

#include <netinet/in.h> // for sockaddr_in

#include <sys/types.h> // for socket

#include <sys/socket.h> // for socket

#include <stdio.h> // for printf

#include <stdlib.h> // for exit

#include <string.h> // for bzero

/*

#include <sys/types.h>

#include <sys/stat.h>

#include <fcntl.h>

#include <unistd.h>

*/

#define HELLO_WORLD_SERVER_PORT 6666

#define BUFFER_SIZE 1024

#define FILE_NAME_MAX_SIZE 512

int main(int argc, char **argv)

{

if (argc != 2)

{

printf("Usage: ./%s ServerIPAddress ",argv[0]);

exit(1);

}

//设置一个socket地址结构client_addr,代表客户机internet地址, 端口

struct sockaddr_in client_addr;

bzero(&client_addr,sizeof(client_addr)); //把一段内存区的内容全部设置为0

client_addr.sin_family = AF_INET; //internet协议族

client_addr.sin_addr.s_addr = htons(INADDR_ANY);//INADDR_ANY表示自动获取本机地址

client_addr.sin_port = htons(0); //0表示让系统自动分配一个空闲端口

//创建用于internet的流协议(TCP)socket,用client_socket代表客户机socket

int client_socket = socket(AF_INET,SOCK_STREAM,0);

if( client_socket < 0)

{

printf("Create Socket Failed! ");

exit(1);

}

//把客户机的socket和客户机的socket地址结构联系起来

if( bind(client_socket,(struct sockaddr*)&client_addr,sizeof(client_addr)))

{

printf("Client Bind Port Failed! ");

exit(1);

}

//设置一个socket地址结构server_addr,代表服务器的internet地址, 端口

struct sockaddr_in server_addr;

bzero(&server_addr,sizeof(server_addr));

server_addr.sin_family = AF_INET;

if(inet_aton(argv[1],&server_addr.sin_addr) == 0) //服务器的IP地址来自程序的参数

{

printf("Server IP Address Error! ");

exit(1);

}

server_addr.sin_port = htons(HELLO_WORLD_SERVER_PORT);

socklen_t server_addr_length = sizeof(server_addr);

//向服务器发起连接,连接成功后client_socket代表了客户机和服务器的一个socket连接

if(connect(client_socket,(struct sockaddr*)&server_addr, server_addr_length) < 0)

{

printf("Can Not Connect To %s! ",argv[1]);

exit(1);

}

char file_name[FILE_NAME_MAX_SIZE+1];

bzero(file_name, FILE_NAME_MAX_SIZE+1);

printf("Please Input File Name On Server:\t");

scanf("%s", file_name);

char buffer[BUFFER_SIZE];

bzero(buffer,BUFFER_SIZE);

strncpy(buffer, file_name, strlen(file_name)>BUFFER_SIZE?BUFFER_SIZE:strlen(file_name));

//向服务器发送buffer中的数据

send(client_socket,buffer,BUFFER_SIZE,0);

// int fp = open(file_name, O_WRONLY|O_CREAT);

// if( fp < 0 )

FILE * fp = fopen(file_name,"r");

if(NULL == fp )

{

printf("nFile:\t%s Not Found ", file_name);

exit(1);

}

//从服务器接收数据到buffer中

bzero(buffer,BUFFER_SIZE);

int file_block_length;

while( (file_block_length = fread(buffer,sizeof(char),BUFFER_SIZE,fp))>0)

{

printf("file_block_length = %d ",file_block_length);

//发送buffer中的字符串到new_server_socket,实际是给服务器端

if(send(client_socket,buffer,file_block_length,0)<0)

{

printf("Send File:\t%s Failed ", file_name);

break;

}

bzero(buffer, BUFFER_SIZE);

}

printf("File:\t%s Transfer Finished ",file_name);

close(fp);

//关闭socket

close(client_socket);

return 0;

}

测试过程:

服务器端: [root@localhost c++]# ./file_serve

Recieve File: cxt From Server Finished

客户端: [root@localhost file_transmit]# ./file_clien 192.168.0.130

Please Input File Name On Server: cxt

file_block_length = 49

File: cxt Transfer Finished

运行结果:客户端将cxt文件传送到服务器端。

基于UDP的文件传输程序:

server.c:

#include<sys/socket.h>;

#include<string.h>;

#include<netinet/in.h>;

#include<stdio.h>;

#include<stdlib.h>;

#include<fcntl.h>;

#include<sys/stat.h>;

#include<unistd.h>;

#include<errno.h>;

#include<sys/select.h>;

#include<sys/time.h>;

#include<unistd.h>;

#include<sys/types.h>;

#define SERV_PORT 2500

#define MAX_SIZE 1024*40

void recvUDP(char name[20],int sockfd)

{

int ret,fd;

mode_t fdmode = (S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH);

char mesg[MAX_SIZE];

fd_set rdset;

struct timeval tv;

int rlen,wlen;

int i;

fd = open(name,O_RDWR|O_CREAT|O_APPEND,fdmode);

if(fd == -1)

{

printf("open file %s error:%n",name,strerror(errno));

exit(-1);

}

for(i=0;;i++)

{

tv.tv_sec = 1;

tv.tv_usec = 0;

FD_ZERO(&rdset);

FD_SET(sockfd,&rdset);

ret = select(sockfd+1,&rdset,NULL,NULL,&tv);

if(ret == -1)

{

printf("select error %s ",strerror(errno));

exit(-1);

}

else if(ret==0)

{

printf("select timeout,continue circle ");

continue;

}

if(FD_ISSET(sockfd,&rdset))

{

memset(mesg,0,MAX_SIZE);

rlen = read(sockfd,mesg,MAX_SIZE);

if(rlen <=0 )

{

printf("read error %s ",strerror(errno));

exit(-1);

}

if(!strcmp(mesg,"end"))

{

printf("recv end. ");

break;

}

wlen = write(fd,mesg,rlen);

if(wlen != rlen )

{

printf("write error %s ",strerror(errno));

exit(-1);

}

}

printf("The %d times write ",i);

}

close(fd);

}

int main(int argc, char *argv[])

{

int sockfd;

int r;

struct sockaddr_in servaddr;

sockfd = socket(AF_INET,SOCK_DGRAM,0); /*create a socket*/

/*init servaddr*/

bzero(&servaddr,sizeof(servaddr));

servaddr.sin_family = AF_INET;

servaddr.sin_addr.s_addr = htonl(INADDR_ANY);

servaddr.sin_port = htons(SERV_PORT);

/*bind address and port to socket*/

if(bind(sockfd,(struct sockaddr *)&servaddr,sizeof(servaddr)) == -1)

{

perror("bind error");

exit(-1);

}

r = fcntl(sockfd, F_GETFL, 0);

fcntl(sockfd, F_SETFL, r & ~O_NONBLOCK);

recvUDP(argv[1],sockfd);

return 0;

}

client.c:

#include<sys/types.h>;

#include<sys/socket.h>;

#include<string.h>;

#include<netinet/in.h>;

#include<stdio.h>;

#include<stdlib.h>;

#include<arpa/inet.h>;

#include<fcntl.h>;

#include<sys/stat.h>;

#include<errno.h>;

#include<sys/sysinfo.h>;

#include<sys/select.h>;

#include<sys/time.h>;

#include<unistd.h>;

#define MAX_SIZE 1024*40

#define SERV_PORT 2500

void connectUDP(char name[20],int sockfd,struct sockaddr *pservaddr,socklen_t servlen)

{

char buf[MAX_SIZE];

fd_set wrset;

struct timeval tv;

int rlen,wlen;

int fd;

int ret;

int i;

if(connect(sockfd,(struct sockaddr *)pservaddr,servlen) == -1)

{

perror("connet error");

exit(1);

}

else

printf("connect server ok! ");

fd = open(name,O_RDONLY);

if(fd==-1)

{

printf("fopen error %s ",strerror(errno));

exit(-1);

}

i=0;

while(1)

{

tv.tv_sec = 1;

tv.tv_usec = 0;

FD_ZERO(&wrset);

FD_SET(sockfd,&wrset);

ret = select(sockfd+1,NULL,&wrset,NULL,&tv);

if(ret == -1)

{

printf("select error %s ",strerror(errno));

exit(-1);

}

else if(ret==0)

{

printf("select timeout,continue circle ");

continue;

}

memset(buf,0,MAX_SIZE);

if(FD_ISSET(sockfd,&wrset))

{

rlen = read(fd,buf,MAX_SIZE);

if(rlen <0)

{

printf("fread data error %s ",strerror(errno));

exit(-1);

}

else if(rlen==0)

{

wlen = write(sockfd,"end",3);

if(wlen !=3)

{

printf("write end error ",strerror(errno));

exit(-1);

}

printf("all complete ");

close(fd);

close(sockfd);

exit(0);

}

wlen = write(sockfd,buf,rlen);

if(wlen != rlen)

{

printf("write data to sockfd error:%s ",strerror(errno));

exit(-1);

}

i++;

usleep(500);

printf("The %d times read ",i);

}

}

}

int main(int argc ,char *argv[])

{

char *fh;

struct sysinfo s_info;

float time1,time2;

int error1,error2;

int sockfd;

struct stat fsize;

struct sockaddr_in servaddr;

error1= sysinfo(&s_info);

time1 = s_info.uptime;

int r;

if(argc != 3)

{

printf("useage:udpclient<IPaddress>; ");

exit(1);

}

bzero(&servaddr,sizeof(servaddr));

servaddr.sin_family= AF_INET;

servaddr.sin_port = htons(SERV_PORT);

if(inet_pton(AF_INET,argv[1],&servaddr.sin_addr) <= 0)

{

printf("[%s]is not a valid IPaddress ",argv[1]);

exit(1);

}

fh = argv[2];

sockfd =socket(AF_INET,SOCK_DGRAM,0);

r = fcntl(sockfd, F_GETFL, 0);

fcntl(sockfd, F_SETFL, r & ~O_NONBLOCK);

connectUDP(argv[2],sockfd,(struct sockaddr *)&servaddr,sizeof(servaddr));

fprintf(stderr,"ServerIP:\t%s ",argv[1]);

if(stat(argv[2],&fsize) == -1)

perror("failed to get fiel statusi ");

else

fprintf(stderr,"file name:\t%s file size:\t%dK ",argv[2],fsize.st_size/1024);

error2=sysinfo(&s_info);

time2 = s_info.uptime;

printf("tranfice file time =%fs ",(time2-time1));

}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: