public static byte[] Serialize(Header header) { using(MemoryStream ms = new MemoryStream()) using(PayloadWriter pw = new PayloadWriter()) { Serializer.Serialize(ms, header); pw.WriteUShort((ushort)ms.Length); pw.WriteBytes(ms.ToArray()); return pw.ToByteArray(); } }
/// <summary> /// Send data to the established connection /// </summary> /// <param name="Message">The data to send</param> /// <param name="Header">The Header to use for adding additional information</param> /// <param name="feature">The Feature that has been used for this Message</param> /// <param name="OpSocket">The OperationalSocket that has been used for this Message</param> internal int SendMessage(IMessage Message, Header Header, Feature feature = null, OperationalSocket OpSocket = null) { lock (SendLock) { if (!Connected) return -1; if (Message == null) throw new ArgumentException("Message cannot be null"); if (Header == null) throw new ArgumentException("Header cannot be null"); ushort HeaderId = OpSocket != null ? OpSocket.Headers.GetHeaderId(Header) : Headers.GetHeaderId(Header); byte[] SerializedHeader = Header.Serialize(Header); uint messageId = OpSocket != null ? OpSocket.MessageHandler.GetMessageId(Message.GetType()) : messageHandler.GetMessageId(Message.GetType()); if (SerializedHeader.Length >= MAX_PACKET_SIZE) throw new ArgumentException("Header length cannot be greater then " + MAX_PAYLOAD); using (MemoryStream outStream = new MemoryStream()) using (PayloadWriter pw = new PayloadWriter(outStream)) { pw.WriteBytes(new byte[HEADER_SIZE], 0, HEADER_SIZE); //reserve space pw.WriteBytes(SerializedHeader); pw.WriteUInteger(messageId); int packetSize = messageHandler.EncryptMessage(this, Message, outStream); if (pw.Length > MAX_PACKET_SIZE) throw new OverflowException("Message size cannot be greater then " + MAX_PACKET_SIZE); int PayloadLength = pw.Length - Connection.HEADER_SIZE; byte CurPacketId = 0; int FeatureId = feature != null ? feature.GetFeatureId() : -1; ushort ConnectionId = OpSocket != null ? OpSocket.ConnectionId : (ushort)0; byte checksum = 0; checksum += (byte)PayloadLength; checksum += CurPacketId; checksum += (byte)ConnectionId; checksum += (byte)HeaderId; checksum += (byte)FeatureId; pw.Position = 0; pw.WriteThreeByteInteger(PayloadLength); //length pw.WriteByte(CurPacketId); //cur packet id pw.WriteUShort(ConnectionId); //Connection Id pw.WriteUShort(HeaderId); //Header Id pw.WriteByte(checksum); pw.WriteInteger(FeatureId); //encrypt the header lock (HeaderEncryption) { HeaderEncryption.Encrypt(pw.GetBuffer(), 0, HEADER_SIZE); byte[] temp = pw.GetBuffer(); headerConfuser.Obfuscate(ref temp, 0); } int SendNum = 0; try { for (int i = 0; i < outStream.Length;) { int len = i + 65535 < outStream.Length ? 65535 : (int)outStream.Length - i; Handle.Send(outStream.GetBuffer(), i, len, SocketFlags.None); i += len; SendNum += len; } } catch (Exception ex) { Disconnect(); return -1; } SysLogger.Log("Send " + outStream.Length, SysLogType.Network); PacketsOut++; DataOut += (ulong)outStream.Length; this.LastPacketSendSW = Stopwatch.StartNew(); return SendNum; } /*using (OptimizedPayloadStream ms = new OptimizedPayloadStream(SerializedHeader, HeaderId, feature, OpSocket)) { ms.Write(BitConverter.GetBytes(messageId), 0, 4); MemoryStream stream = ms.PayloadFrames[ms.PayloadFrames.Count - 1]; int ReservedPos = (int)stream.Position; ms.Write(new byte[3], 0, 3); //reserve space ms.WritingMessage = true; Serializer.Serialize(ms, Message); ms.WritingMessage = false; using (PayloadWriter pw = new PayloadWriter(new MemoryStream(stream.GetBuffer()))) { pw.Position = ReservedPos; //skip MessageId data pw.WriteThreeByteInteger(ms.MessageLength);//Reserved Space + MessageId = 7 } ms.Commit(this); for (int i = 0; i < ms.PayloadFrames.Count; i++) { stream = ms.PayloadFrames[i]; lock (HeaderEncryption) { HeaderEncryption.Encrypt(stream.GetBuffer(), 0, HEADER_SIZE); byte[] temp = stream.GetBuffer(); headerConfuser.Obfuscate(ref temp, 0); } //lock (PayloadEncryption) //{ // PayloadEncryption.Encrypt(stream.GetBuffer(), HEADER_SIZE, (int)stream.Length - HEADER_SIZE); //} Handle.Send(stream.GetBuffer(), 0, (int)stream.Length, SocketFlags.None); } }*/ } }
public byte[] Serialize(object obj) { if (obj == null) { return new byte[] { (byte)ObjectTypes.Null } } ; PayloadWriter pw = new PayloadWriter(); Type ObjectType = obj.GetType(); if (ObjectType == typeof(byte)) { pw.WriteByte((byte)ObjectTypes.Byte); pw.WriteByte((byte)obj); } else if (ObjectType == typeof(byte[])) { byte[] data = (byte[])obj; pw.WriteByte((byte)ObjectTypes.ByteArray); pw.WriteInteger(data.Length); pw.WriteBytes(data); } else if (ObjectType == typeof(short)) { pw.WriteByte((byte)ObjectTypes.Short); pw.WriteShort((short)obj); } else if (ObjectType == typeof(ushort)) { pw.WriteByte((byte)ObjectTypes.UShort); pw.WriteUShort((ushort)obj); } else if (ObjectType == typeof(int)) { pw.WriteByte((byte)ObjectTypes.Int); pw.WriteInteger((int)obj); } else if (ObjectType == typeof(uint)) { pw.WriteByte((byte)ObjectTypes.UINT); pw.WriteUInteger((uint)obj); } else if (ObjectType == typeof(long)) { pw.WriteByte((byte)ObjectTypes.Long); pw.WriteLong((long)obj); } else if (ObjectType == typeof(bool)) { pw.WriteByte((byte)ObjectTypes.Bool); pw.WriteByte((bool)obj ? (byte)1 : (byte)0); } else if (ObjectType == typeof(string)) { pw.WriteByte((byte)ObjectTypes.String); pw.WriteString(obj.ToString()); } else if (ObjectType == typeof(SolidBrush)) { pw.WriteByte((byte)ObjectTypes.SolidBrush); pw.WriteByte(((SolidBrush)obj).Color.A); pw.WriteByte(((SolidBrush)obj).Color.R); pw.WriteByte(((SolidBrush)obj).Color.G); pw.WriteByte(((SolidBrush)obj).Color.B); } else if (ObjectType == typeof(Rectangle)) { pw.WriteByte((byte)ObjectTypes.Rectangle); pw.WriteInteger(((Rectangle)obj).X); pw.WriteInteger(((Rectangle)obj).Y); pw.WriteInteger(((Rectangle)obj).Width); pw.WriteInteger(((Rectangle)obj).Height); } else if (ObjectType == typeof(Size)) { pw.WriteByte((byte)ObjectTypes.Size); pw.WriteInteger(((Size)obj).Width); pw.WriteInteger(((Size)obj).Height); } else if (ObjectType == typeof(Bitmap) || ObjectType == typeof(Image)) { pw.WriteByte((byte)ObjectTypes.Bitmap); lzwCompression.Compress((Bitmap)obj, pw.vStream); } else { MemoryStream ms = new MemoryStream(); BinaryFormatter bf = new BinaryFormatter(); bf.Serialize(ms, obj); pw.WriteByte((byte)ObjectTypes.Other); pw.WriteBytes(ms.GetBuffer(), 0, (int)ms.Length); ms.Close(); } return(pw.ToByteArray()); }
public byte[] Serialize(object obj) { if (obj == null) return new byte[] { (byte)ObjectTypes.Null }; PayloadWriter pw = new PayloadWriter(); Type ObjectType = obj.GetType(); if (ObjectType == typeof(byte)) { pw.WriteByte((byte)ObjectTypes.Byte); pw.WriteByte((byte)obj); } else if (ObjectType == typeof(byte[])) { byte[] data = (byte[])obj; pw.WriteByte((byte)ObjectTypes.ByteArray); pw.WriteInteger(data.Length); pw.WriteBytes(data); } else if (ObjectType == typeof(short)) { pw.WriteByte((byte)ObjectTypes.Short); pw.WriteShort((short)obj); } else if (ObjectType == typeof(ushort)) { pw.WriteByte((byte)ObjectTypes.UShort); pw.WriteUShort((ushort)obj); } else if (ObjectType == typeof(int)) { pw.WriteByte((byte)ObjectTypes.Int); pw.WriteInteger((int)obj); } else if (ObjectType == typeof(uint)) { pw.WriteByte((byte)ObjectTypes.UINT); pw.WriteUInteger((uint)obj); } else if (ObjectType == typeof(long)) { pw.WriteByte((byte)ObjectTypes.Long); pw.WriteLong((long)obj); } else if (ObjectType == typeof(bool)) { pw.WriteByte((byte)ObjectTypes.Bool); pw.WriteByte((bool)obj ? (byte)1 : (byte)0); } else if (ObjectType == typeof(string)) { pw.WriteByte((byte)ObjectTypes.String); pw.WriteString(obj.ToString()); } else if (ObjectType == typeof(SolidBrush)) { pw.WriteByte((byte)ObjectTypes.SolidBrush); pw.WriteByte(((SolidBrush)obj).Color.A); pw.WriteByte(((SolidBrush)obj).Color.R); pw.WriteByte(((SolidBrush)obj).Color.G); pw.WriteByte(((SolidBrush)obj).Color.B); } else if (ObjectType == typeof(Rectangle)) { pw.WriteByte((byte)ObjectTypes.Rectangle); pw.WriteInteger(((Rectangle)obj).X); pw.WriteInteger(((Rectangle)obj).Y); pw.WriteInteger(((Rectangle)obj).Width); pw.WriteInteger(((Rectangle)obj).Height); } else if (ObjectType == typeof(Size)) { pw.WriteByte((byte)ObjectTypes.Size); pw.WriteInteger(((Size)obj).Width); pw.WriteInteger(((Size)obj).Height); } else if (ObjectType == typeof(Bitmap) || ObjectType == typeof(Image)) { pw.WriteByte((byte)ObjectTypes.Bitmap); lzwCompression.Compress((Bitmap)obj, pw.vStream); } else { MemoryStream ms = new MemoryStream(); BinaryFormatter bf = new BinaryFormatter(); bf.Serialize(ms, obj); pw.WriteByte((byte)ObjectTypes.Other); pw.WriteBytes(ms.GetBuffer(), 0, (int)ms.Length); ms.Close(); } return pw.ToByteArray(); }
/// <summary> /// Save the changes and apply Header Information /// </summary> public void Commit(Connection conn) { bool UseFragments = PayloadFrames.Count > 1; int FullLength = 0; byte FragmentId = (byte)(UseFragments ? 1 : 0); //count all the payload, this is the original size for (int i = 0; i < PayloadFrames.Count; i++) { FullLength += (int)PayloadFrames[i].Length - Connection.HEADER_SIZE; } #region Encryption & Compression //compress all the data for (int i = 0; i < PayloadFrames.Count; i++) { MemoryStream stream = PayloadFrames[i]; if (CompressionAlgorithm.QuickLZ == (conn.CompressionAlgorithm & CompressionAlgorithm.QuickLZ)) { UnsafeQuickLZ quickLz = new UnsafeQuickLZ(); byte[] compressed = quickLz.compress(stream.GetBuffer(), (uint)Connection.HEADER_SIZE, (uint)stream.Length - Connection.HEADER_SIZE); if (compressed != null && compressed.Length + Connection.HEADER_SIZE < stream.Length) //only apply compression if it's smaller then the original data { stream.Position = Connection.HEADER_SIZE; stream.Write(compressed, 0, compressed.Length); if (stream.Length != compressed.Length + Connection.HEADER_SIZE) { stream.SetLength(compressed.Length + Connection.HEADER_SIZE); } } } } //encrypt all the data for (int i = 0; i < PayloadFrames.Count; i++) { MemoryStream stream = PayloadFrames[i]; if (EncAlgorithm.HwAES == (conn.EncryptionAlgorithm & EncAlgorithm.HwAES)) { lock (conn.EncAES) { //no need to re-size the stream here, AES will encrypt at the same size or bigger then the stream, so data will be overwritten byte[] encrypted = conn.EncAES.Encrypt(stream.GetBuffer(), Connection.HEADER_SIZE, (int)stream.Length - Connection.HEADER_SIZE); stream.Position = Connection.HEADER_SIZE; stream.Write(encrypted, 0, encrypted.Length); } } if (EncAlgorithm.WopEx == (conn.EncryptionAlgorithm & EncAlgorithm.WopEx)) { lock (conn.PayloadEncryption) { conn.PayloadEncryption.Encrypt(stream.GetBuffer(), Connection.HEADER_SIZE, (int)stream.Length - Connection.HEADER_SIZE); } } } #endregion for (int i = 0; i < PayloadFrames.Count; i++) { PayloadWriter pw = new PayloadWriter(PayloadFrames[i]); pw.Position = 0; //if (pw.Length - Connection.HEADER_SIZE > ushort.MaxValue) // throw new OverflowException(); //should never happen if Write(...) is handled correctly ushort PayloadLength = (ushort)(pw.Length - Connection.HEADER_SIZE); byte CurPacketId = 0; int FeatureId = feature != null?feature.GetFeatureId() : -1; ushort ConnectionId = OpSocket != null ? OpSocket.ConnectionId : (ushort)0; if (i + 1 >= PayloadFrames.Count && UseFragments) { FragmentId += 128; } byte checksum = 0; checksum += (byte)PayloadLength; checksum += FragmentId; checksum += CurPacketId; checksum += (byte)ConnectionId; checksum += (byte)HeaderId; checksum += (byte)FullLength; checksum += (byte)FeatureId; pw.WriteUShort(PayloadLength); //length pw.WriteByte(CurPacketId); //cur packet id pw.WriteUShort(ConnectionId); //Connection Id pw.WriteUShort(HeaderId); //Header Id pw.WriteByte(FragmentId); pw.WriteByte(checksum); pw.WriteThreeByteInteger(FullLength); //the full packet size, mainly used for Fragmentation pw.WriteInteger(FeatureId); if (UseFragments) { FragmentId++; } } }