Linux------小网盘（1）

　　

一：要求

　　利用Linux Socket进行文件传输，本次只支持client端向sever端上传文件

二：实现提示：

　　client.c

　　client的参数有两个，分别是服务器主机名和端口；

　　在while循环中输入你想要传输的文件名，然后建立socket，通过服务器主机名和端口连接服务器；

　　打开文件，在while循环中读取文件，发送文件

　　sever.c

　　在while循环中accept客户端的连接，接受数据并写入文件中

　 （也可以选择select（）accept客户端的连接，在while循环中接受数据并写入文件中）

三：程序

　　sever.c (服务器端)

　　

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


#define  PORT      5168
#define  MSGLEN    1024

int main(int argc, char **argv)
{
    int severFd, clientFd;
    int fp,flags;
    socklen_t addrlen;
    struct sockaddr_in severAddr, clientAddr;
    char recvBuff[MSGLEN];
    char filename[100];
    int recv_len;

    if((severFd = socket(AF_INET, SOCK_STREAM, 0)) == -1){
        perror("sockst() error");
        exit(-1);
    }

    severAddr.sin_family = AF_INET;
    severAddr.sin_port   = htons(PORT);
    severAddr.sin_addr.s_addr = htons(INADDR_ANY);
    bzero(&severAddr.sin_zero, 8);

    if(bind(severFd, (struct sockaddr*)&severAddr, sizeof(struct sockaddr)) == -1) {
        perror("bind() error");
        exit(-1);
    }

    if(listen(severFd, 1) == -1) {
        perror("listen() error");
        exit(-1);
    }
    
    addrlen =sizeof(struct sockaddr);
    while(1) {
        flags = 0;
        if((clientFd = accept(severFd, (struct sockaddr*)&clientAddr, &addrlen)) == -1){
            perror("accept() error");
            exit(-1);
        }
        printf("recv file fome ip:%s  port: %d
", inet_ntoa(clientAddr.sin_addr), ntohs(clientAddr.sin_port));
        if(recv(clientFd, filename, strlen(filename), 0) < 0){
            perror("recv filename error");
            break; //less
        }
        
        fp = open(filename, O_RDWR | O_CREAT, 777);
        while((recv_len = recv(clientFd, recvBuff, MSGLEN, 0)) > 0) {
            flags++;
            if(flags == 1) {
                printf("recv file start");
            }
            else {
                printf(".");
            }
            if(write(fp, recvBuff, recv_len)) {
                bzero(&recvBuff, MSGLEN);    
            }else {
                perror("write() error");
                break;
            }
        } 
        if(flags == 0) 
             perror("recv() error");
        if(flags > 0){
               printf("
recv success
");
             close(clientFd);
        }
    }

    close(severFd);
    return 0;
}

　　client.c（客户端）

　　

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


#define MSGLEN 1024

struct ADDR {
    int port;
    char name[32];
};

void file_send(struct ADDR addr, char *filename)
{
    int sockfd;
    FILE* fp;
    struct sockaddr_in sevrAddr;
    struct hostent *host;
    char readBuff[MSGLEN];
    int len;

    host = gethostbyname(addr.name);

    if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1){
        perror("socket() error");
        exit(-1);
    }else
        printf("socket ok
");

    sevrAddr.sin_family = AF_INET;
    sevrAddr.sin_port   = htons(addr.port);
    sevrAddr.sin_addr   = *((struct in_addr*)host->h_addr);
    bzero(&sevrAddr.sin_zero, 8);
//    printf("ok2
");
    if(connect(sockfd, (struct sockaddr*)&sevrAddr, sizeof(struct sockaddr)) == -1){
        perror("connect() error");
        exit(-1);
    }else
        printf("connect ok
");

//    fp = open(filename, O_RDONLY);
    if(send(sockfd, filename, sizeof(filename), 0) < 0) {
        perror("send filename error");
        exit(-1);
    }
    
    fp = fopen(filename, "rb");
    if(fp != NULL){
        printf("send file");
        while(1) {
            if((len = fread(readBuff, 1, MSGLEN, fp)) > 0) {
                if(send(sockfd, readBuff, len, 0) < 0){
                    perror("send() error");
                    exit(-1);
                } else {
                    printf(".");
                    bzero(&readBuff, MSGLEN);    
                }                
            } else if(len == 0){  //等于0表示文件已到末尾
    //            send(sockfd, readBuff, strlen(readBuff), 0);
                printf("
file send success
");
                break;
            } else {
                perror("read() error");
                exit(-1);
            }
        }
    }else {
        printf("open file failed
");
        exit(-1);
    }
    fclose(fp);
    close(sockfd);
}


int main(int argc, char **argv) 
{
    struct ADDR useraddr;
    char filename[100]; 

    if(argc != 3) {
        printf("Usage: ./client [hostname] [port]");
        exit(-1);
    }

    strcpy(useraddr.name, argv[1]);
    useraddr.port = atoi(argv[2]);

    while(1) {
        printf("please input filename of you want send
");
        fgets(filename, 20, stdin);
        filename[strlen(filename)-1] = 0;
//        printf("ok1
");
        file_send(useraddr, filename);
    }
    return 0;
}

四：现象

　　略（已实验，运行正确）