mirror of
https://github.com/fumiama/WireGold.git
synced 2026-06-05 07:50:24 +08:00
107 lines
2.3 KiB
Go
107 lines
2.3 KiB
Go
package lower
|
|
|
|
import (
|
|
"os"
|
|
"os/exec"
|
|
"strconv"
|
|
|
|
"github.com/fumiama/water"
|
|
"github.com/fumiama/water/waterutil"
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/fumiama/WireGold/gold/head"
|
|
"github.com/fumiama/WireGold/gold/link"
|
|
)
|
|
|
|
// NIC 虚拟网卡
|
|
type NIC struct {
|
|
ifce *water.Interface
|
|
ip string
|
|
subnet string
|
|
hasstart bool
|
|
}
|
|
|
|
// NewNIC 新建 TUN 网络接口卡
|
|
// 网卡地址为 ip, 所属子网为 subnet
|
|
func NewNIC(ip, subnet string) (n *NIC) {
|
|
ifce, err := water.New(water.Config{DeviceType: water.TUN})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
n = &NIC{
|
|
ifce: ifce,
|
|
ip: ip,
|
|
subnet: subnet,
|
|
}
|
|
n.prepare()
|
|
return
|
|
}
|
|
|
|
// Start 开始处理网卡消息，阻塞
|
|
func (nc *NIC) Start(m *link.Me) {
|
|
if nc.hasstart {
|
|
return
|
|
}
|
|
nc.hasstart = true
|
|
go func() { // 接收到 NIC
|
|
for nc.hasstart {
|
|
packet := m.Read()
|
|
n, err := nc.ifce.Write(packet.Data)
|
|
if err != nil {
|
|
logrus.Errorln("[lower] recv write to nic err:", err)
|
|
break
|
|
}
|
|
logrus.Infoln("[lower] recv write", n, "bytes packet to nic")
|
|
}
|
|
}()
|
|
buf := make([]byte, 4096)
|
|
for nc.hasstart { // 从 NIC 发送
|
|
packet := buf
|
|
n, err := nc.ifce.Read(packet)
|
|
if err != nil {
|
|
logrus.Errorln("[lower] send read from nic err:", err)
|
|
break
|
|
}
|
|
if n == 0 {
|
|
continue
|
|
}
|
|
packet = packet[:n]
|
|
if !waterutil.IsIPv4(packet) {
|
|
logrus.Warnln("[lower] skip to send", len(packet), "bytes non-ipv4 packet")
|
|
continue
|
|
}
|
|
dst := waterutil.IPv4Destination(packet)
|
|
srcport := waterutil.IPv4SourcePort(packet)
|
|
dstport := waterutil.IPv4DestinationPort(packet)
|
|
logrus.Infoln("[lower] sending", n, "bytes packet from :"+strconv.Itoa(int(srcport)), "to", dst.String()+":"+strconv.Itoa(int(dstport)))
|
|
lnk, err := m.Connect(dst.String())
|
|
if err != nil {
|
|
logrus.Warnln("[lower] connect to peer", dst.String(), "err:", err)
|
|
continue
|
|
}
|
|
lnk.Write(head.NewPacket(head.ProtoData, srcport, dst, dstport, packet), false)
|
|
}
|
|
}
|
|
|
|
// Stop 停止处理
|
|
func (n *NIC) Stop() {
|
|
n.hasstart = false
|
|
}
|
|
|
|
// Destroy 关闭网卡
|
|
func (n *NIC) Destroy() error {
|
|
return n.ifce.Close()
|
|
}
|
|
|
|
func execute(c string, args ...string) {
|
|
logrus.Printf("[lower] exec cmd: %v %v:", c, args)
|
|
cmd := exec.Command(c, args...)
|
|
cmd.Stderr = os.Stderr
|
|
cmd.Stdout = os.Stdout
|
|
cmd.Stdin = os.Stdin
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
logrus.Panicln("[lower] failed to exec cmd:", err)
|
|
}
|
|
}
|