2014-11-16 13:13:20 -07:00
// Copyright (C) 2014 The Syncthing Authors.
2014-10-04 07:48:33 -07:00
//
2015-03-07 13:36:35 -07:00
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this file,
2017-02-08 23:52:18 -07:00
// You can obtain one at https://mozilla.org/MPL/2.0/.
2014-10-04 07:48:33 -07:00
package model
import (
2017-12-07 01:42:03 -07:00
"bytes"
2017-04-25 17:15:23 -07:00
"context"
2016-12-14 16:30:29 -07:00
"crypto/rand"
2020-07-28 02:13:15 -07:00
"errors"
2016-12-14 16:30:29 -07:00
"io"
2017-12-07 01:42:03 -07:00
"io/ioutil"
2014-10-10 16:27:17 -07:00
"os"
"path/filepath"
2019-01-25 01:52:21 -07:00
"runtime"
2020-07-28 02:13:15 -07:00
"strings"
2014-10-04 07:48:33 -07:00
"testing"
2014-12-09 16:58:58 -07:00
"time"
2014-10-04 07:48:33 -07:00
2018-02-25 02:14:02 -07:00
"github.com/syncthing/syncthing/lib/config"
2018-05-14 12:01:35 -07:00
"github.com/syncthing/syncthing/lib/events"
2016-12-14 16:30:29 -07:00
"github.com/syncthing/syncthing/lib/fs"
2017-12-07 01:42:03 -07:00
"github.com/syncthing/syncthing/lib/ignore"
2019-10-22 12:55:51 -07:00
"github.com/syncthing/syncthing/lib/osutil"
2015-09-22 10:38:46 -07:00
"github.com/syncthing/syncthing/lib/protocol"
2015-08-06 02:29:25 -07:00
"github.com/syncthing/syncthing/lib/scanner"
"github.com/syncthing/syncthing/lib/sync"
2014-10-04 07:48:33 -07:00
)
var blocks = [ ] protocol . BlockInfo {
{ Hash : [ ] uint8 { 0xfa , 0x43 , 0x23 , 0x9b , 0xce , 0xe7 , 0xb9 , 0x7c , 0xa6 , 0x2f , 0x0 , 0x7c , 0xc6 , 0x84 , 0x87 , 0x56 , 0xa , 0x39 , 0xe1 , 0x9f , 0x74 , 0xf3 , 0xdd , 0xe7 , 0x48 , 0x6d , 0xb3 , 0xf9 , 0x8d , 0xf8 , 0xe4 , 0x71 } } , // Zero'ed out block
{ Offset : 0 , Size : 0x20000 , Hash : [ ] uint8 { 0x7e , 0xad , 0xbc , 0x36 , 0xae , 0xbb , 0xcf , 0x74 , 0x43 , 0xe2 , 0x7a , 0x5a , 0x4b , 0xb8 , 0x5b , 0xce , 0xe6 , 0x9e , 0x1e , 0x10 , 0xf9 , 0x8a , 0xbc , 0x77 , 0x95 , 0x2 , 0x29 , 0x60 , 0x9e , 0x96 , 0xae , 0x6c } } ,
{ Offset : 131072 , Size : 0x20000 , Hash : [ ] uint8 { 0x3c , 0xc4 , 0x20 , 0xf4 , 0xb , 0x2e , 0xcb , 0xb9 , 0x5d , 0xce , 0x34 , 0xa8 , 0xc3 , 0x92 , 0xea , 0xf3 , 0xda , 0x88 , 0x33 , 0xee , 0x7a , 0xb6 , 0xe , 0xf1 , 0x82 , 0x5e , 0xb0 , 0xa9 , 0x26 , 0xa9 , 0xc0 , 0xef } } ,
{ Offset : 262144 , Size : 0x20000 , Hash : [ ] uint8 { 0x76 , 0xa8 , 0xc , 0x69 , 0xd7 , 0x5c , 0x52 , 0xfd , 0xdf , 0x55 , 0xef , 0x44 , 0xc1 , 0xd6 , 0x25 , 0x48 , 0x4d , 0x98 , 0x48 , 0x4d , 0xaa , 0x50 , 0xf6 , 0x6b , 0x32 , 0x47 , 0x55 , 0x81 , 0x6b , 0xed , 0xee , 0xfb } } ,
{ Offset : 393216 , Size : 0x20000 , Hash : [ ] uint8 { 0x44 , 0x1e , 0xa4 , 0xf2 , 0x8d , 0x1f , 0xc3 , 0x1b , 0x9d , 0xa5 , 0x18 , 0x5e , 0x59 , 0x1b , 0xd8 , 0x5c , 0xba , 0x7d , 0xb9 , 0x8d , 0x70 , 0x11 , 0x5c , 0xea , 0xa1 , 0x57 , 0x4d , 0xcb , 0x3c , 0x5b , 0xf8 , 0x6c } } ,
{ Offset : 524288 , Size : 0x20000 , Hash : [ ] uint8 { 0x8 , 0x40 , 0xd0 , 0x5e , 0x80 , 0x0 , 0x0 , 0x7c , 0x8b , 0xb3 , 0x8b , 0xf7 , 0x7b , 0x23 , 0x26 , 0x28 , 0xab , 0xda , 0xcf , 0x86 , 0x8f , 0xc2 , 0x8a , 0x39 , 0xc6 , 0xe6 , 0x69 , 0x59 , 0x97 , 0xb6 , 0x1a , 0x43 } } ,
{ Offset : 655360 , Size : 0x20000 , Hash : [ ] uint8 { 0x38 , 0x8e , 0x44 , 0xcb , 0x30 , 0xd8 , 0x90 , 0xf , 0xce , 0x7 , 0x4b , 0x58 , 0x86 , 0xde , 0xce , 0x59 , 0xa2 , 0x46 , 0xd2 , 0xf9 , 0xba , 0xaf , 0x35 , 0x87 , 0x38 , 0xdf , 0xd2 , 0xd , 0xf9 , 0x45 , 0xed , 0x91 } } ,
{ Offset : 786432 , Size : 0x20000 , Hash : [ ] uint8 { 0x32 , 0x28 , 0xcd , 0xf , 0x37 , 0x21 , 0xe5 , 0xd4 , 0x1e , 0x58 , 0x87 , 0x73 , 0x8e , 0x36 , 0xdf , 0xb2 , 0x70 , 0x78 , 0x56 , 0xc3 , 0x42 , 0xff , 0xf7 , 0x8f , 0x37 , 0x95 , 0x0 , 0x26 , 0xa , 0xac , 0x54 , 0x72 } } ,
{ Offset : 917504 , Size : 0x20000 , Hash : [ ] uint8 { 0x96 , 0x6b , 0x15 , 0x6b , 0xc4 , 0xf , 0x19 , 0x18 , 0xca , 0xbb , 0x5f , 0xd6 , 0xbb , 0xa2 , 0xc6 , 0x2a , 0xac , 0xbb , 0x8a , 0xb9 , 0xce , 0xec , 0x4c , 0xdb , 0x78 , 0xec , 0x57 , 0x5d , 0x33 , 0xf9 , 0x8e , 0xaf } } ,
}
2015-09-04 03:01:00 -07:00
var folders = [ ] string { "default" }
2018-01-14 07:30:11 -07:00
var diffTestData = [ ] struct {
a string
b string
s int
d [ ] protocol . BlockInfo
} {
{ "contents" , "contents" , 1024 , [ ] protocol . BlockInfo { } } ,
{ "" , "" , 1024 , [ ] protocol . BlockInfo { } } ,
{ "contents" , "contents" , 3 , [ ] protocol . BlockInfo { } } ,
{ "contents" , "cantents" , 3 , [ ] protocol . BlockInfo { { Offset : 0 , Size : 3 } } } ,
{ "contents" , "contants" , 3 , [ ] protocol . BlockInfo { { Offset : 3 , Size : 3 } } } ,
{ "contents" , "cantants" , 3 , [ ] protocol . BlockInfo { { Offset : 0 , Size : 3 } , { Offset : 3 , Size : 3 } } } ,
{ "contents" , "" , 3 , [ ] protocol . BlockInfo { { Offset : 0 , Size : 0 } } } ,
{ "" , "contents" , 3 , [ ] protocol . BlockInfo { { Offset : 0 , Size : 3 } , { Offset : 3 , Size : 3 } , { Offset : 6 , Size : 2 } } } ,
{ "con" , "contents" , 3 , [ ] protocol . BlockInfo { { Offset : 3 , Size : 3 } , { Offset : 6 , Size : 2 } } } ,
{ "contents" , "con" , 3 , nil } ,
{ "contents" , "cont" , 3 , [ ] protocol . BlockInfo { { Offset : 3 , Size : 1 } } } ,
{ "cont" , "contents" , 3 , [ ] protocol . BlockInfo { { Offset : 3 , Size : 3 } , { Offset : 6 , Size : 2 } } } ,
}
2019-03-04 05:27:10 -07:00
func setupFile ( filename string , blockNumbers [ ] int ) protocol . FileInfo {
2014-10-10 16:27:17 -07:00
// Create existing file
2015-12-21 14:47:26 -07:00
existingBlocks := make ( [ ] protocol . BlockInfo , len ( blockNumbers ) )
for i := range blockNumbers {
existingBlocks [ i ] = blocks [ blockNumbers [ i ] ]
}
return protocol . FileInfo {
2016-07-04 03:40:29 -07:00
Name : filename ,
Blocks : existingBlocks ,
2014-10-04 07:48:33 -07:00
}
2015-12-21 14:47:26 -07:00
}
2014-10-04 07:48:33 -07:00
2019-03-07 07:15:14 -07:00
func createFile ( t * testing . T , name string , fs fs . Filesystem ) protocol . FileInfo {
t . Helper ( )
f , err := fs . Create ( name )
2019-03-09 11:45:36 -07:00
must ( t , err )
2019-03-07 07:15:14 -07:00
f . Close ( )
fi , err := fs . Stat ( name )
2019-03-09 11:45:36 -07:00
must ( t , err )
2019-03-07 07:15:14 -07:00
file , err := scanner . CreateFileInfo ( fi , name , fs )
2019-03-09 11:45:36 -07:00
must ( t , err )
2019-03-07 07:15:14 -07:00
return file
}
2020-05-05 23:34:54 -07:00
// Sets up a folder and model, but makes sure the services aren't actually running.
2020-12-21 04:59:22 -07:00
func setupSendReceiveFolder ( t testing . TB , files ... protocol . FileInfo ) ( * testModel , * sendReceiveFolder ) {
2020-05-05 23:34:54 -07:00
w , fcfg := tmpDefaultWrapper ( )
2020-11-17 05:19:04 -07:00
// Initialise model and stop immediately.
2020-12-21 04:59:22 -07:00
model := setupModel ( t , w )
2020-11-17 05:19:04 -07:00
model . cancel ( )
<- model . stopped
2020-05-05 23:34:54 -07:00
f := model . folderRunners [ fcfg . ID ] . ( * sendReceiveFolder )
2020-07-28 02:13:15 -07:00
f . tempPullErrors = make ( map [ string ] string )
2020-05-05 23:34:54 -07:00
f . ctx = context . Background ( )
2017-04-01 02:58:06 -07:00
2019-04-07 04:29:17 -07:00
// Update index
if files != nil {
f . updateLocalsFromScanning ( files )
}
2019-03-08 13:29:09 -07:00
return model , f
2015-12-21 14:47:26 -07:00
}
2020-11-17 05:19:04 -07:00
func cleanupSRFolder ( f * sendReceiveFolder , m * testModel ) {
2019-08-15 07:29:37 -07:00
os . Remove ( m . cfg . ConfigPath ( ) )
2019-10-22 12:55:51 -07:00
os . RemoveAll ( f . Filesystem ( ) . URI ( ) )
2019-08-15 07:29:37 -07:00
}
2015-12-21 14:47:26 -07:00
// Layout of the files: (indexes from the above array)
// 12345678 - Required file
// 02005008 - Existing file (currently in the index)
// 02340070 - Temp file on the disk
func TestHandleFile ( t * testing . T ) {
// After the diff between required and existing we should:
// Copy: 2, 5, 8
// Pull: 1, 3, 4, 6, 7
2014-10-04 07:48:33 -07:00
2015-12-21 14:47:26 -07:00
existingBlocks := [ ] int { 0 , 2 , 0 , 0 , 5 , 0 , 0 , 8 }
2019-03-04 05:27:10 -07:00
existingFile := setupFile ( "filex" , existingBlocks )
2015-12-21 14:47:26 -07:00
requiredFile := existingFile
requiredFile . Blocks = blocks [ 1 : ]
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t , existingFile )
2019-08-15 07:29:37 -07:00
defer cleanupSRFolder ( f , m )
2019-02-12 08:04:04 -07:00
2014-10-08 15:41:23 -07:00
copyChan := make ( chan copyBlocksState , 1 )
2014-10-04 07:48:33 -07:00
2020-01-21 10:23:08 -07:00
f . handleFile ( requiredFile , f . fset . Snapshot ( ) , copyChan )
2014-10-04 07:48:33 -07:00
// Receive the results
toCopy := <- copyChan
2014-10-08 15:41:23 -07:00
if len ( toCopy . blocks ) != 8 {
t . Errorf ( "Unexpected count of copy blocks: %d != 8" , len ( toCopy . blocks ) )
2014-10-04 07:48:33 -07:00
}
2016-04-15 03:59:41 -07:00
for _ , block := range blocks [ 1 : ] {
found := false
for _ , toCopyBlock := range toCopy . blocks {
if string ( toCopyBlock . Hash ) == string ( block . Hash ) {
found = true
break
}
}
if ! found {
t . Errorf ( "Did not find block %s" , block . String ( ) )
2014-10-04 07:48:33 -07:00
}
}
}
func TestHandleFileWithTemp ( t * testing . T ) {
// After diff between required and existing we should:
// Copy: 2, 5, 8
// Pull: 1, 3, 4, 6, 7
// After dropping out blocks already on the temp file we should:
// Copy: 5, 8
// Pull: 1, 6
2015-12-21 14:47:26 -07:00
existingBlocks := [ ] int { 0 , 2 , 0 , 0 , 5 , 0 , 0 , 8 }
2019-03-04 05:27:10 -07:00
existingFile := setupFile ( "file" , existingBlocks )
2014-10-04 07:48:33 -07:00
requiredFile := existingFile
requiredFile . Blocks = blocks [ 1 : ]
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t , existingFile )
2019-08-15 07:29:37 -07:00
defer cleanupSRFolder ( f , m )
2019-02-12 08:04:04 -07:00
if _ , err := prepareTmpFile ( f . Filesystem ( ) ) ; err != nil {
t . Fatal ( err )
}
2014-10-08 15:41:23 -07:00
copyChan := make ( chan copyBlocksState , 1 )
2014-10-04 07:48:33 -07:00
2020-01-21 10:23:08 -07:00
f . handleFile ( requiredFile , f . fset . Snapshot ( ) , copyChan )
2014-10-04 07:48:33 -07:00
// Receive the results
toCopy := <- copyChan
2014-10-08 15:41:23 -07:00
if len ( toCopy . blocks ) != 4 {
t . Errorf ( "Unexpected count of copy blocks: %d != 4" , len ( toCopy . blocks ) )
2014-10-04 07:48:33 -07:00
}
2016-04-15 03:59:41 -07:00
for _ , idx := range [ ] int { 1 , 5 , 6 , 8 } {
found := false
block := blocks [ idx ]
for _ , toCopyBlock := range toCopy . blocks {
if string ( toCopyBlock . Hash ) == string ( block . Hash ) {
found = true
break
}
}
if ! found {
t . Errorf ( "Did not find block %s" , block . String ( ) )
2014-10-04 07:48:33 -07:00
}
}
}
2014-10-10 16:27:17 -07:00
func TestCopierFinder ( t * testing . T ) {
2020-08-06 22:47:48 -07:00
methods := [ ] fs . CopyRangeMethod { fs . CopyRangeMethodStandard , fs . CopyRangeMethodAllWithFallback }
if runtime . GOOS == "linux" {
methods = append ( methods , fs . CopyRangeMethodSendFile )
}
for _ , method := range methods {
2020-08-03 12:54:42 -07:00
t . Run ( method . String ( ) , func ( t * testing . T ) {
// After diff between required and existing we should:
// Copy: 1, 2, 3, 4, 6, 7, 8
// Since there is no existing file, nor a temp file
2014-10-10 16:27:17 -07:00
2020-08-03 12:54:42 -07:00
// After dropping out blocks found locally:
// Pull: 1, 5, 6, 8
2014-10-10 16:27:17 -07:00
2020-08-03 12:54:42 -07:00
tempFile := fs . TempName ( "file2" )
2014-10-10 16:27:17 -07:00
2020-08-03 12:54:42 -07:00
existingBlocks := [ ] int { 0 , 2 , 3 , 4 , 0 , 0 , 7 , 0 }
existingFile := setupFile ( fs . TempName ( "file" ) , existingBlocks )
existingFile . Size = 1
requiredFile := existingFile
requiredFile . Blocks = blocks [ 1 : ]
requiredFile . Name = "file2"
2014-10-10 16:27:17 -07:00
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t , existingFile )
2020-08-03 12:54:42 -07:00
f . CopyRangeMethod = method
2019-02-12 08:04:04 -07:00
2020-08-03 12:54:42 -07:00
defer cleanupSRFolder ( f , m )
2019-02-12 08:04:04 -07:00
2020-08-03 12:54:42 -07:00
if _ , err := prepareTmpFile ( f . Filesystem ( ) ) ; err != nil {
t . Fatal ( err )
}
2014-10-10 16:27:17 -07:00
2020-08-03 12:54:42 -07:00
copyChan := make ( chan copyBlocksState )
pullChan := make ( chan pullBlockState , 4 )
finisherChan := make ( chan * sharedPullerState , 1 )
2014-10-10 16:27:17 -07:00
2020-08-03 12:54:42 -07:00
// Run a single fetcher routine
go f . copierRoutine ( copyChan , pullChan , finisherChan )
defer close ( copyChan )
2014-10-10 16:27:17 -07:00
2020-08-03 12:54:42 -07:00
f . handleFile ( requiredFile , f . fset . Snapshot ( ) , copyChan )
2019-11-08 10:53:51 -07:00
2020-08-03 12:54:42 -07:00
timeout := time . After ( 10 * time . Second )
pulls := make ( [ ] pullBlockState , 4 )
for i := 0 ; i < 4 ; i ++ {
select {
case pulls [ i ] = <- pullChan :
case <- timeout :
t . Fatalf ( "Timed out before receiving all 4 states on pullChan (already got %v)" , i )
}
}
var finish * sharedPullerState
select {
case finish = <- finisherChan :
case <- timeout :
t . Fatal ( "Timed out before receiving 4 states on pullChan" )
}
2014-10-10 16:27:17 -07:00
2020-08-03 12:54:42 -07:00
defer cleanupSharedPullerState ( finish )
2014-10-10 16:27:17 -07:00
2020-08-03 12:54:42 -07:00
select {
case <- pullChan :
t . Fatal ( "Pull channel has data to be read" )
case <- finisherChan :
t . Fatal ( "Finisher channel has data to be read" )
default :
2016-04-15 03:59:41 -07:00
}
2014-10-10 16:27:17 -07:00
2020-08-03 12:54:42 -07:00
// Verify that the right blocks went into the pull list.
// They are pulled in random order.
for _ , idx := range [ ] int { 1 , 5 , 6 , 8 } {
found := false
block := blocks [ idx ]
for _ , pulledBlock := range pulls {
if string ( pulledBlock . block . Hash ) == string ( block . Hash ) {
found = true
break
}
}
if ! found {
t . Errorf ( "Did not find block %s" , block . String ( ) )
}
if string ( finish . file . Blocks [ idx - 1 ] . Hash ) != string ( blocks [ idx ] . Hash ) {
t . Errorf ( "Block %d mismatch: %s != %s" , idx , finish . file . Blocks [ idx - 1 ] . String ( ) , blocks [ idx ] . String ( ) )
}
}
2014-10-10 16:27:17 -07:00
2020-08-03 12:54:42 -07:00
// Verify that the fetched blocks have actually been written to the temp file
blks , err := scanner . HashFile ( context . TODO ( ) , f . Filesystem ( ) , tempFile , protocol . MinBlockSize , nil , false )
if err != nil {
t . Log ( err )
}
for _ , eq := range [ ] int { 2 , 3 , 4 , 7 } {
if string ( blks [ eq - 1 ] . Hash ) != string ( blocks [ eq ] . Hash ) {
t . Errorf ( "Block %d mismatch: %s != %s" , eq , blks [ eq - 1 ] . String ( ) , blocks [ eq ] . String ( ) )
}
}
} )
2014-10-10 16:27:17 -07:00
}
}
2014-10-22 07:24:11 -07:00
2016-12-14 16:30:29 -07:00
func TestWeakHash ( t * testing . T ) {
2019-02-12 08:04:04 -07:00
// Setup the model/pull environment
2020-12-21 04:59:22 -07:00
model , fo := setupSendReceiveFolder ( t )
2019-08-15 07:29:37 -07:00
defer cleanupSRFolder ( fo , model )
2019-03-08 13:29:09 -07:00
ffs := fo . Filesystem ( )
2019-01-11 05:56:05 -07:00
2019-02-12 08:04:04 -07:00
tempFile := fs . TempName ( "weakhash" )
2016-12-14 16:30:29 -07:00
var shift int64 = 10
var size int64 = 1 << 20
2018-04-16 11:08:50 -07:00
expectBlocks := int ( size / protocol . MinBlockSize )
expectPulls := int ( shift / protocol . MinBlockSize )
2016-12-14 16:30:29 -07:00
if shift > 0 {
expectPulls ++
}
2019-02-12 08:04:04 -07:00
f , err := ffs . Create ( "weakhash" )
2019-03-09 11:45:36 -07:00
must ( t , err )
2016-12-14 16:30:29 -07:00
defer f . Close ( )
2019-02-12 08:04:04 -07:00
_ , err = io . CopyN ( f , rand . Reader , size )
2016-12-14 16:30:29 -07:00
if err != nil {
t . Error ( err )
}
info , err := f . Stat ( )
if err != nil {
t . Error ( err )
}
// Create two files, second file has `shifted` bytes random prefix, yet
// both are of the same length, for example:
// File 1: abcdefgh
// File 2: xyabcdef
f . Seek ( 0 , os . SEEK_SET )
2018-04-16 11:08:50 -07:00
existing , err := scanner . Blocks ( context . TODO ( ) , f , protocol . MinBlockSize , size , nil , true )
2016-12-14 16:30:29 -07:00
if err != nil {
t . Error ( err )
}
f . Seek ( 0 , os . SEEK_SET )
remainder := io . LimitReader ( f , size - shift )
prefix := io . LimitReader ( rand . Reader , shift )
nf := io . MultiReader ( prefix , remainder )
2018-04-16 11:08:50 -07:00
desired , err := scanner . Blocks ( context . TODO ( ) , nf , protocol . MinBlockSize , size , nil , true )
2016-12-14 16:30:29 -07:00
if err != nil {
t . Error ( err )
}
existingFile := protocol . FileInfo {
Name : "weakhash" ,
Blocks : existing ,
Size : size ,
ModifiedS : info . ModTime ( ) . Unix ( ) ,
2020-10-01 23:07:05 -07:00
ModifiedNs : info . ModTime ( ) . Nanosecond ( ) ,
2016-12-14 16:30:29 -07:00
}
desiredFile := protocol . FileInfo {
Name : "weakhash" ,
Size : size ,
Blocks : desired ,
ModifiedS : info . ModTime ( ) . Unix ( ) + 1 ,
}
2019-04-07 04:29:17 -07:00
fo . updateLocalsFromScanning ( [ ] protocol . FileInfo { existingFile } )
2019-02-12 08:04:04 -07:00
2016-12-14 16:30:29 -07:00
copyChan := make ( chan copyBlocksState )
pullChan := make ( chan pullBlockState , expectBlocks )
finisherChan := make ( chan * sharedPullerState , 1 )
// Run a single fetcher routine
go fo . copierRoutine ( copyChan , pullChan , finisherChan )
2019-11-13 02:21:54 -07:00
defer close ( copyChan )
2016-12-14 16:30:29 -07:00
// Test 1 - no weak hashing, file gets fully repulled (`expectBlocks` pulls).
2017-01-04 14:04:13 -07:00
fo . WeakHashThresholdPct = 101
2020-01-21 10:23:08 -07:00
fo . handleFile ( desiredFile , fo . fset . Snapshot ( ) , copyChan )
2016-12-14 16:30:29 -07:00
var pulls [ ] pullBlockState
2019-11-13 02:21:54 -07:00
timeout := time . After ( 10 * time . Second )
2016-12-14 16:30:29 -07:00
for len ( pulls ) < expectBlocks {
select {
case pull := <- pullChan :
pulls = append ( pulls , pull )
2019-11-13 02:21:54 -07:00
case <- timeout :
2020-05-05 23:34:54 -07:00
t . Fatalf ( "timed out, got %d pulls expected %d" , len ( pulls ) , expectPulls )
2016-12-14 16:30:29 -07:00
}
}
finish := <- finisherChan
select {
case <- pullChan :
t . Fatal ( "Pull channel has data to be read" )
case <- finisherChan :
t . Fatal ( "Finisher channel has data to be read" )
default :
}
2019-08-16 00:35:19 -07:00
cleanupSharedPullerState ( finish )
2019-02-12 08:04:04 -07:00
if err := ffs . Remove ( tempFile ) ; err != nil {
t . Fatal ( err )
}
2016-12-14 16:30:29 -07:00
// Test 2 - using weak hash, expectPulls blocks pulled.
2017-01-04 14:04:13 -07:00
fo . WeakHashThresholdPct = - 1
2020-01-21 10:23:08 -07:00
fo . handleFile ( desiredFile , fo . fset . Snapshot ( ) , copyChan )
2016-12-14 16:30:29 -07:00
pulls = pulls [ : 0 ]
for len ( pulls ) < expectPulls {
select {
case pull := <- pullChan :
pulls = append ( pulls , pull )
2016-12-16 05:05:27 -07:00
case <- time . After ( 10 * time . Second ) :
2020-05-05 23:34:54 -07:00
t . Fatalf ( "timed out, got %d pulls expected %d" , len ( pulls ) , expectPulls )
2016-12-14 16:30:29 -07:00
}
}
finish = <- finisherChan
2019-08-16 00:35:19 -07:00
cleanupSharedPullerState ( finish )
2016-12-14 16:30:29 -07:00
expectShifted := expectBlocks - expectPulls
if finish . copyOriginShifted != expectShifted {
t . Errorf ( "did not copy %d shifted" , expectShifted )
}
}
2017-11-04 00:20:11 -07:00
// Test that updating a file removes its old blocks from the blockmap
2014-10-22 07:24:11 -07:00
func TestCopierCleanup ( t * testing . T ) {
2015-01-17 18:12:06 -07:00
iterFn := func ( folder , file string , index int32 ) bool {
2014-10-22 07:24:11 -07:00
return true
}
// Create a file
2019-03-04 05:27:10 -07:00
file := setupFile ( "test" , [ ] int { 0 } )
2020-05-16 05:39:27 -07:00
file . Size = 1
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t , file )
2019-08-15 07:29:37 -07:00
defer cleanupSRFolder ( f , m )
2014-10-22 07:24:11 -07:00
file . Blocks = [ ] protocol . BlockInfo { blocks [ 1 ] }
2019-02-06 01:32:03 -07:00
file . Version = file . Version . Update ( myID . Short ( ) )
2014-10-22 07:24:11 -07:00
// Update index (removing old blocks)
2019-04-07 04:29:17 -07:00
f . updateLocalsFromScanning ( [ ] protocol . FileInfo { file } )
2014-10-22 07:24:11 -07:00
2015-09-04 03:01:00 -07:00
if m . finder . Iterate ( folders , blocks [ 0 ] . Hash , iterFn ) {
2014-10-22 07:24:11 -07:00
t . Error ( "Unexpected block found" )
}
2015-09-04 03:01:00 -07:00
if ! m . finder . Iterate ( folders , blocks [ 1 ] . Hash , iterFn ) {
2014-10-22 07:24:11 -07:00
t . Error ( "Expected block not found" )
}
file . Blocks = [ ] protocol . BlockInfo { blocks [ 0 ] }
2019-02-06 01:32:03 -07:00
file . Version = file . Version . Update ( myID . Short ( ) )
2014-10-22 07:24:11 -07:00
// Update index (removing old blocks)
2019-04-07 04:29:17 -07:00
f . updateLocalsFromScanning ( [ ] protocol . FileInfo { file } )
2014-10-22 07:24:11 -07:00
2015-09-04 03:01:00 -07:00
if ! m . finder . Iterate ( folders , blocks [ 0 ] . Hash , iterFn ) {
2014-10-22 07:24:11 -07:00
t . Error ( "Unexpected block found" )
}
2015-09-04 03:01:00 -07:00
if m . finder . Iterate ( folders , blocks [ 1 ] . Hash , iterFn ) {
2014-10-22 07:24:11 -07:00
t . Error ( "Expected block not found" )
}
}
2014-10-24 15:20:08 -07:00
2015-01-07 16:12:12 -07:00
func TestDeregisterOnFailInCopy ( t * testing . T ) {
2019-03-04 05:27:10 -07:00
file := setupFile ( "filex" , [ ] int { 0 , 2 , 0 , 0 , 5 , 0 , 0 , 8 } )
2015-01-07 16:12:12 -07:00
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2019-08-15 07:29:37 -07:00
defer cleanupSRFolder ( f , m )
2015-03-04 16:33:48 -07:00
2018-12-11 01:42:03 -07:00
// Set up our evet subscription early
2019-08-15 07:29:37 -07:00
s := m . evLogger . Subscribe ( events . ItemFinished )
2018-12-11 01:42:03 -07:00
2015-01-07 16:12:12 -07:00
// queue.Done should be called by the finisher routine
2016-08-06 06:05:59 -07:00
f . queue . Push ( "filex" , 0 , time . Time { } )
2016-04-26 13:19:30 -07:00
f . queue . Pop ( )
2015-01-07 16:12:12 -07:00
2016-04-26 13:19:30 -07:00
if f . queue . lenProgress ( ) != 1 {
2015-01-07 16:12:12 -07:00
t . Fatal ( "Expected file in progress" )
}
pullChan := make ( chan pullBlockState )
2020-05-05 23:34:54 -07:00
finisherBufferChan := make ( chan * sharedPullerState , 1 )
2015-01-07 16:12:12 -07:00
finisherChan := make ( chan * sharedPullerState )
2017-12-07 01:42:03 -07:00
dbUpdateChan := make ( chan dbUpdateJob , 1 )
2020-01-21 10:23:08 -07:00
snap := f . fset . Snapshot ( )
2015-01-07 16:12:12 -07:00
2019-11-14 15:08:40 -07:00
copyChan , copyWg := startCopier ( f , pullChan , finisherBufferChan )
2020-01-21 10:23:08 -07:00
go f . finisherRoutine ( snap , finisherChan , dbUpdateChan , make ( chan string ) )
2019-11-14 15:08:40 -07:00
defer func ( ) {
close ( copyChan )
copyWg . Wait ( )
close ( pullChan )
close ( finisherBufferChan )
close ( finisherChan )
} ( )
2015-01-07 16:12:12 -07:00
2020-01-21 10:23:08 -07:00
f . handleFile ( file , snap , copyChan )
2015-01-07 16:12:12 -07:00
2015-04-28 08:34:55 -07:00
// Receive a block at puller, to indicate that at least a single copier
2015-01-07 16:12:12 -07:00
// loop has been performed.
2020-05-05 23:34:54 -07:00
var toPull pullBlockState
select {
case toPull = <- pullChan :
case <- time . After ( 10 * time . Second ) :
t . Fatal ( "timed out" )
}
2018-12-11 01:42:03 -07:00
2015-01-07 16:12:12 -07:00
// Unblock copier
2018-12-11 01:42:03 -07:00
go func ( ) {
for range pullChan {
}
} ( )
2015-01-07 16:12:12 -07:00
2019-11-14 15:08:40 -07:00
// Close the file, causing errors on further access
toPull . sharedPullerState . fail ( os . ErrNotExist )
2015-01-07 16:12:12 -07:00
select {
case state := <- finisherBufferChan :
// At this point the file should still be registered with both the job
// queue, and the progress emitter. Verify this.
2016-04-26 13:19:30 -07:00
if f . model . progressEmitter . lenRegistry ( ) != 1 || f . queue . lenProgress ( ) != 1 || f . queue . lenQueued ( ) != 0 {
2015-01-07 16:12:12 -07:00
t . Fatal ( "Could not find file" )
}
// Pass the file down the real finisher, and give it time to consume
finisherChan <- state
2018-05-14 12:01:35 -07:00
2018-12-11 01:42:03 -07:00
t0 := time . Now ( )
if ev , err := s . Poll ( time . Minute ) ; err != nil {
2018-05-14 12:01:35 -07:00
t . Fatal ( "Got error waiting for ItemFinished event:" , err )
} else if n := ev . Data . ( map [ string ] interface { } ) [ "item" ] ; n != state . file . Name {
t . Fatal ( "Got ItemFinished event for wrong file:" , n )
}
2018-12-11 01:42:03 -07:00
t . Log ( "event took" , time . Since ( t0 ) )
2015-01-07 16:12:12 -07:00
2015-10-13 22:38:13 -07:00
state . mut . Lock ( )
2019-08-16 00:35:19 -07:00
stateWriter := state . writer
2015-10-13 22:38:13 -07:00
state . mut . Unlock ( )
2019-08-16 00:35:19 -07:00
if stateWriter != nil {
2015-01-07 16:12:12 -07:00
t . Fatal ( "File not closed?" )
}
2016-04-26 13:19:30 -07:00
if f . model . progressEmitter . lenRegistry ( ) != 0 || f . queue . lenProgress ( ) != 0 || f . queue . lenQueued ( ) != 0 {
t . Fatal ( "Still registered" , f . model . progressEmitter . lenRegistry ( ) , f . queue . lenProgress ( ) , f . queue . lenQueued ( ) )
2015-01-07 16:12:12 -07:00
}
// Doing it again should have no effect
finisherChan <- state
2018-12-11 01:42:03 -07:00
if _ , err := s . Poll ( time . Second ) ; err != events . ErrTimeout {
t . Fatal ( "Expected timeout, not another event" , err )
}
2015-01-07 16:12:12 -07:00
2016-04-26 13:19:30 -07:00
if f . model . progressEmitter . lenRegistry ( ) != 0 || f . queue . lenProgress ( ) != 0 || f . queue . lenQueued ( ) != 0 {
t . Fatal ( "Still registered" , f . model . progressEmitter . lenRegistry ( ) , f . queue . lenProgress ( ) , f . queue . lenQueued ( ) )
2015-01-07 16:12:12 -07:00
}
2018-12-11 01:42:03 -07:00
2019-11-13 02:21:54 -07:00
case <- time . After ( 5 * time . Second ) :
2015-01-07 16:12:12 -07:00
t . Fatal ( "Didn't get anything to the finisher" )
}
}
func TestDeregisterOnFailInPull ( t * testing . T ) {
2019-03-04 05:27:10 -07:00
file := setupFile ( "filex" , [ ] int { 0 , 2 , 0 , 0 , 5 , 0 , 0 , 8 } )
2015-01-07 16:12:12 -07:00
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2019-08-15 07:29:37 -07:00
defer cleanupSRFolder ( f , m )
2015-01-07 16:12:12 -07:00
2018-12-11 01:42:03 -07:00
// Set up our evet subscription early
2019-08-15 07:29:37 -07:00
s := m . evLogger . Subscribe ( events . ItemFinished )
2018-12-11 01:42:03 -07:00
2015-01-07 16:12:12 -07:00
// queue.Done should be called by the finisher routine
2016-08-06 06:05:59 -07:00
f . queue . Push ( "filex" , 0 , time . Time { } )
2016-04-26 13:19:30 -07:00
f . queue . Pop ( )
2015-01-07 16:12:12 -07:00
2016-04-26 13:19:30 -07:00
if f . queue . lenProgress ( ) != 1 {
2015-01-07 16:12:12 -07:00
t . Fatal ( "Expected file in progress" )
}
pullChan := make ( chan pullBlockState )
finisherBufferChan := make ( chan * sharedPullerState )
finisherChan := make ( chan * sharedPullerState )
2017-12-07 01:42:03 -07:00
dbUpdateChan := make ( chan dbUpdateJob , 1 )
2020-01-21 10:23:08 -07:00
snap := f . fset . Snapshot ( )
2015-01-07 16:12:12 -07:00
2019-11-14 15:08:40 -07:00
copyChan , copyWg := startCopier ( f , pullChan , finisherBufferChan )
pullWg := sync . NewWaitGroup ( )
pullWg . Add ( 1 )
go func ( ) {
f . pullerRoutine ( pullChan , finisherBufferChan )
pullWg . Done ( )
} ( )
2020-01-21 10:23:08 -07:00
go f . finisherRoutine ( snap , finisherChan , dbUpdateChan , make ( chan string ) )
2019-11-14 15:08:40 -07:00
defer func ( ) {
// Unblock copier and puller
go func ( ) {
for range finisherBufferChan {
}
} ( )
close ( copyChan )
copyWg . Wait ( )
close ( pullChan )
pullWg . Wait ( )
close ( finisherBufferChan )
close ( finisherChan )
} ( )
2015-01-07 16:12:12 -07:00
2020-01-21 10:23:08 -07:00
f . handleFile ( file , snap , copyChan )
2015-01-07 16:12:12 -07:00
2015-11-11 19:20:34 -07:00
// Receive at finisher, we should error out as puller has nowhere to pull
2015-01-07 16:12:12 -07:00
// from.
2018-05-14 12:01:35 -07:00
timeout = time . Second
2019-11-14 15:08:40 -07:00
// Both the puller and copier may send to the finisherBufferChan.
var state * sharedPullerState
after := time . After ( 5 * time . Second )
for {
select {
case state = <- finisherBufferChan :
case <- after :
t . Fatal ( "Didn't get failed state to the finisher" )
2015-01-07 16:12:12 -07:00
}
2019-11-14 15:08:40 -07:00
if state . failed ( ) != nil {
break
}
}
2015-01-07 16:12:12 -07:00
2019-11-14 15:08:40 -07:00
// At this point the file should still be registered with both the job
// queue, and the progress emitter. Verify this.
if f . model . progressEmitter . lenRegistry ( ) != 1 || f . queue . lenProgress ( ) != 1 || f . queue . lenQueued ( ) != 0 {
t . Fatal ( "Could not find file" )
}
2018-05-14 12:01:35 -07:00
2019-11-14 15:08:40 -07:00
// Pass the file down the real finisher, and give it time to consume
finisherChan <- state
2015-01-07 16:12:12 -07:00
2019-11-14 15:08:40 -07:00
t0 := time . Now ( )
if ev , err := s . Poll ( time . Minute ) ; err != nil {
t . Fatal ( "Got error waiting for ItemFinished event:" , err )
} else if n := ev . Data . ( map [ string ] interface { } ) [ "item" ] ; n != state . file . Name {
t . Fatal ( "Got ItemFinished event for wrong file:" , n )
}
t . Log ( "event took" , time . Since ( t0 ) )
2015-01-07 16:12:12 -07:00
2019-11-14 15:08:40 -07:00
state . mut . Lock ( )
stateWriter := state . writer
state . mut . Unlock ( )
if stateWriter != nil {
t . Fatal ( "File not closed?" )
}
2015-01-07 16:12:12 -07:00
2019-11-14 15:08:40 -07:00
if f . model . progressEmitter . lenRegistry ( ) != 0 || f . queue . lenProgress ( ) != 0 || f . queue . lenQueued ( ) != 0 {
t . Fatal ( "Still registered" , f . model . progressEmitter . lenRegistry ( ) , f . queue . lenProgress ( ) , f . queue . lenQueued ( ) )
}
2018-12-11 01:42:03 -07:00
2019-11-14 15:08:40 -07:00
// Doing it again should have no effect
finisherChan <- state
2015-01-07 16:12:12 -07:00
2019-11-14 15:08:40 -07:00
if _ , err := s . Poll ( time . Second ) ; err != events . ErrTimeout {
t . Fatal ( "Expected timeout, not another event" , err )
}
if f . model . progressEmitter . lenRegistry ( ) != 0 || f . queue . lenProgress ( ) != 0 || f . queue . lenQueued ( ) != 0 {
t . Fatal ( "Still registered" , f . model . progressEmitter . lenRegistry ( ) , f . queue . lenProgress ( ) , f . queue . lenQueued ( ) )
2015-01-07 16:12:12 -07:00
}
}
2017-12-07 01:42:03 -07:00
func TestIssue3164 ( t * testing . T ) {
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2019-08-15 07:29:37 -07:00
defer cleanupSRFolder ( f , m )
2019-03-08 13:29:09 -07:00
ffs := f . Filesystem ( )
tmpDir := ffs . URI ( )
2017-12-07 01:42:03 -07:00
2019-02-12 08:04:04 -07:00
ignDir := filepath . Join ( "issue3164" , "oktodelete" )
subDir := filepath . Join ( ignDir , "foobar" )
2019-03-09 11:45:36 -07:00
must ( t , ffs . MkdirAll ( subDir , 0777 ) )
must ( t , ioutil . WriteFile ( filepath . Join ( tmpDir , subDir , "file" ) , [ ] byte ( "Hello" ) , 0644 ) )
must ( t , ioutil . WriteFile ( filepath . Join ( tmpDir , ignDir , "file" ) , [ ] byte ( "Hello" ) , 0644 ) )
2017-12-07 01:42:03 -07:00
file := protocol . FileInfo {
Name : "issue3164" ,
}
2020-06-16 06:20:08 -07:00
must ( t , f . scanSubdirs ( nil ) )
2019-02-12 08:04:04 -07:00
matcher := ignore . New ( ffs )
2019-03-09 11:45:36 -07:00
must ( t , matcher . Parse ( bytes . NewBufferString ( "(?d)oktodelete" ) , "" ) )
2019-03-10 23:28:54 -07:00
f . ignores = matcher
2017-12-07 01:42:03 -07:00
dbUpdateChan := make ( chan dbUpdateJob , 1 )
2020-01-21 10:23:08 -07:00
f . deleteDir ( file , f . fset . Snapshot ( ) , dbUpdateChan , make ( chan string ) )
2017-12-07 01:42:03 -07:00
2019-02-12 08:04:04 -07:00
if _ , err := ffs . Stat ( "issue3164" ) ; ! fs . IsNotExist ( err ) {
2017-12-07 01:42:03 -07:00
t . Fatal ( err )
}
}
2018-01-14 07:30:11 -07:00
func TestDiff ( t * testing . T ) {
for i , test := range diffTestData {
a , _ := scanner . Blocks ( context . TODO ( ) , bytes . NewBufferString ( test . a ) , test . s , - 1 , nil , false )
b , _ := scanner . Blocks ( context . TODO ( ) , bytes . NewBufferString ( test . b ) , test . s , - 1 , nil , false )
_ , d := blockDiff ( a , b )
if len ( d ) != len ( test . d ) {
t . Fatalf ( "Incorrect length for diff %d; %d != %d" , i , len ( d ) , len ( test . d ) )
} else {
for j := range test . d {
if d [ j ] . Offset != test . d [ j ] . Offset {
t . Errorf ( "Incorrect offset for diff %d block %d; %d != %d" , i , j , d [ j ] . Offset , test . d [ j ] . Offset )
}
if d [ j ] . Size != test . d [ j ] . Size {
t . Errorf ( "Incorrect length for diff %d block %d; %d != %d" , i , j , d [ j ] . Size , test . d [ j ] . Size )
}
}
}
}
}
2018-01-14 14:52:41 -07:00
func BenchmarkDiff ( b * testing . B ) {
testCases := make ( [ ] struct { a , b [ ] protocol . BlockInfo } , 0 , len ( diffTestData ) )
for _ , test := range diffTestData {
a , _ := scanner . Blocks ( context . TODO ( ) , bytes . NewBufferString ( test . a ) , test . s , - 1 , nil , false )
b , _ := scanner . Blocks ( context . TODO ( ) , bytes . NewBufferString ( test . b ) , test . s , - 1 , nil , false )
testCases = append ( testCases , struct { a , b [ ] protocol . BlockInfo } { a , b } )
}
b . ReportAllocs ( )
b . ResetTimer ( )
for i := 0 ; i < b . N ; i ++ {
for _ , tc := range testCases {
blockDiff ( tc . a , tc . b )
}
}
}
2018-01-14 07:30:11 -07:00
func TestDiffEmpty ( t * testing . T ) {
emptyCases := [ ] struct {
a [ ] protocol . BlockInfo
b [ ] protocol . BlockInfo
need int
have int
} {
{ nil , nil , 0 , 0 } ,
{ [ ] protocol . BlockInfo { { Offset : 3 , Size : 1 } } , nil , 0 , 0 } ,
{ nil , [ ] protocol . BlockInfo { { Offset : 3 , Size : 1 } } , 1 , 0 } ,
}
for _ , emptyCase := range emptyCases {
h , n := blockDiff ( emptyCase . a , emptyCase . b )
if len ( h ) != emptyCase . have {
t . Errorf ( "incorrect have: %d != %d" , len ( h ) , emptyCase . have )
}
if len ( n ) != emptyCase . need {
t . Errorf ( "incorrect have: %d != %d" , len ( h ) , emptyCase . have )
}
}
}
2018-11-13 01:36:16 -07:00
// TestDeleteIgnorePerms checks, that a file gets deleted when the IgnorePerms
// option is true and the permissions do not match between the file on disk and
// in the db.
func TestDeleteIgnorePerms ( t * testing . T ) {
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2019-08-15 07:29:37 -07:00
defer cleanupSRFolder ( f , m )
2019-03-08 13:29:09 -07:00
ffs := f . Filesystem ( )
2018-11-13 01:36:16 -07:00
f . IgnorePerms = true
name := "deleteIgnorePerms"
file , err := ffs . Create ( name )
if err != nil {
t . Error ( err )
}
defer file . Close ( )
stat , err := file . Stat ( )
2019-03-09 11:45:36 -07:00
must ( t , err )
2018-11-13 01:36:16 -07:00
fi , err := scanner . CreateFileInfo ( stat , name , ffs )
2019-03-09 11:45:36 -07:00
must ( t , err )
2018-11-13 01:36:16 -07:00
ffs . Chmod ( name , 0600 )
2020-09-11 22:45:50 -07:00
scanChan := make ( chan string , 1 )
err = f . checkToBeDeleted ( fi , fi , true , scanChan )
2019-03-09 11:45:36 -07:00
must ( t , err )
2018-11-13 01:36:16 -07:00
}
2019-01-25 01:52:21 -07:00
func TestCopyOwner ( t * testing . T ) {
// Verifies that owner and group are copied from the parent, for both
// files and directories.
if runtime . GOOS == "windows" {
t . Skip ( "copying owner not supported on Windows" )
}
const (
expOwner = 1234
expGroup = 5678
)
// Set up a folder with the CopyParentOwner bit and backed by a fake
// filesystem.
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2019-08-15 07:29:37 -07:00
defer cleanupSRFolder ( f , m )
2019-02-12 08:04:04 -07:00
f . folder . FolderConfiguration = config . NewFolderConfiguration ( m . id , f . ID , f . Label , fs . FilesystemTypeFake , "/TestCopyOwner" )
f . folder . FolderConfiguration . CopyOwnershipFromParent = true
2019-01-25 01:52:21 -07:00
f . fs = f . Filesystem ( )
// Create a parent dir with a certain owner/group.
f . fs . Mkdir ( "foo" , 0755 )
f . fs . Lchown ( "foo" , expOwner , expGroup )
dir := protocol . FileInfo {
Name : "foo/bar" ,
Type : protocol . FileInfoTypeDirectory ,
Permissions : 0755 ,
}
// Have the folder create a subdirectory, verify that it's the correct
// owner/group.
dbUpdateChan := make ( chan dbUpdateJob , 1 )
2020-05-24 23:46:24 -07:00
scanChan := make ( chan string )
2019-01-25 01:52:21 -07:00
defer close ( dbUpdateChan )
2020-05-24 23:46:24 -07:00
f . handleDir ( dir , f . fset . Snapshot ( ) , dbUpdateChan , scanChan )
select {
case <- dbUpdateChan : // empty the channel for later
case toScan := <- scanChan :
t . Fatal ( "Unexpected receive on scanChan:" , toScan )
}
2019-01-25 01:52:21 -07:00
info , err := f . fs . Lstat ( "foo/bar" )
if err != nil {
t . Fatal ( "Unexpected error (dir):" , err )
}
if info . Owner ( ) != expOwner || info . Group ( ) != expGroup {
t . Fatalf ( "Expected dir owner/group to be %d/%d, not %d/%d" , expOwner , expGroup , info . Owner ( ) , info . Group ( ) )
}
// Have the folder create a file, verify it's the correct owner/group.
// File is zero sized to avoid having to handle copies/pulls.
file := protocol . FileInfo {
Name : "foo/bar/baz" ,
Type : protocol . FileInfoTypeFile ,
Permissions : 0644 ,
}
// Wire some stuff. The flow here is handleFile() -[copierChan]->
// copierRoutine() -[finisherChan]-> finisherRoutine() -[dbUpdateChan]->
// back to us and we're done. The copier routine doesn't do anything,
// but it's the way data is passed around. When the database update
// comes the finisher is done.
2020-01-21 10:23:08 -07:00
snap := f . fset . Snapshot ( )
2019-01-25 01:52:21 -07:00
finisherChan := make ( chan * sharedPullerState )
2019-11-14 15:08:40 -07:00
copierChan , copyWg := startCopier ( f , nil , finisherChan )
2020-01-21 10:23:08 -07:00
go f . finisherRoutine ( snap , finisherChan , dbUpdateChan , nil )
2019-11-14 15:08:40 -07:00
defer func ( ) {
close ( copierChan )
copyWg . Wait ( )
close ( finisherChan )
} ( )
2020-01-21 10:23:08 -07:00
f . handleFile ( file , snap , copierChan )
2019-01-25 01:52:21 -07:00
<- dbUpdateChan
info , err = f . fs . Lstat ( "foo/bar/baz" )
if err != nil {
t . Fatal ( "Unexpected error (file):" , err )
}
if info . Owner ( ) != expOwner || info . Group ( ) != expGroup {
t . Fatalf ( "Expected file owner/group to be %d/%d, not %d/%d" , expOwner , expGroup , info . Owner ( ) , info . Group ( ) )
}
// Have the folder create a symlink. Verify it accordingly.
symlink := protocol . FileInfo {
Name : "foo/bar/sym" ,
Type : protocol . FileInfoTypeSymlink ,
Permissions : 0644 ,
SymlinkTarget : "over the rainbow" ,
}
2020-05-24 23:46:24 -07:00
f . handleSymlink ( symlink , snap , dbUpdateChan , scanChan )
select {
case <- dbUpdateChan :
case toScan := <- scanChan :
t . Fatal ( "Unexpected receive on scanChan:" , toScan )
}
2019-01-25 01:52:21 -07:00
info , err = f . fs . Lstat ( "foo/bar/sym" )
if err != nil {
t . Fatal ( "Unexpected error (file):" , err )
}
if info . Owner ( ) != expOwner || info . Group ( ) != expGroup {
t . Fatalf ( "Expected symlink owner/group to be %d/%d, not %d/%d" , expOwner , expGroup , info . Owner ( ) , info . Group ( ) )
}
}
2019-03-07 07:15:14 -07:00
// TestSRConflictReplaceFileByDir checks that a conflict is created when an existing file
// is replaced with a directory and versions are conflicting
func TestSRConflictReplaceFileByDir ( t * testing . T ) {
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2019-08-15 07:29:37 -07:00
defer cleanupSRFolder ( f , m )
2019-03-08 13:29:09 -07:00
ffs := f . Filesystem ( )
2019-03-07 07:15:14 -07:00
name := "foo"
// create local file
file := createFile ( t , name , ffs )
file . Version = protocol . Vector { } . Update ( myID . Short ( ) )
2019-04-07 04:29:17 -07:00
f . updateLocalsFromScanning ( [ ] protocol . FileInfo { file } )
2019-03-07 07:15:14 -07:00
// Simulate remote creating a dir with the same name
file . Type = protocol . FileInfoTypeDirectory
rem := device1 . Short ( )
file . Version = protocol . Vector { } . Update ( rem )
file . ModifiedBy = rem
dbUpdateChan := make ( chan dbUpdateJob , 1 )
scanChan := make ( chan string , 1 )
2020-01-21 10:23:08 -07:00
f . handleDir ( file , f . fset . Snapshot ( ) , dbUpdateChan , scanChan )
2019-03-07 07:15:14 -07:00
if confls := existingConflicts ( name , ffs ) ; len ( confls ) != 1 {
t . Fatal ( "Expected one conflict, got" , len ( confls ) )
} else if scan := <- scanChan ; confls [ 0 ] != scan {
t . Fatal ( "Expected request to scan" , confls [ 0 ] , "got" , scan )
}
}
// TestSRConflictReplaceFileByLink checks that a conflict is created when an existing file
// is replaced with a link and versions are conflicting
func TestSRConflictReplaceFileByLink ( t * testing . T ) {
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2019-08-15 07:29:37 -07:00
defer cleanupSRFolder ( f , m )
2019-03-08 13:29:09 -07:00
ffs := f . Filesystem ( )
2019-03-07 07:15:14 -07:00
name := "foo"
// create local file
file := createFile ( t , name , ffs )
file . Version = protocol . Vector { } . Update ( myID . Short ( ) )
2019-04-07 04:29:17 -07:00
f . updateLocalsFromScanning ( [ ] protocol . FileInfo { file } )
2019-03-07 07:15:14 -07:00
// Simulate remote creating a symlink with the same name
file . Type = protocol . FileInfoTypeSymlink
file . SymlinkTarget = "bar"
rem := device1 . Short ( )
file . Version = protocol . Vector { } . Update ( rem )
file . ModifiedBy = rem
dbUpdateChan := make ( chan dbUpdateJob , 1 )
scanChan := make ( chan string , 1 )
2020-01-21 10:23:08 -07:00
f . handleSymlink ( file , f . fset . Snapshot ( ) , dbUpdateChan , scanChan )
2019-03-07 07:15:14 -07:00
if confls := existingConflicts ( name , ffs ) ; len ( confls ) != 1 {
t . Fatal ( "Expected one conflict, got" , len ( confls ) )
} else if scan := <- scanChan ; confls [ 0 ] != scan {
t . Fatal ( "Expected request to scan" , confls [ 0 ] , "got" , scan )
}
}
2019-08-16 00:35:19 -07:00
2019-10-22 12:55:51 -07:00
// TestDeleteBehindSymlink checks that we don't delete or schedule a scan
// when trying to delete a file behind a symlink.
func TestDeleteBehindSymlink ( t * testing . T ) {
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2019-10-22 12:55:51 -07:00
defer cleanupSRFolder ( f , m )
ffs := f . Filesystem ( )
destDir := createTmpDir ( )
defer os . RemoveAll ( destDir )
destFs := fs . NewFilesystem ( fs . FilesystemTypeBasic , destDir )
link := "link"
file := filepath . Join ( link , "file" )
must ( t , ffs . MkdirAll ( link , 0755 ) )
fi := createFile ( t , file , ffs )
f . updateLocalsFromScanning ( [ ] protocol . FileInfo { fi } )
2020-06-17 23:15:47 -07:00
must ( t , osutil . RenameOrCopy ( fs . CopyRangeMethodStandard , ffs , destFs , file , "file" ) )
2019-10-22 12:55:51 -07:00
must ( t , ffs . RemoveAll ( link ) )
2020-07-28 02:13:15 -07:00
if err := fs . DebugSymlinkForTestsOnly ( destFs , ffs , "" , link ) ; err != nil {
2019-10-22 12:55:51 -07:00
if runtime . GOOS == "windows" {
// Probably we require permissions we don't have.
t . Skip ( "Need admin permissions or developer mode to run symlink test on Windows: " + err . Error ( ) )
} else {
t . Fatal ( err )
}
}
fi . Deleted = true
fi . Version = fi . Version . Update ( device1 . Short ( ) )
scanChan := make ( chan string , 1 )
dbUpdateChan := make ( chan dbUpdateJob , 1 )
2020-01-21 10:23:08 -07:00
f . deleteFile ( fi , f . fset . Snapshot ( ) , dbUpdateChan , scanChan )
2019-10-22 12:55:51 -07:00
select {
case f := <- scanChan :
t . Fatalf ( "Received %v on scanChan" , f )
case u := <- dbUpdateChan :
if u . jobType != dbUpdateDeleteFile {
t . Errorf ( "Expected jobType %v, got %v" , dbUpdateDeleteFile , u . jobType )
}
if u . file . Name != fi . Name {
t . Errorf ( "Expected update for %v, got %v" , fi . Name , u . file . Name )
}
default :
t . Fatalf ( "No db update received" )
}
if _ , err := destFs . Stat ( "file" ) ; err != nil {
t . Errorf ( "Expected no error when stating file behind symlink, got %v" , err )
}
}
2020-04-21 10:55:14 -07:00
// Reproduces https://github.com/syncthing/syncthing/issues/6559
func TestPullCtxCancel ( t * testing . T ) {
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2020-04-21 10:55:14 -07:00
defer cleanupSRFolder ( f , m )
pullChan := make ( chan pullBlockState )
finisherChan := make ( chan * sharedPullerState )
var cancel context . CancelFunc
f . ctx , cancel = context . WithCancel ( context . Background ( ) )
go f . pullerRoutine ( pullChan , finisherChan )
defer close ( pullChan )
emptyState := func ( ) pullBlockState {
return pullBlockState {
2020-05-01 00:36:46 -07:00
sharedPullerState : newSharedPullerState ( protocol . FileInfo { } , nil , f . folderID , "" , nil , nil , false , false , protocol . FileInfo { } , false , false ) ,
2020-04-21 10:55:14 -07:00
block : protocol . BlockInfo { } ,
}
}
cancel ( )
done := make ( chan struct { } )
defer close ( done )
for i := 0 ; i < 2 ; i ++ {
go func ( ) {
select {
case pullChan <- emptyState ( ) :
case <- done :
}
} ( )
select {
case s := <- finisherChan :
if s . failed ( ) == nil {
t . Errorf ( "state %v not failed" , i )
}
case <- time . After ( 5 * time . Second ) :
t . Fatalf ( "timed out before receiving state %v on finisherChan" , i )
}
}
}
2020-06-16 06:20:08 -07:00
func TestPullDeleteUnscannedDir ( t * testing . T ) {
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2020-06-16 06:20:08 -07:00
defer cleanupSRFolder ( f , m )
ffs := f . Filesystem ( )
dir := "foobar"
must ( t , ffs . MkdirAll ( dir , 0777 ) )
fi := protocol . FileInfo {
Name : dir ,
}
scanChan := make ( chan string , 1 )
dbUpdateChan := make ( chan dbUpdateJob , 1 )
f . deleteDir ( fi , f . fset . Snapshot ( ) , dbUpdateChan , scanChan )
if _ , err := ffs . Stat ( dir ) ; fs . IsNotExist ( err ) {
t . Error ( "directory has been deleted" )
}
select {
case toScan := <- scanChan :
if toScan != dir {
t . Errorf ( "expected %v to be scanned, got %v" , dir , toScan )
}
default :
t . Error ( "nothing was scheduled for scanning" )
}
}
2020-07-28 02:13:15 -07:00
func TestPullCaseOnlyPerformFinish ( t * testing . T ) {
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2020-07-28 02:13:15 -07:00
defer cleanupSRFolder ( f , m )
ffs := f . Filesystem ( )
name := "foo"
contents := [ ] byte ( "contents" )
must ( t , writeFile ( ffs , name , contents , 0644 ) )
must ( t , f . scanSubdirs ( nil ) )
var cur protocol . FileInfo
hasCur := false
snap := dbSnapshot ( t , m , f . ID )
defer snap . Release ( )
snap . WithHave ( protocol . LocalDeviceID , func ( i protocol . FileIntf ) bool {
if hasCur {
t . Fatal ( "got more than one file" )
}
cur = i . ( protocol . FileInfo )
hasCur = true
return true
} )
if ! hasCur {
t . Fatal ( "file is missing" )
}
remote := * ( & cur )
remote . Version = protocol . Vector { } . Update ( device1 . Short ( ) )
remote . Name = strings . ToUpper ( cur . Name )
temp := fs . TempName ( remote . Name )
must ( t , writeFile ( ffs , temp , contents , 0644 ) )
scanChan := make ( chan string , 1 )
dbUpdateChan := make ( chan dbUpdateJob , 1 )
err := f . performFinish ( remote , cur , hasCur , temp , snap , dbUpdateChan , scanChan )
select {
case <- dbUpdateChan : // boring case sensitive filesystem
return
case <- scanChan :
t . Error ( "no need to scan anything here" )
default :
}
var caseErr * fs . ErrCaseConflict
if ! errors . As ( err , & caseErr ) {
t . Error ( "Expected case conflict error, got" , err )
}
}
func TestPullCaseOnlyDir ( t * testing . T ) {
testPullCaseOnlyDirOrSymlink ( t , true )
}
func TestPullCaseOnlySymlink ( t * testing . T ) {
if runtime . GOOS == "windows" {
t . Skip ( "symlinks not supported on windows" )
}
testPullCaseOnlyDirOrSymlink ( t , false )
}
func testPullCaseOnlyDirOrSymlink ( t * testing . T , dir bool ) {
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2020-07-28 02:13:15 -07:00
defer cleanupSRFolder ( f , m )
ffs := f . Filesystem ( )
name := "foo"
if dir {
must ( t , ffs . Mkdir ( name , 0777 ) )
} else {
must ( t , ffs . CreateSymlink ( "target" , name ) )
}
must ( t , f . scanSubdirs ( nil ) )
var cur protocol . FileInfo
hasCur := false
snap := dbSnapshot ( t , m , f . ID )
defer snap . Release ( )
snap . WithHave ( protocol . LocalDeviceID , func ( i protocol . FileIntf ) bool {
if hasCur {
t . Fatal ( "got more than one file" )
}
cur = i . ( protocol . FileInfo )
hasCur = true
return true
} )
if ! hasCur {
t . Fatal ( "file is missing" )
}
scanChan := make ( chan string , 1 )
dbUpdateChan := make ( chan dbUpdateJob , 1 )
remote := * ( & cur )
remote . Version = protocol . Vector { } . Update ( device1 . Short ( ) )
remote . Name = strings . ToUpper ( cur . Name )
if dir {
f . handleDir ( remote , snap , dbUpdateChan , scanChan )
} else {
f . handleSymlink ( remote , snap , dbUpdateChan , scanChan )
}
select {
case <- dbUpdateChan : // boring case sensitive filesystem
return
case <- scanChan :
t . Error ( "no need to scan anything here" )
default :
}
if errStr , ok := f . tempPullErrors [ remote . Name ] ; ! ok {
t . Error ( "missing error for" , remote . Name )
} else if ! strings . Contains ( errStr , "differs from name" ) {
t . Error ( "unexpected error" , errStr , "for" , remote . Name )
}
}
2020-09-09 02:47:14 -07:00
func TestPullTempFileCaseConflict ( t * testing . T ) {
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2020-09-09 02:47:14 -07:00
defer cleanupSRFolder ( f , m )
copyChan := make ( chan copyBlocksState , 1 )
file := protocol . FileInfo { Name : "foo" }
confl := "Foo"
tempNameConfl := fs . TempName ( confl )
if fd , err := f . fs . Create ( tempNameConfl ) ; err != nil {
t . Fatal ( err )
} else {
if _ , err := fd . Write ( [ ] byte ( "data" ) ) ; err != nil {
t . Fatal ( err )
}
fd . Close ( )
}
f . handleFile ( file , f . fset . Snapshot ( ) , copyChan )
cs := <- copyChan
if _ , err := cs . tempFile ( ) ; err != nil {
t . Error ( err )
}
}
2020-09-09 23:23:54 -07:00
func TestPullCaseOnlyRename ( t * testing . T ) {
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2020-09-09 23:23:54 -07:00
defer cleanupSRFolder ( f , m )
// tempNameConfl := fs.TempName(confl)
name := "foo"
if fd , err := f . fs . Create ( name ) ; err != nil {
t . Fatal ( err )
} else {
if _ , err := fd . Write ( [ ] byte ( "data" ) ) ; err != nil {
t . Fatal ( err )
}
fd . Close ( )
}
must ( t , f . scanSubdirs ( nil ) )
cur , ok := m . CurrentFolderFile ( f . ID , name )
if ! ok {
t . Fatal ( "file missing" )
}
deleted := cur
deleted . SetDeleted ( myID . Short ( ) )
confl := cur
confl . Name = "Foo"
confl . Version = confl . Version . Update ( device1 . Short ( ) )
dbUpdateChan := make ( chan dbUpdateJob , 2 )
scanChan := make ( chan string , 2 )
snap := f . fset . Snapshot ( )
defer snap . Release ( )
if err := f . renameFile ( cur , deleted , confl , snap , dbUpdateChan , scanChan ) ; err != nil {
t . Error ( err )
}
}
2020-09-10 01:52:38 -07:00
func TestPullSymlinkOverExistingWindows ( t * testing . T ) {
if runtime . GOOS != "windows" {
t . Skip ( )
}
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2020-09-10 01:52:38 -07:00
defer cleanupSRFolder ( f , m )
name := "foo"
if fd , err := f . fs . Create ( name ) ; err != nil {
t . Fatal ( err )
} else {
if _ , err := fd . Write ( [ ] byte ( "data" ) ) ; err != nil {
t . Fatal ( err )
}
fd . Close ( )
}
must ( t , f . scanSubdirs ( nil ) )
file , ok := m . CurrentFolderFile ( f . ID , name )
if ! ok {
t . Fatal ( "file missing" )
}
m . Index ( device1 , f . ID , [ ] protocol . FileInfo { { Name : name , Type : protocol . FileInfoTypeSymlink , Version : file . Version . Update ( device1 . Short ( ) ) } } )
scanChan := make ( chan string )
changed := f . pullerIteration ( scanChan )
if changed != 1 {
t . Error ( "Expected one change in pull, got" , changed )
}
if file , ok := m . CurrentFolderFile ( f . ID , name ) ; ! ok {
t . Error ( "symlink entry missing" )
} else if ! file . IsUnsupported ( ) {
t . Error ( "symlink entry isn't marked as unsupported" )
}
if _ , err := f . fs . Lstat ( name ) ; err == nil {
t . Error ( "old file still exists on disk" )
} else if ! fs . IsNotExist ( err ) {
t . Error ( err )
}
}
2020-09-11 22:45:50 -07:00
func TestPullDeleteCaseConflict ( t * testing . T ) {
2020-12-21 04:59:22 -07:00
m , f := setupSendReceiveFolder ( t )
2020-09-11 22:45:50 -07:00
defer cleanupSRFolder ( f , m )
name := "foo"
fi := protocol . FileInfo { Name : "Foo" }
dbUpdateChan := make ( chan dbUpdateJob , 1 )
scanChan := make ( chan string )
if fd , err := f . fs . Create ( name ) ; err != nil {
t . Fatal ( err )
} else {
if _ , err := fd . Write ( [ ] byte ( "data" ) ) ; err != nil {
t . Fatal ( err )
}
fd . Close ( )
}
f . deleteFileWithCurrent ( fi , protocol . FileInfo { } , false , dbUpdateChan , scanChan )
select {
case <- dbUpdateChan :
default :
t . Error ( "Missing db update for file" )
}
snap := f . fset . Snapshot ( )
defer snap . Release ( )
f . deleteDir ( fi , snap , dbUpdateChan , scanChan )
select {
case <- dbUpdateChan :
default :
t . Error ( "Missing db update for dir" )
}
}
2019-08-16 00:35:19 -07:00
func cleanupSharedPullerState ( s * sharedPullerState ) {
s . mut . Lock ( )
defer s . mut . Unlock ( )
if s . writer == nil {
return
}
s . writer . mut . Lock ( )
s . writer . fd . Close ( )
s . writer . mut . Unlock ( )
}
2019-11-14 15:08:40 -07:00
func startCopier ( f * sendReceiveFolder , pullChan chan <- pullBlockState , finisherChan chan <- * sharedPullerState ) ( chan copyBlocksState , sync . WaitGroup ) {
copyChan := make ( chan copyBlocksState )
wg := sync . NewWaitGroup ( )
wg . Add ( 1 )
go func ( ) {
f . copierRoutine ( copyChan , pullChan , finisherChan )
wg . Done ( )
} ( )
return copyChan , wg
}