Как ограничить количество символов, читаемых StreamReader.ReadLine () в .NET? - PullRequest
7 голосов
/ 30 декабря 2008

Я пишу приложение веб-сервера на C # и использую класс StreamReader для чтения из базового NetworkStream:

 NetworkStream ns = new NetworkStream(clientSocket);
 StreamReader sr = new StreamReader(ns);
 String request = sr.ReadLine();

Этот код подвержен DoS-атакам, потому что если злоумышленник никогда не отключится, мы никогда не закончим чтение строки. Есть ли способ ограничить количество символов, читаемых StreamReader.ReadLine () в .NET?

Ответы [ 4 ]

10 голосов
/ 30 декабря 2008

Вы должны будете использовать перегрузку Read(char[], int, int) (которая ограничивает длину) и выполнить свое собственное обнаружение конца строки; не должно быть слишком сложно.

Для слегка ленивой версии (в которой используется версия с односимвольным чтением):

static IEnumerable<string> ReadLines(string path, int maxLineLength)
{
    StringBuilder currentLine = new StringBuilder(maxLineLength);
    using (var reader = File.OpenText(path))
    {
        int i;
        while((i = reader.Read()) > 0) {
            char c = (char) i;
            if(c == '\r' || c == '\n') {
                yield return currentLine.ToString();
                currentLine.Length = 0;
                continue;
            }
            currentLine.Append((char)c);
            if (currentLine.Length > maxLineLength)
            {
                throw new InvalidOperationException("Max length exceeded");
            }
        }
        if (currentLine.Length > 0)
        {
            yield return currentLine.ToString();
        }                
    }
}
4 голосов
/ 30 декабря 2008

Вам может понадобиться одна из StreamReader.Read перегрузок:

Взято из http://msdn.microsoft.com/en-us/library/9kstw824.aspx

    using (StreamReader sr = new StreamReader(path)) 
    {
        //This is an arbitrary size for this example.
        char[] c = null;

        while (sr.Peek() >= 0) 
        {
            c = new char[5];
            sr.Read(c, 0, c.Length);
            //The output will look odd, because
            //only five characters are read at a time.
            Console.WriteLine(c);
        }
    }

Сфокусируйтесь на линии sr.Read(c, 0, c.Length). Это читает только 5 символов из потока и помещается в массив c. Вы можете изменить значение 5 на желаемое.

2 голосов
/ 02 января 2009

Вот мое собственное решение, основанное на решении Марка Гравелла:

using System;
using System.IO;
using System.Text;

namespace MyProject
{
    class StreamReaderExt : StreamReader
    {

        public StreamReaderExt(Stream s, Encoding e) : base(s, e)
        {            
        }

        /// <summary>
        /// Reads a line of characters terminated by CR+LF from the current stream and returns the data as a string
        /// </summary>
        /// <param name="maxLineLength">Maximum allowed line length</param>
        /// <exception cref="System.IO.IOException" />
        /// <exception cref="System.InvalidOperationException">When string read by this method exceeds the maximum allowed line length</exception>
        /// <returns></returns>
        public string ReadLineCRLF(int maxLineLength)
        {
            StringBuilder currentLine = new StringBuilder(maxLineLength);

            int i;
            bool foundCR = false;
            bool readData = false;

            while ((i = Read()) > 0)
            {

                readData = true;

                char c = (char)i;

                if (foundCR)
                {
                    if (c == '\r')
                    {
                        // If CR was found before , and the next character is also CR,
                        // adding previously skipped CR to the result string
                        currentLine.Append('\r');
                        continue;
                    }
                    else if (c == '\n')
                    {
                        // LF found, finished reading the string
                        return currentLine.ToString();
                    }
                    else
                    {
                        // If CR was found before , but the next character is not LF,
                        // adding previously skipped CR to the result string
                        currentLine.Append('\r');
                        foundCR = false;
                    }
                }
                else // CR not found
                {
                    if (c == '\r')
                    {
                        foundCR = true;
                        continue;
                    }
                }

                currentLine.Append((char)c);
                if (currentLine.Length > maxLineLength)
                {
                    throw new InvalidOperationException("Max line length exceeded");
                }
            }

            if (foundCR)
            {
                // If CR was found before, and the end of the stream has been reached, appending the skipped CR character
                currentLine.Append('\r');
            }

            if (readData)
            {
                return currentLine.ToString();
            }

            // End of the stream reached
            return null;

        }
    }
}

Этот фрагмент кода предоставляется "КАК ЕСТЬ" БЕЗ ГАРАНТИИ.

1 голос
/ 30 декабря 2008

Вы всегда можете использовать «.Read (...)», и MSDN рекомендует делать это в такой ситуации, как ваша.

http://msdn.microsoft.com/en-us/library/system.io.streamreader.readline.aspx

...