send captured packets to server
This commit is contained in:
@@ -2,6 +2,7 @@ package main
|
||||
|
||||
import (
|
||||
"encoding/binary"
|
||||
"encoding/hex"
|
||||
"fmt"
|
||||
"log"
|
||||
"net"
|
||||
@@ -11,7 +12,6 @@ import (
|
||||
|
||||
"baalvpn/shared"
|
||||
|
||||
"github.com/songgao/packets/ethernet"
|
||||
"github.com/songgao/water"
|
||||
)
|
||||
|
||||
@@ -26,6 +26,7 @@ var (
|
||||
peers = map[string]*Peer{}
|
||||
peersMutex sync.Mutex
|
||||
nextIP = 3
|
||||
iface *water.Interface
|
||||
)
|
||||
|
||||
func main() {
|
||||
@@ -36,47 +37,25 @@ func main() {
|
||||
config := water.Config{DeviceType: water.TAP}
|
||||
config.Name = IFACE_NAME
|
||||
|
||||
iface, err := water.New(config)
|
||||
var err error
|
||||
iface, err = water.New(config)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
if err := exec.Command("ip", "addr", "add", "10.1.0.10/24", "dev", IFACE_NAME).Run(); err != nil {
|
||||
if err := exec.Command("ip", "addr", "add", "10.1.0.2/24", "dev", IFACE_NAME).Run(); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
if err := exec.Command("ip", "link", "set", "dev", IFACE_NAME, "up").Run(); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
log.Println("Listening...")
|
||||
|
||||
go receivePackets()
|
||||
sendPackets(iface)
|
||||
}
|
||||
|
||||
func sendPackets(iface *water.Interface) {
|
||||
var frame ethernet.Frame
|
||||
for {
|
||||
frame.Resize(1500)
|
||||
n, err := iface.Read([]byte(frame))
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
frame = frame[:n]
|
||||
|
||||
if frame.Destination()[0]&0x01 == 0x01 {
|
||||
log.Println("multicast")
|
||||
continue
|
||||
}
|
||||
|
||||
log.Printf("Dst: %s\n", frame.Destination())
|
||||
log.Printf("Src: %s\n", frame.Source())
|
||||
log.Printf("Ethertype: % x\n", frame.Ethertype())
|
||||
log.Printf("Payload: % x\n", frame.Payload())
|
||||
}
|
||||
receivePackets()
|
||||
}
|
||||
|
||||
func receivePackets() {
|
||||
log.Println("Listening for packets...")
|
||||
|
||||
listenAddr, err := net.ResolveUDPAddr("udp", ":38000")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
@@ -102,7 +81,7 @@ func receivePackets() {
|
||||
}
|
||||
}
|
||||
|
||||
func receivePacket(c *net.UDPConn, pkt []byte, addr *net.UDPAddr) {
|
||||
func receivePacket(conn *net.UDPConn, pkt []byte, addr *net.UDPAddr) {
|
||||
if len(pkt) < 4 {
|
||||
log.Println("packet too short")
|
||||
return
|
||||
@@ -135,9 +114,12 @@ func receivePacket(c *net.UDPConn, pkt []byte, addr *net.UDPAddr) {
|
||||
}
|
||||
|
||||
resp := shared.BuildPkt(shared.RESP_REGISTER, net.ParseIP(internalIP).To4())
|
||||
if _, err := c.WriteToUDP(resp, addr); err != nil {
|
||||
log.Printf("register WriteToUDP: %v\n", err)
|
||||
if _, err := conn.WriteToUDP(resp, addr); err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
case shared.REQ_DATA:
|
||||
fmt.Println("received packet from " + peerIP + ":")
|
||||
fmt.Println(hex.EncodeToString(pkt[4:]))
|
||||
default:
|
||||
log.Println("unknown packet type")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user