X-Git-Url: http://www.git.cypherpunks.ru/?p=goredo.git;a=blobdiff_plain;f=main.go;h=4de78be52fa786830cfbb92862c1c495cd59140a;hp=873582afd0a942f66c02a70082f372ee7dd1cde6;hb=314f58ec690c7321535d6718e8d3a0ecb4cac019;hpb=150988618524bf17378bd8770231cda7c9bbb43a

diff --git a/main.go b/main.go
index 873582a..4de78be 100644
--- a/main.go
+++ b/main.go
@@ -1,5 +1,5 @@
 /*
-goredo -- redo implementation on pure Go
+goredo -- djb's redo implementation on pure Go
 Copyright (C) 2020-2021 Sergey Matveev <stargrave@stargrave.org>
 
 This program is free software: you can redistribute it and/or modify
@@ -18,6 +18,8 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 package main
 
 import (
+	"bufio"
+	"bytes"
 	"crypto/rand"
 	"flag"
 	"fmt"
@@ -30,6 +32,7 @@ import (
 	"strconv"
 
 	"go.cypherpunks.ru/recfile"
+	"golang.org/x/sys/unix"
 )
 
 var (
@@ -96,11 +99,12 @@ func main() {
 	var err error
 	Cwd, err = os.Getwd()
 	if err != nil {
-		panic(err)
+		log.Fatalln(err)
 	}
 
 	NoColor = os.Getenv(EnvNoColor) != ""
 	NoSync = os.Getenv(EnvNoSync) == "1"
+	InodeTrust = os.Getenv(EnvInodeNoTrust) == ""
 
 	TopDir = os.Getenv(EnvTopDir)
 	if TopDir == "" {
@@ -158,6 +162,46 @@ func main() {
 	}
 
 	// Those are internal envs
+	FdOODTgts, err = ioutil.TempFile("", "ood-tgts")
+	if err != nil {
+		panic(err)
+	}
+	if err = os.Remove(FdOODTgts.Name()); err != nil {
+		panic(err)
+	}
+	FdOODTgtsLock, err = ioutil.TempFile("", "ood-tgts.lock")
+	if err != nil {
+		panic(err)
+	}
+	if err = os.Remove(FdOODTgtsLock.Name()); err != nil {
+		panic(err)
+	}
+
+	if v := os.Getenv(EnvOODTgtsFd); v != "" {
+		fd := mustParseFd(v, EnvOODTgtsFd)
+		fdLock := mustParseFd(v, EnvOODTgtsLockFd)
+		if err = unix.Flock(int(fdLock.Fd()), unix.LOCK_EX); err != nil {
+			panic(err)
+		}
+		if _, err = fd.Seek(0, os.SEEK_SET); err != nil {
+			panic(err)
+		}
+		tgtsRaw, err := ioutil.ReadAll(bufio.NewReader(fd))
+		if err != nil {
+			panic(err)
+		}
+		unix.Flock(int(fdLock.Fd()), unix.LOCK_UN)
+		OODTgts = make(map[string]struct{})
+		for _, tgtRaw := range bytes.Split(tgtsRaw, []byte{0}) {
+			t := string(tgtRaw)
+			if t == "" {
+				continue
+			}
+			OODTgts[t] = struct{}{}
+			trace(CDebug, "ood: known to be: %s", t)
+		}
+	}
+
 	StderrPrefix = os.Getenv(EnvStderrPrefix)
 	if v := os.Getenv(EnvLevel); v != "" {
 		Level, err = strconv.Atoi(v)
@@ -168,16 +212,18 @@ func main() {
 			panic("negative " + EnvLevel)
 		}
 	}
+
 	var fdDep *os.File
 	if v := os.Getenv(EnvDepFd); v != "" {
 		fdDep = mustParseFd(v, EnvDepFd)
 	}
-	BuildUUID = os.Getenv(EnvBuildUUID)
+
 	tgts := flag.Args()
+	BuildUUID = os.Getenv(EnvBuildUUID)
 	if BuildUUID == "" {
 		raw := new([16]byte)
 		if _, err = io.ReadFull(rand.Reader, raw[:]); err != nil {
-			panic(err)
+			log.Fatalln(err)
 		}
 		raw[6] = (raw[6] & 0x0F) | uint8(4<<4) // version 4
 		BuildUUID = fmt.Sprintf(
@@ -188,6 +234,7 @@ func main() {
 			tgts = []string{"all"}
 		}
 	}
+
 	statusInit()
 
 	for i, tgt := range tgts {
@@ -263,17 +310,19 @@ CmdSwitch:
 			log.Fatalln("single target expected")
 		}
 		var fdTmp *os.File
-		fdTmp, err = ioutil.TempFile("", "")
+		fdTmp, err = ioutil.TempFile("", "whichdo")
 		if err != nil {
 			break
 		}
-		os.Remove(fdTmp.Name())
+		if err = os.Remove(fdTmp.Name()); err != nil {
+			break
+		}
 		cwd, tgt := cwdAndTgt(tgts[0])
 		doFile, upLevels, err := findDo(fdTmp, cwd, tgt)
 		if err != nil {
 			break
 		}
-		_, err = fdTmp.Seek(0, 0)
+		_, err = fdTmp.Seek(0, os.SEEK_SET)
 		if err != nil {
 			break
 		}