mirror of
https://github.com/wneessen/apg-go.git
synced 2024-12-23 11:40:38 +01:00
Merge pull request #56 from wneessen/goreport_cleanup
GoReportCard cleanup
This commit is contained in:
commit
3b1ce9009a
3 changed files with 61 additions and 55 deletions
|
@ -65,26 +65,8 @@ func main() {
|
|||
}
|
||||
|
||||
// Old style character modes
|
||||
if humanReadable {
|
||||
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeHumanReadable)
|
||||
}
|
||||
if lowerCase {
|
||||
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeLowerCase)
|
||||
}
|
||||
if upperCase {
|
||||
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeUpperCase)
|
||||
}
|
||||
if numeric {
|
||||
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeNumeric)
|
||||
}
|
||||
if special {
|
||||
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeSpecial)
|
||||
}
|
||||
if complexPass {
|
||||
config.Mode = apg.MaskSetMode(config.Mode, apg.ModeLowerCase|apg.ModeNumeric|
|
||||
apg.ModeSpecial|apg.ModeUpperCase)
|
||||
config.Mode = apg.MaskClearMode(config.Mode, apg.ModeHumanReadable)
|
||||
}
|
||||
configOldStyle(config, humanReadable, lowerCase, upperCase, numeric,
|
||||
special, complexPass)
|
||||
|
||||
// New style character modes (has higher priority than the old style modes)
|
||||
if modeString != "" {
|
||||
|
@ -93,6 +75,21 @@ func main() {
|
|||
|
||||
// For the "minimum amount of" modes we need to imply at the type
|
||||
// of character mode is set
|
||||
configMinRequirement(config)
|
||||
|
||||
// Check if algorithm is supported
|
||||
config.Algorithm = apg.IntToAlgo(algorithm)
|
||||
if config.Algorithm == apg.AlgoUnsupported {
|
||||
_, _ = fmt.Fprintf(os.Stderr, "unsupported algorithm value: %d\n", algorithm)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// Generate the password based on the given flags and print it to stdout
|
||||
generate(config)
|
||||
}
|
||||
|
||||
// configMinRequirement configures the "minimum amount" feature
|
||||
func configMinRequirement(config *apg.Config) {
|
||||
if config.MinLowerCase > 0 {
|
||||
if float64(config.MinLength)/2 < float64(config.MinNumeric) {
|
||||
_, _ = os.Stderr.WriteString(MinimumAmountTooHigh)
|
||||
|
@ -117,15 +114,35 @@ func main() {
|
|||
}
|
||||
config.Mode = apg.MaskSetMode(config.Mode, apg.ModeUpperCase)
|
||||
}
|
||||
}
|
||||
|
||||
// Check if algorithm is supported
|
||||
config.Algorithm = apg.IntToAlgo(algorithm)
|
||||
if config.Algorithm == apg.AlgoUnsupported {
|
||||
_, _ = fmt.Fprintf(os.Stderr, "unsupported algorithm value: %d\n", algorithm)
|
||||
os.Exit(1)
|
||||
// configOldStyle configures the old style character modes
|
||||
func configOldStyle(config *apg.Config, humanReadable, lowerCase, upperCase,
|
||||
numeric, special, complexPass bool,
|
||||
) {
|
||||
if humanReadable {
|
||||
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeHumanReadable)
|
||||
}
|
||||
if lowerCase {
|
||||
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeLowerCase)
|
||||
}
|
||||
if upperCase {
|
||||
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeUpperCase)
|
||||
}
|
||||
if numeric {
|
||||
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeNumeric)
|
||||
}
|
||||
if special {
|
||||
config.Mode = apg.MaskToggleMode(config.Mode, apg.ModeSpecial)
|
||||
}
|
||||
if complexPass {
|
||||
config.Mode = apg.MaskSetMode(config.Mode, apg.ModeLowerCase|apg.ModeNumeric|
|
||||
apg.ModeSpecial|apg.ModeUpperCase)
|
||||
config.Mode = apg.MaskClearMode(config.Mode, apg.ModeHumanReadable)
|
||||
}
|
||||
}
|
||||
|
||||
// Generate the password based on the given flags
|
||||
func generate(config *apg.Config) {
|
||||
generator := apg.New(config)
|
||||
for i := int64(0); i < config.NumberPass; i++ {
|
||||
password, err := generator.Generate()
|
||||
|
|
45
random.go
45
random.go
|
@ -229,13 +229,7 @@ func (g *Generator) checkMinimumRequirements(password string) bool {
|
|||
charRange = CharRangeAlphaLower
|
||||
}
|
||||
|
||||
count := 0
|
||||
for _, char := range charRange {
|
||||
count += strings.Count(password, string(char))
|
||||
}
|
||||
if int64(count) < g.config.MinLowerCase {
|
||||
ok = false
|
||||
}
|
||||
matchesMinimumAmount(charRange, password, g.config.MinLowerCase, &ok)
|
||||
}
|
||||
if g.config.MinNumeric > 0 {
|
||||
var charRange string
|
||||
|
@ -246,13 +240,7 @@ func (g *Generator) checkMinimumRequirements(password string) bool {
|
|||
charRange = CharRangeNumeric
|
||||
}
|
||||
|
||||
count := 0
|
||||
for _, char := range charRange {
|
||||
count += strings.Count(password, string(char))
|
||||
}
|
||||
if int64(count) < g.config.MinNumeric {
|
||||
ok = false
|
||||
}
|
||||
matchesMinimumAmount(charRange, password, g.config.MinNumeric, &ok)
|
||||
}
|
||||
if g.config.MinSpecial > 0 {
|
||||
var charRange string
|
||||
|
@ -263,13 +251,7 @@ func (g *Generator) checkMinimumRequirements(password string) bool {
|
|||
charRange = CharRangeSpecial
|
||||
}
|
||||
|
||||
count := 0
|
||||
for _, char := range charRange {
|
||||
count += strings.Count(password, string(char))
|
||||
}
|
||||
if int64(count) < g.config.MinSpecial {
|
||||
ok = false
|
||||
}
|
||||
matchesMinimumAmount(charRange, password, g.config.MinSpecial, &ok)
|
||||
}
|
||||
if g.config.MinUpperCase > 0 {
|
||||
var charRange string
|
||||
|
@ -280,13 +262,7 @@ func (g *Generator) checkMinimumRequirements(password string) bool {
|
|||
charRange = CharRangeAlphaUpper
|
||||
}
|
||||
|
||||
count := 0
|
||||
for _, char := range charRange {
|
||||
count += strings.Count(password, string(char))
|
||||
}
|
||||
if int64(count) < g.config.MinUpperCase {
|
||||
ok = false
|
||||
}
|
||||
matchesMinimumAmount(charRange, password, g.config.MinUpperCase, &ok)
|
||||
}
|
||||
return ok
|
||||
}
|
||||
|
@ -359,3 +335,16 @@ func (g *Generator) generateRandom() (string, error) {
|
|||
|
||||
return password, nil
|
||||
}
|
||||
|
||||
// matchesMinimumAmount checks if the number of occurrences of characters in
|
||||
// charRange in the password is less than minAmount and updates the
|
||||
// value of ok accordingly.
|
||||
func matchesMinimumAmount(charRange, password string, minAmount int64, ok *bool) {
|
||||
count := 0
|
||||
for _, char := range charRange {
|
||||
count += strings.Count(password, string(char))
|
||||
}
|
||||
if int64(count) < minAmount {
|
||||
*ok = false
|
||||
}
|
||||
}
|
||||
|
|
|
@ -98,7 +98,7 @@ func TestGenerator_RandomBytes(t *testing.T) {
|
|||
}
|
||||
bl := len(rb)
|
||||
if int64(bl) != tc.l {
|
||||
t.Errorf("lenght of provided bytes does not match requested length: got: %d, expected: %d",
|
||||
t.Errorf("length of provided bytes does not match requested length: got: %d, expected: %d",
|
||||
bl, tc.l)
|
||||
}
|
||||
if bytes.Equal(rb, make([]byte, tc.l)) {
|
||||
|
|
Loading…
Reference in a new issue