137 lines
3.0 KiB
Go
137 lines
3.0 KiB
Go
package ui
|
|
|
|
import (
|
|
"fmt"
|
|
"go_dreamfactory/cmd/v2/lib/common"
|
|
"go_dreamfactory/cmd/v2/service"
|
|
"go_dreamfactory/cmd/v2/service/observer"
|
|
"net"
|
|
"runtime"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"fyne.io/fyne/v2"
|
|
"fyne.io/fyne/v2/container"
|
|
"fyne.io/fyne/v2/theme"
|
|
"fyne.io/fyne/v2/widget"
|
|
"github.com/go-ping/ping"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/cast"
|
|
)
|
|
|
|
type portStatus struct {
|
|
port int
|
|
err string
|
|
}
|
|
type appPing struct {
|
|
appAdapter
|
|
resultCh chan portStatus
|
|
}
|
|
|
|
func (this *appPing) LazyInit(ptService service.PttService, obs observer.Observer) error {
|
|
this.tabItem = container.NewTabItemWithIcon(common.TOOLBAR_PING, theme.DownloadIcon(), nil)
|
|
|
|
content := container.NewMax()
|
|
content.Objects = []fyne.CanvasObject{}
|
|
|
|
targetHost := widget.NewEntry()
|
|
targetHost.PlaceHolder = "目标主机Ip"
|
|
|
|
portEntry := widget.NewMultiLineEntry()
|
|
portEntry.Text = "20,21,22,80,1521,2379,2380,3306,6379,8080,8020,8500,9000,9090,13306,50070,27019,10013,8300,8301,8600,10012,10011"
|
|
form := widget.NewForm(
|
|
widget.NewFormItem("主机", targetHost),
|
|
widget.NewFormItem("端口", portEntry),
|
|
)
|
|
// result
|
|
result := widget.NewMultiLineEntry()
|
|
result.Disable()
|
|
|
|
form.OnSubmit = func() {
|
|
result.Text = ""
|
|
if portEntry.Text == "" {
|
|
return
|
|
}
|
|
this.resultCh = make(chan portStatus)
|
|
ports := strings.Split(portEntry.Text, ",")
|
|
|
|
pinger, err := ping.NewPinger(targetHost.Text)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
pinger.Count = 3
|
|
if runtime.GOOS == "windows" {
|
|
pinger.SetPrivileged(true)
|
|
}
|
|
var msgs []string
|
|
pinger.OnRecv = func(p *ping.Packet) {
|
|
msgs = append(msgs, fmt.Sprintf("来自:%s 的回复 字节=%d 时间=%dms TTL=%d",
|
|
p.IPAddr.String(),
|
|
p.Nbytes,
|
|
p.Rtt.Milliseconds(),
|
|
p.Ttl))
|
|
result.Text = strings.Join(msgs, "\n")
|
|
result.Refresh()
|
|
}
|
|
|
|
if err := pinger.Run(); err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
|
|
ip, err := net.ResolveIPAddr("ip", targetHost.Text)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
return
|
|
}
|
|
this.ping(ip.String(), ports)
|
|
|
|
for p := range this.resultCh {
|
|
msgs = append(msgs, fmt.Sprintf("端口:%d %s", p.port, p.err))
|
|
result.Text = strings.Join(msgs, "\n")
|
|
result.Refresh()
|
|
}
|
|
|
|
}
|
|
form.Items[1].HintText = "多个端口使用英文,号分隔"
|
|
|
|
result.OnChanged = func(s string) {
|
|
result.Refresh()
|
|
}
|
|
content.Objects = append(content.Objects, container.NewBorder(form, nil, nil, nil, result))
|
|
this.tabItem.Content = content
|
|
return nil
|
|
}
|
|
|
|
func (this *appPing) ping(targetHost string, ports []string) {
|
|
var wg sync.WaitGroup
|
|
wg.Add(len(ports))
|
|
for _, port := range ports {
|
|
go func(p int) {
|
|
defer wg.Done()
|
|
_, err := net.DialTimeout("tcp", fmt.Sprintf("%s:%d", targetHost, p), time.Millisecond*500)
|
|
var errStr string
|
|
if err == nil {
|
|
errStr = "OK"
|
|
} else {
|
|
errStr = "ERR"
|
|
logrus.Error(err)
|
|
}
|
|
this.resultCh <- portStatus{port: p, err: errStr}
|
|
}(cast.ToInt(port))
|
|
}
|
|
|
|
go func() {
|
|
defer close(this.resultCh)
|
|
wg.Wait()
|
|
}()
|
|
}
|
|
|
|
func (a *appPing) GetAppName() string {
|
|
return common.TOOLBAR_PING
|
|
}
|
|
|
|
func (a *appPing) Icon() fyne.Resource {
|
|
return theme.ComputerIcon()
|
|
}
|