+
+#ifdef BB_FEATURE_TAR_CREATE
+
+/* Some info to be carried along when creating a new tarball */
+struct TarBallInfo
+{
+ char* fileName; /* File name of the tarball */
+ int tarFd; /* Open-for-write file descriptor
+ for the tarball */
+ struct stat statBuf; /* Stat info for the tarball, letting
+ us know the inode and device that the
+ tarball lives, so we can avoid trying
+ to include the tarball into itself */
+ int verboseFlag; /* Whether to print extra stuff or not */
+ char** excludeList; /* List of files to not include */
+};
+typedef struct TarBallInfo TarBallInfo;
+
+
+/* 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 tempBuffer[32];
+ char *tempString = tempBuffer;
+
+ /* Create a string of the specified length with an initial space,
+ * leading zeroes and the octal number, and a trailing null. */
+ sprintf (tempString, "%0*lo", len - 1, value);
+
+ /* If the string is too large, suppress the leading space. */
+ tempLength = strlen (tempString) + 1;
+ 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;
+}
+
+/* Write out a tar header for the specified file/directory/whatever */
+static int
+writeTarHeader(struct TarBallInfo *tbInfo, const char *fileName, struct stat *statbuf)
+{
+ long chksum=0;
+ struct TarHeader header;
+#if defined BB_FEATURE_TAR_EXCLUDE
+ char** tmpList;
+#endif
+ const unsigned char *cp = (const unsigned char *) &header;
+ ssize_t size = sizeof(struct TarHeader);
+
+ memset( &header, 0, size);
+
+ if (*fileName=='/') {
+ static int alreadyWarned=FALSE;
+ if (alreadyWarned==FALSE) {
+ errorMsg("Removing leading '/' from member names\n");
+ alreadyWarned=TRUE;
+ }
+ strncpy(header.name, fileName+1, sizeof(header.name));
+ }
+ else {
+ strncpy(header.name, fileName, sizeof(header.name));
+ }
+
+#if defined BB_FEATURE_TAR_EXCLUDE
+ /* Check for excluded files.... */
+ for (tmpList=tbInfo->excludeList; tmpList && *tmpList; tmpList++) {
+ /* Do some extra hoop jumping for when directory names
+ * end in '/' but the entry in tmpList doesn't */
+ if (strncmp( *tmpList, header.name, strlen(*tmpList))==0 || (
+ header.name[strlen(header.name)-1]=='/'
+ && strncmp( *tmpList, header.name,
+ MIN(strlen(header.name)-1, strlen(*tmpList)))==0)) {
+ /* Set the mode to something that is not a regular file, thereby
+ * faking out writeTarFile into thinking that nothing further need
+ * be done for this file. Yes, I know this is ugly, but it works. */
+ statbuf->st_mode = 0;
+ return( TRUE);
+ }
+ }
+#endif
+
+ putOctal(header.mode, sizeof(header.mode), statbuf->st_mode);
+ putOctal(header.uid, sizeof(header.uid), statbuf->st_uid);
+ putOctal(header.gid, sizeof(header.gid), statbuf->st_gid);
+ putOctal(header.size, sizeof(header.size), 0); /* Regular file size is handled later */
+ putOctal(header.mtime, sizeof(header.mtime), statbuf->st_mtime);
+ strncpy(header.magic, TAR_MAGIC TAR_VERSION,
+ TAR_MAGIC_LEN + TAR_VERSION_LEN );
+
+ /* Enter the user and group names (default to root if it fails) */
+ my_getpwuid(header.uname, statbuf->st_uid);
+ if (! *header.uname)
+ strcpy(header.uname, "root");
+ my_getgrgid(header.gname, statbuf->st_gid);
+ if (! *header.uname)
+ strcpy(header.uname, "root");
+
+ /* WARNING/NOTICE: I break Hard Links */
+ if (S_ISLNK(statbuf->st_mode)) {
+ int link_size=0;
+ char buffer[BUFSIZ];
+ header.typeflag = SYMTYPE;
+ link_size = readlink(fileName, buffer, sizeof(buffer) - 1);
+ if ( link_size < 0) {
+ errorMsg("Error reading symlink '%s': %s\n", header.name, strerror(errno));
+ return ( FALSE);
+ }
+ buffer[link_size] = '\0';
+ strncpy(header.linkname, buffer, sizeof(header.linkname));
+ } else if (S_ISDIR(statbuf->st_mode)) {
+ header.typeflag = DIRTYPE;
+ strncat(header.name, "/", sizeof(header.name));
+ } else if (S_ISCHR(statbuf->st_mode)) {
+ header.typeflag = CHRTYPE;
+ putOctal(header.devmajor, sizeof(header.devmajor), MAJOR(statbuf->st_rdev));
+ putOctal(header.devminor, sizeof(header.devminor), MINOR(statbuf->st_rdev));
+ } else if (S_ISBLK(statbuf->st_mode)) {
+ header.typeflag = BLKTYPE;
+ putOctal(header.devmajor, sizeof(header.devmajor), MAJOR(statbuf->st_rdev));
+ putOctal(header.devminor, sizeof(header.devminor), MINOR(statbuf->st_rdev));
+ } else if (S_ISFIFO(statbuf->st_mode)) {
+ header.typeflag = FIFOTYPE;
+ } else if (S_ISREG(statbuf->st_mode)) {
+ header.typeflag = REGTYPE;
+ putOctal(header.size, sizeof(header.size), statbuf->st_size);
+ } else {
+ errorMsg("%s: Unknown file type\n", fileName);
+ return ( FALSE);
+ }
+
+ /* Calculate and store the checksum (i.e. the sum of all of the bytes of
+ * the header). The checksum field must be filled with blanks for the
+ * calculation. The checksum field is formatted differently from the
+ * other fields: it has [6] digits, a null, then a space -- rather than
+ * digits, followed by a null like the other fields... */
+ memset(header.chksum, ' ', sizeof(header.chksum));
+ cp = (const unsigned char *) &header;
+ while (size-- > 0)
+ chksum += *cp++;
+ putOctal(header.chksum, 7, chksum);
+
+ /* Now write the header out to disk */
+ if ((size=fullWrite(tbInfo->tarFd, (char*)&header, sizeof(struct TarHeader))) < 0) {
+ errorMsg(io_error, fileName, strerror(errno));
+ return ( FALSE);
+ }
+ /* Pad the header up to the tar block size */
+ for (; size<TAR_BLOCK_SIZE; size++) {
+ write(tbInfo->tarFd, "\0", 1);
+ }
+ /* Now do the verbose thing (or not) */
+ if (tbInfo->verboseFlag==TRUE) {
+ FILE *vbFd = stdout;
+ if (tbInfo->tarFd == fileno(stdout)) // If the archive goes to stdout, verbose to stderr
+ vbFd = stderr;
+ fprintf(vbFd, "%s\n", header.name);
+ }
+
+ return ( TRUE);
+}
+
+
+static int writeFileToTarball(const char *fileName, struct stat *statbuf, void* userData)
+{
+ struct TarBallInfo *tbInfo = (struct TarBallInfo *)userData;
+
+ /* It is against the rules to archive a socket */
+ if (S_ISSOCK(statbuf->st_mode)) {
+ errorMsg("%s: socket ignored\n", fileName);
+ return( TRUE);
+ }
+
+ /* It is a bad idea to store the archive we are in the process of creating,
+ * so check the device and inode to be sure that this particular file isn't
+ * the new tarball */
+ if (tbInfo->statBuf.st_dev == statbuf->st_dev &&
+ tbInfo->statBuf.st_ino == statbuf->st_ino) {
+ errorMsg("%s: file is the archive; skipping\n", fileName);
+ return( TRUE);
+ }
+
+ if (writeTarHeader(tbInfo, fileName, statbuf)==FALSE) {
+ return( FALSE);
+ }
+
+ /* Now, if the file is a regular file, copy it out to the tarball */
+ if (S_ISREG(statbuf->st_mode)) {
+ int inputFileFd;
+ char buffer[BUFSIZ];
+ ssize_t size=0, readSize=0;
+
+ /* open the file we want to archive, and make sure all is well */
+ if ((inputFileFd = open(fileName, O_RDONLY)) < 0) {
+ errorMsg("%s: Cannot open: %s\n", fileName, strerror(errno));
+ return( FALSE);
+ }
+
+ /* write the file to the archive */
+ while ( (size = fullRead(inputFileFd, buffer, sizeof(buffer))) > 0 ) {
+ if (fullWrite(tbInfo->tarFd, buffer, size) != size ) {
+ /* Output file seems to have a problem */
+ errorMsg(io_error, fileName, strerror(errno));
+ return( FALSE);
+ }
+ readSize+=size;
+ }
+ if (size == -1) {
+ errorMsg(io_error, fileName, strerror(errno));
+ return( FALSE);
+ }
+ /* Pad the file up to the tar block size */
+ for (; (readSize%TAR_BLOCK_SIZE) != 0; readSize++) {
+ write(tbInfo->tarFd, "\0", 1);
+ }
+ close( inputFileFd);
+ }
+
+ return( TRUE);
+}
+
+static int writeTarFile(const char* tarName, int tostdoutFlag,
+ int verboseFlag, int argc, char **argv, char** excludeList)
+{
+ int tarFd=-1;
+ int errorFlag=FALSE;
+ ssize_t size;
+ struct TarBallInfo tbInfo;
+ tbInfo.verboseFlag = verboseFlag;
+
+ /* Make sure there is at least one file to tar up. */
+ if (argc <= 0)
+ fatalError("Cowardly refusing to create an empty archive\n");
+
+ /* Open the tar file for writing. */
+ if (tostdoutFlag == TRUE)
+ tbInfo.tarFd = fileno(stdout);
+ else
+ tbInfo.tarFd = open (tarName, O_WRONLY | O_CREAT | O_TRUNC, 0644);
+ if (tbInfo.tarFd < 0) {
+ errorMsg( "Error opening '%s': %s\n", tarName, strerror(errno));
+ return ( FALSE);
+ }
+ tbInfo.excludeList=excludeList;
+ /* Store the stat info for the tarball's file, so
+ * can avoid including the tarball into itself.... */
+ if (fstat(tbInfo.tarFd, &tbInfo.statBuf) < 0)
+ fatalError(io_error, tarName, strerror(errno));
+
+ /* Set the umask for this process so it doesn't
+ * screw up permission setting for us later. */
+ umask(0);
+
+ /* Read the directory/files and iterate over them one at a time */
+ while (argc-- > 0) {
+ if (recursiveAction(*argv++, TRUE, FALSE, FALSE,
+ writeFileToTarball, writeFileToTarball,
+ (void*) &tbInfo) == FALSE) {
+ errorFlag = TRUE;
+ }
+ }
+ /* Write two empty blocks to the end of the archive */
+ for (size=0; size<(2*TAR_BLOCK_SIZE); size++) {
+ write(tbInfo.tarFd, "\0", 1);
+ }
+
+ /* To be pedantically correct, we would check if the tarball
+ * is smaller then 20 tar blocks, and pad it if it was smaller,
+ * but that isn't necessary for GNU tar interoperability, and
+ * so is considered a waste of space */
+
+ /* Hang up the tools, close up shop, head home */
+ close(tarFd);
+ if (errorFlag == TRUE) {
+ errorMsg("Error exit delayed from previous errors\n");
+ return(FALSE);
+ }
+ return( TRUE);
+}
+
+
+#endif
+