support for the -stdlib flag

internal/client/daemon.go

@@ -11,6 +11,7 @@ import (
 	"os/user"
 	"path/filepath"
 	"runtime"
+	"strings"
 	"sync"
 	"syscall"
 	"time"
@@ -274,15 +275,17 @@ func (daemon *Daemon) FallbackToLocalCxx(req DaemonSockRequest, reason error) Da
 	return reply
 }
 
-func (daemon *Daemon) GetOrCreateIncludesCache(cxxName string) *IncludesCache {
+func (daemon *Daemon) GetOrCreateIncludesCache(cxxName string, cxxArgs []string) *IncludesCache {
+	cxxCacheName := cxxName + strings.Join(cxxArgs, " ")
+
 	daemon.mu.Lock()
-	includesCache := daemon.includesCache[cxxName]
+	includesCache := daemon.includesCache[cxxCacheName]
 	if includesCache == nil {
 		var err error
-		if includesCache, err = MakeIncludesCache(cxxName); err != nil {
+		if includesCache, err = MakeIncludesCache(cxxName, cxxArgs); err != nil {
 			logClient.Error("failed to calc default include dirs for", cxxName, err)
 		}
-		daemon.includesCache[cxxName] = includesCache
+		daemon.includesCache[cxxCacheName] = includesCache
 	}
 	daemon.mu.Unlock()
 	return includesCache

internal/client/includes-cache.go

@@ -27,8 +27,8 @@ type IncludesCache struct {
 	mu sync.RWMutex
 }
 
-func MakeIncludesCache(cxxName string) (*IncludesCache, error) {
-	cxxDefIDirs, err := GetDefaultCxxIncludeDirsOnLocal(cxxName)
+func MakeIncludesCache(cxxName string, cxxArgs []string) (*IncludesCache, error) {
+	cxxDefIDirs, err := GetDefaultCxxIncludeDirsOnLocal(cxxName, cxxArgs)
 
 	return &IncludesCache{
 		cxxName:         cxxName,

internal/client/includes-collector.go

@@ -120,8 +120,12 @@ func CollectDependentIncludesByCxxM(includesCache *IncludesCache, cwd string, cx
 // GetDefaultCxxIncludeDirsOnLocal retrieves default include dirs on a local machine.
 // This is done by -Wp,-v option for a no input file.
 // This result is cached once nocc-daemon is started.
-func GetDefaultCxxIncludeDirsOnLocal(cxxName string) (IncludeDirs, error) {
-	cxxWpCommand := exec.Command(cxxName, "-Wp,-v", "-x", "c++", "/dev/null", "-fsyntax-only")
+func GetDefaultCxxIncludeDirsOnLocal(cxxName string, cxxArgs []string) (IncludeDirs, error) {
+	cxxCmdLine := make([]string, 0, len(cxxArgs)+5)
+	cxxCmdLine = append(cxxCmdLine, cxxArgs...)
+	cxxCmdLine = append(cxxCmdLine, "-Wp,-v", "-x", "c++", "/dev/null", "-fsyntax-only")
+
+	cxxWpCommand := exec.Command(cxxName, cxxCmdLine...)
 	var cxxWpStderr bytes.Buffer
 	cxxWpCommand.Stderr = &cxxWpStderr
 	if err := cxxWpCommand.Run(); err != nil {

internal/client/invocation.go

@@ -27,12 +27,13 @@ type Invocation struct {
 	sessionID  uint32    // incremental while a daemon is alive
 
 	// cmdLine is parsed to the following fields:
-	cppInFile  string      // input file as specified in cmd line (.cpp for compilation, .h for pch generation)
-	objOutFile string      // output file as specified in cmd line (.o for compilation, .gch/.pch for pch generation)
-	cxxName    string      // g++ / clang / etc.
-	cxxArgs    []string    // args like -Wall, -fpch-preprocess and many more, except:
-	cxxIDirs   IncludeDirs // -I / -iquote / -isystem go here
-	depsFlags  DepCmdFlags // -MD -MF file and others, used for .d files generation (not passed to server)
+	cppInFile       string      // input file as specified in cmd line (.cpp for compilation, .h for pch generation)
+	objOutFile      string      // output file as specified in cmd line (.o for compilation, .gch/.pch for pch generation)
+	cxxName         string      // g++ / clang / etc.
+	cxxArgs         []string    // args like -Wall, -fpch-preprocess and many more, except:
+	cxxArgsIncludes []string    // TODO
+	cxxIDirs        IncludeDirs // -I / -iquote / -isystem go here
+	depsFlags       DepCmdFlags // -MD -MF file and others, used for .d files generation (not passed to server)
 
 	waitUploads int32 // files still waiting for upload to finish; 0 releases wgUpload; see Invocation.DoneUploadFile
 	doneRecv    int32 // 1 if o file received or failed receiving; 1 releases wgRecv; see Invocation.DoneRecvObj
@@ -73,13 +74,14 @@ func pathAbs(cwd string, relPath string) string {
 
 func ParseCmdLineInvocation(daemon *Daemon, cwd string, cmdLine []string) (invocation *Invocation) {
 	invocation = &Invocation{
-		createTime:    time.Now(),
-		sessionID:     atomic.AddUint32(&daemon.totalInvocations, 1),
-		cxxName:       cmdLine[0],
-		cxxArgs:       make([]string, 0, 10),
-		cxxIDirs:      MakeIncludeDirs(),
-		summary:       MakeInvocationSummary(),
-		includesCache: daemon.GetOrCreateIncludesCache(cmdLine[0]),
+		createTime:      time.Now(),
+		sessionID:       atomic.AddUint32(&daemon.totalInvocations, 1),
+		cxxName:         cmdLine[0],
+		cxxArgs:         make([]string, 0, 10),
+		cxxArgsIncludes: make([]string, 0, 1),
+		cxxIDirs:        MakeIncludeDirs(),
+		summary:         MakeInvocationSummary(),
+		includesCache:   nil, // TODO
 	}
 
 	parseArgFile := func(key string, arg string, argIndex *int) (string, bool) {
@@ -113,6 +115,14 @@ func ParseCmdLineInvocation(daemon *Daemon, cwd string, cmdLine []string) (invoc
 		return ""
 	}
 
+	parsePartsArgString := func(arg string) string {
+		parts := strings.SplitN(arg, "=", 2)
+		if len(parts) != 2 {
+			return arg
+		}
+		return parts[1]
+	}
+
 	for i := 1; i < len(cmdLine); i++ {
 		arg := cmdLine[i]
 		if len(arg) == 0 {
@@ -183,6 +193,9 @@ func ParseCmdLineInvocation(daemon *Daemon, cwd string, cmdLine []string) (invoc
 				invocation.cxxArgs = append(invocation.cxxArgs, "-Xclang", xArg)
 				i++
 				continue
+			} else if strings.HasPrefix(arg, "-stdlib=") {
+				invocation.cxxArgsIncludes = append(invocation.cxxArgsIncludes, parsePartsArgString(arg))
+				continue
 			}
 		} else if isSourceFileName(arg) || isHeaderFileName(arg) {
 			if invocation.cppInFile != "" {
@@ -198,6 +211,8 @@ func ParseCmdLineInvocation(daemon *Daemon, cwd string, cmdLine []string) (invoc
 		invocation.cxxArgs = append(invocation.cxxArgs, arg)
 	}
 
+	invocation.includesCache = daemon.GetOrCreateIncludesCache(invocation.cxxName, invocation.cxxArgsIncludes)
+
 	if invocation.err != nil {
 		return
 	}