Micro optimization pt. 2

.gitignore

@@ -1,6 +1,3 @@
 /.idea/
 .DS_Store
-/od-database-crawler
+/oddb-go
-*.log
-/queue/
-/crawled/

.travis.yml

@@ -1,5 +0,0 @@
-language: go
-
-go:
-  - "1.11.x"
-  - master

README.md

@@ -1,24 +1,2 @@
-# od-database Go crawler 🚀
+# oddb Go crawler
-[![Build Status](https://travis-ci.org/terorie/od-database-crawler.svg?branch=master)](https://travis-ci.org/terorie/od-database-crawler)
 > by terorie 2018 :P
-
- * Crawler for [__OD-Database__](https://github.com/simon987/od-database)
- * Crawls HTTP open directories (standard Web Server Listings)
- * Gets name, path, size and modification time of all files
- * Lightweight and fast: __over 9000 requests per second__ on a standard laptop
-
-https://od-db.the-eye.eu/
-
-#### Usage
-
- 1. With Config File (if `config.yml` found in working dir)
-    - Download [default config](https://github.com/terorie/od-database-crawler/blob/master/config.yml)
-    - Set `server.url` and `server.token`
-    - Start with `./od-database-crawler server --config <file>`
- 
- 2. With Flags or env
-    - Override config file if it exists
-    - `--help` for list of flags
-    - Every flag is available as an environment variable:
-      `--server.crawl_stats` ➡️ `OD_SERVER_CRAWL_STATS`
-    - Start with `./od-database-crawler server <flags>`

config.go

@@ -1,176 +1,62 @@
 package main
 
 import (
-	"bufio"
 	"fmt"
 	"github.com/sirupsen/logrus"
 	"github.com/spf13/viper"
-	"io"
 	"os"
-	"path/filepath"
-	"strings"
 	"time"
 )
 
 var config struct {
 	ServerUrl  string
 	Token      string
-	ServerTimeout time.Duration
-	Recheck    time.Duration
-	ChunkSize  int64
 	Retries    int
 	Workers    int
-	UserAgent  string
 	Tasks      int32
+	CrawlStats time.Duration
+	AllocStats time.Duration
 	Verbose    bool
-	PrintHTTP  bool
-	JobBufferSize int
 }
 
 const (
 	ConfServerUrl  = "server.url"
 	ConfToken      = "server.token"
-	ConfServerTimeout = "server.timeout"
-	ConfRecheck    = "server.recheck"
-	ConfCooldown   = "server.cooldown"
-	ConfChunkSize  = "server.upload_chunk"
-	ConfUploadRetries = "server.upload_retries"
-	ConfUploadRetryInterval = "server.upload_retry_interval"
-
 	ConfTasks      = "crawl.tasks"
 	ConfRetries    = "crawl.retries"
 	ConfWorkers    = "crawl.connections"
-	ConfUserAgent  = "crawl.user-agent"
-	ConfDialTimeout = "crawl.dial_timeout"
-	ConfTimeout    = "crawl.timeout"
-	ConfJobBufferSize = "crawl.job_buffer"
-
 	ConfCrawlStats = "output.crawl_stats"
 	ConfAllocStats = "output.resource_stats"
 	ConfVerbose    = "output.verbose"
-	ConfPrintHTTP  = "output.http"
-	ConfLogFile    = "output.log"
 )
 
 func prepareConfig() {
-	pf := rootCmd.PersistentFlags()
+	viper.SetDefault(ConfRetries, 5)
-
+	viper.SetDefault(ConfWorkers, 2)
-	bind := func(s string) {
+	viper.SetDefault(ConfTasks, 3)
-		if err := viper.BindPFlag(s, pf.Lookup(s)); err != nil {
+	viper.SetDefault(ConfCrawlStats, 3 * time.Second)
-			panic(err)
+	viper.SetDefault(ConfAllocStats, 0)
-		}
+	viper.SetDefault(ConfVerbose, false)
-		var envKey string
-		envKey = strings.Replace(s, ".", "_", -1)
-		envKey = strings.ToUpper(envKey)
-		envKey = "OD_" + envKey
-		if err := viper.BindEnv(s, envKey); err != nil {
-			panic(err)
-		}
-	}
-
-	pf.SortFlags = false
-	pf.StringVar(&configFile, "config", "", "Config file")
-	configFile = os.Getenv("OD_CONFIG")
-
-	pf.String(ConfServerUrl, "http://od-db.the-eye.eu/api", "OD-DB server URL")
-	bind(ConfServerUrl)
-
-	pf.String(ConfToken, "", "OD-DB access token (env OD_SERVER_TOKEN)")
-	bind(ConfToken)
-
-	pf.Duration(ConfServerTimeout, 60 * time.Second, "OD-DB request timeout")
-	bind(ConfServerTimeout)
-
-	pf.Duration(ConfRecheck, 1 * time.Second, "OD-DB: Poll interval for new jobs")
-	bind(ConfRecheck)
-
-	pf.Duration(ConfCooldown, 30 * time.Second, "OD-DB: Time to wait after a server-side error")
-	bind(ConfCooldown)
-
-	pf.String(ConfChunkSize, "1 MB", "OD-DB: Result upload chunk size")
-	bind(ConfChunkSize)
-
-	pf.Uint(ConfUploadRetries, 10, "OD-DB: Max upload retries")
-	bind(ConfUploadRetries)
-
-	pf.Duration(ConfUploadRetryInterval, 30 * time.Second, "OD-DB: Time to wait between upload retries")
-	bind(ConfUploadRetryInterval)
-
-	pf.Uint(ConfTasks, 100, "Crawler: Max concurrent tasks")
-	bind(ConfTasks)
-
-	pf.Uint(ConfWorkers, 4, "Crawler: Connections per server")
-	bind(ConfWorkers)
-
-	pf.Uint(ConfRetries, 5, "Crawler: Request retries")
-	bind(ConfRetries)
-
-	pf.Duration(ConfDialTimeout, 10 * time.Second, "Crawler: Handshake timeout")
-	bind(ConfDialTimeout)
-
-	pf.Duration(ConfTimeout, 30 * time.Second, "Crawler: Request timeout")
-	bind(ConfTimeout)
-
-	pf.String(ConfUserAgent, "Mozilla/5.0 (X11; od-database-crawler) Gecko/20100101 Firefox/52.0", "Crawler: User-Agent")
-	bind(ConfUserAgent)
-
-	pf.Uint(ConfJobBufferSize, 5000, "Crawler: Task queue cache size")
-	bind(ConfJobBufferSize)
-
-	pf.Duration(ConfCrawlStats, time.Second, "Log: Crawl stats interval")
-	bind(ConfCrawlStats)
-
-	pf.Duration(ConfAllocStats, 10 * time.Second, "Log: Resource stats interval")
-	bind(ConfAllocStats)
-
-	pf.Bool(ConfVerbose, false, "Log: Print every listed dir")
-	bind(ConfVerbose)
-
-	pf.Bool(ConfPrintHTTP, false, "Log: Print HTTP client errors")
-	bind(ConfPrintHTTP)
-
-	pf.String(ConfLogFile, "crawler.log", "Log file")
-	bind(ConfLogFile)
 }
 
 func readConfig() {
-	// If config.yml in working dir, use it
+	viper.AddConfigPath(".")
-	if _, err := os.Stat("config.yml"); err == nil {
+	viper.SetConfigName("config")
-		configFile = "config.yml"
+	err := viper.ReadInConfig()
-	}
-
-	if configFile != "" {
-		var err error
-		confPath, err := filepath.Abs(configFile)
-		if err != nil { panic(err) }
-
-		viper.SetConfigFile(confPath)
-		err = viper.ReadInConfig()
 	if err != nil {
 		fmt.Fprintln(os.Stderr, err)
 		os.Exit(1)
 	}
-	}
 
 	config.ServerUrl = viper.GetString(ConfServerUrl)
-	if config.ServerUrl == "" {
+	//if config.ServerUrl == "" {
-		configMissing(ConfServerUrl)
+	//	configMissing(ConfServerUrl)
-	}
+	//}
-	config.ServerUrl = strings.TrimRight(config.ServerUrl, "/")
 
 	config.Token = viper.GetString(ConfToken)
-	if config.Token == "" {
+	//if config.Token == "" {
-		configMissing(ConfToken)
+	//	configMissing(ConfToken)
-	}
+	//}
-
-	config.ServerTimeout = viper.GetDuration(ConfServerTimeout)
-
-	config.Recheck = viper.GetDuration(ConfRecheck)
-
-	config.ChunkSize = int64(viper.GetSizeInBytes(ConfChunkSize))
-	if config.ChunkSize < 100 {
-		configOOB(ConfChunkSize, config.ChunkSize)
-	}
 
 	config.Retries = viper.GetInt(ConfRetries)
 	if config.Retries < 0 {
@@ -187,31 +73,14 @@ func readConfig() {
 		configOOB(ConfTasks, int(config.Tasks))
 	}
 
-	config.UserAgent = viper.GetString(ConfUserAgent)
+	config.CrawlStats = viper.GetDuration(ConfCrawlStats)
 
-	setDialTimeout(viper.GetDuration(ConfDialTimeout))
+	config.AllocStats = viper.GetDuration(ConfAllocStats)
-
-	setTimeout(viper.GetDuration(ConfTimeout))
-
-	config.JobBufferSize = viper.GetInt(ConfJobBufferSize)
 
 	config.Verbose = viper.GetBool(ConfVerbose)
 	if config.Verbose {
 		logrus.SetLevel(logrus.DebugLevel)
 	}
-
-	if filePath := viper.GetString(ConfLogFile); filePath != "" {
-		f, err := os.OpenFile(filePath, os.O_CREATE | os.O_WRONLY | os.O_APPEND, 0644)
-		bufWriter := bufio.NewWriter(f)
-		if err != nil { panic(err) }
-		exitHooks.Add(func() {
-			bufWriter.Flush()
-			f.Close()
-		})
-		logrus.SetOutput(io.MultiWriter(os.Stdout, bufWriter))
-	}
-
-	config.PrintHTTP = viper.GetBool(ConfPrintHTTP)
 }
 
 func configMissing(key string) {
@@ -219,7 +88,7 @@ func configMissing(key string) {
 	os.Exit(1)
 }
 
-func configOOB(key string, v interface{}) {
+func configOOB(key string, v int) {
-	fmt.Fprintf(os.Stderr, "config: illegal value %v for key %s!\n", v, key)
+	fmt.Fprintf(os.Stderr, "config: illegal value %d for %key!\n", v, key)
 	os.Exit(1)
 }

config.yml

@@ -1,84 +1,26 @@
 # OD-Database server settings
 server:
   # Connection URL
-  url: http://od-db.mine.terorie.com/api
+  url: localhost:6969
-
   # Server auth token
   token:
 
-  # Request timeout
-  timeout: 60s
-
-  # Recheck interval
-  # The crawler periodically asks the server
-  # for new jobs. Sets the minimum wait time
-  # between /task/get requests to the server.
-  recheck: 1s
-
-  # Time to wait after receiving an error
-  # from the server. Doesn't apply to uploads.
-  cooldown: 30s
-
-  # Upload chunk size
-  # If the value is too high, the upload fails.
-  upload_chunk: 1 MB
-
-  upload_retries: 10
-  upload_retry_interval: 30s
-
 # Log output settings
 output:
   # Crawl statistics
   crawl_stats: 1s
-
   # CPU/RAM/Job queue stats
-  resource_stats: 10s
+  resource_stats: 1s
-
   # More output? (Every listed dir)
   verbose: false
 
-  # Print HTTP errors (Super spammy)
-  http: false
-
-  # Log file
-  # If empty, no log file is created.
-  log: crawler.log
-
 # Crawler settings
 crawl:
-  # Number of sites that can be processed at once
+  # Number of sites that can be
-  tasks: 100
+  # processed at once
-
+  tasks: 3
   # Number of connections per site
-  # Please be careful with this setting!
+  connections: 2
-  # The crawler fires fast and more than
-  # ten connections can overwhelm a server.
-  connections: 4
-
   # How often to retry getting data
   # from the site before giving up
   retries: 5
-
-  # Time before discarding a failed connection attempt
-  dial_timeout: 10s
-
-  # Time before discarding a network request
-  timeout: 30s
-
-  # Crawler User-Agent
-  # If empty, no User-Agent header is sent.
-  user-agent: "Mozilla/5.0 (X11; od-database-crawler) Gecko/20100101 Firefox/52.0"
-
-  # Job buffer size (per task)
-  # Higher values cause less disk writes
-  # but require more memory.
-  #
-  # The job queue contains all URLs
-  # that should be crawled next.
-  # As it grows very large over time,
-  # it's kept mainly on disk.
-  # This sets how many jobs are kept
-  # in memory.
-  # A negative value will cause all jobs
-  # to be stored in memory. (Don't do this)
-  job_buffer: 5000

crawl.go

@@ -2,45 +2,28 @@ package main
 
 import (
 	"bytes"
-	"crypto/tls"
+	"encoding/base64"
-	"github.com/terorie/od-database-crawler/ds/redblackhash"
+	"fmt"
-	"github.com/terorie/od-database-crawler/fasturl"
+	"github.com/sirupsen/logrus"
 	"github.com/valyala/fasthttp"
 	"golang.org/x/crypto/blake2b"
 	"golang.org/x/net/html"
-	"net"
+	"golang.org/x/net/html/atom"
+	"net/url"
 	"path"
 	"strconv"
 	"strings"
 	"time"
 )
 
-var client = fasthttp.Client {
+var client fasthttp.Client
-	TLSConfig: &tls.Config{
-		InsecureSkipVerify: true,
-	},
-}
 
-func setDialTimeout(d time.Duration) {
+func GetDir(j *Job, f *File) (links []url.URL, err error) {
-	client.Dial = func(addr string) (net.Conn, error) {
-		return fasthttp.DialTimeout(addr, d)
-	}
-}
-
-func setTimeout(d time.Duration) {
-	client.ReadTimeout = d
-	client.WriteTimeout = d / 2
-}
-
-func GetDir(j *Job, f *File) (links []fasturl.URL, err error) {
 	f.IsDir = true
 	f.Name = path.Base(j.Uri.Path)
 
 	req := fasthttp.AcquireRequest()
-	if config.UserAgent != "" {
+	req.SetRequestURI(j.Uri.String())
-		req.Header.SetUserAgent(config.UserAgent)
-	}
-	req.SetRequestURI(j.UriStr)
 
 	res := fasthttp.AcquireResponse()
 	defer fasthttp.ReleaseResponse(res)
@@ -49,99 +32,98 @@ func GetDir(j *Job, f *File) (links []fasturl.URL, err error) {
 	fasthttp.ReleaseRequest(req)
 
 	if err != nil {
+		logrus.Error(err)
 		return
 	}
 
 	err = checkStatusCode(res.StatusCode())
-	if err != nil {
+	if err != nil { return }
-		return
-	}
 
 	body := res.Body()
-	return ParseDir(body, &j.Uri)
-}
-
-func ParseDir(body []byte, baseUri *fasturl.URL) (links []fasturl.URL, err error) {
 	doc := html.NewTokenizer(bytes.NewReader(body))
 
 	var linkHref string
+	var linkTexts []string
 	for {
-		err = nil
-
 		tokenType := doc.Next()
+		token := doc.Token()
 		if tokenType == html.ErrorToken {
 			break
 		}
 
 		switch tokenType {
 		case html.StartTagToken:
-			name, hasAttr := doc.TagName()
+			if token.DataAtom == atom.A {
-			if len(name) == 1 && name[0] == 'a' {
+				for _, attr := range token.Attr {
-				for hasAttr {
+					if attr.Key == "href" {
-					var ks, vs []byte
+						linkHref = attr.Val
-					ks, vs, hasAttr = doc.TagAttr()
-					if bytes.Equal(ks, []byte("href")) {
-						// TODO Check escape
-						linkHref = string(vs)
 						break
 					}
 				}
 			}
 
+		case html.TextToken:
+			if linkHref != "" {
+				linkTexts = append(linkTexts, token.Data)
+			}
+
 		case html.EndTagToken:
-			name, _ := doc.TagName()
+			if linkHref != "" && token.DataAtom == atom.A {
-			if len(name) == 1 && name[0] == 'a' {
 				// Copy params
 				href := linkHref
+				linkText := strings.Join(linkTexts, " ")
 
 				// Reset params
 				linkHref = ""
+				linkTexts = nil
 
-				if strings.LastIndexByte(href, '?') != -1 {
+				// TODO Optimized decision tree
-					continue
+				for _, entry := range urlBlackList {
+					if href == entry {
+						goto nextToken
+					}
+				}
+				for _, entry := range urlPartBlackList {
+					if strings.Contains(href, entry) {
+						goto nextToken
+					}
+				}
+				for _, entry := range fileNameBlackList {
+					if strings.Contains(linkText, entry) {
+						goto nextToken
+					}
 				}
 
-				switch href {
+				subref, err := url.Parse(href)
-				case "", " ", ".", "..", "/":
+				if err != nil { continue }
-					continue
-				}
 
-				if strings.Contains(href, "../") {
+				link := *j.Uri.ResolveReference(subref)
-					continue
-				}
 
-				var link fasturl.URL
+				if link.Scheme != j.Uri.Scheme ||
-				err = baseUri.ParseRel(&link, href)
+					link.Host != j.Uri.Host ||
-				if err != nil {
+					link.Path == j.Uri.Path ||
-					continue
+					!strings.HasPrefix(link.Path, j.Uri.Path) {
-				}
-
-				if link.Scheme != baseUri.Scheme ||
-					link.Host != baseUri.Host ||
-					link.Path == baseUri.Path ||
-					!strings.HasPrefix(link.Path, baseUri.Path) {
 					continue
 				}
 
 				links = append(links, link)
 			}
 		}
+
+		nextToken:
 	}
 
 	return
 }
 
-func GetFile(u fasturl.URL, f *File) (err error) {
+func GetFile(u url.URL, f *File) (err error) {
 	f.IsDir = false
 	u.Path = path.Clean(u.Path)
 	f.Name = path.Base(u.Path)
-	f.Path = strings.Trim(path.Dir(u.Path), "/")
+	f.Path = strings.Trim(u.Path, "/")
 
 	req := fasthttp.AcquireRequest()
 	req.Header.SetMethod("HEAD")
-	if config.UserAgent != "" {
-		req.Header.SetUserAgent(config.UserAgent)
-	}
 	req.SetRequestURI(u.String())
 
 	res := fasthttp.AcquireResponse()
@@ -151,22 +133,19 @@ func GetFile(u fasturl.URL, f *File) (err error) {
 	err = client.Do(req, res)
 	fasthttp.ReleaseRequest(req)
 
-	if err != nil {
+	if err != nil { return }
-		return
-	}
 
 	err = checkStatusCode(res.StatusCode())
-	if err != nil {
+	if err != nil { return }
-		return
-	}
 
-	f.applyContentLength(string(res.Header.Peek("content-length")))
+	// TODO Inefficient af
-	f.applyLastModified(string(res.Header.Peek("last-modified")))
+	header := res.Header.Header()
+	f.ParseHeader(header)
 
 	return nil
 }
 
-func (f *File) HashDir(links []fasturl.URL) (o redblackhash.Key) {
+func (f *File) HashDir(links []url.URL) string {
 	h, _ := blake2b.New256(nil)
 	h.Write([]byte(f.Name))
 	for _, link := range links {
@@ -174,46 +153,67 @@ func (f *File) HashDir(links []fasturl.URL) (o redblackhash.Key) {
 		h.Write([]byte(fileName))
 	}
 	sum := h.Sum(nil)
-	copy(o[:redblackhash.KeySize], sum)
+	b64sum := base64.StdEncoding.EncodeToString(sum)
-	return
+	return b64sum
 }
 
-func (f *File) applyContentLength(v string) {
+func (f *File) ParseHeader(h []byte) {
-	if v == "" {
+	var k1, k2 int
-		return
+	var v1, v2 int
+
+	// Simple finite state machine
+	state := 0
+	for i, b := range h {
+		switch state {
+		case 0:
+			if b == byte(':') {
+				state = 1
+				k2 = i
 			}
+
+		case 1:
+			state = 2
+
+		case 2:
+			state = 3
+			v1 = i
+
+		case 3:
+			if b == byte('\r') {
+				state = 4
+			}
+
+		case 4:
+			state = 0
+			v2 = i - 1
+
+			key := string(h[k1:k2])
+			val := string(h[v1:v2])
+			k1 = i
+
+			f.applyHeader(key, val)
+		}
+	}
+
+}
+
+func (f *File) applyHeader(k, v string) {
+	switch k {
+	case "content-length":
 		size, err := strconv.ParseInt(v, 10, 64)
-	if err != nil {
+		if err != nil { break }
-		return
+		if size < 0 { break }
-	}
-	if size < 0 {
-		return
-	}
 		f.Size = size
-}
 
-// TODO Cleanup
+	case "last-modified":
-func (f *File) applyLastModified(v string) {
-	if v == "" {
-		return
-	}
-	var t time.Time
 		var err error
-	t, err = time.Parse(time.RFC1123, v)
+		f.MTime, err = time.Parse(time.RFC1123, v)
-	if err == nil {
+		if err == nil { break }
-		f.MTime = t.Unix()
+		f.MTime, err = time.Parse(time.RFC850, v)
-		return
+		if err == nil { break }
-	}
-	t, err = time.Parse(time.RFC850, v)
-	if err == nil {
-		f.MTime = t.Unix()
-		return
-	}
 		// TODO Parse asctime
-	t, err = time.Parse("2006-01-02", v[:10])
+		f.MTime, err = time.Parse("2006-01-02", v[:10])
-	if err == nil {
+		if err == nil { break }
-		f.MTime = t.Unix()
-		return
 	}
 }
 
@@ -221,7 +221,53 @@ func checkStatusCode(status int) error {
 	switch status {
 	case fasthttp.StatusOK:
 		return nil
+
+	case fasthttp.StatusTooManyRequests:
+		return ErrRateLimit
+
+	case fasthttp.StatusForbidden,
+		fasthttp.StatusUnauthorized:
+		return ErrForbidden
+
 	default:
-		return &HttpError{status}
+		return fmt.Errorf("got HTTP status %d", status)
 	}
 }
+
+var urlBlackList = [...]string {
+	"",
+	" ",
+	".",
+	"..",
+	"/",
+}
+
+var urlPartBlackList = [...]string {
+	"?C=N&O=D",
+	"?C=M&O=A",
+	"?C=S&O=A",
+	"?C=D&O=A",
+	"?C=N;O=D",
+	"?C=M;O=A",
+	"?C=M&O=D",
+	"?C=S;O=A",
+	"?C=S&O=D",
+	"?C=D;O=A",
+	"?MA",
+	"?SA",
+	"?DA",
+	"?ND",
+	"?C=N&O=A",
+	"?C=N&O=A",
+	"?M=A",
+	"?N=D",
+	"?S=A",
+	"?D=A",
+}
+
+var fileNameBlackList = [...]string {
+	"Parent Directory",
+	" Parent Directory",
+	"../",
+}
+

crawl_nginx_test.go

@@ -1,117 +0,0 @@
-package main
-
-import (
-	"github.com/terorie/od-database-crawler/fasturl"
-	"testing"
-)
-
-func TestParseDirNginx(t *testing.T) {
-	var u fasturl.URL
-	err := u.Parse("https://the-eye.eu/public/")
-	if err != nil {
-		t.Fatal("Failed to parse URL", err)
-	}
-
-	links, err := ParseDir([]byte(nginxListing), &u)
-	if err != nil {
-		t.Fatal("Failed to extract links", err)
-	}
-
-	if len(links) != len(nginxLinks) {
-		t.Fatalf("Expected %d links, got %d",
-			len(nginxLinks), len(links))
-	}
-
-	for i := 0; i < len(links); i++ {
-		gotLink := links[i].String()
-		expLink := nginxLinks[i]
-
-		if gotLink != expLink {
-			t.Errorf(`Expected "%s" got "%s"`,
-				expLink, gotLink)
-		}
-	}
-}
-
-var nginxLinks = []string {
-	"https://the-eye.eu/public/AppleArchive/",
-	"https://the-eye.eu/public/AudioBooks/",
-	"https://the-eye.eu/public/Books/",
-	"https://the-eye.eu/public/Comics/",
-	"https://the-eye.eu/public/Games/",
-	"https://the-eye.eu/public/Icons/",
-	"https://the-eye.eu/public/Images/",
-	"https://the-eye.eu/public/JFK_Files/",
-	"https://the-eye.eu/public/MSDN/",
-	"https://the-eye.eu/public/Music/",
-	"https://the-eye.eu/public/Operating%20Systems/",
-	"https://the-eye.eu/public/Posters/",
-	"https://the-eye.eu/public/Psychedelics/",
-	"https://the-eye.eu/public/Psychoactives/",
-	"https://the-eye.eu/public/Radio/",
-	"https://the-eye.eu/public/Random/",
-	"https://the-eye.eu/public/Site-Dumps/",
-	"https://the-eye.eu/public/Software/",
-	"https://the-eye.eu/public/Strategic%20Intelligence%20Network/",
-	"https://the-eye.eu/public/WorldTracker.org/",
-	"https://the-eye.eu/public/concen.org/",
-	"https://the-eye.eu/public/freenrg.info/",
-	"https://the-eye.eu/public/murdercube.com/",
-	"https://the-eye.eu/public/parazite/",
-	"https://the-eye.eu/public/ripreddit/",
-	"https://the-eye.eu/public/rom/",
-	"https://the-eye.eu/public/touhou/",
-	"https://the-eye.eu/public/vns/",
-	"https://the-eye.eu/public/xbins/",
-	"https://the-eye.eu/public/xbins.diodematrix/",
-	"https://the-eye.eu/public/Rclone_for_Scrubs.pdf",
-	"https://the-eye.eu/public/Wget_Linux_Guide.pdf",
-	"https://the-eye.eu/public/Wget_Windows_Guide.pdf",
-	"https://the-eye.eu/public/rclone_guide.pdf",
-	"https://the-eye.eu/public/wget-noobs-guide.pdf",
-	"https://the-eye.eu/public/xbox-scene_Aug2014.7z",
-}
-
-const nginxListing =
-`<html>
-<head><title>Index of /public/</title></head>
-<body bgcolor="white">
-<h1>Index of /public/</h1><hr><pre><a href="../">../</a>
-<a href="AppleArchive/">AppleArchive/</a>                                      03-Nov-2017 18:13       -
-<a href="AudioBooks/">AudioBooks/</a>                                        29-Sep-2018 19:47       -
-<a href="Books/">Books/</a>                                             27-Nov-2018 17:50       -
-<a href="Comics/">Comics/</a>                                            05-Nov-2018 21:37       -
-<a href="Games/">Games/</a>                                             28-Nov-2018 11:54       -
-<a href="Icons/">Icons/</a>                                             22-May-2018 07:47       -
-<a href="Images/">Images/</a>                                            21-Jan-2018 03:21       -
-<a href="JFK_Files/">JFK_Files/</a>                                         03-Nov-2017 17:03       -
-<a href="MSDN/">MSDN/</a>                                              03-Nov-2017 15:48       -
-<a href="Music/">Music/</a>                                             02-Mar-2018 15:47       -
-<a href="Operating%20Systems/">Operating Systems/</a>                                 25-Apr-2018 07:18       -
-<a href="Posters/">Posters/</a>                                           07-Jul-2018 01:12       -
-<a href="Psychedelics/">Psychedelics/</a>                                      11-Apr-2018 05:45       -
-<a href="Psychoactives/">Psychoactives/</a>                                     18-May-2018 02:58       -
-<a href="Radio/">Radio/</a>                                             09-Jun-2018 15:49       -
-<a href="Random/">Random/</a>                                            04-Dec-2018 12:33       -
-<a href="Site-Dumps/">Site-Dumps/</a>                                        15-Dec-2018 11:04       -
-<a href="Software/">Software/</a>                                          27-Nov-2017 00:22       -
-<a href="Strategic%20Intelligence%20Network/">Strategic Intelligence Network/</a>                    17-Nov-2017 16:35       -
-<a href="WorldTracker.org/">WorldTracker.org/</a>                                  12-Apr-2018 04:16       -
-<a href="concen.org/">concen.org/</a>                                        08-Oct-2018 14:08       -
-<a href="freenrg.info/">freenrg.info/</a>                                      19-Dec-2017 10:59       -
-<a href="murdercube.com/">murdercube.com/</a>                                    06-Dec-2017 10:45       -
-<a href="parazite/">parazite/</a>                                          20-Nov-2017 21:25       -
-<a href="ripreddit/">ripreddit/</a>                                         04-Aug-2018 14:30       -
-<a href="rom/">rom/</a>                                               28-Nov-2018 14:15       -
-<a href="touhou/">touhou/</a>                                            03-Nov-2017 11:07       -
-<a href="vns/">vns/</a>                                               03-Nov-2017 11:36       -
-<a href="xbins/">xbins/</a>                                             03-Nov-2017 17:23       -
-<a href="xbins.diodematrix/">xbins.diodematrix/</a>                                 21-Sep-2018 22:33       -
-<a href="Rclone_for_Scrubs.pdf">Rclone_for_Scrubs.pdf</a>                              04-Sep-2018 13:31    315K
-<a href="Wget_Linux_Guide.pdf">Wget_Linux_Guide.pdf</a>                               21-Dec-2017 20:28    168K
-<a href="Wget_Windows_Guide.pdf">Wget_Windows_Guide.pdf</a>                             25-Nov-2017 17:59    867K
-<a href="rclone_guide.pdf">rclone_guide.pdf</a>                                   03-Sep-2018 23:37    315K
-<a href="wget-noobs-guide.pdf">wget-noobs-guide.pdf</a>                               21-Dec-2017 20:29    168K
-<a href="xbox-scene_Aug2014.7z">xbox-scene_Aug2014.7z</a>                              26-Oct-2017 23:09      1G
-</pre><hr></body>
-</html>`

crawl_test.go

@@ -1,59 +0,0 @@
-package main
-
-import (
-	"bytes"
-	"github.com/PuerkitoBio/goquery"
-	"github.com/terorie/od-database-crawler/fasturl"
-	"net/url"
-	"strings"
-	"testing"
-)
-
-func BenchmarkParseDir(b *testing.B) {
-	for n := 0; n < b.N; n++ {
-		var u fasturl.URL
-		err := u.Parse("http://archive.ubuntu.com/ubuntu/indices/")
-		if err != nil {
-			b.Fatal("Failed to parse URL", err)
-		}
-
-		_, err = ParseDir([]byte(apache2Listing), &u)
-		if err != nil {
-			b.Fatal("Failed to extract links", err)
-		}
-	}
-}
-
-func BenchmarkParseDirReference(b *testing.B) {
-	for n := 0; n < b.N; n++ {
-		u, err := url.Parse("http://archive.ubuntu.com/ubuntu/indices/")
-		if err != nil {
-			b.Fatal("Failed to parse URL", err)
-		}
-
-		_, err = referenceParseDir([]byte(apache2Listing), u)
-		if err != nil {
-			b.Fatal("Failed to extract links", err)
-		}
-	}
-}
-
-func referenceParseDir(body []byte, baseUri *url.URL) (links []*url.URL, err error) {
-	doc, err := goquery.NewDocumentFromReader(bytes.NewReader(body))
-	if err != nil { return nil, err }
-
-	doc.Find("a[href]").Each(func(i int, s *goquery.Selection) {
-		href, _ := s.Attr("href")
-
-		sub, err := baseUri.Parse(href)
-		if err != nil { return } // continue
-
-		if !strings.HasPrefix(sub.String(), baseUri.String()) {
-			return // continue
-		}
-
-		links = append(links, sub)
-	})
-
-	return
-}

ds/redblackhash/redblack.go

@@ -1,489 +0,0 @@
-// Copyright (c) 2015, Emir Pasic. All rights reserved.
-// Use of this source code is governed by a BSD-style
-// license that can be found in the LICENSE file.
-
-// Modifications by terorie
-
-// Package redblacktree implements a red-black tree.
-//
-// Used by TreeSet and TreeMap.
-//
-// Structure is not thread safe.
-//
-// References: http://en.wikipedia.org/wiki/Red%E2%80%93black_tree
-package redblackhash
-
-import (
-	"bytes"
-	"fmt"
-	"sync"
-)
-
-const (
-	black, red color = true, false
-	KeySize = 64
-)
-
-type color bool
-type Key [KeySize]byte
-
-// Tree holds elements of the red-black tree
-type Tree struct {
-	sync.Mutex
-	Root       *Node
-	size       int
-}
-
-// Node is a single element within the tree
-type Node struct {
-	Key    Key
-	color  color
-	Left   *Node
-	Right  *Node
-	Parent *Node
-}
-
-func (k *Key) Compare(o *Key) int {
-	return bytes.Compare(k[:], o[:])
-}
-
-// Put inserts node into the tree.
-// Key should adhere to the comparator's type assertion, otherwise method panics.
-func (tree *Tree) Put(key *Key) {
-	var insertedNode *Node
-	if tree.Root == nil {
-		// Assert key is of comparator's type for initial tree
-		tree.Root = &Node{Key: *key, color: red}
-		insertedNode = tree.Root
-	} else {
-		node := tree.Root
-		loop := true
-		for loop {
-			compare := key.Compare(&node.Key)
-			switch {
-			case compare == 0:
-				node.Key = *key
-				return
-			case compare < 0:
-				if node.Left == nil {
-					node.Left = &Node{Key: *key, color: red}
-					insertedNode = node.Left
-					loop = false
-				} else {
-					node = node.Left
-				}
-			case compare > 0:
-				if node.Right == nil {
-					node.Right = &Node{Key: *key, color: red}
-					insertedNode = node.Right
-					loop = false
-				} else {
-					node = node.Right
-				}
-			}
-		}
-		insertedNode.Parent = node
-	}
-	tree.insertCase1(insertedNode)
-	tree.size++
-}
-
-// Get searches the node in the tree by key and returns its value or nil if key is not found in tree.
-// Second return parameter is true if key was found, otherwise false.
-// Key should adhere to the comparator's type assertion, otherwise method panics.
-func (tree *Tree) Get(key *Key) (found bool) {
-	node := tree.lookup(key)
-	return node != nil
-}
-
-// Remove remove the node from the tree by key.
-// Key should adhere to the comparator's type assertion, otherwise method panics.
-func (tree *Tree) Remove(key *Key) {
-	var child *Node
-	node := tree.lookup(key)
-	if node == nil {
-		return
-	}
-	if node.Left != nil && node.Right != nil {
-		pred := node.Left.maximumNode()
-		node.Key = pred.Key
-		node = pred
-	}
-	if node.Left == nil || node.Right == nil {
-		if node.Right == nil {
-			child = node.Left
-		} else {
-			child = node.Right
-		}
-		if node.color == black {
-			node.color = nodeColor(child)
-			tree.deleteCase1(node)
-		}
-		tree.replaceNode(node, child)
-		if node.Parent == nil && child != nil {
-			child.color = black
-		}
-	}
-	tree.size--
-}
-
-// Empty returns true if tree does not contain any nodes
-func (tree *Tree) Empty() bool {
-	return tree.size == 0
-}
-
-// Size returns number of nodes in the tree.
-func (tree *Tree) Size() int {
-	return tree.size
-}
-
-// Left returns the left-most (min) node or nil if tree is empty.
-func (tree *Tree) Left() *Node {
-	var parent *Node
-	current := tree.Root
-	for current != nil {
-		parent = current
-		current = current.Left
-	}
-	return parent
-}
-
-// Right returns the right-most (max) node or nil if tree is empty.
-func (tree *Tree) Right() *Node {
-	var parent *Node
-	current := tree.Root
-	for current != nil {
-		parent = current
-		current = current.Right
-	}
-	return parent
-}
-
-// Floor Finds floor node of the input key, return the floor node or nil if no floor is found.
-// Second return parameter is true if floor was found, otherwise false.
-//
-// Floor node is defined as the largest node that is smaller than or equal to the given node.
-// A floor node may not be found, either because the tree is empty, or because
-// all nodes in the tree are larger than the given node.
-//
-// Key should adhere to the comparator's type assertion, otherwise method panics.
-func (tree *Tree) Floor(key *Key) (floor *Node, found bool) {
-	found = false
-	node := tree.Root
-	for node != nil {
-		compare := key.Compare(&node.Key)
-		switch {
-		case compare == 0:
-			return node, true
-		case compare < 0:
-			node = node.Left
-		case compare > 0:
-			floor, found = node, true
-			node = node.Right
-		}
-	}
-	if found {
-		return floor, true
-	}
-	return nil, false
-}
-
-// Ceiling finds ceiling node of the input key, return the ceiling node or nil if no ceiling is found.
-// Second return parameter is true if ceiling was found, otherwise false.
-//
-// Ceiling node is defined as the smallest node that is larger than or equal to the given node.
-// A ceiling node may not be found, either because the tree is empty, or because
-// all nodes in the tree are smaller than the given node.
-//
-// Key should adhere to the comparator's type assertion, otherwise method panics.
-func (tree *Tree) Ceiling(key *Key) (ceiling *Node, found bool) {
-	found = false
-	node := tree.Root
-	for node != nil {
-		compare := key.Compare(&node.Key)
-		switch {
-		case compare == 0:
-			return node, true
-		case compare < 0:
-			ceiling, found = node, true
-			node = node.Left
-		case compare > 0:
-			node = node.Right
-		}
-	}
-	if found {
-		return ceiling, true
-	}
-	return nil, false
-}
-
-// Clear removes all nodes from the tree.
-func (tree *Tree) Clear() {
-	tree.Root = nil
-	tree.size = 0
-}
-
-// String returns a string representation of container
-func (tree *Tree) String() string {
-	str := "RedBlackTree\n"
-	if !tree.Empty() {
-		output(tree.Root, "", true, &str)
-	}
-	return str
-}
-
-func (node *Node) String() string {
-	return fmt.Sprintf("%v", node.Key)
-}
-
-func output(node *Node, prefix string, isTail bool, str *string) {
-	if node.Right != nil {
-		newPrefix := prefix
-		if isTail {
-			newPrefix += "│   "
-		} else {
-			newPrefix += "    "
-		}
-		output(node.Right, newPrefix, false, str)
-	}
-	*str += prefix
-	if isTail {
-		*str += "└── "
-	} else {
-		*str += "┌── "
-	}
-	*str += node.String() + "\n"
-	if node.Left != nil {
-		newPrefix := prefix
-		if isTail {
-			newPrefix += "    "
-		} else {
-			newPrefix += "│   "
-		}
-		output(node.Left, newPrefix, true, str)
-	}
-}
-
-func (tree *Tree) lookup(key *Key) *Node {
-	node := tree.Root
-	for node != nil {
-		compare := key.Compare(&node.Key)
-		switch {
-		case compare == 0:
-			return node
-		case compare < 0:
-			node = node.Left
-		case compare > 0:
-			node = node.Right
-		}
-	}
-	return nil
-}
-
-func (node *Node) grandparent() *Node {
-	if node != nil && node.Parent != nil {
-		return node.Parent.Parent
-	}
-	return nil
-}
-
-func (node *Node) uncle() *Node {
-	if node == nil || node.Parent == nil || node.Parent.Parent == nil {
-		return nil
-	}
-	return node.Parent.sibling()
-}
-
-func (node *Node) sibling() *Node {
-	if node == nil || node.Parent == nil {
-		return nil
-	}
-	if node == node.Parent.Left {
-		return node.Parent.Right
-	}
-	return node.Parent.Left
-}
-
-func (tree *Tree) rotateLeft(node *Node) {
-	right := node.Right
-	tree.replaceNode(node, right)
-	node.Right = right.Left
-	if right.Left != nil {
-		right.Left.Parent = node
-	}
-	right.Left = node
-	node.Parent = right
-}
-
-func (tree *Tree) rotateRight(node *Node) {
-	left := node.Left
-	tree.replaceNode(node, left)
-	node.Left = left.Right
-	if left.Right != nil {
-		left.Right.Parent = node
-	}
-	left.Right = node
-	node.Parent = left
-}
-
-func (tree *Tree) replaceNode(old *Node, new *Node) {
-	if old.Parent == nil {
-		tree.Root = new
-	} else {
-		if old == old.Parent.Left {
-			old.Parent.Left = new
-		} else {
-			old.Parent.Right = new
-		}
-	}
-	if new != nil {
-		new.Parent = old.Parent
-	}
-}
-
-func (tree *Tree) insertCase1(node *Node) {
-	if node.Parent == nil {
-		node.color = black
-	} else {
-		tree.insertCase2(node)
-	}
-}
-
-func (tree *Tree) insertCase2(node *Node) {
-	if nodeColor(node.Parent) == black {
-		return
-	}
-	tree.insertCase3(node)
-}
-
-func (tree *Tree) insertCase3(node *Node) {
-	uncle := node.uncle()
-	if nodeColor(uncle) == red {
-		node.Parent.color = black
-		uncle.color = black
-		node.grandparent().color = red
-		tree.insertCase1(node.grandparent())
-	} else {
-		tree.insertCase4(node)
-	}
-}
-
-func (tree *Tree) insertCase4(node *Node) {
-	grandparent := node.grandparent()
-	if node == node.Parent.Right && node.Parent == grandparent.Left {
-		tree.rotateLeft(node.Parent)
-		node = node.Left
-	} else if node == node.Parent.Left && node.Parent == grandparent.Right {
-		tree.rotateRight(node.Parent)
-		node = node.Right
-	}
-	tree.insertCase5(node)
-}
-
-func (tree *Tree) insertCase5(node *Node) {
-	node.Parent.color = black
-	grandparent := node.grandparent()
-	grandparent.color = red
-	if node == node.Parent.Left && node.Parent == grandparent.Left {
-		tree.rotateRight(grandparent)
-	} else if node == node.Parent.Right && node.Parent == grandparent.Right {
-		tree.rotateLeft(grandparent)
-	}
-}
-
-func (node *Node) maximumNode() *Node {
-	if node == nil {
-		return nil
-	}
-	for node.Right != nil {
-		node = node.Right
-	}
-	return node
-}
-
-func (tree *Tree) deleteCase1(node *Node) {
-	if node.Parent == nil {
-		return
-	}
-	tree.deleteCase2(node)
-}
-
-func (tree *Tree) deleteCase2(node *Node) {
-	sibling := node.sibling()
-	if nodeColor(sibling) == red {
-		node.Parent.color = red
-		sibling.color = black
-		if node == node.Parent.Left {
-			tree.rotateLeft(node.Parent)
-		} else {
-			tree.rotateRight(node.Parent)
-		}
-	}
-	tree.deleteCase3(node)
-}
-
-func (tree *Tree) deleteCase3(node *Node) {
-	sibling := node.sibling()
-	if nodeColor(node.Parent) == black &&
-		nodeColor(sibling) == black &&
-		nodeColor(sibling.Left) == black &&
-		nodeColor(sibling.Right) == black {
-		sibling.color = red
-		tree.deleteCase1(node.Parent)
-	} else {
-		tree.deleteCase4(node)
-	}
-}
-
-func (tree *Tree) deleteCase4(node *Node) {
-	sibling := node.sibling()
-	if nodeColor(node.Parent) == red &&
-		nodeColor(sibling) == black &&
-		nodeColor(sibling.Left) == black &&
-		nodeColor(sibling.Right) == black {
-		sibling.color = red
-		node.Parent.color = black
-	} else {
-		tree.deleteCase5(node)
-	}
-}
-
-func (tree *Tree) deleteCase5(node *Node) {
-	sibling := node.sibling()
-	if node == node.Parent.Left &&
-		nodeColor(sibling) == black &&
-		nodeColor(sibling.Left) == red &&
-		nodeColor(sibling.Right) == black {
-		sibling.color = red
-		sibling.Left.color = black
-		tree.rotateRight(sibling)
-	} else if node == node.Parent.Right &&
-		nodeColor(sibling) == black &&
-		nodeColor(sibling.Right) == red &&
-		nodeColor(sibling.Left) == black {
-		sibling.color = red
-		sibling.Right.color = black
-		tree.rotateLeft(sibling)
-	}
-	tree.deleteCase6(node)
-}
-
-func (tree *Tree) deleteCase6(node *Node) {
-	sibling := node.sibling()
-	sibling.color = nodeColor(node.Parent)
-	node.Parent.color = black
-	if node == node.Parent.Left && nodeColor(sibling.Right) == red {
-		sibling.Right.color = black
-		tree.rotateLeft(node.Parent)
-	} else if nodeColor(sibling.Left) == red {
-		sibling.Left.color = black
-		tree.rotateRight(node.Parent)
-	}
-}
-
-func nodeColor(node *Node) color {
-	if node == nil {
-		return black
-	}
-	return node.color
-}

errors.go

@@ -1,17 +1,8 @@
 package main
 
-import (
+import "errors"
-	"errors"
-	"fmt"
-)
 
 var ErrRateLimit = errors.New("too many requests")
+var ErrForbidden = errors.New("access denied")
 var ErrKnown     = errors.New("already crawled")
 
-type HttpError struct {
-	code int
-}
-
-func (e HttpError) Error() string {
-	return fmt.Sprintf("http status %d", e.code)
-}

fasturl/url.go

@@ -1,869 +0,0 @@
-// Copyright 2009 The Go Authors. All rights reserved.
-// Use of this source code is governed by a BSD-style
-// license that can be found in the LICENSE file.
-
-// Package fasturl parses URLs and implements query escaping.
-package fasturl
-
-// Modifications by terorie
-
-// See RFC 3986. This package generally follows RFC 3986, except where
-// it deviates for compatibility reasons. When sending changes, first
-// search old issues for history on decisions. Unit tests should also
-// contain references to issue numbers with details.
-
-import (
-	"errors"
-	"fmt"
-	"strconv"
-	"strings"
-)
-
-type Scheme int
-const (
-	SchemeInvalid = iota
-	SchemeHTTP
-	SchemeHTTPS
-	SchemeCount
-)
-
-var Schemes = [SchemeCount]string {
-	"",
-	"http",
-	"https",
-}
-
-var ErrUnknownScheme = errors.New("unknown protocol scheme")
-
-// Error reports an error and the operation and URL that caused it.
-type Error struct {
-	Op  string
-	URL string
-	Err error
-}
-
-func (e *Error) Error() string { return e.Op + " " + e.URL + ": " + e.Err.Error() }
-
-type timeout interface {
-	Timeout() bool
-}
-
-func (e *Error) Timeout() bool {
-	t, ok := e.Err.(timeout)
-	return ok && t.Timeout()
-}
-
-type temporary interface {
-	Temporary() bool
-}
-
-func (e *Error) Temporary() bool {
-	t, ok := e.Err.(temporary)
-	return ok && t.Temporary()
-}
-
-func ishex(c byte) bool {
-	switch {
-	case '0' <= c && c <= '9':
-		return true
-	case 'a' <= c && c <= 'f':
-		return true
-	case 'A' <= c && c <= 'F':
-		return true
-	}
-	return false
-}
-
-func unhex(c byte) byte {
-	switch {
-	case '0' <= c && c <= '9':
-		return c - '0'
-	case 'a' <= c && c <= 'f':
-		return c - 'a' + 10
-	case 'A' <= c && c <= 'F':
-		return c - 'A' + 10
-	}
-	return 0
-}
-
-type encoding int
-
-const (
-	encodePath encoding = 1 + iota
-	encodePathSegment
-	encodeHost
-	encodeZone
-	encodeUserPassword
-	encodeQueryComponent
-	encodeFragment
-)
-
-type EscapeError string
-
-func (e EscapeError) Error() string {
-	return "invalid URL escape " + strconv.Quote(string(e))
-}
-
-type InvalidHostError string
-
-func (e InvalidHostError) Error() string {
-	return "invalid character " + strconv.Quote(string(e)) + " in host name"
-}
-
-// Return true if the specified character should be escaped when
-// appearing in a URL string, according to RFC 3986.
-//
-// Please be informed that for now shouldEscape does not check all
-// reserved characters correctly. See golang.org/issue/5684.
-func shouldEscape(c byte, mode encoding) bool {
-	// §2.3 Unreserved characters (alphanum)
-	if 'A' <= c && c <= 'Z' || 'a' <= c && c <= 'z' || '0' <= c && c <= '9' {
-		return false
-	}
-
-	if mode == encodeHost || mode == encodeZone {
-		// §3.2.2 Host allows
-		//	sub-delims = "!" / "$" / "&" / "'" / "(" / ")" / "*" / "+" / "," / ";" / "="
-		// as part of reg-name.
-		// We add : because we include :port as part of host.
-		// We add [ ] because we include [ipv6]:port as part of host.
-		// We add < > because they're the only characters left that
-		// we could possibly allow, and Parse will reject them if we
-		// escape them (because hosts can't use %-encoding for
-		// ASCII bytes).
-		switch c {
-		case '!', '$', '&', '\'', '(', ')', '*', '+', ',', ';', '=', ':', '[', ']', '<', '>', '"':
-			return false
-		}
-	}
-
-	switch c {
-	case '-', '_', '.', '~': // §2.3 Unreserved characters (mark)
-		return false
-
-	case '$', '&', '+', ',', '/', ':', ';', '=', '?', '@': // §2.2 Reserved characters (reserved)
-		// Different sections of the URL allow a few of
-		// the reserved characters to appear unescaped.
-		switch mode {
-		case encodePath: // §3.3
-			// The RFC allows : @ & = + $ but saves / ; , for assigning
-			// meaning to individual path segments. This package
-			// only manipulates the path as a whole, so we allow those
-			// last three as well. That leaves only ? to escape.
-			return c == '?'
-
-		case encodePathSegment: // §3.3
-			// The RFC allows : @ & = + $ but saves / ; , for assigning
-			// meaning to individual path segments.
-			return c == '/' || c == ';' || c == ',' || c == '?'
-
-		case encodeUserPassword: // §3.2.1
-			// The RFC allows ';', ':', '&', '=', '+', '$', and ',' in
-			// userinfo, so we must escape only '@', '/', and '?'.
-			// The parsing of userinfo treats ':' as special so we must escape
-			// that too.
-			return c == '@' || c == '/' || c == '?' || c == ':'
-
-		case encodeQueryComponent: // §3.4
-			// The RFC reserves (so we must escape) everything.
-			return true
-
-		case encodeFragment: // §4.1
-			// The RFC text is silent but the grammar allows
-			// everything, so escape nothing.
-			return false
-		}
-	}
-
-	if mode == encodeFragment {
-		// RFC 3986 §2.2 allows not escaping sub-delims. A subset of sub-delims are
-		// included in reserved from RFC 2396 §2.2. The remaining sub-delims do not
-		// need to be escaped. To minimize potential breakage, we apply two restrictions:
-		// (1) we always escape sub-delims outside of the fragment, and (2) we always
-		// escape single quote to avoid breaking callers that had previously assumed that
-		// single quotes would be escaped. See issue #19917.
-		switch c {
-		case '!', '(', ')', '*':
-			return false
-		}
-	}
-
-	// Everything else must be escaped.
-	return true
-}
-
-// unescape unescapes a string; the mode specifies
-// which section of the URL string is being unescaped.
-func unescape(s string, mode encoding) (string, error) {
-	// Count %, check that they're well-formed.
-	n := 0
-	hasPlus := false
-	for i := 0; i < len(s); {
-		switch s[i] {
-		case '%':
-			n++
-			if i+2 >= len(s) || !ishex(s[i+1]) || !ishex(s[i+2]) {
-				s = s[i:]
-				if len(s) > 3 {
-					s = s[:3]
-				}
-				return "", EscapeError(s)
-			}
-			// Per https://tools.ietf.org/html/rfc3986#page-21
-			// in the host component %-encoding can only be used
-			// for non-ASCII bytes.
-			// But https://tools.ietf.org/html/rfc6874#section-2
-			// introduces %25 being allowed to escape a percent sign
-			// in IPv6 scoped-address literals. Yay.
-			if mode == encodeHost && unhex(s[i+1]) < 8 && s[i:i+3] != "%25" {
-				return "", EscapeError(s[i : i+3])
-			}
-			if mode == encodeZone {
-				// RFC 6874 says basically "anything goes" for zone identifiers
-				// and that even non-ASCII can be redundantly escaped,
-				// but it seems prudent to restrict %-escaped bytes here to those
-				// that are valid host name bytes in their unescaped form.
-				// That is, you can use escaping in the zone identifier but not
-				// to introduce bytes you couldn't just write directly.
-				// But Windows puts spaces here! Yay.
-				v := unhex(s[i+1])<<4 | unhex(s[i+2])
-				if s[i:i+3] != "%25" && v != ' ' && shouldEscape(v, encodeHost) {
-					return "", EscapeError(s[i : i+3])
-				}
-			}
-			i += 3
-		case '+':
-			hasPlus = mode == encodeQueryComponent
-			i++
-		default:
-			if (mode == encodeHost || mode == encodeZone) && s[i] < 0x80 && shouldEscape(s[i], mode) {
-				return "", InvalidHostError(s[i : i+1])
-			}
-			i++
-		}
-	}
-
-	if n == 0 && !hasPlus {
-		return s, nil
-	}
-
-	t := make([]byte, len(s)-2*n)
-	j := 0
-	for i := 0; i < len(s); {
-		switch s[i] {
-		case '%':
-			t[j] = unhex(s[i+1])<<4 | unhex(s[i+2])
-			j++
-			i += 3
-		case '+':
-			if mode == encodeQueryComponent {
-				t[j] = ' '
-			} else {
-				t[j] = '+'
-			}
-			j++
-			i++
-		default:
-			t[j] = s[i]
-			j++
-			i++
-		}
-	}
-	return string(t), nil
-}
-
-func escape(s string, mode encoding) string {
-	spaceCount, hexCount := 0, 0
-	for i := 0; i < len(s); i++ {
-		c := s[i]
-		if shouldEscape(c, mode) {
-			if c == ' ' && mode == encodeQueryComponent {
-				spaceCount++
-			} else {
-				hexCount++
-			}
-		}
-	}
-
-	if spaceCount == 0 && hexCount == 0 {
-		return s
-	}
-
-	t := make([]byte, len(s)+2*hexCount)
-	j := 0
-	for i := 0; i < len(s); i++ {
-		switch c := s[i]; {
-		case c == ' ' && mode == encodeQueryComponent:
-			t[j] = '+'
-			j++
-		case shouldEscape(c, mode):
-			t[j] = '%'
-			t[j+1] = "0123456789ABCDEF"[c>>4]
-			t[j+2] = "0123456789ABCDEF"[c&15]
-			j += 3
-		default:
-			t[j] = s[i]
-			j++
-		}
-	}
-	return string(t)
-}
-
-// A URL represents a parsed URL (technically, a URI reference).
-//
-// The general form represented is:
-//
-//	[scheme:][//[userinfo@]host][/]path[?query][#fragment]
-//
-// URLs that do not start with a slash after the scheme are interpreted as:
-//
-//	scheme:opaque[?query][#fragment]
-//
-// Note that the Path field is stored in decoded form: /%47%6f%2f becomes /Go/.
-// A consequence is that it is impossible to tell which slashes in the Path were
-// slashes in the raw URL and which were %2f. This distinction is rarely important,
-// but when it is, code must not use Path directly.
-// The Parse function sets both Path and RawPath in the URL it returns,
-// and URL's String method uses RawPath if it is a valid encoding of Path,
-// by calling the EscapedPath method.
-type URL struct {
-	Scheme     Scheme
-	Host       string    // host or host:port
-	Path       string    // path (relative paths may omit leading slash)
-}
-
-// Maybe rawurl is of the form scheme:path.
-// (Scheme must be [a-zA-Z][a-zA-Z0-9+-.]*)
-// If so, return scheme, path; else return "", rawurl.
-func getscheme(rawurl string) (scheme Scheme, path string, err error) {
-	for i := 0; i < len(rawurl); i++ {
-		c := rawurl[i]
-		switch {
-		case 'a' <= c && c <= 'z' || 'A' <= c && c <= 'Z':
-			// do nothing
-		case '0' <= c && c <= '9' || c == '+' || c == '-' || c == '.':
-			if i == 0 {
-				return SchemeInvalid, rawurl, nil
-			}
-		case c == ':':
-			if i == 0 {
-				return SchemeInvalid, "", errors.New("missing protocol scheme")
-			}
-			switch rawurl[:i] {
-			case "http":
-				scheme = SchemeHTTP
-			case "https":
-				scheme = SchemeHTTPS
-			default:
-				return SchemeInvalid, "", ErrUnknownScheme
-			}
-
-			path = rawurl[i+1:]
-			return
-		default:
-			// we have encountered an invalid character,
-			// so there is no valid scheme
-			return SchemeInvalid, rawurl, nil
-		}
-	}
-	return SchemeInvalid, rawurl, nil
-}
-
-// Maybe s is of the form t c u.
-// If so, return t, c u (or t, u if cutc == true).
-// If not, return s, "".
-func split(s string, c string, cutc bool) (string, string) {
-	i := strings.Index(s, c)
-	if i < 0 {
-		return s, ""
-	}
-	if cutc {
-		return s[:i], s[i+len(c):]
-	}
-	return s[:i], s[i:]
-}
-
-// Parse parses rawurl into a URL structure.
-//
-// The rawurl may be relative (a path, without a host) or absolute
-// (starting with a scheme). Trying to parse a hostname and path
-// without a scheme is invalid but may not necessarily return an
-// error, due to parsing ambiguities.
-func (u *URL) Parse(rawurl string) error {
-	// Cut off #frag
-	s, frag := split(rawurl, "#", true)
-	err := u.parse(s, false)
-	if err != nil {
-		return &Error{"parse", s, err}
-	}
-	if frag == "" {
-		return nil
-	}
-	return nil
-}
-
-// ParseRequestURI parses rawurl into a URL structure. It assumes that
-// rawurl was received in an HTTP request, so the rawurl is interpreted
-// only as an absolute URI or an absolute path.
-// The string rawurl is assumed not to have a #fragment suffix.
-// (Web browsers strip #fragment before sending the URL to a web server.)
-func (u *URL) ParseRequestURI(rawurl string) error {
-	err := u.parse(rawurl, true)
-	if err != nil {
-		return &Error{"parse", rawurl, err}
-	}
-	return nil
-}
-
-// parse parses a URL from a string in one of two contexts. If
-// viaRequest is true, the URL is assumed to have arrived via an HTTP request,
-// in which case only absolute URLs or path-absolute relative URLs are allowed.
-// If viaRequest is false, all forms of relative URLs are allowed.
-func (u *URL) parse(rawurl string, viaRequest bool) error {
-	var rest string
-	var err error
-
-	if rawurl == "" && viaRequest {
-		return errors.New("empty url")
-	}
-
-	if rawurl == "*" {
-		u.Path = "*"
-		return nil
-	}
-
-	// Split off possible leading "http:", "mailto:", etc.
-	// Cannot contain escaped characters.
-	if u.Scheme, rest, err = getscheme(rawurl); err != nil {
-		return err
-	}
-
-	if strings.HasSuffix(rest, "?") && strings.Count(rest, "?") == 1 {
-		rest = rest[:len(rest)-1]
-	} else {
-		rest, _ = split(rest, "?", true)
-	}
-
-	if !strings.HasPrefix(rest, "/") {
-		if u.Scheme != SchemeInvalid {
-			// We consider rootless paths per RFC 3986 as opaque.
-			return nil
-		}
-		if viaRequest {
-			return errors.New("invalid URI for request")
-		}
-
-		// Avoid confusion with malformed schemes, like cache_object:foo/bar.
-		// See golang.org/issue/16822.
-		//
-		// RFC 3986, §3.3:
-		// In addition, a URI reference (Section 4.1) may be a relative-path reference,
-		// in which case the first path segment cannot contain a colon (":") character.
-		colon := strings.Index(rest, ":")
-		slash := strings.Index(rest, "/")
-		if colon >= 0 && (slash < 0 || colon < slash) {
-			// First path segment has colon. Not allowed in relative URL.
-			return errors.New("first path segment in URL cannot contain colon")
-		}
-	}
-
-	if (u.Scheme != SchemeInvalid || !viaRequest && !strings.HasPrefix(rest, "///")) && strings.HasPrefix(rest, "//") {
-		var authority string
-		authority, rest = split(rest[2:], "/", false)
-		u.Host, err = parseAuthority(authority)
-		if err != nil {
-			return err
-		}
-	}
-	u.Path = rest
-	return nil
-}
-
-func parseAuthority(authority string) (host string, err error) {
-	i := strings.LastIndex(authority, "@")
-	if i < 0 {
-		host, err = parseHost(authority)
-	} else {
-		host, err = parseHost(authority[i+1:])
-	}
-	if err != nil {
-		return "", err
-	}
-	if i < 0 {
-		return host, nil
-	}
-	userinfo := authority[:i]
-	if !validUserinfo(userinfo) {
-		return "", errors.New("fasturl: invalid userinfo")
-	}
-	return host, nil
-}
-
-// parseHost parses host as an authority without user
-// information. That is, as host[:port].
-func parseHost(host string) (string, error) {
-	if strings.HasPrefix(host, "[") {
-		// Parse an IP-Literal in RFC 3986 and RFC 6874.
-		// E.g., "[fe80::1]", "[fe80::1%25en0]", "[fe80::1]:80".
-		i := strings.LastIndex(host, "]")
-		if i < 0 {
-			return "", errors.New("missing ']' in host")
-		}
-		colonPort := host[i+1:]
-		if !validOptionalPort(colonPort) {
-			return "", fmt.Errorf("invalid port %q after host", colonPort)
-		}
-
-		// RFC 6874 defines that %25 (%-encoded percent) introduces
-		// the zone identifier, and the zone identifier can use basically
-		// any %-encoding it likes. That's different from the host, which
-		// can only %-encode non-ASCII bytes.
-		// We do impose some restrictions on the zone, to avoid stupidity
-		// like newlines.
-		zone := strings.Index(host[:i], "%25")
-		if zone >= 0 {
-			host1, err := unescape(host[:zone], encodeHost)
-			if err != nil {
-				return "", err
-			}
-			host2, err := unescape(host[zone:i], encodeZone)
-			if err != nil {
-				return "", err
-			}
-			host3, err := unescape(host[i:], encodeHost)
-			if err != nil {
-				return "", err
-			}
-			return host1 + host2 + host3, nil
-		}
-	}
-
-	var err error
-	if host, err = unescape(host, encodeHost); err != nil {
-		return "", err
-	}
-	return host, nil
-}
-
-// validOptionalPort reports whether port is either an empty string
-// or matches /^:\d*$/
-func validOptionalPort(port string) bool {
-	if port == "" {
-		return true
-	}
-	if port[0] != ':' {
-		return false
-	}
-	for _, b := range port[1:] {
-		if b < '0' || b > '9' {
-			return false
-		}
-	}
-	return true
-}
-
-// String reassembles the URL into a valid URL string.
-// The general form of the result is one of:
-//
-//	scheme:opaque?query#fragment
-//	scheme://userinfo@host/path?query#fragment
-//
-// If u.Opaque is non-empty, String uses the first form;
-// otherwise it uses the second form.
-// To obtain the path, String uses u.EscapedPath().
-//
-// In the second form, the following rules apply:
-//	- if u.Scheme is empty, scheme: is omitted.
-//	- if u.User is nil, userinfo@ is omitted.
-//	- if u.Host is empty, host/ is omitted.
-//	- if u.Scheme and u.Host are empty and u.User is nil,
-//	   the entire scheme://userinfo@host/ is omitted.
-//	- if u.Host is non-empty and u.Path begins with a /,
-//	   the form host/path does not add its own /.
-//	- if u.RawQuery is empty, ?query is omitted.
-//	- if u.Fragment is empty, #fragment is omitted.
-func (u *URL) String() string {
-	var buf strings.Builder
-	if u.Scheme != SchemeInvalid {
-		buf.WriteString(Schemes[u.Scheme])
-		buf.WriteByte(':')
-	}
-	if u.Scheme != SchemeInvalid || u.Host != "" {
-		if u.Host != "" || u.Path != "" {
-			buf.WriteString("//")
-		}
-		if h := u.Host; h != "" {
-			buf.WriteString(escape(h, encodeHost))
-		}
-	}
-	path := u.Path
-	if path != "" && path[0] != '/' && u.Host != "" {
-		buf.WriteByte('/')
-	}
-	if buf.Len() == 0 {
-		// RFC 3986 §4.2
-		// A path segment that contains a colon character (e.g., "this:that")
-		// cannot be used as the first segment of a relative-path reference, as
-		// it would be mistaken for a scheme name. Such a segment must be
-		// preceded by a dot-segment (e.g., "./this:that") to make a relative-
-		// path reference.
-		if i := strings.IndexByte(path, ':'); i > -1 && strings.IndexByte(path[:i], '/') == -1 {
-			buf.WriteString("./")
-		}
-	}
-	buf.WriteString(path)
-	return buf.String()
-}
-
-func isRunesDot(r []rune) bool {
-	return len(r) == 1 && r[0] == '.'
-}
-
-func isRunesDoubleDot(r []rune) bool {
-	return len(r) == 2 && r[0] == '.' && r[1] == '.'
-}
-
-// resolvePath applies special path segments from refs and applies
-// them to base, per RFC 3986.
-func resolvePath(base, ref string) string {
-	var full string
-	if ref == "" {
-		full = base
-	} else if ref[0] != '/' {
-		i := strings.LastIndex(base, "/")
-		full = base[:i+1] + ref
-	} else {
-		full = ref
-	}
-	if full == "" {
-		return ""
-	} else if full == "/" {
-		return "/"
-	}
-
-	dst := make([]rune, len(full))
-	dst = dst[0:0]
-
-	start := 0
-	rs := []rune(full)
-	if len(rs) != 0 && rs[0] == '/' {
-		rs = rs[1:]
-	}
-	var stack []int
-	stack = append(stack, 0)
-	for i, c := range rs {
-		if i == len(rs) - 1 {
-			closingSlash := false
-			part := rs[start:]
-			if len(part) == 0 {
-				dst = append(dst, '/')
-			} else if part[len(part)-1] == '/' {
-				part = part[:len(part)-1]
-				closingSlash = true
-			}
-			switch {
-			case isRunesDot(part):
-				dst = append(dst, '/')
-			case isRunesDoubleDot(part):
-				// Cut to the last slash
-				start = i+1
-				dst = dst[:stack[len(stack)-1]]
-				if len(stack) != 1 {
-					stack = stack[:len(stack)-1]
-				}
-				dst = append(dst, '/')
-			default:
-				dst = append(dst, '/')
-				dst = append(dst, part...)
-			}
-			if closingSlash && len(dst) != 0 && dst[len(dst)-1] != '/' {
-				dst = append(dst, '/')
-			}
-		} else if c == '/' {
-			part := rs[start:i]
-			switch {
-			case isRunesDot(part):
-				start = i+1
-			case isRunesDoubleDot(part):
-				// Cut to the last slash
-				start = i+1
-				dst = dst[:stack[len(stack)-1]]
-				if len(stack) != 1 {
-					stack = stack[:len(stack)-1]
-				}
-			default:
-				start = i+1
-				stack = append(stack, len(dst))
-				dst = append(dst, '/')
-				dst = append(dst, part...)
-			}
-		}
-	}
-	return string(dst)
-
-	/*var dst []string
-	src := strings.Split(full, "/")
-	for _, elem := range src {
-		switch elem {
-		case ".":
-			// drop
-		case "..":
-			if len(dst) > 0 {
-				dst = dst[:len(dst)-1]
-			}
-		default:
-			dst = append(dst, elem)
-		}
-	}
-	if last := src[len(src)-1]; last == "." || last == ".." {
-		// Add final slash to the joined path.
-		dst = append(dst, "")
-	}
-	return "/" + strings.TrimPrefix(strings.Join(dst, "/"), "/")*/
-}
-
-// IsAbs reports whether the URL is absolute.
-// Absolute means that it has a non-empty scheme.
-func (u *URL) IsAbs() bool {
-	return u.Scheme != SchemeInvalid
-}
-
-// ParseRel parses a URL in the context of the receiver. The provided URL
-// may be relative or absolute. Parse returns nil, err on parse
-// failure, otherwise its return value is the same as ResolveReference.
-func (u *URL) ParseRel(out *URL, ref string) error {
-	var refurl URL
-
-	err := refurl.Parse(ref)
-	if err != nil {
-		return err
-	}
-
-	u.ResolveReference(out, &refurl)
-	return nil
-}
-
-// ResolveReference resolves a URI reference to an absolute URI from
-// an absolute base URI u, per RFC 3986 Section 5.2. The URI reference
-// may be relative or absolute. ResolveReference always returns a new
-// URL instance, even if the returned URL is identical to either the
-// base or reference. If ref is an absolute URL, then ResolveReference
-// ignores base and returns a copy of ref.
-func (u *URL) ResolveReference(url *URL, ref *URL) {
-	*url = *ref
-	if ref.Scheme == SchemeInvalid {
-		url.Scheme = u.Scheme
-	}
-	if ref.Scheme != SchemeInvalid || ref.Host != "" {
-		// The "absoluteURI" or "net_path" cases.
-		// We can ignore the error from setPath since we know we provided a
-		// validly-escaped path.
-		url.Path = resolvePath(ref.Path, "")
-		return
-	}
-	// The "abs_path" or "rel_path" cases.
-	url.Host = u.Host
-	url.Path = resolvePath(u.Path, ref.Path)
-	return
-}
-
-// Marshaling interface implementations.
-// Would like to implement MarshalText/UnmarshalText but that will change the JSON representation of URLs.
-
-func (u *URL) MarshalBinary() (text []byte, err error) {
-	return []byte(u.String()), nil
-}
-
-func (u *URL) UnmarshalBinary(text []byte) error {
-	var u1 URL
-	err := u1.Parse(string(text))
-	if err != nil {
-		return err
-	}
-	*u = u1
-	return nil
-}
-
-// validUserinfo reports whether s is a valid userinfo string per RFC 3986
-// Section 3.2.1:
-//     userinfo    = *( unreserved / pct-encoded / sub-delims / ":" )
-//     unreserved  = ALPHA / DIGIT / "-" / "." / "_" / "~"
-//     sub-delims  = "!" / "$" / "&" / "'" / "(" / ")"
-//                   / "*" / "+" / "," / ";" / "="
-//
-// It doesn't validate pct-encoded. The caller does that via func unescape.
-func validUserinfo(s string) bool {
-	for _, r := range s {
-		if 'A' <= r && r <= 'Z' {
-			continue
-		}
-		if 'a' <= r && r <= 'z' {
-			continue
-		}
-		if '0' <= r && r <= '9' {
-			continue
-		}
-		switch r {
-		case '-', '.', '_', ':', '~', '!', '$', '&', '\'',
-			'(', ')', '*', '+', ',', ';', '=', '%', '@':
-			continue
-		default:
-			return false
-		}
-	}
-	return true
-}
-
-func PathUnescape(s string) string {
-	newStr, err := pathUnescape(s)
-	if err != nil {
-		return s
-	} else {
-		return newStr
-	}
-}
-
-func pathUnescape(s string) (string, error) {
-	// Count %, check that they're well-formed.
-	n := 0
-	for i := 0; i < len(s); {
-		switch s[i] {
-		case '%':
-			n++
-			if i+2 >= len(s) || !ishex(s[i+1]) || !ishex(s[i+2]) {
-				s = s[i:]
-				if len(s) > 3 {
-					s = s[:3]
-				}
-				return "", EscapeError(s)
-			}
-			i += 3
-		default:
-			i++
-		}
-	}
-
-	if n == 0 {
-		return s, nil
-	}
-
-	t := make([]byte, len(s)-2*n)
-	j := 0
-	for i := 0; i < len(s); {
-		switch s[i] {
-		case '%':
-			t[j] = unhex(s[i+1])<<4 | unhex(s[i+2])
-			j++
-			i += 3
-		case '+':
-			t[j] = '+'
-			j++
-			i++
-		default:
-			t[j] = s[i]
-			j++
-			i++
-		}
-	}
-	return string(t), nil
-}

fasturl/url_test.go

@@ -1,897 +0,0 @@
-// Copyright 2009 The Go Authors. All rights reserved.
-// Use of this source code is governed by a BSD-style
-// license that can be found in the LICENSE file.
-
-package fasturl
-
-import (
-	"bytes"
-	encodingPkg "encoding"
-	"encoding/gob"
-	"encoding/json"
-	"fmt"
-	"io"
-	"net"
-	"reflect"
-	"testing"
-)
-
-type URLTest struct {
-	in        string
-	out       *URL   // expected parse; RawPath="" means same as Path
-	roundtrip string // expected result of reserializing the URL; empty means same as "in".
-}
-
-var urltests = []URLTest{
-	// no path
-	{
-		"http://www.google.com",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "www.google.com",
-		},
-		"",
-	},
-	// path
-	{
-		"http://www.google.com/",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "www.google.com",
-			Path:   "/",
-		},
-		"",
-	},
-	// %20 outside query
-	{
-		"http://www.google.com/a%20b",
-		&URL{
-			Scheme:   SchemeHTTP,
-			Host:     "www.google.com",
-			Path:     "/a%20b",
-		},
-		"",
-	},
-	// leading // without scheme should create an authority
-	{
-		"//foo",
-		&URL{
-			Host: "foo",
-		},
-		"",
-	},
-	// Three leading slashes isn't an authority, but doesn't return an error.
-	// (We can't return an error, as this code is also used via
-	// ServeHTTP -> ReadRequest -> Parse, which is arguably a
-	// different URL parsing context, but currently shares the
-	// same codepath)
-	{
-		"///threeslashes",
-		&URL{
-			Path: "///threeslashes",
-		},
-		"",
-	},
-	// unescaped @ in username should not confuse host
-	{
-		"http://j@ne:password@google.com",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "google.com",
-		},
-		"http://google.com",
-	},
-	// unescaped @ in password should not confuse host
-	{
-		"http://jane:p@ssword@google.com",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "google.com",
-		},
-		"http://google.com",
-	},
-	// Relative path
-	{
-		"a/b/c",
-		&URL{
-			Path: "a/b/c",
-		},
-		"a/b/c",
-	},
-	// host subcomponent; IPv4 address in RFC 3986
-	{
-		"http://192.168.0.1/",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "192.168.0.1",
-			Path:   "/",
-		},
-		"",
-	},
-	// host and port subcomponents; IPv4 address in RFC 3986
-	{
-		"http://192.168.0.1:8080/",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "192.168.0.1:8080",
-			Path:   "/",
-		},
-		"",
-	},
-	// host subcomponent; IPv6 address in RFC 3986
-	{
-		"http://[fe80::1]/",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "[fe80::1]",
-			Path:   "/",
-		},
-		"",
-	},
-	// host and port subcomponents; IPv6 address in RFC 3986
-	{
-		"http://[fe80::1]:8080/",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "[fe80::1]:8080",
-			Path:   "/",
-		},
-		"",
-	},
-	// host subcomponent; IPv6 address with zone identifier in RFC 6874
-	{
-		"http://[fe80::1%25en0]/", // alphanum zone identifier
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "[fe80::1%en0]",
-			Path:   "/",
-		},
-		"",
-	},
-	// host and port subcomponents; IPv6 address with zone identifier in RFC 6874
-	{
-		"http://[fe80::1%25en0]:8080/", // alphanum zone identifier
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "[fe80::1%en0]:8080",
-			Path:   "/",
-		},
-		"",
-	},
-	// host subcomponent; IPv6 address with zone identifier in RFC 6874
-	{
-		"http://[fe80::1%25%65%6e%301-._~]/", // percent-encoded+unreserved zone identifier
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "[fe80::1%en01-._~]",
-			Path:   "/",
-		},
-		"http://[fe80::1%25en01-._~]/",
-	},
-	// host and port subcomponents; IPv6 address with zone identifier in RFC 6874
-	{
-		"http://[fe80::1%25%65%6e%301-._~]:8080/", // percent-encoded+unreserved zone identifier
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "[fe80::1%en01-._~]:8080",
-			Path:   "/",
-		},
-		"http://[fe80::1%25en01-._~]:8080/",
-	},
-	// golang.org/issue/12200 (colon with empty port)
-	{
-		"http://192.168.0.2:8080/foo",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "192.168.0.2:8080",
-			Path:   "/foo",
-		},
-		"",
-	},
-	{
-		"http://192.168.0.2:/foo",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "192.168.0.2:",
-			Path:   "/foo",
-		},
-		"",
-	},
-	{
-		// Malformed IPv6 but still accepted.
-		"http://2b01:e34:ef40:7730:8e70:5aff:fefe:edac:8080/foo",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "2b01:e34:ef40:7730:8e70:5aff:fefe:edac:8080",
-			Path:   "/foo",
-		},
-		"",
-	},
-	{
-		// Malformed IPv6 but still accepted.
-		"http://2b01:e34:ef40:7730:8e70:5aff:fefe:edac:/foo",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "2b01:e34:ef40:7730:8e70:5aff:fefe:edac:",
-			Path:   "/foo",
-		},
-		"",
-	},
-	{
-		"http://[2b01:e34:ef40:7730:8e70:5aff:fefe:edac]:8080/foo",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "[2b01:e34:ef40:7730:8e70:5aff:fefe:edac]:8080",
-			Path:   "/foo",
-		},
-		"",
-	},
-	{
-		"http://[2b01:e34:ef40:7730:8e70:5aff:fefe:edac]:/foo",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "[2b01:e34:ef40:7730:8e70:5aff:fefe:edac]:",
-			Path:   "/foo",
-		},
-		"",
-	},
-	// golang.org/issue/7991 and golang.org/issue/12719 (non-ascii %-encoded in host)
-	{
-		"http://hello.世界.com/foo",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "hello.世界.com",
-			Path:   "/foo",
-		},
-		"http://hello.%E4%B8%96%E7%95%8C.com/foo",
-	},
-	{
-		"http://hello.%e4%b8%96%e7%95%8c.com/foo",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "hello.世界.com",
-			Path:   "/foo",
-		},
-		"http://hello.%E4%B8%96%E7%95%8C.com/foo",
-	},
-	{
-		"http://hello.%E4%B8%96%E7%95%8C.com/foo",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "hello.世界.com",
-			Path:   "/foo",
-		},
-		"",
-	},
-	// golang.org/issue/10433 (path beginning with //)
-	{
-		"http://example.com//foo",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "example.com",
-			Path:   "//foo",
-		},
-		"",
-	},
-	// test that we can reparse the host names we accept.
-	{
-		"http://authority<\"hi\">/foo",
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "authority<\"hi\">",
-			Path:   "/foo",
-		},
-		"",
-	},
-}
-
-// more useful string for debugging than fmt's struct printer
-func ufmt(u *URL) string {
-	return fmt.Sprintf("scheme=%q, host=%q, path=%q",
-		Schemes[u.Scheme], u.Host, u.Path)
-}
-
-func BenchmarkString(b *testing.B) {
-	b.StopTimer()
-	b.ReportAllocs()
-	for _, tt := range urltests {
-		var u URL
-		err := u.Parse(tt.in)
-		if err != nil {
-			b.Errorf("Parse(%q) returned error %s", tt.in, err)
-			continue
-		}
-		if tt.roundtrip == "" {
-			continue
-		}
-		b.StartTimer()
-		var g string
-		for i := 0; i < b.N; i++ {
-			g = u.String()
-		}
-		b.StopTimer()
-		if w := tt.roundtrip; b.N > 0 && g != w {
-			b.Errorf("Parse(%q).String() == %q, want %q", tt.in, g, w)
-		}
-	}
-}
-
-func TestParse(t *testing.T) {
-	for _, tt := range urltests {
-		var u URL
-		err := u.Parse(tt.in)
-		if err != nil {
-			t.Errorf("Parse(%q) returned error %v", tt.in, err)
-			continue
-		}
-		if !reflect.DeepEqual(&u, tt.out) {
-			t.Errorf("Parse(%q):\n\tgot  %v\n\twant %v\n", tt.in, ufmt(&u), ufmt(tt.out))
-		}
-	}
-}
-
-const pathThatLooksSchemeRelative = "//not.a.user@not.a.host/just/a/path"
-
-var parseRequestURLTests = []struct {
-	url           string
-	expectedValid bool
-}{
-	{"http://foo.com", true},
-	{"http://foo.com/", true},
-	{"http://foo.com/path", true},
-	{"/", true},
-	{pathThatLooksSchemeRelative, true},
-	{"//not.a.user@%66%6f%6f.com/just/a/path/also", true},
-	{"*", true},
-	{"http://192.168.0.1/", true},
-	{"http://192.168.0.1:8080/", true},
-	{"http://[fe80::1]/", true},
-	{"http://[fe80::1]:8080/", true},
-
-	// Tests exercising RFC 6874 compliance:
-	{"http://[fe80::1%25en0]/", true},                 // with alphanum zone identifier
-	{"http://[fe80::1%25en0]:8080/", true},            // with alphanum zone identifier
-	{"http://[fe80::1%25%65%6e%301-._~]/", true},      // with percent-encoded+unreserved zone identifier
-	{"http://[fe80::1%25%65%6e%301-._~]:8080/", true}, // with percent-encoded+unreserved zone identifier
-
-	{"foo.html", false},
-	{"../dir/", false},
-	{"http://192.168.0.%31/", false},
-	{"http://192.168.0.%31:8080/", false},
-	{"http://[fe80::%31]/", false},
-	{"http://[fe80::%31]:8080/", false},
-	{"http://[fe80::%31%25en0]/", false},
-	{"http://[fe80::%31%25en0]:8080/", false},
-
-	// These two cases are valid as textual representations as
-	// described in RFC 4007, but are not valid as address
-	// literals with IPv6 zone identifiers in URIs as described in
-	// RFC 6874.
-	{"http://[fe80::1%en0]/", false},
-	{"http://[fe80::1%en0]:8080/", false},
-}
-
-func TestParseRequestURI(t *testing.T) {
-	for _, test := range parseRequestURLTests {
-		var u URL
-		err := u.ParseRequestURI(test.url)
-		if test.expectedValid && err != nil {
-			t.Errorf("ParseRequestURI(%q) gave err %v; want no error", test.url, err)
-		} else if !test.expectedValid && err == nil {
-			t.Errorf("ParseRequestURI(%q) gave nil error; want some error", test.url)
-		}
-	}
-
-	var url URL
-	err := url.ParseRequestURI(pathThatLooksSchemeRelative)
-	if err != nil {
-		t.Fatalf("Unexpected error %v", err)
-	}
-	if url.Path != pathThatLooksSchemeRelative {
-		t.Errorf("ParseRequestURI path:\ngot  %q\nwant %q", url.Path, pathThatLooksSchemeRelative)
-	}
-}
-
-var stringURLTests = []struct {
-	url  URL
-	want string
-}{
-	// No leading slash on path should prepend slash on String() call
-	{
-		url: URL{
-			Scheme: SchemeHTTP,
-			Host:   "www.google.com",
-			Path:   "search",
-		},
-		want: "http://www.google.com/search",
-	},
-	// Relative path with first element containing ":" should be prepended with "./", golang.org/issue/17184
-	{
-		url: URL{
-			Path: "this:that",
-		},
-		want: "./this:that",
-	},
-	// Relative path with second element containing ":" should not be prepended with "./"
-	{
-		url: URL{
-			Path: "here/this:that",
-		},
-		want: "here/this:that",
-	},
-	// Non-relative path with first element containing ":" should not be prepended with "./"
-	{
-		url: URL{
-			Scheme: SchemeHTTP,
-			Host:   "www.google.com",
-			Path:   "this:that",
-		},
-		want: "http://www.google.com/this:that",
-	},
-}
-
-func TestURLString(t *testing.T) {
-	for _, tt := range urltests {
-		var u URL
-		err := u.Parse(tt.in)
-		if err != nil {
-			t.Errorf("Parse(%q) returned error %s", tt.in, err)
-			continue
-		}
-		expected := tt.in
-		if tt.roundtrip != "" {
-			expected = tt.roundtrip
-		}
-		s := u.String()
-		if s != expected {
-			t.Errorf("Parse(%q).String() == %q (expected %q)", tt.in, s, expected)
-		}
-	}
-
-	for _, tt := range stringURLTests {
-		if got := tt.url.String(); got != tt.want {
-			t.Errorf("%+v.String() = %q; want %q", tt.url, got, tt.want)
-		}
-	}
-}
-
-var resolvePathTests = []struct {
-	base, ref, expected string
-}{
-	{"a/b", ".", "/a/"},
-	{"a/b", "c", "/a/c"},
-	{"a/b", "..", "/"},
-	{"a/", "..", "/"},
-	{"a/", "../..", "/"},
-	{"a/b/c", "..", "/a/"},
-	{"a/b/c", "../d", "/a/d"},
-	{"a/b/c", ".././d", "/a/d"},
-	{"a/b", "./..", "/"},
-	{"a/./b", ".", "/a/"},
-	{"a/../", ".", "/"},
-	{"a/.././b", "c", "/c"},
-}
-
-func TestResolvePath(t *testing.T) {
-	for _, test := range resolvePathTests {
-		got := resolvePath(test.base, test.ref)
-		if got != test.expected {
-			t.Errorf("For %q + %q got %q; expected %q", test.base, test.ref, got, test.expected)
-		}
-	}
-}
-
-var resolveReferenceTests = []struct {
-	base, rel, expected string
-}{
-	// Absolute URL references
-	{"http://foo.com?a=b", "https://bar.com/", "https://bar.com/"},
-	{"http://foo.com/", "https://bar.com/?a=b", "https://bar.com/"},
-	{"http://foo.com/", "https://bar.com/?", "https://bar.com/"},
-
-	// Path-absolute references
-	{"http://foo.com/bar", "/baz", "http://foo.com/baz"},
-	{"http://foo.com/bar?a=b#f", "/baz", "http://foo.com/baz"},
-	{"http://foo.com/bar?a=b", "/baz?", "http://foo.com/baz"},
-	{"http://foo.com/bar?a=b", "/baz?c=d", "http://foo.com/baz"},
-
-	// Multiple slashes
-	{"http://foo.com/bar", "http://foo.com//baz", "http://foo.com//baz"},
-	{"http://foo.com/bar", "http://foo.com///baz/quux", "http://foo.com///baz/quux"},
-
-	// Scheme-relative
-	{"https://foo.com/bar?a=b", "//bar.com/quux", "https://bar.com/quux"},
-
-	// Path-relative references:
-
-	// ... current directory
-	{"http://foo.com", ".", "http://foo.com/"},
-	{"http://foo.com/bar", ".", "http://foo.com/"},
-	{"http://foo.com/bar/", ".", "http://foo.com/bar/"},
-
-	// ... going down
-	{"http://foo.com", "bar", "http://foo.com/bar"},
-	{"http://foo.com/", "bar", "http://foo.com/bar"},
-	{"http://foo.com/bar/baz", "quux", "http://foo.com/bar/quux"},
-
-	// ... going up
-	{"http://foo.com/bar/baz", "../quux", "http://foo.com/quux"},
-	{"http://foo.com/bar/baz", "../../../../../quux", "http://foo.com/quux"},
-	{"http://foo.com/bar", "..", "http://foo.com/"},
-	{"http://foo.com/bar/baz", "./..", "http://foo.com/"},
-	// ".." in the middle (issue 3560)
-	{"http://foo.com/bar/baz", "quux/dotdot/../tail", "http://foo.com/bar/quux/tail"},
-	{"http://foo.com/bar/baz", "quux/./dotdot/../tail", "http://foo.com/bar/quux/tail"},
-	{"http://foo.com/bar/baz", "quux/./dotdot/.././tail", "http://foo.com/bar/quux/tail"},
-	{"http://foo.com/bar/baz", "quux/./dotdot/./../tail", "http://foo.com/bar/quux/tail"},
-	{"http://foo.com/bar/baz", "quux/./dotdot/dotdot/././../../tail", "http://foo.com/bar/quux/tail"},
-	{"http://foo.com/bar/baz", "quux/./dotdot/dotdot/./.././../tail", "http://foo.com/bar/quux/tail"},
-	{"http://foo.com/bar/baz", "quux/./dotdot/dotdot/dotdot/./../../.././././tail", "http://foo.com/bar/quux/tail"},
-	{"http://foo.com/bar/baz", "quux/./dotdot/../dotdot/../dot/./tail/..", "http://foo.com/bar/quux/dot/"},
-
-	// Remove any dot-segments prior to forming the target URI.
-	// http://tools.ietf.org/html/rfc3986#section-5.2.4
-	{"http://foo.com/dot/./dotdot/../foo/bar", "../baz", "http://foo.com/dot/baz"},
-
-	// Triple dot isn't special
-	{"http://foo.com/bar", "...", "http://foo.com/..."},
-
-	// Fragment
-	{"http://foo.com/bar", ".#frag", "http://foo.com/"},
-	{"http://example.org/", "#!$&%27()*+,;=", "http://example.org/"},
-
-	// Paths with escaping (issue 16947).
-	{"http://foo.com/foo%2fbar/", "../baz", "http://foo.com/baz"},
-	{"http://foo.com/1/2%2f/3%2f4/5", "../../a/b/c", "http://foo.com/1/a/b/c"},
-	{"http://foo.com/1/2/3", "./a%2f../../b/..%2fc", "http://foo.com/1/2/b/..%2fc"},
-	{"http://foo.com/1/2%2f/3%2f4/5", "./a%2f../b/../c", "http://foo.com/1/2%2f/3%2f4/a%2f../c"},
-	{"http://foo.com/foo%20bar/", "../baz", "http://foo.com/baz"},
-	{"http://foo.com/foo", "../bar%2fbaz", "http://foo.com/bar%2fbaz"},
-	{"http://foo.com/foo%2dbar/", "./baz-quux", "http://foo.com/foo%2dbar/baz-quux"},
-
-	// RFC 3986: Normal Examples
-	// http://tools.ietf.org/html/rfc3986#section-5.4.1
-	{"http://a/b/c/d;p?q", "g", "http://a/b/c/g"},
-	{"http://a/b/c/d;p?q", "./g", "http://a/b/c/g"},
-	{"http://a/b/c/d;p?q", "g/", "http://a/b/c/g/"},
-	{"http://a/b/c/d;p?q", "/g", "http://a/g"},
-	{"http://a/b/c/d;p?q", "//g", "http://g"},
-	{"http://a/b/c/d;p?q", "?y", "http://a/b/c/d;p"},
-	{"http://a/b/c/d;p?q", "g?y", "http://a/b/c/g"},
-	{"http://a/b/c/d;p?q", "#s", "http://a/b/c/d;p"},
-	{"http://a/b/c/d;p?q", "g#s", "http://a/b/c/g"},
-	{"http://a/b/c/d;p?q", "g?y#s", "http://a/b/c/g"},
-	{"http://a/b/c/d;p?q", ";x", "http://a/b/c/;x"},
-	{"http://a/b/c/d;p?q", "g;x", "http://a/b/c/g;x"},
-	{"http://a/b/c/d;p?q", "g;x?y#s", "http://a/b/c/g;x"},
-	{"http://a/b/c/d;p?q", "", "http://a/b/c/d;p"},
-	{"http://a/b/c/d;p?q", ".", "http://a/b/c/"},
-	{"http://a/b/c/d;p?q", "./", "http://a/b/c/"},
-	{"http://a/b/c/d;p?q", "..", "http://a/b/"},
-	{"http://a/b/c/d;p?q", "../", "http://a/b/"},
-	{"http://a/b/c/d;p?q", "../g", "http://a/b/g"},
-	{"http://a/b/c/d;p?q", "../..", "http://a/"},
-	{"http://a/b/c/d;p?q", "../../", "http://a/"},
-	{"http://a/b/c/d;p?q", "../../g", "http://a/g"},
-
-	// RFC 3986: Abnormal Examples
-	// http://tools.ietf.org/html/rfc3986#section-5.4.2
-	{"http://a/b/c/d;p?q", "../../../g", "http://a/g"},
-	{"http://a/b/c/d;p?q", "../../../../g", "http://a/g"},
-	{"http://a/b/c/d;p?q", "/./g", "http://a/g"},
-	{"http://a/b/c/d;p?q", "/../g", "http://a/g"},
-	{"http://a/b/c/d;p?q", "g.", "http://a/b/c/g."},
-	{"http://a/b/c/d;p?q", ".g", "http://a/b/c/.g"},
-	{"http://a/b/c/d;p?q", "g..", "http://a/b/c/g.."},
-	{"http://a/b/c/d;p?q", "..g", "http://a/b/c/..g"},
-	{"http://a/b/c/d;p?q", "./../g", "http://a/b/g"},
-	{"http://a/b/c/d;p?q", "./g/.", "http://a/b/c/g/"},
-	{"http://a/b/c/d;p?q", "g/./h", "http://a/b/c/g/h"},
-	{"http://a/b/c/d;p?q", "g/../h", "http://a/b/c/h"},
-	{"http://a/b/c/d;p?q", "g;x=1/./y", "http://a/b/c/g;x=1/y"},
-	{"http://a/b/c/d;p?q", "g;x=1/../y", "http://a/b/c/y"},
-	{"http://a/b/c/d;p?q", "g?y/./x", "http://a/b/c/g"},
-	{"http://a/b/c/d;p?q", "g?y/../x", "http://a/b/c/g"},
-	{"http://a/b/c/d;p?q", "g#s/./x", "http://a/b/c/g"},
-	{"http://a/b/c/d;p?q", "g#s/../x", "http://a/b/c/g"},
-
-	// Extras.
-	{"https://a/b/c/d;p?q", "//g?q", "https://g"},
-	{"https://a/b/c/d;p?q", "//g#s", "https://g"},
-	{"https://a/b/c/d;p?q", "//g/d/e/f?y#s", "https://g/d/e/f"},
-	{"https://a/b/c/d;p#s", "?y", "https://a/b/c/d;p"},
-	{"https://a/b/c/d;p?q#s", "?y", "https://a/b/c/d;p"},
-}
-
-func TestResolveReference(t *testing.T) {
-	mustParse := func(url string) *URL {
-		u := new(URL)
-		err := u.Parse(url)
-		if err != nil {
-			t.Fatalf("Parse(%q) got err %v", url, err)
-		}
-		return u
-	}
-	for _, test := range resolveReferenceTests {
-		base := mustParse(test.base)
-		rel := mustParse(test.rel)
-		var url URL
-		base.ResolveReference(&url, rel)
-		if got := url.String(); got != test.expected {
-			t.Errorf("URL(%q).ResolveReference(%q)\ngot  %q\nwant %q", test.base, test.rel, got, test.expected)
-		}
-	}
-}
-
-type RequestURITest struct {
-	url *URL
-	out string
-}
-
-var requritests = []RequestURITest{
-	{
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "example.com",
-			Path:   "",
-		},
-		"/",
-	},
-	{
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "example.com",
-			Path:   "/a b",
-		},
-		"/a%20b",
-	},
-	{
-		&URL{
-			Scheme: SchemeHTTP,
-			Host:   "example.com",
-			Path:   "//foo",
-		},
-		"//foo",
-	},
-}
-
-func TestParseErrors(t *testing.T) {
-	tests := []struct {
-		in      string
-		wantErr bool
-	}{
-		{"http://[::1]", false},
-		{"http://[::1]:80", false},
-		{"http://[::1]:namedport", true}, // rfc3986 3.2.3
-		{"http://[::1]/", false},
-		{"http://[::1]a", true},
-		{"http://[::1]%23", true},
-		{"http://[::1%25en0]", false},     // valid zone id
-		{"http://[::1]:", false},          // colon, but no port OK
-		{"http://[::1]:%38%30", true},     // not allowed: % encoding only for non-ASCII
-		{"http://[::1%25%41]", false},     // RFC 6874 allows over-escaping in zone
-		{"http://[%10::1]", true},         // no %xx escapes in IP address
-		{"http://[::1]/%48", false},       // %xx in path is fine
-		{"http://%41:8080/", true},        // not allowed: % encoding only for non-ASCII
-
-		{"http://[]%20%48%54%54%50%2f%31%2e%31%0a%4d%79%48%65%61%64%65%72%3a%20%31%32%33%0a%0a/", true}, // golang.org/issue/11208
-		{"http://a b.com/", true},    // no space in host name please
-	}
-	for _, tt := range tests {
-		var u URL
-		err := u.Parse(tt.in)
-		if tt.wantErr {
-			if err == nil {
-				t.Errorf("Parse(%q) = %#v; want an error", tt.in, u)
-			}
-			continue
-		}
-		if err != nil {
-			t.Logf("Parse(%q) = %v; want no error", tt.in, err)
-		}
-	}
-}
-
-type shouldEscapeTest struct {
-	in     byte
-	mode   encoding
-	escape bool
-}
-
-var shouldEscapeTests = []shouldEscapeTest{
-	// Unreserved characters (§2.3)
-	{'a', encodePath, false},
-	{'a', encodeUserPassword, false},
-	{'a', encodeQueryComponent, false},
-	{'a', encodeFragment, false},
-	{'a', encodeHost, false},
-	{'z', encodePath, false},
-	{'A', encodePath, false},
-	{'Z', encodePath, false},
-	{'0', encodePath, false},
-	{'9', encodePath, false},
-	{'-', encodePath, false},
-	{'-', encodeUserPassword, false},
-	{'-', encodeQueryComponent, false},
-	{'-', encodeFragment, false},
-	{'.', encodePath, false},
-	{'_', encodePath, false},
-	{'~', encodePath, false},
-
-	// User information (§3.2.1)
-	{':', encodeUserPassword, true},
-	{'/', encodeUserPassword, true},
-	{'?', encodeUserPassword, true},
-	{'@', encodeUserPassword, true},
-	{'$', encodeUserPassword, false},
-	{'&', encodeUserPassword, false},
-	{'+', encodeUserPassword, false},
-	{',', encodeUserPassword, false},
-	{';', encodeUserPassword, false},
-	{'=', encodeUserPassword, false},
-
-	// Host (IP address, IPv6 address, registered name, port suffix; §3.2.2)
-	{'!', encodeHost, false},
-	{'$', encodeHost, false},
-	{'&', encodeHost, false},
-	{'\'', encodeHost, false},
-	{'(', encodeHost, false},
-	{')', encodeHost, false},
-	{'*', encodeHost, false},
-	{'+', encodeHost, false},
-	{',', encodeHost, false},
-	{';', encodeHost, false},
-	{'=', encodeHost, false},
-	{':', encodeHost, false},
-	{'[', encodeHost, false},
-	{']', encodeHost, false},
-	{'0', encodeHost, false},
-	{'9', encodeHost, false},
-	{'A', encodeHost, false},
-	{'z', encodeHost, false},
-	{'_', encodeHost, false},
-	{'-', encodeHost, false},
-	{'.', encodeHost, false},
-}
-
-func TestShouldEscape(t *testing.T) {
-	for _, tt := range shouldEscapeTests {
-		if shouldEscape(tt.in, tt.mode) != tt.escape {
-			t.Errorf("shouldEscape(%q, %v) returned %v; expected %v", tt.in, tt.mode, !tt.escape, tt.escape)
-		}
-	}
-}
-
-type timeoutError struct {
-	timeout bool
-}
-
-func (e *timeoutError) Error() string { return "timeout error" }
-func (e *timeoutError) Timeout() bool { return e.timeout }
-
-type temporaryError struct {
-	temporary bool
-}
-
-func (e *temporaryError) Error() string   { return "temporary error" }
-func (e *temporaryError) Temporary() bool { return e.temporary }
-
-type timeoutTemporaryError struct {
-	timeoutError
-	temporaryError
-}
-
-func (e *timeoutTemporaryError) Error() string { return "timeout/temporary error" }
-
-var netErrorTests = []struct {
-	err       error
-	timeout   bool
-	temporary bool
-}{{
-	err:       &Error{"Get", "http://google.com/", &timeoutError{timeout: true}},
-	timeout:   true,
-	temporary: false,
-}, {
-	err:       &Error{"Get", "http://google.com/", &timeoutError{timeout: false}},
-	timeout:   false,
-	temporary: false,
-}, {
-	err:       &Error{"Get", "http://google.com/", &temporaryError{temporary: true}},
-	timeout:   false,
-	temporary: true,
-}, {
-	err:       &Error{"Get", "http://google.com/", &temporaryError{temporary: false}},
-	timeout:   false,
-	temporary: false,
-}, {
-	err:       &Error{"Get", "http://google.com/", &timeoutTemporaryError{timeoutError{timeout: true}, temporaryError{temporary: true}}},
-	timeout:   true,
-	temporary: true,
-}, {
-	err:       &Error{"Get", "http://google.com/", &timeoutTemporaryError{timeoutError{timeout: false}, temporaryError{temporary: true}}},
-	timeout:   false,
-	temporary: true,
-}, {
-	err:       &Error{"Get", "http://google.com/", &timeoutTemporaryError{timeoutError{timeout: true}, temporaryError{temporary: false}}},
-	timeout:   true,
-	temporary: false,
-}, {
-	err:       &Error{"Get", "http://google.com/", &timeoutTemporaryError{timeoutError{timeout: false}, temporaryError{temporary: false}}},
-	timeout:   false,
-	temporary: false,
-}, {
-	err:       &Error{"Get", "http://google.com/", io.EOF},
-	timeout:   false,
-	temporary: false,
-}}
-
-// Test that url.Error implements net.Error and that it forwards
-func TestURLErrorImplementsNetError(t *testing.T) {
-	for i, tt := range netErrorTests {
-		err, ok := tt.err.(net.Error)
-		if !ok {
-			t.Errorf("%d: %T does not implement net.Error", i+1, tt.err)
-			continue
-		}
-		if err.Timeout() != tt.timeout {
-			t.Errorf("%d: err.Timeout(): got %v, want %v", i+1, err.Timeout(), tt.timeout)
-			continue
-		}
-		if err.Temporary() != tt.temporary {
-			t.Errorf("%d: err.Temporary(): got %v, want %v", i+1, err.Temporary(), tt.temporary)
-		}
-	}
-}
-
-var _ encodingPkg.BinaryMarshaler = (*URL)(nil)
-var _ encodingPkg.BinaryUnmarshaler = (*URL)(nil)
-
-func TestJSON(t *testing.T) {
-	var u URL
-	err := u.Parse("https://www.google.com/x?y=z")
-	if err != nil {
-		t.Fatal(err)
-	}
-	js, err := json.Marshal(&u)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	// If only we could implement TextMarshaler/TextUnmarshaler,
-	// this would work:
-	//
-	// if string(js) != strconv.Quote(u.String()) {
-	// 	t.Errorf("json encoding: %s\nwant: %s\n", js, strconv.Quote(u.String()))
-	// }
-
-	u1 := new(URL)
-	err = json.Unmarshal(js, u1)
-	if err != nil {
-		t.Fatal(err)
-	}
-	if u1.String() != u.String() {
-		t.Errorf("json decoded to: %s\nwant: %s\n", u1, &u)
-	}
-}
-
-func TestGob(t *testing.T) {
-	var u URL
-	err := u.Parse("https://www.google.com/x?y=z")
-	if err != nil {
-		t.Fatal(err)
-	}
-	var w bytes.Buffer
-	err = gob.NewEncoder(&w).Encode(&u)
-	if err != nil {
-		t.Fatal(err)
-	}
-
-	u1 := new(URL)
-	err = gob.NewDecoder(&w).Decode(u1)
-	if err != nil {
-		t.Fatal(err)
-	}
-	if u1.String() != u.String() {
-		t.Errorf("json decoded to: %s\nwant: %s\n", u1, &u)
-	}
-}

go.mod

@@ -1,14 +0,0 @@
-module github.com/syndtr/od-database-crawler
-
-require (
-	github.com/beeker1121/goque v2.0.1+incompatible
-	github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db // indirect
-	github.com/sirupsen/logrus v1.3.0
-	github.com/spf13/cobra v0.0.3
-	github.com/spf13/viper v1.3.1
-	github.com/syndtr/goleveldb v0.0.0-20181128100959-b001fa50d6b2 // indirect
-	github.com/terorie/od-database-crawler v1.1.1
-	github.com/valyala/fasthttp v1.1.0
-	golang.org/x/crypto v0.0.0-20190131182504-b8fe1690c613
-	golang.org/x/net v0.0.0-20180911220305-26e67e76b6c3
-)

go.sum

@@ -1,66 +0,0 @@
-github.com/armon/consul-api v0.0.0-20180202201655-eb2c6b5be1b6/go.mod h1:grANhF5doyWs3UAsr3K4I6qtAmlQcZDesFNEHPZAzj8=
-github.com/beeker1121/goque v2.0.1+incompatible h1:5nJHPMqQLxUvGFc8m/NW2QzxKyc0zICmqs/JUsmEjwE=
-github.com/beeker1121/goque v2.0.1+incompatible/go.mod h1:L6dOWBhDOnxUVQsb0wkLve0VCnt2xJW/MI8pdRX4ANw=
-github.com/coreos/etcd v3.3.10+incompatible/go.mod h1:uF7uidLiAD3TWHmW31ZFd/JWoc32PjwdhPthX9715RE=
-github.com/coreos/go-etcd v2.0.0+incompatible/go.mod h1:Jez6KQU2B/sWsbdaef3ED8NzMklzPG4d5KIOhIy30Tk=
-github.com/coreos/go-semver v0.2.0/go.mod h1:nnelYz7RCh+5ahJtPPxZlU+153eP4D4r3EedlOD2RNk=
-github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
-github.com/fsnotify/fsnotify v1.4.7 h1:IXs+QLmnXW2CcXuY+8Mzv/fWEsPGWxqefPtCP5CnV9I=
-github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
-github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db h1:woRePGFeVFfLKN/pOkfl+p/TAqKOfFu+7KPlMVpok/w=
-github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
-github.com/hashicorp/hcl v1.0.0 h1:0Anlzjpi4vEasTeNFn2mLJgTSwt0+6sfsiTG8qcWGx4=
-github.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ=
-github.com/klauspost/compress v1.4.0 h1:8nsMz3tWa9SWWPL60G1V6CUsf4lLjWLTNEtibhe8gh8=
-github.com/klauspost/compress v1.4.0/go.mod h1:RyIbtBH6LamlWaDj8nUwkbUhJ87Yi3uG0guNDohfE1A=
-github.com/klauspost/cpuid v0.0.0-20180405133222-e7e905edc00e h1:+lIPJOWl+jSiJOc70QXJ07+2eg2Jy2EC7Mi11BWujeM=
-github.com/klauspost/cpuid v0.0.0-20180405133222-e7e905edc00e/go.mod h1:Pj4uuM528wm8OyEC2QMXAi2YiTZ96dNQPGgoMS4s3ek=
-github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
-github.com/magiconair/properties v1.8.0 h1:LLgXmsheXeRoUOBOjtwPQCWIYqM/LU1ayDtDePerRcY=
-github.com/magiconair/properties v1.8.0/go.mod h1:PppfXfuXeibc/6YijjN8zIbojt8czPbwD3XqdrwzmxQ=
-github.com/mitchellh/mapstructure v1.1.2 h1:fmNYVwqnSfB9mZU6OS2O6GsXM+wcskZDuKQzvN1EDeE=
-github.com/mitchellh/mapstructure v1.1.2/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y=
-github.com/pelletier/go-toml v1.2.0 h1:T5zMGML61Wp+FlcbWjRDT7yAxhJNAiPPLOFECq181zc=
-github.com/pelletier/go-toml v1.2.0/go.mod h1:5z9KED0ma1S8pY6P1sdut58dfprrGBbd/94hg7ilaic=
-github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
-github.com/sirupsen/logrus v1.3.0 h1:hI/7Q+DtNZ2kINb6qt/lS+IyXnHQe9e90POfeewL/ME=
-github.com/sirupsen/logrus v1.3.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo=
-github.com/spf13/afero v1.1.2 h1:m8/z1t7/fwjysjQRYbP0RD+bUIF/8tJwPdEZsI83ACI=
-github.com/spf13/afero v1.1.2/go.mod h1:j4pytiNVoe2o6bmDsKpLACNPDBIoEAkihy7loJ1B0CQ=
-github.com/spf13/cast v1.3.0 h1:oget//CVOEoFewqQxwr0Ej5yjygnqGkvggSE/gB35Q8=
-github.com/spf13/cast v1.3.0/go.mod h1:Qx5cxh0v+4UWYiBimWS+eyWzqEqokIECu5etghLkUJE=
-github.com/spf13/cobra v0.0.3 h1:ZlrZ4XsMRm04Fr5pSFxBgfND2EBVa1nLpiy1stUsX/8=
-github.com/spf13/cobra v0.0.3/go.mod h1:1l0Ry5zgKvJasoi3XT1TypsSe7PqH0Sj9dhYf7v3XqQ=
-github.com/spf13/jwalterweatherman v1.0.0 h1:XHEdyB+EcvlqZamSM4ZOMGlc93t6AcsBEu9Gc1vn7yk=
-github.com/spf13/jwalterweatherman v1.0.0/go.mod h1:cQK4TGJAtQXfYWX+Ddv3mKDzgVb68N+wFjFa4jdeBTo=
-github.com/spf13/pflag v1.0.3 h1:zPAT6CGy6wXeQ7NtTnaTerfKOsV6V6F8agHXFiazDkg=
-github.com/spf13/pflag v1.0.3/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4=
-github.com/spf13/viper v1.3.1 h1:5+8j8FTpnFV4nEImW/ofkzEt8VoOiLXxdYIDsB73T38=
-github.com/spf13/viper v1.3.1/go.mod h1:ZiWeW+zYFKm7srdB9IoDzzZXaJaI5eL9QjNiN/DMA2s=
-github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
-github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
-github.com/syndtr/goleveldb v0.0.0-20181128100959-b001fa50d6b2 h1:GnOzE5fEFN3b2zDhJJABEofdb51uMRNb8eqIVtdducs=
-github.com/syndtr/goleveldb v0.0.0-20181128100959-b001fa50d6b2/go.mod h1:Z4AUp2Km+PwemOoO/VB5AOx9XSsIItzFjoJlOSiYmn0=
-github.com/terorie/od-database-crawler v1.1.1 h1:Ca+ZqbZX3rVWBR8SDRzvroyxjBtUs75MQXZ9YG0gqGo=
-github.com/terorie/od-database-crawler v1.1.1/go.mod h1:vVJ7pLkudrlUNp9qu24JCzQ8N6mFsrOmX1tPXr155DQ=
-github.com/ugorji/go/codec v0.0.0-20181204163529-d75b2dcb6bc8/go.mod h1:VFNgLljTbGfSG7qAOspJ7OScBnGdDN/yBr0sguwnwf0=
-github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
-github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc=
-github.com/valyala/fasthttp v1.1.0 h1:3BohG7mqwj4lq7PTX//7gLbUlzNvZSPmuHFnloXT0lw=
-github.com/valyala/fasthttp v1.1.0/go.mod h1:4vX61m6KN+xDduDNwXrhIAVZaZaZiQ1luJk8LWSxF3s=
-github.com/valyala/tcplisten v0.0.0-20161114210144-ceec8f93295a/go.mod h1:v3UYOV9WzVtRmSR+PDvWpU/qWl4Wa5LApYYX4ZtKbio=
-github.com/xordataexchange/crypt v0.0.3-0.20170626215501-b2862e3d0a77/go.mod h1:aYKd//L2LvnjZzWKhF00oedf4jCCReLcmhLdhm1A27Q=
-golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
-golang.org/x/crypto v0.0.0-20181203042331-505ab145d0a9/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
-golang.org/x/crypto v0.0.0-20190131182504-b8fe1690c613 h1:MQ/ZZiDsUapFFiMS+vzwXkCTeEKaum+Do5rINYJDmxc=
-golang.org/x/crypto v0.0.0-20190131182504-b8fe1690c613/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
-golang.org/x/net v0.0.0-20180911220305-26e67e76b6c3 h1:czFLhve3vsQetD6JOJ8NZZvGQIXlnN3/yXxbT6/awxI=
-golang.org/x/net v0.0.0-20180911220305-26e67e76b6c3/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
-golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
-golang.org/x/sys v0.0.0-20181205085412-a5c9d58dba9a h1:1n5lsVfiQW3yfsRGu98756EH1YthsFqr/5mxHduZW2A=
-golang.org/x/sys v0.0.0-20181205085412-a5c9d58dba9a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
-golang.org/x/text v0.3.0 h1:g61tztE5qeGQ89tm6NTjjM9VPIm088od1l6aSorWRWg=
-golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
-gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
-gopkg.in/yaml.v2 v2.2.2 h1:ZCJp+EgiOT7lHqUV2J862kp8Qj64Jo6az82+3Td9dZw=
-gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=

help.go

@@ -1,15 +0,0 @@
-package main
-
-const helpText =
-`HTTP crawler for the OD-Database
-  DB      >> https://od-db.the-eye.eu <<
-  Crawler >> https://github.com/terorie/od-database-crawler <<
-  Server  >> https://github.com/simon987/od-database <<
-
-Quick start:
- - get config file  (config.yml in working dir)
- - get OD-DB server ("server.url": Database URL + /api)
- - get access token ("server.token": e.g. c010b6dd-20...)
- - ./od-database-crawler server
-
-Questions? Discord @terorie#2664 / Telegram @terorie`

main.go

@@ -2,162 +2,86 @@ package main
 
 import (
 	"context"
-	"fmt"
 	"github.com/sirupsen/logrus"
-	"github.com/spf13/cobra"
+	"github.com/urfave/cli"
-	"github.com/spf13/viper"
+	"log"
-	"github.com/terorie/od-database-crawler/fasturl"
+	"net/http"
+	_ "net/http/pprof"
+	"net/url"
 	"os"
 	"strings"
-	"sync/atomic"
 	"time"
 )
 
-var configFile string
+var app = cli.App {
-
+	Name:         "oddb-go",
-var rootCmd = cobra.Command {
+	Usage:        "OD-Database Go crawler",
-	Use: "od-database-crawler",
+	Version:      "0.1",
-	Version: "1.2.0",
+	BashComplete: cli.DefaultAppComplete,
-	Short: "OD-Database Go crawler",
+	Writer:       os.Stdout,
-	Long: helpText,
+	Compiled:     buildDate,
-	PersistentPreRunE: preRun,
+	Commands:     []cli.Command{
-	PersistentPostRun: func(cmd *cobra.Command, args []string) {
+		{
-		exitHooks.Execute()
+			Name:      "crawl",
+			Usage:     "Crawl a list of URLs",
+			ArgsUsage: "[site, site, ...]",
+			Action:    cmdCrawler,
+		},
 	},
 }
 
-var serverCmd = cobra.Command {
-	Use: "server",
-	Short: "Start crawl server",
-	Long: "Connect to the OD-Database and contribute to the database\n" +
-		"by crawling the web for open directories!",
-	Run: cmdBase,
-}
-
-var crawlCmd = cobra.Command {
-	Use: "crawl",
-	Short: "Crawl an URL",
-	Long: "Crawl the URL specified.\n" +
-		"Results will not be uploaded to the database,\n" +
-		"they're saved under crawled/0.json instead.\n" +
-		"Primarily used for testing and benchmarking.",
-	RunE: cmdCrawler,
-	Args: cobra.ExactArgs(1),
-}
-
-var exitHooks Hooks
-
 func init() {
-	rootCmd.AddCommand(&crawlCmd)
-	rootCmd.AddCommand(&serverCmd)
-
 	prepareConfig()
 }
 
-func preRun(cmd *cobra.Command, args []string) error {
+func main() {
-	if err := os.MkdirAll("crawled", 0755);
+	go func() {
-		err != nil { panic(err) }
+		log.Println(http.ListenAndServe("localhost:42069", nil))
-
+	}()
-	if err := os.MkdirAll("queue", 0755);
+	app.Run(os.Args)
-		err != nil { panic(err) }
+}
 
+func cmdCrawler(clic *cli.Context) error {
 	readConfig()
 
-	return nil
+	if clic.NArg() == 0 {
+		cli.ShowCommandHelpAndExit(clic, "crawl", 1)
 	}
 
-func main() {
+	args := clic.Args()
-	err := rootCmd.Execute()
+	remotes := make([]*OD, len(args))
-	if err != nil {
+	for i, arg := range args {
-		fmt.Fprintln(os.Stderr, err)
-		os.Exit(1)
-	}
-}
-
-func cmdBase(_ *cobra.Command, _ []string) {
-	// TODO Graceful shutdown
-	appCtx := context.Background()
-	forceCtx := context.Background()
-
-	inRemotes := make(chan *OD)
-	go Schedule(forceCtx, inRemotes)
-
-	ticker := time.NewTicker(config.Recheck)
-	defer ticker.Stop()
-	for {
-		select {
-		case <-appCtx.Done():
-			return
-		case <-ticker.C:
-			t, err := FetchTask()
-			if err != nil {
-				logrus.WithError(err).
-					Error("Failed to get new task")
-				time.Sleep(viper.GetDuration(ConfCooldown))
-				continue
-			}
-			if t == nil {
-				// No new task
-				if atomic.LoadInt32(&numActiveTasks) == 0 {
-					logrus.Info("Waiting …")
-				}
-				continue
-			}
-
-			var baseUri fasturl.URL
-			err = baseUri.Parse(t.Url)
-			if urlErr, ok := err.(*fasturl.Error); ok && urlErr.Err == fasturl.ErrUnknownScheme {
-				// Not an error
-				err = nil
-
-				// Give back task
-				//err2 := CancelTask(t.WebsiteId)
-				//if err2 != nil {
-				//	logrus.Error(err2)
-				//}
-
-				continue
-			} else if err != nil {
-				logrus.WithError(err).
-					Error("Failed to get new task")
-				time.Sleep(viper.GetDuration(ConfCooldown))
-				continue
-			}
-			ScheduleTask(inRemotes, t, &baseUri)
-		}
-	}
-}
-
-func cmdCrawler(_ *cobra.Command, args []string) error {
-	arg := args[0]
 		// https://github.com/golang/go/issues/19779
 		if !strings.Contains(arg, "://") {
 			arg = "http://" + arg
 		}
-	var u fasturl.URL
+		u, err := url.Parse(arg)
-	err := u.Parse(arg)
 		if !strings.HasSuffix(u.Path, "/") {
 			u.Path += "/"
 		}
 		if err != nil { return err }
+		remotes[i] = &OD{ BaseUri: *u }
+	}
 
-	// TODO Graceful shutdown
+	c := context.Background()
-	forceCtx := context.Background()
 
 	inRemotes := make(chan *OD)
-	go Schedule(forceCtx, inRemotes)
+	go Schedule(c, inRemotes)
 
-	ticker := time.NewTicker(3 * time.Second)
+	for _, remote := range remotes {
-	defer ticker.Stop()
+		globalWait.Add(1)
-
+		inRemotes <- remote
-	task := Task {
-		WebsiteId: 0,
-		Url: u.String(),
 	}
-	ScheduleTask(inRemotes, &task, &u)
 
 	// Wait for all jobs to finish
 	globalWait.Wait()
 
+	logrus.Info("All dirs processed!")
+
 	return nil
 }
+
+var buildDate = time.Date(
+	2018, 10, 28,
+	17, 10, 0, 0,
+	time.UTC)

model.go

@@ -1,63 +1,32 @@
 package main
 
 import (
-	"github.com/terorie/od-database-crawler/ds/redblackhash"
+	"net/url"
-	"github.com/terorie/od-database-crawler/fasturl"
 	"sync"
 	"time"
 )
 
-type Task struct {
-	WebsiteId uint64 `json:"website_id"`
-	Url       string `json:"url"`
-}
-
-type TaskResult struct {
-	StatusCode    string    `json:"status_code"`
-	FileCount     uint64    `json:"file_count"`
-	ErrorCount    uint64    `json:"-"`
-	StartTime     time.Time `json:"-"`
-	StartTimeUnix int64     `json:"start_time"`
-	EndTimeUnix   int64     `json:"end_time"`
-	WebsiteId     uint64    `json:"website_id"`
-}
-
 type Job struct {
-	Uri       fasturl.URL
+	OD        *OD
+	Uri       url.URL
 	UriStr    string
 	Fails     int
 	LastError error
 }
 
 type OD struct {
-	Task    Task
-	Result  TaskResult
 	Wait    sync.WaitGroup
-	BaseUri fasturl.URL
+	BaseUri url.URL
+	lock    sync.Mutex
+	Files   []File
 	WCtx    WorkerContext
-	Scanned redblackhash.Tree
+	Scanned sync.Map
 }
 
 type File struct {
 	Name  string    `json:"name"`
 	Size  int64     `json:"size"`
-	MTime int64  `json:"mtime"`
+	MTime time.Time `json:"mtime"`
 	Path  string    `json:"path"`
 	IsDir bool      `json:"-"`
 }
-
-func (o *OD) LoadOrStoreKey(k *redblackhash.Key) (exists bool) {
-	o.Scanned.Lock()
-	defer o.Scanned.Unlock()
-
-	exists = o.Scanned.Get(k)
-	if exists { return true }
-
-	o.Scanned.Put(k)
-	return false
-}
-
-type errorString string
-func (e errorString) Error() string {
-	return string(e)
-}

queue.go

@@ -1,129 +0,0 @@
-package main
-
-import (
-	"github.com/beeker1121/goque"
-	"os"
-	"sync"
-	"sync/atomic"
-)
-
-type BufferedQueue struct {
-	dataDir string
-	q       *goque.Queue
-	buf     []Job
-	m       sync.Mutex
-}
-
-func OpenQueue(dataDir string) (bq *BufferedQueue, err error) {
-	bq = new(BufferedQueue)
-	if config.JobBufferSize < 0 {
-		return
-	}
-	bq.dataDir = dataDir
-	bq.q, err = goque.OpenQueue(dataDir)
-	if err != nil { return nil, err }
-	return
-}
-
-func (q *BufferedQueue) Enqueue(job *Job) error {
-	atomic.AddInt64(&totalQueued, 1)
-	if q.directEnqueue(job) {
-		return nil
-	}
-
-	var gob JobGob
-	gob.ToGob(job)
-	_, err := q.q.EnqueueObject(gob)
-	return err
-}
-
-func (q *BufferedQueue) Dequeue() (job Job, err error) {
-	if q.directDequeue(&job) {
-		atomic.AddInt64(&totalQueued, -1)
-		return job, nil
-	}
-
-	if config.JobBufferSize < 0 {
-		err = goque.ErrEmpty
-		return
-	}
-
-	var item *goque.Item
-	item, err = q.q.Dequeue()
-	if err != nil { return }
-
-	atomic.AddInt64(&totalQueued, -1)
-
-	var gob JobGob
-	err = item.ToObject(&gob)
-	if err != nil { return }
-	gob.FromGob(&job)
-
-	return
-}
-
-func (q *BufferedQueue) directEnqueue(job *Job) bool {
-	q.m.Lock()
-	defer q.m.Unlock()
-
-	bs := config.JobBufferSize
-	if len(q.buf) < bs || bs < 0 {
-		q.buf = append(q.buf, *job)
-		return true
-	} else {
-		return false
-	}
-}
-
-func (q *BufferedQueue) directDequeue(job *Job) bool {
-	q.m.Lock()
-	defer q.m.Unlock()
-
-	if len(q.buf) > 0 {
-		*job = q.buf[0]
-		q.buf = q.buf[1:]
-		return true
-	} else {
-		return false
-	}
-}
-
-// Always returns nil (But implements io.Closer)
-func (q *BufferedQueue) Close() error {
-	if config.JobBufferSize < 0 {
-		return nil
-	}
-
-	// Close ignoring errors
-	q.q.Close()
-
-	// Delete files
-	if err := os.RemoveAll(q.dataDir);
-		err != nil { panic(err) }
-
-	return nil
-}
-
-type JobGob struct {
-	Uri string
-	Fails int
-	LastError string
-}
-
-func (g *JobGob) ToGob(j *Job) {
-	g.Uri = j.UriStr
-	g.Fails = j.Fails
-	if j.LastError != nil {
-		g.LastError = j.LastError.Error()
-	}
-}
-
-func (g *JobGob) FromGob(j *Job) {
-	if err := j.Uri.Parse(g.Uri);
-		err != nil { panic(err) }
-	j.UriStr = g.Uri
-	j.Fails = g.Fails
-	if g.LastError != "" {
-		j.LastError = errorString(g.LastError)
-	}
-}

release.sh

@@ -1,25 +0,0 @@
-#!/usr/bin/env bash
-
-appname="od-database-crawler"
-tag=$1
-[ -z "$tag" ] && echo "Usage: build <version>" && exit 1
-
-name=${appname}-${tag}-windows.exe
-GOOS="windows" GOARCH="amd64" go build -ldflags="-s -w" -o $name
-gzip -f $name
-echo $name
-
-name=${appname}-${tag}-linux
-GOOS="linux" GOARCH="amd64" go build -ldflags="-s -w" -o $name
-gzip -f $name
-echo $name
-
-name=${appname}-${tag}-mac
-GOOS="darwin" GOARCH="amd64" go build -ldflags="-s -w" -o $name
-gzip -f $name
-echo $name
-
-name=${appname}-${tag}-freebsd
-GOOS="freebsd" GOARCH="amd64" go build -ldflags="-s -w" -o $name
-gzip -f $name
-echo $name

scheduler.go

@@ -2,215 +2,101 @@ package main
 
 import (
 	"context"
-	"encoding/json"
-	"fmt"
 	"github.com/sirupsen/logrus"
-	"github.com/terorie/od-database-crawler/fasturl"
-	"os"
-	"path"
-	"sync"
 	"sync/atomic"
-	"time"
 )
 
-var activeTasksLock sync.Mutex
+var activeTasks int32
-var activeTasks = make(map[uint64]bool)
+var totalBuffered int64
-var numActiveTasks int32
-var totalQueued int64
 
 func Schedule(c context.Context, remotes <-chan *OD) {
 	go Stats(c)
 
-	for remote := range remotes {
+	for {
+		select {
+		case <-c.Done():
+			return
+
+		case remote := <-remotes:
 			logrus.WithField("url", remote.BaseUri.String()).
 				Info("Starting crawler")
 
-		// Collect results
-		results := make(chan File)
-
-		remote.WCtx.OD = remote
-
-		// Get queue path
-		queuePath := path.Join("queue", fmt.Sprintf("%d", remote.Task.WebsiteId))
-
-		// Delete existing queue
-		if err := os.RemoveAll(queuePath);
-			err != nil { panic(err) }
-
-		// Start new queue
-		var err error
-		remote.WCtx.Queue, err = OpenQueue(queuePath)
-		if err != nil { panic(err) }
-
 			// Spawn workers
+			remote.WCtx.in, remote.WCtx.out = makeJobBuffer(c)
 			for i := 0; i < config.Workers; i++ {
-			go remote.WCtx.Worker(results)
+				go remote.WCtx.Worker()
 			}
 
 			// Enqueue initial job
-		atomic.AddInt32(&numActiveTasks, 1)
+			atomic.AddInt32(&activeTasks, 1)
 			remote.WCtx.queueJob(Job{
+				OD:     remote,
 				Uri:    remote.BaseUri,
 				UriStr: remote.BaseUri.String(),
 				Fails:  0,
 			})
+			globalWait.Done()
 
 			// Upload result when ready
-		go remote.Watch(results)
+			go remote.Watch()
-
-		// Sleep if max number of tasks are active
-		for atomic.LoadInt32(&numActiveTasks) > config.Tasks {
-			select {
-			case <-c.Done():
-				return
-			case <-time.After(time.Second):
-				continue
-			}
 		}
 	}
 }
 
-func ScheduleTask(remotes chan<- *OD, t *Task, u *fasturl.URL) {
+func (r *OD) Watch() {
-	if !t.register() {
-		return
-	}
-
-	globalWait.Add(1)
-	now := time.Now()
-	od := &OD {
-		Task: *t,
-		BaseUri: *u,
-		Result: TaskResult {
-			WebsiteId: t.WebsiteId,
-			StartTime: now,
-			StartTimeUnix: now.Unix(),
-		},
-	}
-	remotes <- od
-}
-
-func (t *Task) register() bool {
-	activeTasksLock.Lock()
-	defer activeTasksLock.Unlock()
-
-	if _, known := activeTasks[t.WebsiteId]; known {
-		return false
-	} else {
-		activeTasks[t.WebsiteId] = true
-		return true
-	}
-}
-
-func (t *Task) unregister() {
-	activeTasksLock.Lock()
-	delete(activeTasks, t.WebsiteId)
-	activeTasksLock.Unlock()
-}
-
-func (o *OD) Watch(results chan File) {
-	// Mark job as completely done
-	defer globalWait.Done()
-	defer o.Task.unregister()
-
-	filePath := path.Join("crawled", fmt.Sprintf("%d.json", o.Task.WebsiteId))
-
-	// Open crawl results file
-	f, err := os.OpenFile(
-		filePath,
-		os.O_CREATE | os.O_RDWR | os.O_TRUNC,
-		0644,
-	)
-	if err != nil {
-		logrus.WithError(err).
-			Error("Failed saving crawl results")
-		return
-	}
-	defer f.Close()
-	defer os.Remove(filePath)
-
-	// Listen for exit code of Collect()
-	collectErrC := make(chan error)
-
-	// Block until all results are written
-	// (closes results channel)
-	o.handleCollect(results, f, collectErrC)
-
-	// Exit code of Collect()
-	err = <-collectErrC
-	close(collectErrC)
-	if err != nil {
-		logrus.WithError(err).
-			Error("Failed saving crawl results")
-		return
-	}
-
-	// Upload results
-	err = PushResult(&o.Result, f)
-	if err != nil {
-		logrus.WithError(err).
-			Error("Failed uploading crawl results")
-		return
-	}
-}
-
-func (o *OD) handleCollect(results chan File, f *os.File, collectErrC chan error) {
-	// Begin collecting results
-	go o.Task.Collect(results, f, collectErrC)
-	defer close(results)
-
 	// Wait for all jobs on remote to finish
-	o.Wait.Wait()
+	r.Wait.Wait()
+	close(r.WCtx.in)
+	atomic.AddInt32(&activeTasks, -1)
 
-	// Close queue
+	logrus.WithField("url", r.BaseUri.String()).
-	if err := o.WCtx.Queue.Close(); err != nil {
+		Info("Crawler finished")
-		panic(err)
-	}
-	atomic.AddInt32(&numActiveTasks, -1)
-
-	// Log finish
-
-	logrus.WithFields(logrus.Fields{
-		"id":  o.Task.WebsiteId,
-		"url": o.BaseUri.String(),
-		"duration": time.Since(o.Result.StartTime),
-	}).Info("Crawler finished")
-
-	// Set status code
-	now := time.Now()
-	o.Result.EndTimeUnix = now.Unix()
-	fileCount := atomic.LoadUint64(&o.Result.FileCount)
-	if fileCount == 0 {
-		errorCount := atomic.LoadUint64(&o.Result.ErrorCount)
-		if errorCount == 0 {
-			o.Result.StatusCode = "empty"
-		} else {
-			o.Result.StatusCode = "directory listing failed"
-		}
-	} else {
-		o.Result.StatusCode = "success"
-	}
 }
 
-func (t *Task) Collect(results chan File, f *os.File, errC chan<- error) {
+func makeJobBuffer(c context.Context) (chan<- Job, <-chan Job) {
-	err := t.collect(results, f)
+	in  := make(chan Job)
-	if err != nil {
+	out := make(chan Job)
-		logrus.WithError(err).
+	go bufferJobs(c, in, out)
-			Error("Failed saving crawl results")
+	return in, out
-	}
-	errC <- err
-}
-
-func (t *Task) collect(results chan File, f *os.File) error {
-	for result := range results {
-		result.Path = fasturl.PathUnescape(result.Path)
-		result.Name = fasturl.PathUnescape(result.Name)
-		resJson, err := json.Marshal(result)
-		if err != nil { panic(err) }
-		_, err = f.Write(resJson)
-		if err != nil { return err }
-		_, err = f.Write([]byte{'\n'})
-		if err != nil { return err }
 }
 
+func bufferJobs(c context.Context, in chan Job, out chan Job) {
+	defer close(out)
+	var inQueue []Job
+	outCh := func() chan Job {
+		if len(inQueue) == 0 {
 			return nil
 		}
+		return out
+	}
+	for len(inQueue) > 0 || in != nil {
+		if len(inQueue) == 0 {
+			select {
+			case v, ok := <-in:
+				if !ok {
+					in = nil
+				} else {
+					atomic.AddInt64(&totalBuffered, 1)
+					inQueue = append(inQueue, v)
+				}
+			case <-c.Done():
+				return
+			}
+		} else {
+			select {
+			case v, ok := <-in:
+				if !ok {
+					in = nil
+				} else {
+					atomic.AddInt64(&totalBuffered, 1)
+					inQueue = append(inQueue, v)
+				}
+			case outCh() <- inQueue[0]:
+				atomic.AddInt64(&totalBuffered, -1)
+				inQueue = inQueue[1:]
+			case <-c.Done():
+				return
+			}
+		}
+	}
+}

server.go

@@ -5,34 +5,39 @@ import (
 	"encoding/json"
 	"fmt"
 	"github.com/sirupsen/logrus"
-	"github.com/spf13/viper"
 	"io"
 	"mime/multipart"
 	"net/http"
 	"net/url"
 	"os"
+	"path/filepath"
 	"strconv"
-	"time"
+	"strings"
 )
 
-var serverClient = http.Client {
+const (
-	Timeout: config.ServerTimeout,
+	fileListChunkSize int64 = 5000000 // 5 mb
-}
+)
+
+var serverClient = http.DefaultClient
 
 func FetchTask() (t *Task, err error) {
-	res, err := serverClient.PostForm(
+	escToken, _ := json.Marshal(config.Token)
+	payload := `{"token":` + string(escToken) + `}`
+
+	req, err := http.NewRequest(
+		http.MethodPost,
 		config.ServerUrl + "/task/get",
-		url.Values{ "token": {config.Token} })
+		strings.NewReader(payload))
+	if err != nil { return }
+
+	res, err := serverClient.Do(req)
 	if err != nil { return }
 	defer res.Body.Close()
 
-	switch res.StatusCode {
+	if res.StatusCode != 200 {
-	case 200:
+		err = fmt.Errorf("http %s", res.Status)
-		break
+		return
-	case 404, 500:
-		return nil, nil
-	default:
-		return nil, fmt.Errorf("http %s", res.Status)
 	}
 
 	t = new(Task)
@@ -42,17 +47,21 @@ func FetchTask() (t *Task, err error) {
 	return
 }
 
-func PushResult(result *TaskResult, f *os.File) (err error) {
+func PushResult(result *TaskResult) (err error) {
-	if result.WebsiteId == 0 {
+	filePath := filepath.Join(
-		// Not a real result, don't push
+		".", "crawled",
-		return nil
+		fmt.Sprintf("%d.json", result.WebsiteId))
-	}
 
-	// Rewind to the beginning of the file
+	defer os.Remove(filePath)
-	_, err = f.Seek(0, 0)
+
-	if err != nil {
+	f, err := os.Open(filePath)
+	if os.IsNotExist(err) {
+		err = fmt.Errorf("cannot upload result: %s does not exist", filePath)
+		return
+	} else if err != nil {
 		return
 	}
+	defer f.Close()
 
 	err = uploadChunks(result.WebsiteId, f)
 	if err != nil {
@@ -64,109 +73,104 @@ func PushResult(result *TaskResult, f *os.File) (err error) {
 		return
 	}
 
-	// Upload result ignoring errors
+	err = uploadResult(result)
-	uploadResult(result)
+	if err != nil {
+		logrus.Errorf("Failed to upload result: %s", err)
+		err2 := CancelTask(result.WebsiteId)
+		if err2 != nil {
+			logrus.Error(err2)
+		}
+		return
+	}
 
 	return
 }
 
-func uploadChunks(websiteId uint64, f *os.File) error {
+func uploadChunks(websiteId uint64, f *os.File) (err error) {
-	eof := false
+	for iter := 1; iter > 0; iter++ {
-	for iter := 1; !eof; iter++ {
 		// TODO Stream with io.Pipe?
 		var b bytes.Buffer
 
 		multi := multipart.NewWriter(&b)
 
 		// Set upload fields
-		var err error
 		err = multi.WriteField("token", config.Token)
-		if err != nil { return err }
+		if err != nil { return }
 		err = multi.WriteField("website_id", fmt.Sprintf("%d", websiteId))
-		if err != nil { return err }
+		if err != nil { return }
 
 		// Copy chunk to file_list
 		formFile, err := multi.CreateFormFile("file_list", "file_list")
-		var n int64
+		_, err = io.CopyN(formFile, f, fileListChunkSize)
-		n, err = io.CopyN(formFile, f, config.ChunkSize)
+		if err == io.EOF {
-		if err != io.EOF && err != nil {
+			break
-			return err
+		} else if err == io.ErrUnexpectedEOF {
-		}
-		if n == 0 {
-			// Don't upload, no content
-			return nil
-		} else if n < config.ChunkSize {
 			err = nil
 			// Break at end of iteration
-			eof = true
+			iter = -420
-		}
-
-		multi.Close()
-
-		for retries := 0; retries < viper.GetInt(ConfUploadRetries); retries++ {
-			if retries > 0 {
-				// Error occurred, retry upload
-				time.Sleep(viper.GetDuration(ConfUploadRetryInterval))
 		}
 
 		req, err := http.NewRequest(
 			http.MethodPost,
 			config.ServerUrl + "/task/upload",
 			&b)
-			req.Header.Set("content-type", multi.FormDataContentType())
+		if err != nil { return err }
-			if err != nil { continue }
 
 		res, err := serverClient.Do(req)
-			if err != nil { continue }
+		if err != nil { return err }
 		res.Body.Close()
 
 		if res.StatusCode != http.StatusOK {
-				logrus.WithField("status", res.Status).
+			return fmt.Errorf("failed to upload list part %d: %s",
-					WithField("part", iter).
+				iter, res.Status)
-					Errorf("Upload failed")
-				continue
 		}
 
-			// Upload successful
+		logrus.Infof("Uploading file list part %d: %s",
-			break
+			iter, res.Status)
 	}
-
+	return
-		logrus.WithField("id", websiteId).
-			WithField("part", iter).
-			Infof("Uploaded files chunk")
-	}
-	return nil
 }
 
 func uploadResult(result *TaskResult) (err error) {
 	resultEnc, err := json.Marshal(result)
 	if err != nil { panic(err) }
 
-	res, err := serverClient.PostForm(
+	payload := url.Values {
-		config.ServerUrl + "/task/complete",
-		url.Values {
 		"token": {config.Token},
 		"result": {string(resultEnc)},
-		},
+	}.Encode()
-	)
+
+	req, err := http.NewRequest(
+		http.MethodPost,
+		config.ServerUrl + "/task/complete",
+		strings.NewReader(payload))
 	if err != nil { return }
-	res.Body.Close()
 
-	if res.StatusCode != http.StatusOK {
+	res, err := serverClient.Do(req)
-		return HttpError{res.StatusCode}
+	if err != nil { return }
-	}
+	res.Body.Close()
-
+
-	return
+	if res.StatusCode != http.StatusOK {
-}
+		return fmt.Errorf("failed to cancel task: %s", res.Status)
-
+	}
-func CancelTask(websiteId uint64) (err error) {
+
-	res, err := serverClient.PostForm(
+	return
-		config.ServerUrl + "/task/cancel",
+}
-		url.Values{
+
-			"token": {config.Token},
+func CancelTask(websiteId uint64) (err error) {
-			"website_id": {strconv.FormatUint(websiteId, 10)},
+	form := url.Values{
-		},
+		"token": {config.Token},
-	)
+		"website_id": {strconv.FormatUint(websiteId, 10)},
+	}
+	encForm := form.Encode()
+
+	req, err := http.NewRequest(
+		http.MethodPost,
+		config.ServerUrl + "/task/cancel",
+		strings.NewReader(encForm))
+	if err != nil { return }
+
+	res, err := serverClient.Do(req)
 	if err != nil { return }
 	res.Body.Close()
 

stats.go

@@ -3,7 +3,6 @@ package main
 import (
 	"context"
 	"github.com/sirupsen/logrus"
-	"github.com/spf13/viper"
 	"math"
 	"runtime"
 	"sync/atomic"
@@ -20,14 +19,11 @@ func Stats(c context.Context) {
 	var crawlTicker <-chan time.Time
 	var allocTicker <-chan time.Time
 
-	crawlInterval := viper.GetDuration(ConfCrawlStats)
+	if config.CrawlStats != 0 {
-	allocInterval := viper.GetDuration(ConfAllocStats)
+		crawlTicker = time.NewTicker(config.CrawlStats).C
-
-	if crawlInterval != 0 {
-		crawlTicker = time.Tick(crawlInterval)
 	}
-	if allocInterval != 0 {
+	if config.AllocStats != 0 {
-		allocTicker = time.Tick(allocInterval)
+		allocTicker = time.NewTicker(config.AllocStats).C
 	}
 
 	for {
@@ -36,17 +32,13 @@ func Stats(c context.Context) {
 			startedNow := atomic.LoadUint64(&totalStarted)
 
 			perSecond := float64(startedNow - startedLast) /
-				crawlInterval.Seconds()
+				config.CrawlStats.Seconds()
 
 			// Round to .5
 			perSecond *= 2
 			perSecond = math.Round(perSecond)
 			perSecond /= 2
 
-			if perSecond <= 0 {
-				continue
-			}
-
 			logrus.WithFields(logrus.Fields{
 				"per_second": perSecond,
 				"done":    atomic.LoadUint64(&totalDone),
@@ -61,7 +53,7 @@ func Stats(c context.Context) {
 			runtime.ReadMemStats(&mem)
 
 			logrus.WithFields(logrus.Fields{
-				"queue_count": atomic.LoadInt64(&totalQueued),
+				"queue_count": totalBuffered,
 				"heap": FormatByteCount(mem.Alloc),
 				"objects": mem.HeapObjects,
 				"num_gc": mem.NumGC,

tasks.go

@@ -0,0 +1,16 @@
+package main
+
+import "time"
+
+type Task struct {
+	WebsiteId int `json:"website_id"`
+	Url string `json:"url"`
+}
+
+type TaskResult struct {
+	StatusCode int `json:"status_code"`
+	FileCount uint64 `json:"file_count"`
+	StartTime time.Time `json:"start_time"`
+	EndTime time.Time `json:"end_time"`
+	WebsiteId uint64 `json:"website_id"`
+}

util.go

@@ -1,9 +1,6 @@
 package main
 
-import (
+import "fmt"
-	"fmt"
-	"sync"
-)
 
 // https://programming.guide/go/formatting-byte-size-to-human-readable-format.html
 func FormatByteCount(b uint64) string {
@@ -19,20 +16,3 @@ func FormatByteCount(b uint64) string {
 		return fmt.Sprintf("%.1f %ciB", float64(b)/float64(div), "KMGTPE"[exp])
 	}
 }
-
-type Hooks struct {
-	m sync.Mutex
-	l []func()
-}
-
-func (h *Hooks) Add(hook func()) {
-	h.m.Lock()
-	h.l = append(h.l, hook)
-	h.m.Unlock()
-}
-
-func (h *Hooks) Execute() {
-	for _, hook := range h.l {
-		hook()
-	}
-}

worker.go

@@ -1,12 +1,8 @@
 package main
 
 import (
-	"github.com/beeker1121/goque"
 	"github.com/sirupsen/logrus"
-	"github.com/valyala/fasthttp"
 	"math"
-	"sort"
-	"strings"
 	"sync"
 	"sync/atomic"
 	"time"
@@ -15,38 +11,24 @@ import (
 var globalWait sync.WaitGroup
 
 type WorkerContext struct {
-	OD *OD
+	in            chan<- Job
-	Queue *BufferedQueue
+	out           <-chan Job
 	lastRateLimit time.Time
 	numRateLimits int
 }
 
-func (w *WorkerContext) Worker(results chan<- File) {
+func (w WorkerContext) Worker() {
-	for {
+	for job := range w.out {
-		job, err := w.Queue.Dequeue()
+		w.step(job)
-		switch err {
-		case goque.ErrEmpty:
-			time.Sleep(500 * time.Millisecond)
-			continue
-
-		case goque.ErrDBClosed:
-			return
-
-		case nil:
-			w.step(results, job)
-
-		default:
-			panic(err)
-		}
 	}
 }
 
-func (w *WorkerContext) step(results chan<- File, job Job) {
+func (w WorkerContext) step(job Job) {
-	defer w.finishJob()
+	defer w.finishJob(&job)
 
 	var f File
 
-	newJobs, err := w.DoJob(&job, &f)
+	newJobs, err := DoJob(&job, &f)
 	atomic.AddUint64(&totalStarted, 1)
 	if err == ErrKnown {
 		return
@@ -55,15 +37,10 @@ func (w *WorkerContext) step(results chan<- File, job Job) {
 	if err != nil {
 		job.Fails++
 
-		if httpErr, ok := err.(*HttpError); ok {
+		if err == ErrForbidden {
-			switch httpErr.code {
+			// Don't attempt crawling again
-			case fasthttp.StatusTooManyRequests:
-				err = ErrRateLimit
-			default:
-				// Don't retry HTTP error codes
 			return
 		}
-		}
 
 		if job.Fails > config.Retries {
 			atomic.AddUint64(&totalAborted, 1)
@@ -85,22 +62,17 @@ func (w *WorkerContext) step(results chan<- File, job Job) {
 		w.queueJob(job)
 	}
 
-	if !f.IsDir {
+	job.OD.Files = append(job.OD.Files, f)
-		results <- f
-	}
 }
 
-func (w *WorkerContext) DoJob(job *Job, f *File) (newJobs []Job, err error) {
+func DoJob(job *Job, f *File) (newJobs []Job, err error) {
-	if len(job.Uri.Path) == 0 { return }
+	if len(job.Uri.Path) != 0 && job.Uri.Path[len(job.Uri.Path)-1] == '/' {
-	if job.Uri.Path[len(job.Uri.Path)-1] == '/' {
 		// Load directory
 		links, err := GetDir(job, f)
 		if err != nil {
-			if !isErrSilent(err) {
 			logrus.WithError(err).
-					WithField("url", job.UriStr).
+				WithField("url", job.Uri.String()).
-					Error("Failed to crawl dir")
+				Error("Failed getting dir")
-			}
 			return nil, err
 		}
 
@@ -108,58 +80,43 @@ func (w *WorkerContext) DoJob(job *Job, f *File) (newJobs []Job, err error) {
 		hash := f.HashDir(links)
 
 		// Skip symlinked dirs
-		if w.OD.LoadOrStoreKey(&hash) {
+		if _, old := job.OD.Scanned.LoadOrStore(hash, true); old {
 			return nil, ErrKnown
 		}
 
-		// Sort by path
-		sort.Slice(links, func(i, j int) bool {
-			return strings.Compare(links[i].Path, links[j].Path) < 0
-		})
-
-		var newJobCount int
-		var lastLink string
 		for _, link := range links {
-			uriStr := link.String()
+			// Skip already queued links
-
+			if _, old := job.OD.Scanned.LoadOrStore(link, true); old {
-			// Ignore dupes
-			if uriStr == lastLink {
 				continue
 			}
-			lastLink = uriStr
+			job.OD.Wait.Add(1)
-
 			newJobs = append(newJobs, Job{
+				OD:     job.OD,
 				Uri:    link,
-				UriStr: uriStr,
+				UriStr: link.String(),
 				Fails:  0,
 			})
-
-			newJobCount++
 		}
-		if config.Verbose {
 		logrus.WithFields(logrus.Fields{
 			"url":   job.UriStr,
-				"files": newJobCount,
+			"files": len(links),
 		}).Debug("Listed")
-		}
 	} else {
 		// Load file
 		err := GetFile(job.Uri, f)
 		if err != nil {
-			if !isErrSilent(err) {
 			logrus.WithError(err).
-					WithField("url", job.UriStr).
+				WithField("url", job.Uri.String()).
-					Error("Failed to crawl file")
+				Error("Failed getting file")
-			}
 			return nil, err
 		}
-		atomic.AddUint64(&w.OD.Result.FileCount, 1)
 	}
 	return
 }
 
-func (w *WorkerContext) queueJob(job Job) {
+func (w WorkerContext) queueJob(job Job) {
-	w.OD.Wait.Add(1)
+	job.OD.Wait.Add(1)
+	globalWait.Add(1)
 
 	if w.numRateLimits > 0 {
 		if time.Since(w.lastRateLimit) > 5*time.Second {
@@ -167,23 +124,14 @@ func (w *WorkerContext) queueJob(job Job) {
 		} else {
 			time.Sleep(time.Duration(math.Sqrt(float64(50*w.numRateLimits))) *
 				100 * time.Millisecond)
+			w.in <- job
+		}
+	} else {
+		w.in <- job
 	}
 }
 
-	if err := w.Queue.Enqueue(&job); err != nil {
+func (w WorkerContext) finishJob(job *Job) {
-		panic(err)
+	job.OD.Wait.Done()
-	}
+	globalWait.Done()
-}
-
-func (w *WorkerContext) finishJob() {
-	w.OD.Wait.Done()
-}
-
-func isErrSilent(err error) bool {
-	if !config.PrintHTTP {
-		if _, ok := err.(*HttpError); ok {
-			return true
-		}
-	}
-	return false
 }