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[];
    int recv_len;

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

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

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

    if(listen(severFd, ) == -) {
        perror("listen() error");
        exit(-);
    }

    addrlen =sizeof(struct sockaddr);
    while() {
        flags = ;
        if((clientFd = accept(severFd, (struct sockaddr*)&clientAddr, &addrlen)) == -){
            perror("accept() error");
            exit(-);
        }
        printf("recv file fome ip:%s  port: %d\n", inet_ntoa(clientAddr.sin_addr), ntohs(clientAddr.sin_port));
        if(recv(clientFd, filename, strlen(filename), ) < ){
            perror("recv filename error");
            break; //less
        }

        fp = open(filename, O_RDWR | O_CREAT, );
        while((recv_len = recv(clientFd, recvBuff, MSGLEN, )) > ) {
            flags++;
            if(flags == ) {
                printf("recv file start");
            }
            else {
                printf(".");
            }
            if(write(fp, recvBuff, recv_len)) {
                bzero(&recvBuff, MSGLEN);
            }else {
                perror("write() error");
                break;
            }
        }
        if(flags == )
             perror("recv() error");
        if(flags > ){
               printf("\nrecv success\n");
             close(clientFd);
        }
    }

    close(severFd);
    return ;
}

　　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[];
};

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, )) == -){
        perror("socket() error");
        exit(-);
    }else
        printf("socket ok\n");

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

//    fp = open(filename, O_RDONLY);
    if(send(sockfd, filename, sizeof(filename), ) < ) {
        perror("send filename error");
        exit(-);
    }

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

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

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

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

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

四：现象

　　略（已实验，运行正确）