Merge pull request #1 from evilpacket/master

Add --root option for flexability

main.go

@@ -15,6 +15,7 @@ import (
 )
 
 var dir *string
+var root *string
 var host *string
 var port *string
 var iport int
@@ -27,11 +28,12 @@ func init() {
 	host = flag.String("host", "localhost", "host for the file server to listen on")
 	port = flag.String("port", "9999", "port for the file server to listen on")
 	dir = flag.String("dir", "", "directory with code to scan")
+	root = flag.String("root", "", "root to serve from")
 	burpBuddyURL = flag.String("burpbuddy", "http://localhost:8001", "HTTP API URL for burpbuddy")
 	flag.Parse()
 
 	if *dir == "" {
-		log.Fatal("dir is required")
+		log.Fatal("--dir is required")
 	}
 
 	d, err := filepath.Abs(*dir)
@@ -45,13 +47,13 @@ func init() {
 		log.Fatalf("Error converting port to int: %s", err.Error())
 	}
 	count = 0
-	localURL = fmt.Sprintf("http://%s:%s", *host, *port)
+	localURL = fmt.Sprintf("http://%s:%s%s", *host, *port, *root)
 }
 
 func main() {
 
 	go func() {
-		log.Fatal(http.ListenAndServe(*host+":"+*port, http.FileServer(http.Dir(*dir))))
+		log.Fatal(http.ListenAndServe(*host+":"+*port, http.StripPrefix(*root + "/", http.FileServer(http.Dir(*dir)))))
 	}()
 
 	log.Printf("Static file server listening on %s, serving %s", localURL, *dir)