1. package main
2.  
3. import (
4.  "flag"
5.  "fmt"
6.  "math/rand"
7.  "net/http"
8.  "strings"
9.  "sync"
10.  "time"
11. )
12.  
13. var (
14.  url             string
15.  payload         string
16.  threads         int
17.  requestCounter  int
18.  printedMsgs     []string
19.  waitGroup       sync.WaitGroup
20. )
21.  
22. func printMsg(msg string) {
23.  if !contains(printedMsgs, msg) {
24.   fmt.Printf("\n%s after %d requests\n", msg, requestCounter)
25.   printedMsgs = append(printedMsgs, msg)
26.  }
27. }
28.  
29. func contains(slice []string, item string) bool {
30.  for _, s := range slice {
31.   if s == item {
32.    return true
33.   }
34.  }
35.  return false
36. }
37.  
38. func init() {
39.  rand.Seed(time.Now().UnixNano())
40. }
41.  
42. func handleStatusCodes(statusCode int) {
43.  requestCounter++
44.  fmt.Printf("\r%d requests have been sent", requestCounter)
45.  
46.  if statusCode == 429 {
47.   printMsg("You have been throttled")
48.  }
49.  if statusCode == 500 {
50.   printMsg("Status code 500 received")
51.  }
52. }
53.  
54. func sendGET() {
55.  defer waitGroup.Done()
56.  
57.  resp, err := http.Get(url)
58.  if err != nil {
59.   return
60.  }
61.  defer resp.Body.Close()
62.  
63.  handleStatusCodes(resp.StatusCode)
64. }
65.  
66. func sendPOST() {
67.  defer waitGroup.Done()
68.  
69.  resp, err := http.Post(url, "application/x-www-form-urlencoded", strings.NewReader(payload))
70.  if err != nil {
71.   return
72.  }
73.  defer resp.Body.Close()
74.  
75.  handleStatusCodes(resp.StatusCode)
76. }
77.  
78. func main() {
79.  flag.StringVar(&url;, "g", "", "Specify GET request. Usage: -g '<url>'")
80.  flag.StringVar(&url;, "p", "", "Specify POST request. Usage: -p '<url>'")
81.  flag.StringVar(&payload;, "d", "", "Specify data payload for POST request")
82.  flag.IntVar(&threads;, "t", 500, "Specify number of threads to be used")
83.  flag.Parse()
84.  
85.  if url == "" {
86.   flag.Usage()
87.   return
88.  }
89.  
90.  if flag.NFlag() == 0 {
91.   flag.Usage()
92.   return
93.  }
94.  
95.  if flag.NFlag() == 1 {
96.   fmt.Println("You must specify either a GET (-g) or POST (-p) request.")
97.   return
98.  }
99.  
100.  waitGroup.Add(threads)
101.  
102.  for i := 0; i < threads; i++ {
103.   if url != "" {
104.    if payload != "" {
105.     go sendPOST()
106.    } else {
107.     go sendGET()
108.    }
109.   }
110.  }
111.  waitGroup.Wait()
112. }