|
5 | 5 | package webhook
|
6 | 6 |
|
7 | 7 | import (
|
| 8 | + "net" |
8 | 9 | "net/http"
|
9 | 10 | "net/url"
|
10 | 11 | "testing"
|
@@ -37,3 +38,84 @@ func TestWebhookProxy(t *testing.T) {
|
37 | 38 | }
|
38 | 39 | }
|
39 | 40 | }
|
| 41 | + |
| 42 | +func TestIsWebhookRequestAllowed(t *testing.T) { |
| 43 | + type tc struct { |
| 44 | + host string |
| 45 | + ip net.IP |
| 46 | + expected bool |
| 47 | + } |
| 48 | + |
| 49 | + ah, an := setting.ParseWebhookAllowedHostList("private, global, *.google.com, 169.254.1.0/24") |
| 50 | + cases := []tc{ |
| 51 | + {"", net.IPv4zero, false}, |
| 52 | + |
| 53 | + {"", net.ParseIP("127.0.0.1"), false}, |
| 54 | + |
| 55 | + {"", net.ParseIP("10.0.1.1"), true}, |
| 56 | + {"", net.ParseIP("192.168.1.1"), true}, |
| 57 | + |
| 58 | + {"", net.ParseIP("8.8.8.8"), true}, |
| 59 | + |
| 60 | + {"google.com", net.IPv4zero, false}, |
| 61 | + {"sub.google.com", net.IPv4zero, true}, |
| 62 | + |
| 63 | + {"", net.ParseIP("169.254.1.1"), true}, |
| 64 | + {"", net.ParseIP("169.254.2.2"), false}, |
| 65 | + } |
| 66 | + for _, c := range cases { |
| 67 | + assert.Equalf(t, c.expected, isWebhookRequestAllowed(ah, an, c.host, c.ip), "case %s(%v)", c.host, c.ip) |
| 68 | + } |
| 69 | + |
| 70 | + ah, an = setting.ParseWebhookAllowedHostList("loopback") |
| 71 | + cases = []tc{ |
| 72 | + {"", net.IPv4zero, false}, |
| 73 | + {"", net.ParseIP("127.0.0.1"), true}, |
| 74 | + {"", net.ParseIP("10.0.1.1"), false}, |
| 75 | + {"", net.ParseIP("192.168.1.1"), false}, |
| 76 | + {"", net.ParseIP("8.8.8.8"), false}, |
| 77 | + {"google.com", net.IPv4zero, false}, |
| 78 | + } |
| 79 | + for _, c := range cases { |
| 80 | + assert.Equalf(t, c.expected, isWebhookRequestAllowed(ah, an, c.host, c.ip), "case %s(%v)", c.host, c.ip) |
| 81 | + } |
| 82 | + |
| 83 | + ah, an = setting.ParseWebhookAllowedHostList("private") |
| 84 | + cases = []tc{ |
| 85 | + {"", net.IPv4zero, false}, |
| 86 | + {"", net.ParseIP("127.0.0.1"), false}, |
| 87 | + {"", net.ParseIP("10.0.1.1"), true}, |
| 88 | + {"", net.ParseIP("192.168.1.1"), true}, |
| 89 | + {"", net.ParseIP("8.8.8.8"), false}, |
| 90 | + {"google.com", net.IPv4zero, false}, |
| 91 | + } |
| 92 | + for _, c := range cases { |
| 93 | + assert.Equalf(t, c.expected, isWebhookRequestAllowed(ah, an, c.host, c.ip), "case %s(%v)", c.host, c.ip) |
| 94 | + } |
| 95 | + |
| 96 | + ah, an = setting.ParseWebhookAllowedHostList("global") |
| 97 | + cases = []tc{ |
| 98 | + {"", net.IPv4zero, false}, |
| 99 | + {"", net.ParseIP("127.0.0.1"), false}, |
| 100 | + {"", net.ParseIP("10.0.1.1"), false}, |
| 101 | + {"", net.ParseIP("192.168.1.1"), false}, |
| 102 | + {"", net.ParseIP("8.8.8.8"), true}, |
| 103 | + {"google.com", net.IPv4zero, false}, |
| 104 | + } |
| 105 | + for _, c := range cases { |
| 106 | + assert.Equalf(t, c.expected, isWebhookRequestAllowed(ah, an, c.host, c.ip), "case %s(%v)", c.host, c.ip) |
| 107 | + } |
| 108 | + |
| 109 | + ah, an = setting.ParseWebhookAllowedHostList("all") |
| 110 | + cases = []tc{ |
| 111 | + {"", net.IPv4zero, true}, |
| 112 | + {"", net.ParseIP("127.0.0.1"), true}, |
| 113 | + {"", net.ParseIP("10.0.1.1"), true}, |
| 114 | + {"", net.ParseIP("192.168.1.1"), true}, |
| 115 | + {"", net.ParseIP("8.8.8.8"), true}, |
| 116 | + {"google.com", net.IPv4zero, true}, |
| 117 | + } |
| 118 | + for _, c := range cases { |
| 119 | + assert.Equalf(t, c.expected, isWebhookRequestAllowed(ah, an, c.host, c.ip), "case %s(%v)", c.host, c.ip) |
| 120 | + } |
| 121 | +} |
0 commit comments