CodeMirror/Picocrypt
1
0
Fork 0
mirror of https://github.com/HACKERALERT/Picocrypt.git synced 2024-05-16 10:12:55 +12:00
Code Issues Projects Releases Wiki Activity

Add option to compress a single file

Browse source
This commit is contained in:
Evan Su 2022-05-19 23:56:45 -04:00 committed by GitHub
parent ba275ff342
commit 9aa62adf51
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 29 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
src/Picocrypt.go
View file

@ -471,10 +471,16 @@ func draw() {
giu.Checkbox("Paranoid mode", &paranoid),
giu.Tooltip("Provides the highest level of security attainable."),
giu.Dummy(-170, 0),
giu.Style().SetDisabled(!(len(allFiles) > 1 || len(onlyFolders) > 0)).To(
giu.Checkbox("Compress files", &compress),
giu.Tooltip("Compress files with Deflate before encrypting."),
),
giu.Checkbox("Compress files", &compress).OnChange(func() {
if !(len(allFiles) > 1 || len(onlyFolders) > 0) {
if compress {
outputFile = filepath.Join(filepath.Dir(outputFile), "Encrypted") + ".zip.pcv"
} else {
outputFile = filepath.Join(filepath.Dir(outputFile), filepath.Base(inputFile)) + ".pcv"
}
}
}),
giu.Tooltip("Compress files with Deflate before encrypting."),
).Build()
giu.Row(
@ -538,7 +544,7 @@ func draw() {
// Prefill the filename
tmp := strings.TrimSuffix(filepath.Base(outputFile), ".pcv")
f.SetInitFilename(strings.TrimSuffix(tmp, filepath.Ext(tmp)))
if mode == "encrypt" && (len(allFiles) > 1 || len(onlyFolders) > 0) {
if mode == "encrypt" && (len(allFiles) > 1 || len(onlyFolders) > 0 || compress) {
f.SetInitFilename("Encrypted")
}
@ -550,7 +556,7 @@ func draw() {
// Add the correct extensions
if mode == "encrypt" {
if len(allFiles) > 1 || len(onlyFolders) > 0 {
if len(allFiles) > 1 || len(onlyFolders) > 0 || compress {
file += ".zip.pcv"
} else {
file += filepath.Ext(inputFile) + ".pcv"
@ -881,7 +887,13 @@ func work() {
var authTag []byte // 64-byte authentication tag (BLAKE2b or HMAC-SHA3)
// Combine/compress all files into a .zip file if needed
if len(allFiles) > 1 || len(onlyFolders) > 0 {
if len(allFiles) > 1 || len(onlyFolders) > 0 || compress {
// Consider case where compressing only one file
files := allFiles
if len(allFiles) == 0 {
files = onlyFiles
}
// Get the root directory of the selected files
var rootDir string
if len(onlyFolders) > 0 {
@ -900,7 +912,7 @@ func work() {
// Calculate total size of uncompressed files
compressTotal = 0
for _, path := range allFiles {
for _, path := range files {
stat, _ := os.Stat(path)
compressTotal += stat.Size()
}
@ -910,8 +922,9 @@ func work() {
compressStart = time.Now()
// Add each file to the .zip
for i, path := range allFiles {
progressInfo = fmt.Sprintf("%d/%d", i+1, len(allFiles))
for i, path := range files {
progressInfo = fmt.Sprintf("%d/%d", i+1, len(files))
giu.Update()
// Create file info header (size, last modified, etc.)
@ -1350,7 +1363,7 @@ func work() {
cancel()
fin.Close()
fout.Close()
if recombine || len(allFiles) > 1 || len(onlyFolders) > 0 {
if recombine || len(allFiles) > 1 || len(onlyFolders) > 0 || compress {
os.Remove(inputFile)
}
os.Remove(outputFile)
@ -1476,12 +1489,13 @@ func work() {
serpent.XORKeyStream(dst, src)
}
}
_, err = fout.Write(dst)
if err != nil {
insufficientSpace()
fin.Close()
fout.Close()
if recombine || len(allFiles) > 1 || len(onlyFolders) > 0 {
if recombine || len(allFiles) > 1 || len(onlyFolders) > 0 || compress {
os.Remove(inputFile)
}
os.Remove(outputFile)
@ -1610,7 +1624,7 @@ func work() {
cancel()
fin.Close()
fout.Close()
if len(allFiles) > 1 || len(onlyFolders) > 0 {
if len(allFiles) > 1 || len(onlyFolders) > 0 || compress {
os.Remove(inputFile)
}
os.Remove(outputFile)
@ -1630,7 +1644,7 @@ func work() {
insufficientSpace()
fin.Close()
fout.Close()
if len(allFiles) > 1 || len(onlyFolders) > 0 {
if len(allFiles) > 1 || len(onlyFolders) > 0 || compress {
os.Remove(inputFile)
}
os.Remove(outputFile)
@ -1681,7 +1695,7 @@ func work() {
}
// Delete the temporary .zip used to encrypt files
if len(allFiles) > 1 || len(onlyFolders) > 0 {
if len(allFiles) > 1 || len(onlyFolders) > 0 || compress {
os.Remove(inputFile)
}