X-Git-Url: http://www.git.cypherpunks.ru/?a=blobdiff_plain;f=src%2Fsp.go;h=eca9544945b1e0d734d6b21d692cce85dc68f372;hb=2570c230757df9e463ea072d9110564b53031ce3;hp=4db358742a17b1f584aa102ac853a1c6386c6d13;hpb=0367cce2741e1ce6a89a49fd5c4e9df6005c9744;p=nncp.git

diff --git a/src/sp.go b/src/sp.go
index 4db3587..eca9544 100644
--- a/src/sp.go
+++ b/src/sp.go
@@ -27,6 +27,7 @@ import (
 	"os"
 	"path/filepath"
 	"sort"
+	"strconv"
 	"sync"
 	"time"
 
@@ -39,6 +40,7 @@ const (
 	MaxSPSize      = 1<<16 - 256
 	PartSuffix     = ".part"
 	SPHeadOverhead = 4
+	CfgDeadline    = "NNCPDEADLINE"
 )
 
 type MTHAndOffset struct {
@@ -132,6 +134,14 @@ type ConnDeadlined interface {
 }
 
 func init() {
+	if v := os.Getenv(CfgDeadline); v != "" {
+		i, err := strconv.Atoi(v)
+		if err != nil {
+			log.Fatalln("Can not convert", CfgDeadline, "to integer:", err)
+		}
+		DefaultDeadline = time.Duration(i) * time.Second
+	}
+
 	var buf bytes.Buffer
 	spHead := SPHead{Type: SPTypeHalt}
 	if _, err := xdr.Marshal(&buf, spHead); err != nil {
@@ -1104,7 +1114,7 @@ func (state *SPState) StartWorkers(
 	return nil
 }
 
-func (state *SPState) Wait() {
+func (state *SPState) Wait() bool {
 	state.wg.Wait()
 	close(state.payloads)
 	close(state.pings)
@@ -1120,12 +1130,15 @@ func (state *SPState) Wait() {
 	if txDuration > 0 {
 		state.TxSpeed = state.TxBytes / txDuration
 	}
+	nothingLeft := len(state.queueTheir) == 0
 	for _, s := range state.fds {
+		nothingLeft = false
 		s.fd.Close()
 	}
 	for pktName := range state.progressBars {
 		ProgressKill(pktName)
 	}
+	return nothingLeft
 }
 
 func (state *SPState) ProcessSP(payload []byte) ([][]byte, error) {
@@ -1451,13 +1464,15 @@ func (state *SPState) ProcessSP(payload []byte) ([][]byte, error) {
 					humanize.IBytes(uint64(fullsize)),
 				)
 			}
-			err = fd.Sync()
-			if err != nil {
-				state.Ctx.LogE("sp-file-sync", lesp, err, func(les LEs) string {
-					return logMsg(les) + ": syncing"
-				})
-				state.closeFd(filePathPart)
-				continue
+			if !NoSync {
+				err = fd.Sync()
+				if err != nil {
+					state.Ctx.LogE("sp-file-sync", lesp, err, func(les LEs) string {
+						return logMsg(les) + ": syncing"
+					})
+					state.closeFd(filePathPart)
+					continue
+				}
 			}
 			if hasherAndOffset != nil {
 				delete(state.fileHashers, filePath)