overflow_cloud/container_network
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ing

Browse Source
This commit is contained in:
crusader 2018-07-04 12:11:33 +09:00
parent 9a91d883d4
commit 8f73e25dd7
1 changed files with 41 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

89
crawler/ssh/SSHCrawler.go
View File

@ -1,15 +1,8 @@
package ssh
import (
"bufio"
"bytes"
"fmt"
uuid "github.com/satori/go.uuid"
ocmsc "git.loafle.net/overflow/commons-go/model/sensorconfig"
"git.loafle.net/overflow/container_network/crawler/ssh/client"
"git.loafle.net/overflow/container_network/crawler/ssh/parser"
crawler "git.loafle.net/overflow/crawler-go"
)
@ -30,59 +23,59 @@ func (c *SSHCrawler) Auth(auth map[string]interface{}) error {
}
func (c *SSHCrawler) Get(config *ocmsc.SensorConfig) (map[string]string, error) {
sshClient, err := client.New(config.Target)
_, err := client.New(config.Connection)
if nil != err {
return nil, err
}
itemCount := len(config.Items)
// itemCount := len(config.Items.MetaCollectionItems)
results := make(map[string]string, 0)
boundary := uuid.NewV4().String()
commands := ""
// boundary := uuid.NewV4().String()
// commands := ""
for i := 0; i < itemCount; i++ {
switch i {
case 0:
commands = config.Items[i].QueryInfo.Query
default:
commands = fmt.Sprintf("%s ; echo \"--%s\" ; %s ", commands, boundary, config.Items[i].QueryInfo.Query)
}
}
commands = fmt.Sprintf("%s ; echo \"--%s--\" ", commands, boundary)
// for i := 0; i < itemCount; i++ {
// switch i {
// case 0:
// commands = config.Items[i].QueryInfo.Query
// default:
// commands = fmt.Sprintf("%s ; echo \"--%s\" ; %s ", commands, boundary, config.Items[i].QueryInfo.Query)
// }
// }
// commands = fmt.Sprintf("%s ; echo \"--%s--\" ", commands, boundary)
buf, err := sshClient.RunCommand(commands)
if nil != err {
return nil, err
}
r := bytes.NewReader(buf)
scanner := bufio.NewScanner(r)
// buf, err := sshClient.RunCommand(commands)
// if nil != err {
// return nil, err
// }
// r := bytes.NewReader(buf)
// scanner := bufio.NewScanner(r)
pScanner := parser.NewParserScanner(scanner, boundary)
// pScanner := parser.NewParserScanner(scanner, boundary)
for i := 0; i < itemCount; i++ {
item := config.Items[i]
mode := item.QueryInfo.Extend["mode"].(string)
p := parser.GetParser(mode)
if nil == p {
return nil, fmt.Errorf("Container: Parser[%s] is not exist", mode)
}
// for i := 0; i < itemCount; i++ {
// item := config.Items[i]
// mode := item.QueryInfo.Extend["mode"].(string)
// p := parser.GetParser(mode)
// if nil == p {
// return nil, fmt.Errorf("Container: Parser[%s] is not exist", mode)
// }
rm, err := p.Parse(pScanner)
if nil != err {
return nil, err
}
// rm, err := p.Parse(pScanner)
// if nil != err {
// return nil, err
// }
if nil != rm {
mm := ocmsc.KeysToMap(item.Keys)
for key, value := range mm {
results[value] = rm[key]
}
}
// if nil != rm {
// mm := ocmsc.KeysToMap(item.Keys)
// for key, value := range mm {
// results[value] = rm[key]
// }
// }
if !pScanner.Clean() {
break
}
}
// if !pScanner.Clean() {
// break
// }
// }
return results, nil
}