// +build linux,go1.16 aix darwin dragonfly freebsd illumos netbsd openbsd solaris

package main

import (
	"log"
	"os"
	"os/user"
	"strconv"
	"syscall"
)

type userInfo struct {
	uid int
	euid int
	gid int
	egid int
	supp_groups []int
	is_setuid bool
	is_setgid bool
	root_user bool
	root_prim_group bool
	root_supp_group bool
	need_drop bool
	unpriv_uid int
	unpriv_gid int
}

func getUserInfo(config Config) userInfo {
	var ui userInfo
	ui.uid = os.Getuid()
	ui.euid = os.Geteuid()
	ui.gid = os.Getgid()
	ui.egid = os.Getegid()
	supp_groups, err := os.Getgroups()
	if err != nil {
		log.Fatal("Could not get supplementary groups: ", err.Error())
	}
	ui.supp_groups = supp_groups
	ui.unpriv_uid = -1
	ui.unpriv_gid = -1

	ui.is_setuid = ui.uid != ui.euid
	ui.is_setgid = ui.gid != ui.egid
	ui.root_user = ui.uid == 0 || ui.euid == 0
	ui.root_prim_group = ui.gid == 0 || ui.egid == 0
	for _, gid := range ui.supp_groups {
		if gid == 0 {
			ui.root_supp_group = true
			break
		}
	}
	ui.need_drop = ui.is_setuid || ui.is_setgid || ui.root_user || ui.root_prim_group || ui.root_supp_group

	if ui.need_drop {
		nobody_user, err := user.Lookup(config.UnprivUsername)
		if err != nil {
			log.Fatal("Running as root but could not lookup UID for user " + config.UnprivUsername + ": " + err.Error())
		}
		ui.unpriv_uid, err = strconv.Atoi(nobody_user.Uid)
		ui.unpriv_gid, err = strconv.Atoi(nobody_user.Gid)
		if err != nil {
			log.Fatal("Running as root but could not lookup UID for user " + config.UnprivUsername + ": " + err.Error())
		}
	}

	return ui
}
func DropPrivs(ui userInfo, errorLog *log.Logger) {

	// If we're already unprivileged, all good
	if !ui.need_drop {
		return
	}

	// Drop supplementary groups
	if ui.root_supp_group {
		err := syscall.Setgroups([]int{})
		if err != nil {
			errorLog.Println("Could not unset supplementary groups: " + err.Error())
			log.Fatal(err)
		}
	}

	// Setguid()
	if ui.root_prim_group {
		err := syscall.Setgid(ui.unpriv_gid)
		if err != nil {
			errorLog.Println("Could not setgid to " + strconv.Itoa(ui.unpriv_gid) + ": " + err.Error())
			log.Fatal(err)
		}
	}

	// Setuid()
	if ui.root_user {
		err := syscall.Setuid(ui.unpriv_uid)
		if err != nil {
			errorLog.Println("Could not setuid to " + strconv.Itoa(ui.unpriv_uid) + ": " + err.Error())
			log.Fatal(err)
		}
	}

}