-
Notifications
You must be signed in to change notification settings - Fork 659
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
84 changed files
with
6,289 additions
and
257 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,171 @@ | ||
| package util | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "github.com/shirou/gopsutil/v3/cpu" | ||
| "github.com/shirou/gopsutil/v3/disk" | ||
| "github.com/shirou/gopsutil/v3/host" | ||
| "github.com/shirou/gopsutil/v3/mem" | ||
| "github.com/shirou/gopsutil/v3/net" | ||
| systemNet "net" | ||
| "strconv" | ||
| "time" | ||
| ) | ||
|
|
||
| // 分区 | ||
| type Part struct { | ||
| Path string `json:"path"` | ||
| FsType string `json:"fstype"` | ||
| Total float64 `json:"total"` | ||
| Free float64 `json:"free"` | ||
| Used float64 `json:"used"` | ||
| UsedPercent int `json:"usedPercent"` | ||
| } | ||
|
|
||
| // 分区集合 | ||
| type Parts []Part | ||
|
|
||
| // CPU | ||
| type CpuSingle struct { | ||
| Num string `json:"num"` | ||
| Percent int `json:"percent"` | ||
| } | ||
|
|
||
| type CpuInfo struct { | ||
| CpuAvg float64 `json:"cpuAvg"` | ||
| CpuAll []CpuSingle `json:"cpuAll"` | ||
| } | ||
|
|
||
| const GB = 1024 * 1024 * 1024 | ||
|
|
||
| func decimal(v string) float64 { | ||
| value, _ := strconv.ParseFloat(v, 64) | ||
| return value | ||
| } | ||
|
|
||
| // 1.主机IP | ||
| func GetLocalIP() (ip string) { | ||
| addresses, err := systemNet.InterfaceAddrs() | ||
| if err != nil { | ||
| return "" | ||
| } | ||
| for _, addr := range addresses { | ||
| ipAddr, ok := addr.(*systemNet.IPNet) | ||
| if !ok { | ||
| continue | ||
| } | ||
| if ipAddr.IP.IsLoopback() { | ||
| continue | ||
| } | ||
| if !ipAddr.IP.IsGlobalUnicast() { | ||
| continue | ||
| } | ||
| return ipAddr.IP.String() | ||
| } | ||
| return "" | ||
| } | ||
|
|
||
| // 2.主机信息 | ||
| func GetHostInfo() (result *host.InfoStat, err error) { | ||
| result, err = host.Info() | ||
| return result, err | ||
| } | ||
|
|
||
| // 3.磁盘信息 | ||
| func GetDiskInfo() (result Parts, err error) { | ||
| parts, err := disk.Partitions(true) | ||
| if err != nil { | ||
| return result, err | ||
| } | ||
| for _, part := range parts { | ||
| diskInfo, err := disk.Usage(part.Mountpoint) | ||
| if err == nil { | ||
| result = append(result, Part{ | ||
| Path: diskInfo.Path, | ||
| FsType: diskInfo.Fstype, | ||
| Total: decimal(fmt.Sprintf("%.2f", float64(diskInfo.Total/GB))), | ||
| Free: decimal(fmt.Sprintf("%.2f", float64(diskInfo.Free/GB))), | ||
| Used: decimal(fmt.Sprintf("%.2f", float64(diskInfo.Used/GB))), | ||
| UsedPercent: int(diskInfo.UsedPercent), | ||
| }) | ||
| } else { | ||
| return result, err | ||
| } | ||
| } | ||
| return result, err | ||
| } | ||
|
|
||
| // 4.CPU使用率 | ||
| func GetCpuPercent() (result CpuInfo, err error) { | ||
| infos, err := cpu.Percent(1*time.Second, true) | ||
| if err != nil { | ||
| return result, err | ||
| } | ||
| var total float64 = 0 | ||
| for index, value := range infos { | ||
| result.CpuAll = append(result.CpuAll, CpuSingle{ | ||
| Num: fmt.Sprintf("#%d", index+1), | ||
| Percent: int(value), | ||
| }) | ||
| total += value | ||
| } | ||
| result.CpuAvg = decimal(fmt.Sprintf("%.1f", total/float64(len(infos)))) | ||
| return result, err | ||
| } | ||
|
|
||
| // 5.内存信息 | ||
| func GetMemInfo() (float64, []map[string]interface{}) { | ||
| info, err := mem.VirtualMemory() | ||
| if err != nil { | ||
| fmt.Println(err) | ||
| return 0, nil | ||
| } | ||
| return decimal(fmt.Sprintf("%.1f", info.UsedPercent)), []map[string]interface{}{ | ||
| {"key": "使用率[%]", "value": decimal(fmt.Sprintf("%.1f", info.UsedPercent))}, | ||
| {"key": "总量[GB]", "value": int(info.Total / GB)}, | ||
| {"key": "使用量[GB]", "value": int(info.Used / GB)}, | ||
| {"key": "剩余量[GB]", "value": int(info.Free / GB)}, | ||
| } | ||
| } | ||
|
|
||
| // 6.获取网卡信息 | ||
| func GetNetInfo() (result []net.IOCountersStat, err error) { | ||
| info, err := net.IOCounters(true) | ||
| if err != nil { | ||
| return result, err | ||
| } | ||
| return info, err | ||
| } | ||
|
|
||
| // 7.计算上下行带宽 | ||
| func GetNetSpeed() (speed map[string]map[string]uint64, err error) { | ||
| speed = map[string]map[string]uint64{} | ||
| info, err := net.IOCounters(true) | ||
| if err != nil { | ||
| return speed, err | ||
| } | ||
| for _, item := range info { | ||
| if item.BytesSent != 0 { | ||
| speed[item.Name] = map[string]uint64{ | ||
| "send": item.BytesSent, | ||
| "recv": item.BytesRecv, | ||
| } | ||
| } | ||
| } | ||
|
|
||
| time.Sleep(1 * time.Second) | ||
|
|
||
| info, err = net.IOCounters(true) | ||
| if err != nil { | ||
| return speed, err | ||
| } | ||
| for _, item := range info { | ||
| if item.BytesSent != 0 { | ||
| speed[item.Name] = map[string]uint64{ | ||
| "send": item.BytesSent - speed[item.Name]["send"], | ||
| "recv": item.BytesRecv - speed[item.Name]["recv"], | ||
| } | ||
| } | ||
| } | ||
| return speed, nil | ||
| } |
5 changes: 2 additions & 3 deletions
5
vendor/github.com/projectdiscovery/nuclei/v2/pkg/protocols/http/request.go
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
Oops, something went wrong.