kev/rinseoff
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

switch to enum for error codes

Browse Source
This commit is contained in:
Kevin Froman 2020-12-21 04:16:58 +00:00
parent 8a74b57724
commit cd723272d4
1 changed files with 34 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
rinseoffcli/RinseOffCLI.cs
View File

@ -1,21 +1,37 @@
using System; using System;
using System.IO; using System.IO;
using System.Text;
using System.Collections.Generic; using System.Collections.Generic;
using rinseoff; using rinseoff;
namespace rinseoffcli namespace rinseoffcli
{ {
class Program class Program
{ {
public static string version = "0.0.0";
enum ErrorCode : byte
{
InvaildArgs = 1,
NoDataFileFound,
NoPermissionToReadDataFile,
FailedToReadDataFile,
FailedToWriteDataFile,
InvalidKeyFileSize,
InvalidKeyFile,
KeyFileNotFound,
NoPermissionToWriteKeyFile,
FailedToReadKeyFile,
FailedToWriteKeyFile
}
static void showHelp(int exitCode = 1){
static void showHelp(ErrorCode exitCode = ErrorCode.InvaildArgs){
Console.WriteLine("RinseOff " + version + " High level secure-erasure utility");
Console.WriteLine("Must specify keygen <key path> or store/load, then a file name followed by a key file.\nFormat: <verb> <data file> <key file>"); Console.WriteLine("Must specify keygen <key path> or store/load, then a file name followed by a key file.\nFormat: <verb> <data file> <key file>");
Environment.Exit(exitCode); Environment.Exit((int) exitCode);
} }
static void validateArgCount(string[] args, int count){ static void validateArgCount(string[] args, int count){
if (args.Length != count){ if (args.Length != count){
stderrWrite("Invalid number of arguments"); stderrWrite("Invalid number of arguments");
showHelp(2); showHelp(ErrorCode.InvaildArgs);
} }
} }
static void stderrWrite(string msg){ static void stderrWrite(string msg){
@ -34,15 +50,15 @@ namespace rinseoffcli
} }
catch(FileNotFoundException){ catch(FileNotFoundException){
stderrWrite(file + " is not found"); stderrWrite(file + " is not found");
Environment.Exit(9); Environment.Exit((int)ErrorCode.NoDataFileFound);
} }
catch(UnauthorizedAccessException){ catch(UnauthorizedAccessException){
stderrWrite("No permssion to read " + file); stderrWrite("No permssion to read " + file);
Environment.Exit(10); Environment.Exit((int)ErrorCode.NoPermissionToReadDataFile);
} }
catch(IOException){ catch(IOException){
stderrWrite("Failed to read " + file); stderrWrite("Failed to read " + file);
Environment.Exit(11); Environment.Exit((int)ErrorCode.FailedToReadDataFile);
} }
return bytesToRead; return bytesToRead;
} }
@ -56,11 +72,11 @@ namespace rinseoffcli
} }
catch(Sodium.Exceptions.KeyOutOfRangeException){ catch(Sodium.Exceptions.KeyOutOfRangeException){
stderrWrite("Keyfile is not appropriate size for key"); stderrWrite("Keyfile is not appropriate size for key");
Environment.Exit(4); Environment.Exit((int)ErrorCode.InvalidKeyFileSize);
} }
catch(System.Security.Cryptography.CryptographicException){ catch(System.Security.Cryptography.CryptographicException){
stderrWrite("Could not decrypt " + filepath + " with " + keypath); stderrWrite("Could not decrypt " + filepath + " with " + keypath);
Environment.Exit(12); Environment.Exit((int)ErrorCode.InvalidKeyFile);
} }
// print the plaintext and exit // print the plaintext and exit
foreach(byte b in plaintext){ foreach(byte b in plaintext){
@ -93,15 +109,15 @@ namespace rinseoffcli
} }
catch(FileNotFoundException){ catch(FileNotFoundException){
stderrWrite("Key file " + keypath + " does not exist"); stderrWrite("Key file " + keypath + " does not exist");
Environment.Exit(3); Environment.Exit((int)ErrorCode.KeyFileNotFound);
} }
catch(Sodium.Exceptions.KeyOutOfRangeException){ catch(Sodium.Exceptions.KeyOutOfRangeException){
stderrWrite("Keyfile is not appropriate size for key"); stderrWrite("Keyfile is not appropriate size for key");
Environment.Exit(4); Environment.Exit((int)ErrorCode.InvalidKeyFileSize);
} }
catch(IOException){ catch(IOException){
stderrWrite("Failed to read key file " + keypath); stderrWrite("Failed to read key file " + keypath);
Environment.Exit(5); Environment.Exit((int)ErrorCode.FailedToReadKeyFile);
} }
try{ try{
File.WriteAllBytes(filepath, encryptedInput); File.WriteAllBytes(filepath, encryptedInput);
@ -112,11 +128,11 @@ namespace rinseoffcli
} }
catch(DirectoryNotFoundException){ catch(DirectoryNotFoundException){
stderrWrite("Output path " + filepath + " not found"); stderrWrite("Output path " + filepath + " not found");
Environment.Exit(7); Environment.Exit((int)ErrorCode.NoDataFileFound);
} }
catch(IOException){ catch(IOException){
stderrWrite("Could not write to " + filepath); stderrWrite("Could not write to " + filepath);
Environment.Exit(8); Environment.Exit((int)ErrorCode.FailedToWriteDataFile);
} }
} }
static void Main(string[] args) static void Main(string[] args)
@ -133,15 +149,15 @@ namespace rinseoffcli
} }
catch(UnauthorizedAccessException){ catch(UnauthorizedAccessException){
stderrWrite("Cannot write to key file due to lack of perms " + args[1]); stderrWrite("Cannot write to key file due to lack of perms " + args[1]);
Environment.Exit(6); Environment.Exit((int)ErrorCode.NoPermissionToWriteKeyFile);
} }
catch(DirectoryNotFoundException){ catch(DirectoryNotFoundException){
stderrWrite("Path not found " + args[1]); stderrWrite("Path not found " + args[1]);
Environment.Exit(6); Environment.Exit((int)ErrorCode.KeyFileNotFound);
} }
catch(IOException){ catch(IOException){
stderrWrite("Path not found " + args[1]); stderrWrite("Error writing keyfile " + args[1]);
Environment.Exit(6); Environment.Exit((int)ErrorCode.FailedToWriteKeyFile);
} }
break; break;
case "store": case "store":