1 /* 

     2 7zMain.c

     3 Test application for 7z Decoder

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

     5 */

     6 

     7 #include <stdio.h>

     8 #include <stdlib.h>

     9 #include <string.h>

    10 

    11 #ifdef _WIN32

    12 #define USE_WINDOWS_FUNCTIONS

    13 #endif

    14 

    15 #ifdef USE_WINDOWS_FUNCTIONS

    16 #include <windows.h>

    17 #endif

    18 

    19 #include "7zIn.h"

    20 #include "7zExtract.h"

    21 

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

    23 

    24 

    25 #ifdef USE_WINDOWS_FUNCTIONS

    26 typedef HANDLE MY_FILE_HANDLE;

    27 #else

    28 typedef FILE *MY_FILE_HANDLE;

    29 #endif

    30 

    31 void ConvertNumberToString(CFileSize value, char *s)

    32 {

    33   char temp[32];

    34   int pos = 0;

    35   do 

    36   {

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

    38     value /= 10;

    39   }

    40   while (value != 0);

    41   do

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

    43   while(pos > 0);

    44   *s = '\0';

    45 }

    46 

    47 #define PERIOD_4 (4 * 365 + 1)

    48 #define PERIOD_100 (PERIOD_4 * 25 - 1)

    49 #define PERIOD_400 (PERIOD_100 * 4 + 1)

    50 

    51 void ConvertFileTimeToString(CArchiveFileTime *ft, char *s)

    52 {

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

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

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

    56   unsigned temp;

    57   UInt32 v; 

    58   v64 /= 10000000;

    59   sec = (unsigned)(v64 % 60);

    60   v64 /= 60;

    61   min = (unsigned)(v64 % 60);

    62   v64 /= 60;

    63   hour = (unsigned)(v64 % 24);

    64   v64 /= 24;

    65 

    66   v = (UInt32)v64;

    67 

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

    69   v %= PERIOD_400;

    70 

    71   temp = (unsigned)(v / PERIOD_100);

    72   if (temp == 4)

    73     temp = 3;

    74   year += temp * 100;

    75   v -= temp * PERIOD_100;

    76 

    77   temp = v / PERIOD_4;

    78   if (temp == 25)

    79     temp = 24;

    80   year += temp * 4;

    81   v -= temp * PERIOD_4;

    82 

    83   temp = v / 365;

    84   if (temp == 4)

    85     temp = 3;

    86   year += temp;

    87   v -= temp * 365;

    88 

    89   if (year % 4 == 0 && (year % 100 != 0 || year % 400 == 0))

    90     ms[1] = 29;

    91   for (mon = 1; mon <= 12; mon++)

    92   {

    93     unsigned s = ms[mon - 1];

    94     if (v < s)

    95       break;

    96     v -= s;

    97   }

    98   day = (unsigned)v + 1;

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

   100 }

   101 

   102 

   103 #ifdef USE_WINDOWS_FUNCTIONS

   104 /*

   105    ReadFile and WriteFile functions in Windows have BUG:

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

   107    from/to Network file, it returns ERROR_NO_SYSTEM_RESOURCES 

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

   109 */

   110 #define kChunkSizeMax (1 << 24)

   111 #endif

   112 

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

   114 { 

   115   if (size == 0)

   116     return 0;

   117   #ifdef USE_WINDOWS_FUNCTIONS

   118   {

   119     size_t processedSize = 0;

   120     do

   121     {

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

   123       DWORD processedLoc = 0;

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

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

   126       size -= processedLoc;

   127       processedSize += processedLoc;

   128       if (!res || processedLoc == 0)

   129         break;

   130     }

   131     while (size > 0);

   132     return processedSize;

   133   }

   134   #else

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

   136   #endif

   137 }

   138 

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

   140 { 

   141   if (size == 0)

   142     return 0;

   143   #ifdef USE_WINDOWS_FUNCTIONS

   144   {

   145     size_t processedSize = 0;

   146     do

   147     {

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

   149       DWORD processedLoc = 0;

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

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

   152       size -= processedLoc;

   153       processedSize += processedLoc;

   154       if (!res)

   155         break;

   156     }

   157     while (size > 0);

   158     return processedSize;

   159   }

   160   #else

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

   162   #endif

   163 }

   164 

   165 int MyCloseFile(MY_FILE_HANDLE file)

   166 { 

   167   #ifdef USE_WINDOWS_FUNCTIONS

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

   169   #else

   170   return fclose(file); 

   171   #endif

   172 }

   173 

   174 typedef struct _CFileInStream

   175 {

   176   ISzInStream InStream;

   177   MY_FILE_HANDLE File;

   178 } CFileInStream;

   179 

   180 #ifdef _LZMA_IN_CB

   181 

   182 #define kBufferSize (1 << 12)

   183 Byte g_Buffer[kBufferSize];

   184 

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

   186 {

   187   CFileInStream *s = (CFileInStream *)object;

   188   size_t processedSizeLoc;

   189   if (maxRequiredSize > kBufferSize)

   190     maxRequiredSize = kBufferSize;

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

   192   *buffer = g_Buffer;

   193   if (processedSize != 0)

   194     *processedSize = processedSizeLoc;

   195   return SZ_OK;

   196 }

   197 

   198 #else

   199 

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

   201 {

   202   CFileInStream *s = (CFileInStream *)object;

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

   204   if (processedSize != 0)

   205     *processedSize = processedSizeLoc;

   206   return SZ_OK;

   207 }

   208 

   209 #endif

   210 

   211 SZ_RESULT SzFileSeekImp(void *object, CFileSize pos)

   212 {

   213   CFileInStream *s = (CFileInStream *)object;

   214 

   215   #ifdef USE_WINDOWS_FUNCTIONS

   216   {

   217     LARGE_INTEGER value;

   218     value.LowPart = (DWORD)pos;

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

   220     #ifdef _SZ_FILE_SIZE_32

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

   222     value.HighPart = 0;

   223     #endif

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

   225     if (value.LowPart == 0xFFFFFFFF)

   226       if(GetLastError() != NO_ERROR) 

   227         return SZE_FAIL;

   228     return SZ_OK;

   229   }

   230   #else

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

   232   if (res == 0)

   233     return SZ_OK;

   234   return SZE_FAIL;

   235   #endif

   236 }

   237 

   238 void PrintError(char *sz)

   239 {

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

   241 }

   242 

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

   244 {

   245   CFileInStream archiveStream;

   246   CArchiveDatabaseEx db;

   247   SZ_RESULT res;

   248   ISzAlloc allocImp;

   249   ISzAlloc allocTempImp;

   250 

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

   252   if (numargs == 1)

   253   {

   254     printf(

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

   256       "<Commands>\n"

   257       "  e: Extract files from archive\n"

   258       "  l: List contents of archive\n"

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

   260     return 0;

   261   }

   262   if (numargs < 3)

   263   {

   264     PrintError("incorrect command");

   265     return 1;

   266   }

   267 

   268   archiveStream.File = 

   269   #ifdef USE_WINDOWS_FUNCTIONS

   270   CreateFile(args[2], GENERIC_READ, FILE_SHARE_READ, 

   271       NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);

   272   if (archiveStream.File == INVALID_HANDLE_VALUE)

   273   #else

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

   275   if (archiveStream.File == 0)

   276   #endif

   277   {

   278     PrintError("can not open input file");

   279     return 1;

   280   }

   281 

   282   archiveStream.InStream.Read = SzFileReadImp;

   283   archiveStream.InStream.Seek = SzFileSeekImp;

   284 

   285   allocImp.Alloc = SzAlloc;

   286   allocImp.Free = SzFree;

   287 

   288   allocTempImp.Alloc = SzAllocTemp;

   289   allocTempImp.Free = SzFreeTemp;

   290 

   291   CrcGenerateTable();

   292 

   293   SzArDbExInit(&db);

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

   295   if (res == SZ_OK)

   296   {

   297     char *command = args[1];

   298     int listCommand = 0;

   299     int testCommand = 0;

   300     int extractCommand = 0;

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

   302       listCommand = 1;

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

   304       testCommand = 1;

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

   306       extractCommand = 1;

   307 

   308     if (listCommand)

   309     {

   310       UInt32 i;

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

   312       {

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

   314         char s[32], t[32];

   315         ConvertNumberToString(f->Size, s);

   316         if (f->IsLastWriteTimeDefined)

   317           ConvertFileTimeToString(&f->LastWriteTime, t);

   318         else

   319           strcpy(t, "                   ");

   320 

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

   322       }

   323     }

   324     else if (testCommand || extractCommand)

   325     {

   326       UInt32 i;

   327 

   328       /*

   329       if you need cache, use these 3 variables.

   330       if you use external function, you can make these variable as static.

   331       */

   332       UInt32 blockIndex = 0xFFFFFFFF; /* it can have any value before first call (if outBuffer = 0) */

   333       Byte *outBuffer = 0; /* it must be 0 before first call for each new archive. */

   334       size_t outBufferSize = 0;  /* it can have any value before first call (if outBuffer = 0) */

   335 

   336       printf("\n");

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

   338       {

   339         size_t offset;

   340         size_t outSizeProcessed;

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

   342         if (f->IsDirectory)

   343           printf("Directory ");

   344         else

   345           printf(testCommand ? 

   346             "Testing   ":

   347             "Extracting");

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

   349         if (f->IsDirectory)

   350         {

   351           printf("\n");

   352           continue;

   353         }

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

   355             &blockIndex, &outBuffer, &outBufferSize, 

   356             &offset, &outSizeProcessed, 

   357             &allocImp, &allocTempImp);

   358         if (res != SZ_OK)

   359           break;

   360         if (!testCommand)

   361         {

   362           MY_FILE_HANDLE outputHandle;

   363           size_t processedSize;

   364           char *fileName = f->Name;

   365           size_t nameLen = strlen(f->Name);

   366           for (; nameLen > 0; nameLen--)

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

   368             {

   369               fileName = f->Name + nameLen;

   370               break;

   371             }

   372             

   373           outputHandle = 

   374           #ifdef USE_WINDOWS_FUNCTIONS

   375             CreateFile(fileName, GENERIC_WRITE, FILE_SHARE_READ, 

   376                 NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);

   377           if (outputHandle == INVALID_HANDLE_VALUE)

   378           #else

   379           fopen(fileName, "wb+");

   380           if (outputHandle == 0)

   381           #endif

   382           {

   383             PrintError("can not open output file");

   384             res = SZE_FAIL;

   385             break;

   386           }

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

   388           if (processedSize != outSizeProcessed)

   389           {

   390             PrintError("can not write output file");

   391             res = SZE_FAIL;

   392             break;

   393           }

   394           if (MyCloseFile(outputHandle))

   395           {

   396             PrintError("can not close output file");

   397             res = SZE_FAIL;

   398             break;

   399           }

   400         }

   401         printf("\n");

   402       }

   403       allocImp.Free(outBuffer);

   404     }

   405     else

   406     {

   407       PrintError("incorrect command");

   408       res = SZE_FAIL;

   409     }

   410   }

   411   SzArDbExFree(&db, allocImp.Free);

   412 

   413   MyCloseFile(archiveStream.File);

   414   if (res == SZ_OK)

   415   {

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

   417     return 0;

   418   }

   419   if (res == (SZ_RESULT)SZE_NOTIMPL)

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

   421   else if (res == (SZ_RESULT)SZE_OUTOFMEMORY)

   422     PrintError("can not allocate memory");

   423   else if (res == (SZ_RESULT)SZE_CRC_ERROR)

   424     PrintError("CRC error");

   425   else     

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

   427   return 1;

   428 }