main: restrict dll search path

zx2c4 · zx2c4 · commit a313165f3bc6 · 2020-11-27T11:40:20.000+01:00
Signed-off-by: Jason A. Donenfeld &lt;Jason@zx2c4.com&gt;
diff --git a/main.go b/main.go
@@ -134,6 +134,10 @@ func pipeFromHandleArgument(handleStr string) (*os.File, error) {
 }
 
 func main() {
+	if windows.SetDllDirectory("") != nil || windows.SetDefaultDllDirectories(windows.LOAD_LIBRARY_SEARCH_SYSTEM32) != nil {
+		panic("failed to restrict dll search path")
+	}
+
 	checkForWow64()
 
 	if len(os.Args) <= 1 {

Original file line number	Diff line number	Diff line change
`@@ -134,6 +134,10 @@ func pipeFromHandleArgument(handleStr string) (*os.File, error) {`
`134`	`134`	`}`
`135`	`135`
`136`	`136`	`func main() {`
	`137`	`+ if windows.SetDllDirectory("") != nil \|\| windows.SetDefaultDllDirectories(windows.LOAD_LIBRARY_SEARCH_SYSTEM32) != nil {`
	`138`	`+ panic("failed to restrict dll search path")`
	`139`	`+ }`
	`140`	`+`
`137`	`141`	`checkForWow64()`
`138`	`142`
`139`	`143`	`if len(os.Args) <= 1 {`