400 lines
10 KiB
C
400 lines
10 KiB
C
/* -*- C -*-
|
|
* Copyright (c) 1998-2006 Motoyuki Kasahara
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
* 3. Neither the name of the project nor the names of its contributors
|
|
* may be used to endorse or promote products derived from this software
|
|
* without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
* SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "ebzip.h"
|
|
#include "ebutils.h"
|
|
|
|
/*
|
|
* File name to be deleted and file to be closed when signal is received.
|
|
*/
|
|
static const char *trap_file_name = NULL;
|
|
static int trap_file = -1;
|
|
|
|
/*
|
|
* Unexported function.
|
|
*/
|
|
static void trap(int signal_number);
|
|
|
|
|
|
/*
|
|
* Copy a file.
|
|
* If it succeeds, 0 is returned. Otherwise -1 is returned.
|
|
*/
|
|
int
|
|
ebzip_copy_file(const char *out_file_name, const char *in_file_name)
|
|
{
|
|
unsigned char buffer[EB_SIZE_PAGE];
|
|
off_t total_length;
|
|
struct stat in_status, out_status;
|
|
int in_file = -1, out_file = -1;
|
|
ssize_t in_length;
|
|
int progress_interval;
|
|
int total_slices;
|
|
struct utimbuf utim;
|
|
int i;
|
|
|
|
/*
|
|
* Output file name information.
|
|
*/
|
|
if (!ebzip_quiet_flag) {
|
|
fprintf(stderr, _("==> copy %s <==\n"), in_file_name);
|
|
fprintf(stderr, _("output to %s\n"), out_file_name);
|
|
fflush(stderr);
|
|
}
|
|
|
|
/*
|
|
* Check for the input file.
|
|
*/
|
|
if (stat(in_file_name, &in_status) != 0 || !S_ISREG(in_status.st_mode)) {
|
|
fprintf(stderr, _("%s: no such file: %s\n"), invoked_name,
|
|
in_file_name);
|
|
goto failed;
|
|
}
|
|
|
|
/*
|
|
* Do nothing if the `in_file_name' and `out_file_name' are the same.
|
|
*/
|
|
if (is_same_file(out_file_name, in_file_name)) {
|
|
if (!ebzip_quiet_flag) {
|
|
fprintf(stderr,
|
|
_("the input and output files are the same, skipped.\n\n"));
|
|
fflush(stderr);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* When test mode, return immediately.
|
|
*/
|
|
if (ebzip_test_flag) {
|
|
#if defined(PRINTF_LL_MODIFIER)
|
|
fprintf(stderr, _("completed (%llu / %llu bytes)\n"),
|
|
(unsigned long long) in_status.st_size,
|
|
(unsigned long long) in_status.st_size);
|
|
#elif defined(PRINTF_I64_MODIFIER)
|
|
fprintf(stderr, _("completed (%I64u / %I64u bytes)\n"),
|
|
(unsigned __int64) in_status.st_size,
|
|
(unsigned __int64) in_status.st_size);
|
|
#else
|
|
fprintf(stderr, _("completed (%lu / %lu bytes)\n"),
|
|
(unsigned long) in_status.st_size,
|
|
(unsigned long) in_status.st_size);
|
|
#endif
|
|
fputc('\n', stderr);
|
|
fflush(stderr);
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* If the file to be output already exists, confirm and unlink it.
|
|
*/
|
|
if (stat(out_file_name, &out_status) == 0
|
|
&& S_ISREG(out_status.st_mode)) {
|
|
if (ebzip_overwrite_mode == EBZIP_OVERWRITE_NO) {
|
|
if (!ebzip_quiet_flag) {
|
|
fputs(_("already exists, skip the file\n\n"), stderr);
|
|
fflush(stderr);
|
|
}
|
|
return 0;
|
|
}
|
|
if (ebzip_overwrite_mode == EBZIP_OVERWRITE_CONFIRM) {
|
|
int y_or_n;
|
|
|
|
fprintf(stderr, _("\nthe file already exists: %s\n"),
|
|
out_file_name);
|
|
y_or_n = query_y_or_n(_("do you wish to overwrite (y or n)? "));
|
|
fputc('\n', stderr);
|
|
fflush(stderr);
|
|
if (!y_or_n)
|
|
return 0;
|
|
}
|
|
if (unlink(out_file_name) < 0) {
|
|
fprintf(stderr, _("%s: failed to unlink the file: %s\n"),
|
|
invoked_name, out_file_name);
|
|
goto failed;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Open files.
|
|
*/
|
|
in_file = open(in_file_name, O_RDONLY | O_BINARY);
|
|
if (in_file < 0) {
|
|
fprintf(stderr, _("%s: failed to open the file, %s: %s\n"),
|
|
invoked_name, strerror(errno), in_file_name);
|
|
goto failed;
|
|
}
|
|
|
|
trap_file_name = out_file_name;
|
|
#ifdef SIGHUP
|
|
signal(SIGHUP, trap);
|
|
#endif
|
|
signal(SIGINT, trap);
|
|
#ifdef SIGQUIT
|
|
signal(SIGQUIT, trap);
|
|
#endif
|
|
#ifdef SIGTERM
|
|
signal(SIGTERM, trap);
|
|
#endif
|
|
|
|
#ifdef O_CREAT
|
|
out_file = open(out_file_name, O_CREAT | O_TRUNC | O_WRONLY | O_BINARY,
|
|
0666 ^ get_umask());
|
|
#else
|
|
out_file = creat(out_file_name, 0666 ^ get_umask());
|
|
#endif
|
|
if (out_file < 0) {
|
|
fprintf(stderr, _("%s: failed to open the file, %s: %s\n"),
|
|
invoked_name, strerror(errno), out_file_name);
|
|
goto failed;
|
|
}
|
|
trap_file = out_file;
|
|
|
|
/*
|
|
* Read data from the input file, compress the data, and then
|
|
* write them to the output file.
|
|
*/
|
|
total_length = 0;
|
|
total_slices = (in_status.st_size + EB_SIZE_PAGE - 1) / EB_SIZE_PAGE;
|
|
progress_interval = EBZIP_PROGRESS_INTERVAL_FACTOR;
|
|
for (i = 0; i < total_slices; i++) {
|
|
/*
|
|
* Read data from `in_file', and write them to `out_file'.
|
|
*/
|
|
in_length = read(in_file, buffer, EB_SIZE_PAGE);
|
|
if (in_length < 0) {
|
|
fprintf(stderr, _("%s: failed to read from the file, %s: %s\n"),
|
|
invoked_name, strerror(errno), in_file_name);
|
|
goto failed;
|
|
} else if (in_length == 0) {
|
|
fprintf(stderr, _("%s: unexpected EOF: %s\n"),
|
|
invoked_name, in_file_name);
|
|
goto failed;
|
|
} else if (in_length != EB_SIZE_PAGE
|
|
&& total_length + in_length != in_status.st_size) {
|
|
fprintf(stderr, _("%s: unexpected EOF: %s\n"),
|
|
invoked_name, in_file_name);
|
|
goto failed;
|
|
}
|
|
|
|
/*
|
|
* Write decoded data to `out_file'.
|
|
*/
|
|
if (write(out_file, buffer, in_length) != in_length) {
|
|
fprintf(stderr, _("%s: failed to write to the file, %s: %s\n"),
|
|
invoked_name, strerror(errno), out_file_name);
|
|
goto failed;
|
|
}
|
|
total_length += in_length;
|
|
|
|
/*
|
|
* Output status information unless `quiet' mode.
|
|
*/
|
|
if (!ebzip_quiet_flag && (i + 1) % progress_interval == 0) {
|
|
#if defined(PRINTF_LL_MODIFIER)
|
|
fprintf(stderr, _("%4.1f%% done (%llu / %llu bytes)\n"),
|
|
(double) (i + 1) * 100.0 / (double) total_slices,
|
|
(unsigned long long) total_length,
|
|
(unsigned long long) in_status.st_size);
|
|
#elif defined(PRINTF_I64_MODIFIER)
|
|
fprintf(stderr, _("%4.1f%% done (%I64u / %I64u bytes)\n"),
|
|
(double) (i + 1) * 100.0 / (double) total_slices,
|
|
(unsigned __int64) total_length,
|
|
(unsigned __int64) in_status.st_size);
|
|
#else
|
|
fprintf(stderr, _("%4.1f%% done (%lu / %lu bytes)\n"),
|
|
(double) (i + 1) * 100.0 / (double) total_slices,
|
|
(unsigned long) total_length,
|
|
(unsigned long) in_status.st_size);
|
|
#endif
|
|
fflush(stderr);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Output the result unless quiet mode.
|
|
*/
|
|
if (!ebzip_quiet_flag) {
|
|
#if defined(PRINTF_LL_MODIFIER)
|
|
fprintf(stderr, _("completed (%llu / %llu bytes)\n"),
|
|
(unsigned long long) total_length,
|
|
(unsigned long long) in_status.st_size);
|
|
#elif defined(PRINTF_I64_MODIFIER)
|
|
fprintf(stderr, _("completed (%I64u / %I64u bytes)\n"),
|
|
(unsigned __int64) total_length,
|
|
(unsigned __int64) in_status.st_size);
|
|
#else
|
|
fprintf(stderr, _("completed (%lu / %lu bytes)\n"),
|
|
(unsigned long) total_length,
|
|
(unsigned long) in_status.st_size);
|
|
#endif
|
|
fputc('\n', stderr);
|
|
fflush(stderr);
|
|
}
|
|
|
|
/*
|
|
* Close files.
|
|
*/
|
|
close(in_file);
|
|
in_file = -1;
|
|
|
|
close(out_file);
|
|
out_file = -1;
|
|
trap_file = -1;
|
|
trap_file_name = NULL;
|
|
#ifdef SIGHUP
|
|
signal(SIGHUP, SIG_DFL);
|
|
#endif
|
|
signal(SIGINT, SIG_DFL);
|
|
#ifdef SIGQUIT
|
|
signal(SIGQUIT, SIG_DFL);
|
|
#endif
|
|
#ifdef SIGTERM
|
|
signal(SIGTERM, SIG_DFL);
|
|
#endif
|
|
|
|
/*
|
|
* Delete an original file unless the keep flag is set.
|
|
*/
|
|
if (!ebzip_test_flag && !ebzip_keep_flag)
|
|
unlink_files_add(in_file_name);
|
|
|
|
/*
|
|
* Set owner, group, permission, atime and mtime of `out_file'.
|
|
* We ignore return values of `chown', `chmod' and `utime'.
|
|
*/
|
|
utim.actime = in_status.st_atime;
|
|
utim.modtime = in_status.st_mtime;
|
|
utime(out_file_name, &utim);
|
|
|
|
return 0;
|
|
|
|
/*
|
|
* An error occurs...
|
|
*/
|
|
failed:
|
|
if (0 <= in_file)
|
|
close(in_file);
|
|
if (0 <= out_file)
|
|
close(out_file);
|
|
|
|
fputc('\n', stderr);
|
|
fflush(stderr);
|
|
|
|
return -1;
|
|
}
|
|
|
|
|
|
/*
|
|
* Copy all files in a directory.
|
|
* If it succeeds, 0 is returned. Otherwise -1 is returned.
|
|
*/
|
|
int
|
|
ebzip_copy_files_in_directory(const char *out_directory_name,
|
|
const char *in_directory_name)
|
|
{
|
|
struct dirent *entry;
|
|
DIR *dir = NULL;
|
|
struct stat in_status;
|
|
char in_path_name[PATH_MAX + 1];
|
|
char out_path_name[PATH_MAX + 1];
|
|
|
|
/*
|
|
* Check for the input directory.
|
|
*/
|
|
if (stat(in_directory_name, &in_status) != 0
|
|
|| !S_ISDIR(in_status.st_mode))
|
|
return 0;
|
|
|
|
|
|
/*
|
|
* Make the output directory if missing.
|
|
*/
|
|
if (!ebzip_test_flag
|
|
&& make_missing_directory(out_directory_name, 0777 ^ get_umask())
|
|
< 0) {
|
|
fprintf(stderr, _("%s: failed to create a directory, %s: %s\n"),
|
|
invoked_name, strerror(errno), out_directory_name);
|
|
goto failed;
|
|
}
|
|
|
|
/*
|
|
* Open the directory `path'.
|
|
*/
|
|
dir = opendir(in_directory_name);
|
|
if (dir == NULL) {
|
|
fprintf(stderr, _("%s: failed to open the directory, %s: %s\n"),
|
|
invoked_name, strerror(errno), in_directory_name);
|
|
goto failed;
|
|
}
|
|
|
|
for (;;) {
|
|
/*
|
|
* Read the directory entry.
|
|
*/
|
|
entry = readdir(dir);
|
|
if (entry == NULL)
|
|
break;
|
|
|
|
eb_compose_path_name(in_directory_name, entry->d_name, in_path_name);
|
|
eb_compose_path_name(out_directory_name, entry->d_name, out_path_name);
|
|
|
|
if (stat(in_path_name, &in_status) != 0
|
|
|| !S_ISREG(in_status.st_mode))
|
|
continue;
|
|
|
|
ebzip_copy_file(out_path_name, in_path_name);
|
|
}
|
|
|
|
closedir(dir);
|
|
return 0;
|
|
|
|
/*
|
|
* An error occurs...
|
|
*/
|
|
failed:
|
|
if (dir != NULL)
|
|
closedir(dir);
|
|
return -1;
|
|
}
|
|
|
|
|
|
/*
|
|
* Signal handler.
|
|
*/
|
|
static void
|
|
trap(int signal_number)
|
|
{
|
|
if (0 <= trap_file)
|
|
close(trap_file);
|
|
if (trap_file_name != NULL)
|
|
unlink(trap_file_name);
|
|
|
|
exit(1);
|
|
}
|