nhyatt/istheinternetonfire
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

small tweaks

Browse Source
This commit is contained in:
Hyatt 2024-01-19 17:43:40 -06:00
parent c1739856ce
commit 567cfe8083
Signed by: nhyatt
GPG Key ID: C50D0BBB5BC40BEA
6 changed files with 20 additions and 12795 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
assets/html/index.tplt
View File

@ -19,7 +19,7 @@
</head>
<body style="text-align:center;">
<small>
[<a href="./status.txt">txt</a>]&nbsp;<tt>dig +short txt istheinternetonfire.app | sed 's/" "//g' |sed 's/%n/\n/g'</tt>&nbsp;[<a href="./status.json">json</a>]
[<a href="./status.txt">txt</a>]&nbsp;<tt>dig +short txt istheinternetonfire.app | sed 's/" "//g' | sed 's/%n/\n/g'</tt>&nbsp;[<a href="./status.json">json</a>]
</small>
<hr align="CENTER" noshade="noshade" size="2" width="100%">
{{- if gt (len .CVEs) 0 }}

18
internal/cisa/cisa.go
View File

@ -2,10 +2,12 @@ package cisa
import (
"encoding/json"
"net/http"
"sync"
"time"
"istheinternetonfire.app/internal/config"
"istheinternetonfire.app/internal/httpclient"
)
var (
@ -22,16 +24,12 @@ func Read() CisaJSON {
func Start() {
for {
/*
c := httpclient.NewClient(http.DefaultClient)
d, err := c.Get(config.Cfg.RemoteURL)
if err != nil {
time.Sleep(time.Second * 120)
continue
}
*/
d := GetExampleData()
c := httpclient.NewClient(http.DefaultClient)
d, err := c.Get(config.Cfg.RemoteURL)
if err != nil {
time.Sleep(time.Second * 120)
continue
}
mu.Lock()
if err := json.Unmarshal(d, &Cisa); err != nil {

12780
internal/cisa/example.go
View File

File diff suppressed because it is too large Load Diff

1
internal/nsupdate/nsupdate.go
View File

@ -57,6 +57,7 @@ func (c NsUpdateStruct) Update(record, recordType, value string) error {
if err != nil {
config.Cfg.Log.Info("unable to get existing record", "record", record, "error", err)
}
config.Cfg.Log.Debug("comparing records", "old value", r, "new value", value)
if r != value {
config.Cfg.Log.Debug("deleting record", "record", record)

7
internal/webserver/httpTemplate.go
View File

@ -55,6 +55,7 @@ func tmpltWebRoot(w http.ResponseWriter, r *http.Request) {
var (
msgBuffer bytes.Buffer
cves []cisa.VulStruct
num int = 3
)
c := cisa.Read()
@ -65,10 +66,14 @@ func tmpltWebRoot(w http.ResponseWriter, r *http.Request) {
}
}
if len(cves) < 3 {
num = len(cves)
}
if err := tmplt.Execute(&msgBuffer, struct {
CVEs []cisa.VulStruct
}{
CVEs: cves[len(cves)-3:],
CVEs: cves[len(cves)-num:],
}); err != nil {
config.Cfg.Log.Debug("unable to execute html template", err)
tmpltError(w, http.StatusInternalServerError, "Template Parse Error.")

7
main.go
View File

@ -56,8 +56,8 @@ func main() {
}
}
if len(cves) == 0 {
if err := dns.Delete("istheinternetonfire.app", "TXT"); err != nil {
config.Cfg.Log.Error("unable to delete dns record", "error", err)
if err := dns.Update("istheinternetonfire.app", "TXT", "Safe for now!"); err != nil {
config.Cfg.Log.Error("unable to update dns record", "error", err)
time.Sleep(time.Second * time.Duration(config.Cfg.RefreshSeconds))
continue
}
@ -70,10 +70,11 @@ func main() {
txtData += fmt.Sprintf("%s - %s - %s\n", v.CveID, v.Product, v.ShortDescription)
}
if err := dns.Update("istheinternetonfire.app", "TXT", txtData); err != nil {
config.Cfg.Log.Error("unable to add dns record", "error", err)
config.Cfg.Log.Error("unable to update dns record", "error", err)
time.Sleep(time.Second * time.Duration(config.Cfg.RefreshSeconds))
continue
}
time.Sleep(time.Second * time.Duration(config.Cfg.RefreshSeconds))
}
}()