libnetwork_test.go 24 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129
  1. package libnetwork_test
  2. import (
  3. "bytes"
  4. "flag"
  5. "fmt"
  6. "io/ioutil"
  7. "net"
  8. "os"
  9. "runtime"
  10. "strconv"
  11. "sync"
  12. "testing"
  13. log "github.com/Sirupsen/logrus"
  14. "github.com/docker/docker/pkg/reexec"
  15. "github.com/docker/libnetwork"
  16. "github.com/docker/libnetwork/netutils"
  17. "github.com/docker/libnetwork/pkg/netlabel"
  18. "github.com/docker/libnetwork/pkg/options"
  19. "github.com/vishvananda/netns"
  20. )
  21. const (
  22. bridgeNetType = "bridge"
  23. bridgeName = "docker0"
  24. )
  25. func TestMain(m *testing.M) {
  26. if reexec.Init() {
  27. return
  28. }
  29. os.Exit(m.Run())
  30. }
  31. func createTestNetwork(networkType, networkName string, option options.Generic, netOption options.Generic) (libnetwork.Network, error) {
  32. controller := libnetwork.New()
  33. genericOption := make(map[string]interface{})
  34. genericOption[netlabel.GenericData] = option
  35. err := controller.ConfigureNetworkDriver(networkType, genericOption)
  36. if err != nil {
  37. return nil, err
  38. }
  39. network, err := controller.NewNetwork(networkType, networkName,
  40. libnetwork.NetworkOptionGeneric(netOption))
  41. if err != nil {
  42. return nil, err
  43. }
  44. return network, nil
  45. }
  46. func getEmptyGenericOption() map[string]interface{} {
  47. genericOption := make(map[string]interface{})
  48. genericOption[netlabel.GenericData] = options.Generic{}
  49. return genericOption
  50. }
  51. func getPortMapping() []netutils.PortBinding {
  52. return []netutils.PortBinding{
  53. netutils.PortBinding{Proto: netutils.TCP, Port: uint16(230), HostPort: uint16(23000)},
  54. netutils.PortBinding{Proto: netutils.UDP, Port: uint16(200), HostPort: uint16(22000)},
  55. netutils.PortBinding{Proto: netutils.TCP, Port: uint16(120), HostPort: uint16(12000)},
  56. }
  57. }
  58. func TestNull(t *testing.T) {
  59. network, err := createTestNetwork("null", "testnetwork", options.Generic{},
  60. options.Generic{})
  61. if err != nil {
  62. t.Fatal(err)
  63. }
  64. ep, err := network.CreateEndpoint("testep")
  65. if err != nil {
  66. t.Fatal(err)
  67. }
  68. _, err = ep.Join("null_container",
  69. libnetwork.JoinOptionHostname("test"),
  70. libnetwork.JoinOptionDomainname("docker.io"),
  71. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  72. if err != nil {
  73. t.Fatal(err)
  74. }
  75. err = ep.Leave("null_container")
  76. if err != nil {
  77. t.Fatal(err)
  78. }
  79. if err := ep.Delete(); err != nil {
  80. t.Fatal(err)
  81. }
  82. if err := network.Delete(); err != nil {
  83. t.Fatal(err)
  84. }
  85. }
  86. func TestHost(t *testing.T) {
  87. network, err := createTestNetwork("host", "testnetwork", options.Generic{}, options.Generic{})
  88. if err != nil {
  89. t.Fatal(err)
  90. }
  91. ep, err := network.CreateEndpoint("testep")
  92. if err != nil {
  93. t.Fatal(err)
  94. }
  95. _, err = ep.Join("host_container",
  96. libnetwork.JoinOptionHostname("test"),
  97. libnetwork.JoinOptionDomainname("docker.io"),
  98. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"),
  99. libnetwork.JoinOptionUseDefaultSandbox())
  100. if err != nil {
  101. t.Fatal(err)
  102. }
  103. err = ep.Leave("host_container")
  104. if err != nil {
  105. t.Fatal(err)
  106. }
  107. if err := ep.Delete(); err != nil {
  108. t.Fatal(err)
  109. }
  110. if err := network.Delete(); err != nil {
  111. t.Fatal(err)
  112. }
  113. }
  114. func TestBridge(t *testing.T) {
  115. defer netutils.SetupTestNetNS(t)()
  116. ip, subnet, err := net.ParseCIDR("192.168.100.1/24")
  117. if err != nil {
  118. t.Fatal(err)
  119. }
  120. subnet.IP = ip
  121. ip, cidr, err := net.ParseCIDR("192.168.100.2/28")
  122. if err != nil {
  123. t.Fatal(err)
  124. }
  125. cidr.IP = ip
  126. ip, cidrv6, err := net.ParseCIDR("fe90::1/96")
  127. if err != nil {
  128. t.Fatal(err)
  129. }
  130. cidrv6.IP = ip
  131. log.Debug("Adding a bridge")
  132. option := options.Generic{
  133. "EnableIPForwarding": true,
  134. }
  135. netOption := options.Generic{
  136. "BridgeName": bridgeName,
  137. "AddressIPv4": subnet,
  138. "FixedCIDR": cidr,
  139. "FixedCIDRv6": cidrv6,
  140. "EnableIPv6": true,
  141. "EnableIPTables": true,
  142. "EnableIPMasquerade": true,
  143. "EnableICC": true,
  144. "AllowNonDefaultBridge": true}
  145. network, err := createTestNetwork(bridgeNetType, "testnetwork", option, netOption)
  146. if err != nil {
  147. t.Fatal(err)
  148. }
  149. ep, err := network.CreateEndpoint("testep", libnetwork.CreateOptionPortMapping(getPortMapping()))
  150. if err != nil {
  151. t.Fatal(err)
  152. }
  153. epInfo, err := ep.Info()
  154. if err != nil {
  155. t.Fatal(err)
  156. }
  157. pmd, ok := epInfo[netlabel.PortMap]
  158. if !ok {
  159. t.Fatalf("Could not find expected info in endpoint data")
  160. }
  161. pm, ok := pmd.([]netutils.PortBinding)
  162. if !ok {
  163. t.Fatalf("Unexpected format for port mapping in endpoint operational data")
  164. }
  165. if len(pm) != 3 {
  166. t.Fatalf("Incomplete data for port mapping in endpoint operational data: %d", len(pm))
  167. }
  168. if err := ep.Delete(); err != nil {
  169. t.Fatal(err)
  170. }
  171. if err := network.Delete(); err != nil {
  172. t.Fatal(err)
  173. }
  174. }
  175. func TestUnknownDriver(t *testing.T) {
  176. defer netutils.SetupTestNetNS(t)()
  177. _, err := createTestNetwork("unknowndriver", "testnetwork", options.Generic{}, options.Generic{})
  178. if err == nil {
  179. t.Fatal("Expected to fail. But instead succeeded")
  180. }
  181. if _, ok := err.(libnetwork.NetworkTypeError); !ok {
  182. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  183. }
  184. }
  185. func TestNilDriver(t *testing.T) {
  186. controller := libnetwork.New()
  187. _, err := controller.NewNetwork("framerelay", "dummy",
  188. libnetwork.NetworkOptionGeneric(getEmptyGenericOption()))
  189. if err == nil {
  190. t.Fatal("Expected to fail. But instead succeeded")
  191. }
  192. if err != libnetwork.ErrInvalidNetworkDriver {
  193. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  194. }
  195. }
  196. func TestNoInitDriver(t *testing.T) {
  197. controller := libnetwork.New()
  198. _, err := controller.NewNetwork("ppp", "dummy",
  199. libnetwork.NetworkOptionGeneric(getEmptyGenericOption()))
  200. if err == nil {
  201. t.Fatal("Expected to fail. But instead succeeded")
  202. }
  203. if err != libnetwork.ErrInvalidNetworkDriver {
  204. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  205. }
  206. }
  207. func TestDuplicateNetwork(t *testing.T) {
  208. defer netutils.SetupTestNetNS(t)()
  209. controller := libnetwork.New()
  210. genericOption := make(map[string]interface{})
  211. genericOption[netlabel.GenericData] = options.Generic{}
  212. err := controller.ConfigureNetworkDriver(bridgeNetType, genericOption)
  213. if err != nil {
  214. t.Fatal(err)
  215. }
  216. _, err = controller.NewNetwork(bridgeNetType, "testnetwork",
  217. libnetwork.NetworkOptionGeneric(genericOption))
  218. if err != nil {
  219. t.Fatal(err)
  220. }
  221. _, err = controller.NewNetwork(bridgeNetType, "testnetwork")
  222. if err == nil {
  223. t.Fatal("Expected to fail. But instead succeeded")
  224. }
  225. if _, ok := err.(libnetwork.NetworkNameError); !ok {
  226. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  227. }
  228. }
  229. func TestNetworkName(t *testing.T) {
  230. defer netutils.SetupTestNetNS(t)()
  231. networkName := "testnetwork"
  232. n, err := createTestNetwork(bridgeNetType, networkName, options.Generic{}, options.Generic{})
  233. if err != nil {
  234. t.Fatal(err)
  235. }
  236. if n.Name() != networkName {
  237. t.Fatalf("Expected network name %s, got %s", networkName, n.Name())
  238. }
  239. }
  240. func TestNetworkType(t *testing.T) {
  241. defer netutils.SetupTestNetNS(t)()
  242. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  243. if err != nil {
  244. t.Fatal(err)
  245. }
  246. if n.Type() != bridgeNetType {
  247. t.Fatalf("Expected network type %s, got %s", bridgeNetType, n.Type())
  248. }
  249. }
  250. func TestNetworkID(t *testing.T) {
  251. defer netutils.SetupTestNetNS(t)()
  252. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  253. if err != nil {
  254. t.Fatal(err)
  255. }
  256. if n.ID() == "" {
  257. t.Fatal("Expected non-empty network id")
  258. }
  259. }
  260. func TestDeleteNetworkWithActiveEndpoints(t *testing.T) {
  261. defer netutils.SetupTestNetNS(t)()
  262. option := options.Generic{
  263. "BridgeName": bridgeName,
  264. "AllowNonDefaultBridge": true}
  265. network, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, option)
  266. if err != nil {
  267. t.Fatal(err)
  268. }
  269. ep, err := network.CreateEndpoint("testep")
  270. if err != nil {
  271. t.Fatal(err)
  272. }
  273. err = network.Delete()
  274. if err == nil {
  275. t.Fatal("Expected to fail. But instead succeeded")
  276. }
  277. if _, ok := err.(*libnetwork.ActiveEndpointsError); !ok {
  278. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  279. }
  280. // Done testing. Now cleanup.
  281. if err := ep.Delete(); err != nil {
  282. t.Fatal(err)
  283. }
  284. if err := network.Delete(); err != nil {
  285. t.Fatal(err)
  286. }
  287. }
  288. func TestUnknownNetwork(t *testing.T) {
  289. defer netutils.SetupTestNetNS(t)()
  290. option := options.Generic{
  291. "BridgeName": bridgeName,
  292. "AllowNonDefaultBridge": true}
  293. network, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, option)
  294. if err != nil {
  295. t.Fatal(err)
  296. }
  297. err = network.Delete()
  298. if err != nil {
  299. t.Fatal(err)
  300. }
  301. err = network.Delete()
  302. if err == nil {
  303. t.Fatal("Expected to fail. But instead succeeded")
  304. }
  305. if _, ok := err.(*libnetwork.UnknownNetworkError); !ok {
  306. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  307. }
  308. }
  309. func TestUnknownEndpoint(t *testing.T) {
  310. defer netutils.SetupTestNetNS(t)()
  311. ip, subnet, err := net.ParseCIDR("192.168.100.1/24")
  312. if err != nil {
  313. t.Fatal(err)
  314. }
  315. subnet.IP = ip
  316. option := options.Generic{
  317. "BridgeName": bridgeName,
  318. "AddressIPv4": subnet,
  319. "AllowNonDefaultBridge": true}
  320. network, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, option)
  321. if err != nil {
  322. t.Fatal(err)
  323. }
  324. ep, err := network.CreateEndpoint("testep")
  325. if err != nil {
  326. t.Fatal(err)
  327. }
  328. err = ep.Delete()
  329. if err != nil {
  330. t.Fatal(err)
  331. }
  332. err = ep.Delete()
  333. if err == nil {
  334. t.Fatal("Expected to fail. But instead succeeded")
  335. }
  336. if _, ok := err.(*libnetwork.UnknownEndpointError); !ok {
  337. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  338. }
  339. // Done testing. Now cleanup
  340. if err := network.Delete(); err != nil {
  341. t.Fatal(err)
  342. }
  343. }
  344. func TestNetworkEndpointsWalkers(t *testing.T) {
  345. defer netutils.SetupTestNetNS(t)()
  346. controller := libnetwork.New()
  347. err := controller.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
  348. if err != nil {
  349. t.Fatal(err)
  350. }
  351. // Create network 1 and add 2 endpoint: ep11, ep12
  352. net1, err := controller.NewNetwork(bridgeNetType, "network1")
  353. if err != nil {
  354. t.Fatal(err)
  355. }
  356. ep11, err := net1.CreateEndpoint("ep11")
  357. if err != nil {
  358. t.Fatal(err)
  359. }
  360. ep12, err := net1.CreateEndpoint("ep12")
  361. if err != nil {
  362. t.Fatal(err)
  363. }
  364. // Test list methods on net1
  365. epList1 := net1.Endpoints()
  366. if len(epList1) != 2 {
  367. t.Fatalf("Endpoints() returned wrong number of elements: %d instead of 2", len(epList1))
  368. }
  369. // endpoint order is not guaranteed
  370. for _, e := range epList1 {
  371. if e != ep11 && e != ep12 {
  372. t.Fatal("Endpoints() did not return all the expected elements")
  373. }
  374. }
  375. // Test Endpoint Walk method
  376. var epName string
  377. var epWanted libnetwork.Endpoint
  378. wlk := func(ep libnetwork.Endpoint) bool {
  379. if ep.Name() == epName {
  380. epWanted = ep
  381. return true
  382. }
  383. return false
  384. }
  385. // Look for ep1 on network1
  386. epName = "ep11"
  387. net1.WalkEndpoints(wlk)
  388. if epWanted == nil {
  389. t.Fatal(err)
  390. }
  391. if ep11 != epWanted {
  392. t.Fatal(err)
  393. }
  394. // Test Network Walk method
  395. var netName string
  396. var netWanted libnetwork.Network
  397. nwWlk := func(nw libnetwork.Network) bool {
  398. if nw.Name() == netName {
  399. netWanted = nw
  400. return true
  401. }
  402. return false
  403. }
  404. // Look for network named "network1"
  405. netName = "network1"
  406. controller.WalkNetworks(nwWlk)
  407. if netWanted == nil {
  408. t.Fatal(err)
  409. }
  410. if net1 != netWanted {
  411. t.Fatal(err)
  412. }
  413. }
  414. func TestControllerQuery(t *testing.T) {
  415. defer netutils.SetupTestNetNS(t)()
  416. controller := libnetwork.New()
  417. err := controller.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
  418. if err != nil {
  419. t.Fatal(err)
  420. }
  421. // Create network 1
  422. net1, err := controller.NewNetwork(bridgeNetType, "network1")
  423. if err != nil {
  424. t.Fatal(err)
  425. }
  426. g := controller.NetworkByName("")
  427. if g != nil {
  428. t.Fatalf("NetworkByName() succeeded with invalid target name")
  429. }
  430. g = controller.NetworkByID("")
  431. if g != nil {
  432. t.Fatalf("NetworkByID() succeeded with invalid target id: %v", g)
  433. }
  434. g = controller.NetworkByID("network1")
  435. if g != nil {
  436. t.Fatalf("NetworkByID() succeeded with invalid target name")
  437. }
  438. g = controller.NetworkByName("network1")
  439. if g == nil {
  440. t.Fatalf("NetworkByName() did not find the network")
  441. }
  442. if g != net1 {
  443. t.Fatalf("NetworkByName() returned the wrong network")
  444. }
  445. g = controller.NetworkByID(net1.ID())
  446. if net1 != g {
  447. t.Fatalf("NetworkByID() returned unexpected element: %v", g)
  448. }
  449. }
  450. func TestNetworkQuery(t *testing.T) {
  451. defer netutils.SetupTestNetNS(t)()
  452. controller := libnetwork.New()
  453. err := controller.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
  454. if err != nil {
  455. t.Fatal(err)
  456. }
  457. // Create network 1 and add 2 endpoint: ep11, ep12
  458. net1, err := controller.NewNetwork(bridgeNetType, "network1")
  459. if err != nil {
  460. t.Fatal(err)
  461. }
  462. ep11, err := net1.CreateEndpoint("ep11")
  463. if err != nil {
  464. t.Fatal(err)
  465. }
  466. ep12, err := net1.CreateEndpoint("ep12")
  467. if err != nil {
  468. t.Fatal(err)
  469. }
  470. e := net1.EndpointByName("ep11")
  471. if ep11 != e {
  472. t.Fatalf("EndpointByName() returned %v instead of %v", e, ep11)
  473. }
  474. e = net1.EndpointByName("")
  475. if e != nil {
  476. t.Fatalf("EndpointByName(): expected nil, got %v", e)
  477. }
  478. e = net1.EndpointByName("IamNotAnEndpoint")
  479. if e != nil {
  480. t.Fatalf("EndpointByName(): expected nil, got %v", e)
  481. }
  482. e = net1.EndpointByID(ep12.ID())
  483. if ep12 != e {
  484. t.Fatalf("EndpointByID() returned %v instead of %v", e, ep12)
  485. }
  486. e = net1.EndpointByID("")
  487. if e != nil {
  488. t.Fatalf("EndpointByID(): expected nil, got %v", e)
  489. }
  490. }
  491. const containerID = "valid_container"
  492. func TestEndpointJoin(t *testing.T) {
  493. defer netutils.SetupTestNetNS(t)()
  494. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  495. if err != nil {
  496. t.Fatal(err)
  497. }
  498. ep, err := n.CreateEndpoint("ep1")
  499. if err != nil {
  500. t.Fatal(err)
  501. }
  502. _, err = ep.Join(containerID,
  503. libnetwork.JoinOptionHostname("test"),
  504. libnetwork.JoinOptionDomainname("docker.io"),
  505. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  506. if err != nil {
  507. t.Fatal(err)
  508. }
  509. defer func() {
  510. err = ep.Leave(containerID)
  511. if err != nil {
  512. t.Fatal(err)
  513. }
  514. }()
  515. }
  516. func TestEndpointJoinInvalidContainerId(t *testing.T) {
  517. defer netutils.SetupTestNetNS(t)()
  518. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  519. if err != nil {
  520. t.Fatal(err)
  521. }
  522. ep, err := n.CreateEndpoint("ep1")
  523. if err != nil {
  524. t.Fatal(err)
  525. }
  526. _, err = ep.Join("")
  527. if err == nil {
  528. t.Fatal("Expected to fail join with empty container id string")
  529. }
  530. if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
  531. t.Fatalf("Failed for unexpected reason: %v", err)
  532. }
  533. }
  534. func TestEndpointDeleteWithActiveContainer(t *testing.T) {
  535. defer netutils.SetupTestNetNS(t)()
  536. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  537. if err != nil {
  538. t.Fatal(err)
  539. }
  540. ep, err := n.CreateEndpoint("ep1")
  541. if err != nil {
  542. t.Fatal(err)
  543. }
  544. _, err = ep.Join(containerID,
  545. libnetwork.JoinOptionHostname("test"),
  546. libnetwork.JoinOptionDomainname("docker.io"),
  547. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  548. if err != nil {
  549. t.Fatal(err)
  550. }
  551. defer func() {
  552. err = ep.Leave(containerID)
  553. if err != nil {
  554. t.Fatal(err)
  555. }
  556. err = ep.Delete()
  557. if err != nil {
  558. t.Fatal(err)
  559. }
  560. }()
  561. err = ep.Delete()
  562. if err == nil {
  563. t.Fatal("Expected to fail. But instead succeeded")
  564. }
  565. if _, ok := err.(*libnetwork.ActiveContainerError); !ok {
  566. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  567. }
  568. }
  569. func TestEndpointMultipleJoins(t *testing.T) {
  570. defer netutils.SetupTestNetNS(t)()
  571. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  572. if err != nil {
  573. t.Fatal(err)
  574. }
  575. ep, err := n.CreateEndpoint("ep1")
  576. if err != nil {
  577. t.Fatal(err)
  578. }
  579. _, err = ep.Join(containerID,
  580. libnetwork.JoinOptionHostname("test"),
  581. libnetwork.JoinOptionDomainname("docker.io"),
  582. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  583. if err != nil {
  584. t.Fatal(err)
  585. }
  586. defer func() {
  587. err = ep.Leave(containerID)
  588. if err != nil {
  589. t.Fatal(err)
  590. }
  591. }()
  592. _, err = ep.Join("container2")
  593. if err == nil {
  594. t.Fatal("Expected to fail multiple joins for the same endpoint")
  595. }
  596. if err != libnetwork.ErrInvalidJoin {
  597. t.Fatalf("Failed for unexpected reason: %v", err)
  598. }
  599. }
  600. func TestEndpointInvalidLeave(t *testing.T) {
  601. defer netutils.SetupTestNetNS(t)()
  602. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{}, options.Generic{})
  603. if err != nil {
  604. t.Fatal(err)
  605. }
  606. ep, err := n.CreateEndpoint("ep1")
  607. if err != nil {
  608. t.Fatal(err)
  609. }
  610. err = ep.Leave(containerID)
  611. if err == nil {
  612. t.Fatal("Expected to fail leave from an endpoint which has no active join")
  613. }
  614. if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
  615. if err != libnetwork.ErrNoContainer {
  616. t.Fatalf("Failed for unexpected reason: %v", err)
  617. }
  618. }
  619. _, err = ep.Join(containerID,
  620. libnetwork.JoinOptionHostname("test"),
  621. libnetwork.JoinOptionDomainname("docker.io"),
  622. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  623. if err != nil {
  624. t.Fatal(err)
  625. }
  626. defer func() {
  627. err = ep.Leave(containerID)
  628. if err != nil {
  629. t.Fatal(err)
  630. }
  631. }()
  632. err = ep.Leave("")
  633. if err == nil {
  634. t.Fatal("Expected to fail leave with empty container id")
  635. }
  636. if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
  637. t.Fatalf("Failed for unexpected reason: %v", err)
  638. }
  639. err = ep.Leave("container2")
  640. if err == nil {
  641. t.Fatal("Expected to fail leave with wrong container id")
  642. }
  643. if _, ok := err.(libnetwork.InvalidContainerIDError); !ok {
  644. t.Fatalf("Failed for unexpected reason: %v", err)
  645. }
  646. }
  647. func TestEndpointUpdateParent(t *testing.T) {
  648. defer netutils.SetupTestNetNS(t)()
  649. n, err := createTestNetwork("bridge", "testnetwork", options.Generic{}, options.Generic{})
  650. if err != nil {
  651. t.Fatal(err)
  652. }
  653. ep1, err := n.CreateEndpoint("ep1", nil)
  654. if err != nil {
  655. t.Fatal(err)
  656. }
  657. _, err = ep1.Join(containerID,
  658. libnetwork.JoinOptionHostname("test1"),
  659. libnetwork.JoinOptionDomainname("docker.io"),
  660. libnetwork.JoinOptionExtraHost("web", "192.168.0.1"))
  661. if err != nil {
  662. t.Fatal(err)
  663. }
  664. defer func() {
  665. err = ep1.Leave(containerID)
  666. if err != nil {
  667. t.Fatal(err)
  668. }
  669. }()
  670. ep2, err := n.CreateEndpoint("ep2", nil)
  671. if err != nil {
  672. t.Fatal(err)
  673. }
  674. _, err = ep2.Join("container2",
  675. libnetwork.JoinOptionHostname("test2"),
  676. libnetwork.JoinOptionDomainname("docker.io"),
  677. libnetwork.JoinOptionHostsPath("/var/lib/docker/test_network/container2/hosts"),
  678. libnetwork.JoinOptionParentUpdate(ep1.ID(), "web", "192.168.0.2"))
  679. if err != nil {
  680. t.Fatal(err)
  681. }
  682. defer func() {
  683. err = ep2.Leave("container2")
  684. if err != nil {
  685. t.Fatal(err)
  686. }
  687. }()
  688. }
  689. func TestEnableIPv6(t *testing.T) {
  690. defer netutils.SetupTestNetNS(t)()
  691. tmpResolvConf := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\nnameserver 2001:4860:4860::8888")
  692. //take a copy of resolv.conf for restoring after test completes
  693. resolvConfSystem, err := ioutil.ReadFile("/etc/resolv.conf")
  694. if err != nil {
  695. t.Fatal(err)
  696. }
  697. //cleanup
  698. defer func() {
  699. if err := ioutil.WriteFile("/etc/resolv.conf", resolvConfSystem, 0644); err != nil {
  700. t.Fatal(err)
  701. }
  702. }()
  703. netOption := options.Generic{
  704. netlabel.EnableIPv6: true,
  705. }
  706. n, err := createTestNetwork("bridge", "testnetwork", options.Generic{}, netOption)
  707. if err != nil {
  708. t.Fatal(err)
  709. }
  710. ep1, err := n.CreateEndpoint("ep1", nil)
  711. if err != nil {
  712. t.Fatal(err)
  713. }
  714. if err := ioutil.WriteFile("/etc/resolv.conf", tmpResolvConf, 0644); err != nil {
  715. t.Fatal(err)
  716. }
  717. resolvConfPath := "/tmp/libnetwork_test/resolv.conf"
  718. _, err = ep1.Join(containerID,
  719. libnetwork.JoinOptionResolvConfPath(resolvConfPath))
  720. if err != nil {
  721. t.Fatal(err)
  722. }
  723. defer func() {
  724. err = ep1.Leave(containerID)
  725. if err != nil {
  726. t.Fatal(err)
  727. }
  728. }()
  729. content, err := ioutil.ReadFile(resolvConfPath)
  730. if err != nil {
  731. t.Fatal(err)
  732. }
  733. if !bytes.Equal(content, tmpResolvConf) {
  734. t.Fatalf("Expected %s, Got %s", string(tmpResolvConf), string(content))
  735. }
  736. if err != nil {
  737. t.Fatal(err)
  738. }
  739. }
  740. func TestNoEnableIPv6(t *testing.T) {
  741. defer netutils.SetupTestNetNS(t)()
  742. tmpResolvConf := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\nnameserver 2001:4860:4860::8888")
  743. expectedResolvConf := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\n")
  744. //take a copy of resolv.conf for restoring after test completes
  745. resolvConfSystem, err := ioutil.ReadFile("/etc/resolv.conf")
  746. if err != nil {
  747. t.Fatal(err)
  748. }
  749. //cleanup
  750. defer func() {
  751. if err := ioutil.WriteFile("/etc/resolv.conf", resolvConfSystem, 0644); err != nil {
  752. t.Fatal(err)
  753. }
  754. }()
  755. n, err := createTestNetwork("bridge", "testnetwork", options.Generic{}, options.Generic{})
  756. if err != nil {
  757. t.Fatal(err)
  758. }
  759. ep1, err := n.CreateEndpoint("ep1", nil)
  760. if err != nil {
  761. t.Fatal(err)
  762. }
  763. if err := ioutil.WriteFile("/etc/resolv.conf", tmpResolvConf, 0644); err != nil {
  764. t.Fatal(err)
  765. }
  766. resolvConfPath := "/tmp/libnetwork_test/resolv.conf"
  767. _, err = ep1.Join(containerID,
  768. libnetwork.JoinOptionResolvConfPath(resolvConfPath))
  769. if err != nil {
  770. t.Fatal(err)
  771. }
  772. defer func() {
  773. err = ep1.Leave(containerID)
  774. if err != nil {
  775. t.Fatal(err)
  776. }
  777. }()
  778. content, err := ioutil.ReadFile(resolvConfPath)
  779. if err != nil {
  780. t.Fatal(err)
  781. }
  782. if !bytes.Equal(content, expectedResolvConf) {
  783. t.Fatalf("Expected %s, Got %s", string(expectedResolvConf), string(content))
  784. }
  785. if err != nil {
  786. t.Fatal(err)
  787. }
  788. }
  789. var (
  790. once sync.Once
  791. ctrlr libnetwork.NetworkController
  792. start = make(chan struct{})
  793. done = make(chan chan struct{}, numThreads-1)
  794. origns = netns.None()
  795. testns = netns.None()
  796. )
  797. const (
  798. iterCnt = 25
  799. numThreads = 3
  800. first = 1
  801. last = numThreads
  802. debug = false
  803. )
  804. func createGlobalInstance(t *testing.T) {
  805. var err error
  806. defer close(start)
  807. origns, err = netns.Get()
  808. if err != nil {
  809. t.Fatal(err)
  810. }
  811. //testns = origns
  812. testns, err = netns.New()
  813. if err != nil {
  814. t.Fatal(err)
  815. }
  816. ctrlr = libnetwork.New()
  817. err = ctrlr.ConfigureNetworkDriver(bridgeNetType, getEmptyGenericOption())
  818. if err != nil {
  819. t.Fatal("configure driver")
  820. }
  821. net, err := ctrlr.NewNetwork(bridgeNetType, "network1")
  822. if err != nil {
  823. t.Fatal("new network")
  824. }
  825. _, err = net.CreateEndpoint("ep1")
  826. if err != nil {
  827. t.Fatal("createendpoint")
  828. }
  829. }
  830. func debugf(format string, a ...interface{}) (int, error) {
  831. if debug {
  832. return fmt.Printf(format, a...)
  833. }
  834. return 0, nil
  835. }
  836. func parallelJoin(t *testing.T, ep libnetwork.Endpoint, thrNumber int) {
  837. debugf("J%d.", thrNumber)
  838. _, err := ep.Join("racing_container")
  839. runtime.LockOSThread()
  840. if err != nil {
  841. if err != libnetwork.ErrNoContainer && err != libnetwork.ErrInvalidJoin {
  842. t.Fatal(err)
  843. }
  844. debugf("JE%d(%v).", thrNumber, err)
  845. }
  846. debugf("JD%d.", thrNumber)
  847. }
  848. func parallelLeave(t *testing.T, ep libnetwork.Endpoint, thrNumber int) {
  849. debugf("L%d.", thrNumber)
  850. err := ep.Leave("racing_container")
  851. runtime.LockOSThread()
  852. if err != nil {
  853. if err != libnetwork.ErrNoContainer && err != libnetwork.ErrInvalidJoin {
  854. t.Fatal(err)
  855. }
  856. debugf("LE%d(%v).", thrNumber, err)
  857. }
  858. debugf("LD%d.", thrNumber)
  859. }
  860. func runParallelTests(t *testing.T, thrNumber int) {
  861. var err error
  862. t.Parallel()
  863. pTest := flag.Lookup("test.parallel")
  864. if pTest == nil {
  865. t.Skip("Skipped because test.parallel flag not set;")
  866. }
  867. numParallel, err := strconv.Atoi(pTest.Value.String())
  868. if err != nil {
  869. t.Fatal(err)
  870. }
  871. if numParallel < numThreads {
  872. t.Skip("Skipped because t.parallel was less than ", numThreads)
  873. }
  874. runtime.LockOSThread()
  875. defer runtime.UnlockOSThread()
  876. if thrNumber == first {
  877. createGlobalInstance(t)
  878. }
  879. if thrNumber != first {
  880. select {
  881. case <-start:
  882. }
  883. thrdone := make(chan struct{})
  884. done <- thrdone
  885. defer close(thrdone)
  886. if thrNumber == last {
  887. defer close(done)
  888. }
  889. err = netns.Set(testns)
  890. if err != nil {
  891. t.Fatal(err)
  892. }
  893. }
  894. defer netns.Set(origns)
  895. net := ctrlr.NetworkByName("network1")
  896. if net == nil {
  897. t.Fatal("Could not find network1")
  898. }
  899. ep := net.EndpointByName("ep1")
  900. if ep == nil {
  901. t.Fatal("Could not find ep1")
  902. }
  903. for i := 0; i < iterCnt; i++ {
  904. parallelJoin(t, ep, thrNumber)
  905. parallelLeave(t, ep, thrNumber)
  906. }
  907. debugf("\n")
  908. if thrNumber == first {
  909. for thrdone := range done {
  910. select {
  911. case <-thrdone:
  912. }
  913. }
  914. testns.Close()
  915. err = ep.Delete()
  916. if err != nil {
  917. t.Fatal(err)
  918. }
  919. }
  920. }
  921. func TestParallel1(t *testing.T) {
  922. runParallelTests(t, 1)
  923. }
  924. func TestParallel2(t *testing.T) {
  925. runParallelTests(t, 2)
  926. }
  927. func TestParallel3(t *testing.T) {
  928. runParallelTests(t, 3)
  929. }