1 // System.Runtime.Remoting.Channels.Tcp.TcpMessageIO.cs
3 // Author: Lluis Sanchez Gual (lluis@ideary.com)
5 // (C) 2002 Lluis Sanchez Gual
8 // Permission is hereby granted, free of charge, to any person obtaining
9 // a copy of this software and associated documentation files (the
10 // "Software"), to deal in the Software without restriction, including
11 // without limitation the rights to use, copy, modify, merge, publish,
12 // distribute, sublicense, and/or sell copies of the Software, and to
13 // permit persons to whom the Software is furnished to do so, subject to
14 // the following conditions:
16 // The above copyright notice and this permission notice shall be
17 // included in all copies or substantial portions of the Software.
19 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
20 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
21 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
22 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
23 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
24 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
25 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
29 using System.Runtime.Serialization;
30 using System.Runtime.Serialization.Formatters.Binary;
31 using System.Collections;
34 using System.Net.Sockets;
36 namespace System.Runtime.Remoting.Channels.Tcp
38 enum MessageStatus { MethodMessage = 0, CancelSignal = 1, Unknown = 10}
40 internal class TcpMessageIO
42 static byte[][] _msgHeaders =
44 new byte[] { (byte)'.', (byte)'N', (byte)'E', (byte)'T', 1, 0 },
45 new byte[] { 255, 255, 255, 255, 255, 255 }
48 public static int DefaultStreamBufferSize = 1000;
50 // Identifies an incoming message
51 public static MessageStatus ReceiveMessageStatus (Stream networkStream, byte[] buffer)
54 StreamRead (networkStream, buffer, 6);
55 } catch (Exception ex) {
56 throw new RemotingException ("Tcp transport error.", ex);
61 bool[] isOnTrack = new bool[_msgHeaders.Length];
62 bool atLeastOneOnTrack = true;
65 while (atLeastOneOnTrack)
67 atLeastOneOnTrack = false;
69 for (int n = 0; n<_msgHeaders.Length; n++)
71 if (i > 0 && !isOnTrack[n]) continue;
73 isOnTrack[n] = (c == _msgHeaders[n][i]);
74 if (isOnTrack[n] && (i == _msgHeaders[n].Length-1)) return (MessageStatus) n;
75 atLeastOneOnTrack = atLeastOneOnTrack || isOnTrack[n];
79 return MessageStatus.Unknown;
81 catch (Exception ex) {
82 throw new RemotingException ("Tcp transport error.", ex);
86 static bool StreamRead (Stream networkStream, byte[] buffer, int count)
90 int pr = networkStream.Read (buffer, nr, count - nr);
92 throw new RemotingException ("Connection closed");
98 public static void SendMessageStream (Stream networkStream, Stream data, ITransportHeaders requestHeaders, byte[] buffer)
100 if (buffer == null) buffer = new byte[DefaultStreamBufferSize];
102 // Writes the message start header
103 byte[] dotnetHeader = _msgHeaders[(int) MessageStatus.MethodMessage];
104 networkStream.Write(dotnetHeader, 0, dotnetHeader.Length);
106 // Writes header tag (0x0000 if request stream, 0x0002 if response stream)
107 if(requestHeaders[CommonTransportKeys.RequestUri]!=null) buffer [0] = (byte) 0;
108 else buffer[0] = (byte) 2;
109 buffer [1] = (byte) 0 ;
112 buffer [2] = (byte) 0;
114 // Writes assemblyID????
115 buffer [3] = (byte) 0;
117 // Writes the length of the stream being sent (not including the headers)
118 int num = (int)data.Length;
119 buffer [4] = (byte) num;
120 buffer [5] = (byte) (num >> 8);
121 buffer [6] = (byte) (num >> 16);
122 buffer [7] = (byte) (num >> 24);
123 networkStream.Write(buffer, 0, 8);
125 // Writes the message headers
126 SendHeaders (networkStream, requestHeaders, buffer);
129 if (data is MemoryStream)
131 // The copy of the stream can be optimized. The internal
132 // buffer of MemoryStream can be used.
133 MemoryStream memStream = (MemoryStream)data;
134 networkStream.Write (memStream.GetBuffer(), 0, (int)memStream.Length);
138 int nread = data.Read (buffer, 0, buffer.Length);
141 networkStream.Write (buffer, 0, nread);
142 nread = data.Read (buffer, 0, buffer.Length);
147 static byte[] msgUriTransportKey = new byte[] { 4, 0, 1, 1 };
148 static byte[] msgContentTypeTransportKey = new byte[] { 6, 0, 1, 1 };
149 static byte[] msgDefaultTransportKey = new byte[] { 1, 0, 1 };
150 static byte[] msgHeaderTerminator = new byte[] { 0, 0 };
152 private static void SendHeaders(Stream networkStream, ITransportHeaders requestHeaders, byte[] buffer)
154 // Writes the headers as a sequence of strings
155 if (networkStream != null)
157 IEnumerator e = requestHeaders.GetEnumerator();
160 DictionaryEntry hdr = (DictionaryEntry)e.Current;
161 switch (hdr.Key.ToString())
163 case CommonTransportKeys.RequestUri:
164 networkStream.Write (msgUriTransportKey, 0, 4);
167 networkStream.Write (msgContentTypeTransportKey, 0, 4);
170 networkStream.Write (msgDefaultTransportKey, 0, 3);
171 SendString (networkStream, hdr.Key.ToString(), buffer);
172 networkStream.WriteByte (1);
175 SendString (networkStream, hdr.Value.ToString(), buffer);
178 networkStream.Write (msgHeaderTerminator, 0, 2); // End of headers
181 public static ITransportHeaders ReceiveHeaders (Stream networkStream, byte[] buffer)
183 StreamRead (networkStream, buffer, 2);
185 byte headerType = buffer [0];
186 TransportHeaders headers = new TransportHeaders ();
188 while (headerType != 0)
191 StreamRead (networkStream, buffer, 1); // byte 1
194 case 4: key = CommonTransportKeys.RequestUri; break;
195 case 6: key = "Content-Type"; break;
196 case 1: key = ReceiveString (networkStream, buffer); break;
197 default: throw new NotSupportedException ("Unknown header code: " + headerType);
199 StreamRead (networkStream, buffer, 1); // byte 1
200 headers[key] = ReceiveString (networkStream, buffer);
202 StreamRead (networkStream, buffer, 2);
203 headerType = buffer [0];
209 public static Stream ReceiveMessageStream (Stream networkStream, out ITransportHeaders headers, byte[] buffer)
213 if (buffer == null) buffer = new byte[DefaultStreamBufferSize];
215 // Reads header tag: 0 -> Stream with headers or 2 -> Response Stream
217 // Gets the length of the data stream
218 StreamRead (networkStream, buffer, 8);
220 int byteCount = (buffer [4] | (buffer [5] << 8) |
221 (buffer [6] << 16) | (buffer [7] << 24));
224 headers = ReceiveHeaders (networkStream, buffer);
226 byte[] resultBuffer = new byte[byteCount];
227 StreamRead (networkStream, resultBuffer, byteCount);
229 return new MemoryStream (resultBuffer);
232 private static void SendString (Stream networkStream, string str, byte[] buffer)
234 // Allocates a buffer. Use the internal buffer if it is
235 // big enough. If not, create a new one.
237 int maxBytes = Encoding.UTF8.GetMaxByteCount(str.Length)+4; //+4 bytes for storing the string length
238 if (maxBytes > buffer.Length)
239 buffer = new byte[maxBytes];
241 int num = Encoding.UTF8.GetBytes (str, 0, str.Length, buffer, 4);
243 // store number of bytes (not number of chars!)
245 buffer [0] = (byte) num;
246 buffer [1] = (byte) (num >> 8);
247 buffer [2] = (byte) (num >> 16);
248 buffer [3] = (byte) (num >> 24);
250 // Write the string bytes
251 networkStream.Write (buffer, 0, num + 4);
254 private static string ReceiveString (Stream networkStream, byte[] buffer)
256 StreamRead (networkStream, buffer, 4);
258 // Reads the number of bytes (not chars!)
260 int byteCount = (buffer [0] | (buffer [1] << 8) |
261 (buffer [2] << 16) | (buffer [3] << 24));
263 if (byteCount == 0) return string.Empty;
265 // Allocates a buffer of the correct size. Use the
266 // internal buffer if it is big enough
268 if (byteCount > buffer.Length)
269 buffer = new byte[byteCount];
273 StreamRead (networkStream, buffer, byteCount);
274 char[] chars = Encoding.UTF8.GetChars (buffer, 0, byteCount);
276 return new string (chars);