85 lines
2.0 KiB
Go
85 lines
2.0 KiB
Go
|
package dbus
|
||
|
|
|
||
|
|
import "net/url"
|
||
|
|
|
||
|
|
// EscapeBusAddressValue implements a requirement to escape the values
|
||
|
|
// in D-Bus server addresses, as defined by the D-Bus specification at
|
||
|
|
// https://dbus.freedesktop.org/doc/dbus-specification.html#addresses.
|
||
|
|
func EscapeBusAddressValue(val string) string {
|
||
|
|
toEsc := strNeedsEscape(val)
|
||
|
|
if toEsc == 0 {
|
||
|
|
// Avoid unneeded allocation/copying.
|
||
|
|
return val
|
||
|
|
}
|
||
|
|
|
||
|
|
// Avoid allocation for short paths.
|
||
|
|
var buf [64]byte
|
||
|
|
var out []byte
|
||
|
|
// Every to-be-escaped byte needs 2 extra bytes.
|
||
|
|
required := len(val) + 2*toEsc
|
||
|
|
if required <= len(buf) {
|
||
|
|
out = buf[:required]
|
||
|
|
} else {
|
||
|
|
out = make([]byte, required)
|
||
|
|
}
|
||
|
|
|
||
|
|
j := 0
|
||
|
|
for i := 0; i < len(val); i++ {
|
||
|
|
if ch := val[i]; needsEscape(ch) {
|
||
|
|
// Convert ch to %xx, where xx is hex value.
|
||
|
|
out[j] = '%'
|
||
|
|
out[j+1] = hexchar(ch >> 4)
|
||
|
|
out[j+2] = hexchar(ch & 0x0F)
|
||
|
|
j += 3
|
||
|
|
} else {
|
||
|
|
out[j] = ch
|
||
|
|
j++
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return string(out)
|
||
|
|
}
|
||
|
|
|
||
|
|
// UnescapeBusAddressValue unescapes values in D-Bus server addresses,
|
||
|
|
// as defined by the D-Bus specification at
|
||
|
|
// https://dbus.freedesktop.org/doc/dbus-specification.html#addresses.
|
||
|
|
func UnescapeBusAddressValue(val string) (string, error) {
|
||
|
|
// Looks like url.PathUnescape does exactly what is required.
|
||
|
|
return url.PathUnescape(val)
|
||
|
|
}
|
||
|
|
|
||
|
|
// hexchar returns an octal representation of a n, where n < 16.
|
||
|
|
// For invalid values of n, the function panics.
|
||
|
|
func hexchar(n byte) byte {
|
||
|
|
const hex = "0123456789abcdef"
|
||
|
|
|
||
|
|
// For n >= len(hex), runtime will panic.
|
||
|
|
return hex[n]
|
||
|
|
}
|
||
|
|
|
||
|
|
// needsEscape tells if a byte is NOT one of optionally-escaped bytes.
|
||
|
|
func needsEscape(c byte) bool {
|
||
|
|
if 'a' <= c && c <= 'z' || 'A' <= c && c <= 'Z' || '0' <= c && c <= '9' {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
switch c {
|
||
|
|
case '-', '_', '/', '\\', '.', '*':
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
// strNeedsEscape tells how many bytes in the string need escaping.
|
||
|
|
func strNeedsEscape(val string) int {
|
||
|
|
count := 0
|
||
|
|
|
||
|
|
for i := 0; i < len(val); i++ {
|
||
|
|
if needsEscape(val[i]) {
|
||
|
|
count++
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return count
|
||
|
|
}
|