SamWaf/global/global.go

package global

import (
	"SamWaf/cache"
	"SamWaf/common/gwebsocket"
	"SamWaf/common/queue"
	"SamWaf/model"
	"SamWaf/model/spec"
	"SamWaf/wafnotify"
	"SamWaf/wafowasp"
	"SamWaf/wafsnowflake"
	"github.com/bytedance/godlp/dlpheader"
	"github.com/lionsoul2014/ip2region/binding/golang/xdb"
	"github.com/oschwald/geoip2-golang"
	"gorm.io/gorm"
	"strconv"
	"time"
)

const (
	GWAF_NAME   = "SamWaf"
	Version_num = 1
)

var (

	/*本机信息**/
	GWAF_RUNTIME_IP          string = "127.0.0.1" //本机当前外网IP
	GWAF_RUNTIME_AREA        string = ""          //本机当前所在区域
	GWAF_RUNTIME_SERVER_TYPE bool   = false       //当前是是否以服务形式启动

	GWAF_RUNTIME_NEW_VERSION      string = ""      //最新版本号
	GWAF_RUNTIME_NEW_VERSION_DESC string = ""      //最新版本描述
	GWAF_RUNTIME_WIN7_VERSION     string = "false" //是否是win7内部版本
	GWAF_RUNTIME_QPS              uint64 = 0       //当前qps
	GWAF_RUNTIME_LOG_PROCESS      uint64 = 0       //log 处理速度

	GWAF_RUNTIME_DNS_SERVER string = "119.29.29.29" //反向查询DNS的IP

	GWAF_RUNTIME_RECORD_LOG_TYPE string = "all" // 记录日志形式： 全部(all),非正常(abnormal)
	GWAF_RUNTIME_IS_UPDATETING   bool   = false //是否正在升级中

	GWAF_RUNTIME_CURRENT_EXEPATH                 string = "" //当前程序运行路径
	GWAF_RUNTIME_CURRENT_WEBPORT                 string = "" //当前程序所占用端口
	GWAF_RUNTIME_CURRENT_EXPORT_DB_LOG_FILE_PATH string = "" //生成的日志导出文件路径

	/**
	遥测数据
	*/
	GWAF_MEASURE_PROCESS_DEQUEENGINE cache.WafOnlyLockWriteData //遥测数据-队列处理时间

	GWAF_GLOBAL_HOST_NAME        string              = "全局网站:0"    //全局网站
	GWAF_GLOBAL_HOST_CODE        string              = "0"         //全局网站代码
	GWAF_LOCAL_DB                *gorm.DB                          //通用本地数据库，尊重用户隐私
	GWAF_LOCAL_LOG_DB            *gorm.DB                          //通用本地数据库存日志数据，尊重用户隐私
	GWAF_LOCAL_STATS_DB          *gorm.DB                          //通用本地数据库存放统计数据，尊重用户隐私
	GWAF_REMOTE_DB               *gorm.DB                          //仅当用户使用云数据库
	GWAF_LOCAL_SERVER_PORT       int                 = 26666       // 本地local端口
	GWAF_LOCAL_INDEX_HTML        string                            //本地首页HTML信息
	GWAF_USER_CODE               string                            // 当前识别号
	GWAF_CUSTOM_SERVER_NAME      string                            // 当前服务器自定义名称
	GWAF_TENANT_ID               string              = "SamWafCom" // 当前租户ID
	GWAF_RELEASE                 string              = "false"     // 当前是否为发行版
	GWAF_RELEASE_VERSION_NAME    string              = "20241028"  // 发行版的版本号名称
	GWAF_RELEASE_VERSION         string              = "v1.0.0"    // 发行版的版本号
	GWAF_LAST_UPDATE_TIME        time.Time                         // 上次时间
	GWAF_NOTICE_ENABLE           bool                = false       // 是否开启通知
	GWAF_CAN_EXPORT_DOWNLOAD_LOG bool                = false       //是否可以导出下载日志
	GWAF_DLP                     dlpheader.EngineAPI               // 脱敏引擎
	GWAF_DLP_CONFIG              string                            // 脱敏引擎配置数据

	GWAF_OWASP *wafowasp.WafOWASP //owasp引擎
	/**链聚合**/
	GWAF_CHAN_HOST      = make(chan model.Hosts, 10)         //主机链
	GWAF_CHAN_ENGINE    = make(chan int, 10)                 //引擎链
	GWAF_CHAN_MSG       = make(chan spec.ChanCommonHost, 10) //全局通讯包
	GWAF_CHAN_UPDATE    = make(chan int, 10)                 //升级后处理链
	GWAF_CHAN_SENSITIVE = make(chan int, 10)                 //敏感词处理链
	GWAF_CHAN_SSL       = make(chan string, 10)              //证书处理链
	/*****CACHE相关*********/
	GCACHE_WAFCACHE      *cache.WafCache      //cache
	GCACHE_WECHAT_ACCESS string          = "" //微信访问密钥

	/*********IP相关**************/
	GCACHE_IP_CBUFF            []byte         // IP相关缓存
	GCACHE_IP_V6_COUNTRY_CBUFF []byte         // IPv6国家相关缓存
	GCACHE_IPV4_SEARCHER       *xdb.Searcher  //IPV4得查询器
	GCACHE_IPV6_SEARCHER       *geoip2.Reader // IPV6得查询器

	GDATA_DELETE_INTERVAL int64 = 180 // 删除180天前的数据

	/****队列相关*****/
	GQEQUE_DB              *queue.Queue //正常DB队列
	GQEQUE_UPDATE_DB       *queue.Queue //正常DB 更新队列
	GQEQUE_LOG_DB          *queue.Queue //日志DB队列
	GQEQUE_STATS_DB        *queue.Queue //统计DB队列
	GQEQUE_STATS_UPDATE_DB *queue.Queue //统计更新DB队列
	GQEQUE_MESSAGE_DB      *queue.Queue //发送消息队列

	/*******通知相关*************/
	GNOTIFY_KAKFA_SERVICE           *wafnotify.WafNotifyService                                  //通知服务
	GNOTIFY_SEND_MAX_LIMIT_MINTUTES                             = time.Duration(5) * time.Minute // 规则相关信息最大发送抑止 默认5分钟

	/******数据库处理参数*****/
	GDATA_BATCH_INSERT       int                 = 1000        //最大批量插入
	GDATA_SHARE_DB_SIZE      int64               = 100 * 10000 //100w 进行分库 100*10000
	GDATA_CURRENT_CHANGE     bool                = false       //当前是否正在切换
	GDATA_CURRENT_LOG_DB_MAP map[string]*gorm.DB               //当前自定义的数据库连接 TODO 如果用户打开了多个 会不会影响内存
	/******WebSocket*********/
	GWebSocket *gwebsocket.WebSocketOnline

	//升级相关
	GUPDATE_VERSION_URL string = "https://update.samwaf.com/" //

	GWAF_SNOWFLAKE_GEN *wafsnowflake.Snowflake //雪花算法

	//任务开关信息
	GWAF_SWITCH_TASK_COUNTER bool

	//数据密码
	GWAF_PWD_COREDB = "3Y)(27EtO^tK8Bj~CORE" //加密
	GWAF_PWD_STATDB = "3Y)(27EtO^tK8Bj~STAT" //加密
	GWAF_PWD_LOGDB  = "3Y)(27EtO^tK8Bj~LOG"  //加密

	//默认创建的账户和密码
	GWAF_DEFAULT_ACCOUNT      string = "admin"         //默认创建的账户
	GWAF_DEFAULT_ACCOUNT_PWD  string = "admin868"      //默认创建的密码
	GWAF_DEFAULT_ACCOUNT_SALT string = "%@311*abDop#*" //盐值

	//通讯加密
	GWAF_COMMUNICATION_KEY = []byte("7E@u*has$d*@s5YX") //通讯加密密钥

	//资源下载
	GWAF_BOT_IP_URL_MAIN string = "https://raw.githubusercontent.com/samwafgo/SamWafBotIPDatabase/main/allowlist/index.json"

	/**
	中心管控部分
	*/
	GWAF_CENTER_ENABLE        string                 = "false"                    //中心管控激活状态
	GWAF_CENTER_URL           string                 = "http://127.0.0.1:26666"   //中心管控默认URL
	GWAF_REG_INFO             model.RegistrationInfo                              //当前注册信息
	GWAF_REG_VERSION                                 = "v1"                       //注册信息版本
	GWAF_REG_KEY                                     = []byte("5F!vion$k@a7QZ&)") //注册信息加密密钥
	GWAF_REG_PUBLIC_KEY       string                 = ""                         //注册信息加密公钥
	GWAF_REG_TMP_REG          []byte                                              //用户传来的信息
	GWAF_REG_FREE_COUNT       int64                  = 99999                      //免费版授权用户数
	GWAF_REG_CUR_CLIENT_COUNT int64                  = 3                          //当前客户端用户数
)

func GetCurrentVersionInt() int {
	version, _ := strconv.Atoi(GWAF_RELEASE_VERSION)
	return version
}