xenadmin/xva_verify/verify_main.cs
auron-muc 6d0c07d5d6 Some Enhancements to input stream handling in xva_verify (#1870)
* Some enhancements to input stream handling

- Open input file in read mode to allow readonly data
- Allow reading from stdin with filename "-"
- test for compressed data, use decompression if needed.

* Fixed missing reset of stream whan not gzipped

* BufferedStream not necessary
2018-01-19 14:08:06 +00:00

109 lines
3.9 KiB
C#

/* Copyright (c) Citrix Systems, Inc.
* All rights reserved.
*
* Redistribution and use in source and binary forms,
* with or without modification, are permitted provided
* that the following conditions are met:
*
* * Redistributions of source code must retain the above
* copyright notice, this list of conditions and the
* following disclaimer.
* * Redistributions in binary form must reproduce the above
* copyright notice, this list of conditions and the
* following disclaimer in the documentation and/or other
* materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
* CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
using System;
using System.IO;
using System.IO.Compression;
class MainClass
{
public static void Main(string[] args)
{
Export.verbose_debugging = true;
if ((args.Length != 1) && (args.Length != 2))
{
Console.WriteLine("Usage: ");
Console.WriteLine(" verify <archive> [optional copy]");
Environment.Exit(1);
}
try
{
string filename = args[0];
Stream f;
Stream g = null;
if (args.Length == 2)
{
g = new FileStream(args[1], FileMode.Create);
}
if (! args[0].Equals("-")) {
f = new FileStream(filename, FileMode.Open, FileAccess.Read);
} else
{
f = Console.OpenStandardInput();
}
// check for gzip compression ( only on seekable inputs - i.e. not the stdin stream )
if (f.CanSeek)
{
try
{
GZipStream zip = new GZipStream(f, CompressionMode.Decompress);
// try reading a byte
zip.ReadByte();
// success - reset stream, use the gunzipped stream from now on
f.Seek(0, SeekOrigin.Begin);
f = new GZipStream(f, CompressionMode.Decompress);
}
catch (InvalidDataException e)
{
// just reset the stream - Exception means the stream is not compressed
f.Seek(0, SeekOrigin.Begin);
}
}
bool cancelling = false;
new Export().verify(f, g, (Export.cancellingCallback)delegate() { return cancelling; });
}
catch (UnauthorizedAccessException)
{
Console.WriteLine("Permission denied, check access rights to file");
}
catch (FileNotFoundException)
{
Console.WriteLine("File not found, verify filename is correct");
}
catch (IOException)
{
Console.WriteLine("IO Exception, file may be truncated.");
}
catch (BlockChecksumFailed)
{
Console.WriteLine("Verification failed, file appears to be corrupt");
}
catch (Exception e)
{
Console.WriteLine(e.Message);
}
}
}