Zabbix

/*

** Zabbix

** Copyright (C) 2001-2025 Zabbix SIA

**

** This program is free software; you can redistribute it and/or modify

** it under the terms of the GNU General Public License as published by

** the Free Software Foundation; either version 2 of the License, or

** (at your option) any later version.

**

** This program is distributed in the hope that it will be useful,

** but WITHOUT ANY WARRANTY; without even the implied warranty of

** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

** GNU General Public License for more details.

**

** You should have received a copy of the GNU General Public License

** along with this program; if not, write to the Free Software

** Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

**/

​

package uname

​

import (

    "errors"

    "os"

    "runtime"

    "strings"

    "syscall"

​

    "golang.org/x/sys/windows"

    "golang.zabbix.com/agent2/pkg/wmi"

)

​

func getHostname(params []string) (uname string, err error) {

    if len(params) > 2 {

        return "", errors.New("Too many parameters.")

    }

​

    var mode, transform string

​

    if len(params) > 0 {

        mode = params[0]

        if len(params) > 1 {

            transform = params[1]

        }

    }

​

    switch mode {

    case "netbios", "":

        w := make([]uint16, windows.MAX_COMPUTERNAME_LENGTH+1)

        sz := uint32(len(w))

        if err = syscall.GetComputerName(&w[0], &sz); err != nil {

            return "", err

        }

        uname = windows.UTF16ToString(w)

    case "host":

        if uname, err = os.Hostname(); err != nil {

            return "", err

        }

    case "shorthost":

        if uname, err = os.Hostname(); err != nil {

            return "", err

        }

        if idx := strings.Index(uname, "."); idx > 0 {

            uname = uname[:idx]

        }

    default:

        return "", errors.New("Invalid first parameter.")

    }

​

    switch transform {

    case "lower":

        uname = strings.ToLower(uname)

    case "none", "":

        break

    default:

        return "", errors.New("Invalid second parameter.")

    }

​

    return

}

​

func getArch() string {

    switch runtime.GOARCH {

    case "386":

        return "x86"

    case "amd64":

        return "x64"

    default:

        return runtime.GOARCH

    }

}