From 9b4b49a6a85f9044ebaffff71b2498a3cf6c8ae4 Mon Sep 17 00:00:00 2001
From: "frsyuki (none)" <frsyuki@vcore.>
Date: Thu, 26 Feb 2009 01:33:34 +0900
Subject: [PATCH] example: limit message size

---
 example/stream.cc | 4 ++++
 example/stream.rb | 5 +++++
 2 files changed, 9 insertions(+)

diff --git a/example/stream.cc b/example/stream.cc
index 8fd92f9d..1e2e733b 100644
--- a/example/stream.cc
+++ b/example/stream.cc
@@ -43,6 +43,10 @@ public:
 
 			process_message(msg, life);
 		}
+
+		if(m_pac.message_size() > 10*1024*1024) {
+			throw std::runtime_error("message is too large");
+		}
 	}
 
 private:
diff --git a/example/stream.rb b/example/stream.rb
index e53ce82d..a72f5b98 100644
--- a/example/stream.rb
+++ b/example/stream.rb
@@ -34,12 +34,17 @@ class Server
 				@pk.reset
 				@buffer.slice!(0, @nread)
 				@nread = 0
+
 				next unless @buffer.empty?
 			end
 
 			break
 		end
 
+		if @buffer.length > 10*1024*1024
+			raise "message is too large"
+		end
+
 	rescue
 		puts "error while processing client packet: #{$!}"
 	end