1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120
|
package shellescape_test
import (
"bufio"
"fmt"
"strings"
"github.com/google/shlex"
"al.essio.dev/pkg/shellescape"
)
func ExampleQuote() {
filename := "myfile; rm -rf /"
prog := "/bin/ls -lh"
unsafe := strings.Join([]string{prog, filename}, " ")
safe := strings.Join([]string{prog, shellescape.Quote(filename)}, " ")
fmt.Println("unsafe:", unsafe)
fmt.Println("safe:", safe)
for i, part := range strings.Split(unsafe, " ") {
fmt.Printf("unsafe[%d] = %s\n", i, part)
}
for i, part := range strings.Split(safe, " ") {
fmt.Printf("safe[%d] = %s\n", i, part)
}
// Output:
// unsafe: /bin/ls -lh myfile; rm -rf /
// safe: /bin/ls -lh 'myfile; rm -rf /'
// unsafe[0] = /bin/ls
// unsafe[1] = -lh
// unsafe[2] = myfile;
// unsafe[3] = rm
// unsafe[4] = -rf
// unsafe[5] = /
// safe[0] = /bin/ls
// safe[1] = -lh
// safe[2] = 'myfile;
// safe[3] = rm
// safe[4] = -rf
// safe[5] = /'
}
func ExampleQuoteCommand_simple() {
filename := "filename with space"
prog := "/usr/bin/ls"
args := "-lh"
unsafe := strings.Join([]string{prog, args, filename}, " ")
safe := strings.Join([]string{prog, shellescape.QuoteCommand([]string{args, filename})}, " ")
fmt.Println("unsafe:", unsafe)
fmt.Println("safe:", safe)
// Output:
// unsafe: /usr/bin/ls -lh filename with space
// safe: /usr/bin/ls -lh 'filename with space'
}
func ExampleQuoteCommand() {
filename := "myfile; rm -rf /"
unsafe := fmt.Sprintf("ls -l %s", filename)
command := fmt.Sprintf("ls -l %s", shellescape.Quote(filename))
splitCommand, _ := shlex.Split(command)
fmt.Println("unsafe:", unsafe)
fmt.Println("command:", command)
fmt.Println("splitCommand:", splitCommand)
remoteCommandUnsafe := fmt.Sprintf("ssh host.domain %s", command)
remoteCommand := fmt.Sprintf("ssh host.domain %s", shellescape.Quote(command))
splitRemoteCommand, _ := shlex.Split(remoteCommand)
fmt.Println("remoteCommandUnsafe:", remoteCommandUnsafe)
fmt.Println("remoteCommand:", remoteCommand)
fmt.Println("splitRemoteCommand:", splitRemoteCommand)
lastSplit, _ := shlex.Split(splitRemoteCommand[2])
fmt.Println("lastSplit[0]:", lastSplit[0])
fmt.Println("lastSplit[1]:", lastSplit[1])
fmt.Println("lastSplit[2]:", lastSplit[2])
// Output:
// unsafe: ls -l myfile; rm -rf /
// command: ls -l 'myfile; rm -rf /'
// splitCommand: [ls -l myfile; rm -rf /]
// remoteCommandUnsafe: ssh host.domain ls -l 'myfile; rm -rf /'
// remoteCommand: ssh host.domain 'ls -l '"'"'myfile; rm -rf /'"'"''
// splitRemoteCommand: [ssh host.domain ls -l 'myfile; rm -rf /']
// lastSplit[0]: ls
// lastSplit[1]: -l
// lastSplit[2]: myfile; rm -rf /
}
func ExampleStripUnsafe() {
safeString := `"printable!" #$%^characters '' 12321312"`
unsafeString := "these runes shall be removed: \u0000\u0081\u001f"
fmt.Println("safe:", shellescape.StripUnsafe(safeString))
fmt.Println("unsafe:", shellescape.StripUnsafe(unsafeString))
// Output:
// safe: "printable!" #$%^characters '' 12321312"
// unsafe: these runes shall be removed:
}
func ExampleScanTokens() {
words := "'tis\x00but\x00a\x00scratch!\x00"
scanner := bufio.NewScanner(strings.NewReader(words))
scanner.Split(shellescape.ScanTokens)
for scanner.Scan() {
fmt.Println(scanner.Text())
}
// Output:
// 'tis
// but
// a
// scratch!
}
|