使用socket下载文件(C++)

刚学习socket编程，写了一个下载文件的小程序，写的不好大家不要吐槽。

客户端：

 

#include<stdio.h>
#include<stdlib.h>
#include<arpa/inet.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<string.h>
#include<unistd.h>
#define BUFF_SIZE 100
int main(int argv,char *args[]){
    int servsock;
    struct sockaddr_in servaddr;
    if(argv!=3){
        printf("Usage:%s <ip><port>",args[0]);
        exit(1);
    }
    servsock = socket(PF_INET,SOCK_STREAM,0);
    if(servsock == -1)
    {
        printf("socket() error\n");
        exit(1);
    }
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = inet_addr(args[1]);
    servaddr.sin_port = htons(atoi(args[2]));
    if(connect(servsock,(struct sockaddr*)&servaddr,sizeof(servaddr)) ==-1)
    {
        printf("connect() error\n");
        exit(1);
    }else{
        printf("connected.....\n");
    }
    printf("please input filename:");
    char filename[BUFF_SIZE];
    memset(filename,0,sizeof(char)*BUFF_SIZE);
    scanf("%s",filename);
    write(servsock,filename,strlen(filename));
    int successFlag;
    read(servsock,&successFlag,sizeof(successFlag));
    if(successFlag!=1){
        printf("input file is not existed!\n successFlag is %d\n",successFlag);
        exit(1);
    }else{
        FILE *fp = fopen(filename,"w");
        while(1){
            memset(filename,0,BUFF_SIZE);
            int revlen = read(servsock,filename,BUFF_SIZE);
            if(revlen ==0)
                break;
            fputs(filename,fp);
        }
        fclose(fp);
        printf("complete!\n");
    }
    close(servsock);
    return 0;
}

服务器：

 

 

#include<stdio.h>
#include<stdlib.h>
#include<arpa/inet.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<string.h>
#include<unistd.h>
#define BUFF_SIZE 100
int main(int argv,char *args[]){
    int servsock,clntsock;
    struct sockaddr_in servaddr,clntaddr;
    if(argv!=2){
        printf("Usage:%s <port>",args[0]);
        exit(1);
    }
    servsock = socket(PF_INET,SOCK_STREAM,0);
    memset(&servaddr,0,sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port = htons(atoi(args[1]));
    if(bind(servsock,(struct sockaddr*)&servaddr,sizeof(servaddr)) ==-1)
    {
        printf("bind() error\n");
        exit(1);
    }
    if(listen(servsock,5) ==-1)
    {
        printf("listen() error\n");
        exit(1);
    }
    socklen_t len = sizeof(clntaddr);
    clntsock = accept(servsock,(struct sockaddr*)&clntaddr,&len);
    char filename[BUFF_SIZE];
    int msglen=read(clntsock,filename,BUFF_SIZE);
    filename[msglen]=0;
    printf("receiver filename is: %s\n",filename);
    FILE *fp = fopen(filename,"r");
    int successFlag = 0;
    if(fp)
        successFlag = 1;
    write(clntsock,&successFlag,sizeof(successFlag));
    if(fp == NULL){
        printf("NULL\n");
        close(clntsock);
        close(servsock);
        return 0;
    }
    while(fgets(filename,BUFF_SIZE,fp)!=NULL){
        int ll = strlen(filename);
        printf("length:%d content:%s",ll,filename);
        write(clntsock,filename,strlen(filename));
    }
    fclose(fp);
    close(clntsock);
    close(servsock);
    return 0;
}