RemoteView/Server.cs


using RemoteView.PageHandlers;
using System;
using System.Collections.Generic;
using System.IO;
using System.Net;

namespace RemoteView
{
    class Server
    {
        Dictionary<String, PageHandler> decoder = new Dictionary<string, PageHandler>();
        HttpListener listener;

        private volatile bool running;

        /// <summary>
        /// Constructor
        /// </summary>
        public Server()
        {
            // Homepage 
            decoder.Add("", new HomePageHandler());
            decoder.Add("home", new HomePageHandler());

            decoder.Add("script", new JavascriptPageHandler());

            // information about the system
            decoder.Add("info", new InfoPageHandler());

            // this page processes clicks into windows events
            decoder.Add("click", new ClickPageHandler());

            // image of choosen device as a png
            decoder.Add("screen", new ScreenPageHandler());

            // image of choosen device as a png
            decoder.Add("favicon.ico", new IconPageHandler());

            // 404 error page
            decoder.Add("404", new NotFoundPageHandler());
        }

        /// <summary>
        /// Start server
        /// </summary>
        /// <param name="port">port to listen to</param>
        public void start(int port)
        {
            try
            {
                listener = new HttpListener();
                listener.Prefixes.Add("http://*:" + port + "/");
                listener.Start();
            }
            catch
            {
                Console.WriteLine("Could not listen on port: {0}.", port);
                return;
            }

            this.running = true;

            do
            {
                Stream output = null;
                try
                {
                    // Note: The GetContext method blocks while waiting for a request. 
                    HttpListenerContext context = listener.GetContext();

                    String[] uri = context.Request.RawUrl.Split('/');

#if DEBUG
                    Console.WriteLine(context.Request.RawUrl);
#endif

                    PageHandler page;
                    bool found = decoder.TryGetValue(uri[1], out page);
                    if (!found)
                    {
                        // if page not found display 404 page
                        page = decoder["404"];
                    }

                    HttpListenerResponse response = context.Response;

                    byte[] buffer = page.handleRequest(response, uri);

                    // Get a response stream and write the response to it.
                    response.ContentLength64 = buffer.Length;
                    output = response.OutputStream;
                    output.Write(buffer, 0, buffer.Length);

                    // must close the output stream.
                    output.Close();
                }
                catch (Exception e)
                {
                    Console.WriteLine(e.Message);
                    try
                    {
                        if (output != null) output.Close();
                    }
                    catch { }
                }
            }
            while (this.running);

            listener.Stop();
        }

        /// <summary>
        /// Stop server
        /// </summary>
        public void stop()
        {
            this.running = false;
            this.listener.Stop();
        }

        /// <summary>
        /// is server running?
        /// </summary>
        /// <returns></returns>
        public bool isRunning()
        {
            return this.running == true;
        }
    }
}