mirror of https://github.com/qwqdanchun/fscan.git
44 lines
1.1 KiB
Go
44 lines
1.1 KiB
Go
|
package Plugins
|
||
|
|
||
|
import (
|
||
|
"database/sql"
|
||
|
"fmt"
|
||
|
_ "github.com/lib/pq"
|
||
|
"github.com/shadow1ng/fscan/common"
|
||
|
"strings"
|
||
|
"time"
|
||
|
)
|
||
|
|
||
|
func PostgresScan(info *common.HostInfo) (tmperr error) {
|
||
|
for _, user := range common.Userdict["postgresql"] {
|
||
|
for _, pass := range common.Passwords {
|
||
|
pass = strings.Replace(pass, "{user}", string(user), -1)
|
||
|
flag, err := PostgresConn(info, user, pass)
|
||
|
if flag == true && err == nil {
|
||
|
return err
|
||
|
} else {
|
||
|
tmperr = err
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
return tmperr
|
||
|
}
|
||
|
|
||
|
func PostgresConn(info *common.HostInfo, user string, pass string) (flag bool, err error) {
|
||
|
flag = false
|
||
|
Host, Port, Username, Password := info.Host, common.PORTList["psql"], user, pass
|
||
|
dataSourceName := fmt.Sprintf("postgres://%v:%v@%v:%v/%v?sslmode=%v", Username, Password, Host, Port, "postgres", "disable")
|
||
|
db, err := sql.Open("mysql", dataSourceName)
|
||
|
if err == nil {
|
||
|
db.SetConnMaxLifetime(time.Duration(info.Timeout) * time.Second)
|
||
|
defer db.Close()
|
||
|
err = db.Ping()
|
||
|
if err == nil {
|
||
|
result := fmt.Sprintf("Postgres:%v:%v:%v %v", Host, Port, Username, Password)
|
||
|
common.LogSuccess(result)
|
||
|
flag = true
|
||
|
}
|
||
|
}
|
||
|
return flag, err
|
||
|
}
|