1 package SevenZip; 2 3 public class LzmaAlone 4 { 5 static public class CommandLine 6 { 7 public static final int kEncode = 0; 8 public static final int kDecode = 1; 9 public static final int kBenchmak = 2; 10 11 public int Command = -1; 12 public int NumBenchmarkPasses = 10; 13 14 public int DictionarySize = 1 << 23; 15 public boolean DictionarySizeIsDefined = false; 16 17 public int Lc = 3; 18 public int Lp = 0; 19 public int Pb = 2; 20 21 public int Fb = 128; 22 public boolean FbIsDefined = false; 23 24 public boolean Eos = false; 25 26 public int Algorithm = 2; 27 public int MatchFinder = 1; 28 29 public String InFile; 30 public String OutFile; 31 ParseSwitch(String s)32 boolean ParseSwitch(String s) 33 { 34 if (s.startsWith("d")) 35 { 36 DictionarySize = 1 << Integer.parseInt(s.substring(1)); 37 DictionarySizeIsDefined = true; 38 } 39 else if (s.startsWith("fb")) 40 { 41 Fb = Integer.parseInt(s.substring(2)); 42 FbIsDefined = true; 43 } 44 else if (s.startsWith("a")) 45 Algorithm = Integer.parseInt(s.substring(1)); 46 else if (s.startsWith("lc")) 47 Lc = Integer.parseInt(s.substring(2)); 48 else if (s.startsWith("lp")) 49 Lp = Integer.parseInt(s.substring(2)); 50 else if (s.startsWith("pb")) 51 Pb = Integer.parseInt(s.substring(2)); 52 else if (s.startsWith("eos")) 53 Eos = true; 54 else if (s.startsWith("mf")) 55 { 56 String mfs = s.substring(2); 57 if (mfs.equals("bt2")) 58 MatchFinder = 0; 59 else if (mfs.equals("bt4")) 60 MatchFinder = 1; 61 else if (mfs.equals("bt4b")) 62 MatchFinder = 2; 63 else 64 return false; 65 } 66 else 67 return false; 68 return true; 69 } 70 Parse(String[] args)71 public boolean Parse(String[] args) throws Exception 72 { 73 int pos = 0; 74 boolean switchMode = true; 75 for (int i = 0; i < args.length; i++) 76 { 77 String s = args[i]; 78 if (s.length() == 0) 79 return false; 80 if (switchMode) 81 { 82 if (s.compareTo("--") == 0) 83 { 84 switchMode = false; 85 continue; 86 } 87 if (s.charAt(0) == '-') 88 { 89 String sw = s.substring(1).toLowerCase(); 90 if (sw.length() == 0) 91 return false; 92 try 93 { 94 if (!ParseSwitch(sw)) 95 return false; 96 } 97 catch (NumberFormatException e) 98 { 99 return false; 100 } 101 continue; 102 } 103 } 104 if (pos == 0) 105 { 106 if (s.equalsIgnoreCase("e")) 107 Command = kEncode; 108 else if (s.equalsIgnoreCase("d")) 109 Command = kDecode; 110 else if (s.equalsIgnoreCase("b")) 111 Command = kBenchmak; 112 else 113 return false; 114 } 115 else if(pos == 1) 116 { 117 if (Command == kBenchmak) 118 { 119 try 120 { 121 NumBenchmarkPasses = Integer.parseInt(s); 122 if (NumBenchmarkPasses < 1) 123 return false; 124 } 125 catch (NumberFormatException e) 126 { 127 return false; 128 } 129 } 130 else 131 InFile = s; 132 } 133 else if(pos == 2) 134 OutFile = s; 135 else 136 return false; 137 pos++; 138 continue; 139 } 140 return true; 141 } 142 } 143 144 PrintHelp()145 static void PrintHelp() 146 { 147 System.out.println( 148 "\nUsage: LZMA <e|d> [<switches>...] inputFile outputFile\n" + 149 " e: encode file\n" + 150 " d: decode file\n" + 151 " b: Benchmark\n" + 152 "<Switches>\n" + 153 // " -a{N}: set compression mode - [0, 1], default: 1 (max)\n" + 154 " -d{N}: set dictionary - [0,28], default: 23 (8MB)\n" + 155 " -fb{N}: set number of fast bytes - [5, 273], default: 128\n" + 156 " -lc{N}: set number of literal context bits - [0, 8], default: 3\n" + 157 " -lp{N}: set number of literal pos bits - [0, 4], default: 0\n" + 158 " -pb{N}: set number of pos bits - [0, 4], default: 2\n" + 159 " -mf{MF_ID}: set Match Finder: [bt2, bt4], default: bt4\n" + 160 " -eos: write End Of Stream marker\n" 161 ); 162 } 163 main(String[] args)164 public static void main(String[] args) throws Exception 165 { 166 System.out.println("\nLZMA (Java) 4.61 2008-11-23\n"); 167 168 if (args.length < 1) 169 { 170 PrintHelp(); 171 return; 172 } 173 174 CommandLine params = new CommandLine(); 175 if (!params.Parse(args)) 176 { 177 System.out.println("\nIncorrect command"); 178 return; 179 } 180 181 if (params.Command == CommandLine.kBenchmak) 182 { 183 int dictionary = (1 << 21); 184 if (params.DictionarySizeIsDefined) 185 dictionary = params.DictionarySize; 186 if (params.MatchFinder > 1) 187 throw new Exception("Unsupported match finder"); 188 SevenZip.LzmaBench.LzmaBenchmark(params.NumBenchmarkPasses, dictionary); 189 } 190 else if (params.Command == CommandLine.kEncode || params.Command == CommandLine.kDecode) 191 { 192 java.io.File inFile = new java.io.File(params.InFile); 193 java.io.File outFile = new java.io.File(params.OutFile); 194 195 java.io.BufferedInputStream inStream = new java.io.BufferedInputStream(new java.io.FileInputStream(inFile)); 196 java.io.BufferedOutputStream outStream = new java.io.BufferedOutputStream(new java.io.FileOutputStream(outFile)); 197 198 boolean eos = false; 199 if (params.Eos) 200 eos = true; 201 if (params.Command == CommandLine.kEncode) 202 { 203 SevenZip.Compression.LZMA.Encoder encoder = new SevenZip.Compression.LZMA.Encoder(); 204 if (!encoder.SetAlgorithm(params.Algorithm)) 205 throw new Exception("Incorrect compression mode"); 206 if (!encoder.SetDictionarySize(params.DictionarySize)) 207 throw new Exception("Incorrect dictionary size"); 208 if (!encoder.SetNumFastBytes(params.Fb)) 209 throw new Exception("Incorrect -fb value"); 210 if (!encoder.SetMatchFinder(params.MatchFinder)) 211 throw new Exception("Incorrect -mf value"); 212 if (!encoder.SetLcLpPb(params.Lc, params.Lp, params.Pb)) 213 throw new Exception("Incorrect -lc or -lp or -pb value"); 214 encoder.SetEndMarkerMode(eos); 215 encoder.WriteCoderProperties(outStream); 216 long fileSize; 217 if (eos) 218 fileSize = -1; 219 else 220 fileSize = inFile.length(); 221 for (int i = 0; i < 8; i++) 222 outStream.write((int)(fileSize >>> (8 * i)) & 0xFF); 223 encoder.Code(inStream, outStream, -1, -1, null); 224 } 225 else 226 { 227 int propertiesSize = 5; 228 byte[] properties = new byte[propertiesSize]; 229 if (inStream.read(properties, 0, propertiesSize) != propertiesSize) 230 throw new Exception("input .lzma file is too short"); 231 SevenZip.Compression.LZMA.Decoder decoder = new SevenZip.Compression.LZMA.Decoder(); 232 if (!decoder.SetDecoderProperties(properties)) 233 throw new Exception("Incorrect stream properties"); 234 long outSize = 0; 235 for (int i = 0; i < 8; i++) 236 { 237 int v = inStream.read(); 238 if (v < 0) 239 throw new Exception("Can't read stream size"); 240 outSize |= ((long)v) << (8 * i); 241 } 242 if (!decoder.Code(inStream, outStream, outSize)) 243 throw new Exception("Error in data stream"); 244 } 245 outStream.flush(); 246 outStream.close(); 247 inStream.close(); 248 } 249 else 250 throw new Exception("Incorrect command"); 251 return; 252 } 253 } 254