From 60231b10b0fa77355be9f7034c7624e651b2d723 Mon Sep 17 00:00:00 2001
From: Pagran <67878280+pagran@users.noreply.github.com>
Date: Thu, 13 Aug 2020 23:28:21 +0300
Subject: [PATCH] Apply file name protection only for transformed code

---
 main.go | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/main.go b/main.go
index 6ed6e4a..527b0cf 100644
--- a/main.go
+++ b/main.go
@@ -483,6 +483,7 @@ func transformCompile(args []string) ([]string, error) {
 
 	// TODO: randomize the order and names of the files
 	for i, file := range files {
+		protectFileName := false
 		origName := filepath.Base(filepath.Clean(paths[i]))
 		name := origName
 		switch {
@@ -509,6 +510,7 @@ func transformCompile(args []string) ([]string, error) {
 			// messy.
 			name = "_cgo_" + name
 		default:
+			protectFileName = true
 			file = transformLineInfo(i, file)
 			file = transformGo(file, info, blacklist)
 			name = fmt.Sprintf("z%d.go", i)
@@ -537,6 +539,13 @@ func transformCompile(args []string) ([]string, error) {
 			printWriter = io.MultiWriter(tempFile, debugFile)
 		}
 
+		if protectFileName {
+			// File name leak protection
+			_, err = printWriter.Write([]byte("//line :1\n"))
+			if err != nil {
+				return nil, err
+			}
+		}
 		if err := printConfig.Fprint(printWriter, fset, file); err != nil {
 			return nil, err
 		}