mirrors/fscan
1
0
Fork 0
mirror of https://github.com/shadow1ng/fscan.git synced 2025-07-14 05:12:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update postgres.go

Browse Source
This commit is contained in:
影舞者 2023-11-13 10:56:54 +08:00 committed by GitHub
parent 8bcbdf1f38
commit 52c680af0d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
Plugins/postgres.go
View File

@ -3,23 +3,22 @@ package Plugins
import ( import (
"database/sql" "database/sql"
"fmt" "fmt"
"strings"
"time"
_ "github.com/lib/pq" _ "github.com/lib/pq"
"github.com/shadow1ng/fscan/common" "github.com/shadow1ng/fscan/common"
"strings"
"time"
) )
func PostgresScan(info common.HostInfo, flags common.Flags) (tmperr error) { func PostgresScan(info *common.HostInfo) (tmperr error) {
if flags.IsBrute { if common.IsBrute {
return return
} }
starttime := time.Now().Unix() starttime := time.Now().Unix()
for _, user := range common.Userdict["postgresql"] { for _, user := range common.Userdict["postgresql"] {
for _, pass := range common.Passwords { for _, pass := range common.Passwords {
pass = strings.Replace(pass, "{user}", string(user), -1) pass = strings.Replace(pass, "{user}", string(user), -1)
flag, err := PostgresConn(info, user, pass, flags.Timeout) flag, err := PostgresConn(info, user, pass)
if flag && err == nil { if flag == true && err == nil {
return err return err
} else { } else {
errlog := fmt.Sprintf("[-] psql %v:%v %v %v %v", info.Host, info.Ports, user, pass, err) errlog := fmt.Sprintf("[-] psql %v:%v %v %v %v", info.Host, info.Ports, user, pass, err)
@ -28,7 +27,7 @@ func PostgresScan(info common.HostInfo, flags common.Flags) (tmperr error) {
if common.CheckErrs(err) { if common.CheckErrs(err) {
return err return err
} }
if time.Now().Unix()-starttime > (int64(len(common.Userdict["postgresql"])*len(common.Passwords)) * flags.Timeout) { if time.Now().Unix()-starttime > (int64(len(common.Userdict["postgresql"])*len(common.Passwords)) * common.Timeout) {
return err return err
} }
} }
@ -37,13 +36,13 @@ func PostgresScan(info common.HostInfo, flags common.Flags) (tmperr error) {
return tmperr return tmperr
} }
func PostgresConn(info common.HostInfo, user string, pass string, timeout int64) (flag bool, err error) { func PostgresConn(info *common.HostInfo, user string, pass string) (flag bool, err error) {
flag = false flag = false
Host, Port, Username, Password := info.Host, info.Ports, user, pass Host, Port, Username, Password := info.Host, info.Ports, user, pass
dataSourceName := fmt.Sprintf("postgres://%v:%v@%v:%v/%v?sslmode=%v", Username, Password, Host, Port, "postgres", "disable") dataSourceName := fmt.Sprintf("postgres://%v:%v@%v:%v/%v?sslmode=%v", Username, Password, Host, Port, "postgres", "disable")
db, err := sql.Open("postgres", dataSourceName) db, err := sql.Open("postgres", dataSourceName)
if err == nil { if err == nil {
db.SetConnMaxLifetime(time.Duration(timeout) * time.Second) db.SetConnMaxLifetime(time.Duration(common.Timeout) * time.Second)
defer db.Close() defer db.Close()
err = db.Ping() err = db.Ping()
if err == nil { if err == nil {