зеркало из https://github.com/mozilla/mig.git
126 строки
2.8 KiB
Go
126 строки
2.8 KiB
Go
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
//
|
|
// Contributor: Aaron Meihm ameihm@mozilla.com [:alm]
|
|
|
|
package sshkey /* import "github.com/mozilla/mig/modules/sshkey" */
|
|
|
|
import (
|
|
"bufio"
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func printHelp(isCmd bool) {
|
|
dash := " "
|
|
if isCmd {
|
|
dash = "-"
|
|
}
|
|
fmt.Printf(`Query parameters
|
|
----------------
|
|
%vpath <string> - specify a path to search for keys, more than one path option
|
|
can be specified on the command line. if the path option is not
|
|
present an OS specific default will be used (/root and /home for
|
|
Linux and Darwin, and c:\Users for Windows).
|
|
|
|
%vmaxdepth <int> - override default search depth of 8.
|
|
`, dash, dash)
|
|
}
|
|
|
|
// ParamsCreator is used by mig-console to create parameters for the module
|
|
func (r *run) ParamsCreator() (interface{}, error) {
|
|
p := newParameters()
|
|
scanner := bufio.NewScanner(os.Stdin)
|
|
for {
|
|
fmt.Printf("search> ")
|
|
scanmore := scanner.Scan()
|
|
if err := scanner.Err(); err != nil {
|
|
fmt.Println("Invalid input. Try again")
|
|
continue
|
|
}
|
|
if !scanmore {
|
|
goto exit
|
|
}
|
|
input := scanner.Text()
|
|
if input == "done" {
|
|
goto exit
|
|
} else if input == "help" {
|
|
printHelp(false)
|
|
continue
|
|
}
|
|
arr := strings.SplitN(input, " ", 2)
|
|
checkType := arr[0]
|
|
checkValue := ""
|
|
if len(arr) > 1 {
|
|
checkValue = arr[1]
|
|
}
|
|
switch checkType {
|
|
case "path":
|
|
if checkValue == "" {
|
|
fmt.Println("Missing parameter, try again")
|
|
continue
|
|
}
|
|
p.Paths = append(p.Paths, checkValue)
|
|
case "maxdepth":
|
|
var err error
|
|
if checkValue == "" {
|
|
fmt.Println("Missing parameter, try again")
|
|
}
|
|
p.MaxDepth, err = strconv.Atoi(checkValue)
|
|
if err != nil {
|
|
fmt.Println("maxdepth argument must be an integer")
|
|
continue
|
|
}
|
|
default:
|
|
fmt.Printf("Invalid command, try help\n")
|
|
continue
|
|
}
|
|
}
|
|
|
|
exit:
|
|
r.Parameters = *p
|
|
return r.Parameters, r.ValidateParameters()
|
|
}
|
|
|
|
// ParamsParser is used by the mig command line tool to parse parameters for the module
|
|
func (r *run) ParamsParser(args []string) (interface{}, error) {
|
|
var (
|
|
paths flagParam
|
|
fs flag.FlagSet
|
|
maxdepth int
|
|
)
|
|
if len(args) > 0 && args[0] == "help" {
|
|
printHelp(true)
|
|
return nil, nil
|
|
}
|
|
|
|
fs.Init("sshkey", flag.ContinueOnError)
|
|
fs.Var(&paths, "path", "see help")
|
|
fs.IntVar(&maxdepth, "maxdepth", 0, "see help")
|
|
err := fs.Parse(args)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
p := newParameters()
|
|
p.Paths = paths
|
|
p.MaxDepth = maxdepth
|
|
r.Parameters = *p
|
|
return r.Parameters, r.ValidateParameters()
|
|
}
|
|
|
|
type flagParam []string
|
|
|
|
func (f *flagParam) String() string {
|
|
return fmt.Sprint([]string(*f))
|
|
}
|
|
|
|
func (f *flagParam) Set(value string) error {
|
|
*f = append(*f, value)
|
|
return nil
|
|
}
|