1. 程式人生 > >基於TCP伺服器檔案上傳與下載

基於TCP伺服器檔案上傳與下載

**

## service.c

**

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>     
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include <sys/types.h>
#include <dirent.h>

#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#define  N  128


int process_list(int newsockfd,char *buf)
{

    DIR * dirp;
    int ret;
    struct dirent * direntp;
    dirp = opendir("./");
    if(dirp == NULL)
    {
        perror("opendir");
        exit(-1);
    }
    while( (direntp = readdir(dirp)) != NULL)
    {
        if(strncmp(direntp->d_name,".",1) == 0 ) continue; 
        strcpy(buf,direntp->d_name);
        ret = write(newsockfd,buf,N);
        if(ret < 0)
        {
            perror("write");
            exit(-1);
        }
    }
    close(newsockfd);


    return 0;
}
int process_get(int newsockfd,char *buf)
{

    int ret,fd,nbytes;

    fd = open(buf+4,O_RDWR);
    if(fd < 0)
    {
        perror("open");
        exit(-1);
    }
    while( (nbytes= read(fd,buf,N)) > 0)
    {
        ret = write(newsockfd,buf,nbytes);
        if(ret < 0)
        {
            perror("write");
            exit(-1);
        }
    }
    close(newsockfd);


    return 0;
}
int process_put(int newsockfd,char *buf)
{

    int ret,fd,nbytes;

    fd = open(buf+4,O_RDWR|O_CREAT|O_TRUNC,0664);
    if(fd < 0)
    {
        perror("open");
        exit(-1);
    }
    while( (nbytes= read(newsockfd,buf,N)) > 0)
    {
        ret = write(fd,buf,nbytes);
        if(ret < 0)
        {
            perror("write");
            exit(-1);
        }
    }
    close(newsockfd);
    close(fd);


    return 0;
}
int main(int argc, const char *argv[])
{
    char buf[N] ={0};
    int sockfd,ret,newsockfd;
    struct sockaddr_in myaddr,client;
    if(argc != 3)
    {
        printf("執行程式時帶入ip 和 port\n");
        exit(-1);
    }
    socklen_t addrlen = sizeof(client);
    sockfd = socket(AF_INET,SOCK_STREAM,0);
    if(sockfd < 0)
    {
        perror("socket");
        exit(-1);
    }
    printf("sockfd=%d\n",sockfd);
    memset(&myaddr,0,sizeof(myaddr));
    myaddr.sin_family = AF_INET ;
    myaddr.sin_port   = htons(atoi(argv[2]));
    myaddr.sin_addr.s_addr   = inet_addr(argv[1]); 
    ret = bind(sockfd,(struct sockaddr *)&myaddr,sizeof(myaddr)) ; // 給socket 一個固定的ip 和埠 
    if(ret < 0)
    {
        perror("bind");
        exit(-1);
    }
    ret = listen(sockfd,10);
    if(ret < 0)
    {
        perror("listen");
        exit(-1);
    }
    while(1)
    {
        newsockfd = accept(sockfd,(struct sockaddr *)&client,&addrlen) ; 
        if(newsockfd < 0)
        {
            printf("accept");
            exit(-1);
        }
        printf("newsockfd = %d\n",newsockfd);
        ret = read(newsockfd,buf,N);
        if(ret  < 0)
        {
            printf("read");
            exit(-1);
        }
        printf("buf=%s\n",buf);
        if( strncmp(buf,"list",4) == 0 ) 
        {
            process_list(newsockfd,buf);
        }
        else if( strncmp(buf,"get",3) == 0 ) 
        {
            process_get(newsockfd,buf);
        }
        else if( strncmp(buf,"put",3) == 0 ) 
        {
            process_put(newsockfd,buf);
        }
    }


    
    return 0;
}


## client.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>       
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#define  N  128 


int process_list(struct sockaddr_in *addr,char * buf)
{
    int ret,sockfd,nbytes;
    sockfd = socket(AF_INET,SOCK_STREAM,0);
    if(sockfd < 0)
    {
        perror("socket");
        exit(-1);
    }
    ret = connect(sockfd,(struct sockaddr *)addr,sizeof(*addr)) ;
    if(ret < 0)
    {
        perror("connect");
        exit(-1);
    }
    ret = write(sockfd,buf,N);
    if(ret < 0)
    {
        perror("write");
        exit(-1);
    }
    while( (nbytes= read(sockfd,buf,N)) > 0)
    {
        printf("%s\n",buf);
    }
    close(sockfd);

    return 0;
}
int process_get(struct sockaddr_in *addr,char * buf)
{
    int ret,sockfd,nbytes,fd;
    sockfd = socket(AF_INET,SOCK_STREAM,0);
    if(sockfd < 0)
    {
        perror("socket");
        exit(-1);
    }
    ret = connect(sockfd,(struct sockaddr *)addr,sizeof(*addr)) ;
    if(ret < 0)
    {
        perror("connect");
        exit(-1);
    }
    ret = write(sockfd,buf,N);
    if(ret < 0)
    {
        perror("write");
        exit(-1);
    }
    fd = open(buf+4,O_RDWR|O_CREAT|O_TRUNC,0664);
    if(fd < 0)
    {
        perror("open");
        exit(-1);
    }
    while( (nbytes= read(sockfd,buf,N)) > 0)
    {
        ret = write(fd,buf,nbytes);
        if(ret < 0)
        {
            perror("write");
            exit(-1);
        }
    }
    close(sockfd);
    return 0;
}
int process_put(struct sockaddr_in *addr,char * buf)
{
    int ret,sockfd,nbytes,fd;
    sockfd = socket(AF_INET,SOCK_STREAM,0);
    if(sockfd < 0)
    {
        perror("socket");
        exit(-1);
    }
    ret = connect(sockfd,(struct sockaddr *)addr,sizeof(*addr)) ;
    if(ret < 0)
    {
        perror("connect");
        exit(-1);
    }
    ret = write(sockfd,buf,N);
    if(ret < 0)
    {
        perror("write");
        exit(-1);
    }
    fd = open(buf+4,O_RDWR);
    if(fd < 0)
    {
        perror("open");
        exit(-1);
    }
    while( (nbytes= read(fd,buf,N)) > 0)
    {
        ret = write(sockfd,buf,nbytes);
        if(ret < 0)
        {
            perror("write");
            exit(-1);
        }
    }
    close(sockfd);
    close(fd);
    return 0;
}
int main(int argc, const char *argv[])
{

    char buf[N] = {0};
    int sockfd,ret;
    struct sockaddr_in server,client;
    if(argc != 3)
    {
        printf("執行程式時帶入ip 和 port\n");
        exit(-1);
    }
    memset(&server,0,sizeof(server));
    server.sin_family = AF_INET ;
    server.sin_port   = htons( atoi(argv[2]) );
    server.sin_addr.s_addr   = inet_addr(argv[1]); 
    while(1)
    {
        printf(">:");
        fgets(buf,N,stdin);
        buf[strlen(buf)-1] = '\0';
        if( strncmp(buf,"list",4) == 0 ) 
        {
            process_list(&server,buf);
        }
        else if( strncmp(buf,"get",3) == 0 ) 
        {
            process_get(&server,buf);
        }
        else if( strncmp(buf,"put",3) == 0 ) 
        {
            process_put(&server,buf);
        }
        else if( strncmp(buf,"quit",4) == 0 ) 
        {
            break;
        }
        else if( strncmp(buf,"exit",4) == 0 ) 
        {
            break;
        }
        else 
        {
            printf("命令錯誤,請重新輸入\n");
        }
        
    }


#if 0 
    printf(">:");
    fgets(buf,N,stdin);
    ret  = write(sockfd,buf,N) ;
    if(ret < 0)
    {
        perror("write");
        exit(-1);
    }
    ret = read(sockfd,buf,N);
    if(ret  < 0)
    {
        printf("read");
        exit(-1);
    }
    printf("%s\n",buf);
    close(sockfd);


#endif
    
    return 0;
}

基於tcp的檔案上傳下載伺服器與客戶端的開發;
該專案具體如下:
1、基於tcp伺服器部分
分成三個部分:
第一個是list如果客戶端傳送一個"list",伺服器會將當前的目錄開啟,並將當前的檔名傳送給客戶端介面,供使用者參考。
第二個是get 如果客戶端送一個"get+filename" , 伺服器會開啟當前的filename讀此檔案的所有內容寫入給客戶端。
第三個是put 如果客戶端送一個"put+filename",伺服器會開啟filename如果不存在就建立一個新的檔案否則則截斷清空等待讀客戶端傳送的資料並寫入。
2、客戶端部分
相應的分成三部分
第一部分是list 客戶端選擇的是list 首先是將選擇的命令傳送給伺服器,之後就等待獲取當前伺服器上有所有有效檔案的檔名。
第二部分是get 客戶端選擇是"get+filename" 將在客戶端方,開啟或是建立一個同伺服器相同的filename檔案,並讀取伺服器傳送的資料寫入當前filename檔案。
第三部分是put 客戶端選擇是"put+filename" 開啟要上傳的檔案,將資料讀出傳送到伺服器。