ms-runner/runner/runner.go

package runner

import (
    "errors"
    "fmt"
    "log"
    "os"
    "strings"
    exec "os/exec"
    cgroups "github.com/containerd/cgroups"
    //cgroup2 "github.com/containerd/cgroups/v3/cgroup2"
    //specs "github.com/opencontainers/runtime-spec/specs-go"
    rand "math/rand"
    "time"
    "os/user"
)

const runner_username string = "gaterunner"
const runIdLength = 20
var coresIsolated []int

type Limits struct {
    core    int
    memory  int
}

func extractNumbers(s string) (result []int) {
    lastNumber,isNumber := false,false
    var curNumber=0
    for _,char :=range s {
        isNumber=(char>='0' && char<='9')
        if(isNumber) {
            curNumber*=10
            curNumber+=int(char-'0')
        }
        if(!isNumber && lastNumber) {
            result=append(result,curNumber)
            curNumber=0
        }
        lastNumber=isNumber
    }
    if(lastNumber) {
        result=append(result,curNumber)
    }
    return
}

func Init() error {
    rand.Seed(time.Now().UnixNano())
    //croup initialisation:
    if cgroups.Mode() == cgroups.Unified {
        log.Println("cgroups v2 usage approved")
    } else {
        return fmt.Errorf("cgroups v2 are not enabled")//TODO: trouble description
    }
    //isolated cores initialisation:
    cmdlineBytes := make([]byte, 400)
    cmdlineFile,_:=os.Open("/proc/cmdline");
    countCmdlineBytes,_:=cmdlineFile.Read(cmdlineBytes);
    cmdline:=string(cmdlineBytes[:countCmdlineBytes])
    kernelParams:=strings.Split(cmdline," ")
    for _,param := range kernelParams{
        if(len(param)>=9 && param[:9]=="isolcpus=") {
            coresIsolated=append(coresIsolated, extractNumbers(param[9:])...)
        }
    }
    if(len(coresIsolated)==0) {
        return fmt.Errorf("no free cores available")//TODO: trouble description
    }
    log.Println("running on cores:",coresIsolated)
    //user setup:
    _,err:=user.Lookup(runner_username)
    if err!=nil{
        if errors.As(err, new(user.UnknownUserError)) {
            exec.Command("useradd", runner_username).Run();
        } else {
            return fmt.Errorf("user error:",err)
        }
    }
    log.Println("Runner initialisation successful!")
    return nil
}

func IsolatedRun(command []string,core int) {
    var runId string
    for i:=0;i<runIdLength;i++{runId+=string('a'+byte(rand.Int31n(26)))}
    os.mkdir("starter/"+runId,0777);
    exec.Command("starter/starter", );
}