implify multitap core options

[pcsx_rearmed.git] / deps / lzma-16.04 / C / Util / Lzma / LzmaUtil.c

/* LzmaUtil.c -- Test application for LZMA compression\r
2015-11-08 : Igor Pavlov : Public domain */\r
\r
#include "../../Precomp.h"\r
\r
#include <stdio.h>\r
#include <stdlib.h>\r
#include <string.h>\r
\r
#include "../../Alloc.h"\r
#include "../../7zFile.h"\r
#include "../../7zVersion.h"\r
#include "../../LzmaDec.h"\r
#include "../../LzmaEnc.h"\r
\r
const char *kCantReadMessage = "Can not read input file";\r
const char *kCantWriteMessage = "Can not write output file";\r
const char *kCantAllocateMessage = "Can not allocate memory";\r
const char *kDataErrorMessage = "Data error";\r
\r
void PrintHelp(char *buffer)\r
{\r
  strcat(buffer, "\nLZMA Utility " MY_VERSION_COPYRIGHT_DATE "\n"\r
      "\nUsage:  lzma <e|d> inputFile outputFile\n"\r
             "  e: encode file\n"\r
             "  d: decode file\n");\r
}\r
\r
int PrintError(char *buffer, const char *message)\r
{\r
  strcat(buffer, "\nError: ");\r
  strcat(buffer, message);\r
  strcat(buffer, "\n");\r
  return 1;\r
}\r
\r
int PrintErrorNumber(char *buffer, SRes val)\r
{\r
  sprintf(buffer + strlen(buffer), "\nError code: %x\n", (unsigned)val);\r
  return 1;\r
}\r
\r
int PrintUserError(char *buffer)\r
{\r
  return PrintError(buffer, "Incorrect command");\r
}\r
\r
#define IN_BUF_SIZE (1 << 16)\r
#define OUT_BUF_SIZE (1 << 16)\r
\r
static SRes Decode2(CLzmaDec *state, ISeqOutStream *outStream, ISeqInStream *inStream,\r
    UInt64 unpackSize)\r
{\r
  int thereIsSize = (unpackSize != (UInt64)(Int64)-1);\r
  Byte inBuf[IN_BUF_SIZE];\r
  Byte outBuf[OUT_BUF_SIZE];\r
  size_t inPos = 0, inSize = 0, outPos = 0;\r
  LzmaDec_Init(state);\r
  for (;;)\r
  {\r
    if (inPos == inSize)\r
    {\r
      inSize = IN_BUF_SIZE;\r
      RINOK(inStream->Read(inStream, inBuf, &inSize));\r
      inPos = 0;\r
    }\r
    {\r
      SRes res;\r
      SizeT inProcessed = inSize - inPos;\r
      SizeT outProcessed = OUT_BUF_SIZE - outPos;\r
      ELzmaFinishMode finishMode = LZMA_FINISH_ANY;\r
      ELzmaStatus status;\r
      if (thereIsSize && outProcessed > unpackSize)\r
      {\r
        outProcessed = (SizeT)unpackSize;\r
        finishMode = LZMA_FINISH_END;\r
      }\r
      \r
      res = LzmaDec_DecodeToBuf(state, outBuf + outPos, &outProcessed,\r
        inBuf + inPos, &inProcessed, finishMode, &status);\r
      inPos += inProcessed;\r
      outPos += outProcessed;\r
      unpackSize -= outProcessed;\r
      \r
      if (outStream)\r
        if (outStream->Write(outStream, outBuf, outPos) != outPos)\r
          return SZ_ERROR_WRITE;\r
        \r
      outPos = 0;\r
      \r
      if (res != SZ_OK || (thereIsSize && unpackSize == 0))\r
        return res;\r
      \r
      if (inProcessed == 0 && outProcessed == 0)\r
      {\r
        if (thereIsSize || status != LZMA_STATUS_FINISHED_WITH_MARK)\r
          return SZ_ERROR_DATA;\r
        return res;\r
      }\r
    }\r
  }\r
}\r
\r
static SRes Decode(ISeqOutStream *outStream, ISeqInStream *inStream)\r
{\r
  UInt64 unpackSize;\r
  int i;\r
  SRes res = 0;\r
\r
  CLzmaDec state;\r
\r
  /* header: 5 bytes of LZMA properties and 8 bytes of uncompressed size */\r
  unsigned char header[LZMA_PROPS_SIZE + 8];\r
\r
  /* Read and parse header */\r
\r
  RINOK(SeqInStream_Read(inStream, header, sizeof(header)));\r
\r
  unpackSize = 0;\r
  for (i = 0; i < 8; i++)\r
    unpackSize += (UInt64)header[LZMA_PROPS_SIZE + i] << (i * 8);\r
\r
  LzmaDec_Construct(&state);\r
  RINOK(LzmaDec_Allocate(&state, header, LZMA_PROPS_SIZE, &g_Alloc));\r
  res = Decode2(&state, outStream, inStream, unpackSize);\r
  LzmaDec_Free(&state, &g_Alloc);\r
  return res;\r
}\r
\r
static SRes Encode(ISeqOutStream *outStream, ISeqInStream *inStream, UInt64 fileSize, char *rs)\r
{\r
  CLzmaEncHandle enc;\r
  SRes res;\r
  CLzmaEncProps props;\r
\r
  UNUSED_VAR(rs);\r
\r
  enc = LzmaEnc_Create(&g_Alloc);\r
  if (enc == 0)\r
    return SZ_ERROR_MEM;\r
\r
  LzmaEncProps_Init(&props);\r
  res = LzmaEnc_SetProps(enc, &props);\r
\r
  if (res == SZ_OK)\r
  {\r
    Byte header[LZMA_PROPS_SIZE + 8];\r
    size_t headerSize = LZMA_PROPS_SIZE;\r
    int i;\r
\r
    res = LzmaEnc_WriteProperties(enc, header, &headerSize);\r
    for (i = 0; i < 8; i++)\r
      header[headerSize++] = (Byte)(fileSize >> (8 * i));\r
    if (outStream->Write(outStream, header, headerSize) != headerSize)\r
      res = SZ_ERROR_WRITE;\r
    else\r
    {\r
      if (res == SZ_OK)\r
        res = LzmaEnc_Encode(enc, outStream, inStream, NULL, &g_Alloc, &g_Alloc);\r
    }\r
  }\r
  LzmaEnc_Destroy(enc, &g_Alloc, &g_Alloc);\r
  return res;\r
}\r
\r
int main2(int numArgs, const char *args[], char *rs)\r
{\r
  CFileSeqInStream inStream;\r
  CFileOutStream outStream;\r
  char c;\r
  int res;\r
  int encodeMode;\r
  Bool useOutFile = False;\r
\r
  FileSeqInStream_CreateVTable(&inStream);\r
  File_Construct(&inStream.file);\r
\r
  FileOutStream_CreateVTable(&outStream);\r
  File_Construct(&outStream.file);\r
\r
  if (numArgs == 1)\r
  {\r
    PrintHelp(rs);\r
    return 0;\r
  }\r
\r
  if (numArgs < 3 || numArgs > 4 || strlen(args[1]) != 1)\r
    return PrintUserError(rs);\r
\r
  c = args[1][0];\r
  encodeMode = (c == 'e' || c == 'E');\r
  if (!encodeMode && c != 'd' && c != 'D')\r
    return PrintUserError(rs);\r
\r
  {\r
    size_t t4 = sizeof(UInt32);\r
    size_t t8 = sizeof(UInt64);\r
    if (t4 != 4 || t8 != 8)\r
      return PrintError(rs, "Incorrect UInt32 or UInt64");\r
  }\r
\r
  if (InFile_Open(&inStream.file, args[2]) != 0)\r
    return PrintError(rs, "Can not open input file");\r
\r
  if (numArgs > 3)\r
  {\r
    useOutFile = True;\r
    if (OutFile_Open(&outStream.file, args[3]) != 0)\r
      return PrintError(rs, "Can not open output file");\r
  }\r
  else if (encodeMode)\r
    PrintUserError(rs);\r
\r
  if (encodeMode)\r
  {\r
    UInt64 fileSize;\r
    File_GetLength(&inStream.file, &fileSize);\r
    res = Encode(&outStream.s, &inStream.s, fileSize, rs);\r
  }\r
  else\r
  {\r
    res = Decode(&outStream.s, useOutFile ? &inStream.s : NULL);\r
  }\r
\r
  if (useOutFile)\r
    File_Close(&outStream.file);\r
  File_Close(&inStream.file);\r
\r
  if (res != SZ_OK)\r
  {\r
    if (res == SZ_ERROR_MEM)\r
      return PrintError(rs, kCantAllocateMessage);\r
    else if (res == SZ_ERROR_DATA)\r
      return PrintError(rs, kDataErrorMessage);\r
    else if (res == SZ_ERROR_WRITE)\r
      return PrintError(rs, kCantWriteMessage);\r
    else if (res == SZ_ERROR_READ)\r
      return PrintError(rs, kCantReadMessage);\r
    return PrintErrorNumber(rs, res);\r
  }\r
  return 0;\r
}\r
\r
int MY_CDECL main(int numArgs, const char *args[])\r
{\r
  char rs[800] = { 0 };\r
  int res = main2(numArgs, args, rs);\r
  fputs(rs, stdout);\r
  return res;\r
}\r