3.15作业

#include<myhead.h>
#define SER_IP "192.168.117.36"
#define SER_PORT 69
 
//客户端
void menu()
{
    printf("*************************\n");
    printf("*\t 1.下载  \t*\n");
    printf("*\t 2.上传  \t*\n");
    printf("*\t 3.退出  \t*\n");
    printf("*************************\n");
}
int rev()
{
    //下载
    //1、创建用于通信的套接字文件描述符
    int cfd = socket(AF_INET, SOCK_DGRAM, 0);
    if(cfd == -1)
    {
        perror("socket error");
        return -1;
    }
    printf("cfd = %d\n", cfd);          //
 
    //3.填充要连接的服务器地址信息结构体
    struct sockaddr_in sin;
    sin.sin_family = AF_INET;   //地址族
    sin.sin_port = htons(SER_PORT);   //端口号
    sin.sin_addr.s_addr = inet_addr(SER_IP);   //ip地址
 
    //3、收发数据
    //校验收到的块编号
    char filename[128]="";
    printf("请输入要下载的文件名>>>");
    scanf("%s",filename);
 
    //下载请求
    char buf[516] = "";
    int size = sprintf(buf, "%c%c%s%c%s%c", 0, 1, filename, 0, "octet", 0);
 
    if(sendto(cfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin))<0)
    {
        perror("sendto error");
        return -1;
    }
    int flag = 0;
    int fd ;
 
    //循环接收发送应答包
    ssize_t recv_len;
    unsigned short num = 1;
    socklen_t addrlen = sizeof(sin);
    while(1)
    {
        bzero(buf, 516);
        recv_len = recvfrom(cfd, buf, 516, 0, (struct sockaddr*)&sin, &addrlen);
        if(recv_len < 0)
        {
            perror("recvfrom error");
            return -1;
        }
        if(3 == buf[1]) //如果是数据包
        {
            if(0 == flag) //防止文件重复打开
            {
                //存储从服务器下载下来的文件,接收数据成功后再创建数据；
                //创建并打开filename文件
                fd = open(filename, O_WRONLY|O_CREAT|O_TRUNC, 0664);
                if(fd < 0)
                {
                    perror("open error");
                    return -1;
                }
                flag = 1;
            }
            //判断当前的快编号，是否与我期望的快编号一致
            if(htons(num) == *(unsigned short*)(buf+2))     //防止数据包重复到达
            {
                //提取数据，写入到文件中
                if(write(fd, buf+4, recv_len-4) < 0)
                {
                    printf("fd:%d recv_len=%ld\n", fd, recv_len);
                    perror("write error");
                    break;
                }
 
                //回复ack包
                //由于数据包前四个字节与ack包除了操作码不一致外，其余全部一致。
                //所以我们直接将数据包的操作码修改成4即可，发送数据包的前四个字节。
                buf[1] = 4;     
                if(sendto(cfd, buf, 4, 0, (struct sockaddr*)&sin, sizeof(sin)) < 0)
                {
                    perror("sendto error");
                }
 
                //判断数据包的大小是否小于 512+2+2;
                if(recv_len < 512+2+2)
                {
                    printf("-----文件下载完毕!!-----\n");
                    break;
                }
                num++;
            }
        }else if(5 == buf[1])     //错误包
        {
            //打印错误信息
            printf("-----ERROR:%s-----\n", buf+4);
            break;
        }
    }
    //5、关闭套接字
    close(cfd);
}
 
int snd()
{
    //上传
    //1、创建用于通信的套接字文件描述符
    int cfd = socket(AF_INET, SOCK_DGRAM, 0);
    if(cfd == -1)
    {
        perror("socket error");
        return -1;
    }
    printf("cfd = %d\n", cfd);          //3
 
    //3.填充要连接的服务器地址信息结构体
    struct sockaddr_in sin;
    sin.sin_family = AF_INET;   //地址族
    sin.sin_port = htons(SER_PORT);   //端口号
    sin.sin_addr.s_addr = inet_addr(SER_IP);   //ip地址
 
    char filename[128] = "";
    printf("请输入要上传的文件名>>>");
    scanf("%s",filename);
 
    //判断该文件是否存在
    int fd = open(filename, O_RDONLY);
    if(fd < 0)
    {
        perror("open error");
        return -1;
    }
 
    //发送上传请求
    //上传协议
    char buf[516] = "";
    int size = sprintf(buf, "%c%c%s%c%s%c", 0, 2, filename, 0, "octet", 0);
 
    if(sendto(cfd, buf, size, 0, (struct sockaddr*)&sin, sizeof(sin))<0)
    {
        perror("sendto error");
        return -1;
    }
 
    //循环接收发送数据包
    int recv_len;
    unsigned short num = 0;
    socklen_t addrlen = sizeof(sin);
    while(1)
    {
        bzero(buf, 516);
        recv_len = recvfrom(cfd, buf, sizeof(buf), 0, (struct sockaddr*)&sin, &addrlen);
        if(recv_len < 0)
        {
            perror("recvfrom error");
            return -1;
        }
 
        //操作码的范围是1-5,因为是网络字节序
        if(4 == buf[1])
        {
            //判断当前数据包的编号是否等于应答包的编号
            if(num == ntohs(*(unsigned short*)(buf+2)))
            {
                //修改操作码为数据包
                buf[1] = 3;
                //填充块编号
                num++;
                *(unsigned short*)(buf+2) = htons(num);
                //读取数据
                //发送数据
                int res = read(fd, buf+4, 516-4);
                if(res < 0)
                {
                    perror("read error");
                    return -1;
                }
                else if(0 == res)
                {
                    printf("-----文件上传完毕-----\n");
                    break;
                }
 
                //发送数据包
                //发送的数据包大小为,读取到的字节数(res)+操作码(2byte)+快编号(2byte)
                if(sendto(cfd, buf, res+4, 0, (struct sockaddr*)&sin, sizeof(sin)) <0)
                {
                    perror("sendto error");
                    return -1;
                }
            }else
            {
                printf("-----文件上传失败,请检查网络环境-----\n");
                break;
            }
        }else if(5 == buf[1])
        {
            printf("-----ERROR:%s-----\n", buf+4);
            break;
        }
 
    }
    //5、关闭套接字
    close(cfd);
}
 
 
int main(int argc, const char *argv[])
{
    int choice = 0;
    while (1)
    {
        menu();
        printf("请输入操作:");
        scanf("%d",&choice);
        switch (choice)
        {
        case 1: rev();
                break;
        case 2: snd();
                break;
        case 3: exit(0);
                break;
        default:printf("输入错误,请重新输入\n");
                break;
        }
        getchar();
        getchar();
        system("clear");
    }
    return 0;
}