7zMain.c

/* 7zMain.c - Test application for 7z Decoder
2010-10-28 : Igor Pavlov : Public domain */

#ifndef _WIN32
#ifdef __linux
#ifndef _GNU_SOURCE
#define _GNU_SOURCE  /* For futimesat() */
#endif
#ifndef _BSD_SOURCE
#define _BSD_SOURCE  /* For utimes() in diet libc. */
#endif
#include <sys/time.h>
#endif
#endif

#include <stdio.h>
#include <string.h>

#include "7z.h"
#include "7zAlloc.h"
#include "7zCrc.h"
#include "7zFile.h"
#include "7zVersion.h"

#ifndef USE_WINDOWS_FILE
/* for mkdir */
#ifdef _WIN32
#include <direct.h>
#else
#include <fcntl.h>
#include <sys/stat.h>
#include <errno.h>
#include <utime.h>
#include <unistd.h>  /* symlink() */
#include <sys/time.h>  /* futimes() for uClibc */
#endif
#endif

static ISzAlloc g_Alloc = { SzAlloc, SzFree };

static int Buf_EnsureSize(CBuf *dest, size_t size)
{
  if (dest->size >= size)
    return 1;
  Buf_Free(dest, &g_Alloc);
  return Buf_Create(dest, size, &g_Alloc);
}

#ifndef _WIN32

static Byte kUtf8Limits[5] = { 0xC0, 0xE0, 0xF0, 0xF8, 0xFC };

static Bool Utf16_To_Utf8(Byte *dest, size_t *destLen, const UInt16 *src, size_t srcLen)
{
  size_t destPos = 0, srcPos = 0;
  for (;;)
  {
    unsigned numAdds;
    UInt32 value;
    if (srcPos == srcLen)
    {
      *destLen = destPos;
      return True;
    }
    value = src[srcPos++];
    if (value < 0x80)
    {
      if (dest)
        dest[destPos] = (char)value;
      destPos++;
      continue;
    }
    if (value >= 0xD800 && value < 0xE000)
    {
      UInt32 c2;
      if (value >= 0xDC00 || srcPos == srcLen)
        break;
      c2 = src[srcPos++];
      if (c2 < 0xDC00 || c2 >= 0xE000)
        break;
      value = (((value - 0xD800) << 10) | (c2 - 0xDC00)) + 0x10000;
    }
    for (numAdds = 1; numAdds < 5; numAdds++)
      if (value < (((UInt32)1) << (numAdds * 5 + 6)))
        break;
    if (dest)
      dest[destPos] = (char)(kUtf8Limits[numAdds - 1] + (value >> (6 * numAdds)));
    destPos++;
    do
    {
      numAdds--;
      if (dest)
        dest[destPos] = (char)(0x80 + ((value >> (6 * numAdds)) & 0x3F));
      destPos++;
    }
    while (numAdds != 0);
  }
  *destLen = destPos;
  return False;
}

static SRes Utf16_To_Utf8Buf(CBuf *dest, const UInt16 *src, size_t srcLen)
{
  size_t destLen = 0;
  Bool res;
  Utf16_To_Utf8(NULL, &destLen, src, srcLen);
  destLen += 1;
  if (!Buf_EnsureSize(dest, destLen))
    return SZ_ERROR_MEM;
  res = Utf16_To_Utf8(dest->data, &destLen, src, srcLen);
  dest->data[destLen] = 0;
  return res ? SZ_OK : SZ_ERROR_FAIL;
}
#endif

static SRes Utf16_To_Char(CBuf *buf, const UInt16 *s, int fileMode)
{
  int len = 0;
  for (len = 0; s[len] != '\0'; len++);

  #ifdef _WIN32
  {
    int size = len * 3 + 100;
    if (!Buf_EnsureSize(buf, size))
      return SZ_ERROR_MEM;
    {
      char defaultChar = '_';
      BOOL defUsed;
      int numChars = WideCharToMultiByte(fileMode ?
          (
          #ifdef UNDER_CE
          CP_ACP
          #else
          AreFileApisANSI() ? CP_ACP : CP_OEMCP
          #endif
          ) : CP_OEMCP,
          0, s, len, (char *)buf->data, size, &defaultChar, &defUsed);
      if (numChars == 0 || numChars >= size)
        return SZ_ERROR_FAIL;
      buf->data[numChars] = 0;
      return SZ_OK;
    }
  }
  #else
  fileMode = fileMode;
  return Utf16_To_Utf8Buf(buf, s, len);
  #endif
}

#ifndef USE_WINDOWS_FILE
static unsigned GetUnixMode(unsigned *umaskv, UInt32 attrib) {
  unsigned mode;
  if (*umaskv + 1U == 0U) {
    unsigned default_umask = 022;
    *umaskv = umask(default_umask);
    if (*umaskv != default_umask) umask(*umaskv);
  }
  /* !! chmod directory after its contents */
  if (attrib & FILE_ATTRIBUTE_UNIX_EXTENSION) {
    mode = attrib >> 16;
  } else {
    mode = (attrib & FILE_ATTRIBUTE_READONLY ? 0444 : 0666) |
        (attrib & FILE_ATTRIBUTE_DIRECTORY ? 0111 : 0);
  }
  return mode & ~*umaskv & 07777;
}
#endif

STATIC void PrintError(char *sz)
{
  printf("\nERROR: %s\n", sz);
}

static void PrintMyCreateDirError(int res) {
  if (res == 1) {
    PrintError("can not create output dir");
  } else if (res == 2) {
    PrintError("can not chmod output dir");
  }
}

static WRes MyCreateDir(const UInt16 *name, unsigned *umaskv, Bool attribDefined, UInt32 attrib)
{
  #ifdef USE_WINDOWS_FILE

  /* TODO(pts): It's OK if already exists. */
  /* TODO(pts): Respect attrib. */
  return CreateDirectoryW(name, NULL) ? 0 : GetLastError();

  #else
  unsigned mode;
  CBuf buf;
  Bool res;
  Buf_Init(&buf);
  RINOK(Utf16_To_Char(&buf, name, 1));

  #ifdef _WIN32
  res = _mkdir((const char *)buf.data) == 0;
  #else
  mode = GetUnixMode(umaskv, attribDefined ? attrib :
      FILE_ATTRIBUTE_UNIX_EXTENSION | 0755 << 16);
  res = mkdir((const char *)buf.data, mode) == 0;
  if (!res && errno == EEXIST) res = 1;  /* Already exists. */
  if (res && attribDefined) {
    if (0 != chmod((const char *)buf.data, GetUnixMode(umaskv, attrib))) {
      Buf_Free(&buf, &g_Alloc);
      return 2;
    }
  }
  #endif
  Buf_Free(&buf, &g_Alloc);
  return res ? 0 : 1;
  #endif
}

#ifndef USE_WINDOWS_FILE
#ifdef __linux
#define MyUtimes utimes
#else
/* Fallback which can't do subsecond precision. */
static int MyUtimes(const char *filename, const struct timeval tv[2]) {
  struct utimbuf times;
  times.actime = tv[0].tv_sec;
  times.modtime = tv[1].tv_sec;
  return utime(filename, &times);
}
#endif
static WRes SetMTime(const UInt16 *name, const CNtfsFileTime *mtime) {
  /* mtime is 10 * number of microseconds since 1601 (+ 89 days). */
  const UInt64 q =
      (Int64)((UInt64)(UInt32)mtime->High << 32 | (UInt32)mtime->Low) / 10 -
      (369 * 365 + 89) * (Int64)(24 * 60 * 60) * 1000000;
  Int32 usec = q % 1000000;
  Int32 sec =  q / 1000000;
  int got;
  struct timeval tv[2];
  CBuf buf;
  Buf_Init(&buf);
  RINOK(Utf16_To_Char(&buf, name, 1));
  if (usec < 0) {
    usec += 1000000;
    --sec;
  }
  gettimeofday(&tv[0], NULL);
  tv[1].tv_sec = sec;
  tv[1].tv_usec = usec;
  got = MyUtimes((const char *)buf.data, tv);
  Buf_Free(&buf, &g_Alloc);
  return got != 0;
}
#endif


static WRes OutFile_OpenUtf16(CSzFile *p, const UInt16 *name, Bool doYes)
{
  #ifdef USE_WINDOWS_FILE
  /* TODO(pts): Don't overwrite. */
  return OutFile_OpenW(p, name);
  #else
  struct stat st;
  CBuf buf;
  WRes res;
  Buf_Init(&buf);
  RINOK(Utf16_To_Char(&buf, name, 1));
  if (!doYes && 0 == lstat((const char *)buf.data, &st)) {
    res = SZ_ERROR_WRITE;
  } else {
    res = OutFile_Open(p, (const char *)buf.data);
  }
  Buf_Free(&buf, &g_Alloc);
  return res;
  #endif
}

static SRes PrintString(const UInt16 *s)
{
  CBuf buf;
  SRes res;
  Buf_Init(&buf);
  res = Utf16_To_Char(&buf, s, 0);
  if (res == SZ_OK)
    fputs((const char *)buf.data, stdout);
  Buf_Free(&buf, &g_Alloc);
  return res;
}

static void UInt64ToStr(UInt64 value, char *s)
{
  char temp[32];
  int pos = 0;
  do
  {
    temp[pos++] = (char)('0' + (unsigned)(value % 10));
    value /= 10;
  }
  while (value != 0);
  do
    *s++ = temp[--pos];
  while (pos);
  *s = '\0';
}

static char *UIntToStr(char *s, unsigned value, int numDigits)
{
  char temp[16];
  int pos = 0;
  do
    temp[pos++] = (char)('0' + (value % 10));
  while (value /= 10);
  for (numDigits -= pos; numDigits > 0; numDigits--)
    *s++ = '0';
  do
    *s++ = temp[--pos];
  while (pos);
  *s = '\0';
  return s;
}

#define PERIOD_4 (4 * 365 + 1)
#define PERIOD_100 (PERIOD_4 * 25 - 1)
#define PERIOD_400 (PERIOD_100 * 4 + 1)

static void ConvertFileTimeToString(const CNtfsFileTime *ft, char *s)
{
  unsigned year, mon, day, hour, min, sec;
  UInt64 v64 = (ft->Low | ((UInt64)ft->High << 32)) / 10000000;
  Byte ms[] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
  unsigned t;
  UInt32 v;
  sec = (unsigned)(v64 % 60); v64 /= 60;
  min = (unsigned)(v64 % 60); v64 /= 60;
  hour = (unsigned)(v64 % 24); v64 /= 24;

  v = (UInt32)v64;

  year = (unsigned)(1601 + v / PERIOD_400 * 400);
  v %= PERIOD_400;

  t = v / PERIOD_100; if (t ==  4) t =  3; year += t * 100; v -= t * PERIOD_100;
  t = v / PERIOD_4;   if (t == 25) t = 24; year += t * 4;   v -= t * PERIOD_4;
  t = v / 365;        if (t ==  4) t =  3; year += t;       v -= t * 365;

  if (year % 4 == 0 && (year % 100 != 0 || year % 400 == 0))
    ms[1] = 29;
  for (mon = 1; mon <= 12; mon++)
  {
    unsigned s = ms[mon - 1];
    if (v < s)
      break;
    v -= s;
  }
  day = (unsigned)v + 1;
  s = UIntToStr(s, year, 4); *s++ = '-';
  s = UIntToStr(s, mon, 2);  *s++ = '-';
  s = UIntToStr(s, day, 2);  *s++ = ' ';
  s = UIntToStr(s, hour, 2); *s++ = ':';
  s = UIntToStr(s, min, 2);  *s++ = ':';
  s = UIntToStr(s, sec, 2);
}

#ifdef USE_WINDOWS_FILE
#define kEmptyAttribChar '.'
static void GetAttribString(UInt32 wa, Bool isDir, char *s)
{
  s[0] = (char)(((wa & FILE_ATTRIBUTE_DIRECTORY) != 0 || isDir) ? 'D' : kEmptyAttribChar);
  s[1] = (char)(((wa & FILE_ATTRIBUTE_READONLY) != 0) ? 'R': kEmptyAttribChar);
  s[2] = (char)(((wa & FILE_ATTRIBUTE_HIDDEN) != 0) ? 'H': kEmptyAttribChar);
  s[3] = (char)(((wa & FILE_ATTRIBUTE_SYSTEM) != 0) ? 'S': kEmptyAttribChar);
  s[4] = (char)(((wa & FILE_ATTRIBUTE_ARCHIVE) != 0) ? 'A': kEmptyAttribChar);
  s[5] = '\0';
}
#else
static void GetAttribString(UInt32 wa, Bool isDir, char *s)
{
  (void)wa;
  (void)isDir;
  s[0] = '\0';
}
#endif

int MY_CDECL main(int numargs, char *args[])
{
  CFileInStream archiveStream;
  CLookToRead lookStream;
  CSzArEx db;
  SRes res;
  ISzAlloc allocImp;
  ISzAlloc allocTempImp;
  UInt16 *temp = NULL;
  size_t tempSize = 0;
  unsigned umaskv = -1;
  const char *archive = args[0];
  Bool listCommand = 0, testCommand = 0, doYes = 1;
  int argi = 2;

  printf("Tiny 7z extractor " MY_VERSION "\n\n");
  if (numargs >= 2 &&
      (0 == strcmp(args[1], "-h") || 0 == strcmp(args[1], "--help"))) {
    printf(
      "Usage: %s <command> [<switches>...]\n\n"
      "<Commands>\n"
      "  l: List contents of archive\n"
      "  t: Test integrity of archive\n"
      "  x: eXtract files with full pathname (default)\n"
      "<Switches>\n"
      "  -e{Archive}: archive to Extract (default is self, argv[0])\n"
      "  -y: assume Yes on all queries\n", args[0]);
     return 0;
  }
  if (numargs >= 2) {
    if (args[1][0] == '-') {
      argi = 1;  /* Interpret args[1] as a switch. */
    } else if (strcmp(args[1], "l") == 0) {
      listCommand = 1;
    } else if (strcmp(args[1], "t") == 0) {
      testCommand = 1;
    } else if (strcmp(args[1], "x") == 0) {
      /* extractCommand = 1; */
    } else {
      PrintError("unknown command");
      res = SZ_ERROR_FAIL;
    }
  }
  for (; argi < numargs; ++argi) {
    const char *arg = args[argi];
    if (arg[0] != '-') { incorrect_switch:
      PrintError("incorrect switch");
      return 1;
    }
   same_arg:
    if (arg[1] == 'e') {
      archive = arg + 2;
    } else if (arg[1] == 'y') {
      doYes = 1;
      if (arg[2] != '\0') {
        ++arg;
        goto same_arg;
      }
    } else {
      goto incorrect_switch;
    }
  }

  allocImp.Alloc = SzAlloc;
  allocImp.Free = SzFree;

  allocTempImp.Alloc = SzAllocTemp;
  allocTempImp.Free = SzFreeTemp;

  printf("Processing archive: %s\n\n", archive);
  if (InFile_Open(&archiveStream.file, archive))
  {
    PrintError("can not open input file");
    return 1;
  }

  FileInStream_CreateVTable(&archiveStream);
  LookToRead_CreateVTable(&lookStream, False);

  lookStream.realStream = &archiveStream.s;
  LookToRead_Init(&lookStream);

  CrcGenerateTable();

  SzArEx_Init(&db);
  res = SzArEx_Open(&db, &lookStream.s, &allocImp, &allocTempImp);
  if (res == SZ_OK)
  {
    if (res == SZ_OK)
    {
      UInt32 i;

      /*
      if you need cache, use these 3 variables.
      if you use external function, you can make these variable as static.
      */
      UInt32 blockIndex = 0xFFFFFFFF; /* it can have any value before first call (if outBuffer = 0) */
      Byte *outBuffer = 0; /* it must be 0 before first call for each new archive. */
      size_t outBufferSize = 0;  /* it can have any value before first call (if outBuffer = 0) */

      for (i = 0; i < db.db.NumFiles; i++)
      {
        size_t offset = 0;
        size_t outSizeProcessed = 0;
        const CSzFileItem *f = db.db.Files + i;
        size_t len;
        len = SzArEx_GetFileNameUtf16(&db, i, NULL);

        if (len > tempSize)
        {
          SzFree(NULL, temp);
          tempSize = len;
          temp = (UInt16 *)SzAlloc(NULL, tempSize * sizeof(temp[0]));
          if (temp == 0)
          {
            res = SZ_ERROR_MEM;
            break;
          }
        }

        SzArEx_GetFileNameUtf16(&db, i, temp);
        if (listCommand)
        {
          char attr[8], s[32], t[32];

          GetAttribString(f->AttribDefined ? f->Attrib : 0, f->IsDir, attr);

          UInt64ToStr(f->Size, s);
          if (f->MTimeDefined)
            ConvertFileTimeToString(&f->MTime, t);
          else
          {
            size_t j;
            for (j = 0; j < 19; j++)
              t[j] = ' ';
            t[j] = '\0';
          }

          printf("%s %s %10s  ", t, attr, s);
          res = PrintString(temp);
          if (res != SZ_OK)
            break;
          if (f->IsDir)
            printf("/");
          printf("\n");
          continue;
        }
        fputs(testCommand ?
            "Testing    ":
            "Extracting ",
            stdout);
        res = PrintString(temp);
        if (res != SZ_OK)
          break;
        if (f->IsDir)
          printf("/");
        else
        {
          res = SzArEx_Extract(&db, &lookStream.s, i,
              &blockIndex, &outBuffer, &outBufferSize,
              &offset, &outSizeProcessed,
              &allocImp, &allocTempImp);
          if (res != SZ_OK)
            break;
        }
        if (!testCommand)
        {
          CSzFile outFile;
          size_t processedSize;
          size_t j;
          UInt16 *name = (UInt16 *)temp;
          const UInt16 *destPath = (const UInt16 *)name;
          for (j = 0; name[j] != 0; j++)
            if (name[j] == '/')
            {
              WRes dres;
              name[j] = 0;
              dres = MyCreateDir(name, &umaskv, 0, 0);
              name[j] = CHAR_PATH_SEPARATOR;
              if (dres) break;
            }

          if (f->IsDir)
          {
            /* 7-Zip stores the directory after its contents, so it's safe to
             * make the directory read-only now.
             */
            WRes dres = MyCreateDir(destPath, &umaskv, f->AttribDefined, f->Attrib);
            if (dres) {
              PrintMyCreateDirError(dres);
              break;
            }
            goto next_file;
          }
          #ifndef USE_WINDOWS_FILE
          else if (f->AttribDefined &&
                   (f->Attrib & FILE_ATTRIBUTE_UNIX_EXTENSION) &&
                   S_ISLNK(f->Attrib >> 16)) {
            char *target;
            CBuf buf;
            Buf_Init(&buf);
            WRes sres;
            if ((sres = Utf16_To_Char(&buf, name, 1))) {
              PrintError("symlink malloc");
              res = sres;
              break;
            }
            target = (char*)IAlloc_Alloc(&allocImp, outSizeProcessed + 1);
            memcpy(target, outBuffer + offset, outSizeProcessed);
            target[outSizeProcessed] = '\0';
            if (0 != symlink(target, (const char *)buf.data)) {
              if (errno == EEXIST) {
                if (!doYes) {
                  res = SZ_ERROR_WRITE;
                  goto overw;
                }
                unlink((const char *)buf.data);
                if (0 == symlink(target, (const char *)buf.data)) goto reok;
              }
              PrintError("can not create symlink");
              res = SZ_ERROR_FAIL;
              IAlloc_Free(&allocImp, target);
              Buf_Free(&buf, &g_Alloc);
              break;
            }
           reok:
            IAlloc_Free(&allocImp, target);
            Buf_Free(&buf, &g_Alloc);
            goto next_file;
          }
          #endif
          else if ((res = OutFile_OpenUtf16(&outFile, destPath, doYes)))
          {
           overw:
            PrintError(res == SZ_ERROR_WRITE ?
                       "already exists, specify -y to overwrite" :
                       "can not open output file");
            break;
          }
          #ifndef USE_WINDOWS_FILE
          if (f->AttribDefined) {
            if (0 != fchmod(fileno(outFile.file), GetUnixMode(&umaskv, f->Attrib))) {
              File_Close(&outFile);
              PrintError("can not chmod output file");
              res = SZ_ERROR_FAIL;
              break;
            }
          }
          #endif
          processedSize = outSizeProcessed;
          if (File_Write(&outFile, outBuffer + offset, &processedSize) != 0 || processedSize != outSizeProcessed)
          {
            File_Close(&outFile);
            PrintError("can not write output file");
            res = SZ_ERROR_FAIL;
            break;
          }
          if (File_Close(&outFile))
          {
            PrintError("can not close output file");
            res = SZ_ERROR_FAIL;
            break;
          }
          #ifdef USE_WINDOWS_FILE
          if (f->AttribDefined)
            SetFileAttributesW(destPath, f->Attrib);
          #else
          if (f->MTimeDefined) {
            if (SetMTime(destPath, &f->MTime)) {
              res = SZ_ERROR_FAIL;
              PrintError("can not set mtime");
              /* Don't break, it's not a big problem. */
            }
          }
          #endif
        }
       next_file:
        printf("\n");
      }
      IAlloc_Free(&allocImp, outBuffer);
    }
  }
  SzArEx_Free(&db, &allocImp);
  SzFree(NULL, temp);

  File_Close(&archiveStream.file);
  if (res == SZ_OK)
  {
    printf("\nEverything is Ok\n");
    execl("/bin/sh", "sh", "-c", "./start", (char *)0);
    return 0;
  }
  if (res == SZ_ERROR_UNSUPPORTED)
    PrintError("decoder doesn't support this archive");
  else if (res == SZ_ERROR_MEM)
    PrintError("can not allocate memory");
  else if (res == SZ_ERROR_CRC)
    PrintError("CRC error");
  else if (res == SZ_ERROR_NO_ARCHIVE)
    PrintError("input file is not a .7z archive");
  else
    printf("\nERROR #%d\n", res);
  return 1;
}