win

Signed-off-by: Takeshi Yoneda <takeshi@tetrate.io>
tetratelabs · Aug 18, 2022 · 3d94880 · 3d94880
1 parent ede6d29
commit 3d94880
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/internal/platform/mmap_windows.go b/internal/platform/mmap_windows.go
@@ -74,7 +74,7 @@ func mmapCodeSegmentAMD64(code io.Reader, size int) ([]byte, error) {
 
 	w := &bufWriter{underlying: mem}
 	_, err = io.CopyN(w, code, int64(size))
-	return mem, nil
+	return mem, err
 }
 
 func mmapCodeSegmentARM64(code io.Reader, size int) ([]byte, error) {
@@ -90,6 +90,9 @@ func mmapCodeSegmentARM64(code io.Reader, size int) ([]byte, error) {
 	sh.Cap = size
 	w := &bufWriter{underlying: mem}
 	_, err = io.CopyN(w, code, int64(size))
+	if err != nil {
+		return nil, err
+	}
 
 	old := uint32(windows_PAGE_READWRITE)
 	err = virtualProtect(p, uintptr(size), windows_PAGE_EXECUTE_READ, &old)