Beispiel #1
0
 /// <summary>
 /// Push new data into the parser
 /// </summary>
 /// <param name="s"></param>
 public void Receive(IDataStream s)
 {
     lock (contentLock)
     {
         while (s.Length > 0)
         {
             while (!readingPayload && s.Length > 0)
             {
                 missing = parser.ParseHeader(s);
                 if (missing <= 0)
                 {
                     var inner = s;
                     if (missing < 0)
                     {
                         var split = s.Length + missing;
                         inner = new StreamSegment(s, 0, split);
                         s     = new StreamSegment(s, split);
                     }
                     else
                     {
                         s = EmptyData.Instance;
                     }
                     var ret = parser.Close(inner);
                     parser = new WebSocketMessage.Parser();
                     onMessageParsed(ret);
                 }
                 else
                 {
                     readingPayload = true;
                     dataBuilder.Append(s);
                     return;
                 }
             }
             if (readingPayload)
             {
                 if (s.Length < missing)
                 {
                     dataBuilder.Append(s);
                     missing -= s.Length;
                     return;
                 }
                 else
                 {
                     dataBuilder.Append(new StreamSegment(s, 0, missing));
                     s              = new StreamSegment(s, missing);
                     missing        = 0;
                     readingPayload = false;
                     var bin = dataBuilder.Close();
                     dataBuilder = new ThresholdedDataBuilder(logger);
                     var msg = parser.Close(bin);
                     parser = new WebSocketMessage.Parser();
                     onMessageParsed(msg);
                 }
             }
         }
     }
 }
Beispiel #2
0
 public HttpData Receive(IDataStream s)
 {
     lock (contentLock)
     {
         if (useContentData)
         {
             contentData.Add(s);
         }
         dataBuilder.Append(s);
         if (state == ParsingState.Header)
         {
             try
             {
                 if (parser.Parse(contentData, isResponse))
                 {
                     state = ParsingState.Body;
                     try
                     {
                         var code = parser.StatusCode;
                         // header parsing is finished
                         if (LastRequestMethod == HttpRequestMethod.HEAD)
                         {
                             info = new HttpBodyLengthInfo((long)0);
                         }
                         else if (LastRequestMethod == HttpRequestMethod.CONNECT && code > 199 && code < 300)
                         {
                             info = new HttpBodyLengthInfo((long)0);
                         }
                         else
                         {
                             info = parser.GetBodyLengthInfo(!isResponse);
                         }
                         useContentData = info.Type == HttpBodyLengthInfo.LengthType.Chunked;
                         headerLength   = parser.HeaderLength;
                         if (contentData.Length - parser.HeaderLength > 0)
                         {
                             var tmpList      = new StreamList();
                             var contentStart = new ByteArray(contentData, parser.HeaderLength);
                             tmpList.Add(contentStart);
                             contentData = tmpList;
                         }
                         else
                         {
                             contentData = new StreamList();
                         }
                     }
                     catch (BadRequestException)
                     {
                         throw new HttpInvalidMessageException();
                     }
                 }
             }
             catch (HttpHeaderParserException)
             {
                 throw new HttpInvalidMessageException();
             }
         }
         if (state == ParsingState.Body)
         {
             if (info.Type == HttpBodyLengthInfo.LengthType.Exact && info.Length <= dataBuilder.Length - headerLength)
             {
                 var resStream = dataBuilder.Close();
                 return(parser.Create(new StreamSegment(resStream, 0, headerLength), new StreamSegment(resStream, headerLength, info.Length), resStream));
             }
             else if (info.Type == HttpBodyLengthInfo.LengthType.Chunked)
             {
                 parseChunkedBody();
             }
         }
     }
     if (state == ParsingState.CheckingChunedTrailer)
     {
         if (contentData.Length >= 2)
         {
             // check CRLF
             bool chunkedEnd = contentData.ReadByte(0) == 13 && contentData.ReadByte(1) == 10;
             if (chunkedEnd)
             {
                 contentData = new StreamList();
                 var resStream = dataBuilder.Close();
                 return(parser.Create(new StreamSegment(resStream, 0, headerLength), new StreamSegment(resStream, headerLength), resStream));
             }
             else
             {
                 parser.ParseChunkedTrailer();
                 state = ParsingState.ParsingChunkedTrailer;
             }
         }
     }
     if (state == ParsingState.ParsingChunkedTrailer)
     {
         if (parser.Parse(contentData, isResponse))
         {
             contentData = new StreamList();
             var resStream = dataBuilder.Close();
             return(parser.Create(new StreamSegment(resStream, 0, headerLength), new StreamSegment(resStream, headerLength), resStream));
         }
     }
     return(null);
 }