linux auto_ftp自动上传更新的文件

/prj/fup 192.168.0.1 linshao woaienen /woaienen/ /woaienen/file_list

#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <dirent.h>
#include <string.h>
#include <time.h>
#include <stdlib.h>
#define FILE_NAME_MAX 256
#define FLAGS O_WRONLY | O_CREAT | O_TRUNC
#define MODE S_IRWXU | S_IRGRP | S_IXGRP | S_IROTH | S_IXOTH

typedef struct _en_file {
    char d_name[FILE_NAME_MAX + 1];
    time_t ctime;
} en_file;

typedef struct _file_list {
    en_file file;
    struct _file_list * next;
} file_list;

void get_file_list(const char * pathname, file_list * f)
{
    DIR * dir = opendir(pathname);
    if (dir == NULL)
    {
        printf("No such directory ... \n");
        return;
    }
    
    struct dirent * dt = NULL;
    file_list * list = f;
    file_list * fl = NULL;
    
    while (dt = readdir(dir))
    {
        if (strcmp(dt->d_name, ".") == 0 || 
            strcmp(dt->d_name, "..") == 0 || 
            strcmp(dt->d_name, "file_list") == 0 || 
            strcmp(dt->d_name, "ftp_sh") == 0 || 
            strcmp(dt->d_name, "fl.log") == 0)
            continue;
            
        fl = (file_list *)malloc(sizeof(file_list));
        fl->next = NULL;
        strcpy(fl->file.d_name, pathname);
        strcat(fl->file.d_name, dt->d_name);
        
        struct stat sbuf;
        stat(fl->file.d_name, &sbuf);
        fl->file.ctime = sbuf.st_ctime;
        list->next = fl;
        list = list->next;
    }
    closedir(dir);
    return;
}

void read_file_list(const char * pathname, file_list * f)
{
    int len, i, j, is_num = 0;
    int fd;
    if ((fd = open(pathname, O_RDONLY)) == -1)
    {
        printf("error, open file failed ... \n");
        return;
    }
    
    file_list * list = f;
    file_list * fl = NULL;
    char buf[1024];
    while (1)
    {
        len = read(fd, buf, 1024);
        if (len <= 0)
            break;
        for (i = 0; i < len; i++)
        {
            if (buf[i] == ':')
            {
                is_num = 1;
                continue;
            }
            
            if (buf[i] == '\n')
            {
                is_num = 0;
                fl->file.d_name[j] = '\0';
                list->next = fl;
                list = list->next;
                fl = NULL;
                continue;
            }
            
            if (!is_num) 
            {
                if (fl == NULL) 
                {
                    fl = (file_list *)malloc(sizeof(file_list));
                    fl->next = NULL;
                    fl->file.ctime = 0;
                    j = 0;
                }
                fl->file.d_name[j++] = buf[i];
            }
            else
            {
                if (buf[i] >= '0' && buf[i] <= '9') 
                {
                    fl->file.ctime = fl->file.ctime * 10 + buf[i] - '0';
                }
            }
        }
        if (len < 1024)
            break;
    }
    close(fd);
    return;
}

void write_file_list(const char * pathname, file_list * f)
{
    int fd;
    if ((fd = open(pathname, FLAGS, MODE)) == -1)
    {
        printf("error, open file failed ... \n");
        return;
    }
    
    file_list * list = f->next;
    char num[13];
    while (list)
    {
        write(fd, list->file.d_name, strlen(list->file.d_name));
        sprintf(num, ":%10d\n", list->file.ctime);
        write(fd, num, 12);
        list = list->next;
    }
    close(fd);
    return;
}

void printf_file_list(file_list * f)
{
    int total = 0;
    file_list * list = f->next;
    printf("The file list is : \n\n");
    printf("%-60s  %16s\n", "  file name", "time");
    printf("================================================================================\n");
    while (list)
    {
        printf("%-60s  %18d\n", list->file.d_name, list->file.ctime);
        list = list->next;
        total++;
    }
    printf("================================================================================\n");
    printf("  toatl: %d\n", total);
    return;
}

int get_send_file_list(file_list * r, file_list * w, file_list * f)
{
    file_list * list, * re, * wr, * fl;
    list = f;
    wr = w->next;
    int state = 0;
    int n = 0;
    while (wr)
    {
        re = r->next;
        while (re)
        {
            if (strcmp(wr->file.d_name, re->file.d_name) == 0)
            {
                state = 1;
                if (wr->file.ctime > re->file.ctime)
                {
                    fl = (file_list *)malloc(sizeof(file_list));
                    fl->file = wr->file;
                    fl->next = NULL;
                    list->next = fl;
                    list = list->next;
                    n++;
                }
                break;
            }
            re = re->next;
        }
        if (!state)
        {
            fl = (file_list *)malloc(sizeof(file_list));
            fl->file = wr->file;
            fl->next = NULL;
            list->next = fl;
            list = list->next;
            n++;
        }
        state = 0;
        wr = wr->next;
    }
    return n;
}

void get_win_file_name(const char * p, char * q)
{
    int i = 0, j = 0, m = 0, n = 0;
    while (p[i])
        if (p[i++] == '/')
            m = i;
    i = 0;
    while (p[m])
        q[i++] = p[m++];
    q[i] = '\0';
    return;
}

void send_file(const char * ip, const char * user, 
    const char * password, file_list * f, int num)
{
    int i;
    file_list * list = f->next;
    FILE * fp, * log;
    printf("\n\nSend %d files ...\n", num);
    if ((fp=popen("ftp -n","w")) == NULL) 
    { 
        puts("Run ftp fail !\n"); 
        exit(1); 
    }
    log = fopen("/var/log/ftp.update.log", "aw+");
    char name[FILE_NAME_MAX + 1];
    char str[255];
    fprintf(fp,"open %s\n", ip); 
    fprintf(fp,"user %s %s\n", user, password);
    while (list)
    {
        get_win_file_name(list->file.d_name, name);
        fprintf(fp,"put %s %s\n", list->file.d_name, name);
        printf("Upload the %s ...\n", list->file.d_name);
        fprintf(log, "Upload the %s ...\n", list->file.d_name);
        list = list->next;
    }
    while ((fgets(str, 255, fp)) != NULL) 
        fputs(str, log); 
    fprintf(fp,"bye\n"); 
    pclose(fp);
    fclose(log);
    printf("Ok! That is right ... \n\n");
    return;
}

int main(int argc, char * argv[])
{
    file_list re, wr, send;
    if (argc != 6){ 
        printf("%s\n%s\n", 
        "Usage:", 
        "fup <ip> <user> <password> <dir> <filelist>"); 
        exit(0); 
    }
    re.next = NULL;
    wr.next = NULL;
    send.next = NULL;
    int num = 0;
    read_file_list(argv[5], &re);
    get_file_list(argv[4], &wr);
    num = get_send_file_list(&re, &wr, &send);
    printf_file_list(&send);
    send_file(argv[1], argv[2], argv[3], &send, num);
    write_file_list(argv[5], &wr);
    return 0;
}