利用TCP/IP网络编程实现文件传输

//发送端：
#include <stdio.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <pthread.h>

//#define PAHT "./txt.txt"

void *recv_pthread(void *arg);

int main(int argc,const char **argv)
{
    if(argc < 2)
    {
        printf("please input file path\n");
        return 0;
    }
    int fd = socket(AF_INET,SOCK_STREAM,0);
    if(fd == -1)
    {
        perror("socket failed!\n");
        return -1;
    }
    struct sockaddr_in server_addr;
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = inet_addr("192.168.1.105");//ip地址
    server_addr.sin_port = htons(9000);//端口号
    int flag = connect(fd,(struct sockaddr *)&server_addr,sizeof(server_addr));
    if(flag == -1)
    {
        perror("connect failed!\n");
        return -1;
    }

    pthread_t id = 0;
    int ret = pthread_create(&id,NULL,recv_pthread,&fd);//创建子线程
    char *data = calloc(1,1024);

    FILE *fp = fopen(argv[1],"r");//打开文件
    if(fp == NULL)
    {
        perror("file open failed!\n");
        return -1;
    }
    //发送一个文件名
    char *buf = calloc(1,1024);
    while(1)
    {
        bzero(buf,1024);
        long flag1 = ftell(fp); 
        size_t items = fread(buf,1024,1,fp);
        if(items <= 0)
        {
            long flag2 = ftell(fp);
            write(fd,buf,flag2-flag1);
            if(ferror(fp))
            {
                printf("fread error!\n");
            }
            else if(feof(fp))
            {
                printf("fread finshed!\n");
            }
            break;
        }
        else 
        {
            int size_len = write(fd,buf,1024);
            if(size_len < 0)
            {
                perror("write failed!\n");
                break;
            }
        }
        //printrf("%d\n",);
    }
    fclose(fp);
    return 0;
}

void *recv_pthread(void *arg)
{
    int sockfd = *((int*)arg);
    char *buf = calloc(1,1024);
    int i = 0;
    while(1)
    {   
        //bzero(buf,1024);
        int len = read(sockfd,&i,sizeof(int));
        if(len <= 0)
        {
            printf("离线\n");
            break;
        }
        printf("read:%d\n",i);
    }
}

//接收端：
#include <stdio.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <pthread.h>
int network_init(unsigned short port);

int main(int argc,const char **argv)
{
    if(argc < 2)
    {
        printf("pls input arg\n");
        return 0;
    }
    int sockfd =  network_init(9000);
    struct sockaddr_in sendaddr = {};
    int len = sizeof(sendaddr);
    int connfd = accept(sockfd,(struct sockaddr*)&sendaddr,&len);
    if(connfd == -1)
    {
        perror("accept failed!\n");
        return -1;
    }
    FILE* fp = fopen(argv[1],"a+");
    //接收到文件名
    int size_len = 0;
    char *buf = calloc(1,1024);
    while(1)
    {
       bzero(buf,1024); 
       size_len = read(connfd,buf,1024);
       if(size_len <= 0)
       {
           printf("结束\n");
           break;
       }
        fwrite(buf,size_len,1,fp);
        printf("%d\n",size_len);
    }
    return 0;
}

int network_init(unsigned short port)
{
    int fd = socket(AF_INET,SOCK_STREAM,0);
    if(fd == -1)
    {
        perror("socket failed!\n");
        return 0;
    }
    struct sockaddr_in myaddr;
    memset(&myaddr,'\0',sizeof(myaddr));
    myaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    myaddr.sin_family = AF_INET;
    myaddr.sin_port = htons(port);

    if(bind(fd,(struct sockaddr*)&myaddr,sizeof(myaddr)) == -1)
    {
        perror("bind failed!\n");
        return -1;
    }
    int flag = listen(fd,5);
    if(flag == -1)
    {
        perror("listen failed!\n");
        return -1;
    }
    return fd; 
}