func proxy(sm *subnet.SubnetManager, tun *os.File, conn *net.UDPConn, tunMTU uint, port int) {
log.Info("Running slow proxy loop")
rtr := NewRouter(port)
go proxyTunToUdp(rtr, tun, conn, tunMTU)
go proxyUdpToTun(conn, tun, tunMTU)
log.Info("Watching for new subnet leases")
evts := make(chan subnet.EventBatch)
sm.Start(evts)
for evtBatch := range evts {
for _, evt := range evtBatch {
if evt.Type == subnet.SubnetAdded {
log.Info("Subnet added: ", evt.Lease.Network)
var attrs subnet.BaseAttrs
if err := json.Unmarshal([]byte(evt.Lease.Data), &attrs); err != nil {
log.Error("Error decoding subnet lease JSON: ", err)
continue
}
rtr.SetRoute(evt.Lease.Network, attrs.PublicIP)
} else if evt.Type == subnet.SubnetRemoved {
log.Info("Subnet removed: ", evt.Lease.Network)
rtr.DelRoute(evt.Lease.Network)
} else {
log.Error("Internal error: unknown event type: ", int(evt.Type))
}
}
}
}
func fastProxy(sm *subnet.SubnetManager, tun *os.File, conn *net.UDPConn, tunIP ip.IP4, tunMTU uint, port int) {
log.Info("Running fast proxy loop")
c, err := conn.File()
if err != nil {
log.Error("Converting UDPConn to File failed: ", err)
return
}
defer c.Close()
ctl, peerCtl, err := newCtlSockets()
if err != nil {
log.Error("Failed to create control socket: ", err)
return
}
defer ctl.Close()
defer peerCtl.Close()
go runCProxy(tun, c, peerCtl, tunIP, tunMTU)
log.Info("Watching for new subnet leases")
evts := make(chan subnet.EventBatch)
sm.Start(evts)
for evtBatch := range evts {
for _, evt := range evtBatch {
if evt.Type == subnet.SubnetAdded {
log.Info("Subnet added: ", evt.Lease.Network)
var attrs subnet.BaseAttrs
if err := json.Unmarshal([]byte(evt.Lease.Data), &attrs); err != nil {
log.Error("Error decoding subnet lease JSON: ", err)
continue
}
cmd := C.command{
cmd: C.CMD_SET_ROUTE,
dest_net: C.in_addr_t(evt.Lease.Network.IP.NetworkOrder()),
dest_net_len: C.int(evt.Lease.Network.PrefixLen),
next_hop_ip: C.in_addr_t(attrs.PublicIP.NetworkOrder()),
next_hop_port: C.short(port),
}
writeCommand(ctl, &cmd)
} else if evt.Type == subnet.SubnetRemoved {
log.Info("Subnet removed: ", evt.Lease.Network)
cmd := C.command{
cmd: C.CMD_DEL_ROUTE,
dest_net: C.in_addr_t(evt.Lease.Network.IP.NetworkOrder()),
dest_net_len: C.int(evt.Lease.Network.PrefixLen),
}
writeCommand(ctl, &cmd)
} else {
log.Error("Internal error: unknown event type: ", int(evt.Type))
}
}
}
}
