udp传送文件

本文介绍了一个基于UDP协议的简单文件传输程序,包括服务器端和客户端的C语言代码实现。通过使用select模型处理I/O多路复用,实现文件的数据发送与接收。同时,提供了Makefile用于编译程序。
务器端:

#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\n",strerror(errno));
                        exit(-1);
                }
                else if(ret==0)
                {
                        printf("select timeout,continue circle\n");
                        continue;
                }
                
                if(FD_ISSET(sockfd,&rdset))
                {
                        memset(mesg,0,MAX_SIZE);
                        rlen = read(sockfd,mesg,MAX_SIZE);
                        if(rlen <=0 )
                        {
                                printf("read error %s\n",strerror(errno));
                                exit(-1);
                        }
                        
                        if(!strcmp(mesg,"end"))
                        {
                                printf("recv end.\n");
                                break;
                        }

                        wlen = write(fd,mesg,rlen);
                        if(wlen != rlen )
                        {
                                printf("write error %s\n",strerror(errno));
                                exit(-1);
                        }        
                }
                
                printf("The %d times write\n",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;

}

客户端:

#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!\n");

                fd = open(name,O_RDONLY);
                if(fd==-1)
                {
                        printf("fopen error %s\n",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\n",strerror(errno));
                                exit(-1);
                        }
                        else if(ret==0)
                        {
                                printf("select timeout,continue circle\n");
                                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\n",strerror(errno));
                                        exit(-1);
                                }
                                else if(rlen==0)
                                {
                                        wlen = write(sockfd,"end",3);
                                        if(wlen !=3)
                                        {
                                                printf("write end error\n",strerror(errno));
                                                exit(-1);
                                        }

                                        printf("all complete\n");

                                        close(fd);
                                        close(sockfd);
                                        exit(0);
                                }

                                wlen = write(sockfd,buf,rlen);
                                if(wlen != rlen)
                                {
                                        printf("write data to sockfd error:%s\n",strerror(errno));
                                        exit(-1);
                                }
                                
                                i++;
                                
                                usleep(500);
                                printf("The %d times read\n",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>;\n");
                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\n",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\n",argv[1]);        
        if(stat(argv[2],&fsize) == -1)
                perror("failed to get fiel statusi\n");
        else        
                fprintf(stderr,"file name:\t%s\nfile size:\t%dK\n",argv[2],fsize.st_size/1024);
        error2=sysinfo(&s_info);
        time2 = s_info.uptime;
        printf("tranfice file time =%fs\n",(time2-time1));
}               

Makefile

all:send recv

send:send.c
        gcc -Wall send.c -o send
recv:recv.c
        gcc -Wall recv.c -o recv
        
clean:
        rm -rf send recv

### 实现基于UDP协议的文件传输 虽然提到过TCP更适合文件传输[^1],但在某些特定场景下(如实时性要求较高而对可靠性要求较低的情况),也可以考虑使用UDP协议进行文件传输。下面是一个简单的基于UDP协议的文件传输示例。 #### 客户端代码 以下是客户端代码,负责向服务器发送文件: ```python import socket def send_file_udp(server_ip, server_port, file_path): try: sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) with open(file_path, 'rb') as f: data = f.read() chunk_size = 1024 # 设置每次发送的数据大小 chunks = [data[i:i+chunk_size] for i in range(0, len(data), chunk_size)] for chunk in chunks: sock.sendto(chunk, (server_ip, server_port)) except Exception as e: print(f"Error occurred while sending the file: {e}") finally: sock.close() if __name__ == "__main__": SERVER_IP = "127.0.0.1" SERVER_PORT = 12345 FILE_PATH = "./example.txt" send_file_udp(SERVER_IP, SERVER_PORT, FILE_PATH) ``` 此代码片段实现了将指定路径下的文件分片并通过UDP协议发送到目标服务器的功能[^3]。 #### 服务器端代码 下面是服务器端代码,用于接收并保存来自客户端的文件数据: ```python import socket def receive_file_udp(port, output_file_path): try: sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) sock.bind(("0.0.0.0", port)) received_data = [] while True: data, addr = sock.recvfrom(1024) # 缓冲区大小设置为1024字节 if not data: break received_data.append(data) with open(output_file_path, 'wb') as f: for d in received_data: f.write(d) except Exception as e: print(f"An error occurred during receiving the file: {e}") finally: sock.close() if __name__ == "__main__": PORT = 12345 OUTPUT_FILE_PATH = "./received_example.txt" receive_file_udp(PORT, OUTPUT_FILE_PATH) ``` 上述代码展示了如何通过UDP协议接收文件,并将其写入本地磁盘。 需要注意的是,在实际应用中可能还需要加入更多的错误检测和纠正机制以提高系统的健壮性[^4]。 ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值