我的文件传输器

最新推荐文章于 2024-05-02 19:42:36 发布

m0_73003633

最新推荐文章于 2024-05-02 19:42:36 发布

阅读量80

点赞数

文章标签： c语言

本文链接：https://blog.csdn.net/m0_73003633/article/details/126688193

版权

int get(int);
int put(int);

int main(int argc, char *argv[])
{
    int sockfd = socket(AF_INET,SOCK_STREAM,0);
    if(-1 == sockfd){
        perror("socket");
        return -1;
    }
    printf("%d\n",sockfd);
struct sockaddr_in saddr;//服务器的地址结构
bzero(&saddr,sizeof(saddr));//清空结构体
saddr.sin_family = AF_INET;//指定协议族 ipv4
saddr.sin_port = htons(12580);//端口号 并转为网络字节序
saddr.sin_addr.s_addr = inet_addr("192.168.1.116");//将ip地址转换为二进制网络字节序

int ret = bind(sockfd,(struct sockaddr *)&saddr,sizeof(saddr));
if(-1 == ret){
    perror("bind");
    close(sockfd);
    return -1;
}
printf("bing成功\n");

ret = listen(sockfd,4);
if(-1 == ret){
    perror("listen");
    close(sockfd);
    return -1;
}
printf("listen成功\n");

struct sockaddr_in caddr;
bzero(&caddr,sizeof(caddr));
socklen_t cad = sizeof(caddr);

int rws = accept(sockfd,(struct sockaddr *)&caddr,&cad);
if(-1 == rws){
    perror("accept");
    close(sockfd);
    return -1;
}

char str[SIZE] = {0};
char str1[SIZE] = {"连接成功，get下载文件，put上传文件,quit退出"};

while(1){
    sleep(1);
    ret = write(rws,str1,sizeof(str1));
    if(-1 == ret){
        perror("write");
        close(rws);
        close(sockfd);
        return -1;
    }

    bzero(str,sizeof(str));
    ret = read(rws,str,sizeof(str));
    if(-1 == ret){
        perror("read");
        close(rws);
        close(sockfd);
        return -1;
    }
    if(strcmp("get\n",str) == 0){
        printf("客户端下载文件\n");
        ret = get(rws);
        if(ret == -1){
            perror("get");
            return -1;
        }

    }
    else if(strcmp("put\n",str) == 0){
        printf("客户端上传文件\n");
        ret = put(rws);
        if(ret == -1){
            perror("put");
            return -1;
        }
    }
    else if(strcmp("quit\n",str) == 0){
        printf("客户端退出\n");
        break;
    }
    else{
        char str2[SIZE] = {"无效选项"};
        write(rws,str2,sizeof(str2));
    }
}
char str3[SIZE] = {"传输成功"};
ret = write(rws,str3,sizeof(str3));
if(-1 == ret){
    perror("write");
    close(rws);
    close(sockfd);
    return -1;
}

close(rws);
close(sockfd);
return 0;
} 

int get(int rws){
    DIR * fdr = opendir(".");
    if(fdr == NULL){
        perror("get_opendir");
        return -1;
    }
    struct dirent *adir;
    char str[SIZE] = {0};
    int ret;
while( NULL != (adir = readdir(fdr))){
    bzero(str,sizeof(str));
    strcpy(str,adir->d_name);
    if(adir->d_name[0] != '.'){
        ret = write(rws,str,sizeof(str));
        if(ret == -1){
            perror("get_write");
            return -1;
        }
    }
}
bzero(str,sizeof(str));
ret = write(rws,str,sizeof(str));
if(ret == -1){
    perror("get_write");
    return -1;
}
ret = read(rws,str,sizeof(str));
if(ret == -1){
    perror("get_write");
    return -1;
}
int i = strlen(str);
str[i-1] = 0;

int fd = open(str,O_RDONLY);
if(ret == -1){
    perror("get_open");
    return -1;
}

do{
    bzero(str,sizeof(str));
    ret = read(fd,str,sizeof(str));
    if(ret == -1){
        perror("get_write_read");
        return -1;
    }

    ret = write(rws,str,ret);
    if(ret == -1){
        perror("get_write_read");
        return -1;
    }

}while(0 != ret);

bzero(str,sizeof(str));
ret = write(rws,str,sizeof(str));
if(ret == -1){
perror("get_write");
    return -1;
}
printf("下载成功\n");
return 0;
}
int put(int rws){
    char str[SIZE] = {0};
    int ret = read(rws,str,sizeof(str));
    if(ret == -1){
        perror("put_read");
        return -1;
    }
int fd = open(str,O_WRONLY | O_TRUNC | O_CREAT,0664);
if(ret == -1){
    perror("get_open");
    return -1;
}
while(0 != str[0]){
    bzero(str,sizeof(str));
    ret = read(rws,str,sizeof(str));
    if(ret == -1){
        perror("get_open_read");
        return -1;
    }   
    ret = write(fd,str,strlen(str));
    if(ret == -1){
        perror("get_open_write");
        return -1;
    }
}

printf("上传成功\n");
return 0;

}

int get(int);
int put(int);

int main(int argc, char *argv[])
{
    int sockfd = socket(AF_INET,SOCK_STREAM,0);
    if(-1 == sockfd){
        perror("socket");
        return -1;
    }
    printf("%d\n",sockfd);
struct sockaddr_in saddr;//服务器的地址结构
bzero(&saddr,sizeof(saddr));//清空结构体
saddr.sin_family = AF_INET;//指定协议族 ipv4
saddr.sin_port = htons(12580);//端口号 并转为网络字节序
saddr.sin_addr.s_addr = inet_addr("192.168.1.116");//将ip地址转换为二进制网络字节序

int ret = connect(sockfd,(struct sockaddr *)&saddr,sizeof(saddr));
if(-1 == ret){
    perror("connect");
    close(sockfd);
    return -1;
}
printf("connect成功\n");

char str[SIZE] = {0};
while(1){
    bzero(str,sizeof(str));
ret = read(sockfd,str,sizeof(str));
if(-1 == ret){
    perror("read");
    close(sockfd);
    return -1;
}
printf("%s\n",str);

    fgets(str,SIZE,stdin);

    ret = write(sockfd,str,sizeof(str));
    if(ret == -1){
        perror("write");
        close(sockfd);
        return -1;
    }
    if(strcmp("quit\n",str) == 0){
        break;
    }
    else if(strcmp("get\n",str) == 0){
        ret = get(sockfd);
        if(ret == -1){
            perror("get");
            return -1;
        }

    }
    else if(strcmp("put\n",str) == 0){
        ret = put(sockfd);
        if(ret == -1){
            perror("put");
            return -1;
        }

    }
    else{
        printf("错误选项\n");
        bzero(str,sizeof(str));
        ret = read(sockfd,str,sizeof(str));
        if(-1 == ret){
            perror("read");
            close(sockfd);
            return -1;
        }
        printf("%s",str);

    }
}

ret = read(sockfd,str,sizeof(str));
if(-1 == ret){
    perror("read");
    close(sockfd);
    return -1;
}

printf("%s\n",str);
return 0;
}

int get(int rws){
char str[SIZE] = {'1'};
int ret ;
while(0 != str[0]){
    bzero(str,sizeof(str));
    ret = read(rws,str,sizeof(str));
    if(ret < 0){
        perror("get_read");
        return -1;
    }
    printf("%s\n",str);
}

bzero(str,sizeof(str));
printf("请输入下载的文件\n");
fgets(str,sizeof(str),stdin);
ret = write(rws,str,sizeof(str));
if(ret < 0){
    perror("get_write");
    return -1;
}
int i = strlen(str);
str[i-1] = 0;

int fd = open(str,O_WRONLY | O_TRUNC | O_CREAT,0664);
if(ret == -1){
    perror("get_open");
    return -1;
}

while(0 != str[0]){
bzero(str,sizeof(str));
    ret = read(rws,str,sizeof(str));
    if(ret == -1){
        perror("get_write_read");
        return -1;
    }
    ret = write(fd,str,strlen(str));
    if(ret == -1){
        perror("get_write_read");
        return -1;
    }
}

return 0;
}

int put(int rws){
    char str[SIZE] = {0};
    printf("请输入要传输的文件名\n");
    fgets(str,sizeof(str),stdin);
    int i = strlen(str);
    str[i-1] = 0;
int ret = write(rws,str,sizeof(str));
if(ret == -1){
    perror("put_write");
    return -1;
}
int fd = open(str,O_RDONLY);
if(ret == -1){
    perror("get_open");
    return -1;
}
while(0 != ret){
    bzero(str,sizeof(str));
    ret = read(fd,str,sizeof(str));
    if(ret == -1){
        perror("get_open_read");
        return -1;
    }   
    ret = write(rws,str,ret);
    if(ret == -1){
        perror("get_open_write");
        return -1;
    }

}
bzero(str,sizeof(str));
 ret = write(rws,str,sizeof(str));
 if(ret == -1){
     perror("get_write");
     return -1;
 }
return 0;
 }