update

Plugins/smb.go

@@ -0,0 +1,75 @@
+package Plugins
+
+import (
+	"../common"
+	"fmt"
+	"github.com/stacktitan/smb/smb"
+	"strings"
+	"sync"
+	"time"
+	"context"
+)
+func SmbScan(info *common.HostInfo,ch chan int,wg *sync.WaitGroup) {
+
+Loop:
+	for _,user:=range common.Userdict["smb"]{
+		for _,pass:=range common.Passwords{
+			pass = strings.Replace(pass, "{user}", string(user), -1)
+			//fmt.Println(user,pass)
+			//flag,err := SmblConn(info,user,pass)
+			flag,err := doWithTimeOut(info,user,pass)
+			//fmt.Println(user,pass,flag,err)
+			if flag==true && err==nil {
+				break Loop
+			}
+		}
+	}
+	wg.Done()
+	<- ch
+
+}
+
+func SmblConn(info *common.HostInfo,user string,pass string)(flag bool,err error){
+	flag = false
+	Host,Port,Username,Password := info.Host, common.PORTList["smb"],user, pass
+	options := smb.Options{
+		Host:        Host,
+		Port:        Port,
+		User:        Username,
+		Password:    Password,
+		Domain:      "",
+		Workstation: "",
+		Timeout:     info.Timeout,
+
+	}
+
+	session, err := smb.NewSession(options, false)
+	//fmt.Println(err)
+	if err == nil {
+		defer session.Close()
+		if session.IsAuthenticated {
+			result := fmt.Sprintf("SMB:%v:%v:%v %v",Host,Port,Username,Password)
+			common.LogSuccess(result)
+			flag = true
+		}
+	}
+	return flag,err
+}
+
+func doWithTimeOut(info *common.HostInfo,user string,pass string)(flag bool,err error){
+	ctx,cancel := context.WithTimeout(context.Background(),time.Duration(info.Timeout)*time.Second)
+	//ctx,cancel := context.WithTimeout(context.Background(),1*time.Second)
+	defer cancel()
+	signal := make(chan int,1)
+	go func() {
+		flag,err = SmblConn(info,user,pass)
+		signal <- 1
+	}()
+
+	select {
+	case <-signal:
+		return flag,err
+	case <-ctx.Done():
+		return false,err
+	}
+}