From 99b17d68b0777402b17056ee1962581c031c8266 Mon Sep 17 00:00:00 2001
From: Justin Reichardt <mrjars@gmail.com>
Date: Thu, 9 Dec 2021 16:08:57 -0600
Subject: [PATCH] Performs downloads to tmp directory

---
 src/rhosts.go | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/src/rhosts.go b/src/rhosts.go
index fdbc29d..89e95bd 100644
--- a/src/rhosts.go
+++ b/src/rhosts.go
@@ -24,8 +24,10 @@ package main
 import (
 	"runtime"
 	"os"
+	"io"
 	"bufio"
 	"log"
+	"net/http"
 )
 
 func main() {
@@ -38,6 +40,8 @@ func main() {
 	sites, downloads := cfgparse(cfgloc)
 	log.Print("Sites:\n",sites)
 	log.Print("Downloads:\n",downloads)
+
+	downloadcontent(downloads, tmpdir)
 	
 }
 
@@ -132,3 +136,27 @@ func cfgparseline(buf string) (uint8, string){
 	}
 	return state, body
 }
+
+func downloadcontent(downloads []string, tmpdir string) {
+	fileloc := tmpdir + "rhostsdown"
+	log.Print("Opening: ", fileloc)
+	file,err := os.Create(fileloc)
+	if (err != nil) {
+		log.Fatal(err)
+	}
+	defer file.Close()
+
+	for _, d := range downloads {
+		log.Print("Downloading: ",d)
+		response, err := http.Get(d)
+		if (err !=nil) {
+			log.Print(err)
+		}else{
+			_,err := io.Copy(file,response.Body)
+			if (err != nil){
+				log.Print(err)
+			}
+		}
+		defer response.Body.Close()
+	}
+}