PostModule.cs

using System;
using System.Collections.Generic;
using System.Text;
using MTSC.Exceptions;
using MTSC.Server;
using MTSC.Server.Handlers;

namespace MTSC.Common.Http.ServerModules
{
    public class PostModule : IHttpModule
    {
        private static string urlEncodedHeader = "application/x-www-form-urlencoded";
        private static string multipartHeader = "multipart/form-data";

        public event EventHandler<Dictionary<string, string>> FormReceived;

        #region Private Methods
        private Dictionary<string, string> ParseFormUrlEncoded(HttpMessage request)
        {
            if (!request.ContainsHeader(HttpMessage.EntityHeadersEnum.ContentLength))
            {
                throw new InvalidPostFormException("Form is missing content-length header!");
            }
            else
            {
                Dictionary<string, string> formData = new Dictionary<string, string>();
                int length = int.Parse(request[HttpMessage.EntityHeadersEnum.ContentLength]);
                int step = 0, parsedLength = 0;
                StringBuilder fieldBuilder = new StringBuilder();
                StringBuilder valueBuilder = new StringBuilder();
                /*
                 * Parse request body to obtain the form data.
                 * Example of a form: field1=value1&field2=value2
                 */
                for (int i = 0; i < request.Body.Length; i++)
                {
                    if(request.Body[i] == '\n')
                    {
                        continue;
                    }
                    else if(step == 0)
                    {
                        parsedLength++;
                        if(request.Body[i] == '=')
                        {
                            step = 1;
                        }
                        else
                        {
                            fieldBuilder.Append((char)request.Body[i]);
                        }
                    }
                    else
                    {
                        parsedLength++;
                        if(request.Body[i] == '&')
                        {
                            step = 0;
                            formData[fieldBuilder.ToString()] = valueBuilder.ToString();
                            fieldBuilder.Clear();
                            valueBuilder.Clear();
                        }
                        else
                        {
                            valueBuilder.Append((char)request.Body[i]);
                            if(parsedLength == length)
                            {
                                formData[fieldBuilder.ToString()] = valueBuilder.ToString();
                            }
                        }
                    }
                }
                return formData;
            }
        }

        private Dictionary<string, string> ParseFormMultipart(HttpMessage request)
        {
            throw new NotImplementedException("Multipart form parsing not implemented yet.");
        }
        #endregion

        #region Interface Implementation
        bool IHttpModule.HandleRequest(Server.Server server, HttpHandler handler, ClientData client, HttpMessage request, ref HttpMessage response)
        {
            if(request.Method == HttpMessage.MethodEnum.Post &&
               request.ContainsHeader(HttpMessage.EntityHeadersEnum.ContentType))
            {
                if (request[HttpMessage.EntityHeadersEnum.ContentType].Contains(urlEncodedHeader))
                {
                    Dictionary<string, string> form = ParseFormUrlEncoded(request);
                    FormReceived?.Invoke(this, form);
                    server.LogDebug("Received POST form of " + form.Keys.Count + " keys!");
                    return true;
                }
                else if (request[HttpMessage.EntityHeadersEnum.ContentType].Contains(multipartHeader))
                {
                    Dictionary<string, string> form = ParseFormMultipart(request);
                    FormReceived?.Invoke(this, form);
                    server.LogDebug("Received POST form of " + form.Keys.Count + " keys!");
                    return true;
                }
                else
                {
                    return false;
                }
            }
            else
            {
                return false;
            }
        }

        void IHttpModule.Tick(Server.Server server, HttpHandler handler)
        {
            
        }
        #endregion
    }
}