1 // Copyright 2016 The Go Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style 3 // license that can be found in the LICENSE file. 4 5 package net 6 7 import ( 8 "crypto/sha256" 9 "encoding/hex" 10 "fmt" 11 "io" 12 "os" 13 "testing" 14 ) 15 16 const ( 17 twain = "testdata/Mark.Twain-Tom.Sawyer.txt" 18 twainLen = 387851 19 twainSHA256 = "461eb7cb2d57d293fc680c836464c9125e4382be3596f7d415093ae9db8fcb0e" 20 ) 21 22 func TestSendfile(t *testing.T) { 23 ln, err := newLocalListener("tcp") 24 if err != nil { 25 t.Fatal(err) 26 } 27 defer ln.Close() 28 29 errc := make(chan error, 1) 30 go func(ln Listener) { 31 // Wait for a connection. 32 conn, err := ln.Accept() 33 if err != nil { 34 errc <- err 35 close(errc) 36 return 37 } 38 39 go func() { 40 defer close(errc) 41 defer conn.Close() 42 43 f, err := os.Open(twain) 44 if err != nil { 45 errc <- err 46 return 47 } 48 defer f.Close() 49 50 // Return file data using io.Copy, which should use 51 // sendFile if available. 52 sbytes, err := io.Copy(conn, f) 53 if err != nil { 54 errc <- err 55 return 56 } 57 58 if sbytes != twainLen { 59 errc <- fmt.Errorf("sent %d bytes; expected %d", sbytes, twainLen) 60 return 61 } 62 }() 63 }(ln) 64 65 // Connect to listener to retrieve file and verify digest matches 66 // expected. 67 c, err := Dial("tcp", ln.Addr().String()) 68 if err != nil { 69 t.Fatal(err) 70 } 71 defer c.Close() 72 73 h := sha256.New() 74 rbytes, err := io.Copy(h, c) 75 if err != nil { 76 t.Error(err) 77 } 78 79 if rbytes != twainLen { 80 t.Errorf("received %d bytes; expected %d", rbytes, twainLen) 81 } 82 83 if res := hex.EncodeToString(h.Sum(nil)); res != twainSHA256 { 84 t.Error("retrieved data hash did not match") 85 } 86 87 for err := range errc { 88 t.Error(err) 89 } 90 } 91
