2 NNCP -- Node-to-Node CoPy
3 Copyright (C) 2016-2017 Sergey Matveev <stargrave@stargrave.org>
5 This program is free software: you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, either version 3 of the License, or
8 (at your option) any later version.
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with this program. If not, see <http://www.gnu.org/licenses/>.
31 "github.com/davecgh/go-xdr/xdr2"
32 "golang.org/x/crypto/blake2b"
35 func TestTx(t *testing.T) {
36 f := func(hops uint8, pathSrc, data string, nice uint8) bool {
37 if len(pathSrc) > int(MaxPathSize) {
38 pathSrc = pathSrc[:MaxPathSize]
42 spool, err := ioutil.TempDir("", "testtx")
46 defer os.RemoveAll(spool)
47 nodeOur, err := NewNodeGenerate()
51 nodeTgtOur, err := NewNodeGenerate()
55 nodeTgt := nodeTgtOur.Their()
58 LogPath: path.Join(spool, "log.log"),
61 Neigh: make(map[NodeId]*Node, hops),
62 Alias: make(map[string]*NodeId),
64 ctx.Neigh[*nodeOur.Id] = nodeOur.Their()
65 ctx.Neigh[*nodeTgt.Id] = nodeTgt
66 privates := make(map[NodeId]*NodeOur, int(hops)+1)
67 privates[*nodeTgt.Id] = nodeTgtOur
68 privates[*nodeOur.Id] = nodeOur
69 for i := uint8(0); i < hops; i++ {
70 node, err := NewNodeGenerate()
74 ctx.Neigh[*node.Id] = node.Their()
75 privates[*node.Id] = node
76 nodeTgt.Via = append(nodeTgt.Via, node.Id)
78 pkt, err := NewPkt(PktTypeMail, pathSrc)
79 src := strings.NewReader(data)
80 dstNode, err := ctx.Tx(nodeTgt, pkt, 123, int64(src.Len()), src)
85 sentJobs := make([]Job, 0, 1)
86 for txJob := range ctx.Jobs(dstNode.Id, TTx) {
87 sentJobs = append(sentJobs, txJob)
89 if len(sentJobs) != 1 {
93 defer txJob.Fd.Close()
95 if _, err = io.Copy(&bufR, txJob.Fd); err != nil {
99 vias := append(nodeTgt.Via, nodeTgt.Id)
100 for i, hopId := range vias {
101 hopOur := privates[*hopId]
102 foundNode, err := PktEncRead(hopOur, ctx.Neigh, &bufR, &bufW)
106 if *foundNode.Id != *nodeOur.Id {
109 bufR, bufW = bufW, bufR
112 if _, err = xdr.Unmarshal(&bufR, &pkt); err != nil {
115 if *hopId == *nodeTgt.Id {
116 if pkt.Type != PktTypeMail {
119 if !bytes.HasPrefix(pkt.Path[:], []byte(pathSrc)) {
122 if bytes.Compare(bufR.Bytes(), []byte(data)) != 0 {
126 if pkt.Type != PktTypeTrns {
129 if bytes.Compare(pkt.Path[:blake2b.Size256], vias[i+1][:]) != 0 {
136 if err := quick.Check(f, nil); err != nil {