更新 main.go
parent
b8ca5d7643
commit
6bdb7884d6
366
main.go
366
main.go
|
|
@ -1,183 +1,183 @@
|
||||||
//go:generate goversioninfo -icon=assets/iconwin.ico
|
//go:generate goversioninfo -icon=assets/iconwin.ico
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"fmt"
|
"fmt"
|
||||||
"log"
|
"log"
|
||||||
"net"
|
"net"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
"syscall"
|
"syscall"
|
||||||
"unsafe"
|
"unsafe"
|
||||||
|
|
||||||
_ "embed"
|
_ "embed"
|
||||||
|
|
||||||
"github.com/getlantern/systray"
|
"github.com/getlantern/systray"
|
||||||
"github.com/google/gopacket"
|
"github.com/google/gopacket"
|
||||||
"github.com/google/gopacket/pcap"
|
"github.com/google/gopacket/pcap"
|
||||||
"github.com/tiagomdiogo/ScaGo/sniffer"
|
"github.com/tiagomdiogo/ScaGo/sniffer"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
devName string
|
devName string
|
||||||
localIp string
|
localIp string
|
||||||
sendIpList []string
|
sendIpList []string
|
||||||
)
|
)
|
||||||
|
|
||||||
//go:embed assets/iconwin.ico
|
//go:embed assets/iconwin.ico
|
||||||
var iconBytes []byte
|
var iconBytes []byte
|
||||||
|
|
||||||
func sendPacket(packet gopacket.Packet, dstIp string) {
|
func sendPacket(packet gopacket.Packet, dstIp string) {
|
||||||
socket, err := net.DialUDP("udp",
|
socket, err := net.DialUDP("udp",
|
||||||
&net.UDPAddr{IP: net.ParseIP(localIp), Port: 6112},
|
&net.UDPAddr{IP: net.ParseIP(localIp), Port: 6112},
|
||||||
&net.UDPAddr{IP: net.ParseIP(dstIp), Port: 6112},
|
&net.UDPAddr{IP: net.ParseIP(dstIp), Port: 6112},
|
||||||
)
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Println(err)
|
log.Println(err)
|
||||||
}
|
}
|
||||||
defer socket.Close()
|
defer socket.Close()
|
||||||
socket.Write(packet.ApplicationLayer().Payload())
|
socket.Write(packet.ApplicationLayer().Payload())
|
||||||
}
|
}
|
||||||
|
|
||||||
func handlePacket(packet gopacket.Packet) {
|
func handlePacket(packet gopacket.Packet) {
|
||||||
log.Println("Sniffed packet:", packet.ApplicationLayer().Payload())
|
log.Println("Sniffed packet:", packet.ApplicationLayer().Payload())
|
||||||
logBuf := bytes.Buffer{}
|
logBuf := bytes.Buffer{}
|
||||||
for _, ip := range sendIpList {
|
for _, ip := range sendIpList {
|
||||||
sendPacket(packet, ip)
|
sendPacket(packet, ip)
|
||||||
logBuf.WriteString(fmt.Sprintf("发送数据包到 %s\n", ip))
|
logBuf.WriteString(fmt.Sprintf("发送数据包到 %s\n", ip))
|
||||||
}
|
}
|
||||||
log.Println(logBuf.String())
|
log.Println(logBuf.String())
|
||||||
}
|
}
|
||||||
|
|
||||||
func buildSendIpList() {
|
func buildSendIpList() {
|
||||||
for i := 1; i <= 50; i++ {
|
for i := 1; i <= 50; i++ {
|
||||||
sendIpList = append(sendIpList, fmt.Sprintf("192.168.73.%d", i))
|
sendIpList = append(sendIpList, fmt.Sprintf("192.168.73.%d", i))
|
||||||
}
|
}
|
||||||
for i := 10; i <= 30; i++ {
|
for i := 10; i <= 30; i++ {
|
||||||
sendIpList = append(sendIpList, fmt.Sprintf("192.168.72.%d", i))
|
sendIpList = append(sendIpList, fmt.Sprintf("192.168.72.%d", i))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func IsInCIDR(ip net.IP, cidr string) bool {
|
func IsInCIDR(ip net.IP, cidr string) bool {
|
||||||
_, ipnet, err := net.ParseCIDR(cidr)
|
_, ipnet, err := net.ParseCIDR(cidr)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
return ipnet.Contains(ip)
|
return ipnet.Contains(ip)
|
||||||
}
|
}
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
// 设置日志输出到文件
|
// 设置日志输出到文件
|
||||||
// 获取当前路径
|
// 获取当前路径
|
||||||
dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
|
dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal("无法获取当前路径:", err)
|
log.Fatal("无法获取当前路径:", err)
|
||||||
}
|
}
|
||||||
logFile, err := os.OpenFile(filepath.Join(dir, "war3-broadcaster.log"), os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
|
logFile, err := os.OpenFile(filepath.Join(dir, "war3-broadcaster.log"), os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal("无法创建日志文件:", err)
|
log.Fatal("无法创建日志文件:", err)
|
||||||
}
|
}
|
||||||
log.SetOutput(logFile)
|
log.SetOutput(logFile)
|
||||||
}
|
}
|
||||||
|
|
||||||
func onReady() {
|
func onReady() {
|
||||||
systray.SetIcon(iconBytes)
|
systray.SetIcon(iconBytes)
|
||||||
systray.SetTitle("WAR3开房助手")
|
systray.SetTitle("WAR3开房助手")
|
||||||
systray.SetTooltip("WAR3开房助手 - 运行中")
|
systray.SetTooltip("WAR3开房助手 - 运行中")
|
||||||
|
|
||||||
mOpenLog := systray.AddMenuItem("打开日志", "打开日志文件")
|
mOpenLog := systray.AddMenuItem("打开日志", "打开日志文件")
|
||||||
mQuit := systray.AddMenuItem("退出", "关闭程序")
|
mQuit := systray.AddMenuItem("退出", "关闭程序")
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
for {
|
for {
|
||||||
select {
|
select {
|
||||||
case <-mOpenLog.ClickedCh:
|
case <-mOpenLog.ClickedCh:
|
||||||
exec.Command("notepad.exe", "war3-broadcaster.log").Start()
|
exec.Command("notepad.exe", "war3-broadcaster.log").Start()
|
||||||
case <-mQuit.ClickedCh:
|
case <-mQuit.ClickedCh:
|
||||||
systray.Quit()
|
systray.Quit()
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
}
|
}
|
||||||
|
|
||||||
func onExit() {
|
func onExit() {
|
||||||
// 清理工作
|
// 清理工作
|
||||||
}
|
}
|
||||||
|
|
||||||
func checkProcessRunning(processName string) bool {
|
func checkProcessRunning(processName string) bool {
|
||||||
cmd := exec.Command("tasklist")
|
cmd := exec.Command("tasklist")
|
||||||
output, err := cmd.Output()
|
output, err := cmd.Output()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
// 判断存在两个war3-broadcaster.exe
|
// 判断存在两个war3-broadcaster.exe
|
||||||
return strings.Count(string(output), processName) >= 2
|
return strings.Count(string(output), processName) >= 2
|
||||||
}
|
}
|
||||||
|
|
||||||
func checkPcapInstalled() bool {
|
func checkPcapInstalled() bool {
|
||||||
cmd := exec.Command("sc", "query", "npcap")
|
cmd := exec.Command("sc", "query", "npcap")
|
||||||
output, err := cmd.Output()
|
output, err := cmd.Output()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
return strings.Contains(string(output), "RUNNING")
|
return strings.Contains(string(output), "RUNNING")
|
||||||
}
|
}
|
||||||
|
|
||||||
// 添加新的辅助函数
|
// 添加新的辅助函数
|
||||||
func showMessageBox(message string, title string, flags uint32) {
|
func showMessageBox(message string, title string, flags uint32) {
|
||||||
syscall.MustLoadDLL("user32.dll").MustFindProc("MessageBoxW").Call(
|
syscall.MustLoadDLL("user32.dll").MustFindProc("MessageBoxW").Call(
|
||||||
0,
|
0,
|
||||||
uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(message))),
|
uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(message))),
|
||||||
uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(title))),
|
uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(title))),
|
||||||
uintptr(flags))
|
uintptr(flags))
|
||||||
}
|
}
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
// 检查pcap是否安装
|
// 检查pcap是否安装
|
||||||
if !checkPcapInstalled() {
|
if !checkPcapInstalled() {
|
||||||
showMessageBox("请先安装Npcap", "错误", 0x10)
|
showMessageBox("请先安装Npcap", "错误", 0x10)
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
// 检查是否已经有进程在运行
|
// 检查是否已经有进程在运行
|
||||||
if checkProcessRunning("war3-broadcaster.exe") {
|
if checkProcessRunning("war3-broadcaster.exe") {
|
||||||
showMessageBox("WAR3开房助手已经在运行中", "提示", 0x40)
|
showMessageBox("WAR3开房助手已经在运行中", "提示", 0x40)
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
go func() {
|
go func() {
|
||||||
buildSendIpList()
|
buildSendIpList()
|
||||||
devices, err := pcap.FindAllDevs()
|
devices, err := pcap.FindAllDevs()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
log.Println("搜索网络设备中...")
|
log.Println("搜索网络设备中...")
|
||||||
devName = ""
|
devName = ""
|
||||||
for _, device := range devices {
|
for _, device := range devices {
|
||||||
for _, address := range device.Addresses {
|
for _, address := range device.Addresses {
|
||||||
if IsInCIDR(address.IP, "192.168.72.3/22") {
|
if IsInCIDR(address.IP, "192.168.72.0/22") {
|
||||||
devName = device.Name
|
devName = device.Name
|
||||||
localIp = address.IP.String()
|
localIp = address.IP.String()
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if devName == "" {
|
if devName == "" {
|
||||||
showMessageBox("找不到网络设备，请确认是否已连接樱花祭世界", "错误", 0x10)
|
showMessageBox("找不到网络设备，请确认是否已连接樱花祭世界", "错误", 0x10)
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
log.Fatalln("找不到网络设备，请确认是否已连接樱花祭世界")
|
log.Fatalln("找不到网络设备，请确认是否已连接樱花祭世界")
|
||||||
}
|
}
|
||||||
showMessageBox("成功找到设备，开始捕获War3广播数据", "提示", 0x40)
|
showMessageBox("成功找到设备，开始捕获War3广播数据", "提示", 0x40)
|
||||||
log.Printf("找到设备: %s, 本地IP: %s\n", devName, localIp)
|
log.Printf("找到设备: %s, 本地IP: %s\n", devName, localIp)
|
||||||
log.Println("开始捕获War3广播数据")
|
log.Println("开始捕获War3广播数据")
|
||||||
sniffer, err := sniffer.NewSniffer(devName, "dst net 255.255.255.255 and udp", handlePacket)
|
sniffer, err := sniffer.NewSniffer(devName, "dst net 255.255.255.255 and udp", handlePacket)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatalln(err)
|
log.Fatalln(err)
|
||||||
}
|
}
|
||||||
sniffer.Start()
|
sniffer.Start()
|
||||||
}()
|
}()
|
||||||
|
|
||||||
systray.Run(onReady, onExit)
|
systray.Run(onReady, onExit)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue