uNet2/SequenceContext.cs at master · maddnias/uNet2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using uNet2.Network;
using uNet2.Packet;
using uNet2.Peer;

namespace uNet2
{
    public class SequenceContext
    {
        public Guid SequenceGuid { get; set; }
        internal SequenceInitPacket InitPacket { get; set; }
        internal ISequencePacket[] SequencePackets { get; set; }
        internal byte[] InitPacketBuffer { get; set; }
        internal List<byte[]> SequencePacketBuffers { get; set; }

        internal SequenceContext(byte[] initPacketBuffer, List<byte[]> sequencePacketBuffers)
        {
            InitPacketBuffer = initPacketBuffer;
            SequencePacketBuffers = sequencePacketBuffers;
        }

        internal SequenceContext(SequenceInitPacket initPacket, ISequencePacket[] sequencePackets,
            byte[] initPacketBuffer, List<byte[]> sequencePacketBuffers)
        {
            InitPacket = initPacket;
            SequencePackets = sequencePackets;
            InitPacketBuffer = initPacketBuffer;
            SequencePacketBuffers = sequencePacketBuffers;
        }

        internal static SequenceContext OperationCreateFromPacket(IDataPacket data, int fragmentSize, Guid operationGuid)
        {
            var seqCtx = CreateFromPacket(data, fragmentSize);
            seqCtx.InitPacket.IsOperation = true;
            seqCtx.InitPacket.OperationGuid = operationGuid;
            return seqCtx;
        }

        public static SequenceContext CreateFromPacket(IDataPacket data, int fragmentSize)
        {
            SequenceInitPacket initPacket;
            var ms = new MemoryStream();
            data.SerializeTo(ms);
            NetworkWriter.PrependStreamSize(ms);
            var completeBuff = ms.ToArray();

            Guid seqGuid;
            var sequence = new SequenceHandler().CreateSequence(data, completeBuff, fragmentSize, out initPacket,
                out seqGuid);
            var initPacketStream = new MemoryStream();

            initPacket.SerializeTo(initPacketStream);

#if DEBUG
          //  var sequenceStreams = new MemoryStream[sequence.Length];
          //  var sequenceBuffs = new List<byte[]>();
            //for (var i = 0; i < sequence.Count; i++)
            //{
            //    var seq = sequence[i];
            //    sequenceStreams[i] = new MemoryStream();
            //    seq.SerializeTo(sequenceStreams[i]);
            //    sequenceBuffs.Add(sequenceStreams[i].ToArray());
            //}
#endif
            var seqCtx = new SequenceContext(initPacket, sequence, initPacketStream.ToArray(), null)
            {
                SequenceGuid = seqGuid
            };
            return seqCtx;
        }
    }
}