/* vi: set sw=4 ts=4: */
/*
* Mini tar implementation for busybox based on code taken from sash.
*
* Copyright (c) 1999 by David I. Bell
* Permission is granted to use, distribute, or modify this source,
* provided that this copyright notice remains intact.
*
* Permission to distribute this code under the GPL has been granted.
*
* Modified for busybox by Erik Andersen <andersee@debian.org>
* Adjusted to grok stdin/stdout options.
*
* Modified to handle device special files by Matt Porter
* <porter@debian.org>
*
* 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
*/
#include "internal.h"
#include <stdio.h>
#include <dirent.h>
#include <errno.h>
#include <fcntl.h>
#include <signal.h>
#include <time.h>
#include <utime.h>
#include <sys/types.h>
#include <sys/sysmacros.h>
#include <sys/param.h> /* for PATH_MAX */
#ifdef BB_FEATURE_TAR_CREATE
static const char tar_usage[] =
"tar -[cxtvOf] [tarFileName] [FILE] ...\n\n"
"Create, extract, or list files from a tar file.\n\n"
"Options:\n"
"\tc=create, x=extract, t=list contents, v=verbose,\n"
"\tO=extract to stdout, f=tarfile or \"-\" for stdin\n";
#else
static const char tar_usage[] =
"tar -[xtvOf] [tarFileName] [FILE] ...\n\n"
"Extract, or list files stored in a tar file. This\n"
"version of tar does not support creation of tar files.\n\n"
"Options:\n"
"\tx=extract, t=list contents, v=verbose,\n"
"\tO=extract to stdout, f=tarfile or \"-\" for stdin\n";
#endif
/* Tar file constants */
/* POSIX tar Header Block, from POSIX 1003.1-1990 */
struct TarHeader
{
/* byte offset */
char name[100]; /* 0 */
char mode[8]; /* 100 */
char uid[8]; /* 108 */
char gid[8]; /* 116 */
char size[12]; /* 124 */
char mtime[12]; /* 136 */
char chksum[8]; /* 148 */
char typeflag; /* 156 */
char linkname[100]; /* 157 */
char magic[6]; /* 257 */
char version[2]; /* 263 */
char uname[32]; /* 265 */
char gname[32]; /* 297 */
char devmajor[8]; /* 329 */
char devminor[8]; /* 337 */
char prefix[155]; /* 345 */
/* padding 500 */
};
typedef struct TarHeader TarHeader;
/* A few useful constants */
#define TAR_MAGIC "ustar" /* ustar and a null */
#define TAR_VERSION "00" /* 00 and no null */
#define TAR_MAGIC_LEN 6
#define TAR_VERSION_LEN 2
#define TAR_NAME_LEN 100
#define TAR_BLOCK_SIZE 512
/* A nice enum with all the possible tar file content types */
enum TarFileType
{
REGTYPE = '0', /* regular file */
REGTYPE0 = '\0', /* regular file (ancient bug compat)*/
LNKTYPE = '1', /* hard link */
SYMTYPE = '2', /* symbolic link */
CHRTYPE = '3', /* character special */
BLKTYPE = '4', /* block special */
DIRTYPE = '5', /* directory */
FIFOTYPE = '6', /* FIFO special */
CONTTYPE = '7', /* reserved */
};
typedef enum TarFileType TarFileType;
/* This struct ignores magic, non-numeric user name,
* non-numeric group name, and the checksum, since
* these are all ignored by BusyBox tar. */
struct TarInfo
{
int tarFd; /* An open file descriptor for reading from the tarball */
char * name; /* File name */
mode_t mode; /* Unix mode, including device bits. */
uid_t uid; /* Numeric UID */
gid_t gid; /* Numeric GID */
size_t size; /* Size of file */
time_t mtime; /* Last-modified time */
enum TarFileType type; /* Regular, directory, link, etc */
char * linkname; /* Name for symbolic and hard links */
dev_t device; /* Special device for mknod() */
};
typedef struct TarInfo TarInfo;
/* Static data */
static const unsigned long TarChecksumOffset = (const unsigned long)&(((TarHeader *)0)->chksum);
/* Local procedures to restore files from a tar file. */
static int readTarFile(const char* tarName, int extractFlag, int listFlag,
int tostdoutFlag, int verboseFlag);
static long getOctal(const char *cp, int len);
static int parseTarHeader(struct TarHeader *rawHeader, struct TarInfo *header);
#ifdef BB_FEATURE_TAR_CREATE
/*
* Local procedures to save files into a tar file.
*/
static void saveFile(const char *fileName, int seeLinks);
static void saveRegularFile(const char *fileName,
const struct stat *statbuf);
static void saveDirectory(const char *fileName,
const struct stat *statbuf);
static void writeHeader(const char *fileName, const struct stat *statbuf);
static void writeTarFile(int argc, char **argv);
static void writeTarBlock(const char *buf, int len);
static int putOctal(char *cp, int len, long value);
#endif
extern int tar_main(int argc, char **argv)
{
const char *tarName=NULL;
const char *options;
int listFlag = FALSE;
int extractFlag = FALSE;
int createFlag = FALSE;
int verboseFlag = FALSE;
int tostdoutFlag = FALSE;
argc--;
argv++;
if (argc < 1)
usage(tar_usage);
/* Parse options */
if (**argv == '-')
options = (*argv++) + 1;
else
options = (*argv++);
argc--;
for (; *options; options++) {
switch (*options) {
case 'f':
if (tarName != NULL)
fatalError( "Only one 'f' option allowed\n");
tarName = *argv++;
if (tarName == NULL)
fatalError( "Option requires an argument: No file specified\n");
argc--;
break;
case 't':
if (extractFlag == TRUE || createFlag == TRUE)
goto flagError;
listFlag = TRUE;
break;
case 'x':
if (listFlag == TRUE || createFlag == TRUE)
goto flagError;
extractFlag = TRUE;
break;
case 'c':
if (extractFlag == TRUE || listFlag == TRUE)
goto flagError;
createFlag = TRUE;
break;
case 'v':
verboseFlag = TRUE;
break;
case 'O':
tostdoutFlag = TRUE;
break;
case '-':
usage(tar_usage);
break;
default:
fatalError( "Unknown tar flag '%c'\n"
"Try `tar --help' for more information\n", *options);
}
}
/*
* Do the correct type of action supplying the rest of the
* command line arguments as the list of files to process.
*/
if (createFlag == TRUE) {
#ifndef BB_FEATURE_TAR_CREATE
fatalError( "This version of tar was not compiled with tar creation support.\n");
#else
exit(writeTarFile(argc, argv));
#endif
} else {
exit(readTarFile(tarName, extractFlag, listFlag, tostdoutFlag, verboseFlag));
}
flagError:
fatalError( "Exactly one of 'c', 'x' or 't' must be specified\n");
}
static void
tarExtractRegularFile(TarInfo *header, int extractFlag, int listFlag, int tostdoutFlag, int verboseFlag)
{
return;
}
static void
tarExtractDirectory(TarInfo *header, int extractFlag, int listFlag, int tostdoutFlag, int verboseFlag)
{
return;
}
static void
tarExtractHardLink(TarInfo *header, int extractFlag, int listFlag, int tostdoutFlag, int verboseFlag)
{
return;
}
static void
tarExtractSymLink(TarInfo *header, int extractFlag, int listFlag, int tostdoutFlag, int verboseFlag)
{
return;
}
static void
tarExtractSpecial(TarInfo *header, int extractFlag, int listFlag, int tostdoutFlag, int verboseFlag)
{
return;
}
/*
* Read a tar file and extract or list the specified files within it.
* If the list is empty than all files are extracted or listed.
*/
static int readTarFile(const char* tarName, int extractFlag, int listFlag,
int tostdoutFlag, int verboseFlag)
{
int status, tarFd=0;
int errorFlag=FALSE;
TarHeader rawHeader;
TarInfo header;
/* Open the tar file for reading. */
if (!strcmp(tarName, "-"))
tarFd = fileno(stdin);
else
tarFd = open(tarName, O_RDONLY);
if (tarFd < 0) {
errorMsg( "Error opening '%s': %s", tarName, strerror(errno));
return ( FALSE);
}
/* Read the tar file */
while ( (status = fullRead(tarFd, (char*)&rawHeader, TAR_BLOCK_SIZE)) == TAR_BLOCK_SIZE ) {
/* Now see if the header looks ok */
if ( parseTarHeader(&rawHeader, &header) == FALSE ) {
close( tarFd);
if ( *(header.name) == '\0' ) {
goto endgame;
} else {
errorFlag=TRUE;
errorMsg("Bad tar header, skipping\n");
continue;
}
}
if ( *(header.name) == '\0' )
goto endgame;
if (extractFlag == FALSE) {
if (verboseFlag == TRUE) {
printf("%s %3d/%-d ", modeString(header.mode), header.uid, header.gid);
if (header.type==CHRTYPE || header.type==BLKTYPE)
printf("%4d,%4d %s ", MAJOR(header.device),
MINOR(header.device), timeString(header.mtime));
else
printf("%9ld %s ", header.size, timeString(header.mtime));
}
printf("%s", header.name);
if (header.type==LNKTYPE)
printf(" (link to \"%s\")", hp->linkName);
else if (header.type==SYMTYPE)
printf(" (symlink to \"%s\")", hp->linkName);
printf("\n");
continue;
}
/* If we got here, we can be certain we have a legitimate
* header to work with. So work with it. */
switch ( header.type ) {
case REGTYPE:
case REGTYPE0:
/* If the name ends in a '/' then assume it is
* supposed to be a directory, and fall through */
if (header.name[strlen(header.name)-1] != '/') {
tarExtractRegularFile(&header, extractFlag, listFlag, tostdoutFlag, verboseFlag);
break;
}
case DIRTYPE:
tarExtractDirectory( &header, extractFlag, listFlag, tostdoutFlag, verboseFlag);
break;
case LNKTYPE:
tarExtractHardLink( &header, extractFlag, listFlag, tostdoutFlag, verboseFlag);
break;
case SYMTYPE:
tarExtractSymLink( &header, extractFlag, listFlag, tostdoutFlag, verboseFlag);
break;
case CHRTYPE:
case BLKTYPE:
case FIFOTYPE:
tarExtractSpecial( &header, extractFlag, listFlag, tostdoutFlag, verboseFlag);
break;
default:
close( tarFd);
return( FALSE);
}
}
close(tarFd);
if (status > 0) {
/* Bummer - we read a partial header */
errorMsg( "Error reading '%s': %s", tarName, strerror(errno));
return ( FALSE);
}
else
return( status);
/* Stuff we do when we know we are done with the file */
endgame:
close( tarFd);
if ( *(header.name) == '\0' ) {
if (errorFlag==FALSE)
return( TRUE);
}
return( FALSE);
}
/*
* Read an octal value in a field of the specified width, with optional
* spaces on both sides of the number and with an optional null character
* at the end. Returns -1 on an illegal format.
*/
static long getOctal(const char *cp, int size)
{
long val = 0;
for(;(size > 0) && (*cp == ' '); cp++, size--);
if ((size == 0) || !isOctal(*cp))
return -1;
for(; (size > 0) && isOctal(*cp); size--) {
val = val * 8 + *cp++ - '0';
}
for (;(size > 0) && (*cp == ' '); cp++, size--);
if ((size > 0) && *cp)
return -1;
return val;
}
/* Parse the tar header and fill in the nice struct with the details */
static int
parseTarHeader(struct TarHeader *rawHeader, struct TarInfo *header)
{
int i;
long chksum, sum;
unsigned char *s = (unsigned char *)rawHeader;
header->name = rawHeader->name;
header->mode = getOctal(rawHeader->mode, sizeof(rawHeader->mode));
header->uid = getOctal(rawHeader->uid, sizeof(rawHeader->uid));
header->gid = getOctal(rawHeader->gid, sizeof(rawHeader->gid));
header->size = getOctal(rawHeader->size, sizeof(rawHeader->size));
header->mtime = getOctal(rawHeader->mtime, sizeof(rawHeader->mtime));
chksum = getOctal(rawHeader->chksum, sizeof(rawHeader->chksum));
header->type = rawHeader->typeflag;
header->linkname = rawHeader->linkname;
header->device = MAJOR(getOctal(rawHeader->devmajor, sizeof(rawHeader->devmajor))) |
MINOR(getOctal(rawHeader->devminor, sizeof(rawHeader->devminor)));
/* Check the checksum */
sum = ' ' * sizeof(rawHeader->chksum);
for ( i = TarChecksumOffset; i > 0; i-- )
sum += *s++;
s += sizeof(rawHeader->chksum);
for ( i = (512 - TarChecksumOffset - sizeof(rawHeader->chksum)); i > 0; i-- )
sum += *s++;
if (sum == chksum )
return ( TRUE);
return( FALSE);
}
#if 0
if ((header->mode < 0) || (header->uid < 0) ||
(header->gid < 0) || (header->size < 0)) {
errorMsg(stderr, "Bad tar header, skipping\n");
return( FALSE);
}
badHeader = FALSE;
skipFileFlag = FALSE;
devFileFlag = FALSE;
/*
* Check for the file modes.
*/
hardLink = ((hp->typeFlag == TAR_TYPE_HARD_LINK) ||
(hp->typeFlag == TAR_TYPE_HARD_LINK - '0'));
softLink = ((hp->typeFlag == TAR_TYPE_SOFT_LINK) ||
(hp->typeFlag == TAR_TYPE_SOFT_LINK - '0'));
/*
* Check for a directory.
*/
if (outName[strlen(outName) - 1] == '/')
mode |= S_IFDIR;
/*
* Check for absolute paths in the file.
* If we find any, then warn the user and make them relative.
*/
if (*outName == '/') {
while (*outName == '/')
outName++;
if (warnedRoot == FALSE) {
fprintf(stderr,
"Absolute path detected, removing leading slashes\n");
}
warnedRoot = TRUE;
}
/*
* See if we want this file to be restored.
* If not, then set up to skip it.
*/
if (wantFileName(outName, argc, argv) == FALSE) {
if (!hardLink && !softLink && (S_ISREG(mode) || S_ISCHR(mode)
|| S_ISBLK(mode) || S_ISSOCK(mode)
|| S_ISFIFO(mode))) {
inHeader = (size == 0) ? TRUE : FALSE;
dataCc = size;
}
skipFileFlag = TRUE;
return;
}
/*
* This file is to be handled.
* If we aren't extracting then just list information about the file.
*/
if (extractFlag == FALSE) {
if (verboseFlag == TRUE) {
printf("%s %3d/%-d ", modeString(mode), uid, gid);
if (S_ISCHR(mode) || S_ISBLK(mode))
printf("%4d,%4d %s ", major, minor, timeString(mtime));
else
printf("%9ld %s ", size, timeString(mtime));
}
printf("%s", outName);
if (hardLink)
printf(" (link to \"%s\")", hp->linkName);
else if (softLink)
printf(" (symlink to \"%s\")", hp->linkName);
else if (S_ISREG(mode) || S_ISCHR(mode) || S_ISBLK(mode) ||
S_ISSOCK(mode) || S_ISFIFO(mode)) {
inHeader = (size == 0) ? TRUE : FALSE;
dataCc = size;
}
printf("\n");
return;
}
/*
* We really want to extract the file.
*/
if (verboseFlag == TRUE)
printf("x %s\n", outName);
if (hardLink) {
if (link(hp->linkName, outName) < 0) {
perror(outName);
return;
}
/* Set the file time */
utb.actime = mtime;
utb.modtime = mtime;
utime(outName, &utb);
/* Set the file permissions */
chown(outName, uid, gid);
chmod(outName, mode);
return;
}
if (softLink) {
#ifdef S_ISLNK
if (symlink(hp->linkName, outName) < 0) {
perror(outName);
return;
}
/* Try to change ownership of the symlink.
* If libs doesn't support that, don't bother.
* Changing the pointed-to file is the Wrong Thing(tm).
*/
#if (__GLIBC__ >= 2) && (__GLIBC_MINOR__ >= 1)
lchown(outName, uid, gid);
#endif
/* Do not change permissions or date on symlink,
* since it changes the pointed to file instead. duh. */
#else
fprintf(stderr, "Cannot create symbolic links\n");
#endif
return;
}
/* Set the umask for this process so it doesn't
* screw things up. */
umask(0);
/*
* If the file is a directory, then just create the path.
*/
if (S_ISDIR(mode)) {
if (createPath(outName, mode) == TRUE) {
/* make the final component, just in case it was
* omitted by createPath() (which will skip the
* directory if it doesn't have a terminating '/')
*/
mkdir(outName, mode);
/* Set the file time */
utb.actime = mtime;
utb.modtime = mtime;
utime(outName, &utb);
/* Set the file permissions */
chown(outName, uid, gid);
chmod(outName, mode);
return;
}
return;
}
/*
* There is a file to write.
* First create the path to it if necessary with default permissions.
*/
createPath(outName, 0777);
inHeader = (size == 0) ? TRUE : FALSE;
dataCc = size;
/*
* Start the output file.
*/
if (tostdoutFlag == TRUE)
outFd = fileno(stdout);
else {
if (S_ISCHR(mode) || S_ISBLK(mode) || S_ISSOCK(mode)) {
devFileFlag = TRUE;
outFd = mknod(outName, mode, makedev(major, minor));
} else if (S_ISFIFO(mode)) {
devFileFlag = TRUE;
outFd = mkfifo(outName, mode);
} else {
outFd = open(outName, O_WRONLY | O_CREAT | O_TRUNC, mode);
}
if (outFd < 0) {
perror(outName);
skipFileFlag = TRUE;
return;
}
/* Set the file time */
utb.actime = mtime;
utb.modtime = mtime;
utime(outName, &utb);
/* Set the file permissions */
chown(outName, uid, gid);
chmod(outName, mode);
}
/*
* If the file is empty, then that's all we need to do.
*/
if (size == 0 && (tostdoutFlag == FALSE) && (devFileFlag == FALSE)) {
close(outFd);
outFd = -1;
}
}
/*
* Handle a data block of some specified size that was read.
*/
static void readData(const char *cp, int count)
{
/*
* Reduce the amount of data left in this file.
* If there is no more data left, then we need to read
* the header again.
*/
dataCc -= count;
if (dataCc <= 0)
inHeader = TRUE;
/*
* If we aren't extracting files or this file is being
* skipped then do nothing more.
*/
if (extractFlag == FALSE || skipFileFlag == TRUE)
return;
/*
* Write the data to the output file.
*/
if (fullWrite(outFd, cp, count) < 0) {
perror(outName);
if (tostdoutFlag == FALSE) {
close(outFd);
outFd = -1;
}
skipFileFlag = TRUE;
return;
}
/*
* Check if we are done writing to the file now.
*/
if (dataCc <= 0 && tostdoutFlag == FALSE) {
struct utimbuf utb;
if (close(outFd))
perror(outName);
/* Set the file time */
utb.actime = mtime;
utb.modtime = mtime;
utime(outName, &utb);
/* Set the file permissions */
chown(outName, uid, gid);
chmod(outName, mode);
outFd = -1;
}
}
/*
* See if the specified file name belongs to one of the specified list
* of path prefixes. An empty list implies that all files are wanted.
* Returns TRUE if the file is selected.
*/
static int
wantFileName(const char *fileName, int argc, char **argv)
{
const char *pathName;
int fileLength;
int pathLength;
/*
* If there are no files in the list, then the file is wanted.
*/
if (argc == 0)
return TRUE;
fileLength = strlen(fileName);
/*
* Check each of the test paths.
*/
while (argc-- > 0) {
pathName = *argv++;
pathLength = strlen(pathName);
if (fileLength < pathLength)
continue;
if (memcmp(fileName, pathName, pathLength) != 0)
continue;
if ((fileLength == pathLength) || (fileName[pathLength] == '/')) {
return TRUE;
}
}
return FALSE;
}
/* From here to the end of the file is the tar writing stuff.
* If you do not have BB_FEATURE_TAR_CREATE defined, this will
* not be built.
* */
#ifdef BB_FEATURE_TAR_CREATE
/*
* Write a tar file containing the specified files.
*/
static void writeTarFile(int argc, char **argv)
{
struct stat statbuf;
/*
* Make sure there is at least one file specified.
*/
if (argc <= 0) {
fprintf(stderr, "No files specified to be saved\n");
errorFlag = TRUE;
}
/*
* Create the tar file for writing.
*/
if ((tarName == NULL) || !strcmp(tarName, "-")) {
tostdoutFlag = TRUE;
tarFd = fileno(stdout);
} else
tarFd = open(tarName, O_WRONLY | O_CREAT | O_TRUNC, 0666);
if (tarFd < 0) {
perror(tarName);
errorFlag = TRUE;
return;
}
/*
* Get the device and inode of the tar file for checking later.
*/
if (fstat(tarFd, &statbuf) < 0) {
perror(tarName);
errorFlag = TRUE;
goto done;
}
tarDev = statbuf.st_dev;
tarInode = statbuf.st_ino;
/*
* Append each file name into the archive file.
* Follow symbolic links for these top level file names.
*/
while (errorFlag == FALSE && (argc-- > 0)) {
saveFile(*argv++, FALSE);
}
/*
* Now write an empty block of zeroes to end the archive.
*/
writeTarBlock("", 1);
done:
/*
* Close the tar file and check for errors if it was opened.
*/
if ((tostdoutFlag == FALSE) && (tarFd >= 0) && (close(tarFd) < 0))
perror(tarName);
}
/*
* Save one file into the tar file.
* If the file is a directory, then this will recursively save all of
* the files and directories within the directory. The seeLinks
* flag indicates whether or not we want to see symbolic links as
* they really are, instead of blindly following them.
*/
static void saveFile(const char *fileName, int seeLinks)
{
int status;
struct stat statbuf;
if (verboseFlag == TRUE)
printf("a %s\n", fileName);
/*
* Check that the file name will fit in the header.
*/
if (strlen(fileName) >= TAR_NAME_SIZE) {
fprintf(stderr, "%s: File name is too long\n", fileName);
return;
}
/*
* Find out about the file.
*/
#ifdef S_ISLNK
if (seeLinks == TRUE)
status = lstat(fileName, &statbuf);
else
#endif
status = stat(fileName, &statbuf);
if (status < 0) {
perror(fileName);
return;
}
/*
* Make sure we aren't trying to save our file into itself.
*/
if ((statbuf.st_dev == tarDev) && (statbuf.st_ino == tarInode)) {
fprintf(stderr, "Skipping saving of archive file itself\n");
return;
}
/*
* Check the type of file.
*/
mode = statbuf.st_mode;
if (S_ISDIR(mode)) {
saveDirectory(fileName, &statbuf);
return;
}
if (S_ISREG(mode)) {
saveRegularFile(fileName, &statbuf);
return;
}
/* Some day add support for tarring these up... but not today. :) */
// if (S_ISLNK(mode) || S_ISFIFO(mode) || S_ISBLK(mode) || S_ISCHR (mode) ) {
// fprintf (stderr, "%s: This version of tar can't store this type of file\n", fileName);
// }
/*
* The file is a strange type of file, ignore it.
*/
fprintf(stderr, "%s: not a directory or regular file\n", fileName);
}
/*
* Save a regular file to the tar file.
*/
static void
saveRegularFile(const char *fileName, const struct stat *statbuf)
{
int sawEof;
int fileFd;
int cc;
int dataCount;
long fullDataCount;
char data[TAR_BLOCK_SIZE * 16];
/*
* Open the file for reading.
*/
fileFd = open(fileName, O_RDONLY);
if (fileFd < 0) {
perror(fileName);
return;
}
/*
* Write out the header for the file.
*/
writeHeader(fileName, statbuf);
/*
* Write the data blocks of the file.
* We must be careful to write the amount of data that the stat
* buffer indicated, even if the file has changed size. Otherwise
* the tar file will be incorrect.
*/
fullDataCount = statbuf->st_size;
sawEof = FALSE;
while (fullDataCount > 0) {
/*
* Get the amount to write this iteration which is
* the minumum of the amount left to write and the
* buffer size.
*/
dataCount = sizeof(data);
if (dataCount > fullDataCount)
dataCount = (int) fullDataCount;
/*
* Read the data from the file if we haven't seen the
* end of file yet.
*/
cc = 0;
if (sawEof == FALSE) {
cc = fullRead(fileFd, data, dataCount);
if (cc < 0) {
perror(fileName);
(void) close(fileFd);
errorFlag = TRUE;
return;
}
/*
* If the file ended too soon, complain and set
* a flag so we will zero fill the rest of it.
*/
if (cc < dataCount) {
fprintf(stderr, "%s: Short read - zero filling", fileName);
sawEof = TRUE;
}
}
/*
* Zero fill the rest of the data if necessary.
*/
if (cc < dataCount)
memset(data + cc, 0, dataCount - cc);
/*
* Write the buffer to the TAR file.
*/
writeTarBlock(data, dataCount);
fullDataCount -= dataCount;
}
/*
* Close the file.
*/
if ((tostdoutFlag == FALSE) && close(fileFd) < 0)
fprintf(stderr, "%s: close: %s\n", fileName, strerror(errno));
}
/*
* Save a directory and all of its files to the tar file.
*/
static void saveDirectory(const char *dirName, const struct stat *statbuf)
{
DIR *dir;
struct dirent *entry;
int needSlash;
char fullName[PATH_MAX + 1];
/*
* Construct the directory name as used in the tar file by appending
* a slash character to it.
*/
strcpy(fullName, dirName);
strcat(fullName, "/");
/*
* Write out the header for the directory entry.
*/
writeHeader(fullName, statbuf);
/*
* Open the directory.
*/
dir = opendir(dirName);
if (dir == NULL) {
fprintf(stderr, "Cannot read directory \"%s\": %s\n",
dirName, strerror(errno));
return;
}
/*
* See if a slash is needed.
*/
needSlash = (*dirName && (dirName[strlen(dirName) - 1] != '/'));
/*
* Read all of the directory entries and check them,
* except for the current and parent directory entries.
*/
while (errorFlag == FALSE && ((entry = readdir(dir)) != NULL)) {
if ((strcmp(entry->d_name, ".") == 0) ||
(strcmp(entry->d_name, "..") == 0)) {
continue;
}
/*
* Build the full path name to the file.
*/
strcpy(fullName, dirName);
if (needSlash)
strcat(fullName, "/");
strcat(fullName, entry->d_name);
/*
* Write this file to the tar file, noticing whether or not
* the file is a symbolic link.
*/
saveFile(fullName, TRUE);
}
/*
* All done, close the directory.
*/
closedir(dir);
}
/*
* Write a tar header for the specified file name and status.
* It is assumed that the file name fits.
*/
static void writeHeader(const char *fileName, const struct stat *statbuf)
{
long checkSum;
const unsigned char *cp;
int len;
TarHeader header;
/*
* Zero the header block in preparation for filling it in.
*/
memset((char *) &header, 0, sizeof(header));
/*
* Fill in the header.
*/
strcpy(header.name, fileName);
strncpy(header.magic, TAR_MAGIC, sizeof(header.magic));
strncpy(header.version, TAR_VERSION, sizeof(header.version));
putOctal(header.mode, sizeof(header.mode), statbuf->st_mode & 0777);
putOctal(header.uid, sizeof(header.uid), statbuf->st_uid);
putOctal(header.gid, sizeof(header.gid), statbuf->st_gid);
putOctal(header.size, sizeof(header.size), statbuf->st_size);
putOctal(header.mtime, sizeof(header.mtime), statbuf->st_mtime);
header.typeFlag = TAR_TYPE_REGULAR;
/*
* Calculate and store the checksum.
* This is the sum of all of the bytes of the header,
* with the checksum field itself treated as blanks.
*/
memset(header.checkSum, ' ', sizeof(header.checkSum));
cp = (const unsigned char *) &header;
len = sizeof(header);
checkSum = 0;
while (len-- > 0)
checkSum += *cp++;
putOctal(header.checkSum, sizeof(header.checkSum), checkSum);
/*
* Write the tar header.
*/
writeTarBlock((const char *) &header, sizeof(header));
}
/*
* Write data to one or more blocks of the tar file.
* The data is always padded out to a multiple of TAR_BLOCK_SIZE.
* The errorFlag static variable is set on an error.
*/
static void writeTarBlock(const char *buf, int len)
{
int partialLength;
int completeLength;
char fullBlock[TAR_BLOCK_SIZE];
/*
* If we had a write error before, then do nothing more.
*/
if (errorFlag == TRUE)
return;
/*
* Get the amount of complete and partial blocks.
*/
partialLength = len % TAR_BLOCK_SIZE;
completeLength = len - partialLength;
/*
* Write all of the complete blocks.
*/
if ((completeLength > 0) && !fullWrite(tarFd, buf, completeLength)) {
perror(tarName);
errorFlag = TRUE;
return;
}
/*
* If there are no partial blocks left, we are done.
*/
if (partialLength == 0)
return;
/*
* Copy the partial data into a complete block, and pad the rest
* of it with zeroes.
*/
memcpy(fullBlock, buf + completeLength, partialLength);
memset(fullBlock + partialLength, 0, TAR_BLOCK_SIZE - partialLength);
/*
* Write the last complete block.
*/
if (!fullWrite(tarFd, fullBlock, TAR_BLOCK_SIZE)) {
perror(tarName);
errorFlag = TRUE;
}
}
/*
* Put an octal string into the specified buffer.
* The number is zero and space padded and possibly null padded.
* Returns TRUE if successful.
*/
static int putOctal(char *cp, int len, long value)
{
int tempLength;
char *tempString;
char tempBuffer[32];
/*
* Create a string of the specified length with an initial space,
* leading zeroes and the octal number, and a trailing null.
*/
tempString = tempBuffer;
sprintf(tempString, " %0*lo", len - 2, value);
tempLength = strlen(tempString) + 1;
/*
* If the string is too large, suppress the leading space.
*/
if (tempLength > len) {
tempLength--;
tempString++;
}
/*
* If the string is still too large, suppress the trailing null.
*/
if (tempLength > len)
tempLength--;
/*
* If the string is still too large, fail.
*/
if (tempLength > len)
return FALSE;
/*
* Copy the string to the field.
*/
memcpy(cp, tempString, len);
return TRUE;
}
#endif
/* END CODE */
#endif