git.oblomov.eu Git - wine/blob - programs/expand/expand.c

   1 /*
   2  * Copyright 1997 Victor Schneider
   3  * Copyright 2002 Alexandre Julliard
   4  * Copyright 2007 Hans Leidekker
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #define WIN32_LEAN_AND_MEAN
  22
  23 #include <stdio.h>
  24 #include <string.h>
  25 #include <windows.h>
  26 #include <lzexpand.h>
  27 #include <setupapi.h>
  28
  29 static UINT CALLBACK set_outfile( PVOID context, UINT notification, UINT_PTR param1, UINT_PTR param2 )
  30 {
  31     FILE_IN_CABINET_INFO_A *info = (FILE_IN_CABINET_INFO_A *)param1;
  32     char buffer[MAX_PATH];
  33     char* basename;
  34
  35     switch (notification)
  36     {
  37     case SPFILENOTIFY_FILEINCABINET:
  38     {
  39         LPSTR outfile = context;
  40         if (outfile[0] != 0)
  41         {
  42             SetLastError( ERROR_NOT_SUPPORTED );
  43             return FILEOP_ABORT;
  44         }
  45         GetFullPathNameA( info->NameInCabinet, sizeof(buffer), buffer, &basename );
  46         strcpy( outfile, basename );
  47         return FILEOP_SKIP;
  48     }
  49     default: return NO_ERROR;
  50     }
  51 }
  52
  53 static UINT CALLBACK extract_callback( PVOID context, UINT notification, UINT_PTR param1, UINT_PTR param2 )
  54 {
  55     FILE_IN_CABINET_INFO_A *info = (FILE_IN_CABINET_INFO_A *)param1;
  56
  57     switch (notification)
  58     {
  59     case SPFILENOTIFY_FILEINCABINET:
  60     {
  61         LPCSTR targetname = context;
  62
  63         strcpy( info->FullTargetName, targetname );
  64         return FILEOP_DOIT;
  65     }
  66     default: return NO_ERROR;
  67     }
  68 }
  69
  70 static BOOL option_equal(LPCSTR str1, LPCSTR str2)
  71 {
  72     if (str1[0] != '/' && str1[0] != '-')
  73         return FALSE;
  74     return !lstrcmpA( str1 + 1, str2 );
  75 }
  76
  77 int main(int argc, char *argv[])
  78 {
  79     int ret = 0;
  80     char infile[MAX_PATH], outfile[MAX_PATH], actual_name[MAX_PATH];
  81     char outfile_basename[MAX_PATH], *basename_index;
  82     UINT comp;
  83
  84     if (argc < 3)
  85     {
  86         fprintf( stderr, "Usage:\n" );
  87         fprintf( stderr, "\t%s infile outfile\n", argv[0] );
  88         fprintf( stderr, "\t%s /r infile\n", argv[0] );
  89         return 1;
  90     }
  91
  92     if (argc == 3 && (option_equal(argv[1], "R") || option_equal(argv[1], "r")))
  93         GetFullPathNameA( argv[2], sizeof(infile), infile, NULL );
  94     else
  95         GetFullPathNameA( argv[1], sizeof(infile), infile, NULL );
  96
  97     if (!SetupGetFileCompressionInfoExA( infile, actual_name, sizeof(actual_name), NULL, NULL, NULL, &comp ))
  98     {
  99         fprintf( stderr, "%s: can't open input file %s\n", argv[0], infile );
 100         return 1;
 101     }
 102
 103     if (argc == 3 && (option_equal(argv[1], "R") || option_equal(argv[1], "r")))
 104     {
 105         switch (comp)
 106         {
 107         case FILE_COMPRESSION_MSZIP:
 108             outfile_basename[0] = 0;
 109             if (!SetupIterateCabinetA( infile, 0, set_outfile, outfile_basename ))
 110             {
 111                 fprintf( stderr, "%s: can't determine original name\n", argv[0] );
 112                 return 1;
 113             }
 114             GetFullPathNameA( infile, sizeof(outfile), outfile, &basename_index );
 115             *basename_index = 0;
 116             strcat( outfile, outfile_basename );
 117             break;
 118         case FILE_COMPRESSION_WINLZA:
 119             GetExpandedNameA( infile, outfile_basename );
 120             break;
 121         default:
 122             fprintf( stderr, "%s: can't determine original\n", argv[0] );
 123             return 1;
 124         }
 125     }
 126     else
 127         GetFullPathNameA( argv[2], sizeof(outfile), outfile, NULL );
 128
 129     if (!lstrcmpiA( infile, outfile ))
 130     {
 131         fprintf( stderr, "%s: can't expand file to itself\n", argv[0] );
 132         return 1;
 133     }
 134
 135     switch (comp)
 136     {
 137     case FILE_COMPRESSION_MSZIP:
 138         if (!SetupIterateCabinetA( infile, 0, extract_callback, outfile ))
 139         {
 140             fprintf( stderr, "%s: cabinet extraction failed\n", argv[0] );
 141             return 1;
 142         }
 143         break;
 144     case FILE_COMPRESSION_WINLZA:
 145     {
 146         INT hin, hout;
 147         OFSTRUCT ofin, ofout;
 148         LONG error;
 149
 150         if ((hin = LZOpenFileA( infile, &ofin, OF_READ )) < 0)
 151         {
 152             fprintf( stderr, "%s: can't open input file %s\n", argv[0], infile );
 153             return 1;
 154         }
 155         if ((hout = LZOpenFileA( outfile, &ofout, OF_CREATE | OF_WRITE )) < 0)
 156         {
 157             LZClose( hin );
 158             fprintf( stderr, "%s: can't open output file %s\n", argv[0], outfile );
 159             return 1;
 160         }
 161         error = LZCopy( hin, hout );
 162
 163         LZClose( hin );
 164         LZClose( hout );
 165
 166         if (error < 0)
 167         {
 168             fprintf( stderr, "%s: LZCopy failed, error is %d\n", argv[0], error );
 169             return 1;
 170         }
 171         break;
 172     }
 173     default:
 174         if (!CopyFileA( infile, outfile, FALSE ))
 175         {
 176             fprintf( stderr, "%s: CopyFileA failed\n", argv[0] );
 177             return 1;
 178         }
 179         break;
 180     }
 181     return ret;
 182 }