/*

 * untgz.c -- Display contents and extract files from a gzip'd TAR file

 *

 * written by Pedro A. Aranda Gutierrez <paag@tid.es></paag@tid.es>

 * adaptation to Unix by Jean-loup Gailly <jloup@gzip.org></jloup@gzip.org>

 * various fixes by Cosmin Truta <cosmint@cs.ubbcluj.ro></cosmint@cs.ubbcluj.ro>

 */

#include <stdio.h></stdio.h>

#include <stdlib.h></stdlib.h>

#include <string.h></string.h>

#include <time.h></time.h>

#include <errno.h></errno.h>

#include "zlib.h"

#ifdef unix

#  include <unistd.h></unistd.h>

#else

#  include <direct.h></direct.h>

#  include <io.h></io.h>

#endif

#ifdef WIN32

#include <windows.h></windows.h>

#  ifndef F_OK

#    define F_OK  0

#  endif

#  define mkdir(dirname,mode)   _mkdir(dirname)

#  ifdef _MSC_VER

#    define access(path,mode)   _access(path,mode)

#    define chmod(path,mode)    _chmod(path,mode)

#    define strdup(str)         _strdup(str)

#  endif

#else

#  include <utime.h></utime.h>

#endif

/* values used in typeflag field */

#define REGTYPE  '0'            /* regular file */

#define AREGTYPE '\0'           /* regular file */

#define LNKTYPE  '1'            /* link */

#define SYMTYPE  '2'            /* reserved */

#define CHRTYPE  '3'            /* character special */

#define BLKTYPE  '4'            /* block special */

#define DIRTYPE  '5'            /* directory */

#define FIFOTYPE '6'            /* FIFO special */

#define CONTTYPE '7'            /* reserved */

/* GNU tar extensions */

#define GNUTYPE_DUMPDIR  'D'    /* file names from dumped directory */

#define GNUTYPE_LONGLINK 'K'    /* long link name */

#define GNUTYPE_LONGNAME 'L'    /* long file name */

#define GNUTYPE_MULTIVOL 'M'    /* continuation of file from another volume */

#define GNUTYPE_NAMES    'N'    /* file name that does not fit into main hdr */

#define GNUTYPE_SPARSE   'S'    /* sparse file */

#define GNUTYPE_VOLHDR   'V'    /* tape/volume header */

/* tar header */

#define BLOCKSIZE     512

#define SHORTNAMESIZE 100

struct tar_header

{                               /* 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 */

                                /* 500 */

};

union tar_buffer

{

  char               buffer[BLOCKSIZE];

  struct tar_header  header;

};

struct attr_item

{

  struct attr_item  *next;

  char              *fname;

  int                mode;

  time_t             time;

};

enum { TGZ_EXTRACT, TGZ_LIST, TGZ_INVALID };

char *TGZfname          OF((const char *));

void TGZnotfound        OF((const char *));

int getoct              OF((char *, int));

char *strtime           OF((time_t *));

int setfiletime         OF((char *, time_t));

void push_attr          OF((struct attr_item **, char *, int, time_t));

void restore_attr       OF((struct attr_item **));

int ExprMatch           OF((char *, char *));

int makedir             OF((char *));

int matchname           OF((int, int, char **, char *));

void error              OF((const char *));

int tar                 OF((gzFile, int, int, int, char **));

void help               OF((int));

int main                OF((int, char **));

char *prog;

const char *TGZsuffix[] = { "\0", ".tar", ".tar.gz", ".taz", ".tgz", NULL };

/* return the file name of the TGZ archive */

/* or NULL if it does not exist */

char *TGZfname (const char *arcname)

{

  static char buffer[1024];

  int origlen,i;

  strcpy(buffer,arcname);

  origlen = strlen(buffer);

  for (i=0; TGZsuffix[i]; i++)

    {

       strcpy(buffer+origlen,TGZsuffix[i]);

       if (access(buffer,F_OK) == 0)

         return buffer;

    }

  return NULL;

}

/* error message for the filename */

void TGZnotfound (const char *arcname)

{

  int i;

  fprintf(stderr,"%s: Couldn't find ",prog);

  for (i=0;TGZsuffix[i];i++)

    fprintf(stderr,(TGZsuffix[i+1]) ? "%s%s, " : "or %s%s\n",

            arcname,

            TGZsuffix[i]);

  exit(1);

}

/* convert octal digits to int */

/* on error return -1 */

int getoct (char *p,int width)

{

  int result = 0;

  char c;

  while (width--)

    {

      c = *p++;

      if (c == 0)

        break;

      if (c == ' ')

        continue;

      if (c < '0' || c > '7')

        return -1;

      result = result * 8 + (c - '0');

    }

  return result;

}

/* convert time_t to string */

/* use the "YYYY/MM/DD hh:mm:ss" format */

char *strtime (time_t *t)

{

  struct tm   *local;

  static char result[32];

  local = localtime(t);

  sprintf(result,"%4d/%02d/%02d %02d:%02d:%02d",

          local->tm_year+1900, local->tm_mon+1, local->tm_mday,

          local->tm_hour, local->tm_min, local->tm_sec);

  return result;

}

/* set file time */

int setfiletime (char *fname,time_t ftime)

{

#ifdef WIN32

  static int isWinNT = -1;

  SYSTEMTIME st;

  FILETIME locft, modft;

  struct tm *loctm;

  HANDLE hFile;

  int result;

  loctm = localtime(&ftime);

  if (loctm == NULL)

    return -1;

  st.wYear         = (WORD)loctm->tm_year + 1900;

  st.wMonth        = (WORD)loctm->tm_mon + 1;

  st.wDayOfWeek    = (WORD)loctm->tm_wday;

  st.wDay          = (WORD)loctm->tm_mday;

  st.wHour         = (WORD)loctm->tm_hour;

  st.wMinute       = (WORD)loctm->tm_min;

  st.wSecond       = (WORD)loctm->tm_sec;

  st.wMilliseconds = 0;

  if (!SystemTimeToFileTime(&st, &locft) ||

      !LocalFileTimeToFileTime(&locft, &modft))

    return -1;

  if (isWinNT < 0)

    isWinNT = (GetVersion() < 0x80000000) ? 1 : 0;

  hFile = CreateFile(fname, GENERIC_WRITE, 0, NULL, OPEN_EXISTING,

                     (isWinNT ? FILE_FLAG_BACKUP_SEMANTICS : 0),

                     NULL);

  if (hFile == INVALID_HANDLE_VALUE)

    return -1;

  result = SetFileTime(hFile, NULL, NULL, &modft) ? 0 : -1;

  CloseHandle(hFile);

  return result;

#else

  struct utimbuf settime;

  settime.actime = settime.modtime = ftime;

  return utime(fname,&settime);

#endif

}

/* push file attributes */

void push_attr(struct attr_item **list,char *fname,int mode,time_t time)

{

  struct attr_item *item;

  item = (struct attr_item *)malloc(sizeof(struct attr_item));

  if (item == NULL)

    error("Out of memory");

  item->fname = strdup(fname);

  item->mode  = mode;

  item->time  = time;

  item->next  = *list;

  *list       = item;

}

/* restore file attributes */

void restore_attr(struct attr_item **list)

{

  struct attr_item *item, *prev;

  for (item = *list; item != NULL; )

    {

      setfiletime(item->fname,item->time);

      chmod(item->fname,item->mode);

      prev = item;

      item = item->next;

      free(prev);

    }

  *list = NULL;

}

/* match regular expression */

#define ISSPECIAL(c) (((c) == '*') || ((c) == '/'))

int ExprMatch (char *string,char *expr)

{

  while (1)

    {

      if (ISSPECIAL(*expr))

        {

          if (*expr == '/')

            {

              if (*string != '\\' && *string != '/')

                return 0;

              string ++; expr++;

            }

          else if (*expr == '*')

            {

              if (*expr ++ == 0)

                return 1;

              while (*++string != *expr)

                if (*string == 0)

                  return 0;

            }

        }

      else

        {

          if (*string != *expr)

            return 0;

          if (*expr++ == 0)

            return 1;

          string++;

        }

    }

}

/* recursive mkdir */

/* abort on ENOENT; ignore other errors like "directory already exists" */

/* return 1 if OK */

/*        0 on error */

int makedir (char *newdir)

{

  char *buffer = strdup(newdir);

  char *p;

  int  len = strlen(buffer);

  if (len <= 0) {

    free(buffer);

    return 0;

  }

  if (buffer[len-1] == '/') {

    buffer[len-1] = '\0';

  }

  if (mkdir(buffer, 0755) == 0)

    {

      free(buffer);

      return 1;

    }

  p = buffer+1;

  while (1)

    {

      char hold;

      while(*p && *p != '\\' && *p != '/')

        p++;

      hold = *p;

      *p = 0;

      if ((mkdir(buffer, 0755) == -1) && (errno == ENOENT))

        {

          fprintf(stderr,"%s: Couldn't create directory %s\n",prog,buffer);

          free(buffer);

          return 0;

        }

      if (hold == 0)

        break;

      *p++ = hold;

    }

  free(buffer);

  return 1;

}

int matchname (int arg,int argc,char **argv,char *fname)

{

  if (arg == argc)      /* no arguments given (untgz tgzarchive) */

    return 1;

  while (arg < argc)

    if (ExprMatch(fname,argv[arg++]))

      return 1;

  return 0; /* ignore this for the moment being */

}

/* tar file list or extract */

int tar (gzFile in,int action,int arg,int argc,char **argv)

{

  union  tar_buffer buffer;

  int    len;

  int    err;

  int    getheader = 1;

  int    remaining = 0;

  FILE   *outfile = NULL;

  char   fname[BLOCKSIZE];

  int    tarmode;

  time_t tartime;

  struct attr_item *attributes = NULL;

  if (action == TGZ_LIST)

    printf("    date      time     size                       file\n"

           " ---------- -------- --------- -------------------------------------\n");

  while (1)

    {

      len = gzread(in, &buffer, BLOCKSIZE);

      if (len < 0)

        error(gzerror(in, &err));

      /*

       * Always expect complete blocks to process

       * the tar information.

       */

      if (len != BLOCKSIZE)

        {

          action = TGZ_INVALID; /* force error exit */

          remaining = 0;        /* force I/O cleanup */

        }

      /*

       * If we have to get a tar header

       */

      if (getheader >= 1)

        {

          /*

           * if we met the end of the tar

           * or the end-of-tar block,

           * we are done

           */

          if (len == 0 || buffer.header.name[0] == 0)

            break;

          tarmode = getoct(buffer.header.mode,8);

          tartime = (time_t)getoct(buffer.header.mtime,12);

          if (tarmode == -1 || tartime == (time_t)-1)

            {

              buffer.header.name[0] = 0;

              action = TGZ_INVALID;

            }

          if (getheader == 1)

            {

              strncpy(fname,buffer.header.name,SHORTNAMESIZE);

              if (fname[SHORTNAMESIZE-1] != 0)

                  fname[SHORTNAMESIZE] = 0;

            }

          else

            {

              /*

               * The file name is longer than SHORTNAMESIZE

               */

              if (strncmp(fname,buffer.header.name,SHORTNAMESIZE-1) != 0)

                  error("bad long name");

              getheader = 1;

            }

          /*

           * Act according to the type flag

           */

          switch (buffer.header.typeflag)

            {

            case DIRTYPE:

              if (action == TGZ_LIST)

                printf(" %s     <dir> %s\n",strtime(&tartime),fname);</dir>

              if (action == TGZ_EXTRACT)

                {

                  makedir(fname);

                  push_attr(&attributes,fname,tarmode,tartime);

                }

              break;

            case REGTYPE:

            case AREGTYPE:

              remaining = getoct(buffer.header.size,12);

              if (remaining == -1)

                {

                  action = TGZ_INVALID;

                  break;

                }

              if (action == TGZ_LIST)

                printf(" %s %9d %s\n",strtime(&tartime),remaining,fname);

              else if (action == TGZ_EXTRACT)

                {

                  if (matchname(arg,argc,argv,fname))

                    {

                      outfile = fopen(fname,"wb");

                      if (outfile == NULL) {

                        /* try creating directory */

                        char *p = strrchr(fname, '/');

                        if (p != NULL) {

                          *p = '\0';

                          makedir(fname);

                          *p = '/';

                          outfile = fopen(fname,"wb");

                        }

                      }

                      if (outfile != NULL)

                        printf("Extracting %s\n",fname);

                      else

                        fprintf(stderr, "%s: Couldn't create %s",prog,fname);

                    }

                  else

                    outfile = NULL;

                }

              getheader = 0;

              break;

            case GNUTYPE_LONGLINK:

            case GNUTYPE_LONGNAME:

              remaining = getoct(buffer.header.size,12);

              if (remaining < 0 || remaining >= BLOCKSIZE)

                {

                  action = TGZ_INVALID;

                  break;

                }

              len = gzread(in, fname, BLOCKSIZE);

              if (len < 0)

                error(gzerror(in, &err));

              if (fname[BLOCKSIZE-1] != 0 || (int)strlen(fname) > remaining)

                {

                  action = TGZ_INVALID;

                  break;

                }

              getheader = 2;

              break;

            default:

              if (action == TGZ_LIST)

                printf(" %s     <---> %s\n",strtime(&tartime),fname);

              break;

            }

        }

      else

        {

          unsigned int bytes = (remaining > BLOCKSIZE) ? BLOCKSIZE : remaining;

          if (outfile != NULL)

            {

              if (fwrite(&buffer,sizeof(char),bytes,outfile) != bytes)

                {

                  fprintf(stderr,

                    "%s: Error writing %s -- skipping\n",prog,fname);

                  fclose(outfile);

                  outfile = NULL;

                  remove(fname);

                }

            }

          remaining -= bytes;

        }

      if (remaining == 0)

        {

          getheader = 1;

          if (outfile != NULL)

            {

              fclose(outfile);

              outfile = NULL;

              if (action != TGZ_INVALID)

                push_attr(&attributes,fname,tarmode,tartime);

            }

        }

      /*

       * Abandon if errors are found

       */

      if (action == TGZ_INVALID)

        {

          error("broken archive");

          break;

        }

    }

  /*

   * Restore file modes and time stamps

   */

  restore_attr(&attributes);

  if (gzclose(in) != Z_OK)

    error("failed gzclose");

  return 0;

}

/* ============================================================ */

void help(int exitval)

{

  printf("untgz version 0.2.1\n"

         "  using zlib version %s\n\n",

         zlibVersion());

  printf("Usage: untgz file.tgz            extract all files\n"

         "       untgz file.tgz fname ...  extract selected files\n"

         "       untgz -l file.tgz         list archive contents\n"

         "       untgz -h                  display this help\n");

  exit(exitval);

}

void error(const char *msg)

{

  fprintf(stderr, "%s: %s\n", prog, msg);

  exit(1);

}

/* ============================================================ */

#if defined(WIN32) && defined(__GNUC__)

int _CRT_glob = 0;      /* disable argument globbing in MinGW */

#endif

int main(int argc,char **argv)

{

    int         action = TGZ_EXTRACT;

    int         arg = 1;

    char        *TGZfile;

    gzFile      *f;

    prog = strrchr(argv[0],'\\');

    if (prog == NULL)

      {

        prog = strrchr(argv[0],'/');

        if (prog == NULL)

          {

            prog = strrchr(argv[0],':');

            if (prog == NULL)

              prog = argv[0];

            else

              prog++;

          }

        else

          prog++;

      }

    else

      prog++;

    if (argc == 1)

      help(0);

    if (strcmp(argv[arg],"-l") == 0)

      {

        action = TGZ_LIST;

        if (argc == ++arg)

          help(0);

      }

    else if (strcmp(argv[arg],"-h") == 0)

      {

        help(0);

      }

    if ((TGZfile = TGZfname(argv[arg])) == NULL)

      TGZnotfound(argv[arg]);

    ++arg;

    if ((action == TGZ_LIST) && (arg != argc))

      help(1);

/*

 *  Process the TGZ file

 */

    switch(action)

      {

      case TGZ_LIST:

      case TGZ_EXTRACT:

        f = gzopen(TGZfile,"rb");

        if (f == NULL)

          {

            fprintf(stderr,"%s: Couldn't gzopen %s\n",prog,TGZfile);

            return 1;

          }

        exit(tar(f, action, arg, argc, argv));

      break;

      default:

        error("Unknown option");

        exit(1);

      }

    return 0;

}