基于Udp的文件传输程序

一、题目

用标准C编写linux系统下的基于socket通信的UDP服务器与客户机的文件传输程序。

要求：

1、  编写出服务器程序，并命名为udp_server.c.

2、  编写出客户端程序，并命名为udp_client.c.

3、  以客户机发送字符串StartTransfer开始文件传送，发送字符串QuitTransfer退出文件传送并结束两边的程序。

4、  要传输的文件为mysend.txt文件（文件长度500KB至10MB之间），存在服务器程序的同一目录下，客户端收到文件后将其命名为myrecv.txt,存储在客户机程序的同一个目录。

二、程序

udp_server.c

#include <stdio.h> 
#include <string.h> 
#include <unistd.h> 
#include <ctype.h> 
#include <sys/socket.h> 
#include <netinet/in.h> 
#include <arpa/inet.h> 
#include <stdlib.h>
#include <sys/types.h> 
#include <sys/stat.h> 
#include <fcntl.h> 
#include <sys/mman.h> 

#define MAXLEN 4096 

int     port = 9012;
int     to_port = 5678;
char    to_ip[] = "127.0.0.1"; 
char    startcmd[13] = "StartTransfer";
char    quitcmd[12] = "QuitTransfer";

int main()
{
    struct sockaddr_in local_sin; 
    struct sockaddr_in to_sin;
    int    mysock; 
        
    FILE *fp;
    int filelen;
    
    char buf[MAXLEN];
    
    int len;
    int i;
    int    recvlen = sizeof(struct sockaddr_in),sendlen = sizeof(struct sockaddr_in);

    bzero( &local_sin, sizeof(local_sin)); 
    local_sin.sin_family = AF_INET; 
    local_sin.sin_addr.s_addr = inet_addr("127.0.0.1"); 
    local_sin.sin_port = htons( port ); 
        
    bzero( &to_sin, sizeof(to_sin)); 
    to_sin.sin_family = AF_INET; 
    to_sin.sin_addr.s_addr = inet_addr(to_ip); 
    to_sin.sin_port = htons(to_port); 

    mysock = socket( AF_INET, SOCK_DGRAM, 0 ); // 创建套接字
    int bind_stats = bind( mysock, ( struct sockaddr * )&local_sin, sizeof(local_sin) );   // 绑定
    if(bind_stats == -1){
    perror("bind error");
    exit(-1);
    }
    len=recvfrom(mysock,buf,MAXLEN,0,(struct sockaddr *)&to_sin,&recvlen);
    if(len == -1){
      perror("recvfrom error");
    exit(-1);
    }
    if(strncmp(buf,startcmd,13)==0)  // 开始传输
    {
        // 打开文件
    if((fp=fopen("mysend.txt","r"))==NULL)
        {
            printf("mysend.txt open failure!!\n");
            return -1;
        }
        fseek(fp,0,SEEK_END);
        filelen = ftell(fp);
        fseek(fp,0,SEEK_SET);
           printf( "filelen = %d\n",filelen); 

           while ( 1 ) 
        { 
            bzero(buf,MAXLEN);
            fread(buf,MAXLEN,1,fp);
            if(filelen>=MAXLEN)
            {
            len = sendto(mysock,buf,MAXLEN,0,(struct sockaddr *)&to_sin,sendlen);
            if(len < 0)
            {
                printf("send failure!!\n");
                break;
            }
            filelen-=MAXLEN;
            }
            else
            {
            sendto(mysock,buf,filelen,0,(struct sockaddr *)&to_sin,sendlen);
            if(len < 0)
            {
                printf("send failure!!\n");
                break;
            }
            sendto(mysock,quitcmd,strlen(quitcmd),0,(struct sockaddr *)&to_sin,sendlen);
            break;
            }    
        }
    } 
    printf("Quit\n");
    return 0;
}

udp_client.c

#include <stdio.h> 
#include <string.h> 
#include <unistd.h> 
#include <ctype.h> 
#include <sys/socket.h> 
#include <netinet/in.h> 
#include <arpa/inet.h> 
#include <stdlib.h>
#include <sys/types.h> 
#include <sys/stat.h> 
#include <fcntl.h> 
#include <sys/mman.h> 

#define MAXLEN 4096 

int     port = 5678;
int     to_port = 9012;
char    to_ip[] = "127.0.0.1"; 
char    startcmd[13] = "StartTransfer";
char    quitcmd[12] = "QuitTransfer";

void main()
{
    struct sockaddr_in local_sin; 
    struct sockaddr_in to_sin;
    int    mysock; 
        
    FILE *fp;
    
    char buf[MAXLEN];
    
    int len;
    int i;
    int    recvlen = sizeof(struct sockaddr_in),sendlen = sizeof(struct sockaddr_in);

    bzero( &local_sin, sizeof(local_sin)); 
    local_sin.sin_family = AF_INET; 
    local_sin.sin_addr.s_addr = inet_addr("127.0.0.1");
    local_sin.sin_port = htons( port ); 
        
    bzero( &to_sin, sizeof(to_sin)); 
    to_sin.sin_family = AF_INET; 
    to_sin.sin_addr.s_addr = inet_addr("127.0.0.1");
    to_sin.sin_port = htons(to_port); 

    mysock = socket( AF_INET, SOCK_DGRAM, 0 ); // 创建套接字
    bind( mysock, ( struct sockaddr * )&local_sin, sizeof(local_sin) );   // 绑定

    i = sendto(mysock,startcmd,13,0,(struct sockaddr *)&to_sin,sendlen);   // 开始传输文件
    if(i == -1){
      perror("sendto error");
    exit(-1);
    }
    // 打开文件
    if((fp=fopen("myrecv.txt","w"))==NULL)
    {
        printf("myrecv.txt open failure!!\n");
        return -1;
    }

    while(1)
    {
    bzero(buf,MAXLEN);
        len=recvfrom(mysock,buf,MAXLEN,0,(struct sockaddr *)&to_sin,&recvlen);
        if(strncmp(buf,quitcmd,12)==0)
        {
            fclose(fp);
            break;
        }
        fwrite(buf,len,1,fp);
    }
    close(mysock);
    printf("Quit\n");
}