From 4c48653ccc17eb041526884a3c010e4c6f0797f6 Mon Sep 17 00:00:00 2001 From: dreamsourcelab tai Date: Thu, 30 Sep 2021 11:17:04 +0800 Subject: [PATCH] suppor make zip in memory --- DSView/pv/ZipMaker.cpp | 199 +++++++++++++++++++++++++++++++++++++++++ DSView/pv/ZipMaker.h | 60 +++++++++++++ 2 files changed, 259 insertions(+) create mode 100644 DSView/pv/ZipMaker.cpp create mode 100644 DSView/pv/ZipMaker.h diff --git a/DSView/pv/ZipMaker.cpp b/DSView/pv/ZipMaker.cpp new file mode 100644 index 00000000..cdda2d33 --- /dev/null +++ b/DSView/pv/ZipMaker.cpp @@ -0,0 +1,199 @@ + +/* + * This file is part of the DSView project. + * DSView is based on PulseView. + * + * Copyright (C) 2012 Joel Holdsworth + * Copyright (C) 2013 DreamSourceLab + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#include "ZipMaker.h" +#include +#include +#include +#include +#include +#include + + +ZipMaker::ZipMaker() : + m_docSource(NULL), + m_errobj(NULL) +{ + m_error[0] = 0; +} + +ZipMaker::~ZipMaker() +{ + Release(); +} + +bool ZipMaker::CreateNew() +{ + Release(); + + m_errobj = new zip_error_t(); + zip_error_init(m_errobj); + + if ((m_docSource = zip_source_buffer_create(NULL, 0, 0, m_errobj)) == NULL) { + sprintf(m_error, "can't create source: %s", zip_error_strerror(m_errobj)); + } + + return m_docSource != NULL; +} + +void ZipMaker::Release() +{ + if (m_errobj){ + zip_error_fini(m_errobj); //release the error obj + delete m_errobj; + m_errobj = NULL; + } + + if (m_docSource){ + zip_source_free(m_docSource); + m_docSource = NULL; + } +} + +bool ZipMaker::SaveToFile(const char *fileName){ + assert(fileName); + assert(m_docSource); + + zip_stat_t zst; + if (zip_source_stat(m_docSource, &zst) < 0) { + sprintf(m_error, "can't stat source: %s\n", zip_error_strerror(zip_source_error(m_docSource))); + return 1; + } + + long long size = zst.size; + void *data = NULL; + FILE *fp = NULL; + bool berr = false; + + if (zip_source_open(m_docSource) < 0) { + sprintf(m_error, "can't open source: %s\n", zip_error_strerror(zip_source_error(m_docSource))); + return false; + } + if (!berr && (data = malloc(size)) == NULL) { + sprintf(m_error, "malloc failed: %s\n", strerror(errno)); + berr = true; + } + + if ((zip_uint64_t)zip_source_read(m_docSource, data, size) < size) { + sprintf(m_error, "can't read data from source: %s\n", zip_error_strerror(zip_source_error(m_docSource))); + berr = true; + } + + if (!berr && (fp = fopen(fileName, "wb")) == NULL) { + sprintf(m_error, "can't open %s: %s\n", fileName, strerror(errno)); + berr = true; + } + if (!berr && fwrite(data, 1, size, fp) < size) { + sprintf(m_error, "can't write %s: %s\n", fileName, strerror(errno)); + berr = true; + fclose(fp); + } + if (!berr && fclose(fp) < 0) { + sprintf(m_error, "can't write %s: %s\n", fileName, strerror(errno)); + berr = true; + } + + zip_source_close(m_docSource); + if (data){ + free(data); + } + + return !berr; +} + +bool ZipMaker::AddFromBuffer(const char *innerFile, const char *buffer, int buferSize) +{ + assert(buffer); + assert(innerFile); + assert(m_docSource); + + zip_t *zip_archive = zip_open_from_source(m_docSource, 0, m_errobj); + if (zip_archive == NULL) + { + sprintf(m_error, "can't open zip from source: %s", zip_error_strerror(m_errobj)); + return false; + } + + zip_source_keep(m_docSource); + + zip_source_t *src = zip_source_buffer(zip_archive, buffer, buferSize, 0); + if (src == NULL){ + sprintf(m_error, "zip_source_buffer error: %s", zip_error_strerror(zip_source_error(m_docSource))); + return false; + } + + if (zip_file_add(zip_archive, innerFile, src, ZIP_FL_OVERWRITE) == -1) + { + sprintf(m_error, "zip_file_add error: %s", zip_error_strerror(zip_source_error(m_docSource))); + return false; + } + + if (zip_close(zip_archive) < 0) { + sprintf(m_error, "zip_close'%s'", zip_strerror(zip_archive)); + return false; + } + + return true; +} + +bool ZipMaker::AddFromFile(const char *localFile, const char *innerFile) +{ + assert(localFile); + assert(innerFile); + assert(m_docSource); + + zip_t *zip_archive = zip_open_from_source(m_docSource, 0, m_errobj); + if (zip_archive == NULL) + { + sprintf(m_error, "can't open zip from source: %s", zip_error_strerror(m_errobj)); + return false; + } + + zip_source_keep(m_docSource); + + zip_source_t *src = zip_source_file(zip_archive, localFile, 0, -1); + if (src == NULL){ + sprintf(m_error, "zip_source_buffer error: %s\n", zip_error_strerror(zip_source_error(m_docSource))); + return false; + } + + if (zip_add(zip_archive, innerFile, src) == -1) + { + sprintf(m_error, "zip_file_add error: %s\n", zip_error_strerror(zip_source_error(m_docSource))); + return false; + } + + if (zip_close(zip_archive) < 0) { + sprintf(m_error, "zip_close'%s'", zip_strerror(zip_archive)); + return false; + } + + return true; +} + +const char *ZipMaker::GetError() +{ + if (m_error[0]) + return m_error; + return NULL; +} diff --git a/DSView/pv/ZipMaker.h b/DSView/pv/ZipMaker.h new file mode 100644 index 00000000..621d16d7 --- /dev/null +++ b/DSView/pv/ZipMaker.h @@ -0,0 +1,60 @@ +/* + * This file is part of the DSView project. + * DSView is based on PulseView. + * + * Copyright (C) 2012 Joel Holdsworth + * Copyright (C) 2013 DreamSourceLab + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#pragma once + +struct zip_source; +struct zip_error; + +typedef struct zip_source zip_archive_source_t; +typedef struct zip_error zip_err_t; + +class ZipMaker +{ +public: + ZipMaker(); + + ~ZipMaker(); + + //create new zip archive in the memory + bool CreateNew(); + + //free the source + void Release(); + + //save data to file + bool SaveToFile(const char *fileName); + + //add a inner file from buffer + bool AddFromBuffer(const char *innerFile, const char *buffer, int buferSize); + + //add a inner file from local file + bool AddFromFile(const char *localFile, const char *innerFile); + + //get the last error + const char *GetError(); + +private: + zip_archive_source_t *m_docSource; //zip file handle + zip_err_t *m_errobj; + char m_error[500]; +};