Mirrors/velopack
1
0
Fork 0
mirror of https://github.com/velopack/velopack.git synced 2025-10-25 15:19:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

SetupBundle code formatting

Browse Source
This commit is contained in:
Caelan Sayler
2024-10-15 17:05:06 +01:00
committed by Caelan
parent de4ed6bea0
commit 118d870554
1 changed files with 31 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
src/vpk/Velopack.Packaging.Windows/SetupBundle.cs
View File

@@ -5,13 +5,13 @@ namespace Velopack.Packaging.Windows;
public static class SetupBundle
{
private readonly static byte[] SquirrelBundleSignature = [
// 64 bytes represent the bundle signature: SHA-256 for "squirrel bundle"
0x94, 0xf0, 0xb1, 0x7b, 0x68, 0x93, 0xe0, 0x29,
0x37, 0xeb, 0x34, 0xef, 0x53, 0xaa, 0xe7, 0xd4,
0x2b, 0x54, 0xf5, 0x70, 0x7e, 0xf5, 0xd6, 0xf5,
0x78, 0x54, 0x98, 0x3e, 0x5e, 0x94, 0xed, 0x7d
];
private static readonly byte[] SquirrelBundleSignature = [
// 64 bytes represent the bundle signature: SHA-256 for "squirrel bundle"
0x94, 0xf0, 0xb1, 0x7b, 0x68, 0x93, 0xe0, 0x29,
0x37, 0xeb, 0x34, 0xef, 0x53, 0xaa, 0xe7, 0xd4,
0x2b, 0x54, 0xf5, 0x70, 0x7e, 0xf5, 0xd6, 0xf5,
0x78, 0x54, 0x98, 0x3e, 0x5e, 0x94, 0xed, 0x7d
];
public static bool IsBundle(string setupPath, out long bundleOffset, out long bundleLength)
{
@@ -61,7 +61,6 @@ public static class SetupBundle
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
// 8 bytes represent the package length
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
.. SquirrelBundleSignature
];
@@ -70,12 +69,14 @@ public static class SetupBundle
Array.Copy(BitConverter.GetBytes(bundleLength), 0, data, 8, 8);
// replace the beginning of the placeholder with the bytes from 'data'
RetryOnIOError(() =>
SearchAndReplace(setupPath, placeholder, data, pad0s: false));
RetryOnIOError(
() =>
SearchAndReplace(setupPath, placeholder, data, pad0s: false));
// memory-mapped write does not updating last write time
RetryOnIOError(() =>
File.SetLastWriteTimeUtc(setupPath, DateTime.UtcNow));
RetryOnIOError(
() =>
File.SetLastWriteTimeUtc(setupPath, DateTime.UtcNow));
if (!IsBundle(setupPath, out var offset, out var length))
throw new InvalidOperationException("Internal logic error writing setup bundle.");
@@ -84,10 +85,10 @@ public static class SetupBundle
}
internal static unsafe void SearchAndReplace(
MemoryMappedViewAccessor accessor,
byte[] searchPattern,
byte[] patternToReplace,
bool pad0s = true)
MemoryMappedViewAccessor accessor,
byte[] searchPattern,
byte[] patternToReplace,
bool pad0s = true)
{
byte* pointer = null;
@@ -125,17 +126,15 @@ public static class SetupBundle
}
}
public static unsafe void SearchAndReplace(
public static void SearchAndReplace(
string filePath,
byte[] searchPattern,
byte[] patternToReplace,
bool pad0s = true)
{
using (var mappedFile = MemoryMappedFile.CreateFromFile(filePath)) {
using (var accessor = mappedFile.CreateViewAccessor()) {
SearchAndReplace(accessor, searchPattern, patternToReplace, pad0s);
}
}
using var mappedFile = MemoryMappedFile.CreateFromFile(filePath);
using var accessor = mappedFile.CreateViewAccessor();
SearchAndReplace(accessor, searchPattern, patternToReplace, pad0s);
}
public static unsafe int SearchInFile(MemoryMappedViewAccessor accessor, byte[] searchPattern)
@@ -144,13 +143,11 @@ public static class SetupBundle
return KMPSearch(searchPattern, (byte*) safeBuffer.DangerousGetHandle(), (long) safeBuffer.ByteLength);
}
public static unsafe int SearchInFile(string filePath, byte[] searchPattern)
public static int SearchInFile(string filePath, byte[] searchPattern)
{
using (var mappedFile = MemoryMappedFile.CreateFromFile(filePath, FileMode.Open, null, 0, MemoryMappedFileAccess.Read)) {
using (var accessor = mappedFile.CreateViewAccessor(0, 0, MemoryMappedFileAccess.Read)) {
return SearchInFile(accessor, searchPattern);
}
}
using var mappedFile = MemoryMappedFile.CreateFromFile(filePath, FileMode.Open, null, 0, MemoryMappedFileAccess.Read);
using var accessor = mappedFile.CreateViewAccessor(0, 0, MemoryMappedFileAccess.Read);
return SearchInFile(accessor, searchPattern);
}
// See: https://en.wikipedia.org/wiki/Knuth%E2%80%93Morris%E2%80%93Pratt_algorithm
@@ -160,6 +157,7 @@ public static class SetupBundle
if (pattern.Length >= 1) {
table[0] = -1;
}
if (pattern.Length >= 2) {
table[1] = 0;
}
@@ -178,6 +176,7 @@ public static class SetupBundle
pos++;
}
}
return table;
}
@@ -193,6 +192,7 @@ public static class SetupBundle
if (i == pattern.Length - 1) {
return m;
}
i++;
} else {
if (table[i] > -1) {
@@ -237,8 +237,7 @@ public static class SetupBundle
fileStream.Write(buf, 0, bytesRead);
pos += bytesRead;
}
}
while (true);
} while (true);
}
public const int NumberOfRetries = 500;
@@ -301,7 +300,7 @@ public static class SetupBundle
func();
break;
} catch (HResultException hrex)
when (i < NumberOfRetries && !IsKnownIrrecoverableError(hrex.Win32HResult)) {
when (i < NumberOfRetries && !IsKnownIrrecoverableError(hrex.Win32HResult)) {
Thread.Sleep(NumMilliSecondsToWait);
}
}
@@ -310,6 +309,7 @@ public static class SetupBundle
public class HResultException : Exception
{
public readonly int Win32HResult;
public HResultException(int hResult) : base(hResult.ToString("X4"))
{
Win32HResult = hResult;