forked from hacklcx/HFish
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
21 changed files
with
961 additions
and
876 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,108 @@ | ||
| package alert | ||
|
|
||
| import ( | ||
| "HFish/utils/try" | ||
| "HFish/core/dbUtil" | ||
| "strconv" | ||
| "strings" | ||
| "HFish/utils/send" | ||
| "bytes" | ||
| "net/http" | ||
| "HFish/utils/log" | ||
| "encoding/json" | ||
| ) | ||
|
|
||
| func AlertMail(model string, typex string, agent string, ipx string, country string, region string, city string, infox string) { | ||
| // 判断邮件通知 | ||
| try.Try(func() { | ||
| // 只有新加入才会发送邮件通知 | ||
| if (model == "new") { | ||
| result, err := dbUtil.DB().Table("hfish_setting").Fields("status", "info").Where("type", "=", "alertMail").First() | ||
|
|
||
| if err != nil { | ||
| log.Pr("HFish", "127.0.0.1", "获取邮件告警配置失败", err) | ||
| } | ||
|
|
||
| status := strconv.FormatInt(result["status"].(int64), 10) | ||
|
|
||
| // 判断是否启用通知 | ||
| if status == "1" { | ||
| info := result["info"] | ||
| config := strings.Split(info.(string), "&&") | ||
|
|
||
| if (country == "本地地址") { | ||
| region = "" | ||
| city = "" | ||
| } else if (country == "局域网") { | ||
| region = "" | ||
| city = "" | ||
| } | ||
|
|
||
| text := ` | ||
| <div><b>Hi，上钩了！</b></div> | ||
| <div><b><br /></b></div> | ||
| <div><b>集群名称：</b>` + agent + `</div> | ||
| <div><b>攻击IP：</b>` + ipx + `</div> | ||
| <div><b>地理信息：</b>` + country + ` ` + region + ` ` + city + `</div> | ||
| <div><b>上钩内容：</b>` + infox + `</div> | ||
| <div><br /></div> | ||
| <div><span style="color: rgb(128, 128, 128); font-size: 10px;">(HFish 自动发送)</span></div> | ||
| ` | ||
|
|
||
| send.SendMail(config[4:], "[HFish]提醒你，"+typex+"有鱼上钩!", text, config) | ||
| } | ||
| } | ||
| }).Catch(func() { | ||
| }) | ||
| } | ||
|
|
||
| func AlertWebHook(id string, model string, typex string, projectName string, agent string, ipx string, country string, region string, city string, infox string, time string) { | ||
| // 判断 WebHook 通知 | ||
| try.Try(func() { | ||
| result, err := dbUtil.DB().Table("hfish_setting").Fields("status", "info").Where("type", "=", "webHook").First() | ||
|
|
||
| if err != nil { | ||
| log.Pr("HFish", "127.0.0.1", "获取WebHook配置失败", err) | ||
| } | ||
|
|
||
| status := strconv.FormatInt(result["status"].(int64), 10) | ||
|
|
||
| // 判断是否启用通知 | ||
| if status == "1" { | ||
| info := result["info"] | ||
|
|
||
| song := make(map[string]interface{}) | ||
| song["id"] = id | ||
| song["model"] = model | ||
| song["project"] = projectName | ||
| song["type"] = typex | ||
| song["agent"] = agent | ||
| song["ip"] = ipx | ||
| song["country"] = country | ||
| song["region"] = region | ||
| song["city"] = city | ||
| song["info"] = infox | ||
| song["time"] = time | ||
|
|
||
| bytesData, _ := json.Marshal(song) | ||
|
|
||
| reader := bytes.NewReader(bytesData) | ||
|
|
||
| request, _ := http.NewRequest("POST", info.(string), reader) | ||
| request.Header.Set("Content-Type", "application/json;charset=UTF-8") | ||
|
|
||
| client := http.Client{} | ||
| resp, err := client.Do(request) | ||
|
|
||
| if err != nil { | ||
| log.Pr("HFish", "127.0.0.1", "WebHook 调用失败", err) | ||
| } else { | ||
| log.Pr("HFish", "127.0.0.1", "WebHook 调用成功") | ||
| } | ||
|
|
||
| defer resp.Body.Close() | ||
| //defer request.Body.Close() | ||
| } | ||
| }).Catch(func() { | ||
| }) | ||
| } |
Oops, something went wrong.