HTTPServer/Client.cs at master · iEmiya/HTTPServer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
using System;
using System.IO;
using System.Net;
using System.Net.Sockets;
using System.Text;
using System.Text.RegularExpressions;
using NLog;

namespace HTTPServer
{
	// Класс-обработчик клиента
	class Client
	{
		// Отправка страницы с ошибкой
		private static void SendError(TcpClient client, HttpStatusCode code)
		{
			// Получаем строку вида "200 OK"
			// HttpStatusCode хранит в себе все статус-коды HTTP/1.1
			string codeStr = ((int)code).ToString() + " " + code.ToString();
			// Код простой HTML-странички
			string html = "<html><body><h1>" + codeStr + "</h1></body></html>";
			// Необходимые заголовки: ответ сервера, тип и длина содержимого. После двух пустых строк - само содержимое
			string str = "HTTP/1.1 " + codeStr + "\nContent-type: text/html\nContent-Length:" + html.Length.ToString() + "\n\n" + html;
			// Приведем строку к виду массива байт
			byte[] buffer = Encoding.ASCII.GetBytes(str);
			// Отправим его клиенту
			client.GetStream().Write(buffer, 0, buffer.Length);
			// Закроем соединение
			client.Close();
		}

		private readonly Logger _log;

		// Конструктор класса. Ему нужно передавать принятого клиента от TcpListener
		public Client(Logger log, TcpClient client)
		{
			_log = log;
			_log.Trace("Init client");

			// Объявим строку, в которой будет хранится запрос клиента
			string request = "";
			// Буфер для хранения принятых от клиента данных
			byte[] buffer = new byte[1024];
			// Переменная для хранения количества байт, принятых от клиента
			int count;

			_log.Trace("Read request");
			// Читаем из потока клиента до тех пор, пока от него поступают данные
			while ((count = client.GetStream().Read(buffer, 0, buffer.Length)) > 0)
			{
				// Преобразуем эти данные в строку и добавим ее к переменной Request
				request += Encoding.ASCII.GetString(buffer, 0, count);
				// Запрос должен обрываться последовательностью \r\n\r\n
				// Либо обрываем прием данных сами, если длина строки Request превышает 4 килобайта
				// Нам не нужно получать данные из POST-запроса (и т. п.), а обычный запрос
				// по идее не должен быть больше 4 килобайт
				if (request.IndexOf("\r\n\r\n", StringComparison.Ordinal) >= 0 || request.Length > 4096)
				{
					break;
				}
			}

			_log.Trace("Request match");
			// Парсим строку запроса с использованием регулярных выражений
			// При этом отсекаем все переменные GET-запроса
			Match reqMatch = Regex.Match(request, @"^\w+\s+([^\s\?]+)[^\s]*\s+HTTP/.*|");

			// Если запрос не удался
			if (reqMatch == Match.Empty)
			{
				// Передаем клиенту ошибку 400 - неверный запрос
				_log.Warn(HttpStatusCode.BadRequest + ": " + request);
				SendError(client, HttpStatusCode.BadRequest);
				return;
			}

			_log.Trace("Get request Uri");
			// Получаем строку запроса
			string requestUri = reqMatch.Groups[1].Value;


			_log.Trace("Unescape request Uri");
			// Приводим ее к изначальному виду, преобразуя экранированные символы
			// Например, "%20" -> " "
			requestUri = Uri.UnescapeDataString(requestUri);

			// Если в строке содержится двоеточие, передадим ошибку 400
			// Это нужно для защиты от URL типа http://example.com/../../file.txt
			if (requestUri.IndexOf("..", StringComparison.Ordinal) >= 0)
			{
				_log.Warn(HttpStatusCode.BadRequest + ": " + request);
				SendError(client, HttpStatusCode.BadRequest);
				return;
			}

			if (string.IsNullOrEmpty(requestUri))
			{
				requestUri += "/";
			}

			// Если строка запроса оканчивается на "/", то добавим к ней index.html
			if (requestUri.EndsWith("/"))
			{
				requestUri += "index.html";
			}

			string filePath;
			if (Settings.IsSite)
			{
				filePath = Settings.Folder + requestUri;
			}
			else
			{
				string fileName;
				try
				{
					fileName = Path.GetFileName(requestUri);
				}
				catch (Exception)
				{
					_log.Warn(HttpStatusCode.BadRequest + ": " + request);
					SendError(client, HttpStatusCode.BadRequest);
					return;
				}

				if (!Settings.Files.TryGetValue(fileName, out filePath))
				{
					log.Warn(HttpStatusCode.Forbidden + ": " + request);
					SendError(client, HttpStatusCode.Forbidden);
					return;
				}
			}

			// Если в папке www не существует данного файла, посылаем ошибку 404
			if (!File.Exists(filePath))
			{
				_log.Warn(HttpStatusCode.NotFound + ": " + request + " => " + filePath);
				SendError(client, HttpStatusCode.NotFound);
				return;
			}

			_log.Trace("Get Mime type");
			// Тип содержимого
			string contentType = MimeTypes.GetMimeType(requestUri);


			_log.Trace("Read file from drive");
			// Открываем файл, страхуясь на случай ошибки
			FileStream fs;
			try
			{
				fs = new FileStream(filePath, FileMode.Open, FileAccess.Read, FileShare.Read);
			}
			catch (Exception)
			{
				// Если случилась ошибка, посылаем клиенту ошибку 500
				_log.Warn(HttpStatusCode.InternalServerError + ": " + request + " => " + filePath);
				SendError(client, HttpStatusCode.InternalServerError);
				return;
			}

			_log.Trace("Create response");
			try
			{
				// Посылаем заголовкиy
				string headers = "HTTP/1.1 200 OK\nContent-Type: " + contentType + "\nContent-Length: " + fs.Length + "\n\n";
				byte[] headersBuffer = Encoding.ASCII.GetBytes(headers);
				client.GetStream().Write(headersBuffer, 0, headersBuffer.Length);

				// Пока не достигнут конец файла
				while (fs.Position < fs.Length)
				{
					// Читаем данные из файла
					count = fs.Read(buffer, 0, buffer.Length);
					// И передаем их клиенту
					client.GetStream().Write(buffer, 0, count);
				}

			}
			catch (Exception ex)
			{
				_log.Trace(ex, "Failed to create response");
			}
			finally
			{
				// Закроем файл и соединение
				fs.Close();
				try
				{
					client.Close();
				}
				catch (Exception ignore)
				{
				}

			}


			_log.Info(HttpStatusCode.OK + ": " + request + " => " + filePath);
		}
	}
}