基于libzip的简易压缩(zip)/解压缩(unzip)程序

/*    zip.c Copyright (c) 2015 zeerd.com.
 *
 *  http://opensource.org/licenses/bsd-license.php
 *
 *    Compile:
 *    ./configure --prefix=/usr
 *    make && sudo make isntall
 *    gcc zip.c -lzip -o zip
 *
 *    Got libzip from http://www.nih.at/libzip
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <dirent.h>
#include <errno.h>
#include <sys/stat.h>

#include <zip.h>

#define DBG_SEQ 0
#define LOGI printf
#define LOGD printf
#define LOGE printf

static int root_len = 0;
static char path[ PATH_MAX ];
static struct zip *z = NULL;

void search_dir ( const char * name )
{
    struct stat _stbuf;
    
    char pathBak[ PATH_MAX ];
    strcpy(pathBak, path);
    strncat( path, name, sizeof( path ));

    if (DBG_SEQ) LOGI( "searching %s\n", path );

    if( stat( path, &_stbuf ) == 0 ) {

        if( S_ISDIR( _stbuf.st_mode )) {
            DIR * _dir;
            struct dirent * _file;
            
            _dir    =    opendir( path );

            if( _dir ) {
                
                if (DBG_SEQ) LOGD("find folder %s\n", path);

                //Caution : no need to do this unless there is a empty folder
                //zip_dir_add(z, name, ZIP_FL_ENC_GUESS); 

                strncat( path, "/", sizeof( path ));

                while(( _file = readdir( _dir )) != NULL ) {
                    if( strncmp( _file->d_name, ".", 1 ) != 0 ) {
                        search_dir( _file->d_name);
                    }
                }

                closedir( _dir );
            }
            else {
                LOGE( "open dir failed: %s\n", path );
            }
        }
        else {
            if (DBG_SEQ) LOGD("find file %s\n", path);

            struct zip_source *s = zip_source_file(z, path, 0, -1);
            if(s != NULL) {
                // the file name give to zip_file_add() would include the path
                zip_file_add(z, &path[root_len+1], s,
                    ZIP_FL_OVERWRITE|ZIP_FL_ENC_GUESS);

                //would be used and freed by zip_close(),
                //so don't free the zip_source here.
                //zip_source_free(s); 
            } else {
                LOGE( "zip_source_file failed for %s with the reason %s\n",
                    path, zip_strerror(z) );
            }
        }
    }
    else {
        LOGE( "stat failed\n" );
    }

    /* remove parsed name */
    strcpy(path, pathBak);
}


int main(int argc, char *argv[])
{
    int err = 0;
    char strerr[1024];

    if (argc < 3) {
        printf(
            "\nArchive a folder to a zip file.\n"
            "Usage:\n\t%s <path> <zip>\n"
            "Example:\n\t%s /home/foo/foo /home/foo/foo.zip\n\n",
            argv[0], argv[0]);
        return -1;
    }

    root_len = strlen(argv[1]);
    z = zip_open(argv[2], ZIP_CREATE|ZIP_EXCL, &err);

    if (z != NULL) {
        search_dir(argv[1]);
        err = zip_close(z);
    } 

    if (err != 0) {
        zip_error_to_str(strerr, 1024, err, errno);
        LOGE("operated zip fail for %s\n", strerr);
    }

    return 0;
}

/*    unzip.c Copyright (c) 2015 zeerd.com.
 *
 *  http://opensource.org/licenses/bsd-license.php
 *
 *    Compile:
 *    ./configure --prefix=/usr
 *    make && sudo make isntall
 *    gcc unzip.c -lzip -o unzip
 *
 *    Got libzip from http://www.nih.at/libzip
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <dirent.h>
#include <errno.h>
#include <sys/stat.h>

#include <zip.h>

#define DBG_SEQ 0
#define LOGI printf
#define LOGD printf
#define LOGE printf

int pmkdir(const char *path) 
{ 
    char name[ PATH_MAX ]; 
    strcpy(name, path); 
    int i, len = strlen(name); 

    if (name[len-1]!='/') {
        strcat(name, "/");
    } 

    len = strlen(name); 

    for (i=1 ; i<len ; i++) { 
        if (name[i]=='/') { 
            name[i] = 0; 
            if ( access(name, NULL) !=0 ) { 
                if (mkdir(name, 0755) == -1) {  
                    LOGE("mkdir error");  
                    return -1;  
                } 
            }
            name[i] = '/'; 
        } 
    } 

    return 0; 
} 

int main(int argc, char *argv[])
{
    int err = 0;
    char strerr[1024];
    struct zip *z = NULL;

    if (argc < 2) {
        printf(
            "\nUn-Archive a zip file to the current path.\n"
            "Usage:\n\t%s <zip>\n"
            "Example:\n\t%s foo.zip\n\n",
            argv[0], argv[0]);
        return -1;
    }

    z = zip_open(argv[1], ZIP_CREATE, &err);
    if (z != NULL) {

        zip_int64_t i, c = zip_get_num_entries(z, ZIP_FL_UNCHANGED);
        for (i=0; i<c ; i++) {
            const char * name = zip_get_name(z, i, ZIP_FL_ENC_GUESS);
            LOGI("find %s\n", name);
            char *d = strdup(name);
            if (d != NULL) {
                char *p = strrchr(d, '/');
                if(p != NULL) {
                    *p = '\0';
                    pmkdir(d);
                }
                free(d);

                FILE *fp = fopen(name, "w+b");
                struct zip_file *f = zip_fopen(z, name, 0);
                if (f != NULL && fp != NULL) {
                    zip_int64_t j, n = 0;
                    char buf[8192] = "";
                    while ((n = zip_fread(f, buf, sizeof(buf))) > 0) {
                        for (j=0;j<n;j++) {
                            putc(buf[j], fp);
                        }
                    }
                    fclose(fp);
                    zip_fclose(f);
                }
            } else {
                LOGE("memory low\n");
            }
        }

        err = zip_close(z);
    } else {
        zip_error_to_str(strerr, 1024, err, errno);
        LOGE("operated zip fail for %s\n", strerr);
    }

    return 0;
}