yuandian/slimming
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

成功

Browse Source
This commit is contained in:
huangsimin 2022-08-26 01:14:32 +08:00
parent 226d8e777b
commit 28df4d850d
2 changed files with 73 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
main.go
View File

@ -1,6 +1,15 @@
package main
import "log"
import (
"fmt"
"log"
"os/exec"
"strings"
"github.com/songgao/packets/ethernet"
"github.com/songgao/water"
"github.com/songgao/water/waterutil"
)
func init() {
log.SetFlags(log.Llongfile | log.LstdFlags)
@ -8,5 +17,53 @@ func init() {
//go:generate bash -c "protoc --go_out=plugins=grpc:. proto/*.proto"
func main() {
// testmain()
NewNetCard().Run()
}
func testmain() {
var a chan bool = make(chan bool)
go func() {
<-a
cmdstr := fmt.Sprintf("ip addr add %s dev stun", config.Network.Self.Virt)
log.Println(cmdstr)
cmd := strings.Split(cmdstr, " ")
err := exec.Command(cmd[0], cmd[1:]...).Run()
if err != nil {
log.Panic(err)
}
cmdstr = "ip link set dev stun up"
cmd = strings.Split(cmdstr, " ")
err = exec.Command(cmd[0], cmd[1:]...).Run()
if err != nil {
log.Panic(err)
}
}()
config := water.Config{
DeviceType: water.TUN,
}
config.Name = "stun"
ifce, err := water.New(config)
if err != nil {
log.Fatal(err)
}
a <- true
var frame ethernet.Frame
for {
frame.Resize(1500)
n, err := ifce.Read([]byte(frame))
if err != nil {
log.Fatal(err)
}
frame = frame[:n]
log.Printf("Dst: %s\n", waterutil.IPv4Destination(frame))
log.Printf("Src: %s\n", waterutil.IPv4Source(frame))
log.Printf("Ethertype: % x\n", frame.Ethertype())
// log.Printf("Payload: % x\n", frame.Payload())
}
}

30
tap.go
View File

@ -7,7 +7,6 @@ import (
"flag"
"fmt"
"log"
"net"
"os/exec"
gen "slimming/proto/gen"
"strings"
@ -17,6 +16,7 @@ import (
"github.com/klauspost/compress/zstd"
"github.com/songgao/packets/ethernet"
"github.com/songgao/water"
"github.com/songgao/water/waterutil"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
)
@ -97,7 +97,7 @@ func (nc *NetCard) Run() {
// go nc.cli.run()
time.Sleep(time.Second)
cmdstr := fmt.Sprintf("ip addr add %s dev stap", config.Network.Self.Virt)
cmdstr := fmt.Sprintf("ip addr add %s dev stun", config.Network.Self.Virt)
log.Println(cmdstr)
cmd := strings.Split(cmdstr, " ")
err := exec.Command(cmd[0], cmd[1:]...).Run()
@ -105,7 +105,7 @@ func (nc *NetCard) Run() {
log.Panic(err)
}
cmdstr = "ip link set dev stap up"
cmdstr = "ip link set dev stun up"
cmd = strings.Split(cmdstr, " ")
err = exec.Command(cmd[0], cmd[1:]...).Run()
if err != nil {
@ -120,7 +120,7 @@ func NewNetCard() *NetCard {
config := water.Config{
DeviceType: water.TUN,
}
config.Name = "stap"
config.Name = "stun"
ifce, err := water.New(config)
if err != nil {
@ -155,28 +155,27 @@ func (nc *NetCard) runRead() {
for atomic.LoadInt32(&isLoop) > 0 {
var rframe ethernet.Frame
rframe.Resize(1500)
n, err := ifce.Read([]byte(rframe))
if err != nil {
log.Fatal(err)
}
rframe = rframe[:n]
log.Printf("%v %v", net.IP(rframe.Source()).To4().String(), net.IP(rframe.Destination()).To4())
// log.Printf("Payload: % x\n", rframe.Payload())
if rframe.Ethertype() != ethernet.IPv4 {
continue
}
log.Println(net.IP(rframe.Destination()).String())
log.Printf("Dst: %s\n", waterutil.IPv4Destination(rframe))
log.Printf("Src: %s\n", waterutil.IPv4Source(rframe))
log.Printf("Ethertype: % x\n", rframe.Ethertype())
var buffer *[][]byte
if buffer, ok = bytesMap[net.IP(rframe.Destination()).String()]; !ok {
mbuffer := make([][]byte, 100)
var key = waterutil.IPv4Destination(rframe).String()
if buffer, ok = bytesMap[key]; !ok {
mbuffer := make([][]byte, 0, 100)
buffer = &mbuffer
bytesMap[net.IP(rframe.Destination()).String()] = buffer
bytesMap[key] = buffer
}
rframe = rframe[:n]
*buffer = append(*buffer, []byte(rframe))
}
@ -192,6 +191,7 @@ func (nc *NetCard) runRead() {
cli.connect(dst)
} else {
cli.FrameChan <- *buffer // 网卡数据 发到对方
*buffer = (*buffer)[0:]
}
}