tor-browser

LzmaAlone.java (6948B)

---

package SevenZip;

public class LzmaAlone
{
static public class CommandLine
{
	public static final int kEncode = 0;
	public static final int kDecode = 1;
	public static final int kBenchmak = 2;
	
	public int Command = -1;
	public int NumBenchmarkPasses = 10;
	
	public int DictionarySize = 1 << 23;
	public boolean DictionarySizeIsDefined = false;
	
	public int Lc = 3;
	public int Lp = 0;
	public int Pb = 2;
	
	public int Fb = 128;
	public boolean FbIsDefined = false;
	
	public boolean Eos = false;
	
	public int Algorithm = 2;
	public int MatchFinder = 1;
	
	public String InFile;
	public String OutFile;
	
	boolean ParseSwitch(String s)
	{
		if (s.startsWith("d"))
		{
			DictionarySize = 1 << Integer.parseInt(s.substring(1));
			DictionarySizeIsDefined = true;
		}
		else if (s.startsWith("fb"))
		{
			Fb = Integer.parseInt(s.substring(2));
			FbIsDefined = true;
		}
		else if (s.startsWith("a"))
			Algorithm = Integer.parseInt(s.substring(1));
		else if (s.startsWith("lc"))
			Lc = Integer.parseInt(s.substring(2));
		else if (s.startsWith("lp"))
			Lp = Integer.parseInt(s.substring(2));
		else if (s.startsWith("pb"))
			Pb = Integer.parseInt(s.substring(2));
		else if (s.startsWith("eos"))
			Eos = true;
		else if (s.startsWith("mf"))
		{
			String mfs = s.substring(2);
			if (mfs.equals("bt2"))
				MatchFinder = 0;
			else if (mfs.equals("bt4"))
				MatchFinder = 1;
			else if (mfs.equals("bt4b"))
				MatchFinder = 2;
			else
				return false;
		}
		else
			return false;
		return true;
	}
	
	public boolean Parse(String[] args) throws Exception
	{
		int pos = 0;
		boolean switchMode = true;
		for (int i = 0; i < args.length; i++)
		{
			String s = args[i];
			if (s.length() == 0)
				return false;
			if (switchMode)
			{
				if (s.compareTo("--") == 0)
				{
					switchMode = false;
					continue;
				}
				if (s.charAt(0) == '-')
				{
					String sw = s.substring(1).toLowerCase();
					if (sw.length() == 0)
						return false;
					try
					{
						if (!ParseSwitch(sw))
							return false;
					}
					catch (NumberFormatException e)
					{
						return false;
					}
					continue;
				}
			}
			if (pos == 0)
			{
				if (s.equalsIgnoreCase("e"))
					Command = kEncode;
				else if (s.equalsIgnoreCase("d"))
					Command = kDecode;
				else if (s.equalsIgnoreCase("b"))
					Command = kBenchmak;
				else
					return false;
			}
			else if(pos == 1)
			{
				if (Command == kBenchmak)
				{
					try
					{
						NumBenchmarkPasses = Integer.parseInt(s);
						if (NumBenchmarkPasses < 1)
							return false;
					}
					catch (NumberFormatException e)
					{
						return false;
					}
				}
				else
					InFile = s;
			}
			else if(pos == 2)
				OutFile = s;
			else
				return false;
			pos++;
			continue;
		}
		return true;
	}
}


static void PrintHelp()
{
	System.out.println(
			"\nUsage:  LZMA <e|d> [<switches>...] inputFile outputFile\n" +
			"  e: encode file\n" +
			"  d: decode file\n" +
			"  b: Benchmark\n" +
			"<Switches>\n" +
			// "  -a{N}:  set compression mode - [0, 1], default: 1 (max)\n" +
			"  -d{N}:  set dictionary - [0,28], default: 23 (8MB)\n" +
			"  -fb{N}: set number of fast bytes - [5, 273], default: 128\n" +
			"  -lc{N}: set number of literal context bits - [0, 8], default: 3\n" +
			"  -lp{N}: set number of literal pos bits - [0, 4], default: 0\n" +
			"  -pb{N}: set number of pos bits - [0, 4], default: 2\n" +
			"  -mf{MF_ID}: set Match Finder: [bt2, bt4], default: bt4\n" +
			"  -eos:   write End Of Stream marker\n"
			);
}

public static void main(String[] args) throws Exception
{
	System.out.println("\nLZMA (Java) 4.61  2008-11-23\n");
	
	if (args.length < 1)
	{
		PrintHelp();
		return;
	}
	
	CommandLine params = new CommandLine();
	if (!params.Parse(args))
	{
		System.out.println("\nIncorrect command");
		return;
	}
	
	if (params.Command == CommandLine.kBenchmak)
	{
		int dictionary = (1 << 21);
		if (params.DictionarySizeIsDefined)
			dictionary = params.DictionarySize;
		if (params.MatchFinder > 1)
			throw new Exception("Unsupported match finder");
		SevenZip.LzmaBench.LzmaBenchmark(params.NumBenchmarkPasses, dictionary);
	}
	else if (params.Command == CommandLine.kEncode || params.Command == CommandLine.kDecode)
	{
		java.io.File inFile = new java.io.File(params.InFile);
		java.io.File outFile = new java.io.File(params.OutFile);
		
		java.io.BufferedInputStream inStream  = new java.io.BufferedInputStream(new java.io.FileInputStream(inFile));
		java.io.BufferedOutputStream outStream = new java.io.BufferedOutputStream(new java.io.FileOutputStream(outFile));
		
		boolean eos = false;
		if (params.Eos)
			eos = true;
		if (params.Command == CommandLine.kEncode)
		{
			SevenZip.Compression.LZMA.Encoder encoder = new SevenZip.Compression.LZMA.Encoder();
			if (!encoder.SetAlgorithm(params.Algorithm))
				throw new Exception("Incorrect compression mode");
			if (!encoder.SetDictionarySize(params.DictionarySize))
				throw new Exception("Incorrect dictionary size");
			if (!encoder.SetNumFastBytes(params.Fb))
				throw new Exception("Incorrect -fb value");
			if (!encoder.SetMatchFinder(params.MatchFinder))
				throw new Exception("Incorrect -mf value");
			if (!encoder.SetLcLpPb(params.Lc, params.Lp, params.Pb))
				throw new Exception("Incorrect -lc or -lp or -pb value");
			encoder.SetEndMarkerMode(eos);
			encoder.WriteCoderProperties(outStream);
			long fileSize;
			if (eos)
				fileSize = -1;
			else
				fileSize = inFile.length();
			for (int i = 0; i < 8; i++)
				outStream.write((int)(fileSize >>> (8 * i)) & 0xFF);
			encoder.Code(inStream, outStream, -1, -1, null);
		}
		else
		{
			int propertiesSize = 5;
			byte[] properties = new byte[propertiesSize];
			if (inStream.read(properties, 0, propertiesSize) != propertiesSize)
				throw new Exception("input .lzma file is too short");
			SevenZip.Compression.LZMA.Decoder decoder = new SevenZip.Compression.LZMA.Decoder();
			if (!decoder.SetDecoderProperties(properties))
				throw new Exception("Incorrect stream properties");
			long outSize = 0;
			for (int i = 0; i < 8; i++)
			{
				int v = inStream.read();
				if (v < 0)
					throw new Exception("Can't read stream size");
				outSize |= ((long)v) << (8 * i);
			}
			if (!decoder.Code(inStream, outStream, outSize))
				throw new Exception("Error in data stream");
		}
		outStream.flush();
		outStream.close();
		inStream.close();
	}
	else
		throw new Exception("Incorrect command");
	return;
}
}