From f3563901047968e76a29483fc8defff2aa9e3f42 Mon Sep 17 00:00:00 2001
From: Justin Reichardt <mrjars@gmail.com>
Date: Thu, 9 Dec 2021 16:55:14 -0600
Subject: [PATCH] Adds sites to tmp file

---
 src/rhosts.go | 27 ++++++++++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

diff --git a/src/rhosts.go b/src/rhosts.go
index c8b8d8e..a04ca00 100644
--- a/src/rhosts.go
+++ b/src/rhosts.go
@@ -42,7 +42,7 @@ func main() {
 	log.Print("Downloads:\n",downloads)
 	copystatichosts(tmpdir, hostsloc)
 	downloadcontent(downloads, tmpdir)
-	
+	writesites(sites, tmpdir)
 }
 
 func sysdetect (tmpdir, hostsloc, cfgloc *string) {
@@ -200,3 +200,28 @@ func downloadcontent(downloads []string, tmpdir string) {
 		}
 	}
 }
+
+func writesites(sites []string, tmpdir string) error {
+	var err error = nil
+	fileloc := tmpdir + "rhosts"
+	log.Print("Opening: " + fileloc)
+	file,err := os.OpenFile(fileloc, os.O_APPEND|os.O_WRONLY, 0644)
+	defer file.Close()
+	if (err != nil) {
+		log.Print(err)
+		return err
+	}
+	_,err = file.WriteString("# rhosts sites\n")
+	if (err != nil){
+		log.Print(err)
+		return err
+	}
+	for _,s := range sites {
+		_,err = file.WriteString(s)
+		if (err != nil){
+			log.Print(err)
+			break
+		}
+	}
+	return err
+}