mono/mcs/class/System.ServiceModel/System.ServiceModel.Channels/MessageImpl.cs

//
// MessageImpl.cs
//
// Author:
//	Atsushi Enomoto <[email protected]>
//
// Copyright (C) 2006 Novell, Inc.  http://www.novell.com
//
// Permission is hereby granted, free of charge, to any person obtaining
// a copy of this software and associated documentation files (the
// "Software"), to deal in the Software without restriction, including
// without limitation the rights to use, copy, modify, merge, publish,
// distribute, sublicense, and/or sell copies of the Software, and to
// permit persons to whom the Software is furnished to do so, subject to
// the following conditions:
// 
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
// 
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
//
using System;
using System.Collections.Generic;
using System.Runtime.Serialization;
using System.Xml;
using System.IO;

namespace System.ServiceModel.Channels
{
	internal class XmlReaderMessage : Message
	{
		MessageVersion version;
		XmlDictionaryReader reader;
		MessageHeaders headers;
		MessageProperties properties = new MessageProperties ();
		bool is_empty, is_fault, body_started, body_consumed;
		int max_headers;
		Dictionary<XmlQualifiedName,string> attributes;

		public XmlReaderMessage (MessageVersion version, XmlDictionaryReader reader, int maxSizeOfHeaders)
		{
			this.version = version;
			this.reader = reader;
			this.max_headers = maxSizeOfHeaders;

			ReadEnvelopeStart ();
		}

		public override MessageHeaders Headers {
			get {
				if (headers == null)
					ReadHeaders ();
				return headers;
			}
		}

		public override bool IsEmpty {
			get {
				ReadBodyStart ();
				return is_empty;
			}
		}

		public override bool IsFault {
			get {
				ReadBodyStart ();
				return is_fault;
			}
		}

		public override MessageProperties Properties {
			get { return properties; }
		}

		public override MessageVersion Version {
			get { return version; }
		}

		protected override MessageBuffer OnCreateBufferedCopy (
			int maxBufferSize)
		{
			ReadBodyStart ();
			var headers = new MessageHeaders (Headers);
			var props = new MessageProperties (Properties);
			return new DefaultMessageBuffer (maxBufferSize, headers, props, new XmlReaderBodyWriter (reader), IsFault, attributes);
		}

		protected override string OnGetBodyAttribute (
			string localName, string ns)
		{
			ReadBodyStart ();
			string v;
			return attributes.TryGetValue (new XmlQualifiedName (localName, ns), out v) ? v : null;
		}

		protected override XmlDictionaryReader OnGetReaderAtBodyContents ()
		{
			if (reader.ReadState == ReadState.Closed)
				return reader; // silly, but that's what our test expects.
			ReadBodyStart ();
			if (is_empty)
				throw new InvalidOperationException ("The message body is empty.");
			body_consumed = true;
			return reader;
		}

		protected override void OnWriteBodyContents (
			XmlDictionaryWriter writer)
		{
			XmlDictionaryReader reader = GetReaderAtBodyContents ();
			while (!reader.EOF && reader.NodeType != XmlNodeType.EndElement)
				writer.WriteNode (reader, false);
		}

		static readonly char [] whitespaceChars = new char [] {' ', '\t', '\r', '\n'};

		void ReadEnvelopeStart ()
		{
			reader.MoveToContent ();
			if (reader.IsEmptyElement)
				throw new ArgumentException ("Missing message content XML.");
			reader.ReadStartElement ("Envelope", Version.Envelope.Namespace);

			// SOAP Header
			reader.MoveToContent ();
		}

		void ReadHeaders ()
		{
			if (headers != null)
				throw new InvalidOperationException ("XmlReader at headers is already consumed.");

			string envNS = Version.Envelope.Namespace;

			headers = new MessageHeaders (version);
			if (reader.LocalName != "Header" || reader.NamespaceURI != envNS)
				return;

			bool isEmptyHeader = reader.IsEmptyElement;
			reader.ReadStartElement ("Header", envNS);
			reader.MoveToContent ();
			if (isEmptyHeader)
				return;

			int nHeaders = 0;
			while (!reader.EOF && reader.NodeType != XmlNodeType.EndElement) {
				if (reader.NodeType == XmlNodeType.Element) {
					if (nHeaders++ == max_headers)
						throw new InvalidOperationException (String.Format ("Message header size has exceeded the maximum header size {0}", max_headers));
					headers.Add (new MessageHeader.RawMessageHeader (reader, envNS));
				}
				else
					reader.Skip ();
				// FIXME: handle UnderstoodHeaders as well.
				reader.MoveToContent ();
			}
			reader.ReadEndElement ();
			reader.MoveToContent ();
		}

		void ReadBodyStart ()
		{
			if (body_consumed)
				throw new InvalidOperationException ("The message body XmlReader is already consumed.");

			if (body_started)
				return;

			// read headers in advance.
			if (headers == null)
				ReadHeaders ();

			// SOAP Body
			body_started = true;
			is_empty = reader.IsEmptyElement;
			attributes = new Dictionary<XmlQualifiedName,string> ();
			reader.MoveToContent ();
			if (reader.MoveToFirstAttribute ()) {
				do {
					attributes [new XmlQualifiedName (reader.LocalName, reader.NamespaceURI)] = reader.Value;
				} while (reader.MoveToNextAttribute ());
				reader.MoveToElement ();
			}
			reader.ReadStartElement ("Body", Version.Envelope.Namespace);
			if (reader.NodeType == XmlNodeType.EndElement) {
				is_empty = true;
				reader.Read ();
			} else {
				reader.MoveToContent ();
				if (reader.NodeType == XmlNodeType.Element &&
				    reader.LocalName == "Fault" &&
				    reader.NamespaceURI == Version.Envelope.Namespace)
					is_fault = true;
			}
		}
	}

	internal abstract class MessageImplBase : Message
	{
		MessageHeaders headers;
		MessageProperties properties = new MessageProperties ();
		Dictionary<XmlQualifiedName,string> attributes;

		public MessageImplBase (MessageVersion version, string action, Dictionary<XmlQualifiedName,string> attributes)
		{
			headers = new MessageHeaders (version);
			if (action != null)
				headers.Action = action;
			this.attributes = attributes;
		}

		internal Dictionary<XmlQualifiedName,string> Attributes {
			get { return attributes; }
		}

		public override MessageHeaders Headers {
			get { return headers; }
		}

		public override MessageProperties Properties {
			get { return properties; }
		}

		public override MessageVersion Version {
			get { return Headers.MessageVersion; }
		}

		protected override string OnGetBodyAttribute (
			string localName, string ns)
		{
			string v;
			return attributes.TryGetValue (new XmlQualifiedName (localName, ns), out v) ? v : null;
		}

		protected override void OnWriteStartBody (
			XmlDictionaryWriter writer)
		{
			var dic = Constants.SoapDictionary;
			writer.WriteStartElement ("s", dic.Add ("Body"), dic.Add (Version.Envelope.Namespace));
			foreach (var p in Attributes)
				writer.WriteAttributeString (p.Key.Name, p.Key.Namespace, p.Value);
		}
	}

	internal class EmptyMessage : MessageImplBase
	{
		static readonly Dictionary<XmlQualifiedName,string> empty_attributes = new Dictionary<XmlQualifiedName,string> ();

		public EmptyMessage (MessageVersion version, string action)
			: base (version, action, empty_attributes)
		{
		}

		public override bool IsEmpty {
			get { return true; }
		}

		protected override void OnWriteBodyContents (
			XmlDictionaryWriter writer)
		{
		}

		protected override MessageBuffer OnCreateBufferedCopy (
			int maxBufferSize)
		{
			return new DefaultMessageBuffer (Headers, Properties, Attributes);
		}
	}

	internal class SimpleMessage : MessageImplBase
	{
		BodyWriter body;
		bool is_fault;

		public SimpleMessage (MessageVersion version,
			string action, BodyWriter body, bool isFault, Dictionary<XmlQualifiedName,string> attributes)
			: base (version, action, attributes)
		{
			this.body = body;
			this.is_fault = isFault;
		}

		public override bool IsEmpty {
			get { return false; }
		}

		public override bool IsFault {
			get { return is_fault; }
		}

		protected override void OnWriteBodyContents (
			XmlDictionaryWriter writer)
		{
			body.WriteBodyContents (writer);
		}

		protected override MessageBuffer OnCreateBufferedCopy (
			int maxBufferSize)
		{
			var headers = new MessageHeaders (Headers);
			var props = new MessageProperties (Properties);
			var atts = new Dictionary<XmlQualifiedName,string> (Attributes);
			return new DefaultMessageBuffer (maxBufferSize, headers, props, body.CreateBufferedCopy (maxBufferSize), IsFault, atts);
		}
	}
}