libnetwork_test.go 49 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271
  1. package libnetwork_test
  2. import (
  3. "bytes"
  4. "encoding/json"
  5. "flag"
  6. "fmt"
  7. "io/ioutil"
  8. "net/http"
  9. "net/http/httptest"
  10. "os"
  11. "os/exec"
  12. "runtime"
  13. "strconv"
  14. "strings"
  15. "sync"
  16. "testing"
  17. log "github.com/Sirupsen/logrus"
  18. "github.com/docker/docker/pkg/plugins"
  19. "github.com/docker/docker/pkg/reexec"
  20. "github.com/docker/libnetwork"
  21. "github.com/docker/libnetwork/config"
  22. "github.com/docker/libnetwork/datastore"
  23. "github.com/docker/libnetwork/driverapi"
  24. "github.com/docker/libnetwork/ipamapi"
  25. "github.com/docker/libnetwork/netlabel"
  26. "github.com/docker/libnetwork/options"
  27. "github.com/docker/libnetwork/osl"
  28. "github.com/docker/libnetwork/testutils"
  29. "github.com/docker/libnetwork/types"
  30. "github.com/opencontainers/runc/libcontainer"
  31. "github.com/opencontainers/runc/libcontainer/configs"
  32. "github.com/vishvananda/netlink"
  33. "github.com/vishvananda/netns"
  34. )
  35. const (
  36. bridgeNetType = "bridge"
  37. )
  38. var controller libnetwork.NetworkController
  39. func TestMain(m *testing.M) {
  40. if reexec.Init() {
  41. return
  42. }
  43. if err := createController(); err != nil {
  44. log.Errorf("Error creating controller: %v", err)
  45. os.Exit(1)
  46. }
  47. //libnetwork.SetTestDataStore(controller, datastore.NewCustomDataStore(datastore.NewMockStore()))
  48. x := m.Run()
  49. controller.Stop()
  50. os.Exit(x)
  51. }
  52. func createController() error {
  53. var err error
  54. // Cleanup local datastore file
  55. os.Remove(datastore.DefaultScopes("")[datastore.LocalScope].Client.Address)
  56. option := options.Generic{
  57. "EnableIPForwarding": true,
  58. }
  59. genericOption := make(map[string]interface{})
  60. genericOption[netlabel.GenericData] = option
  61. cfgOptions, err := libnetwork.OptionBoltdbWithRandomDBFile()
  62. if err != nil {
  63. return err
  64. }
  65. controller, err = libnetwork.New(append(cfgOptions, config.OptionDriverConfig(bridgeNetType, genericOption))...)
  66. if err != nil {
  67. return err
  68. }
  69. return nil
  70. }
  71. func createTestNetwork(networkType, networkName string, netOption options.Generic, ipamV4Configs, ipamV6Configs []*libnetwork.IpamConf) (libnetwork.Network, error) {
  72. return controller.NewNetwork(networkType, networkName,
  73. libnetwork.NetworkOptionGeneric(netOption),
  74. libnetwork.NetworkOptionIpam(ipamapi.DefaultIPAM, "", ipamV4Configs, ipamV6Configs))
  75. }
  76. func getEmptyGenericOption() map[string]interface{} {
  77. genericOption := make(map[string]interface{})
  78. genericOption[netlabel.GenericData] = options.Generic{}
  79. return genericOption
  80. }
  81. func getPortMapping() []types.PortBinding {
  82. return []types.PortBinding{
  83. {Proto: types.TCP, Port: uint16(230), HostPort: uint16(23000)},
  84. {Proto: types.UDP, Port: uint16(200), HostPort: uint16(22000)},
  85. {Proto: types.TCP, Port: uint16(120), HostPort: uint16(12000)},
  86. {Proto: types.TCP, Port: uint16(320), HostPort: uint16(32000), HostPortEnd: uint16(32999)},
  87. {Proto: types.UDP, Port: uint16(420), HostPort: uint16(42000), HostPortEnd: uint16(42001)},
  88. }
  89. }
  90. func TestNull(t *testing.T) {
  91. cnt, err := controller.NewSandbox("null_container",
  92. libnetwork.OptionHostname("test"),
  93. libnetwork.OptionDomainname("docker.io"),
  94. libnetwork.OptionExtraHost("web", "192.168.0.1"))
  95. if err != nil {
  96. t.Fatal(err)
  97. }
  98. network, err := createTestNetwork("null", "testnull", options.Generic{}, nil, nil)
  99. if err != nil {
  100. t.Fatal(err)
  101. }
  102. ep, err := network.CreateEndpoint("testep")
  103. if err != nil {
  104. t.Fatal(err)
  105. }
  106. err = ep.Join(cnt)
  107. if err != nil {
  108. t.Fatal(err)
  109. }
  110. err = ep.Leave(cnt)
  111. if err != nil {
  112. t.Fatal(err)
  113. }
  114. if err := ep.Delete(); err != nil {
  115. t.Fatal(err)
  116. }
  117. if err := cnt.Delete(); err != nil {
  118. t.Fatal(err)
  119. }
  120. // host type is special network. Cannot be removed.
  121. err = network.Delete()
  122. if err == nil {
  123. t.Fatal(err)
  124. }
  125. if _, ok := err.(types.ForbiddenError); !ok {
  126. t.Fatalf("Unexpected error type")
  127. }
  128. }
  129. func TestHost(t *testing.T) {
  130. sbx1, err := controller.NewSandbox("host_c1",
  131. libnetwork.OptionHostname("test1"),
  132. libnetwork.OptionDomainname("docker.io"),
  133. libnetwork.OptionExtraHost("web", "192.168.0.1"),
  134. libnetwork.OptionUseDefaultSandbox())
  135. if err != nil {
  136. t.Fatal(err)
  137. }
  138. defer func() {
  139. if err := sbx1.Delete(); err != nil {
  140. t.Fatal(err)
  141. }
  142. }()
  143. sbx2, err := controller.NewSandbox("host_c2",
  144. libnetwork.OptionHostname("test2"),
  145. libnetwork.OptionDomainname("docker.io"),
  146. libnetwork.OptionExtraHost("web", "192.168.0.1"),
  147. libnetwork.OptionUseDefaultSandbox())
  148. if err != nil {
  149. t.Fatal(err)
  150. }
  151. defer func() {
  152. if err := sbx2.Delete(); err != nil {
  153. t.Fatal(err)
  154. }
  155. }()
  156. network, err := createTestNetwork("host", "testhost", options.Generic{}, nil, nil)
  157. if err != nil {
  158. t.Fatal(err)
  159. }
  160. ep1, err := network.CreateEndpoint("testep1")
  161. if err != nil {
  162. t.Fatal(err)
  163. }
  164. if err := ep1.Join(sbx1); err != nil {
  165. t.Fatal(err)
  166. }
  167. ep2, err := network.CreateEndpoint("testep2")
  168. if err != nil {
  169. t.Fatal(err)
  170. }
  171. if err := ep2.Join(sbx2); err != nil {
  172. t.Fatal(err)
  173. }
  174. if err := ep1.Leave(sbx1); err != nil {
  175. t.Fatal(err)
  176. }
  177. if err := ep2.Leave(sbx2); err != nil {
  178. t.Fatal(err)
  179. }
  180. if err := ep1.Delete(); err != nil {
  181. t.Fatal(err)
  182. }
  183. if err := ep2.Delete(); err != nil {
  184. t.Fatal(err)
  185. }
  186. // Try to create another host endpoint and join/leave that.
  187. cnt3, err := controller.NewSandbox("host_c3",
  188. libnetwork.OptionHostname("test3"),
  189. libnetwork.OptionDomainname("docker.io"),
  190. libnetwork.OptionExtraHost("web", "192.168.0.1"),
  191. libnetwork.OptionUseDefaultSandbox())
  192. if err != nil {
  193. t.Fatal(err)
  194. }
  195. defer func() {
  196. if err := cnt3.Delete(); err != nil {
  197. t.Fatal(err)
  198. }
  199. }()
  200. ep3, err := network.CreateEndpoint("testep3")
  201. if err != nil {
  202. t.Fatal(err)
  203. }
  204. if err := ep3.Join(sbx2); err != nil {
  205. t.Fatal(err)
  206. }
  207. if err := ep3.Leave(sbx2); err != nil {
  208. t.Fatal(err)
  209. }
  210. if err := ep3.Delete(); err != nil {
  211. t.Fatal(err)
  212. }
  213. // host type is special network. Cannot be removed.
  214. err = network.Delete()
  215. if err == nil {
  216. t.Fatal(err)
  217. }
  218. if _, ok := err.(types.ForbiddenError); !ok {
  219. t.Fatalf("Unexpected error type")
  220. }
  221. }
  222. func TestBridge(t *testing.T) {
  223. if !testutils.IsRunningInContainer() {
  224. defer testutils.SetupTestOSContext(t)()
  225. }
  226. netOption := options.Generic{
  227. netlabel.GenericData: options.Generic{
  228. "BridgeName": "testnetwork",
  229. "EnableIPv6": true,
  230. "EnableICC": true,
  231. "EnableIPMasquerade": true,
  232. },
  233. }
  234. ipamV4ConfList := []*libnetwork.IpamConf{&libnetwork.IpamConf{PreferredPool: "192.168.100.0/24", Gateway: "192.168.100.1"}}
  235. ipamV6ConfList := []*libnetwork.IpamConf{&libnetwork.IpamConf{PreferredPool: "fe90::/98", Gateway: "fe90::22"}}
  236. network, err := createTestNetwork(bridgeNetType, "testnetwork", netOption, ipamV4ConfList, ipamV6ConfList)
  237. if err != nil {
  238. t.Fatal(err)
  239. }
  240. ep, err := network.CreateEndpoint("testep", libnetwork.CreateOptionPortMapping(getPortMapping()))
  241. if err != nil {
  242. t.Fatal(err)
  243. }
  244. epInfo, err := ep.DriverInfo()
  245. if err != nil {
  246. t.Fatal(err)
  247. }
  248. pmd, ok := epInfo[netlabel.PortMap]
  249. if !ok {
  250. t.Fatalf("Could not find expected info in endpoint data")
  251. }
  252. pm, ok := pmd.([]types.PortBinding)
  253. if !ok {
  254. t.Fatalf("Unexpected format for port mapping in endpoint operational data")
  255. }
  256. if len(pm) != 5 {
  257. t.Fatalf("Incomplete data for port mapping in endpoint operational data: %d", len(pm))
  258. }
  259. if err := ep.Delete(); err != nil {
  260. t.Fatal(err)
  261. }
  262. if err := network.Delete(); err != nil {
  263. t.Fatal(err)
  264. }
  265. }
  266. func TestUnknownDriver(t *testing.T) {
  267. if !testutils.IsRunningInContainer() {
  268. defer testutils.SetupTestOSContext(t)()
  269. }
  270. _, err := createTestNetwork("unknowndriver", "testnetwork", options.Generic{}, nil, nil)
  271. if err == nil {
  272. t.Fatal("Expected to fail. But instead succeeded")
  273. }
  274. if _, ok := err.(types.NotFoundError); !ok {
  275. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  276. }
  277. }
  278. func TestNilRemoteDriver(t *testing.T) {
  279. _, err := controller.NewNetwork("framerelay", "dummy",
  280. libnetwork.NetworkOptionGeneric(getEmptyGenericOption()))
  281. if err == nil {
  282. t.Fatal("Expected to fail. But instead succeeded")
  283. }
  284. if _, ok := err.(types.NotFoundError); !ok {
  285. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  286. }
  287. }
  288. func TestNetworkName(t *testing.T) {
  289. if !testutils.IsRunningInContainer() {
  290. defer testutils.SetupTestOSContext(t)()
  291. }
  292. netOption := options.Generic{
  293. netlabel.GenericData: options.Generic{
  294. "BridgeName": "testnetwork",
  295. },
  296. }
  297. _, err := createTestNetwork(bridgeNetType, "", netOption, nil, nil)
  298. if err == nil {
  299. t.Fatal("Expected to fail. But instead succeeded")
  300. }
  301. if _, ok := err.(libnetwork.ErrInvalidName); !ok {
  302. t.Fatalf("Expected to fail with ErrInvalidName error. Got %v", err)
  303. }
  304. networkName := "testnetwork"
  305. n, err := createTestNetwork(bridgeNetType, networkName, netOption, nil, nil)
  306. if err != nil {
  307. t.Fatal(err)
  308. }
  309. defer func() {
  310. if err := n.Delete(); err != nil {
  311. t.Fatal(err)
  312. }
  313. }()
  314. if n.Name() != networkName {
  315. t.Fatalf("Expected network name %s, got %s", networkName, n.Name())
  316. }
  317. }
  318. func TestNetworkType(t *testing.T) {
  319. if !testutils.IsRunningInContainer() {
  320. defer testutils.SetupTestOSContext(t)()
  321. }
  322. netOption := options.Generic{
  323. netlabel.GenericData: options.Generic{
  324. "BridgeName": "testnetwork",
  325. },
  326. }
  327. n, err := createTestNetwork(bridgeNetType, "testnetwork", netOption, nil, nil)
  328. if err != nil {
  329. t.Fatal(err)
  330. }
  331. defer func() {
  332. if err := n.Delete(); err != nil {
  333. t.Fatal(err)
  334. }
  335. }()
  336. if n.Type() != bridgeNetType {
  337. t.Fatalf("Expected network type %s, got %s", bridgeNetType, n.Type())
  338. }
  339. }
  340. func TestNetworkID(t *testing.T) {
  341. if !testutils.IsRunningInContainer() {
  342. defer testutils.SetupTestOSContext(t)()
  343. }
  344. netOption := options.Generic{
  345. netlabel.GenericData: options.Generic{
  346. "BridgeName": "testnetwork",
  347. },
  348. }
  349. n, err := createTestNetwork(bridgeNetType, "testnetwork", netOption, nil, nil)
  350. if err != nil {
  351. t.Fatal(err)
  352. }
  353. defer func() {
  354. if err := n.Delete(); err != nil {
  355. t.Fatal(err)
  356. }
  357. }()
  358. if n.ID() == "" {
  359. t.Fatal("Expected non-empty network id")
  360. }
  361. }
  362. func TestDeleteNetworkWithActiveEndpoints(t *testing.T) {
  363. if !testutils.IsRunningInContainer() {
  364. defer testutils.SetupTestOSContext(t)()
  365. }
  366. netOption := options.Generic{
  367. "BridgeName": "testnetwork",
  368. }
  369. option := options.Generic{
  370. netlabel.GenericData: netOption,
  371. }
  372. network, err := createTestNetwork(bridgeNetType, "testnetwork", option, nil, nil)
  373. if err != nil {
  374. t.Fatal(err)
  375. }
  376. ep, err := network.CreateEndpoint("testep")
  377. if err != nil {
  378. t.Fatal(err)
  379. }
  380. err = network.Delete()
  381. if err == nil {
  382. t.Fatal("Expected to fail. But instead succeeded")
  383. }
  384. if _, ok := err.(*libnetwork.ActiveEndpointsError); !ok {
  385. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  386. }
  387. // Done testing. Now cleanup.
  388. if err := ep.Delete(); err != nil {
  389. t.Fatal(err)
  390. }
  391. if err := network.Delete(); err != nil {
  392. t.Fatal(err)
  393. }
  394. }
  395. func TestUnknownNetwork(t *testing.T) {
  396. if !testutils.IsRunningInContainer() {
  397. defer testutils.SetupTestOSContext(t)()
  398. }
  399. netOption := options.Generic{
  400. "BridgeName": "testnetwork",
  401. }
  402. option := options.Generic{
  403. netlabel.GenericData: netOption,
  404. }
  405. network, err := createTestNetwork(bridgeNetType, "testnetwork", option, nil, nil)
  406. if err != nil {
  407. t.Fatal(err)
  408. }
  409. err = network.Delete()
  410. if err != nil {
  411. t.Fatal(err)
  412. }
  413. err = network.Delete()
  414. if err == nil {
  415. t.Fatal("Expected to fail. But instead succeeded")
  416. }
  417. if _, ok := err.(*libnetwork.UnknownNetworkError); !ok {
  418. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  419. }
  420. }
  421. func TestUnknownEndpoint(t *testing.T) {
  422. if !testutils.IsRunningInContainer() {
  423. defer testutils.SetupTestOSContext(t)()
  424. }
  425. netOption := options.Generic{
  426. "BridgeName": "testnetwork",
  427. }
  428. option := options.Generic{
  429. netlabel.GenericData: netOption,
  430. }
  431. ipamV4ConfList := []*libnetwork.IpamConf{&libnetwork.IpamConf{PreferredPool: "192.168.100.0/24"}}
  432. network, err := createTestNetwork(bridgeNetType, "testnetwork", option, ipamV4ConfList, nil)
  433. if err != nil {
  434. t.Fatal(err)
  435. }
  436. _, err = network.CreateEndpoint("")
  437. if err == nil {
  438. t.Fatal("Expected to fail. But instead succeeded")
  439. }
  440. if _, ok := err.(libnetwork.ErrInvalidName); !ok {
  441. t.Fatalf("Expected to fail with ErrInvalidName error. Actual error: %v", err)
  442. }
  443. ep, err := network.CreateEndpoint("testep")
  444. if err != nil {
  445. t.Fatal(err)
  446. }
  447. err = ep.Delete()
  448. if err != nil {
  449. t.Fatal(err)
  450. }
  451. // Done testing. Now cleanup
  452. if err := network.Delete(); err != nil {
  453. t.Fatal(err)
  454. }
  455. }
  456. func TestNetworkEndpointsWalkers(t *testing.T) {
  457. if !testutils.IsRunningInContainer() {
  458. defer testutils.SetupTestOSContext(t)()
  459. }
  460. // Create network 1 and add 2 endpoint: ep11, ep12
  461. netOption := options.Generic{
  462. netlabel.GenericData: options.Generic{
  463. "BridgeName": "network1",
  464. },
  465. }
  466. net1, err := createTestNetwork(bridgeNetType, "network1", netOption, nil, nil)
  467. if err != nil {
  468. t.Fatal(err)
  469. }
  470. defer func() {
  471. if err := net1.Delete(); err != nil {
  472. t.Fatal(err)
  473. }
  474. }()
  475. ep11, err := net1.CreateEndpoint("ep11")
  476. if err != nil {
  477. t.Fatal(err)
  478. }
  479. defer func() {
  480. if err := ep11.Delete(); err != nil {
  481. t.Fatal(err)
  482. }
  483. }()
  484. ep12, err := net1.CreateEndpoint("ep12")
  485. if err != nil {
  486. t.Fatal(err)
  487. }
  488. defer func() {
  489. if err := ep12.Delete(); err != nil {
  490. t.Fatal(err)
  491. }
  492. }()
  493. // Test list methods on net1
  494. epList1 := net1.Endpoints()
  495. if len(epList1) != 2 {
  496. t.Fatalf("Endpoints() returned wrong number of elements: %d instead of 2", len(epList1))
  497. }
  498. // endpoint order is not guaranteed
  499. for _, e := range epList1 {
  500. if e != ep11 && e != ep12 {
  501. t.Fatal("Endpoints() did not return all the expected elements")
  502. }
  503. }
  504. // Test Endpoint Walk method
  505. var epName string
  506. var epWanted libnetwork.Endpoint
  507. wlk := func(ep libnetwork.Endpoint) bool {
  508. if ep.Name() == epName {
  509. epWanted = ep
  510. return true
  511. }
  512. return false
  513. }
  514. // Look for ep1 on network1
  515. epName = "ep11"
  516. net1.WalkEndpoints(wlk)
  517. if epWanted == nil {
  518. t.Fatal(err)
  519. }
  520. if ep11 != epWanted {
  521. t.Fatal(err)
  522. }
  523. current := len(controller.Networks())
  524. // Create network 2
  525. netOption = options.Generic{
  526. netlabel.GenericData: options.Generic{
  527. "BridgeName": "network2",
  528. },
  529. }
  530. net2, err := createTestNetwork(bridgeNetType, "network2", netOption, nil, nil)
  531. if err != nil {
  532. t.Fatal(err)
  533. }
  534. defer func() {
  535. if err := net2.Delete(); err != nil {
  536. t.Fatal(err)
  537. }
  538. }()
  539. // Test Networks method
  540. if len(controller.Networks()) != current+1 {
  541. t.Fatalf("Did not find the expected number of networks")
  542. }
  543. // Test Network Walk method
  544. var netName string
  545. var netWanted libnetwork.Network
  546. nwWlk := func(nw libnetwork.Network) bool {
  547. if nw.Name() == netName {
  548. netWanted = nw
  549. return true
  550. }
  551. return false
  552. }
  553. // Look for network named "network1" and "network2"
  554. netName = "network1"
  555. controller.WalkNetworks(nwWlk)
  556. if netWanted == nil {
  557. t.Fatal(err)
  558. }
  559. if net1.ID() != netWanted.ID() {
  560. t.Fatal(err)
  561. }
  562. netName = "network2"
  563. controller.WalkNetworks(nwWlk)
  564. if netWanted == nil {
  565. t.Fatal(err)
  566. }
  567. if net2.ID() != netWanted.ID() {
  568. t.Fatal(err)
  569. }
  570. }
  571. func TestDuplicateEndpoint(t *testing.T) {
  572. if !testutils.IsRunningInContainer() {
  573. defer testutils.SetupTestOSContext(t)()
  574. }
  575. netOption := options.Generic{
  576. netlabel.GenericData: options.Generic{
  577. "BridgeName": "testnetwork",
  578. },
  579. }
  580. n, err := createTestNetwork(bridgeNetType, "testnetwork", netOption, nil, nil)
  581. if err != nil {
  582. t.Fatal(err)
  583. }
  584. defer func() {
  585. if err := n.Delete(); err != nil {
  586. t.Fatal(err)
  587. }
  588. }()
  589. ep, err := n.CreateEndpoint("ep1")
  590. if err != nil {
  591. t.Fatal(err)
  592. }
  593. defer func() {
  594. if err := ep.Delete(); err != nil {
  595. t.Fatal(err)
  596. }
  597. }()
  598. ep2, err := n.CreateEndpoint("ep1")
  599. defer func() {
  600. // Cleanup ep2 as well, else network cleanup might fail for failure cases
  601. if ep2 != nil {
  602. if err := ep2.Delete(); err != nil {
  603. t.Fatal(err)
  604. }
  605. }
  606. }()
  607. if err == nil {
  608. t.Fatal("Expected to fail. But instead succeeded")
  609. }
  610. if _, ok := err.(types.ForbiddenError); !ok {
  611. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  612. }
  613. }
  614. func TestControllerQuery(t *testing.T) {
  615. if !testutils.IsRunningInContainer() {
  616. defer testutils.SetupTestOSContext(t)()
  617. }
  618. // Create network 1
  619. netOption := options.Generic{
  620. netlabel.GenericData: options.Generic{
  621. "BridgeName": "network1",
  622. },
  623. }
  624. net1, err := createTestNetwork(bridgeNetType, "network1", netOption, nil, nil)
  625. if err != nil {
  626. t.Fatal(err)
  627. }
  628. defer func() {
  629. if err := net1.Delete(); err != nil {
  630. t.Fatal(err)
  631. }
  632. }()
  633. // Create network 2
  634. netOption = options.Generic{
  635. netlabel.GenericData: options.Generic{
  636. "BridgeName": "network2",
  637. },
  638. }
  639. net2, err := createTestNetwork(bridgeNetType, "network2", netOption, nil, nil)
  640. if err != nil {
  641. t.Fatal(err)
  642. }
  643. defer func() {
  644. if err := net2.Delete(); err != nil {
  645. t.Fatal(err)
  646. }
  647. }()
  648. _, err = controller.NetworkByName("")
  649. if err == nil {
  650. t.Fatalf("NetworkByName() succeeded with invalid target name")
  651. }
  652. if _, ok := err.(libnetwork.ErrInvalidName); !ok {
  653. t.Fatalf("Expected NetworkByName() to fail with ErrInvalidName error. Got: %v", err)
  654. }
  655. _, err = controller.NetworkByID("")
  656. if err == nil {
  657. t.Fatalf("NetworkByID() succeeded with invalid target id")
  658. }
  659. if _, ok := err.(libnetwork.ErrInvalidID); !ok {
  660. t.Fatalf("NetworkByID() failed with unexpected error: %v", err)
  661. }
  662. g, err := controller.NetworkByID("network1")
  663. if err == nil {
  664. t.Fatalf("Unexpected success for NetworkByID(): %v", g)
  665. }
  666. if _, ok := err.(libnetwork.ErrNoSuchNetwork); !ok {
  667. t.Fatalf("NetworkByID() failed with unexpected error: %v", err)
  668. }
  669. g, err = controller.NetworkByName("network1")
  670. if err != nil {
  671. t.Fatalf("Unexpected failure for NetworkByName(): %v", err)
  672. }
  673. if g == nil {
  674. t.Fatalf("NetworkByName() did not find the network")
  675. }
  676. if g != net1 {
  677. t.Fatalf("NetworkByName() returned the wrong network")
  678. }
  679. g, err = controller.NetworkByID(net1.ID())
  680. if err != nil {
  681. t.Fatalf("Unexpected failure for NetworkByID(): %v", err)
  682. }
  683. if net1.ID() != g.ID() {
  684. t.Fatalf("NetworkByID() returned unexpected element: %v", g)
  685. }
  686. g, err = controller.NetworkByName("network2")
  687. if err != nil {
  688. t.Fatalf("Unexpected failure for NetworkByName(): %v", err)
  689. }
  690. if g == nil {
  691. t.Fatalf("NetworkByName() did not find the network")
  692. }
  693. if g != net2 {
  694. t.Fatalf("NetworkByName() returned the wrong network")
  695. }
  696. g, err = controller.NetworkByID(net2.ID())
  697. if err != nil {
  698. t.Fatalf("Unexpected failure for NetworkByID(): %v", err)
  699. }
  700. if net2.ID() != g.ID() {
  701. t.Fatalf("NetworkByID() returned unexpected element: %v", g)
  702. }
  703. }
  704. func TestNetworkQuery(t *testing.T) {
  705. if !testutils.IsRunningInContainer() {
  706. defer testutils.SetupTestOSContext(t)()
  707. }
  708. // Create network 1 and add 2 endpoint: ep11, ep12
  709. netOption := options.Generic{
  710. netlabel.GenericData: options.Generic{
  711. "BridgeName": "network1",
  712. },
  713. }
  714. net1, err := createTestNetwork(bridgeNetType, "network1", netOption, nil, nil)
  715. if err != nil {
  716. t.Fatal(err)
  717. }
  718. defer func() {
  719. if err := net1.Delete(); err != nil {
  720. t.Fatal(err)
  721. }
  722. }()
  723. ep11, err := net1.CreateEndpoint("ep11")
  724. if err != nil {
  725. t.Fatal(err)
  726. }
  727. defer func() {
  728. if err := ep11.Delete(); err != nil {
  729. t.Fatal(err)
  730. }
  731. }()
  732. ep12, err := net1.CreateEndpoint("ep12")
  733. if err != nil {
  734. t.Fatal(err)
  735. }
  736. defer func() {
  737. if err := ep12.Delete(); err != nil {
  738. t.Fatal(err)
  739. }
  740. }()
  741. e, err := net1.EndpointByName("ep11")
  742. if err != nil {
  743. t.Fatal(err)
  744. }
  745. if ep11 != e {
  746. t.Fatalf("EndpointByName() returned %v instead of %v", e, ep11)
  747. }
  748. e, err = net1.EndpointByName("")
  749. if err == nil {
  750. t.Fatalf("EndpointByName() succeeded with invalid target name")
  751. }
  752. if _, ok := err.(libnetwork.ErrInvalidName); !ok {
  753. t.Fatalf("Expected EndpointByName() to fail with ErrInvalidName error. Got: %v", err)
  754. }
  755. e, err = net1.EndpointByName("IamNotAnEndpoint")
  756. if err == nil {
  757. t.Fatalf("EndpointByName() succeeded with unknown target name")
  758. }
  759. if _, ok := err.(libnetwork.ErrNoSuchEndpoint); !ok {
  760. t.Fatal(err)
  761. }
  762. if e != nil {
  763. t.Fatalf("EndpointByName(): expected nil, got %v", e)
  764. }
  765. e, err = net1.EndpointByID(ep12.ID())
  766. if err != nil {
  767. t.Fatal(err)
  768. }
  769. if ep12.ID() != e.ID() {
  770. t.Fatalf("EndpointByID() returned %v instead of %v", e, ep12)
  771. }
  772. e, err = net1.EndpointByID("")
  773. if err == nil {
  774. t.Fatalf("EndpointByID() succeeded with invalid target id")
  775. }
  776. if _, ok := err.(libnetwork.ErrInvalidID); !ok {
  777. t.Fatalf("EndpointByID() failed with unexpected error: %v", err)
  778. }
  779. }
  780. const containerID = "valid_c"
  781. func checkSandbox(t *testing.T, info libnetwork.EndpointInfo) {
  782. origns, err := netns.Get()
  783. if err != nil {
  784. t.Fatalf("Could not get the current netns: %v", err)
  785. }
  786. defer origns.Close()
  787. key := info.Sandbox().Key()
  788. f, err := os.OpenFile(key, os.O_RDONLY, 0)
  789. if err != nil {
  790. t.Fatalf("Failed to open network namespace path %q: %v", key, err)
  791. }
  792. defer f.Close()
  793. runtime.LockOSThread()
  794. defer runtime.UnlockOSThread()
  795. nsFD := f.Fd()
  796. if err = netns.Set(netns.NsHandle(nsFD)); err != nil {
  797. t.Fatalf("Setting to the namespace pointed to by the sandbox %s failed: %v", key, err)
  798. }
  799. defer netns.Set(origns)
  800. _, err = netlink.LinkByName("eth0")
  801. if err != nil {
  802. t.Fatalf("Could not find the interface eth0 inside the sandbox: %v", err)
  803. }
  804. _, err = netlink.LinkByName("eth1")
  805. if err != nil {
  806. t.Fatalf("Could not find the interface eth1 inside the sandbox: %v", err)
  807. }
  808. }
  809. func TestEndpointJoin(t *testing.T) {
  810. if !testutils.IsRunningInContainer() {
  811. defer testutils.SetupTestOSContext(t)()
  812. }
  813. // Create network 1 and add 2 endpoint: ep11, ep12
  814. n1, err := createTestNetwork(bridgeNetType, "testnetwork1", options.Generic{
  815. netlabel.GenericData: options.Generic{
  816. "BridgeName": "testnetwork1",
  817. },
  818. }, nil, nil)
  819. if err != nil {
  820. t.Fatal(err)
  821. }
  822. defer func() {
  823. if err := n1.Delete(); err != nil {
  824. t.Fatal(err)
  825. }
  826. }()
  827. ep1, err := n1.CreateEndpoint("ep1")
  828. if err != nil {
  829. t.Fatal(err)
  830. }
  831. defer func() {
  832. if err := ep1.Delete(); err != nil {
  833. t.Fatal(err)
  834. }
  835. }()
  836. // Validate if ep.Info() only gives me IP address info and not names and gateway during CreateEndpoint()
  837. info := ep1.Info()
  838. iface := info.Iface()
  839. if iface.Address() != nil && iface.Address().IP.To4() == nil {
  840. t.Fatalf("Invalid IP address returned: %v", iface.Address())
  841. }
  842. if info.Gateway().To4() != nil {
  843. t.Fatalf("Expected empty gateway for an empty endpoint. Instead found a gateway: %v", info.Gateway())
  844. }
  845. if info.Sandbox() != nil {
  846. t.Fatalf("Expected an empty sandbox key for an empty endpoint. Instead found a non-empty sandbox key: %s", info.Sandbox().Key())
  847. }
  848. // test invalid joins
  849. err = ep1.Join(nil)
  850. if err == nil {
  851. t.Fatalf("Expected to fail join with nil Sandbox")
  852. }
  853. if _, ok := err.(types.BadRequestError); !ok {
  854. t.Fatalf("Unexpected error type returned: %T", err)
  855. }
  856. fsbx := &fakeSandbox{}
  857. if err = ep1.Join(fsbx); err == nil {
  858. t.Fatalf("Expected to fail join with invalid Sandbox")
  859. }
  860. if _, ok := err.(types.BadRequestError); !ok {
  861. t.Fatalf("Unexpected error type returned: %T", err)
  862. }
  863. sb, err := controller.NewSandbox(containerID,
  864. libnetwork.OptionHostname("test"),
  865. libnetwork.OptionDomainname("docker.io"),
  866. libnetwork.OptionExtraHost("web", "192.168.0.1"))
  867. if err != nil {
  868. t.Fatal(err)
  869. }
  870. defer func() {
  871. if err := sb.Delete(); err != nil {
  872. t.Fatal(err)
  873. }
  874. }()
  875. err = ep1.Join(sb)
  876. runtime.LockOSThread()
  877. if err != nil {
  878. t.Fatal(err)
  879. }
  880. defer func() {
  881. err = ep1.Leave(sb)
  882. runtime.LockOSThread()
  883. if err != nil {
  884. t.Fatal(err)
  885. }
  886. }()
  887. // Validate if ep.Info() only gives valid gateway and sandbox key after has container has joined.
  888. info = ep1.Info()
  889. if info.Gateway().To4() == nil {
  890. t.Fatalf("Expected a valid gateway for a joined endpoint. Instead found an invalid gateway: %v", info.Gateway())
  891. }
  892. if info.Sandbox() == nil {
  893. t.Fatalf("Expected an non-empty sandbox key for a joined endpoint. Instead found a empty sandbox key")
  894. }
  895. // Check endpoint provided container information
  896. if ep1.Info().Sandbox().Key() != sb.Key() {
  897. t.Fatalf("Endpoint Info returned unexpected sandbox key: %s", sb.Key())
  898. }
  899. // Attempt retrieval of endpoint interfaces statistics
  900. stats, err := sb.Statistics()
  901. if err != nil {
  902. t.Fatal(err)
  903. }
  904. if _, ok := stats["eth0"]; !ok {
  905. t.Fatalf("Did not find eth0 statistics")
  906. }
  907. // Now test the container joining another network
  908. n2, err := createTestNetwork(bridgeNetType, "testnetwork2",
  909. options.Generic{
  910. netlabel.GenericData: options.Generic{
  911. "BridgeName": "testnetwork2",
  912. },
  913. }, nil, nil)
  914. if err != nil {
  915. t.Fatal(err)
  916. }
  917. defer func() {
  918. if err := n2.Delete(); err != nil {
  919. t.Fatal(err)
  920. }
  921. }()
  922. ep2, err := n2.CreateEndpoint("ep2")
  923. if err != nil {
  924. t.Fatal(err)
  925. }
  926. defer func() {
  927. if err := ep2.Delete(); err != nil {
  928. t.Fatal(err)
  929. }
  930. }()
  931. err = ep2.Join(sb)
  932. if err != nil {
  933. t.Fatal(err)
  934. }
  935. runtime.LockOSThread()
  936. defer func() {
  937. err = ep2.Leave(sb)
  938. runtime.LockOSThread()
  939. if err != nil {
  940. t.Fatal(err)
  941. }
  942. }()
  943. if ep1.Info().Sandbox().Key() != ep2.Info().Sandbox().Key() {
  944. t.Fatalf("ep1 and ep2 returned different container sandbox key")
  945. }
  946. checkSandbox(t, info)
  947. }
  948. type fakeSandbox struct{}
  949. func (f *fakeSandbox) ID() string {
  950. return "fake sandbox"
  951. }
  952. func (f *fakeSandbox) ContainerID() string {
  953. return ""
  954. }
  955. func (f *fakeSandbox) Key() string {
  956. return "fake key"
  957. }
  958. func (f *fakeSandbox) Labels() map[string]interface{} {
  959. return nil
  960. }
  961. func (f *fakeSandbox) Statistics() (map[string]*types.InterfaceStatistics, error) {
  962. return nil, nil
  963. }
  964. func (f *fakeSandbox) Refresh(opts ...libnetwork.SandboxOption) error {
  965. return nil
  966. }
  967. func (f *fakeSandbox) Delete() error {
  968. return nil
  969. }
  970. func (f *fakeSandbox) SetKey(key string) error {
  971. return nil
  972. }
  973. func TestExternalKey(t *testing.T) {
  974. externalKeyTest(t, false)
  975. }
  976. func TestExternalKeyWithReexec(t *testing.T) {
  977. externalKeyTest(t, true)
  978. }
  979. func externalKeyTest(t *testing.T, reexec bool) {
  980. if !testutils.IsRunningInContainer() {
  981. defer testutils.SetupTestOSContext(t)()
  982. }
  983. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{
  984. netlabel.GenericData: options.Generic{
  985. "BridgeName": "testnetwork",
  986. },
  987. }, nil, nil)
  988. if err != nil {
  989. t.Fatal(err)
  990. }
  991. defer func() {
  992. if err := n.Delete(); err != nil {
  993. t.Fatal(err)
  994. }
  995. }()
  996. ep, err := n.CreateEndpoint("ep1")
  997. if err != nil {
  998. t.Fatal(err)
  999. }
  1000. defer func() {
  1001. err = ep.Delete()
  1002. if err != nil {
  1003. t.Fatal(err)
  1004. }
  1005. }()
  1006. ep2, err := n.CreateEndpoint("ep2")
  1007. if err != nil {
  1008. t.Fatal(err)
  1009. }
  1010. defer func() {
  1011. err = ep2.Delete()
  1012. if err != nil {
  1013. t.Fatal(err)
  1014. }
  1015. }()
  1016. cnt, err := controller.NewSandbox(containerID,
  1017. libnetwork.OptionHostname("test"),
  1018. libnetwork.OptionDomainname("docker.io"),
  1019. libnetwork.OptionUseExternalKey(),
  1020. libnetwork.OptionExtraHost("web", "192.168.0.1"))
  1021. defer func() {
  1022. if err := cnt.Delete(); err != nil {
  1023. t.Fatal(err)
  1024. }
  1025. osl.GC()
  1026. }()
  1027. // Join endpoint to sandbox before SetKey
  1028. err = ep.Join(cnt)
  1029. runtime.LockOSThread()
  1030. if err != nil {
  1031. t.Fatal(err)
  1032. }
  1033. defer func() {
  1034. err = ep.Leave(cnt)
  1035. runtime.LockOSThread()
  1036. if err != nil {
  1037. t.Fatal(err)
  1038. }
  1039. }()
  1040. sbox := ep.Info().Sandbox()
  1041. if sbox == nil {
  1042. t.Fatalf("Expected to have a valid Sandbox")
  1043. }
  1044. if reexec {
  1045. err := reexecSetKey("this-must-fail", containerID, controller.ID())
  1046. if err == nil {
  1047. t.Fatalf("SetExternalKey must fail if the corresponding namespace is not created")
  1048. }
  1049. } else {
  1050. // Setting an non-existing key (namespace) must fail
  1051. if err := sbox.SetKey("this-must-fail"); err == nil {
  1052. t.Fatalf("Setkey must fail if the corresponding namespace is not created")
  1053. }
  1054. }
  1055. // Create a new OS sandbox using the osl API before using it in SetKey
  1056. if extOsBox, err := osl.NewSandbox("ValidKey", true); err != nil {
  1057. t.Fatalf("Failed to create new osl sandbox")
  1058. } else {
  1059. defer func() {
  1060. if err := extOsBox.Destroy(); err != nil {
  1061. log.Warnf("Failed to remove os sandbox: %v", err)
  1062. }
  1063. }()
  1064. }
  1065. if reexec {
  1066. err := reexecSetKey("ValidKey", containerID, controller.ID())
  1067. if err != nil {
  1068. t.Fatalf("SetExternalKey failed with %v", err)
  1069. }
  1070. } else {
  1071. if err := sbox.SetKey("ValidKey"); err != nil {
  1072. t.Fatalf("Setkey failed with %v", err)
  1073. }
  1074. }
  1075. // Join endpoint to sandbox after SetKey
  1076. err = ep2.Join(sbox)
  1077. if err != nil {
  1078. t.Fatal(err)
  1079. }
  1080. runtime.LockOSThread()
  1081. defer func() {
  1082. err = ep2.Leave(sbox)
  1083. runtime.LockOSThread()
  1084. if err != nil {
  1085. t.Fatal(err)
  1086. }
  1087. }()
  1088. if ep.Info().Sandbox().Key() != ep2.Info().Sandbox().Key() {
  1089. t.Fatalf("ep1 and ep2 returned different container sandbox key")
  1090. }
  1091. checkSandbox(t, ep.Info())
  1092. }
  1093. func reexecSetKey(key string, containerID string, controllerID string) error {
  1094. var (
  1095. state libcontainer.State
  1096. b []byte
  1097. err error
  1098. )
  1099. state.NamespacePaths = make(map[configs.NamespaceType]string)
  1100. state.NamespacePaths[configs.NamespaceType("NEWNET")] = key
  1101. if b, err = json.Marshal(state); err != nil {
  1102. return err
  1103. }
  1104. cmd := &exec.Cmd{
  1105. Path: reexec.Self(),
  1106. Args: append([]string{"libnetwork-setkey"}, containerID, controllerID),
  1107. Stdin: strings.NewReader(string(b)),
  1108. Stdout: os.Stdout,
  1109. Stderr: os.Stderr,
  1110. }
  1111. return cmd.Run()
  1112. }
  1113. func TestEndpointDeleteWithActiveContainer(t *testing.T) {
  1114. if !testutils.IsRunningInContainer() {
  1115. defer testutils.SetupTestOSContext(t)()
  1116. }
  1117. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{
  1118. netlabel.GenericData: options.Generic{
  1119. "BridgeName": "testnetwork",
  1120. },
  1121. }, nil, nil)
  1122. if err != nil {
  1123. t.Fatal(err)
  1124. }
  1125. defer func() {
  1126. if err := n.Delete(); err != nil {
  1127. t.Fatal(err)
  1128. }
  1129. }()
  1130. ep, err := n.CreateEndpoint("ep1")
  1131. if err != nil {
  1132. t.Fatal(err)
  1133. }
  1134. defer func() {
  1135. err = ep.Delete()
  1136. if err != nil {
  1137. t.Fatal(err)
  1138. }
  1139. }()
  1140. cnt, err := controller.NewSandbox(containerID,
  1141. libnetwork.OptionHostname("test"),
  1142. libnetwork.OptionDomainname("docker.io"),
  1143. libnetwork.OptionExtraHost("web", "192.168.0.1"))
  1144. defer func() {
  1145. if err := cnt.Delete(); err != nil {
  1146. t.Fatal(err)
  1147. }
  1148. }()
  1149. err = ep.Join(cnt)
  1150. runtime.LockOSThread()
  1151. if err != nil {
  1152. t.Fatal(err)
  1153. }
  1154. defer func() {
  1155. err = ep.Leave(cnt)
  1156. runtime.LockOSThread()
  1157. if err != nil {
  1158. t.Fatal(err)
  1159. }
  1160. }()
  1161. err = ep.Delete()
  1162. if err == nil {
  1163. t.Fatal("Expected to fail. But instead succeeded")
  1164. }
  1165. if _, ok := err.(*libnetwork.ActiveContainerError); !ok {
  1166. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  1167. }
  1168. }
  1169. func TestEndpointMultipleJoins(t *testing.T) {
  1170. if !testutils.IsRunningInContainer() {
  1171. defer testutils.SetupTestOSContext(t)()
  1172. }
  1173. n, err := createTestNetwork(bridgeNetType, "testmultiple", options.Generic{
  1174. netlabel.GenericData: options.Generic{
  1175. "BridgeName": "testmultiple",
  1176. },
  1177. }, nil, nil)
  1178. if err != nil {
  1179. t.Fatal(err)
  1180. }
  1181. defer func() {
  1182. if err := n.Delete(); err != nil {
  1183. t.Fatal(err)
  1184. }
  1185. }()
  1186. ep, err := n.CreateEndpoint("ep1")
  1187. if err != nil {
  1188. t.Fatal(err)
  1189. }
  1190. defer func() {
  1191. if err := ep.Delete(); err != nil {
  1192. t.Fatal(err)
  1193. }
  1194. }()
  1195. sbx1, err := controller.NewSandbox(containerID,
  1196. libnetwork.OptionHostname("test"),
  1197. libnetwork.OptionDomainname("docker.io"),
  1198. libnetwork.OptionExtraHost("web", "192.168.0.1"))
  1199. defer func() {
  1200. if err := sbx1.Delete(); err != nil {
  1201. t.Fatal(err)
  1202. }
  1203. }()
  1204. sbx2, err := controller.NewSandbox("c2")
  1205. defer func() {
  1206. if err := sbx2.Delete(); err != nil {
  1207. t.Fatal(err)
  1208. }
  1209. runtime.LockOSThread()
  1210. }()
  1211. err = ep.Join(sbx1)
  1212. runtime.LockOSThread()
  1213. if err != nil {
  1214. t.Fatal(err)
  1215. }
  1216. defer func() {
  1217. err = ep.Leave(sbx1)
  1218. runtime.LockOSThread()
  1219. if err != nil {
  1220. t.Fatal(err)
  1221. }
  1222. }()
  1223. err = ep.Join(sbx2)
  1224. if err == nil {
  1225. t.Fatal("Expected to fail multiple joins for the same endpoint")
  1226. }
  1227. if _, ok := err.(types.ForbiddenError); !ok {
  1228. t.Fatalf("Failed with unexpected error type: %T. Desc: %s", err, err.Error())
  1229. }
  1230. }
  1231. func TestLeaveAll(t *testing.T) {
  1232. if !testutils.IsRunningInContainer() {
  1233. defer testutils.SetupTestOSContext(t)()
  1234. }
  1235. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{
  1236. netlabel.GenericData: options.Generic{
  1237. "BridgeName": "testnetwork",
  1238. },
  1239. }, nil, nil)
  1240. if err != nil {
  1241. t.Fatal(err)
  1242. }
  1243. defer func() {
  1244. // If this goes through, it means cnt.Delete() effectively detached from all the endpoints
  1245. if err := n.Delete(); err != nil {
  1246. t.Fatal(err)
  1247. }
  1248. }()
  1249. ep1, err := n.CreateEndpoint("ep1")
  1250. if err != nil {
  1251. t.Fatal(err)
  1252. }
  1253. ep2, err := n.CreateEndpoint("ep2")
  1254. if err != nil {
  1255. t.Fatal(err)
  1256. }
  1257. cnt, err := controller.NewSandbox("leaveall")
  1258. if err != nil {
  1259. t.Fatal(err)
  1260. }
  1261. err = ep1.Join(cnt)
  1262. if err != nil {
  1263. t.Fatalf("Failed to join ep1: %v", err)
  1264. }
  1265. runtime.LockOSThread()
  1266. err = ep2.Join(cnt)
  1267. if err != nil {
  1268. t.Fatalf("Failed to join ep2: %v", err)
  1269. }
  1270. runtime.LockOSThread()
  1271. err = cnt.Delete()
  1272. if err != nil {
  1273. t.Fatal(err)
  1274. }
  1275. }
  1276. func TestontainerInvalidLeave(t *testing.T) {
  1277. if !testutils.IsRunningInContainer() {
  1278. defer testutils.SetupTestOSContext(t)()
  1279. }
  1280. n, err := createTestNetwork(bridgeNetType, "testnetwork", options.Generic{
  1281. netlabel.GenericData: options.Generic{
  1282. "BridgeName": "testnetwork",
  1283. },
  1284. }, nil, nil)
  1285. if err != nil {
  1286. t.Fatal(err)
  1287. }
  1288. defer func() {
  1289. if err := n.Delete(); err != nil {
  1290. t.Fatal(err)
  1291. }
  1292. }()
  1293. ep, err := n.CreateEndpoint("ep1")
  1294. if err != nil {
  1295. t.Fatal(err)
  1296. }
  1297. defer func() {
  1298. if err := ep.Delete(); err != nil {
  1299. t.Fatal(err)
  1300. }
  1301. }()
  1302. cnt, err := controller.NewSandbox(containerID,
  1303. libnetwork.OptionHostname("test"),
  1304. libnetwork.OptionDomainname("docker.io"),
  1305. libnetwork.OptionExtraHost("web", "192.168.0.1"))
  1306. if err != nil {
  1307. t.Fatal(err)
  1308. }
  1309. defer func() {
  1310. if err := cnt.Delete(); err != nil {
  1311. t.Fatal(err)
  1312. }
  1313. }()
  1314. err = ep.Leave(cnt)
  1315. if err == nil {
  1316. t.Fatal("Expected to fail leave from an endpoint which has no active join")
  1317. }
  1318. if _, ok := err.(types.ForbiddenError); !ok {
  1319. t.Fatalf("Failed with unexpected error type: %T. Desc: %s", err, err.Error())
  1320. }
  1321. if err := ep.Leave(nil); err == nil {
  1322. t.Fatalf("Expected to fail leave nil Sandbox")
  1323. }
  1324. if _, ok := err.(types.BadRequestError); !ok {
  1325. t.Fatalf("Unexpected error type returned: %T", err)
  1326. }
  1327. fsbx := &fakeSandbox{}
  1328. if err = ep.Leave(fsbx); err == nil {
  1329. t.Fatalf("Expected to fail leave with invalid Sandbox")
  1330. }
  1331. if _, ok := err.(types.BadRequestError); !ok {
  1332. t.Fatalf("Unexpected error type returned: %T", err)
  1333. }
  1334. }
  1335. func TestEndpointUpdateParent(t *testing.T) {
  1336. if !testutils.IsRunningInContainer() {
  1337. defer testutils.SetupTestOSContext(t)()
  1338. }
  1339. n, err := createTestNetwork("bridge", "testnetwork", options.Generic{
  1340. netlabel.GenericData: options.Generic{
  1341. "BridgeName": "testnetwork",
  1342. },
  1343. }, nil, nil)
  1344. if err != nil {
  1345. t.Fatal(err)
  1346. }
  1347. defer func() {
  1348. if err := n.Delete(); err != nil {
  1349. t.Fatal(err)
  1350. }
  1351. }()
  1352. ep1, err := n.CreateEndpoint("ep1")
  1353. if err != nil {
  1354. t.Fatal(err)
  1355. }
  1356. ep2, err := n.CreateEndpoint("ep2")
  1357. if err != nil {
  1358. t.Fatal(err)
  1359. }
  1360. sbx1, err := controller.NewSandbox(containerID,
  1361. libnetwork.OptionHostname("test"),
  1362. libnetwork.OptionDomainname("docker.io"),
  1363. libnetwork.OptionExtraHost("web", "192.168.0.1"))
  1364. if err != nil {
  1365. t.Fatal(err)
  1366. }
  1367. defer func() {
  1368. if err := sbx1.Delete(); err != nil {
  1369. t.Fatal(err)
  1370. }
  1371. }()
  1372. sbx2, err := controller.NewSandbox("c2",
  1373. libnetwork.OptionHostname("test2"),
  1374. libnetwork.OptionDomainname("docker.io"),
  1375. libnetwork.OptionHostsPath("/var/lib/docker/test_network/container2/hosts"),
  1376. libnetwork.OptionExtraHost("web", "192.168.0.2"))
  1377. if err != nil {
  1378. t.Fatal(err)
  1379. }
  1380. defer func() {
  1381. if err := sbx2.Delete(); err != nil {
  1382. t.Fatal(err)
  1383. }
  1384. }()
  1385. err = ep1.Join(sbx1)
  1386. runtime.LockOSThread()
  1387. if err != nil {
  1388. t.Fatal(err)
  1389. }
  1390. err = ep2.Join(sbx2)
  1391. runtime.LockOSThread()
  1392. if err != nil {
  1393. t.Fatal(err)
  1394. }
  1395. }
  1396. func TestEnableIPv6(t *testing.T) {
  1397. if !testutils.IsRunningInContainer() {
  1398. defer testutils.SetupTestOSContext(t)()
  1399. }
  1400. tmpResolvConf := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\nnameserver 2001:4860:4860::8888\n")
  1401. //take a copy of resolv.conf for restoring after test completes
  1402. resolvConfSystem, err := ioutil.ReadFile("/etc/resolv.conf")
  1403. if err != nil {
  1404. t.Fatal(err)
  1405. }
  1406. //cleanup
  1407. defer func() {
  1408. if err := ioutil.WriteFile("/etc/resolv.conf", resolvConfSystem, 0644); err != nil {
  1409. t.Fatal(err)
  1410. }
  1411. }()
  1412. netOption := options.Generic{
  1413. netlabel.EnableIPv6: true,
  1414. netlabel.GenericData: options.Generic{
  1415. "BridgeName": "testnetwork",
  1416. },
  1417. }
  1418. ipamV6ConfList := []*libnetwork.IpamConf{&libnetwork.IpamConf{PreferredPool: "fe80::/98"}}
  1419. n, err := createTestNetwork("bridge", "testnetwork", netOption, nil, ipamV6ConfList)
  1420. if err != nil {
  1421. t.Fatal(err)
  1422. }
  1423. defer func() {
  1424. if err := n.Delete(); err != nil {
  1425. t.Fatal(err)
  1426. }
  1427. }()
  1428. ep1, err := n.CreateEndpoint("ep1")
  1429. if err != nil {
  1430. t.Fatal(err)
  1431. }
  1432. if err := ioutil.WriteFile("/etc/resolv.conf", tmpResolvConf, 0644); err != nil {
  1433. t.Fatal(err)
  1434. }
  1435. resolvConfPath := "/tmp/libnetwork_test/resolv.conf"
  1436. defer os.Remove(resolvConfPath)
  1437. sb, err := controller.NewSandbox(containerID, libnetwork.OptionResolvConfPath(resolvConfPath))
  1438. if err != nil {
  1439. t.Fatal(err)
  1440. }
  1441. defer func() {
  1442. if err := sb.Delete(); err != nil {
  1443. t.Fatal(err)
  1444. }
  1445. }()
  1446. err = ep1.Join(sb)
  1447. if err != nil {
  1448. t.Fatal(err)
  1449. }
  1450. content, err := ioutil.ReadFile(resolvConfPath)
  1451. if err != nil {
  1452. t.Fatal(err)
  1453. }
  1454. if !bytes.Equal(content, tmpResolvConf) {
  1455. t.Fatalf("Expected:\n%s\nGot:\n%s", string(tmpResolvConf), string(content))
  1456. }
  1457. if err != nil {
  1458. t.Fatal(err)
  1459. }
  1460. }
  1461. func TestResolvConfHost(t *testing.T) {
  1462. if !testutils.IsRunningInContainer() {
  1463. defer testutils.SetupTestOSContext(t)()
  1464. }
  1465. tmpResolvConf := []byte("search localhost.net\nnameserver 127.0.0.1\nnameserver 2001:4860:4860::8888\n")
  1466. //take a copy of resolv.conf for restoring after test completes
  1467. resolvConfSystem, err := ioutil.ReadFile("/etc/resolv.conf")
  1468. if err != nil {
  1469. t.Fatal(err)
  1470. }
  1471. //cleanup
  1472. defer func() {
  1473. if err := ioutil.WriteFile("/etc/resolv.conf", resolvConfSystem, 0644); err != nil {
  1474. t.Fatal(err)
  1475. }
  1476. }()
  1477. n, err := controller.NetworkByName("testhost")
  1478. if err != nil {
  1479. t.Fatal(err)
  1480. }
  1481. ep1, err := n.CreateEndpoint("ep1", nil)
  1482. if err != nil {
  1483. t.Fatal(err)
  1484. }
  1485. if err := ioutil.WriteFile("/etc/resolv.conf", tmpResolvConf, 0644); err != nil {
  1486. t.Fatal(err)
  1487. }
  1488. resolvConfPath := "/tmp/libnetwork_test/resolv.conf"
  1489. defer os.Remove(resolvConfPath)
  1490. sb, err := controller.NewSandbox(containerID,
  1491. libnetwork.OptionResolvConfPath(resolvConfPath),
  1492. libnetwork.OptionOriginResolvConfPath("/etc/resolv.conf"))
  1493. if err != nil {
  1494. t.Fatal(err)
  1495. }
  1496. defer func() {
  1497. if err := sb.Delete(); err != nil {
  1498. t.Fatal(err)
  1499. }
  1500. }()
  1501. err = ep1.Join(sb)
  1502. if err != nil {
  1503. t.Fatal(err)
  1504. }
  1505. defer func() {
  1506. err = ep1.Leave(sb)
  1507. if err != nil {
  1508. t.Fatal(err)
  1509. }
  1510. }()
  1511. finfo, err := os.Stat(resolvConfPath)
  1512. if err != nil {
  1513. t.Fatal(err)
  1514. }
  1515. fmode := (os.FileMode)(0644)
  1516. if finfo.Mode() != fmode {
  1517. t.Fatalf("Expected file mode %s, got %s", fmode.String(), finfo.Mode().String())
  1518. }
  1519. content, err := ioutil.ReadFile(resolvConfPath)
  1520. if err != nil {
  1521. t.Fatal(err)
  1522. }
  1523. if !bytes.Equal(content, tmpResolvConf) {
  1524. t.Fatalf("Expected:\n%s\nGot:\n%s", string(tmpResolvConf), string(content))
  1525. }
  1526. }
  1527. func TestResolvConf(t *testing.T) {
  1528. if !testutils.IsRunningInContainer() {
  1529. defer testutils.SetupTestOSContext(t)()
  1530. }
  1531. tmpResolvConf1 := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\nnameserver 2001:4860:4860::8888\n")
  1532. expectedResolvConf1 := []byte("search pommesfrites.fr\nnameserver 12.34.56.78\n")
  1533. tmpResolvConf2 := []byte("search pommesfrites.fr\nnameserver 112.34.56.78\nnameserver 2001:4860:4860::8888\n")
  1534. expectedResolvConf2 := []byte("search pommesfrites.fr\nnameserver 112.34.56.78\n")
  1535. tmpResolvConf3 := []byte("search pommesfrites.fr\nnameserver 113.34.56.78\n")
  1536. //take a copy of resolv.conf for restoring after test completes
  1537. resolvConfSystem, err := ioutil.ReadFile("/etc/resolv.conf")
  1538. if err != nil {
  1539. t.Fatal(err)
  1540. }
  1541. //cleanup
  1542. defer func() {
  1543. if err := ioutil.WriteFile("/etc/resolv.conf", resolvConfSystem, 0644); err != nil {
  1544. t.Fatal(err)
  1545. }
  1546. }()
  1547. netOption := options.Generic{
  1548. netlabel.GenericData: options.Generic{
  1549. "BridgeName": "testnetwork",
  1550. },
  1551. }
  1552. n, err := createTestNetwork("bridge", "testnetwork", netOption, nil, nil)
  1553. if err != nil {
  1554. t.Fatal(err)
  1555. }
  1556. defer func() {
  1557. if err := n.Delete(); err != nil {
  1558. t.Fatal(err)
  1559. }
  1560. }()
  1561. ep, err := n.CreateEndpoint("ep")
  1562. if err != nil {
  1563. t.Fatal(err)
  1564. }
  1565. if err := ioutil.WriteFile("/etc/resolv.conf", tmpResolvConf1, 0644); err != nil {
  1566. t.Fatal(err)
  1567. }
  1568. resolvConfPath := "/tmp/libnetwork_test/resolv.conf"
  1569. defer os.Remove(resolvConfPath)
  1570. sb1, err := controller.NewSandbox(containerID, libnetwork.OptionResolvConfPath(resolvConfPath))
  1571. if err != nil {
  1572. t.Fatal(err)
  1573. }
  1574. defer func() {
  1575. if err := sb1.Delete(); err != nil {
  1576. t.Fatal(err)
  1577. }
  1578. }()
  1579. err = ep.Join(sb1)
  1580. runtime.LockOSThread()
  1581. if err != nil {
  1582. t.Fatal(err)
  1583. }
  1584. finfo, err := os.Stat(resolvConfPath)
  1585. if err != nil {
  1586. t.Fatal(err)
  1587. }
  1588. fmode := (os.FileMode)(0644)
  1589. if finfo.Mode() != fmode {
  1590. t.Fatalf("Expected file mode %s, got %s", fmode.String(), finfo.Mode().String())
  1591. }
  1592. content, err := ioutil.ReadFile(resolvConfPath)
  1593. if err != nil {
  1594. t.Fatal(err)
  1595. }
  1596. if !bytes.Equal(content, expectedResolvConf1) {
  1597. fmt.Printf("\n%v\n%v\n", expectedResolvConf1, content)
  1598. t.Fatalf("Expected:\n%s\nGot:\n%s", string(expectedResolvConf1), string(content))
  1599. }
  1600. err = ep.Leave(sb1)
  1601. runtime.LockOSThread()
  1602. if err != nil {
  1603. t.Fatal(err)
  1604. }
  1605. if err := ioutil.WriteFile("/etc/resolv.conf", tmpResolvConf2, 0644); err != nil {
  1606. t.Fatal(err)
  1607. }
  1608. sb2, err := controller.NewSandbox(containerID+"_2", libnetwork.OptionResolvConfPath(resolvConfPath))
  1609. if err != nil {
  1610. t.Fatal(err)
  1611. }
  1612. defer func() {
  1613. if err := sb2.Delete(); err != nil {
  1614. t.Fatal(err)
  1615. }
  1616. }()
  1617. err = ep.Join(sb2)
  1618. runtime.LockOSThread()
  1619. if err != nil {
  1620. t.Fatal(err)
  1621. }
  1622. content, err = ioutil.ReadFile(resolvConfPath)
  1623. if err != nil {
  1624. t.Fatal(err)
  1625. }
  1626. if !bytes.Equal(content, expectedResolvConf2) {
  1627. t.Fatalf("Expected:\n%s\nGot:\n%s", string(expectedResolvConf2), string(content))
  1628. }
  1629. if err := ioutil.WriteFile(resolvConfPath, tmpResolvConf3, 0644); err != nil {
  1630. t.Fatal(err)
  1631. }
  1632. err = ep.Leave(sb2)
  1633. runtime.LockOSThread()
  1634. if err != nil {
  1635. t.Fatal(err)
  1636. }
  1637. err = ep.Join(sb2)
  1638. runtime.LockOSThread()
  1639. if err != nil {
  1640. t.Fatal(err)
  1641. }
  1642. content, err = ioutil.ReadFile(resolvConfPath)
  1643. if err != nil {
  1644. t.Fatal(err)
  1645. }
  1646. if !bytes.Equal(content, tmpResolvConf3) {
  1647. t.Fatalf("Expected:\n%s\nGot:\n%s", string(tmpResolvConf3), string(content))
  1648. }
  1649. }
  1650. func TestInvalidRemoteDriver(t *testing.T) {
  1651. if !testutils.IsRunningInContainer() {
  1652. t.Skip("Skipping test when not running inside a Container")
  1653. }
  1654. mux := http.NewServeMux()
  1655. server := httptest.NewServer(mux)
  1656. if server == nil {
  1657. t.Fatal("Failed to start a HTTP Server")
  1658. }
  1659. defer server.Close()
  1660. type pluginRequest struct {
  1661. name string
  1662. }
  1663. mux.HandleFunc("/Plugin.Activate", func(w http.ResponseWriter, r *http.Request) {
  1664. w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json")
  1665. fmt.Fprintln(w, `{"Implements": ["InvalidDriver"]}`)
  1666. })
  1667. if err := os.MkdirAll("/etc/docker/plugins", 0755); err != nil {
  1668. t.Fatal(err)
  1669. }
  1670. defer func() {
  1671. if err := os.RemoveAll("/etc/docker/plugins"); err != nil {
  1672. t.Fatal(err)
  1673. }
  1674. }()
  1675. if err := ioutil.WriteFile("/etc/docker/plugins/invalid-network-driver.spec", []byte(server.URL), 0644); err != nil {
  1676. t.Fatal(err)
  1677. }
  1678. ctrlr, err := libnetwork.New()
  1679. if err != nil {
  1680. t.Fatal(err)
  1681. }
  1682. defer ctrlr.Stop()
  1683. _, err = ctrlr.NewNetwork("invalid-network-driver", "dummy",
  1684. libnetwork.NetworkOptionGeneric(getEmptyGenericOption()))
  1685. if err == nil {
  1686. t.Fatal("Expected to fail. But instead succeeded")
  1687. }
  1688. if err != plugins.ErrNotImplements {
  1689. t.Fatalf("Did not fail with expected error. Actual error: %v", err)
  1690. }
  1691. }
  1692. func TestValidRemoteDriver(t *testing.T) {
  1693. if !testutils.IsRunningInContainer() {
  1694. t.Skip("Skipping test when not running inside a Container")
  1695. }
  1696. mux := http.NewServeMux()
  1697. server := httptest.NewServer(mux)
  1698. if server == nil {
  1699. t.Fatal("Failed to start a HTTP Server")
  1700. }
  1701. defer server.Close()
  1702. type pluginRequest struct {
  1703. name string
  1704. }
  1705. mux.HandleFunc("/Plugin.Activate", func(w http.ResponseWriter, r *http.Request) {
  1706. w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json")
  1707. fmt.Fprintf(w, `{"Implements": ["%s"]}`, driverapi.NetworkPluginEndpointType)
  1708. })
  1709. mux.HandleFunc(fmt.Sprintf("/%s.CreateNetwork", driverapi.NetworkPluginEndpointType), func(w http.ResponseWriter, r *http.Request) {
  1710. w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json")
  1711. fmt.Fprintf(w, "null")
  1712. })
  1713. if err := os.MkdirAll("/etc/docker/plugins", 0755); err != nil {
  1714. t.Fatal(err)
  1715. }
  1716. defer func() {
  1717. if err := os.RemoveAll("/etc/docker/plugins"); err != nil {
  1718. t.Fatal(err)
  1719. }
  1720. }()
  1721. if err := ioutil.WriteFile("/etc/docker/plugins/valid-network-driver.spec", []byte(server.URL), 0644); err != nil {
  1722. t.Fatal(err)
  1723. }
  1724. n, err := controller.NewNetwork("valid-network-driver", "dummy",
  1725. libnetwork.NetworkOptionGeneric(getEmptyGenericOption()))
  1726. if err != nil {
  1727. // Only fail if we could not find the plugin driver
  1728. if _, ok := err.(types.NotFoundError); ok {
  1729. t.Fatal(err)
  1730. }
  1731. return
  1732. }
  1733. defer func() {
  1734. if err := n.Delete(); err != nil {
  1735. t.Fatal(err)
  1736. }
  1737. }()
  1738. }
  1739. var (
  1740. once sync.Once
  1741. start = make(chan struct{})
  1742. done = make(chan chan struct{}, numThreads-1)
  1743. origns = netns.None()
  1744. testns = netns.None()
  1745. sboxes = make([]libnetwork.Sandbox, numThreads)
  1746. )
  1747. const (
  1748. iterCnt = 25
  1749. numThreads = 3
  1750. first = 1
  1751. last = numThreads
  1752. debug = false
  1753. )
  1754. func createGlobalInstance(t *testing.T) {
  1755. var err error
  1756. defer close(start)
  1757. origns, err = netns.Get()
  1758. if err != nil {
  1759. t.Fatal(err)
  1760. }
  1761. if testutils.IsRunningInContainer() {
  1762. testns = origns
  1763. } else {
  1764. testns, err = netns.New()
  1765. if err != nil {
  1766. t.Fatal(err)
  1767. }
  1768. }
  1769. netOption := options.Generic{
  1770. netlabel.GenericData: options.Generic{
  1771. "BridgeName": "network",
  1772. },
  1773. }
  1774. net1, err := controller.NetworkByName("testhost")
  1775. if err != nil {
  1776. t.Fatal(err)
  1777. }
  1778. net2, err := createTestNetwork("bridge", "network2", netOption, nil, nil)
  1779. if err != nil {
  1780. t.Fatal(err)
  1781. }
  1782. _, err = net1.CreateEndpoint("pep1")
  1783. if err != nil {
  1784. t.Fatal(err)
  1785. }
  1786. _, err = net2.CreateEndpoint("pep2")
  1787. if err != nil {
  1788. t.Fatal(err)
  1789. }
  1790. _, err = net2.CreateEndpoint("pep3")
  1791. if err != nil {
  1792. t.Fatal(err)
  1793. }
  1794. if sboxes[first-1], err = controller.NewSandbox(fmt.Sprintf("%drace", first), libnetwork.OptionUseDefaultSandbox()); err != nil {
  1795. t.Fatal(err)
  1796. }
  1797. for thd := first + 1; thd <= last; thd++ {
  1798. if sboxes[thd-1], err = controller.NewSandbox(fmt.Sprintf("%drace", thd)); err != nil {
  1799. t.Fatal(err)
  1800. }
  1801. }
  1802. }
  1803. func debugf(format string, a ...interface{}) (int, error) {
  1804. if debug {
  1805. return fmt.Printf(format, a...)
  1806. }
  1807. return 0, nil
  1808. }
  1809. func parallelJoin(t *testing.T, rc libnetwork.Sandbox, ep libnetwork.Endpoint, thrNumber int) {
  1810. debugf("J%d.", thrNumber)
  1811. var err error
  1812. sb := sboxes[thrNumber-1]
  1813. err = ep.Join(sb)
  1814. runtime.LockOSThread()
  1815. if err != nil {
  1816. if _, ok := err.(types.ForbiddenError); !ok {
  1817. t.Fatalf("thread %d: %v", thrNumber, err)
  1818. }
  1819. debugf("JE%d(%v).", thrNumber, err)
  1820. }
  1821. debugf("JD%d.", thrNumber)
  1822. }
  1823. func parallelLeave(t *testing.T, rc libnetwork.Sandbox, ep libnetwork.Endpoint, thrNumber int) {
  1824. debugf("L%d.", thrNumber)
  1825. var err error
  1826. sb := sboxes[thrNumber-1]
  1827. err = ep.Leave(sb)
  1828. runtime.LockOSThread()
  1829. if err != nil {
  1830. if _, ok := err.(types.ForbiddenError); !ok {
  1831. t.Fatalf("thread %d: %v", thrNumber, err)
  1832. }
  1833. debugf("LE%d(%v).", thrNumber, err)
  1834. }
  1835. debugf("LD%d.", thrNumber)
  1836. }
  1837. func runParallelTests(t *testing.T, thrNumber int) {
  1838. var (
  1839. ep libnetwork.Endpoint
  1840. sb libnetwork.Sandbox
  1841. err error
  1842. )
  1843. t.Parallel()
  1844. pTest := flag.Lookup("test.parallel")
  1845. if pTest == nil {
  1846. t.Skip("Skipped because test.parallel flag not set;")
  1847. }
  1848. numParallel, err := strconv.Atoi(pTest.Value.String())
  1849. if err != nil {
  1850. t.Fatal(err)
  1851. }
  1852. if numParallel < numThreads {
  1853. t.Skip("Skipped because t.parallel was less than ", numThreads)
  1854. }
  1855. runtime.LockOSThread()
  1856. defer runtime.UnlockOSThread()
  1857. if thrNumber == first {
  1858. createGlobalInstance(t)
  1859. }
  1860. if thrNumber != first {
  1861. select {
  1862. case <-start:
  1863. }
  1864. thrdone := make(chan struct{})
  1865. done <- thrdone
  1866. defer close(thrdone)
  1867. if thrNumber == last {
  1868. defer close(done)
  1869. }
  1870. err = netns.Set(testns)
  1871. if err != nil {
  1872. t.Fatal(err)
  1873. }
  1874. }
  1875. defer netns.Set(origns)
  1876. net1, err := controller.NetworkByName("testhost")
  1877. if err != nil {
  1878. t.Fatal(err)
  1879. }
  1880. if net1 == nil {
  1881. t.Fatal("Could not find testhost")
  1882. }
  1883. net2, err := controller.NetworkByName("network2")
  1884. if err != nil {
  1885. t.Fatal(err)
  1886. }
  1887. if net2 == nil {
  1888. t.Fatal("Could not find network2")
  1889. }
  1890. epName := fmt.Sprintf("pep%d", thrNumber)
  1891. if thrNumber == first {
  1892. ep, err = net1.EndpointByName(epName)
  1893. } else {
  1894. ep, err = net2.EndpointByName(epName)
  1895. }
  1896. if err != nil {
  1897. t.Fatal(err)
  1898. }
  1899. if ep == nil {
  1900. t.Fatal("Got nil ep with no error")
  1901. }
  1902. cid := fmt.Sprintf("%drace", thrNumber)
  1903. controller.WalkSandboxes(libnetwork.SandboxContainerWalker(&sb, cid))
  1904. if sb == nil {
  1905. t.Fatalf("Got nil sandbox for container: %s", cid)
  1906. }
  1907. for i := 0; i < iterCnt; i++ {
  1908. parallelJoin(t, sb, ep, thrNumber)
  1909. parallelLeave(t, sb, ep, thrNumber)
  1910. }
  1911. debugf("\n")
  1912. err = sb.Delete()
  1913. if err != nil {
  1914. t.Fatal(err)
  1915. }
  1916. if thrNumber == first {
  1917. for thrdone := range done {
  1918. select {
  1919. case <-thrdone:
  1920. }
  1921. }
  1922. testns.Close()
  1923. if err := net2.Delete(); err != nil {
  1924. t.Fatal(err)
  1925. }
  1926. } else {
  1927. err = ep.Delete()
  1928. if err != nil {
  1929. t.Fatal(err)
  1930. }
  1931. }
  1932. }
  1933. func TestParallel1(t *testing.T) {
  1934. runParallelTests(t, 1)
  1935. }
  1936. func TestParallel2(t *testing.T) {
  1937. runParallelTests(t, 2)
  1938. }
  1939. func TestParallel3(t *testing.T) {
  1940. runParallelTests(t, 3)
  1941. }