rpaste/modules/paste/sysinfo.go
nazunalika c901ee769b
Add SysInfo Functions
Add sysinfo functions, template updates, stuff like that
2021-12-27 11:45:46 -07:00

297 lines
7.5 KiB
Go

// sysinfo
package paste
import (
"bytes"
"io/ioutil"
"os/exec"
"regexp"
"strconv"
"strings"
"text/template"
"github.com/rocky-linux/rpaste/modules/setting"
)
var (
buf bytes.Buffer
)
type SystemInfo struct {
OsRelease string
DesktopEnvs string
DesktopInstalled string
SELinuxStatus string
SELinuxErrors string
CPUInfo string
Support64bit string
VirtSupport string
LoadAverage string
MemUsage string
TopCPUHogs string
TopMemHogs string
DiskUsage string
BlockDevs string
PciDevs string
USBDevs string
DRMInfo string
XorgModules string
GLSupport string
XorgErrors string
DmesgTail string
LastTenReboots string
DnfRepoList string
DnfRepoFiles string
YumConf string
LastTwentyPackages string
EFISupport string
}
func osRelease() string {
app := "uniq"
args := []string{"/etc/os-release"}
cmd, _ := exec.Command(app, args...).CombinedOutput()
return string(cmd)
}
func desktopEnvs() string {
// The below just returns whatever the match is, rather than each match. The
// original command (and by extension, the original fpaste python script and
// rpaste bash script) would spit out more than just the expected result
app := "ps"
args := []string{
"-eo",
"comm=",
}
cmd, _ := exec.Command(app, args...).CombinedOutput()
re, _ := regexp.Compile(`(gnome-session|startkde|startactive|xfce.?-session|fluxbox|blackbox|hackedbox|ratpoison|enlightenment|icewm-session|od-session|wmaker|wmx|openbox-lxde|openbox-gnome-session|openbox-kde-session|mwm|e16|fvwm|xmonad|sugar-session|mate-session|lxqt-session|cinnamon)`)
values := re.FindString(string(cmd))
return values
}
func desktopInstalled() string {
app := "ls"
args := []string{
"-m",
"/usr/share/xsessions/",
}
cmd, _ := exec.Command(app, args...).CombinedOutput()
re, _ := regexp.Compile(`\.desktop`)
values := re.ReplaceAllString(string(cmd), "")
return values
}
func selinuxStatus() string {
app1 := "getenforce"
app2 := "grep -v '^#' /etc/sysconfig/selinux"
cmd1, _ := exec.Command(app1).CombinedOutput()
cmd2, _ := exec.Command("/bin/sh", "-c", app2).CombinedOutput()
combined := string(cmd1) + string(cmd2)
return combined
}
func selinuxErrors() string {
// I'm not proud of this
app := "journalctl --since yesterday |grep avc: | grep -Eo comm='[^ ]+' | sort |uniq -c |sort -rn"
cmd, _ := exec.Command("/bin/sh", "-c", app).CombinedOutput()
return string(cmd)
}
func cpuInfo() string {
// I'm not proud of this
app := "grep 'model name' /proc/cpuinfo | awk -F: '{print $2}' | uniq -c | sed -re 's/^ +//'"
cmd, _ := exec.Command("/bin/sh", "-c", app).CombinedOutput()
return string(cmd)
}
func support64Bit() string {
text := " lm "
file, _ := ioutil.ReadFile("/proc/cpuinfo")
string := string(file)
return strconv.FormatBool(strings.Contains(string, text))
}
func virtSupport() string {
re, _ := regexp.Compile(`(vmx|svm)`)
file, _ := ioutil.ReadFile("/proc/cpuinfo")
exists := re.Match(file)
return strconv.FormatBool(exists)
}
func loadAverage() string {
app := "uptime"
cmd, _ := exec.Command(app).CombinedOutput()
return string(cmd)
}
func memUsage() string {
app := "free"
args := []string{"-m"}
cmd, _ := exec.Command(app, args...).CombinedOutput()
return string(cmd)
}
func topCPUHogs() string {
// I'm not proud of this
app := "ps axuScnh | sort -rnk3 | head -5"
cmd, _ := exec.Command("/bin/sh", "-c", app).CombinedOutput()
return string(cmd)
}
func topMemHogs() string {
// I'm not proud of this
app := "ps axuScnh | sort -rnk4 | head -5"
cmd, _ := exec.Command("/bin/sh", "-c", app).CombinedOutput()
return string(cmd)
}
func diskUsage() string {
app := "df"
args := []string{"-hT"}
cmd, _ := exec.Command(app, args...).CombinedOutput()
return string(cmd)
}
func blockDevs() string {
app := "/sbin/blkid"
cmd, _ := exec.Command(app).CombinedOutput()
return string(cmd)
}
func pciDevs() string {
app := "/sbin/lspci"
args := []string{"-nn"}
cmd, _ := exec.Command(app, args...).CombinedOutput()
return string(cmd)
}
func usbDevs() string {
app := "/bin/lsusb"
cmd, _ := exec.Command(app).CombinedOutput()
return string(cmd)
}
func drmInfo() string {
// I'm not proud of this
app := "journalctl -k -b | grep -o 'kernel:.*drm.*$' | cut -d ' ' -f 2-"
cmd, _ := exec.Command("/bin/sh", "-c", app).CombinedOutput()
return string(cmd)
}
func xorgModules() string {
// I'm not proud of this
app := "grep LoadModule /var/log/Xorg.0.log ~/.local/share/xorg/Xorg.0.log | cut -d '\"' -f 2 | xargs"
cmd, _ := exec.Command("/bin/sh", "-c", app).CombinedOutput()
return string(cmd)
}
func glSupport() string {
// I'm not proud of this
// turn to golang regexp? we would rather get the error output regardless
app := "glxinfo | grep -E 'OpenGL version|OpenGL renderer'"
cmd, _ := exec.Command("/bin/sh", "-c", app).CombinedOutput()
return string(cmd)
}
func xorgErrors() string {
// I'm not proud of this
app := "grep '^\\[.*(EE)' /var/log/Xorg.0.log ~/.local/share/xorg/Xorg.0.log | cut -d ':' -f 2-"
cmd, _ := exec.Command("/bin/sh", "-c", app).CombinedOutput()
return string(cmd)
}
func dmesgTail() string {
// I'm not proud of this
// Maybe I can implement a tail-like function
app := "dmesg | tail"
cmd, _ := exec.Command("/bin/sh", "-c", app).CombinedOutput()
return string(cmd)
}
func lastTenReboots() string {
app := "last"
args := []string{
"-x",
"-n10",
"reboot",
"runlevel",
}
cmd, _ := exec.Command(app, args...).CombinedOutput()
return string(cmd)
}
func dnfRepoList() string {
app := "dnf"
args := []string{"repolist"}
cmd, _ := exec.Command(app, args...).CombinedOutput()
return string(cmd)
}
func dnfRepoFiles() string {
app := "ls"
args := []string{
"-l",
"/etc/yum.repos.d/",
}
cmd, _ := exec.Command(app, args...).CombinedOutput()
return string(cmd)
}
func yumConf() string {
app := "grep -v '^#' /etc/yum.conf"
cmd, _ := exec.Command("/bin/sh", "-c", app).CombinedOutput()
return string(cmd)
}
func lastTwentyPkgs() string {
app := "rpm -qa --nodigest --nosignature --last | head -20"
cmd, _ := exec.Command("/bin/sh", "-c", app).CombinedOutput()
return string(cmd)
}
func efiSupport() string {
app := "efibootmgr"
args := []string{"-v"}
cmd, _ := exec.Command(app, args...).CombinedOutput()
return string(cmd)
}
func SysInfoGather() string {
sysInfoData := SystemInfo{
OsRelease: osRelease(),
DesktopEnvs: desktopEnvs(),
DesktopInstalled: desktopInstalled(),
SELinuxStatus: selinuxStatus(),
SELinuxErrors: selinuxErrors(),
CPUInfo: cpuInfo(),
Support64bit: support64Bit(),
VirtSupport: virtSupport(),
LoadAverage: loadAverage(),
MemUsage: memUsage(),
TopCPUHogs: topCPUHogs(),
TopMemHogs: topMemHogs(),
DiskUsage: diskUsage(),
BlockDevs: blockDevs(),
PciDevs: pciDevs(),
USBDevs: usbDevs(),
DRMInfo: drmInfo(),
XorgModules: xorgModules(),
GLSupport: glSupport(),
XorgErrors: xorgErrors(),
DmesgTail: dmesgTail(),
LastTenReboots: lastTenReboots(),
DnfRepoList: dnfRepoList(),
DnfRepoFiles: dnfRepoFiles(),
YumConf: yumConf(),
LastTwentyPackages: lastTwentyPkgs(),
EFISupport: efiSupport(),
}
t := template.New("sysinfo")
t, _ = t.Parse(setting.SysInfoTemplate)
t.Execute(&buf, sysInfoData)
sysInfoResults := buf.String()
return sysInfoResults
}