user.go 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. package overlay
  2. import (
  3. "io"
  4. "net/netip"
  5. "github.com/sirupsen/logrus"
  6. "github.com/slackhq/nebula/config"
  7. "github.com/slackhq/nebula/routing"
  8. )
  9. func NewUserDeviceFromConfig(c *config.C, l *logrus.Logger, vpnNetworks []netip.Prefix, routines int) (Device, error) {
  10. return NewUserDevice(vpnNetworks)
  11. }
  12. func NewUserDevice(vpnNetworks []netip.Prefix) (Device, error) {
  13. // these pipes guarantee each write/read will match 1:1
  14. or, ow := io.Pipe()
  15. ir, iw := io.Pipe()
  16. return &UserDevice{
  17. vpnNetworks: vpnNetworks,
  18. outboundReader: or,
  19. outboundWriter: ow,
  20. inboundReader: ir,
  21. inboundWriter: iw,
  22. }, nil
  23. }
  24. type UserDevice struct {
  25. vpnNetworks []netip.Prefix
  26. outboundReader *io.PipeReader
  27. outboundWriter *io.PipeWriter
  28. inboundReader *io.PipeReader
  29. inboundWriter *io.PipeWriter
  30. }
  31. func (d *UserDevice) Activate() error {
  32. return nil
  33. }
  34. func (d *UserDevice) Networks() []netip.Prefix { return d.vpnNetworks }
  35. func (d *UserDevice) Name() string { return "faketun0" }
  36. func (d *UserDevice) RoutesFor(ip netip.Addr) routing.Gateways {
  37. return routing.Gateways{routing.NewGateway(ip, 1)}
  38. }
  39. func (d *UserDevice) NewMultiQueueReader() (io.ReadWriteCloser, error) {
  40. return d, nil
  41. }
  42. func (d *UserDevice) Pipe() (*io.PipeReader, *io.PipeWriter) {
  43. return d.inboundReader, d.outboundWriter
  44. }
  45. func (d *UserDevice) Read(p []byte) (n int, err error) {
  46. return d.outboundReader.Read(p)
  47. }
  48. func (d *UserDevice) Write(p []byte) (n int, err error) {
  49. return d.inboundWriter.Write(p)
  50. }
  51. func (d *UserDevice) Close() error {
  52. d.inboundWriter.Close()
  53. d.outboundWriter.Close()
  54. return nil
  55. }