fixed stack over flow

smasherprog · smasherprog · commit aa6e8b40b07c · 2017-07-30T20:30:18.000-07:00
diff --git a/include/internal/WebSocketProtocol.h b/include/internal/WebSocketProtocol.h
@@ -623,11 +623,15 @@ namespace SL {
             ReadHeaderNext(parent, socket);
         }
         template <class PARENTTYPE, class SOCKETTYPE>inline void ReadHeaderStart(const PARENTTYPE& parent, const SOCKETTYPE& socket, const std::shared_ptr<asio::streambuf>& extradata) {
+            free(socket->ReceiveBuffer);
+            socket->ReceiveBuffer = nullptr;
+            socket->ReceiveBufferSize = 0;
+            socket->LastOpCode = OpCode::INVALID;
             if (extradata->size() == 0) {
-                ReadHeaderStart(parent, socket);
+                ReadHeaderNext(parent, socket);
             }
             else {
-                ReadHeaderStart(parent, socket, extradata);
+                ReadHeaderNext(parent, socket, extradata);
             }
         }
 

Original file line number	Diff line number	Diff line change
`@@ -623,11 +623,15 @@ namespace SL {`
`623`	`623`	`ReadHeaderNext(parent, socket);`
`624`	`624`	`}`
`625`	`625`	`template <class PARENTTYPE, class SOCKETTYPE>inline void ReadHeaderStart(const PARENTTYPE& parent, const SOCKETTYPE& socket, const std::shared_ptr<asio::streambuf>& extradata) {`
	`626`	`+ free(socket->ReceiveBuffer);`
	`627`	`+ socket->ReceiveBuffer = nullptr;`
	`628`	`+ socket->ReceiveBufferSize = 0;`
	`629`	`+ socket->LastOpCode = OpCode::INVALID;`
`626`	`630`	`if (extradata->size() == 0) {`
`627`		`- ReadHeaderStart(parent, socket);`
	`631`	`+ ReadHeaderNext(parent, socket);`
`628`	`632`	`}`
`629`	`633`	`else {`
`630`		`- ReadHeaderStart(parent, socket, extradata);`
	`634`	`+ ReadHeaderNext(parent, socket, extradata);`
`631`	`635`	`}`
`632`	`636`	`}`
`633`	`637`