DWS/hadrian
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improve injections

Browse Source
This commit is contained in:
Gianni Carafa
2023-11-04 20:54:38 +01:00
parent 890d6929e0
commit a1e5d540fe
2 changed files with 18 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
handlers/proxy.go
View File

@@ -166,7 +166,15 @@ func applyRules(domain string, path string, body string) string {
if err != nil {
log.Fatal(err)
}
doc.Find(injection.Position).AppendHtml(injection.Code)
if injection.Replace != "" {
doc.Find(injection.Position).ReplaceWithHtml(injection.Replace)
}
if injection.Append != "" {
doc.Find(injection.Position).AppendHtml(injection.Append)
}
if injection.Prepend != "" {
doc.Find(injection.Position).PrependHtml(injection.Prepend)
}
body, err = doc.Html()
if err != nil {
log.Fatal(err)
@@ -189,6 +197,8 @@ type RuleSet []struct {
RegexRules []Rule `yaml:"regexRules"`
Injections []struct {
Position string `yaml:"position"`
Code string `yaml:"code"`
Append string `yaml:"append"`
Prepend string `yaml:"prepend"`
Replace string `yaml:"replace"`
} `yaml:"injections"`
}