mirror of
https://github.com/xvzc/SpoofDPI.git
synced 2024-12-22 06:15:51 +00:00
chore: refactor
This commit is contained in:
parent
d7e0239c6a
commit
2489cc450d
112
util/config.go
Normal file
112
util/config.go
Normal file
@ -0,0 +1,112 @@
|
||||
package util
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"regexp"
|
||||
"strings"
|
||||
|
||||
"github.com/pterm/pterm"
|
||||
log "github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
Addr *string
|
||||
Port *int
|
||||
Dns *string
|
||||
Debug *bool
|
||||
Banner *bool
|
||||
AllowedPattern *regexp.Regexp
|
||||
AllowedUrls *regexp.Regexp
|
||||
}
|
||||
|
||||
type ArrayFlags []string
|
||||
|
||||
func (i *ArrayFlags) String() string {
|
||||
return "my string representation"
|
||||
}
|
||||
|
||||
func (i *ArrayFlags) Set(value string) error {
|
||||
*i = append(*i, value)
|
||||
return nil
|
||||
}
|
||||
|
||||
var config *Config
|
||||
var allowedHosts ArrayFlags
|
||||
var allowedPattern *string
|
||||
|
||||
func GetConfig() *Config {
|
||||
return config
|
||||
}
|
||||
|
||||
func (c *Config) PatternExists() bool {
|
||||
return c.AllowedPattern != nil || c.AllowedUrls != nil
|
||||
}
|
||||
|
||||
func (c *Config) PatternMatches(bytes []byte) bool {
|
||||
return (c.AllowedPattern != nil && c.AllowedPattern.Match(bytes)) ||
|
||||
(c.AllowedUrls != nil && c.AllowedUrls.Match(bytes))
|
||||
}
|
||||
|
||||
func ParseArgs() {
|
||||
config = &Config{}
|
||||
config.Addr = flag.String("addr", "127.0.0.1", "Listen addr")
|
||||
config.Port = flag.Int("port", 8080, "port")
|
||||
config.Dns = flag.String("dns", "8.8.8.8", "DNS server")
|
||||
config.Debug = flag.Bool("debug", false, "true | false")
|
||||
config.Banner = flag.Bool("banner", true, "true | false")
|
||||
|
||||
flag.Var(&allowedHosts, "url", "Bypass DPI only on this url, can be passed multiple times")
|
||||
allowedPattern = flag.String(
|
||||
"pattern",
|
||||
"",
|
||||
"Bypass DPI only on packets matching this regex pattern",
|
||||
)
|
||||
|
||||
flag.Parse()
|
||||
|
||||
if len(allowedHosts) > 0 {
|
||||
var escapedUrls []string
|
||||
for _, host := range allowedHosts {
|
||||
escapedUrls = append(escapedUrls, regexp.QuoteMeta(host))
|
||||
}
|
||||
|
||||
allowedHostsRegex := strings.Join(escapedUrls, "|")
|
||||
config.AllowedUrls = regexp.MustCompile(allowedHostsRegex)
|
||||
}
|
||||
|
||||
log.Info(*allowedPattern)
|
||||
if *allowedPattern != "" {
|
||||
config.AllowedPattern = regexp.MustCompile(*allowedPattern)
|
||||
}
|
||||
}
|
||||
|
||||
func PrintColoredBanner() {
|
||||
cyan := pterm.NewLettersFromStringWithStyle("Spoof", pterm.NewStyle(pterm.FgCyan))
|
||||
purple := pterm.NewLettersFromStringWithStyle("DPI", pterm.NewStyle(pterm.FgLightMagenta))
|
||||
pterm.DefaultBigText.WithLetters(cyan, purple).Render()
|
||||
|
||||
pterm.DefaultBulletList.WithItems([]pterm.BulletListItem{
|
||||
{Level: 0, Text: "ADDR : " + *config.Addr},
|
||||
{Level: 0, Text: "PORT : " + fmt.Sprint(*config.Port)},
|
||||
{Level: 0, Text: "DNS : " + *config.Dns},
|
||||
{Level: 0, Text: "DEBUG : " + fmt.Sprint(*config.Debug)},
|
||||
}).Render()
|
||||
|
||||
if allowedHosts != nil && len(allowedHosts) > 0 {
|
||||
log.Info("White listed urls: ", allowedHosts)
|
||||
}
|
||||
|
||||
if *allowedPattern != "" {
|
||||
log.Info("Regex Pattern: ", *allowedPattern)
|
||||
}
|
||||
}
|
||||
|
||||
func PrintSimpleInfo() {
|
||||
fmt.Println("")
|
||||
fmt.Println("- ADDR : ", *config.Addr)
|
||||
fmt.Println("- PORT : ", *config.Port)
|
||||
fmt.Println("- DNS : ", *config.Dns)
|
||||
fmt.Println("- DEBUG : ", *config.Debug)
|
||||
fmt.Println("")
|
||||
}
|
Loading…
Reference in New Issue
Block a user