Files
SamWaf/global/global.go
2024-10-28 14:20:35 +08:00

153 lines
7.2 KiB
Go

package global
import (
"SamWaf/cache"
"SamWaf/common/gwebsocket"
"SamWaf/common/queue"
"SamWaf/model"
"SamWaf/model/spec"
"SamWaf/wafnotify"
"SamWaf/wafsnowflake"
"github.com/bytedance/godlp/dlpheader"
"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 = "true" //是否是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_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_DLP dlpheader.EngineAPI // 脱敏引擎
GWAF_DLP_CONFIG string // 脱敏引擎配置数据
/**链聚合**/
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 = "" //微信访问密钥
GCACHE_IP_CBUFF []byte // IP相关缓存
GDATA_DELETE_INTERVAL int64 = 180 // 删除180天前的数据
/****队列相关*****/
GQEQUE_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 //通知服务
/******数据库处理参数*****/
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
/******记录参数配置****************/
GCONFIG_RECORD_MAX_BODY_LENGTH int64 = 1024 * 2 //限制记录最大请求的body长度 record_max_req_body_length
GCONFIG_RECORD_MAX_RES_BODY_LENGTH int64 = 1024 * 4 //限制记录最大响应的body长度 record_max_rep_body_length
GCONFIG_RECORD_RESP int64 = 0 // 是否记录响应记录 record_resp
GCONFIG_RECORD_PROXY_HEADER string = "X-Forwarded-For,X-Real-IP" //配置获取IP头信息
GCONFIG_RECORD_AUTO_LOAD_SSL int64 = 1 //是否每天凌晨3点自动加载ssl证书
GCONFIG_RECORD_KAFKA_ENABLE int64 = 0 //kafka 是否激活
GCONFIG_RECORD_KAFKA_URL string = "127.0.0.1:9092" //kafka url地址
GCONFIG_RECORD_KAFKA_TOPIC string = "samwaf_logs_topic" //kafka topic
//升级相关
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
}