justin
/
rhosts
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Detects whitelist in the options 

Passes the addresses of the variables when parsing the config file
Fills out the whitelist string array with config entries
stable
Justin Reichardt 2022-03-01 10:44:46 -06:00
parent c32bed5041
commit 901f8c537e
1 changed files with 23 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/rhosts.go
View File

@ -60,8 +60,9 @@ func main() {
sysdetect (&tmpdir, &hostsloc, &cfgloc) sysdetect (&tmpdir, &hostsloc, &cfgloc)
for true { for true {
var sites, downloads, whitelist []string
err := error(nil) err := error(nil)
sites, downloads, err := cfgparse(cfgloc) err = cfgparse(&sites, &downloads, &whitelist, cfgloc)
if (err != nil){ if (err != nil){
log.Print("Failed to parse config file") log.Print("Failed to parse config file")
continue continue
@ -124,16 +125,14 @@ func sysdetect (tmpdir, hostsloc, cfgloc *string) {
} }
// cfgparse recieves the location of the config file and returns a list of sites to add and content to download // cfgparse recieves the location of the config file and returns a list of sites to add and content to download
func cfgparse (cfgloc string) ([]string, []string, error){ func cfgparse (sites, downloads, whitelist *[]string, cfgloc string) (error){
var err error=nil var err error=nil
var downloads []string
var sites []string
log.Print("Opening: ", cfgloc) log.Print("Opening: ", cfgloc)
file, err := os.Open(cfgloc) file, err := os.Open(cfgloc)
defer file.Close() defer file.Close()
if err != nil { if err != nil {
log.Print(err) log.Print(err)
return nil, nil,err return err
} }
filebuf := bufio.NewScanner(file) filebuf := bufio.NewScanner(file)
filebuf.Split(bufio.ScanLines) filebuf.Split(bufio.ScanLines)
@ -141,17 +140,19 @@ func cfgparse (cfgloc string) ([]string, []string, error){
state, body := cfgparseline(filebuf.Text()) state, body := cfgparseline(filebuf.Text())
switch state { switch state {
case 3: case 3:
sites =append(sites,body) *sites =append(*sites,body)
case 4: case 4:
downloads = append(downloads,body) *downloads = append(*downloads,body)
case 5:
*whitelist = append(*whitelist,body)
} }
} }
err = filebuf.Err() err = filebuf.Err()
if err != nil { if err != nil {
log.Print(err) log.Print(err)
return nil, nil, err return err
} }
return sites, downloads, err return err
} }
// cfgparseline reads a single line of the config and returns the type and content of the line // cfgparseline reads a single line of the config and returns the type and content of the line
@ -162,6 +163,7 @@ func cfgparseline(buf string) (uint8, string){
// 2 - Comment // 2 - Comment
// 3 - Site // 3 - Site
// 4 - Download // 4 - Download
// 5 - Whitelist
var state uint8= 0 var state uint8= 0
body :=buf[:] body :=buf[:]
for i:=0; i<len(buf);i++ { for i:=0; i<len(buf);i++ {
@ -195,6 +197,18 @@ func cfgparseline(buf string) (uint8, string){
state = 0 state = 0
} }
//compare buf[i:(i+3)] to "site" //compare buf[i:(i+3)] to "site"
case 'w':
if (len(buf) < i+10) {
state = 1
break
}
if (buf[i:(i+10)] == "whitelist=") {
i +=10
state = 5
body = buf[i:]
} else{
state = 1
}
} }
if (state !=0){ if (state !=0){
return state,body return state,body