Compare commits
3 Commits
Author | SHA1 | Date | |
---|---|---|---|
5520b26658 | |||
e8b7b8408c | |||
|
bee08f34e6 |
2
Archiver/.gitignore
vendored
Normal file
2
Archiver/.gitignore
vendored
Normal file
@ -0,0 +1,2 @@
|
||||
bin
|
||||
obj
|
18
Archiver/Archiver.csproj
Normal file
18
Archiver/Archiver.csproj
Normal file
@ -0,0 +1,18 @@
|
||||
<Project Sdk="Microsoft.NET.Sdk">
|
||||
|
||||
<ItemGroup>
|
||||
<ProjectReference Include="..\ProtocolLib\ProtocolLib.csproj" />
|
||||
</ItemGroup>
|
||||
|
||||
<ItemGroup>
|
||||
<PackageReference Include="CommandLineParser" Version="2.9.1" />
|
||||
</ItemGroup>
|
||||
|
||||
<PropertyGroup>
|
||||
<OutputType>Exe</OutputType>
|
||||
<TargetFramework>net8.0</TargetFramework>
|
||||
<ImplicitUsings>enable</ImplicitUsings>
|
||||
<Nullable>enable</Nullable>
|
||||
</PropertyGroup>
|
||||
|
||||
</Project>
|
25
Archiver/Archiver.sln
Normal file
25
Archiver/Archiver.sln
Normal file
@ -0,0 +1,25 @@
|
||||
|
||||
Microsoft Visual Studio Solution File, Format Version 12.00
|
||||
# Visual Studio Version 17
|
||||
VisualStudioVersion = 17.5.002.0
|
||||
MinimumVisualStudioVersion = 10.0.40219.1
|
||||
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "Archiver", "Archiver.csproj", "{EF1E4119-D2B7-420D-9D4B-3EBC40FCE68C}"
|
||||
EndProject
|
||||
Global
|
||||
GlobalSection(SolutionConfigurationPlatforms) = preSolution
|
||||
Debug|Any CPU = Debug|Any CPU
|
||||
Release|Any CPU = Release|Any CPU
|
||||
EndGlobalSection
|
||||
GlobalSection(ProjectConfigurationPlatforms) = postSolution
|
||||
{EF1E4119-D2B7-420D-9D4B-3EBC40FCE68C}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
|
||||
{EF1E4119-D2B7-420D-9D4B-3EBC40FCE68C}.Debug|Any CPU.Build.0 = Debug|Any CPU
|
||||
{EF1E4119-D2B7-420D-9D4B-3EBC40FCE68C}.Release|Any CPU.ActiveCfg = Release|Any CPU
|
||||
{EF1E4119-D2B7-420D-9D4B-3EBC40FCE68C}.Release|Any CPU.Build.0 = Release|Any CPU
|
||||
EndGlobalSection
|
||||
GlobalSection(SolutionProperties) = preSolution
|
||||
HideSolutionNode = FALSE
|
||||
EndGlobalSection
|
||||
GlobalSection(ExtensibilityGlobals) = postSolution
|
||||
SolutionGuid = {92B64C8C-A729-4720-8CCF-F1A1BE5FB4D1}
|
||||
EndGlobalSection
|
||||
EndGlobal
|
85
Archiver/Lib/ArchiveDecoder.cs
Normal file
85
Archiver/Lib/ArchiveDecoder.cs
Normal file
@ -0,0 +1,85 @@
|
||||
using PacketIO;
|
||||
|
||||
namespace Archiver.Lib;
|
||||
|
||||
public static class ArchiveDecoder
|
||||
{
|
||||
public static async ValueTask Decode(string inputPath, string outputPath)
|
||||
{
|
||||
if (!File.Exists(inputPath))
|
||||
{
|
||||
throw new FileNotFoundException($"{inputPath} does not exist", inputPath);
|
||||
}
|
||||
|
||||
FileAttributes attributes = File.GetAttributes(inputPath);
|
||||
if (attributes.HasFlag(FileAttributes.Directory))
|
||||
{
|
||||
throw new IOException("Target does not file.");
|
||||
}
|
||||
|
||||
using (FileStream fs = new FileStream(inputPath, FileMode.Open))
|
||||
{
|
||||
byte[] buffer = new byte[ArchiveEncoder.CreateFileHeader().Length];
|
||||
await fs.ReadAsync(buffer, 0, buffer.Length).ConfigureAwait(false);
|
||||
if (!buffer.SequenceEqual(ArchiveEncoder.CreateFileHeader()))
|
||||
{
|
||||
throw new IOException("Not supported file type.");
|
||||
}
|
||||
|
||||
await ReadFile(fs, outputPath).ConfigureAwait(false);
|
||||
}
|
||||
}
|
||||
|
||||
private static async ValueTask ReadFile(Stream stream, string outputPath)
|
||||
{
|
||||
IPacketFile? file = await TreePacketFile.ReadTreeFile(stream).ConfigureAwait(false);
|
||||
if (file is null)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
await OutputFile(file, outputPath).ConfigureAwait(false);
|
||||
await ReadFile(stream, outputPath).ConfigureAwait(false);
|
||||
}
|
||||
|
||||
private static async ValueTask OutputFile(IPacketFile file, string outputPath)
|
||||
{
|
||||
switch (file.GetFileType())
|
||||
{
|
||||
case PacketFileType.Directory:
|
||||
{
|
||||
string dirPath = Path.Combine(outputPath, file.GetFileName());
|
||||
if (!Directory.Exists(dirPath))
|
||||
{
|
||||
Directory.CreateDirectory(dirPath);
|
||||
}
|
||||
|
||||
foreach (IPacketFile inFile in file.EnumerableFiles())
|
||||
{
|
||||
await OutputFile(inFile, dirPath).ConfigureAwait(false);
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
case PacketFileType.File:
|
||||
{
|
||||
string filePath = Path.Combine(outputPath, file.GetFileName());
|
||||
using (FileStream outputStream = new FileStream(filePath, FileMode.OpenOrCreate))
|
||||
{
|
||||
PacketFile pfile = (PacketFile) file;
|
||||
byte[] buffer = new byte[8192];
|
||||
int c;
|
||||
do
|
||||
{
|
||||
c = await pfile.ReadAsync(buffer, 0, buffer.Length).ConfigureAwait(false);
|
||||
await outputStream.WriteAsync(buffer, 0, c).ConfigureAwait(false);
|
||||
} while (c > 0);
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
file.Dispose();
|
||||
}
|
||||
}
|
186
Archiver/Lib/ArchiveEncoder.cs
Normal file
186
Archiver/Lib/ArchiveEncoder.cs
Normal file
@ -0,0 +1,186 @@
|
||||
using PacketIO;
|
||||
|
||||
namespace Archiver.Lib;
|
||||
|
||||
public static class ArchiveEncoder
|
||||
{
|
||||
public static async ValueTask Encode(string inputPath, string outputPath)
|
||||
{
|
||||
if (!Directory.Exists(inputPath))
|
||||
{
|
||||
throw new FileNotFoundException($"{inputPath} does not exist", inputPath);
|
||||
}
|
||||
|
||||
using (FileStream fs = new FileStream(outputPath, FileMode.OpenOrCreate))
|
||||
{
|
||||
byte[] buffer = CreateFileHeader();
|
||||
await fs.WriteAsync(buffer, 0, buffer.Length).ConfigureAwait(false);
|
||||
|
||||
FileAttributes attributes = File.GetAttributes(inputPath);
|
||||
if (attributes.HasFlag(FileAttributes.Directory))
|
||||
{
|
||||
DirectoryInfo dirInfo = new DirectoryInfo(inputPath);
|
||||
PacketDirectoryModel model = await CreatePacketDirectory(dirInfo).ConfigureAwait(false);
|
||||
try
|
||||
{
|
||||
await TreePacketFile.WriteTreeFiles(fs, model.PacketDirectory).ConfigureAwait(false);
|
||||
}
|
||||
finally
|
||||
{
|
||||
DisposeFiles(model);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
FileInfo fileInfo = new FileInfo(inputPath);
|
||||
PacketFileModel model = await CreatePacketFile(fileInfo).ConfigureAwait(false);
|
||||
try
|
||||
{
|
||||
await TreePacketFile.WriteTreeFiles(fs, model.PacketFile).ConfigureAwait(false);
|
||||
}
|
||||
finally
|
||||
{
|
||||
DisposeFiles(model);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public static async ValueTask Encode(IEnumerable<string> inputFiles, string outputPath)
|
||||
{
|
||||
using (FileStream fs = new FileStream(outputPath, FileMode.OpenOrCreate))
|
||||
{
|
||||
byte[] buffer = CreateFileHeader();
|
||||
await fs.WriteAsync(buffer, 0, buffer.Length).ConfigureAwait(false);
|
||||
|
||||
foreach (string inputPath in inputFiles)
|
||||
{
|
||||
FileAttributes attributes = File.GetAttributes(inputPath);
|
||||
if (attributes.HasFlag(FileAttributes.Directory))
|
||||
{
|
||||
DirectoryInfo dirInfo = new DirectoryInfo(inputPath);
|
||||
PacketDirectoryModel model = await CreatePacketDirectory(dirInfo).ConfigureAwait(false);
|
||||
try
|
||||
{
|
||||
await TreePacketFile.WriteTreeFiles(fs, model.PacketDirectory).ConfigureAwait(false);
|
||||
}
|
||||
finally
|
||||
{
|
||||
DisposeFiles(model);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
FileInfo fileInfo = new FileInfo(inputPath);
|
||||
PacketFileModel model = await CreatePacketFile(fileInfo).ConfigureAwait(false);
|
||||
try
|
||||
{
|
||||
await TreePacketFile.WriteTreeFiles(fs, model.PacketFile).ConfigureAwait(false);
|
||||
}
|
||||
finally
|
||||
{
|
||||
DisposeFiles(model);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public static bool IsArchiveFile(string inputPath)
|
||||
{
|
||||
FileAttributes attributes = File.GetAttributes(inputPath);
|
||||
if (attributes.HasFlag(FileAttributes.Directory))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
using (FileStream fs = File.OpenRead(inputPath))
|
||||
{
|
||||
byte[] buffer = new byte[CreateFileHeader().Length];
|
||||
int c = fs.Read(buffer, 0, buffer.Length);
|
||||
if (c <= 0)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
return buffer.SequenceEqual(CreateFileHeader());
|
||||
}
|
||||
}
|
||||
|
||||
internal static byte[] CreateFileHeader()
|
||||
{
|
||||
byte[] header = new byte[]
|
||||
{
|
||||
0xCC, 0xDE, 0xFF, 0x00,
|
||||
};
|
||||
return header;
|
||||
}
|
||||
|
||||
private static void DisposeFiles(PacketDirectoryModel dirModel)
|
||||
{
|
||||
dirModel.PacketDirectory.Dispose();
|
||||
foreach (PacketDirectoryModel subDirModel in dirModel.Directories)
|
||||
{
|
||||
DisposeFiles(subDirModel);
|
||||
}
|
||||
|
||||
foreach (PacketFileModel fileModel in dirModel.Files)
|
||||
{
|
||||
DisposeFiles(fileModel);
|
||||
}
|
||||
}
|
||||
|
||||
private static void DisposeFiles(PacketFileModel fileModel)
|
||||
{
|
||||
fileModel.PacketFile.Dispose();
|
||||
fileModel.TempFile.Dispose();
|
||||
}
|
||||
|
||||
private static async ValueTask<PacketDirectoryModel> CreatePacketDirectory(DirectoryInfo dirInfo)
|
||||
{
|
||||
PacketDirectory packetDirectory = new PacketDirectory(dirInfo.Name);
|
||||
PacketDirectoryModel dirModel = new PacketDirectoryModel()
|
||||
{
|
||||
PacketDirectory = packetDirectory,
|
||||
Files = new List<PacketFileModel>(),
|
||||
};
|
||||
|
||||
foreach (FileInfo fileInfo in dirInfo.EnumerateFiles())
|
||||
{
|
||||
PacketFileModel model = await CreatePacketFile(fileInfo).ConfigureAwait(false);
|
||||
dirModel.Files.Add(model);
|
||||
packetDirectory.AddFile(model.PacketFile);
|
||||
}
|
||||
|
||||
foreach (DirectoryInfo subInfo in dirInfo.EnumerateDirectories())
|
||||
{
|
||||
PacketDirectoryModel subDirModel = await CreatePacketDirectory(subInfo).ConfigureAwait(false);
|
||||
dirModel.Directories.Add(subDirModel);
|
||||
packetDirectory.AddFile(subDirModel.PacketDirectory);
|
||||
}
|
||||
|
||||
return dirModel;
|
||||
}
|
||||
|
||||
private static async ValueTask<PacketFileModel> CreatePacketFile(FileInfo fileInfo)
|
||||
{
|
||||
TempFile tempFile = new TempFile();
|
||||
Stream outputStream = tempFile.Open(FileMode.Open);
|
||||
|
||||
using (FileStream sourceStream = fileInfo.OpenRead())
|
||||
{
|
||||
await sourceStream.CopyToAsync(outputStream).ConfigureAwait(false);
|
||||
}
|
||||
|
||||
outputStream.Seek(0, SeekOrigin.Begin);
|
||||
PacketFile file = new PacketFile(fileInfo.Name, outputStream);
|
||||
PacketFileModel model = new PacketFileModel()
|
||||
{
|
||||
TempFile = tempFile,
|
||||
PacketFile = file,
|
||||
};
|
||||
|
||||
return model;
|
||||
}
|
||||
}
|
||||
|
10
Archiver/Lib/PacketDirectoryModel.cs
Normal file
10
Archiver/Lib/PacketDirectoryModel.cs
Normal file
@ -0,0 +1,10 @@
|
||||
using PacketIO;
|
||||
|
||||
namespace Archiver.Lib;
|
||||
|
||||
public class PacketDirectoryModel
|
||||
{
|
||||
public PacketDirectory PacketDirectory { get; set; }
|
||||
public List<PacketFileModel> Files { get; set; } = new List<PacketFileModel>();
|
||||
public List<PacketDirectoryModel> Directories { get; set; } = new List<PacketDirectoryModel>();
|
||||
}
|
9
Archiver/Lib/PacketFileModel.cs
Normal file
9
Archiver/Lib/PacketFileModel.cs
Normal file
@ -0,0 +1,9 @@
|
||||
using PacketIO;
|
||||
|
||||
namespace Archiver.Lib;
|
||||
|
||||
public class PacketFileModel
|
||||
{
|
||||
public TempFile TempFile { get; set;}
|
||||
public PacketFile PacketFile { get; set; }
|
||||
}
|
21
Archiver/Lib/TempFile.cs
Normal file
21
Archiver/Lib/TempFile.cs
Normal file
@ -0,0 +1,21 @@
|
||||
namespace Archiver.Lib;
|
||||
|
||||
public class TempFile : IDisposable
|
||||
{
|
||||
public string FileName { get; }
|
||||
|
||||
public TempFile()
|
||||
{
|
||||
FileName = Path.GetTempFileName();
|
||||
}
|
||||
|
||||
public Stream Open(FileMode mode)
|
||||
{
|
||||
return new FileStream(FileName, mode);
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
File.Delete(FileName);
|
||||
}
|
||||
}
|
12
Archiver/Options.cs
Normal file
12
Archiver/Options.cs
Normal file
@ -0,0 +1,12 @@
|
||||
using CommandLine;
|
||||
|
||||
namespace Archiver;
|
||||
|
||||
public class Options
|
||||
{
|
||||
[Option('i', "input", Required = true)]
|
||||
public IEnumerable<string> InputFiles { get; set; }
|
||||
|
||||
[Option('o', "output", Required = true)]
|
||||
public string OutputPath { get; set; }
|
||||
}
|
56
Archiver/Program.cs
Normal file
56
Archiver/Program.cs
Normal file
@ -0,0 +1,56 @@
|
||||
using Archiver.Lib;
|
||||
|
||||
namespace Archiver;
|
||||
|
||||
public class Program
|
||||
{
|
||||
public static void Main(string[] args)
|
||||
{
|
||||
try
|
||||
{
|
||||
Options options = CommandLine.Parser.Default.ParseArguments<Options>(args).Value;
|
||||
if (options is null || options.InputFiles is null || string.IsNullOrEmpty(options.OutputPath))
|
||||
{
|
||||
Console.WriteLine("[Archiver] Usage: archiver -i <input files> -o <output file>");
|
||||
return;
|
||||
}
|
||||
|
||||
int c = options.InputFiles.Count();
|
||||
if (c == 0)
|
||||
{
|
||||
Console.WriteLine("[Archiver] Empty input files.");
|
||||
return;
|
||||
}
|
||||
|
||||
if (c > 1)
|
||||
{
|
||||
Console.WriteLine($"[Archiver] Encode");
|
||||
Console.WriteLine($"[Archiver] Input : {string.Join(", ", options.InputFiles)}");
|
||||
Console.WriteLine($"[Archiver] Output: {options.OutputPath}");
|
||||
ArchiveEncoder.Encode(options.InputFiles, options.OutputPath).ConfigureAwait(false).GetAwaiter().GetResult();
|
||||
}
|
||||
else
|
||||
{
|
||||
string inputFile = options.InputFiles.ToArray()[0];
|
||||
if (ArchiveEncoder.IsArchiveFile(inputFile))
|
||||
{
|
||||
Console.WriteLine($"[Archiver] Decode");
|
||||
Console.WriteLine($"[Archiver] Input : {inputFile}");
|
||||
Console.WriteLine($"[Archiver] Output: {options.OutputPath}");
|
||||
ArchiveDecoder.Decode(inputFile, options.OutputPath).ConfigureAwait(false).GetAwaiter().GetResult();
|
||||
}
|
||||
else
|
||||
{
|
||||
Console.WriteLine($"[Archiver] Encode");
|
||||
Console.WriteLine($"[Archiver] Input : {string.Join(',', options.InputFiles)}");
|
||||
Console.WriteLine($"[Archiver] Output: {options.OutputPath}");
|
||||
ArchiveEncoder.Encode(inputFile, options.OutputPath).ConfigureAwait(false).GetAwaiter().GetResult();
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Console.WriteLine($"[Archiver] Error: {ex.Message}");
|
||||
}
|
||||
}
|
||||
}
|
2
ProtocolLib/.gitignore
vendored
Normal file
2
ProtocolLib/.gitignore
vendored
Normal file
@ -0,0 +1,2 @@
|
||||
bin
|
||||
obj
|
12
ProtocolLib/PacketIO/IPacketFile.cs
Normal file
12
ProtocolLib/PacketIO/IPacketFile.cs
Normal file
@ -0,0 +1,12 @@
|
||||
namespace PacketIO;
|
||||
|
||||
public interface IPacketFile : IDisposable
|
||||
{
|
||||
public PacketFileType GetFileType();
|
||||
|
||||
public string GetFileName();
|
||||
|
||||
public long GetFileSize();
|
||||
|
||||
public IEnumerable<IPacketFile> EnumerableFiles();
|
||||
}
|
53
ProtocolLib/PacketIO/PacketDirectory.cs
Normal file
53
ProtocolLib/PacketIO/PacketDirectory.cs
Normal file
@ -0,0 +1,53 @@
|
||||
namespace PacketIO;
|
||||
|
||||
public class PacketDirectory : IPacketFile
|
||||
{
|
||||
private string fileName;
|
||||
private List<IPacketFile> files;
|
||||
|
||||
public PacketDirectory(string fileName, params IPacketFile[] files)
|
||||
{
|
||||
this.fileName = fileName;
|
||||
this.files = new List<IPacketFile>();
|
||||
this.files.AddRange(files);
|
||||
}
|
||||
|
||||
public IEnumerable<IPacketFile> EnumerableFiles()
|
||||
{
|
||||
return this.files.ToList();
|
||||
}
|
||||
|
||||
public string GetFileName()
|
||||
{
|
||||
return this.fileName;
|
||||
}
|
||||
|
||||
public long GetFileSize()
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
public PacketFileType GetFileType()
|
||||
{
|
||||
return PacketFileType.Directory;
|
||||
}
|
||||
|
||||
public long AddFile(IPacketFile file)
|
||||
{
|
||||
this.files.Add(file);
|
||||
return this.files.LongCount();
|
||||
}
|
||||
|
||||
public bool RemoveFile(IPacketFile file)
|
||||
{
|
||||
return this.files.Remove(file);
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
foreach (IPacketFile file in this.EnumerableFiles())
|
||||
{
|
||||
file.Dispose();
|
||||
}
|
||||
}
|
||||
}
|
66
ProtocolLib/PacketIO/PacketFile.cs
Normal file
66
ProtocolLib/PacketIO/PacketFile.cs
Normal file
@ -0,0 +1,66 @@
|
||||
namespace PacketIO;
|
||||
|
||||
public class PacketFile : IPacketFile
|
||||
{
|
||||
private string fileName;
|
||||
private TempFile? tempFile;
|
||||
private Stream stream;
|
||||
|
||||
public PacketFile(string fileName)
|
||||
{
|
||||
this.fileName = fileName;
|
||||
this.tempFile = new TempFile();
|
||||
this.stream = tempFile.Open(FileMode.Open);
|
||||
}
|
||||
|
||||
public PacketFile(string fileName, Stream stream)
|
||||
{
|
||||
this.fileName = fileName;
|
||||
this.stream = stream;
|
||||
}
|
||||
|
||||
public IEnumerable<IPacketFile> EnumerableFiles()
|
||||
{
|
||||
return [];
|
||||
}
|
||||
|
||||
public string GetFileName()
|
||||
{
|
||||
return this.fileName;
|
||||
}
|
||||
|
||||
public long GetFileSize()
|
||||
{
|
||||
return this.stream.Length;
|
||||
}
|
||||
|
||||
public PacketFileType GetFileType()
|
||||
{
|
||||
return PacketFileType.File;
|
||||
}
|
||||
|
||||
public async ValueTask<int> ReadAsync(byte[] buffer, int offset, int count)
|
||||
{
|
||||
int read = await this.stream.ReadAsync(buffer, offset, count).ConfigureAwait(false);
|
||||
return read;
|
||||
}
|
||||
|
||||
public async ValueTask WriteAsync(byte[] buffer, int offset, int count)
|
||||
{
|
||||
await this.stream.WriteAsync(buffer, offset, count).ConfigureAwait(false);
|
||||
}
|
||||
|
||||
public void Seek(long offset, SeekOrigin origin)
|
||||
{
|
||||
this.stream.Seek(offset, origin);
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
this.stream.Dispose();
|
||||
if (this.tempFile is object)
|
||||
{
|
||||
this.tempFile.Dispose();
|
||||
}
|
||||
}
|
||||
}
|
11
ProtocolLib/PacketIO/PacketFileType.cs
Normal file
11
ProtocolLib/PacketIO/PacketFileType.cs
Normal file
@ -0,0 +1,11 @@
|
||||
namespace PacketIO;
|
||||
|
||||
[Flags]
|
||||
public enum PacketFileType : byte
|
||||
{
|
||||
None = 0,
|
||||
|
||||
File = 0x02 << 0,
|
||||
|
||||
Directory = 0x02 << 1,
|
||||
}
|
21
ProtocolLib/PacketIO/TempFile.cs
Normal file
21
ProtocolLib/PacketIO/TempFile.cs
Normal file
@ -0,0 +1,21 @@
|
||||
namespace PacketIO;
|
||||
|
||||
public class TempFile : IDisposable
|
||||
{
|
||||
public string FileName { get; }
|
||||
|
||||
public TempFile()
|
||||
{
|
||||
FileName = Path.GetTempFileName();
|
||||
}
|
||||
|
||||
public Stream Open(FileMode mode)
|
||||
{
|
||||
return new FileStream(FileName, mode);
|
||||
}
|
||||
|
||||
public void Dispose()
|
||||
{
|
||||
File.Delete(FileName);
|
||||
}
|
||||
}
|
150
ProtocolLib/PacketIO/TreePacketFile.cs
Normal file
150
ProtocolLib/PacketIO/TreePacketFile.cs
Normal file
@ -0,0 +1,150 @@
|
||||
using System.Text;
|
||||
|
||||
namespace PacketIO;
|
||||
|
||||
public class TreePacketFile
|
||||
{
|
||||
public static async ValueTask WriteTreeFiles(Stream stream, params IPacketFile[] files)
|
||||
{
|
||||
foreach (IPacketFile file in files)
|
||||
{
|
||||
await WriteTreeFile(stream, file).ConfigureAwait(false);
|
||||
}
|
||||
}
|
||||
|
||||
public static async ValueTask WriteTreeFile(Stream stream, IPacketFile file)
|
||||
{
|
||||
byte fileType = (byte) file.GetFileType();
|
||||
string fileName = file.GetFileName();
|
||||
long fileSize = file.GetFileSize();
|
||||
|
||||
byte[] fileNameBuffer = Encoding.UTF8.GetBytes(fileName);
|
||||
|
||||
byte[] buffer = new byte[sizeof(byte) + sizeof(int) + sizeof(long) + fileNameBuffer.Length];
|
||||
buffer[0] = fileType;
|
||||
Array.Copy(BitConverter.GetBytes(fileNameBuffer.Length), 0, buffer, 1, sizeof(int));
|
||||
Array.Copy(BitConverter.GetBytes(fileSize), 0, buffer, 5, sizeof(long));
|
||||
Array.Copy(fileNameBuffer, 0, buffer, 13, fileNameBuffer.Length);
|
||||
|
||||
await stream.WriteAsync(buffer, 0, buffer.Length).ConfigureAwait(false);
|
||||
|
||||
switch (file.GetFileType())
|
||||
{
|
||||
case PacketFileType.Directory:
|
||||
{
|
||||
long fileCount = file.EnumerableFiles().LongCount();
|
||||
buffer = BitConverter.GetBytes(fileCount);
|
||||
await stream.WriteAsync(buffer, 0, sizeof(long));
|
||||
|
||||
await WriteTreeFiles(stream, file.EnumerableFiles().ToArray()).ConfigureAwait(false);
|
||||
break;
|
||||
}
|
||||
case PacketFileType.File:
|
||||
{
|
||||
int c;
|
||||
buffer = new byte[8192];
|
||||
PacketFile packetFile = (PacketFile) file;
|
||||
packetFile.Seek(0, SeekOrigin.Begin);
|
||||
do
|
||||
{
|
||||
c = await packetFile.ReadAsync(buffer, 0, buffer.Length).ConfigureAwait(false);
|
||||
|
||||
await stream.WriteAsync(buffer, 0, c).ConfigureAwait(false);
|
||||
} while (c > 0);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
throw new NotImplementedException();
|
||||
}
|
||||
}
|
||||
|
||||
public static async ValueTask<IEnumerable<IPacketFile>> ReadTreeFiles(Stream stream)
|
||||
{
|
||||
List<IPacketFile> files = new List<IPacketFile>();
|
||||
|
||||
IPacketFile? file = null;
|
||||
do
|
||||
{
|
||||
file = await ReadTreeFile(stream).ConfigureAwait(false);
|
||||
if (file is object)
|
||||
{
|
||||
files.Add(file);
|
||||
}
|
||||
} while (file is object);
|
||||
|
||||
return files;
|
||||
}
|
||||
|
||||
public static async ValueTask<IPacketFile?> ReadTreeFile(Stream stream)
|
||||
{
|
||||
byte[] buffer = new byte[sizeof(byte) + sizeof(int) + sizeof(long)];
|
||||
int c = await stream.ReadAsync(buffer, 0, buffer.Length).ConfigureAwait(false);
|
||||
if (c <= 0)
|
||||
{
|
||||
return null;
|
||||
}
|
||||
|
||||
PacketFileType fileType = (PacketFileType) buffer[0];
|
||||
int fileNameBufferSize = BitConverter.ToInt32(buffer, 1);
|
||||
long fileSize = BitConverter.ToInt64(buffer, 5);
|
||||
|
||||
buffer = new byte[fileNameBufferSize];
|
||||
c = await stream.ReadAsync(buffer, 0, buffer.Length).ConfigureAwait(false);
|
||||
if (c <= 0)
|
||||
{
|
||||
return null;
|
||||
}
|
||||
|
||||
string fileName = Encoding.UTF8.GetString(buffer);
|
||||
|
||||
switch (fileType)
|
||||
{
|
||||
case PacketFileType.Directory:
|
||||
{
|
||||
buffer = new byte[sizeof(long)];
|
||||
c = await stream.ReadAsync(buffer, 0, buffer.Length);
|
||||
if (c <= 0)
|
||||
{
|
||||
return null;
|
||||
}
|
||||
|
||||
long fileCount = BitConverter.ToInt64(buffer, 0);
|
||||
|
||||
PacketDirectory directory = new PacketDirectory(fileName);
|
||||
for (long i = 0 ; i < fileCount ; i++)
|
||||
{
|
||||
IPacketFile? file = await ReadTreeFile(stream).ConfigureAwait(false);
|
||||
if (file is null)
|
||||
{
|
||||
break;
|
||||
}
|
||||
|
||||
directory.AddFile(file);
|
||||
}
|
||||
|
||||
return directory;
|
||||
}
|
||||
case PacketFileType.File:
|
||||
{
|
||||
PacketFile file = new PacketFile(fileName);
|
||||
|
||||
buffer = new byte[8192];
|
||||
long remainedSize = fileSize;
|
||||
long readed = 0;
|
||||
do
|
||||
{
|
||||
int leftSize = (int) Math.Min(buffer.Length, fileSize - readed);
|
||||
c = await stream.ReadAsync(buffer, 0, leftSize).ConfigureAwait(false);
|
||||
await file.WriteAsync(buffer, 0, c).ConfigureAwait(false);
|
||||
remainedSize -= c;
|
||||
readed += c;
|
||||
} while (remainedSize > 0);
|
||||
|
||||
file.Seek(0, SeekOrigin.Begin);
|
||||
return file;
|
||||
}
|
||||
default:
|
||||
throw new NotImplementedException();
|
||||
}
|
||||
}
|
||||
}
|
71
ProtocolLib/Program.cs
Normal file
71
ProtocolLib/Program.cs
Normal file
@ -0,0 +1,71 @@
|
||||
using PacketIO;
|
||||
|
||||
namespace ProtocolLib;
|
||||
|
||||
public class Program
|
||||
{
|
||||
public static void Main(string[] args)
|
||||
{
|
||||
// Make tree file.
|
||||
using (FileStream fs = new FileStream("tree.bin", FileMode.OpenOrCreate))
|
||||
{
|
||||
DirectoryInfo directoryInfo = new DirectoryInfo("i");
|
||||
if (directoryInfo.Exists)
|
||||
{
|
||||
foreach (FileInfo fileInfo in directoryInfo.GetFiles())
|
||||
{
|
||||
using (PacketFile file = new PacketFile(fileInfo.Name, fileInfo.Open(FileMode.Open, FileAccess.ReadWrite)))
|
||||
{
|
||||
TreePacketFile.WriteTreeFile(fs, file).ConfigureAwait(false).GetAwaiter().GetResult();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Get files from tree file.
|
||||
using (FileStream fs = new FileStream("tree.bin", FileMode.OpenOrCreate))
|
||||
{
|
||||
IEnumerable<IPacketFile> files = TreePacketFile.ReadTreeFiles(fs).ConfigureAwait(false).GetAwaiter().GetResult();
|
||||
foreach (IPacketFile ifile in files)
|
||||
{
|
||||
PacketFile file = (PacketFile) ifile;
|
||||
using (FileStream outStream = new FileStream(file.GetFileName(), FileMode.OpenOrCreate))
|
||||
{
|
||||
byte[] buffer = new byte[8192];
|
||||
int c;
|
||||
while ((c = file.ReadAsync(buffer, 0, buffer.Length).ConfigureAwait(false).GetAwaiter().GetResult()) > 0)
|
||||
{
|
||||
outStream.Write(buffer, 0, c);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// Paket test.
|
||||
using (MemoryStream ms = new MemoryStream())
|
||||
{
|
||||
PacketStream packetStream = new PacketStream(ms);
|
||||
|
||||
byte[] buffer = System.Text.Encoding.UTF8.GetBytes("Hello, World!!!");
|
||||
foreach (Packet packet in PacketFactory.CreatePacket(DataOperator.Connect, DataType.Binary, buffer))
|
||||
{
|
||||
packetStream.WritePacket(packet).ConfigureAwait(false).GetAwaiter().GetResult();
|
||||
}
|
||||
|
||||
ms.Seek(0, SeekOrigin.Begin);
|
||||
|
||||
Packet? packet1 = null;
|
||||
do
|
||||
{
|
||||
packet1 = packetStream.ReadPacket().ConfigureAwait(false).GetAwaiter().GetResult();
|
||||
if (packet1 is object)
|
||||
{
|
||||
Console.Write($"{System.Text.Encoding.UTF8.GetString(packet1.Payload)}");
|
||||
}
|
||||
} while (packet1 is object);
|
||||
|
||||
Console.WriteLine();
|
||||
}
|
||||
}
|
||||
}
|
10
ProtocolLib/ProtocolLib.csproj
Normal file
10
ProtocolLib/ProtocolLib.csproj
Normal file
@ -0,0 +1,10 @@
|
||||
<Project Sdk="Microsoft.NET.Sdk">
|
||||
|
||||
<PropertyGroup>
|
||||
<OutputType>Library</OutputType>
|
||||
<TargetFramework>net8.0</TargetFramework>
|
||||
<ImplicitUsings>enable</ImplicitUsings>
|
||||
<Nullable>enable</Nullable>
|
||||
</PropertyGroup>
|
||||
|
||||
</Project>
|
25
ProtocolLib/ProtocolLib.sln
Normal file
25
ProtocolLib/ProtocolLib.sln
Normal file
@ -0,0 +1,25 @@
|
||||
|
||||
Microsoft Visual Studio Solution File, Format Version 12.00
|
||||
# Visual Studio Version 17
|
||||
VisualStudioVersion = 17.5.002.0
|
||||
MinimumVisualStudioVersion = 10.0.40219.1
|
||||
Project("{9A19103F-16F7-4668-BE54-9A1E7A4F7556}") = "ProtocolLib", "ProtocolLib.csproj", "{BB1F2133-E83C-4975-A685-D2DCE2FC24C9}"
|
||||
EndProject
|
||||
Global
|
||||
GlobalSection(SolutionConfigurationPlatforms) = preSolution
|
||||
Debug|Any CPU = Debug|Any CPU
|
||||
Release|Any CPU = Release|Any CPU
|
||||
EndGlobalSection
|
||||
GlobalSection(ProjectConfigurationPlatforms) = postSolution
|
||||
{BB1F2133-E83C-4975-A685-D2DCE2FC24C9}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
|
||||
{BB1F2133-E83C-4975-A685-D2DCE2FC24C9}.Debug|Any CPU.Build.0 = Debug|Any CPU
|
||||
{BB1F2133-E83C-4975-A685-D2DCE2FC24C9}.Release|Any CPU.ActiveCfg = Release|Any CPU
|
||||
{BB1F2133-E83C-4975-A685-D2DCE2FC24C9}.Release|Any CPU.Build.0 = Release|Any CPU
|
||||
EndGlobalSection
|
||||
GlobalSection(SolutionProperties) = preSolution
|
||||
HideSolutionNode = FALSE
|
||||
EndGlobalSection
|
||||
GlobalSection(ExtensibilityGlobals) = postSolution
|
||||
SolutionGuid = {AD8CE391-A853-498C-A2E6-7A83701164D4}
|
||||
EndGlobalSection
|
||||
EndGlobal
|
15
ProtocolLib/ProtocolLib/ByteConverter.cs
Normal file
15
ProtocolLib/ProtocolLib/ByteConverter.cs
Normal file
@ -0,0 +1,15 @@
|
||||
namespace ProtocolLib;
|
||||
|
||||
internal static class ByteConverter
|
||||
{
|
||||
public static byte[] Convert(bool value)
|
||||
{
|
||||
byte[] buffer = BitConverter.GetBytes(value);
|
||||
if (BitConverter.IsLittleEndian)
|
||||
{
|
||||
Array.Reverse(buffer);
|
||||
}
|
||||
|
||||
return buffer;
|
||||
}
|
||||
}
|
12
ProtocolLib/ProtocolLib/DataOperator.cs
Normal file
12
ProtocolLib/ProtocolLib/DataOperator.cs
Normal file
@ -0,0 +1,12 @@
|
||||
namespace ProtocolLib;
|
||||
|
||||
public enum DataOperator : byte
|
||||
{
|
||||
None = 0x00,
|
||||
|
||||
Connect = 0x02 << 0,
|
||||
|
||||
Request = 0x02 << 1,
|
||||
|
||||
Response = 0x02 << 2,
|
||||
}
|
8
ProtocolLib/ProtocolLib/DataType.cs
Normal file
8
ProtocolLib/ProtocolLib/DataType.cs
Normal file
@ -0,0 +1,8 @@
|
||||
namespace ProtocolLib;
|
||||
|
||||
public enum DataType : byte
|
||||
{
|
||||
None = 0x00,
|
||||
|
||||
Binary = 0x02 << 0,
|
||||
}
|
79
ProtocolLib/ProtocolLib/PackeFactory.cs
Normal file
79
ProtocolLib/ProtocolLib/PackeFactory.cs
Normal file
@ -0,0 +1,79 @@
|
||||
namespace ProtocolLib;
|
||||
|
||||
public static class PacketFactory
|
||||
{
|
||||
public static readonly int MAX_PAYLOAD_SIZE = 810;
|
||||
|
||||
/// <summary>
|
||||
/// 指定されたバイト配列からパケットを作成する。
|
||||
/// </summary>
|
||||
/// <param name="dataOperator">オペレータの種類</param>
|
||||
/// <param name="dataType">データの種類</param>
|
||||
/// <param name="data">バイトデータ</param>
|
||||
/// <returns></returns>
|
||||
public static IEnumerable<Packet> CreatePacket(DataOperator dataOperator, DataType dataType, byte[] data)
|
||||
{
|
||||
List<Packet> packets = new List<Packet>();
|
||||
|
||||
long packetNumber = 0;
|
||||
int remainedSize = data.Length;
|
||||
do
|
||||
{
|
||||
int packSize = Math.Min(MAX_PAYLOAD_SIZE, remainedSize);
|
||||
byte[] payload = data.AsSpan(data.Length - remainedSize, packSize).ToArray();
|
||||
|
||||
Packet packet = new Packet()
|
||||
{
|
||||
Operator = dataOperator,
|
||||
Type = dataType,
|
||||
PacketNumber = packetNumber,
|
||||
PayloadSize = packSize,
|
||||
Payload = payload,
|
||||
};
|
||||
|
||||
packets.Add(packet);
|
||||
remainedSize -= packSize;
|
||||
} while (remainedSize > 0);
|
||||
|
||||
return packets;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 指定されたバイト配列からパケットを作成する。
|
||||
/// </summary>
|
||||
/// <param name="dataOperator">オペレータの種類</param>
|
||||
/// <param name="dataType">データの種類</param>
|
||||
/// <param name="data">バイトデータ</param>
|
||||
/// <returns></returns>
|
||||
public static IEnumerable<Packet> CreatePacket(DataOperator dataOperator, DataType dataType, Stream stream)
|
||||
{
|
||||
List<Packet> packets = new List<Packet>();
|
||||
|
||||
long packetNumber = 0;
|
||||
long remainedSize = stream.Length;
|
||||
do
|
||||
{
|
||||
int packSize = (int) Math.Min(MAX_PAYLOAD_SIZE, remainedSize);
|
||||
byte[] payload = new byte[packSize];
|
||||
int c = stream.Read(payload, 0, payload.Length);
|
||||
if (c <= 0)
|
||||
{
|
||||
throw new IOException();
|
||||
}
|
||||
|
||||
Packet packet = new Packet()
|
||||
{
|
||||
Operator = dataOperator,
|
||||
Type = dataType,
|
||||
PacketNumber = packetNumber,
|
||||
PayloadSize = packSize,
|
||||
Payload = payload,
|
||||
};
|
||||
|
||||
packets.Add(packet);
|
||||
remainedSize -= packSize;
|
||||
} while (remainedSize > 0);
|
||||
|
||||
return packets;
|
||||
}
|
||||
}
|
14
ProtocolLib/ProtocolLib/Packet.cs
Normal file
14
ProtocolLib/ProtocolLib/Packet.cs
Normal file
@ -0,0 +1,14 @@
|
||||
namespace ProtocolLib;
|
||||
|
||||
public class Packet
|
||||
{
|
||||
public DataOperator Operator { get; set; }
|
||||
|
||||
public DataType Type { get; set; }
|
||||
|
||||
public long PacketNumber { get; set;}
|
||||
|
||||
public int PayloadSize { get; set; }
|
||||
|
||||
public byte[] Payload { get; set; }
|
||||
}
|
70
ProtocolLib/ProtocolLib/PacketStream.cs
Normal file
70
ProtocolLib/ProtocolLib/PacketStream.cs
Normal file
@ -0,0 +1,70 @@
|
||||
namespace ProtocolLib;
|
||||
|
||||
public class PacketStream
|
||||
{
|
||||
private Stream stream;
|
||||
|
||||
public PacketStream(Stream stream)
|
||||
{
|
||||
this.stream = stream;
|
||||
}
|
||||
|
||||
public async ValueTask WritePacket(Packet packet)
|
||||
{
|
||||
byte[] buffer = this.PacketToBytes(packet);
|
||||
await this.stream.WriteAsync(buffer, 0, buffer.Length).ConfigureAwait(false);
|
||||
}
|
||||
|
||||
public async ValueTask<Packet?> ReadPacket()
|
||||
{
|
||||
byte[] buffer = new byte[2 + sizeof(long) + sizeof(int)];
|
||||
int r = await this.stream.ReadAsync(buffer, 0, buffer.Length).ConfigureAwait(false);
|
||||
if (r <= 0)
|
||||
{
|
||||
return null;
|
||||
}
|
||||
|
||||
DataOperator dataOperator = (DataOperator) buffer[0];
|
||||
DataType dataType = (DataType) buffer[1];
|
||||
long packetNumber = BitConverter.ToInt64(buffer, 2);
|
||||
int payloadSize = BitConverter.ToInt32(buffer, 10);
|
||||
if (payloadSize < 0)
|
||||
{
|
||||
throw new IOException();
|
||||
}
|
||||
|
||||
buffer = new byte[payloadSize];
|
||||
r = await this.stream.ReadAsync(buffer, 0, buffer.Length).ConfigureAwait(false);
|
||||
if (r <= 0)
|
||||
{
|
||||
throw new IOException();
|
||||
}
|
||||
|
||||
Packet packet = new Packet()
|
||||
{
|
||||
Operator = dataOperator,
|
||||
Type = dataType,
|
||||
PacketNumber = packetNumber,
|
||||
PayloadSize = payloadSize,
|
||||
Payload = buffer,
|
||||
};
|
||||
|
||||
return packet;
|
||||
}
|
||||
|
||||
private byte[] PacketToBytes(Packet packet)
|
||||
{
|
||||
byte[] buffer = new byte[2 + sizeof(long) + sizeof(int) + packet.PayloadSize];
|
||||
|
||||
buffer[0] = (byte) packet.Operator;
|
||||
buffer[1] = (byte) packet.Type;
|
||||
|
||||
byte[] temp = BitConverter.GetBytes(packet.PacketNumber);
|
||||
Array.Copy(temp, 0, buffer, 2, temp.Length);
|
||||
temp = BitConverter.GetBytes(packet.PayloadSize);
|
||||
Array.Copy(temp, 0, buffer, 10, temp.Length);
|
||||
Array.Copy(packet.Payload, 0, buffer, 14, packet.PayloadSize);
|
||||
|
||||
return buffer;
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user