531e3b7861
- Almost all calls to Stream.Read were broken. These have been patched to all go through ReadBytes which itself has been fixed to function correctly. The key thing to note is that Stream.Read is very much allowed to return less than the requested number of bytes. If this happens and you're not checking the return result, you'll be working with partially initialized arrays and really bad stuff happens when you do that. - Call CopyTo rather than copying between streams manually. - Peek and ReadUInt8 have been changed to avoid a pointless array allocation which is significant overhead for such simple calls.
124 lines
2.7 KiB
C#
124 lines
2.7 KiB
C#
#region Copyright & License Information
|
|
/*
|
|
* Copyright 2007-2011 The OpenRA Developers (see AUTHORS)
|
|
* This file is part of OpenRA, which is free software. It is made
|
|
* available to you under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation. For more information,
|
|
* see COPYING.
|
|
*/
|
|
#endregion
|
|
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Text;
|
|
using ICSharpCode.SharpZipLib.Zip;
|
|
using SZipFile = ICSharpCode.SharpZipLib.Zip.ZipFile;
|
|
|
|
namespace OpenRA.FileSystem
|
|
{
|
|
public class ZipFile : IFolder
|
|
{
|
|
string filename;
|
|
SZipFile pkg;
|
|
int priority;
|
|
|
|
static ZipFile()
|
|
{
|
|
ZipConstants.DefaultCodePage = Encoding.Default.CodePage;
|
|
}
|
|
|
|
public ZipFile(string filename, int priority)
|
|
{
|
|
this.filename = filename;
|
|
this.priority = priority;
|
|
try
|
|
{
|
|
// pull the file into memory, dont keep it open.
|
|
pkg = new SZipFile(new MemoryStream(File.ReadAllBytes(filename)));
|
|
}
|
|
catch (ZipException e)
|
|
{
|
|
Log.Write("debug", "Couldn't load zip file: {0}", e.Message);
|
|
}
|
|
}
|
|
|
|
// Create a new zip with the specified contents
|
|
public ZipFile(string filename, int priority, Dictionary<string, byte[]> contents)
|
|
{
|
|
this.priority = priority;
|
|
this.filename = filename;
|
|
|
|
if (File.Exists(filename))
|
|
File.Delete(filename);
|
|
|
|
pkg = SZipFile.Create(filename);
|
|
Write(contents);
|
|
}
|
|
|
|
public Stream GetContent(string filename)
|
|
{
|
|
using (var z = pkg.GetInputStream(pkg.GetEntry(filename)))
|
|
{
|
|
var ms = new MemoryStream();
|
|
z.CopyTo(ms);
|
|
ms.Seek(0, SeekOrigin.Begin);
|
|
return ms;
|
|
}
|
|
}
|
|
|
|
public IEnumerable<uint> ClassicHashes()
|
|
{
|
|
foreach (ZipEntry entry in pkg)
|
|
yield return PackageEntry.HashFilename(entry.Name, PackageHashType.Classic);
|
|
}
|
|
|
|
public IEnumerable<uint> CrcHashes()
|
|
{
|
|
yield break;
|
|
}
|
|
|
|
public IEnumerable<string> AllFileNames()
|
|
{
|
|
foreach (ZipEntry entry in pkg)
|
|
yield return entry.Name;
|
|
}
|
|
|
|
public bool Exists(string filename)
|
|
{
|
|
return pkg.GetEntry(filename) != null;
|
|
}
|
|
|
|
public int Priority { get { return 500 + priority; } }
|
|
public string Name { get { return filename; } }
|
|
|
|
public void Write(Dictionary<string, byte[]> contents)
|
|
{
|
|
// TODO: Clear existing content?
|
|
pkg.Close();
|
|
pkg = SZipFile.Create(filename);
|
|
pkg.BeginUpdate();
|
|
|
|
foreach (var kvp in contents)
|
|
pkg.Add(new StaticMemoryDataSource(kvp.Value), kvp.Key);
|
|
|
|
pkg.CommitUpdate();
|
|
pkg.Close();
|
|
pkg = new SZipFile(new MemoryStream(File.ReadAllBytes(filename)));
|
|
}
|
|
}
|
|
|
|
class StaticMemoryDataSource : IStaticDataSource
|
|
{
|
|
byte[] data;
|
|
public StaticMemoryDataSource(byte[] data)
|
|
{
|
|
this.data = data;
|
|
}
|
|
|
|
public Stream GetSource()
|
|
{
|
|
return new MemoryStream(data);
|
|
}
|
|
}
|
|
}
|