mirror of
				https://github.com/ciphervance/supercell-wx.git
				synced 2025-10-31 19:20:05 +00:00 
			
		
		
		
	Handle raw message content when istream is EOF
This commit is contained in:
		
							parent
							
								
									ad7f3674d6
								
							
						
					
					
						commit
						465507d6b5
					
				
					 1 changed files with 25 additions and 8 deletions
				
			
		|  | @ -164,6 +164,9 @@ bool TextProductMessage::Parse(std::istream& is) | ||||||
|    if (dataValid) |    if (dataValid) | ||||||
|    { |    { | ||||||
|       // Store raw message content
 |       // Store raw message content
 | ||||||
|  |       if (is.good()) | ||||||
|  |       { | ||||||
|  |          // Read content equal to the message size
 | ||||||
|          std::streampos  messageEnd  = is.tellg(); |          std::streampos  messageEnd  = is.tellg(); | ||||||
|          std::streamsize messageSize = messageEnd - messageStart; |          std::streamsize messageSize = messageEnd - messageStart; | ||||||
|          p->messageContent_.resize(messageEnd - messageStart); |          p->messageContent_.resize(messageEnd - messageStart); | ||||||
|  | @ -174,6 +177,20 @@ bool TextProductMessage::Parse(std::istream& is) | ||||||
|             messageSize--; |             messageSize--; | ||||||
|          } |          } | ||||||
|          is.read(p->messageContent_.data(), messageSize); |          is.read(p->messageContent_.data(), messageSize); | ||||||
|  |       } | ||||||
|  |       else | ||||||
|  |       { | ||||||
|  |          // Read remaining content in the input stream
 | ||||||
|  |          is.clear(); | ||||||
|  |          is.seekg(messageStart); | ||||||
|  |          if (is.peek() == common::Characters::SOH) | ||||||
|  |          { | ||||||
|  |             is.seekg(std::streamoff {1}, std::ios_base::cur); | ||||||
|  |          } | ||||||
|  | 
 | ||||||
|  |          constexpr std::istreambuf_iterator<char> eos; | ||||||
|  |          p->messageContent_ = {std::istreambuf_iterator<char>(is), eos}; | ||||||
|  |       } | ||||||
| 
 | 
 | ||||||
|       // Trim extra characters from raw message
 |       // Trim extra characters from raw message
 | ||||||
|       while (p->messageContent_.size() > 0 && |       while (p->messageContent_.size() > 0 && | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue
	
	 Dan Paulat
						Dan Paulat