Source of plugin_userparameter.go - Zabbix

Raw file
Source viewDiff to previous
4.72 KB
xxxxxxxxxx
 
/*
** 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 Affero General Public License as published by the Free Software Foundation, version 3.
**
** 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 Affero General Public License for more details.
**
** You should have received a copy of the GNU Affero General Public License along with this program.
** If not, see <https://www.gnu.org/licenses/>.
**/
​
package agent
​
import (
    "bytes"
    "fmt"
    "strings"
    "time"
    "unicode"
​
    "golang.zabbix.com/agent2/pkg/itemutil"
    "golang.zabbix.com/agent2/pkg/zbxcmd"
    "golang.zabbix.com/sdk/errs"
    "golang.zabbix.com/sdk/plugin"
)
​
type parameterInfo struct {
    cmd      string
    flexible bool
}
​
// Plugin -
type UserParameterPlugin struct {
    plugin.Base
    parameters           map[string]*parameterInfo
    unsafeUserParameters int
    userParameterDir     string
}
​
var userParameter UserParameterPlugin
​
func (p *UserParameterPlugin) cmd(key string, params []string) (string, error) {
    var b bytes.Buffer
​
    parameter := p.parameters[key]
    s := parameter.cmd
​
    if parameter.flexible {
        var n int
​
        for i := 0; i < len(params); i++ {
            n += len(params[i])
        }
​
        b.Grow(len(s) + n)
​
        for i := strings.IndexByte(s, '$'); i != -1; i = strings.IndexByte(s, '$') {
            b.WriteString(s[:i])
​
            if len(s) > i+1 {
                i++
            }
​
            if s[i] == '0' {
                b.WriteString(parameter.cmd)
            } else if s[i] >= '1' && s[i] <= '9' {
                if int(s[i]-'0') <= len(params) {
                    param := params[s[i]-'0'-1]
                    if p.unsafeUserParameters == 0 {
                        if j := strings.IndexAny(param, "\\'\"`*?[]{}~$!&;()<>|#@\n"); j != -1 {
                            if unicode.IsPrint(rune(param[j])) {
                                return "", fmt.Errorf("Character \"%c\" is not allowed", param[j])
                            }
​
                            return "", fmt.Errorf("Character 0x%02x is not allowed", param[j])
                        }
                    }
                    b.WriteString(param)
                }
            } else {
                if s[i] != '$' {
                    b.WriteByte('$')
                }
                b.WriteByte(s[i])
            }
            s = s[i+1:]
        }
​