libnetwork_test.go 50 KB

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