reniced/util.go

package main

import (
	"fmt"
	"log"
	"os"
	"os/user"
	"runtime"
	"strings"
	"syscall"

	"gopkg.in/yaml.v3"
)

// readConf вычиляет, от какого пользователя запущен процесс и в зависимости от результата, скармливает readConfFile тот
// или иной путь, по которому предположительно лежит конфиг.
func readConf() (Config, error) {
	var (
		cnf Config
		err error
	)

	u, err := user.Current()

	if err != nil {
		log.Fatalf("Unable to get user info for current user: %s", err)
	}

	// Для рута и для других пользователей конфиги находятся в разных местах.
	if u.Uid != "0" || runtime.GOOS == "darwin" {
		if os.Getenv("HOME") == "" {
			log.Fatalln("Unable to get HOME environment variable value")
		}

		cnf, err = readConfFile(os.Getenv("HOME") + "/.reniced.yaml")
	} else {
		if runtime.GOOS == "freebsd" {
			cnf, err = readConfFile("/usr/local/etc/reniced.yaml")
		} else {
			cnf, err = readConfFile("/etc/reniced.yaml")
		}
	}

	return cnf, err
}

// readConfFile парсит даденный конфиг в глобальную структуру, с которой работает программа.
func readConfFile(path string) (Config, error) {
	var cfg Config

	fileInfo, err := os.Stat(path)

	// Предполагаем, что файла либо нет, либо мы не можем его прочитать, второе надо бы логгировать, но пока забьём.
	if err != nil {
		return Config{}, err
	}

	// Конфиг-файл длинноват для конфига, попробуем следующего кандидата.
	if fileInfo.Size() > 65535 {
		err := fmt.Errorf("config file %s is too long for config, skipping", path)

		return Config{}, err
	}

	buf, err := os.ReadFile(path)

	// Не удалось прочитать.
	if err != nil {
		return Config{}, err
	}

	if err = yaml.Unmarshal(buf, &cfg); err != nil {
		return Config{}, err
	}

	if cfg.LoopDelay == 0 {
		cfg.LoopDelay = 2000
		log.Printf("loop_delay set to %d milliseconds", cfg.LoopDelay)
	}

	if cfg.Debug {
		log.Println("debug set to true")
	}

	if cfg.MaxWorkers == 0 {
		cfg.MaxWorkers = 5
		log.Printf("max_workers set to %d", cfg.MaxWorkers)
	}

	if cfg.NbCapacity == 0 {
		cfg.NbCapacity = 100
		log.Printf("nb_capacity set to %d", cfg.NbCapacity)
	}

	if cfg.NbCapacity < cfg.MaxWorkers {
		cfg.NbCapacity = cfg.MaxWorkers
		log.Printf("nb_capacity set to same value as max_workers: %d", cfg.MaxWorkers)
	}

	// Обычный юзер не может задрать nice больше 0, только руту такое можно.
	var maxNice int

	if os.Getuid() == 0 {
		maxNice = -20
	}

	for _, v := range cfg.Prio {
		switch {
		case len(v.Name) != 0 && (v.Nice < 20 && v.Nice > maxNice):
			for _, procName := range v.Name {
				renice[procName] = v.Nice

				if cfg.Debug {
					log.Printf("Add %s to list of nicelevel %d processes.", procName, v.Nice)
				}
			}
		case len(v.Name) == 0:
			log.Println("Skipping empty entry in prio config block")
		case v.Nice > 20 || v.Nice < maxNice:
			log.Printf("Niceness value out of range %d < nice < 20 in prio config block", maxNice)
		}
	}

	for _, v := range cfg.IOPrio {
		switch {
		case len(v.Name) == 0:
			log.Println("Skipping empty name entry in ioprio config block")
		case v.Class > 3:
			log.Printf("Skipping entry %s with class > 3 in ioprio config block", v.Name)
		case v.Class == 0 && v.Prio != 0:
			log.Printf("Skipping entry %s with class 0 and prio not 0 in ioprio config block", v.Name)
		case v.Class == 3 && v.Prio != 0:
			log.Printf("Skipping entry %s with class 3 and prio not 0 in ioprio config block", v.Name)
		case v.Class == 1 && v.Prio > 7:
			log.Printf("Skipping entry %s with class 1 and prio > 7 in ioprio config block", v.Name)
		case v.Class == 2 && v.Prio > 7:
			log.Printf("Skipping entry %s with class 2 and prio > 7 in ioprio config block", v.Name)
		default:
			for _, processName := range v.Name {
				ioreniceClass[processName] = v.Class
				ioreniceClassdata[processName] = v.Prio

				if cfg.Debug {
					log.Printf("Add %s to list of ionice class %d processes.", processName, v.Class)
					log.Printf("Add %s to list of ionice prio %d processes.", processName, v.Prio)
				}
			}
		}
	}

	for _, v := range cfg.Kill {
		if len(v.Name) != 0 {
			switch v.Sig {
			case "kill", "stop", "term", "int", "quit", "abrt", "hup", "usr1", "usr2":
				for _, procName := range v.Name {
					switch v.Sig {
					case "kill":
						kill[procName] = syscall.SIGKILL
					case "stop":
						kill[procName] = syscall.SIGSTOP
					case "term":
						kill[procName] = syscall.SIGTERM
					case "int":
						kill[procName] = syscall.SIGINT
					case "quit":
						kill[procName] = syscall.SIGQUIT
					case "abrt":
						kill[procName] = syscall.SIGABRT
					case "hup":
						kill[procName] = syscall.SIGHUP
					case "usr1":
						kill[procName] = syscall.SIGUSR1
					case "usr2":
						kill[procName] = syscall.SIGUSR2
					default:
						continue
					}

					if cfg.Debug {
						log.Printf(
							"Add %s to list of processes that should be killed with SIG%s signal.",
							procName,
							strings.ToUpper(v.Sig),
						)
					}
				}
			default:
				log.Println("Skipping unsupported signal entry in kill config block")
			}
		} else {
			log.Println("Skipping empty entry in kill config block")
		}
	}

	return cfg, nil
}

/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */