Linux下socket传文件方法更新---不用加延迟等待

server:

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

#define PORT 6679

int main()
{
        int socket_fd;
        int new_fd;
        //建立套接字
        socket_fd=socket(AF_INET,SOCK_STREAM,0);// int socket(int domain, int type, int protocol);
        printf("socket_fd is %d\n",socket_fd);

        //初始化设置地址和端口号
    struct sockaddr_in server_addr;

        memset(&server_addr,0,sizeof(server_addr));
        server_addr.sin_family=AF_INET;
        server_addr.sin_port=htons(PORT);
        server_addr.sin_addr.s_addr=htonl(INADDR_ANY);
    memset(&(server_addr.sin_zero),0,8);
        //设置端口可重用
        int contain;
        setsockopt(socket_fd,SOL_SOCKET, SO_REUSEADDR, &contain, sizeof(int));

        //绑定fd和本机地址
        int res=bind(socket_fd,(struct sockaddr *)&server_addr,sizeof(server_addr));
        printf("bind res is %d\n",res);
        if(res==-1)
                return -1;

        if(listen(socket_fd,20)==-1)
        {
                printf("Failed to listen!\n");
                return -1;
        }
        printf("after listen\n");

        while(1)
        {
            struct sockaddr_in client_addr;
        int size=sizeof(client_addr);
        new_fd=accept(socket_fd, (struct sockaddr *)&client_addr, &size);
                if(new_fd==-1)
                {
                        printf("wait for client to connect!\n");
                        sleep(1);
                        continue;
                }
                printf("A connect and the new fd is %d\n",new_fd);
                //while(1)
                {
                        FILE *in;
                        char input[100];
                        res=read(new_fd,input,100);
                        input[res]=0;
                        char path[100];
                        sprintf(path,"/work/myproject/study/photo/get/%s",input);
                        printf("the path is %s\n",path);
                        in=fopen(path,"w");
                        printf("after path\n");
                        char photo[200];
                        while(1)
                        {
                                res=read(new_fd,photo,200);
                                if(res==0)
                                 return 0;
                fwrite(photo,sizeof(char),res,in);
                        }
                }
        }
    close(socket_fd);
    close(new_fd);

}

client:

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

#define SERVER_IP "127.0.0.1"
#define PORT    6679

int main()
{
        int socket_fd;
    socket_fd=socket(AF_INET,SOCK_STREAM,0);

    struct sockaddr_in server_addr;
        memset(&server_addr,0,sizeof(server_addr));
        server_addr.sin_family=AF_INET;
        server_addr.sin_port=htons(PORT);
        server_addr.sin_addr.s_addr=inet_addr(SERVER_IP);
        memset(&(server_addr.sin_zero),0,8);

        int res =connect(socket_fd,(struct sockaddr *)&server_addr,sizeof(server_addr));
        printf("connect res is %d\n",res);

        char path[100];
        printf("please give the path of the photo\n");
        char input[100];
        gets(input);
        sprintf(path,"/work/myproject/study/photo/%s",input);
    write(socket_fd,input,strlen(input));

        FILE *out=fopen(path,"r");
        sprintf(path,"/work/myproject/study/photo/copy_%s",input);
        FILE *new=fopen(path,"w");
        usleep(1000);
    int nread;
    char buff[100];
    while((nread=fread(buff,sizeof(char),100,out))>0)
    {
        fwrite(buff,sizeof(char),nread,new);
        write(socket_fd,buff,nread);
    }
        close(socket_fd);
}

Makefile:

all: server client
server: server.c
    gcc server.c -o server
client: client.c
    gcc client.c -o client
clean:
    rm -rf server client