利用socket传文件

欢迎批评

也可以看看这里的比较详细http://www.jb51.net/article/53733.htm

服务端
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#define BUFFER_SIZE 1024
#define FILE_NAME_SIZE 512 
void error(char *message);

int main(int argc,char*argv[])
{
    int serv_sock,clnt_sock,i,length,read_length,j;
    struct sockaddr_in serv_addr,clnt_addr;
    socklen_t clnt_addr_sz;
    char file_name[FILE_NAME_SIZE],message[BUFFER_SIZE];
    char buffer[BUFFER_SIZE];

    if(argc!=2)
    {
        printf("usage:%s<port>
",argv[0]);
        exit(1);
    }

    serv_sock=socket(PF_INET,SOCK_STREAM,0);
    if(serv_sock==-1)
        error("socket() error");

    memset(&serv_addr,0,sizeof(serv_addr));
    serv_addr.sin_family=AF_INET;
    serv_addr.sin_addr.s_addr=htonl(INADDR_ANY);
    serv_addr.sin_port=htons(atoi(argv[1]));

    if(bind(serv_sock,(struct sockaddr*)&serv_addr,sizeof(serv_addr))==-1)
        error("bind() error");

    if(listen(serv_sock,3)==-1)
        error("listen() error");

    clnt_addr_sz=sizeof(clnt_addr);
    for(i=1;i<=3;i++)
    //while(1)
    {
        clnt_sock=accept(serv_sock,(struct sockaddr*)&clnt_addr,&clnt_addr_sz);
        if(clnt_sock==-1)
            error("accept() error");
        else
            printf("connected client %d 
`",i);
        
        if(read(clnt_sock,buffer,BUFFER_SIZE)==-1)
            printf("read() error");
        strcpy(file_name,buffer);    
        printf("file's name %s
 ",file_name);
        FILE *fp=fopen(file_name,"r");

        if(fp==NULL)
            printf("file %s can not open 
",file_name);
        
        else 
        {
            bzero(buffer,BUFFER_SIZE);
            while((length=fread(buffer,sizeof(char),BUFFER_SIZE,fp))>0)
            {
                if(write(clnt_sock,buffer,length)==-1)
                {
                    printf("send file %s failed
",file_name);
                    break;
                }
            }
            fclose(fp);
            printf("transfer succeed
");
        }
        close(clnt_sock);
    }
    close(serv_sock);
    return 0;
}

void error(char* message)
{
    fputs(message,stderr);
    fputc('
',stderr);
    exit(1);
}

客户端 
1 #include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#define FILE_NAME_SZ 512
#define BUFFER_SIZE 1024
void error(char* message);

int main(int argc,char*argv[])
{
    int sock,temp=0;
    struct sockaddr_in clnt;
    int length,write_length;
    char buffer[BUFFER_SIZE+1];
    char filename[FILE_NAME_SZ+1];
    char message[BUFFER_SIZE];
    sock=socket(PF_INET,SOCK_STREAM,0);
    if(argc!=3)
    {
        printf("usage:%s<ip><port>
",argv[0]);
    }
    if(sock==-1)
        error("socket() error ");

    memset(&clnt,0,sizeof(clnt));
    clnt.sin_family=AF_INET;
    clnt.sin_addr.s_addr=inet_addr(argv[1]);
    clnt.sin_port=htons(atoi(argv[2]));

    if(connect(sock,(struct sockaddr*)&clnt,sizeof(clnt))==-1)
        error("connect() error");
    else
        puts("connect........");

    printf("input file name	");
    scanf("%s",filename);

    strcpy(buffer,filename);
    write(sock,buffer,BUFFER_SIZE); 
    
    FILE *fp=fopen(filename,"w");
    if(fp==NULL)
    {
        printf("open file %s error
",filename);
        exit(1);
    }
    else
    {
        bzero(buffer,BUFFER_SIZE);
        while((length=read(sock,buffer,BUFFER_SIZE))>0)
        {
            if(fwrite(buffer,sizeof(char),length,fp)<length)
            {
                printf("file %s write  error
",filename);
                    break;
            }
            //printf("receive %s succeed
",filename);
            temp=1;
        }
        if(temp==1) printf("receive %s succedd
",filename);
        else        printf("receive %s failed
",filename);
    }
    fclose(fp);
    close(sock);
    return 0;
}

void error(char* message)
{
    fputs(message,stderr);
    fputc('
',stderr);
    exit(1);
}