feat: 增加Cassandra扫描和测试环境

Plugins/Cassandra.go

@@ -0,0 +1,94 @@
+package Plugins
+
+import (
+	"fmt"
+	"github.com/gocql/gocql"
+	"github.com/shadow1ng/fscan/Common"
+	"strconv"
+	"strings"
+	"time"
+)
+
+func CassandraScan(info *Common.HostInfo) (tmperr error) {
+	if Common.DisableBrute {
+		return
+	}
+
+	starttime := time.Now().Unix()
+
+	// 首先测试无认证访问
+	flag, err := CassandraConn(info, "", "")
+	if flag && err == nil {
+		return err
+	}
+
+	// 尝试用户名密码组合
+	for _, user := range Common.Userdict["cassandra"] {
+		for _, pass := range Common.Passwords {
+			pass = strings.Replace(pass, "{user}", user, -1)
+
+			flag, err := CassandraConn(info, user, pass)
+			if flag && err == nil {
+				return err
+			}
+
+			errlog := fmt.Sprintf("[-] Cassandra服务 %v:%v 尝试失败 用户名: %v 密码: %v 错误: %v", info.Host, info.Ports, user, pass, err)
+			Common.LogError(errlog)
+			tmperr = err
+
+			if Common.CheckErrs(err) {
+				return err
+			}
+
+			if time.Now().Unix()-starttime > (int64(len(Common.Userdict["cassandra"])*len(Common.Passwords)) * Common.Timeout) {
+				return err
+			}
+		}
+	}
+	return tmperr
+}
+
+func CassandraConn(info *Common.HostInfo, user string, pass string) (bool, error) {
+	host, port := info.Host, info.Ports
+	timeout := time.Duration(Common.Timeout) * time.Second
+
+	cluster := gocql.NewCluster(host)
+	cluster.Port, _ = strconv.Atoi(port)
+	cluster.Timeout = timeout
+	cluster.ProtoVersion = 4 // 指定协议版本
+	cluster.Consistency = gocql.One
+
+	if user != "" || pass != "" {
+		cluster.Authenticator = gocql.PasswordAuthenticator{
+			Username: user,
+			Password: pass,
+		}
+	}
+
+	// 增加重试机制
+	cluster.RetryPolicy = &gocql.SimpleRetryPolicy{NumRetries: 3}
+
+	session, err := cluster.CreateSession()
+	if err != nil {
+		return false, err
+	}
+	defer session.Close()
+
+	// 使用更简单的查询测试连接
+	var version string
+	if err := session.Query("SELECT peer FROM system.peers").Scan(&version); err != nil {
+		if err := session.Query("SELECT now() FROM system.local").Scan(&version); err != nil {
+			return false, err
+		}
+	}
+
+	result := fmt.Sprintf("[+] Cassandra服务 %v:%v ", host, port)
+	if user != "" {
+		result += fmt.Sprintf("爆破成功 用户名: %v 密码: %v", user, pass)
+	} else {
+		result += "无需认证即可访问"
+	}
+	Common.LogSuccess(result)
+
+	return true, nil
+}