Detect if insufficient disk space

This commit is contained in:
Evan Su 2022-05-17 20:44:03 -04:00 committed by GitHub
parent 1e9bcefe5c
commit 77691ecf09
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 82 additions and 18 deletions

View File

@ -564,6 +564,8 @@ func draw() {
}
}
outputFile = file
mainStatus = "Ready."
mainStatusColor = WHITE
}).Build()
giu.Tooltip("Save the output with a custom name and path.").Build()
}),
@ -663,7 +665,7 @@ func onDrop(names []string) {
if stat.IsDir() {
folders++
mode = "encrypt"
inputLabel = "1 folder selected."
inputLabel = "1 folder."
startLabel = "Encrypt"
onlyFolders = append(onlyFolders, names[0])
inputFile = filepath.Join(filepath.Dir(names[0]), "Encrypted") + ".zip"
@ -684,7 +686,7 @@ func onDrop(names []string) {
// Decide if encrypting or decrypting
if strings.HasSuffix(names[0], ".pcv") || isSplit {
mode = "decrypt"
inputLabel = "Volume selected for decryption."
inputLabel = "Volume for decryption."
startLabel = "Decrypt"
commentsLabel = "Comments (read-only):"
commentsDisabled = true
@ -696,6 +698,17 @@ func onDrop(names []string) {
inputFile = names[0]
outputFile = names[0][:ind]
recombine = true
totalFiles := 0
// Find out the number of splitted chunks
for {
stat, err := os.Stat(fmt.Sprintf("%s.%d", inputFile, totalFiles))
if err != nil {
break
}
totalFiles++
size += int(stat.Size())
}
} else {
outputFile = names[0][:len(names[0])-4]
}
@ -770,7 +783,7 @@ func onDrop(names []string) {
}
} else { // One file was dropped for encryption
mode = "encrypt"
inputLabel = "1 file selected."
inputLabel = "1 file."
startLabel = "Encrypt"
inputFile = names[0]
outputFile = names[0] + ".pcv"
@ -779,7 +792,9 @@ func onDrop(names []string) {
// Add the file
onlyFiles = append(onlyFiles, names[0])
inputFile = names[0]
size += int(stat.Size())
if !isSplit {
size += int(stat.Size())
}
}
} else { // There are multiple dropped items
mode = "encrypt"
@ -804,18 +819,18 @@ func onDrop(names []string) {
// Update UI with the number of files and folders selected
if folders == 0 {
inputLabel = fmt.Sprintf("%d files selected.", files)
inputLabel = fmt.Sprintf("%d files.", files)
} else if files == 0 {
inputLabel = fmt.Sprintf("%d folders selected.", folders)
inputLabel = fmt.Sprintf("%d folders.", folders)
} else {
if files == 1 && folders > 1 {
inputLabel = fmt.Sprintf("1 file and %d folders selected.", folders)
inputLabel = fmt.Sprintf("1 file and %d folders.", folders)
} else if folders == 1 && files > 1 {
inputLabel = fmt.Sprintf("%d files and 1 folder selected.", files)
inputLabel = fmt.Sprintf("%d files and 1 folder.", files)
} else if folders == 1 && files == 1 {
inputLabel = "1 file and 1 folder selected."
inputLabel = "1 file and 1 folder."
} else {
inputLabel = fmt.Sprintf("%d files and %d folders selected.", files, folders)
inputLabel = fmt.Sprintf("%d files and %d folders.", files, folders)
}
}
@ -927,9 +942,17 @@ func work() {
// Use a passthrough to catch compression progress
passthrough := &compressorProgress{Reader: fin}
buf := make([]byte, MiB)
io.CopyBuffer(entry, passthrough, buf)
_, err = io.CopyBuffer(entry, passthrough, buf)
fin.Close()
if err != nil {
insufficientSpace()
writer.Close()
file.Close()
os.Remove(inputFile)
return
}
if !working {
cancel()
writer.Close()
@ -985,9 +1008,17 @@ func work() {
break
}
data = data[:read]
fout.Write(data)
_, err = fout.Write(data)
done += read
if err != nil {
insufficientSpace()
fin.Close()
fout.Close()
os.Remove(outputFile + ".pcv")
return
}
// Update the stats
progress, speed, eta = statify(int64(done), totalBytes, startTime)
progressInfo = fmt.Sprintf("%d/%d", i+1, totalFiles)
@ -1445,7 +1476,17 @@ func work() {
serpent.XORKeyStream(dst, src)
}
}
fout.Write(dst)
_, err = fout.Write(dst)
if err != nil {
insufficientSpace()
fin.Close()
fout.Close()
if recombine || len(allFiles) > 1 || len(onlyFolders) > 0 {
os.Remove(inputFile)
}
os.Remove(outputFile)
return
}
// Update stats
if mode == "decrypt" && reedsolo {
@ -1584,7 +1625,24 @@ func work() {
}
data = data[:read]
fout.Write(data)
_, err = fout.Write(data)
if err != nil {
insufficientSpace()
fin.Close()
fout.Close()
if len(allFiles) > 1 || len(onlyFolders) > 0 {
os.Remove(inputFile)
}
os.Remove(outputFile)
// If user cancels, remove the unfinished files
for _, j := range splitted {
os.Remove(j)
}
os.Remove(fmt.Sprintf("%s.%d", outputFile, i))
return
}
done += read
if done >= chunkSize {
break
@ -1677,6 +1735,12 @@ func accessDenied(s string) {
mainStatusColor = RED
}
// If there isn't enough disk space
func insufficientSpace() {
mainStatus = "Insufficient disk space."
mainStatusColor = RED
}
// If corruption is detected during decryption
func broken(fin *os.File, fout *os.File, message string) {
fin.Close()
@ -1848,13 +1912,13 @@ func timeify(seconds int) string {
// Convert bytes to KiB, MiB, etc.
func sizeify(size int64) string {
if size >= int64(TiB) {
return fmt.Sprintf("%.1fT", float64(size)/float64(TiB))
return fmt.Sprintf("%.2f TiB", float64(size)/float64(TiB))
} else if size >= int64(GiB) {
return fmt.Sprintf("%.1fG", float64(size)/float64(GiB))
return fmt.Sprintf("%.2f GiB", float64(size)/float64(GiB))
} else if size >= int64(MiB) {
return fmt.Sprintf("%.1fM", float64(size)/float64(MiB))
return fmt.Sprintf("%.2f MiB", float64(size)/float64(MiB))
} else {
return fmt.Sprintf("%.1fK", float64(size)/float64(KiB))
return fmt.Sprintf("%.2f KiB", float64(size)/float64(KiB))
}
}