nhyatt/entropy
1
0
Fork 0
mirror of https://github.com/jlowellwofford/entropy.git synced 2025-04-05 05:00:16 -05:00
Code Issues Packages Projects Releases Wiki Activity

add quality option for addentropy

Browse Source
This commit is contained in:
J. Lowell Wofford 2020-12-01 11:29:01 -07:00
parent 5386961c69
commit 5a3894077e
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
cmd/entropy/entropy.go
View File

@ -71,14 +71,26 @@ func main() {
case "add": case "add":
fallthrough fallthrough
case "addentropy": case "addentropy":
if len(os.Args) != 3 { if len(os.Args) < 3 || len(os.Args) > 4 {
usageFatal("addtoentcnt requires a file path as an option") usageFatal("addtoentcnt requires a file path as an option")
} }
var quality = 1.0
if len(os.Args) == 4 {
quality, err = strconv.ParseFloat(os.Args[3], 32)
if err != nil {
fatal("couldn't parse quality option, must be a float from 0.0 - 1.0: %v", err)
}
if quality < 0 || quality > 1 {
fatal("quality must be in the range of 0.0 - 1.0")
}
}
var buf []byte var buf []byte
if buf, err = ioutil.ReadFile(os.Args[2]); err != nil { if buf, err = ioutil.ReadFile(os.Args[2]); err != nil {
fatal("could not read file %s: %v", os.Args[2], err) fatal("could not read file %s: %v", os.Args[2], err)
} }
if err = entropy.AddEntropy(len(buf), buf); err != nil { bits := int(float64(len(buf)) * 8 * quality)
fmt.Printf("adding %d bytes with %d bits of entropy\n", len(buf), bits)
if err = entropy.AddEntropy(bits, buf); err != nil {
fatal("failed to add entropy: %v", err) fatal("failed to add entropy: %v", err)
} }
case "zap": case "zap":