resolvconf_linux_test.go 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307
  1. package resolvconf
  2. import (
  3. "bytes"
  4. "io/ioutil"
  5. "os"
  6. "testing"
  7. "github.com/docker/docker/libnetwork/types"
  8. "github.com/docker/docker/pkg/ioutils"
  9. )
  10. func TestGet(t *testing.T) {
  11. resolvConfUtils, err := Get()
  12. if err != nil {
  13. t.Fatal(err)
  14. }
  15. resolvConfSystem, err := ioutil.ReadFile("/etc/resolv.conf")
  16. if err != nil {
  17. t.Fatal(err)
  18. }
  19. if string(resolvConfUtils.Content) != string(resolvConfSystem) {
  20. t.Fatalf("/etc/resolv.conf and GetResolvConf have different content.")
  21. }
  22. hashSystem, err := ioutils.HashData(bytes.NewReader(resolvConfSystem))
  23. if err != nil {
  24. t.Fatal(err)
  25. }
  26. if resolvConfUtils.Hash != hashSystem {
  27. t.Fatalf("/etc/resolv.conf and GetResolvConf have different hashes.")
  28. }
  29. }
  30. func TestGetNameservers(t *testing.T) {
  31. for resolv, result := range map[string][]string{`
  32. nameserver 1.2.3.4
  33. nameserver 40.3.200.10
  34. search example.com`: {"1.2.3.4", "40.3.200.10"},
  35. `search example.com`: {},
  36. `nameserver 1.2.3.4
  37. search example.com
  38. nameserver 4.30.20.100`: {"1.2.3.4", "4.30.20.100"},
  39. ``: {},
  40. ` nameserver 1.2.3.4 `: {"1.2.3.4"},
  41. `search example.com
  42. nameserver 1.2.3.4
  43. #nameserver 4.3.2.1`: {"1.2.3.4"},
  44. `search example.com
  45. nameserver 1.2.3.4 # not 4.3.2.1`: {"1.2.3.4"},
  46. } {
  47. test := GetNameservers([]byte(resolv), types.IP)
  48. if !strSlicesEqual(test, result) {
  49. t.Fatalf("Wrong nameserver string {%s} should be %v. Input: %s", test, result, resolv)
  50. }
  51. }
  52. }
  53. func TestGetNameserversAsCIDR(t *testing.T) {
  54. for resolv, result := range map[string][]string{`
  55. nameserver 1.2.3.4
  56. nameserver 40.3.200.10
  57. search example.com`: {"1.2.3.4/32", "40.3.200.10/32"},
  58. `search example.com`: {},
  59. `nameserver 1.2.3.4
  60. search example.com
  61. nameserver 4.30.20.100`: {"1.2.3.4/32", "4.30.20.100/32"},
  62. ``: {},
  63. ` nameserver 1.2.3.4 `: {"1.2.3.4/32"},
  64. `search example.com
  65. nameserver 1.2.3.4
  66. #nameserver 4.3.2.1`: {"1.2.3.4/32"},
  67. `search example.com
  68. nameserver 1.2.3.4 # not 4.3.2.1`: {"1.2.3.4/32"},
  69. } {
  70. test := GetNameserversAsCIDR([]byte(resolv))
  71. if !strSlicesEqual(test, result) {
  72. t.Fatalf("Wrong nameserver string {%s} should be %v. Input: %s", test, result, resolv)
  73. }
  74. }
  75. }
  76. func TestGetSearchDomains(t *testing.T) {
  77. for resolv, result := range map[string][]string{
  78. `search example.com`: {"example.com"},
  79. `search example.com # ignored`: {"example.com"},
  80. ` search example.com `: {"example.com"},
  81. ` search example.com # ignored`: {"example.com"},
  82. `search foo.example.com example.com`: {"foo.example.com", "example.com"},
  83. ` search foo.example.com example.com `: {"foo.example.com", "example.com"},
  84. ` search foo.example.com example.com # ignored`: {"foo.example.com", "example.com"},
  85. ``: {},
  86. `# ignored`: {},
  87. `nameserver 1.2.3.4
  88. search foo.example.com example.com`: {"foo.example.com", "example.com"},
  89. `nameserver 1.2.3.4
  90. search dup1.example.com dup2.example.com
  91. search foo.example.com example.com`: {"foo.example.com", "example.com"},
  92. `nameserver 1.2.3.4
  93. search foo.example.com example.com
  94. nameserver 4.30.20.100`: {"foo.example.com", "example.com"},
  95. } {
  96. test := GetSearchDomains([]byte(resolv))
  97. if !strSlicesEqual(test, result) {
  98. t.Fatalf("Wrong search domain string {%s} should be %v. Input: %s", test, result, resolv)
  99. }
  100. }
  101. }
  102. func TestGetOptions(t *testing.T) {
  103. for resolv, result := range map[string][]string{
  104. `options opt1`: {"opt1"},
  105. `options opt1 # ignored`: {"opt1"},
  106. ` options opt1 `: {"opt1"},
  107. ` options opt1 # ignored`: {"opt1"},
  108. `options opt1 opt2 opt3`: {"opt1", "opt2", "opt3"},
  109. `options opt1 opt2 opt3 # ignored`: {"opt1", "opt2", "opt3"},
  110. ` options opt1 opt2 opt3 `: {"opt1", "opt2", "opt3"},
  111. ` options opt1 opt2 opt3 # ignored`: {"opt1", "opt2", "opt3"},
  112. ``: {},
  113. `# ignored`: {},
  114. `nameserver 1.2.3.4`: {},
  115. `nameserver 1.2.3.4
  116. options opt1 opt2 opt3`: {"opt1", "opt2", "opt3"},
  117. `nameserver 1.2.3.4
  118. options opt1 opt2
  119. options opt3 opt4`: {"opt3", "opt4"},
  120. } {
  121. test := GetOptions([]byte(resolv))
  122. if !strSlicesEqual(test, result) {
  123. t.Fatalf("Wrong options string {%s} should be %v. Input: %s", test, result, resolv)
  124. }
  125. }
  126. }
  127. func strSlicesEqual(a, b []string) bool {
  128. if len(a) != len(b) {
  129. return false
  130. }
  131. for i, v := range a {
  132. if v != b[i] {
  133. return false
  134. }
  135. }
  136. return true
  137. }
  138. func TestBuild(t *testing.T) {
  139. file, err := ioutil.TempFile("", "")
  140. if err != nil {
  141. t.Fatal(err)
  142. }
  143. defer os.Remove(file.Name())
  144. _, err = Build(file.Name(), []string{"ns1", "ns2", "ns3"}, []string{"search1"}, []string{"opt1"})
  145. if err != nil {
  146. t.Fatal(err)
  147. }
  148. content, err := ioutil.ReadFile(file.Name())
  149. if err != nil {
  150. t.Fatal(err)
  151. }
  152. if expected := "search search1\nnameserver ns1\nnameserver ns2\nnameserver ns3\noptions opt1\n"; !bytes.Contains(content, []byte(expected)) {
  153. t.Fatalf("Expected to find '%s' got '%s'", expected, content)
  154. }
  155. }
  156. func TestBuildWithZeroLengthDomainSearch(t *testing.T) {
  157. file, err := ioutil.TempFile("", "")
  158. if err != nil {
  159. t.Fatal(err)
  160. }
  161. defer os.Remove(file.Name())
  162. _, err = Build(file.Name(), []string{"ns1", "ns2", "ns3"}, []string{"."}, []string{"opt1"})
  163. if err != nil {
  164. t.Fatal(err)
  165. }
  166. content, err := ioutil.ReadFile(file.Name())
  167. if err != nil {
  168. t.Fatal(err)
  169. }
  170. if expected := "nameserver ns1\nnameserver ns2\nnameserver ns3\noptions opt1\n"; !bytes.Contains(content, []byte(expected)) {
  171. t.Fatalf("Expected to find '%s' got '%s'", expected, content)
  172. }
  173. if notExpected := "search ."; bytes.Contains(content, []byte(notExpected)) {
  174. t.Fatalf("Expected to not find '%s' got '%s'", notExpected, content)
  175. }
  176. }
  177. func TestBuildWithNoOptions(t *testing.T) {
  178. file, err := ioutil.TempFile("", "")
  179. if err != nil {
  180. t.Fatal(err)
  181. }
  182. defer os.Remove(file.Name())
  183. _, err = Build(file.Name(), []string{"ns1", "ns2", "ns3"}, []string{"search1"}, []string{})
  184. if err != nil {
  185. t.Fatal(err)
  186. }
  187. content, err := ioutil.ReadFile(file.Name())
  188. if err != nil {
  189. t.Fatal(err)
  190. }
  191. if expected := "search search1\nnameserver ns1\nnameserver ns2\nnameserver ns3\n"; !bytes.Contains(content, []byte(expected)) {
  192. t.Fatalf("Expected to find '%s' got '%s'", expected, content)
  193. }
  194. if notExpected := "search ."; bytes.Contains(content, []byte(notExpected)) {
  195. t.Fatalf("Expected to not find '%s' got '%s'", notExpected, content)
  196. }
  197. }
  198. func TestFilterResolvDns(t *testing.T) {
  199. ns0 := "nameserver 10.16.60.14\nnameserver 10.16.60.21\n"
  200. if result, _ := FilterResolvDNS([]byte(ns0), false); result != nil {
  201. if ns0 != string(result.Content) {
  202. t.Fatalf("Failed No Localhost: expected \n<%s> got \n<%s>", ns0, string(result.Content))
  203. }
  204. }
  205. ns1 := "nameserver 10.16.60.14\nnameserver 10.16.60.21\nnameserver 127.0.0.1\n"
  206. if result, _ := FilterResolvDNS([]byte(ns1), false); result != nil {
  207. if ns0 != string(result.Content) {
  208. t.Fatalf("Failed Localhost: expected \n<%s> got \n<%s>", ns0, string(result.Content))
  209. }
  210. }
  211. ns1 = "nameserver 10.16.60.14\nnameserver 127.0.0.1\nnameserver 10.16.60.21\n"
  212. if result, _ := FilterResolvDNS([]byte(ns1), false); result != nil {
  213. if ns0 != string(result.Content) {
  214. t.Fatalf("Failed Localhost: expected \n<%s> got \n<%s>", ns0, string(result.Content))
  215. }
  216. }
  217. ns1 = "nameserver 127.0.1.1\nnameserver 10.16.60.14\nnameserver 10.16.60.21\n"
  218. if result, _ := FilterResolvDNS([]byte(ns1), false); result != nil {
  219. if ns0 != string(result.Content) {
  220. t.Fatalf("Failed Localhost: expected \n<%s> got \n<%s>", ns0, string(result.Content))
  221. }
  222. }
  223. ns1 = "nameserver ::1\nnameserver 10.16.60.14\nnameserver 127.0.2.1\nnameserver 10.16.60.21\n"
  224. if result, _ := FilterResolvDNS([]byte(ns1), false); result != nil {
  225. if ns0 != string(result.Content) {
  226. t.Fatalf("Failed Localhost: expected \n<%s> got \n<%s>", ns0, string(result.Content))
  227. }
  228. }
  229. ns1 = "nameserver 10.16.60.14\nnameserver ::1\nnameserver 10.16.60.21\nnameserver ::1"
  230. if result, _ := FilterResolvDNS([]byte(ns1), false); result != nil {
  231. if ns0 != string(result.Content) {
  232. t.Fatalf("Failed Localhost: expected \n<%s> got \n<%s>", ns0, string(result.Content))
  233. }
  234. }
  235. // with IPv6 disabled (false param), the IPv6 nameserver should be removed
  236. ns1 = "nameserver 10.16.60.14\nnameserver 2002:dead:beef::1\nnameserver 10.16.60.21\nnameserver ::1"
  237. if result, _ := FilterResolvDNS([]byte(ns1), false); result != nil {
  238. if ns0 != string(result.Content) {
  239. t.Fatalf("Failed Localhost+IPv6 off: expected \n<%s> got \n<%s>", ns0, string(result.Content))
  240. }
  241. }
  242. // with IPv6 disabled (false param), the IPv6 link-local nameserver with zone ID should be removed
  243. ns1 = "nameserver 10.16.60.14\nnameserver FE80::BB1%1\nnameserver FE80::BB1%eth0\nnameserver 10.16.60.21\n"
  244. if result, _ := FilterResolvDNS([]byte(ns1), false); result != nil {
  245. if ns0 != string(result.Content) {
  246. t.Fatalf("Failed Localhost+IPv6 off: expected \n<%s> got \n<%s>", ns0, string(result.Content))
  247. }
  248. }
  249. // with IPv6 enabled, the IPv6 nameserver should be preserved
  250. ns0 = "nameserver 10.16.60.14\nnameserver 2002:dead:beef::1\nnameserver 10.16.60.21\n"
  251. ns1 = "nameserver 10.16.60.14\nnameserver 2002:dead:beef::1\nnameserver 10.16.60.21\nnameserver ::1"
  252. if result, _ := FilterResolvDNS([]byte(ns1), true); result != nil {
  253. if ns0 != string(result.Content) {
  254. t.Fatalf("Failed Localhost+IPv6 on: expected \n<%s> got \n<%s>", ns0, string(result.Content))
  255. }
  256. }
  257. // with IPv6 enabled, and no non-localhost servers, Google defaults (both IPv4+IPv6) should be added
  258. ns0 = "\nnameserver 8.8.8.8\nnameserver 8.8.4.4\nnameserver 2001:4860:4860::8888\nnameserver 2001:4860:4860::8844"
  259. ns1 = "nameserver 127.0.0.1\nnameserver ::1\nnameserver 127.0.2.1"
  260. if result, _ := FilterResolvDNS([]byte(ns1), true); result != nil {
  261. if ns0 != string(result.Content) {
  262. t.Fatalf("Failed no Localhost+IPv6 enabled: expected \n<%s> got \n<%s>", ns0, string(result.Content))
  263. }
  264. }
  265. // with IPv6 disabled, and no non-localhost servers, Google defaults (only IPv4) should be added
  266. ns0 = "\nnameserver 8.8.8.8\nnameserver 8.8.4.4"
  267. ns1 = "nameserver 127.0.0.1\nnameserver ::1\nnameserver 127.0.2.1"
  268. if result, _ := FilterResolvDNS([]byte(ns1), false); result != nil {
  269. if ns0 != string(result.Content) {
  270. t.Fatalf("Failed no Localhost+IPv6 enabled: expected \n<%s> got \n<%s>", ns0, string(result.Content))
  271. }
  272. }
  273. }