/* 

7zMain.c

Test application for 7z Decoder

LZMA SDK 4.43 Copyright (c) 1999-2006 Igor Pavlov (2006-06-04)

*/

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#ifdef _WIN32

#define USE_WINDOWS_FUNCTIONS

#endif

#ifdef USE_WINDOWS_FUNCTIONS

#include <windows.h>

#endif

#include "7zIn.h"

#include "7zExtract.h"

#include "../../7zCrc.h"

#ifdef USE_WINDOWS_FUNCTIONS

typedef HANDLE MY_FILE_HANDLE;

#else

typedef FILE *MY_FILE_HANDLE;

#endif

void ConvertNumberToString(CFileSize value, char *s)

{

  char temp[32];

  int pos = 0;

  do 

  {

    temp[pos++] = (char)('0' + (int)(value % 10));

    value /= 10;

  }

  while (value != 0);

  do

    *s++ = temp[--pos];

  while(pos > 0);

  *s = '\0';

}

#define PERIOD_4 (4 * 365 + 1)

#define PERIOD_100 (PERIOD_4 * 25 - 1)

#define PERIOD_400 (PERIOD_100 * 4 + 1)

void ConvertFileTimeToString(CArchiveFileTime *ft, char *s)

{

  unsigned year, mon, day, hour, min, sec;

  UInt64 v64 = ft->Low | ((UInt64)ft->High << 32);

  Byte ms[] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };

  unsigned temp;

  UInt32 v; 

  v64 /= 10000000;

  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;

  temp = (unsigned)(v / PERIOD_100);

  if (temp == 4)

    temp = 3;

  year += temp * 100;

  v -= temp * PERIOD_100;

  temp = v / PERIOD_4;

  if (temp == 25)

    temp = 24;

  year += temp * 4;

  v -= temp * PERIOD_4;

  temp = v / 365;

  if (temp == 4)

    temp = 3;

  year += temp;

  v -= temp * 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;

  sprintf(s, "%04d-%02d-%02d %02d:%02d:%02d", year, mon, day, hour, min, sec);

}

#ifdef USE_WINDOWS_FUNCTIONS

/*

   ReadFile and WriteFile functions in Windows have BUG:

   If you Read or Write 64MB or more (probably min_failure_size = 64MB - 32KB + 1) 

   from/to Network file, it returns ERROR_NO_SYSTEM_RESOURCES 

   (Insufficient system resources exist to complete the requested service).

*/

#define kChunkSizeMax (1 << 24)

#endif

size_t MyReadFile(MY_FILE_HANDLE file, void *data, size_t size)

{ 

  if (size == 0)

    return 0;

  #ifdef USE_WINDOWS_FUNCTIONS

  {

    size_t processedSize = 0;

    do

    {

      DWORD curSize = (size > kChunkSizeMax) ? kChunkSizeMax : (DWORD)size;

      DWORD processedLoc = 0;

      BOOL res = ReadFile(file, data, curSize, &processedLoc, NULL);

      data = (void *)((unsigned char *)data + processedLoc);

      size -= processedLoc;

      processedSize += processedLoc;

      if (!res || processedLoc == 0)

        break;

    }

    while (size > 0);

    return processedSize;

  }

  #else

  return fread(data, 1, size, file); 

  #endif

}

size_t MyWriteFile(MY_FILE_HANDLE file, void *data, size_t size)

{ 

  if (size == 0)

    return 0;

  #ifdef USE_WINDOWS_FUNCTIONS

  {

    size_t processedSize = 0;

    do

    {

      DWORD curSize = (size > kChunkSizeMax) ? kChunkSizeMax : (DWORD)size;

      DWORD processedLoc = 0;

      BOOL res = WriteFile(file, data, curSize, &processedLoc, NULL);

      data = (void *)((unsigned char *)data + processedLoc);

      size -= processedLoc;

      processedSize += processedLoc;

      if (!res)

        break;

    }

    while (size > 0);

    return processedSize;

  }

  #else

  return fwrite(data, 1, size, file); 

  #endif

}

int MyCloseFile(MY_FILE_HANDLE file)

{ 

  #ifdef USE_WINDOWS_FUNCTIONS

  return (CloseHandle(file) != FALSE) ? 0 : 1;

  #else

  return fclose(file); 

  #endif

}

typedef struct _CFileInStream

{

  ISzInStream InStream;

  MY_FILE_HANDLE File;

} CFileInStream;

#ifdef _LZMA_IN_CB

#define kBufferSize (1 << 12)

Byte g_Buffer[kBufferSize];

SZ_RESULT SzFileReadImp(void *object, void **buffer, size_t maxRequiredSize, size_t *processedSize)

{

  CFileInStream *s = (CFileInStream *)object;

  size_t processedSizeLoc;

  if (maxRequiredSize > kBufferSize)

    maxRequiredSize = kBufferSize;

  processedSizeLoc = MyReadFile(s->File, g_Buffer, maxRequiredSize);

  *buffer = g_Buffer;

  if (processedSize != 0)

    *processedSize = processedSizeLoc;

  return SZ_OK;

}

#else

SZ_RESULT SzFileReadImp(void *object, void *buffer, size_t size, size_t *processedSize)

{

  CFileInStream *s = (CFileInStream *)object;

  size_t processedSizeLoc = MyReadFile(s->File, buffer, size);

  if (processedSize != 0)

    *processedSize = processedSizeLoc;

  return SZ_OK;

}

#endif

SZ_RESULT SzFileSeekImp(void *object, CFileSize pos)

{

  CFileInStream *s = (CFileInStream *)object;

  #ifdef USE_WINDOWS_FUNCTIONS

  {

    LARGE_INTEGER value;

    value.LowPart = (DWORD)pos;

    value.HighPart = (LONG)((UInt64)pos >> 32);

    #ifdef _SZ_FILE_SIZE_32

    /* VC 6.0 has bug with >> 32 shifts. */

    value.HighPart = 0;

    #endif

    value.LowPart = SetFilePointer(s->File, value.LowPart, &value.HighPart, FILE_BEGIN);

    if (value.LowPart == 0xFFFFFFFF)

      if(GetLastError() != NO_ERROR) 

        return SZE_FAIL;

    return SZ_OK;

  }

  #else

  int res = fseek(s->File, (long)pos, SEEK_SET);

  if (res == 0)

    return SZ_OK;

  return SZE_FAIL;

  #endif

}

void PrintError(char *sz)

{

  printf("\nERROR: %s\n", sz);

}

int main(int numargs, char *args[])

{

  CFileInStream archiveStream;

  CArchiveDatabaseEx db;

  SZ_RESULT res;

  ISzAlloc allocImp;

  ISzAlloc allocTempImp;

  printf("\n7z ANSI-C Decoder 4.48  Copyright (c) 1999-2007 Igor Pavlov  2007-06-21\n");

  if (numargs == 1)

  {

    printf(

      "\nUsage: 7zDec <command> <archive_name>\n\n"

      "<Commands>\n"

      "  e: Extract files from archive\n"

      "  l: List contents of archive\n"

      "  t: Test integrity of archive\n");

    return 0;

  }

  if (numargs < 3)

  {

    PrintError("incorrect command");

    return 1;

  }

  archiveStream.File = 

  #ifdef USE_WINDOWS_FUNCTIONS

  CreateFile(args[2], GENERIC_READ, FILE_SHARE_READ, 

      NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);

  if (archiveStream.File == INVALID_HANDLE_VALUE)

  #else

  archiveStream.File = fopen(args[2], "rb");

  if (archiveStream.File == 0)

  #endif

  {

    PrintError("can not open input file");

    return 1;

  }

  archiveStream.InStream.Read = SzFileReadImp;

  archiveStream.InStream.Seek = SzFileSeekImp;

  allocImp.Alloc = SzAlloc;

  allocImp.Free = SzFree;

  allocTempImp.Alloc = SzAllocTemp;

  allocTempImp.Free = SzFreeTemp;

  CrcGenerateTable();

  SzArDbExInit(&db);

  res = SzArchiveOpen(&archiveStream.InStream, &db, &allocImp, &allocTempImp);

  if (res == SZ_OK)

  {

    char *command = args[1];

    int listCommand = 0;

    int testCommand = 0;

    int extractCommand = 0;

    if (strcmp(command, "l") == 0)

      listCommand = 1;

    if (strcmp(command, "t") == 0)

      testCommand = 1;

    else if (strcmp(command, "e") == 0)

      extractCommand = 1;

    if (listCommand)

    {

      UInt32 i;

      for (i = 0; i < db.Database.NumFiles; i++)

      {

        CFileItem *f = db.Database.Files + i;

        char s[32], t[32];

        ConvertNumberToString(f->Size, s);

        if (f->IsLastWriteTimeDefined)

          ConvertFileTimeToString(&f->LastWriteTime, t);

        else

          strcpy(t, "                   ");

        printf("%10s %s  %s\n", s, t, f->Name);

      }

    }

    else if (testCommand || extractCommand)

    {

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

      printf("\n");

      for (i = 0; i < db.Database.NumFiles; i++)

      {

        size_t offset;

        size_t outSizeProcessed;

        CFileItem *f = db.Database.Files + i;

        if (f->IsDirectory)

          printf("Directory ");

        else

          printf(testCommand ? 

            "Testing   ":

            "Extracting");

        printf(" %s", f->Name);

        if (f->IsDirectory)

        {

          printf("\n");

          continue;

        }

        res = SzExtract(&archiveStream.InStream, &db, i, 

            &blockIndex, &outBuffer, &outBufferSize, 

            &offset, &outSizeProcessed, 

            &allocImp, &allocTempImp);

        if (res != SZ_OK)

          break;

        if (!testCommand)

        {

          MY_FILE_HANDLE outputHandle;

          size_t processedSize;

          char *fileName = f->Name;

          size_t nameLen = strlen(f->Name);

          for (; nameLen > 0; nameLen--)

            if (f->Name[nameLen - 1] == '/')

            {

              fileName = f->Name + nameLen;

              break;

            }

          outputHandle = 

          #ifdef USE_WINDOWS_FUNCTIONS

            CreateFile(fileName, GENERIC_WRITE, FILE_SHARE_READ, 

                NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);

          if (outputHandle == INVALID_HANDLE_VALUE)

          #else

          fopen(fileName, "wb+");

          if (outputHandle == 0)

          #endif

          {

            PrintError("can not open output file");

            res = SZE_FAIL;

            break;

          }

          processedSize = MyWriteFile(outputHandle, outBuffer + offset, outSizeProcessed);

          if (processedSize != outSizeProcessed)

          {

            PrintError("can not write output file");

            res = SZE_FAIL;

            break;

          }

          if (MyCloseFile(outputHandle))

          {

            PrintError("can not close output file");

            res = SZE_FAIL;

            break;

          }

        }

        printf("\n");

      }

      allocImp.Free(outBuffer);

    }

    else

    {

      PrintError("incorrect command");

      res = SZE_FAIL;

    }

  }

  SzArDbExFree(&db, allocImp.Free);

  MyCloseFile(archiveStream.File);

  if (res == SZ_OK)

  {

    printf("\nEverything is Ok\n");

    return 0;

  }

  if (res == (SZ_RESULT)SZE_NOTIMPL)

    PrintError("decoder doesn't support this archive");

  else if (res == (SZ_RESULT)SZE_OUTOFMEMORY)

    PrintError("can not allocate memory");

  else if (res == (SZ_RESULT)SZE_CRC_ERROR)

    PrintError("CRC error");

  else     

    printf("\nERROR #%d\n", res);

  return 1;

}