1
0

tunnels_test.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573
  1. //go:build e2e_testing
  2. // +build e2e_testing
  3. package e2e
  4. import (
  5. "fmt"
  6. "net/netip"
  7. "testing"
  8. "time"
  9. "github.com/slackhq/nebula/cert"
  10. "github.com/slackhq/nebula/cert_test"
  11. "github.com/slackhq/nebula/e2e/router"
  12. "github.com/slackhq/nebula/header"
  13. "github.com/slackhq/nebula/udp"
  14. "github.com/stretchr/testify/assert"
  15. "github.com/stretchr/testify/require"
  16. "gopkg.in/yaml.v3"
  17. )
  18. func TestDropInactiveTunnels(t *testing.T) {
  19. t.Parallel()
  20. // The goal of this test is to ensure the shortest inactivity timeout will close the tunnel on both sides
  21. // under ideal conditions
  22. ca, _, caKey, _ := cert_test.NewTestCaCert(cert.Version1, cert.Curve_CURVE25519, time.Now(), time.Now().Add(10*time.Minute), nil, nil, []string{})
  23. myControl, myVpnIpNet, myUdpAddr, _ := newSimpleServer(cert.Version1, ca, caKey, "me", "10.128.0.1/24", m{"tunnels": m{"drop_inactive": true, "inactivity_timeout": "5s"}})
  24. theirControl, theirVpnIpNet, theirUdpAddr, _ := newSimpleServer(cert.Version1, ca, caKey, "them", "10.128.0.2/24", m{"tunnels": m{"drop_inactive": true, "inactivity_timeout": "10m"}})
  25. // Share our underlay information
  26. myControl.InjectLightHouseAddr(theirVpnIpNet[0].Addr(), theirUdpAddr)
  27. theirControl.InjectLightHouseAddr(myVpnIpNet[0].Addr(), myUdpAddr)
  28. // Start the servers
  29. myControl.Start()
  30. theirControl.Start()
  31. r := router.NewR(t, myControl, theirControl)
  32. r.Log("Assert the tunnel between me and them works")
  33. assertTunnel(t, myVpnIpNet[0].Addr(), theirVpnIpNet[0].Addr(), myControl, theirControl, r)
  34. r.Log("Go inactive and wait for the tunnels to get dropped")
  35. waitStart := time.Now()
  36. for {
  37. myIndexes := len(myControl.GetHostmap().Indexes)
  38. theirIndexes := len(theirControl.GetHostmap().Indexes)
  39. if myIndexes == 0 && theirIndexes == 0 {
  40. break
  41. }
  42. since := time.Since(waitStart)
  43. r.Logf("my tunnels: %v; their tunnels: %v; duration: %v", myIndexes, theirIndexes, since)
  44. if since > time.Second*30 {
  45. t.Fatal("Tunnel should have been declared inactive after 5 seconds and before 30 seconds")
  46. }
  47. time.Sleep(1 * time.Second)
  48. r.FlushAll()
  49. }
  50. r.Logf("Inactive tunnels were dropped within %v", time.Since(waitStart))
  51. myControl.Stop()
  52. theirControl.Stop()
  53. }
  54. func TestCertUpgrade(t *testing.T) {
  55. t.Parallel()
  56. // The goal of this test is to ensure the shortest inactivity timeout will close the tunnel on both sides
  57. // under ideal conditions
  58. ca, _, caKey, _ := cert_test.NewTestCaCert(cert.Version1, cert.Curve_CURVE25519, time.Now(), time.Now().Add(10*time.Minute), nil, nil, []string{})
  59. caB, err := ca.MarshalPEM()
  60. if err != nil {
  61. panic(err)
  62. }
  63. ca2, _, caKey2, _ := cert_test.NewTestCaCert(cert.Version2, cert.Curve_CURVE25519, time.Now(), time.Now().Add(10*time.Minute), nil, nil, []string{})
  64. ca2B, err := ca2.MarshalPEM()
  65. if err != nil {
  66. panic(err)
  67. }
  68. caStr := fmt.Sprintf("%s\n%s", caB, ca2B)
  69. myCert, _, myPrivKey, _ := cert_test.NewTestCert(cert.Version1, cert.Curve_CURVE25519, ca, caKey, "me", time.Now(), time.Now().Add(5*time.Minute), []netip.Prefix{netip.MustParsePrefix("10.128.0.1/24")}, nil, []string{})
  70. _, myCert2Pem := cert_test.NewTestCertDifferentVersion(myCert, cert.Version2, ca2, caKey2)
  71. theirCert, _, theirPrivKey, _ := cert_test.NewTestCert(cert.Version1, cert.Curve_CURVE25519, ca, caKey, "them", time.Now(), time.Now().Add(5*time.Minute), []netip.Prefix{netip.MustParsePrefix("10.128.0.2/24")}, nil, []string{})
  72. theirCert2, _ := cert_test.NewTestCertDifferentVersion(theirCert, cert.Version2, ca2, caKey2)
  73. myControl, myVpnIpNet, myUdpAddr, myC := newServer([]cert.Certificate{ca, ca2}, []cert.Certificate{myCert}, myPrivKey, m{})
  74. theirControl, theirVpnIpNet, theirUdpAddr, _ := newServer([]cert.Certificate{ca, ca2}, []cert.Certificate{theirCert, theirCert2}, theirPrivKey, m{})
  75. // Share our underlay information
  76. myControl.InjectLightHouseAddr(theirVpnIpNet[0].Addr(), theirUdpAddr)
  77. theirControl.InjectLightHouseAddr(myVpnIpNet[0].Addr(), myUdpAddr)
  78. // Start the servers
  79. myControl.Start()
  80. theirControl.Start()
  81. r := router.NewR(t, myControl, theirControl)
  82. defer r.RenderFlow()
  83. r.Log("Assert the tunnel between me and them works")
  84. assertTunnel(t, myVpnIpNet[0].Addr(), theirVpnIpNet[0].Addr(), myControl, theirControl, r)
  85. r.Log("yay")
  86. //todo ???
  87. time.Sleep(1 * time.Second)
  88. r.FlushAll()
  89. mc := m{
  90. "pki": m{
  91. "ca": caStr,
  92. "cert": string(myCert2Pem),
  93. "key": string(myPrivKey),
  94. },
  95. //"tun": m{"disabled": true},
  96. "firewall": myC.Settings["firewall"],
  97. //"handshakes": m{
  98. // "try_interval": "1s",
  99. //},
  100. "listen": myC.Settings["listen"],
  101. "logging": myC.Settings["logging"],
  102. "timers": myC.Settings["timers"],
  103. }
  104. cb, err := yaml.Marshal(mc)
  105. if err != nil {
  106. panic(err)
  107. }
  108. r.Logf("reload new v2-only config")
  109. err = myC.ReloadConfigString(string(cb))
  110. assert.NoError(t, err)
  111. r.Log("yay, spin until their sees it")
  112. waitStart := time.Now()
  113. for {
  114. assertTunnel(t, myVpnIpNet[0].Addr(), theirVpnIpNet[0].Addr(), myControl, theirControl, r)
  115. c := theirControl.GetHostInfoByVpnAddr(myVpnIpNet[0].Addr(), false)
  116. if c == nil {
  117. r.Log("nil")
  118. } else {
  119. version := c.Cert.Version()
  120. r.Logf("version %d", version)
  121. if version == cert.Version2 {
  122. break
  123. }
  124. }
  125. since := time.Since(waitStart)
  126. if since > time.Second*10 {
  127. t.Fatal("Cert should be new by now")
  128. }
  129. time.Sleep(time.Second)
  130. }
  131. r.RenderHostmaps("Final hostmaps", myControl, theirControl)
  132. myControl.Stop()
  133. theirControl.Stop()
  134. }
  135. func TestCertDowngrade(t *testing.T) {
  136. t.Parallel()
  137. // The goal of this test is to ensure the shortest inactivity timeout will close the tunnel on both sides
  138. // under ideal conditions
  139. ca, _, caKey, _ := cert_test.NewTestCaCert(cert.Version1, cert.Curve_CURVE25519, time.Now(), time.Now().Add(10*time.Minute), nil, nil, []string{})
  140. caB, err := ca.MarshalPEM()
  141. if err != nil {
  142. panic(err)
  143. }
  144. ca2, _, caKey2, _ := cert_test.NewTestCaCert(cert.Version2, cert.Curve_CURVE25519, time.Now(), time.Now().Add(10*time.Minute), nil, nil, []string{})
  145. ca2B, err := ca2.MarshalPEM()
  146. if err != nil {
  147. panic(err)
  148. }
  149. caStr := fmt.Sprintf("%s\n%s", caB, ca2B)
  150. myCert, _, myPrivKey, myCertPem := cert_test.NewTestCert(cert.Version1, cert.Curve_CURVE25519, ca, caKey, "me", time.Now(), time.Now().Add(5*time.Minute), []netip.Prefix{netip.MustParsePrefix("10.128.0.1/24")}, nil, []string{})
  151. myCert2, _ := cert_test.NewTestCertDifferentVersion(myCert, cert.Version2, ca2, caKey2)
  152. theirCert, _, theirPrivKey, _ := cert_test.NewTestCert(cert.Version1, cert.Curve_CURVE25519, ca, caKey, "them", time.Now(), time.Now().Add(5*time.Minute), []netip.Prefix{netip.MustParsePrefix("10.128.0.2/24")}, nil, []string{})
  153. theirCert2, _ := cert_test.NewTestCertDifferentVersion(theirCert, cert.Version2, ca2, caKey2)
  154. myControl, myVpnIpNet, myUdpAddr, myC := newServer([]cert.Certificate{ca, ca2}, []cert.Certificate{myCert2}, myPrivKey, m{})
  155. theirControl, theirVpnIpNet, theirUdpAddr, _ := newServer([]cert.Certificate{ca, ca2}, []cert.Certificate{theirCert, theirCert2}, theirPrivKey, m{})
  156. // Share our underlay information
  157. myControl.InjectLightHouseAddr(theirVpnIpNet[0].Addr(), theirUdpAddr)
  158. theirControl.InjectLightHouseAddr(myVpnIpNet[0].Addr(), myUdpAddr)
  159. // Start the servers
  160. myControl.Start()
  161. theirControl.Start()
  162. r := router.NewR(t, myControl, theirControl)
  163. defer r.RenderFlow()
  164. r.Log("Assert the tunnel between me and them works")
  165. //assertTunnel(t, theirVpnIpNet[0].Addr(), myVpnIpNet[0].Addr(), theirControl, myControl, r)
  166. //r.Log("yay")
  167. assertTunnel(t, myVpnIpNet[0].Addr(), theirVpnIpNet[0].Addr(), myControl, theirControl, r)
  168. r.Log("yay")
  169. //todo ???
  170. time.Sleep(1 * time.Second)
  171. r.FlushAll()
  172. mc := m{
  173. "pki": m{
  174. "ca": caStr,
  175. "cert": string(myCertPem),
  176. "key": string(myPrivKey),
  177. },
  178. "firewall": myC.Settings["firewall"],
  179. "listen": myC.Settings["listen"],
  180. "logging": myC.Settings["logging"],
  181. "timers": myC.Settings["timers"],
  182. }
  183. cb, err := yaml.Marshal(mc)
  184. if err != nil {
  185. panic(err)
  186. }
  187. r.Logf("reload new v1-only config")
  188. err = myC.ReloadConfigString(string(cb))
  189. assert.NoError(t, err)
  190. r.Log("yay, spin until their sees it")
  191. waitStart := time.Now()
  192. for {
  193. assertTunnel(t, myVpnIpNet[0].Addr(), theirVpnIpNet[0].Addr(), myControl, theirControl, r)
  194. c := theirControl.GetHostInfoByVpnAddr(myVpnIpNet[0].Addr(), false)
  195. c2 := myControl.GetHostInfoByVpnAddr(theirVpnIpNet[0].Addr(), false)
  196. if c == nil || c2 == nil {
  197. r.Log("nil")
  198. } else {
  199. version := c.Cert.Version()
  200. theirVersion := c2.Cert.Version()
  201. r.Logf("version %d,%d", version, theirVersion)
  202. if version == cert.Version1 {
  203. break
  204. }
  205. }
  206. since := time.Since(waitStart)
  207. if since > time.Second*5 {
  208. r.Log("it is unusual that the cert is not new yet, but not a failure yet")
  209. }
  210. if since > time.Second*10 {
  211. r.Log("wtf")
  212. t.Fatal("Cert should be new by now")
  213. }
  214. time.Sleep(time.Second)
  215. }
  216. r.RenderHostmaps("Final hostmaps", myControl, theirControl)
  217. myControl.Stop()
  218. theirControl.Stop()
  219. }
  220. func TestCertMismatchCorrection(t *testing.T) {
  221. t.Parallel()
  222. // The goal of this test is to ensure the shortest inactivity timeout will close the tunnel on both sides
  223. // under ideal conditions
  224. ca, _, caKey, _ := cert_test.NewTestCaCert(cert.Version1, cert.Curve_CURVE25519, time.Now(), time.Now().Add(10*time.Minute), nil, nil, []string{})
  225. ca2, _, caKey2, _ := cert_test.NewTestCaCert(cert.Version2, cert.Curve_CURVE25519, time.Now(), time.Now().Add(10*time.Minute), nil, nil, []string{})
  226. myCert, _, myPrivKey, _ := cert_test.NewTestCert(cert.Version1, cert.Curve_CURVE25519, ca, caKey, "me", time.Now(), time.Now().Add(5*time.Minute), []netip.Prefix{netip.MustParsePrefix("10.128.0.1/24")}, nil, []string{})
  227. myCert2, _ := cert_test.NewTestCertDifferentVersion(myCert, cert.Version2, ca2, caKey2)
  228. theirCert, _, theirPrivKey, _ := cert_test.NewTestCert(cert.Version1, cert.Curve_CURVE25519, ca, caKey, "them", time.Now(), time.Now().Add(5*time.Minute), []netip.Prefix{netip.MustParsePrefix("10.128.0.2/24")}, nil, []string{})
  229. theirCert2, _ := cert_test.NewTestCertDifferentVersion(theirCert, cert.Version2, ca2, caKey2)
  230. myControl, myVpnIpNet, myUdpAddr, _ := newServer([]cert.Certificate{ca, ca2}, []cert.Certificate{myCert2}, myPrivKey, m{})
  231. theirControl, theirVpnIpNet, theirUdpAddr, _ := newServer([]cert.Certificate{ca, ca2}, []cert.Certificate{theirCert, theirCert2}, theirPrivKey, m{})
  232. // Share our underlay information
  233. myControl.InjectLightHouseAddr(theirVpnIpNet[0].Addr(), theirUdpAddr)
  234. theirControl.InjectLightHouseAddr(myVpnIpNet[0].Addr(), myUdpAddr)
  235. // Start the servers
  236. myControl.Start()
  237. theirControl.Start()
  238. r := router.NewR(t, myControl, theirControl)
  239. defer r.RenderFlow()
  240. r.Log("Assert the tunnel between me and them works")
  241. //assertTunnel(t, theirVpnIpNet[0].Addr(), myVpnIpNet[0].Addr(), theirControl, myControl, r)
  242. //r.Log("yay")
  243. assertTunnel(t, myVpnIpNet[0].Addr(), theirVpnIpNet[0].Addr(), myControl, theirControl, r)
  244. r.Log("yay")
  245. //todo ???
  246. time.Sleep(1 * time.Second)
  247. r.FlushAll()
  248. waitStart := time.Now()
  249. for {
  250. assertTunnel(t, myVpnIpNet[0].Addr(), theirVpnIpNet[0].Addr(), myControl, theirControl, r)
  251. c := theirControl.GetHostInfoByVpnAddr(myVpnIpNet[0].Addr(), false)
  252. c2 := myControl.GetHostInfoByVpnAddr(theirVpnIpNet[0].Addr(), false)
  253. if c == nil || c2 == nil {
  254. r.Log("nil")
  255. } else {
  256. version := c.Cert.Version()
  257. theirVersion := c2.Cert.Version()
  258. r.Logf("version %d,%d", version, theirVersion)
  259. if version == theirVersion {
  260. break
  261. }
  262. }
  263. since := time.Since(waitStart)
  264. if since > time.Second*5 {
  265. r.Log("wtf")
  266. }
  267. if since > time.Second*10 {
  268. r.Log("wtf")
  269. t.Fatal("Cert should be new by now")
  270. }
  271. time.Sleep(time.Second)
  272. }
  273. r.RenderHostmaps("Final hostmaps", myControl, theirControl)
  274. myControl.Stop()
  275. theirControl.Stop()
  276. }
  277. func TestCrossStackRelaysWork(t *testing.T) {
  278. t.Parallel()
  279. ca, _, caKey, _ := cert_test.NewTestCaCert(cert.Version2, cert.Curve_CURVE25519, time.Now(), time.Now().Add(10*time.Minute), nil, nil, []string{})
  280. myControl, myVpnIpNet, _, _ := newSimpleServer(cert.Version2, ca, caKey, "me ", "10.128.0.1/24,fc00::1/64", m{"relay": m{"use_relays": true}})
  281. relayControl, relayVpnIpNet, relayUdpAddr, _ := newSimpleServer(cert.Version2, ca, caKey, "relay ", "10.128.0.128/24,fc00::128/64", m{"relay": m{"am_relay": true}})
  282. theirUdp := netip.MustParseAddrPort("10.0.0.2:4242")
  283. theirControl, theirVpnIpNet, theirUdpAddr, _ := newSimpleServerWithUdp(cert.Version2, ca, caKey, "them ", "fc00::2/64", theirUdp, m{"relay": m{"use_relays": true}})
  284. //myVpnV4 := myVpnIpNet[0]
  285. myVpnV6 := myVpnIpNet[1]
  286. relayVpnV4 := relayVpnIpNet[0]
  287. relayVpnV6 := relayVpnIpNet[1]
  288. theirVpnV6 := theirVpnIpNet[0]
  289. // Teach my how to get to the relay and that their can be reached via the relay
  290. myControl.InjectLightHouseAddr(relayVpnV4.Addr(), relayUdpAddr)
  291. myControl.InjectLightHouseAddr(relayVpnV6.Addr(), relayUdpAddr)
  292. myControl.InjectRelays(theirVpnV6.Addr(), []netip.Addr{relayVpnV6.Addr()})
  293. relayControl.InjectLightHouseAddr(theirVpnV6.Addr(), theirUdpAddr)
  294. // Build a router so we don't have to reason who gets which packet
  295. r := router.NewR(t, myControl, relayControl, theirControl)
  296. defer r.RenderFlow()
  297. // Start the servers
  298. myControl.Start()
  299. relayControl.Start()
  300. theirControl.Start()
  301. t.Log("Trigger a handshake from me to them via the relay")
  302. myControl.InjectTunPacket(BuildTunUDPPacket(theirVpnV6.Addr(), 80, myVpnV6.Addr(), 80, []byte("Hi from me")))
  303. p := r.RouteForAllUntilTxTun(theirControl)
  304. r.Log("Assert the tunnel works")
  305. assertUdpPacket(t, []byte("Hi from me"), p, myVpnV6.Addr(), theirVpnV6.Addr(), 80, 80)
  306. t.Log("reply?")
  307. theirControl.InjectTunPacket(BuildTunUDPPacket(myVpnV6.Addr(), 80, theirVpnV6.Addr(), 80, []byte("Hi from them")))
  308. p = r.RouteForAllUntilTxTun(myControl)
  309. assertUdpPacket(t, []byte("Hi from them"), p, theirVpnV6.Addr(), myVpnV6.Addr(), 80, 80)
  310. r.RenderHostmaps("Final hostmaps", myControl, relayControl, theirControl)
  311. //t.Log("finish up")
  312. //myControl.Stop()
  313. //theirControl.Stop()
  314. //relayControl.Stop()
  315. }
  316. // TestRelayReplayProtection asserts that a relay (forwarding-type) node rejects
  317. // replayed relay frames. A captured relay frame, re-injected with the same
  318. // message counter, must be dropped by the replay window rather than re-forwarded
  319. // to the relay target. Before the fix, handleOutsideRelayPacket authenticated the
  320. // frame but never advanced the replay window, so every replay was re-forwarded.
  321. func TestRelayReplayProtection(t *testing.T) {
  322. t.Parallel()
  323. ca, _, caKey, _ := cert_test.NewTestCaCert(cert.Version2, cert.Curve_CURVE25519, time.Now(), time.Now().Add(10*time.Minute), nil, nil, []string{})
  324. myControl, myVpnIpNet, _, _ := newSimpleServer(cert.Version2, ca, caKey, "me ", "10.128.0.1/24,fc00::1/64", m{"relay": m{"use_relays": true}})
  325. relayControl, relayVpnIpNet, relayUdpAddr, _ := newSimpleServer(cert.Version2, ca, caKey, "relay ", "10.128.0.128/24,fc00::128/64", m{"relay": m{"am_relay": true}})
  326. theirUdp := netip.MustParseAddrPort("10.0.0.2:4242")
  327. theirControl, theirVpnIpNet, theirUdpAddr, _ := newSimpleServerWithUdp(cert.Version2, ca, caKey, "them ", "fc00::2/64", theirUdp, m{"relay": m{"use_relays": true}})
  328. myVpnV6 := myVpnIpNet[1]
  329. relayVpnV4 := relayVpnIpNet[0]
  330. relayVpnV6 := relayVpnIpNet[1]
  331. theirVpnV6 := theirVpnIpNet[0]
  332. // Teach me how to reach the relay and that them is reachable via the relay
  333. myControl.InjectLightHouseAddr(relayVpnV4.Addr(), relayUdpAddr)
  334. myControl.InjectLightHouseAddr(relayVpnV6.Addr(), relayUdpAddr)
  335. myControl.InjectRelays(theirVpnV6.Addr(), []netip.Addr{relayVpnV6.Addr()})
  336. relayControl.InjectLightHouseAddr(theirVpnV6.Addr(), theirUdpAddr)
  337. r := router.NewR(t, myControl, relayControl, theirControl)
  338. defer r.RenderFlow()
  339. myControl.Start()
  340. relayControl.Start()
  341. theirControl.Start()
  342. // Establish the relayed tunnel in both directions so all handshakes complete.
  343. t.Log("Establish the relayed tunnel")
  344. myControl.InjectTunPacket(BuildTunUDPPacket(theirVpnV6.Addr(), 80, myVpnV6.Addr(), 80, []byte("Hi from me")))
  345. p := r.RouteForAllUntilTxTun(theirControl)
  346. assertUdpPacket(t, []byte("Hi from me"), p, myVpnV6.Addr(), theirVpnV6.Addr(), 80, 80)
  347. theirControl.InjectTunPacket(BuildTunUDPPacket(myVpnV6.Addr(), 80, theirVpnV6.Addr(), 80, []byte("Hi from them")))
  348. p = r.RouteForAllUntilTxTun(myControl)
  349. assertUdpPacket(t, []byte("Hi from them"), p, theirVpnV6.Addr(), myVpnV6.Addr(), 80, 80)
  350. // Drain anything still queued on me's UDP tx so the next packet we pull is the
  351. // relay frame we are about to generate.
  352. for myControl.GetFromUDP(false) != nil {
  353. }
  354. // Capture a single legitimate relay frame that me transmits toward the relay.
  355. t.Log("Capture a relay frame from me -> relay")
  356. myControl.InjectTunPacket(BuildTunUDPPacket(theirVpnV6.Addr(), 80, myVpnV6.Addr(), 80, []byte("replay me")))
  357. relayFrame := myControl.GetFromUDP(true)
  358. require.Equal(t, relayUdpAddr, relayFrame.To, "captured frame should be addressed to the relay")
  359. var fh header.H
  360. require.NoError(t, fh.Parse(relayFrame.Data))
  361. require.Equal(t, header.Message, fh.Type)
  362. require.Equal(t, header.MessageRelay, fh.Subtype)
  363. // drainForwards counts relay frames the relay forwards toward them within the
  364. // settle window. We match on destination + (Message, MessageRelay) so the
  365. // relay's own direct traffic to them can't be miscounted.
  366. drainForwards := func(settle time.Duration) int {
  367. ch := relayControl.GetUDPTxChan()
  368. count := 0
  369. for {
  370. select {
  371. case pkt := <-ch:
  372. var ph header.H
  373. if pkt.To == theirUdpAddr && ph.Parse(pkt.Data) == nil &&
  374. ph.Type == header.Message && ph.Subtype == header.MessageRelay {
  375. count++
  376. }
  377. pkt.Release()
  378. case <-time.After(settle):
  379. return count
  380. }
  381. }
  382. }
  383. // First delivery of the captured frame: the relay should forward it once.
  384. t.Log("Deliver the captured frame once; relay forwards it to them")
  385. relayControl.InjectUDPPacket(relayFrame)
  386. require.Equal(t, 1, drainForwards(200*time.Millisecond), "relay should forward the first, legitimate copy")
  387. // Replay the exact same frame several times. A correct replay window rejects
  388. // these duplicates so the relay forwards none of them.
  389. t.Log("Replay the captured frame; relay must drop the duplicates")
  390. const replays = 3
  391. for i := 0; i < replays; i++ {
  392. relayControl.InjectUDPPacket(relayFrame)
  393. }
  394. forwarded := drainForwards(200 * time.Millisecond)
  395. assert.Equal(t, 0, forwarded, "relay re-forwarded %d/%d replayed relay frames; replay protection is ineffective on relay tunnels", forwarded, replays)
  396. r.RenderHostmaps("Final hostmaps", myControl, relayControl, theirControl)
  397. }
  398. func TestCloseTunnelAuthenticated(t *testing.T) {
  399. t.Parallel()
  400. ca, _, caKey, _ := cert_test.NewTestCaCert(cert.Version1, cert.Curve_CURVE25519, time.Now(), time.Now().Add(10*time.Minute), nil, nil, []string{})
  401. myControl, myVpnIpNet, myUdpAddr, _ := newSimpleServer(cert.Version1, ca, caKey, "me", "10.128.0.1/24", m{"tunnels": m{"drop_inactive": true, "inactivity_timeout": "5s"}})
  402. theirControl, theirVpnIpNet, theirUdpAddr, _ := newSimpleServer(cert.Version1, ca, caKey, "them", "10.128.0.2/24", m{"tunnels": m{"drop_inactive": true, "inactivity_timeout": "10m"}})
  403. // Share our underlay information
  404. myControl.InjectLightHouseAddr(theirVpnIpNet[0].Addr(), theirUdpAddr)
  405. theirControl.InjectLightHouseAddr(myVpnIpNet[0].Addr(), myUdpAddr)
  406. // Start the servers
  407. myControl.Start()
  408. theirControl.Start()
  409. r := router.NewR(t, myControl, theirControl)
  410. r.Log("Assert the tunnel between me and them works")
  411. assertTunnel(t, myVpnIpNet[0].Addr(), theirVpnIpNet[0].Addr(), myControl, theirControl, r)
  412. r.Log("Close the tunnel")
  413. myControl.CloseTunnel(theirVpnIpNet[0].Addr(), false)
  414. r.FlushAll()
  415. waitStart := time.Now()
  416. for {
  417. myIndexes := len(myControl.GetHostmap().Indexes)
  418. theirIndexes := len(theirControl.GetHostmap().Indexes)
  419. if myIndexes == 0 && theirIndexes == 0 {
  420. break
  421. }
  422. since := time.Since(waitStart)
  423. r.Logf("my tunnels: %v; their tunnels: %v; duration: %v", myIndexes, theirIndexes, since)
  424. if since > time.Second*6 {
  425. t.Fatal("Tunnel should have been declared inactive after 2 seconds and before 6 seconds")
  426. }
  427. time.Sleep(1 * time.Second)
  428. //r.FlushAll()
  429. }
  430. r.Logf("Happy path success, tunnels were dropped within %v", time.Since(waitStart))
  431. myControl.InjectLightHouseAddr(theirVpnIpNet[0].Addr(), theirUdpAddr)
  432. theirControl.InjectLightHouseAddr(myVpnIpNet[0].Addr(), myUdpAddr)
  433. r.Log("Assert another tunnel between me and them works")
  434. assertTunnel(t, myVpnIpNet[0].Addr(), theirVpnIpNet[0].Addr(), myControl, theirControl, r)
  435. hi := myControl.GetHostInfoByVpnAddr(theirVpnIpNet[0].Addr(), false)
  436. if hi == nil {
  437. t.Fatal("There is no hostinfo for this tunnel")
  438. }
  439. myHi := theirControl.GetHostInfoByVpnAddr(myVpnIpNet[0].Addr(), false)
  440. if myHi == nil {
  441. t.Fatal("There is no hostinfo for my tunnel")
  442. }
  443. r.Log("It does")
  444. buf := make([]byte, 1024)
  445. hdr := header.H{
  446. Version: 1,
  447. Type: header.CloseTunnel,
  448. Subtype: 0,
  449. Reserved: 0,
  450. RemoteIndex: hi.RemoteIndex,
  451. MessageCounter: 5,
  452. }
  453. out, err := hdr.Encode(buf)
  454. if err != nil {
  455. t.Fatal(err)
  456. }
  457. pkt := &udp.Packet{
  458. To: hi.CurrentRemote,
  459. From: myHi.CurrentRemote,
  460. Data: out,
  461. }
  462. r.InjectUDPPacket(myControl, theirControl, pkt)
  463. r.Log("Injected bogus close tunnel. Let's see!")
  464. waitStart = time.Now()
  465. for {
  466. myIndexes := len(myControl.GetHostmap().Indexes)
  467. theirIndexes := len(theirControl.GetHostmap().Indexes)
  468. if myIndexes == 0 {
  469. t.Fatal("myIndexes should not be 0")
  470. }
  471. if theirIndexes == 0 {
  472. t.Fatal("theirIndexes should not be 0, they should have rejected this bogus packet")
  473. }
  474. since := time.Since(waitStart)
  475. r.Logf("my tunnels: %v; their tunnels: %v; duration: %v", myIndexes, theirIndexes, since)
  476. if since > time.Second*4 {
  477. t.Log("The tunnel would have been gone by now")
  478. break
  479. }
  480. time.Sleep(1 * time.Second)
  481. r.FlushAll()
  482. }
  483. myControl.Stop()
  484. theirControl.Stop()
  485. }