mirror of
https://github.com/SagerNet/sing-box.git
synced 2024-12-05 04:46:40 +00:00
120 lines
2.8 KiB
Go
120 lines
2.8 KiB
Go
|
package sip003
|
|||
|
|
|||
|
import (
|
|||
|
"bytes"
|
|||
|
"fmt"
|
|||
|
)
|
|||
|
|
|||
|
// mod from https://github.com/shadowsocks/v2ray-plugin/blob/master/args.go
|
|||
|
|
|||
|
// Args maps a string key to a list of values. It is similar to url.Values.
|
|||
|
type Args map[string][]string
|
|||
|
|
|||
|
// Get the first value associated with the given key. If there are any values
|
|||
|
// associated with the key, the value return has the value and ok is set to
|
|||
|
// true. If there are no values for the given key, value is "" and ok is false.
|
|||
|
// If you need access to multiple values, use the map directly.
|
|||
|
func (args Args) Get(key string) (value string, ok bool) {
|
|||
|
if args == nil {
|
|||
|
return "", false
|
|||
|
}
|
|||
|
vals, ok := args[key]
|
|||
|
if !ok || len(vals) == 0 {
|
|||
|
return "", false
|
|||
|
}
|
|||
|
return vals[0], true
|
|||
|
}
|
|||
|
|
|||
|
// Add Append value to the list of values for key.
|
|||
|
func (args Args) Add(key, value string) {
|
|||
|
args[key] = append(args[key], value)
|
|||
|
}
|
|||
|
|
|||
|
// Return the index of the next unescaped byte in s that is in the term set, or
|
|||
|
// else the length of the string if no terminators appear. Additionally return
|
|||
|
// the unescaped string up to the returned index.
|
|||
|
func indexUnescaped(s string, term []byte) (int, string, error) {
|
|||
|
var i int
|
|||
|
unesc := make([]byte, 0)
|
|||
|
for i = 0; i < len(s); i++ {
|
|||
|
b := s[i]
|
|||
|
// A terminator byte?
|
|||
|
if bytes.IndexByte(term, b) != -1 {
|
|||
|
break
|
|||
|
}
|
|||
|
if b == '\\' {
|
|||
|
i++
|
|||
|
if i >= len(s) {
|
|||
|
return 0, "", fmt.Errorf("nothing following final escape in %q", s)
|
|||
|
}
|
|||
|
b = s[i]
|
|||
|
}
|
|||
|
unesc = append(unesc, b)
|
|||
|
}
|
|||
|
return i, string(unesc), nil
|
|||
|
}
|
|||
|
|
|||
|
// ParsePluginOptions Parse a name–value mapping as from SS_PLUGIN_OPTIONS.
|
|||
|
//
|
|||
|
// "<value> is a k=v string value with options that are to be passed to the
|
|||
|
// transport. semicolons, equal signs and backslashes must be escaped
|
|||
|
// with a backslash."
|
|||
|
// Example: secret=nou;cache=/tmp/cache;secret=yes
|
|||
|
func ParsePluginOptions(s string) (opts Args, err error) {
|
|||
|
opts = make(Args)
|
|||
|
if len(s) == 0 {
|
|||
|
return
|
|||
|
}
|
|||
|
i := 0
|
|||
|
for {
|
|||
|
var key, value string
|
|||
|
var offset, begin int
|
|||
|
|
|||
|
if i >= len(s) {
|
|||
|
break
|
|||
|
}
|
|||
|
begin = i
|
|||
|
// Read the key.
|
|||
|
offset, key, err = indexUnescaped(s[i:], []byte{'=', ';'})
|
|||
|
if err != nil {
|
|||
|
return
|
|||
|
}
|
|||
|
if len(key) == 0 {
|
|||
|
err = fmt.Errorf("empty key in %q", s[begin:i])
|
|||
|
return
|
|||
|
}
|
|||
|
i += offset
|
|||
|
// End of string or no equals sign?
|
|||
|
if i >= len(s) || s[i] != '=' {
|
|||
|
opts.Add(key, "1")
|
|||
|
// Skip the semicolon.
|
|||
|
i++
|
|||
|
continue
|
|||
|
}
|
|||
|
// Skip the equals sign.
|
|||
|
i++
|
|||
|
// Read the value.
|
|||
|
offset, value, err = indexUnescaped(s[i:], []byte{';'})
|
|||
|
if err != nil {
|
|||
|
return
|
|||
|
}
|
|||
|
i += offset
|
|||
|
opts.Add(key, value)
|
|||
|
// Skip the semicolon.
|
|||
|
i++
|
|||
|
}
|
|||
|
return opts, nil
|
|||
|
}
|
|||
|
|
|||
|
// Escape backslashes and all the bytes that are in set.
|
|||
|
func backslashEscape(s string, set []byte) string {
|
|||
|
var buf bytes.Buffer
|
|||
|
for _, b := range []byte(s) {
|
|||
|
if b == '\\' || bytes.IndexByte(set, b) != -1 {
|
|||
|
buf.WriteByte('\\')
|
|||
|
}
|
|||
|
buf.WriteByte(b)
|
|||
|
}
|
|||
|
return buf.String()
|
|||
|
}
|