From cbc4204e500b35fd9cb0bcd6e08de548c4ab6f63 Mon Sep 17 00:00:00 2001
From: Leonid Bugaev <leonsbox@gmail.com>
Date: Mon, 30 Sep 2013 09:02:27 +0200
Subject: [PATCH] Some minor changes

---
 listener/raw_tcp_listener.go      | 9 ++++-----
 listener/raw_tcp_listener_test.go | 2 +-
 listener/tcp_packet.go            | 2 +-
 3 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/listener/raw_tcp_listener.go b/listener/raw_tcp_listener.go
index 789f9bc..81c2717 100644
--- a/listener/raw_tcp_listener.go
+++ b/listener/raw_tcp_listener.go
@@ -28,9 +28,9 @@ type RAWTCPListener struct {
 func RAWTCPListen(addr string, port int) (listener *RAWTCPListener) {
 	listener = &RAWTCPListener{}
 
-	listener.c_packets = make(chan *TCPPacket, 100)
-	listener.c_messages = make(chan *TCPMessage, 50)
-	listener.c_del_message = make(chan *TCPMessage, 10)
+	listener.c_packets = make(chan *TCPPacket, 1000)
+	listener.c_messages = make(chan *TCPMessage, 1000)
+	listener.c_del_message = make(chan *TCPMessage, 1000)
 
 	listener.addr = addr
 	listener.port = port
@@ -112,8 +112,7 @@ func (t *RAWTCPListener) parsePacket(buf []byte) {
 		new_buf := make([]byte, len(buf))
 		copy(new_buf, buf)
 
-		packet := NewTCPPacket(new_buf)
-		t.c_packets <- packet
+		t.c_packets <- ParseTCPPacket(new_buf)
 	}
 }
 
diff --git a/listener/raw_tcp_listener_test.go b/listener/raw_tcp_listener_test.go
index 8c32908..ca82bbd 100644
--- a/listener/raw_tcp_listener_test.go
+++ b/listener/raw_tcp_listener_test.go
@@ -70,7 +70,7 @@ func TestRawTCPListener(t *testing.T) {
 		}
 	}()
 
-	for i := 0; i < 100000; i++ {
+	for i := 0; i < 10000; i++ {
 		wg.Add(1)
 
 		packets := getPackets(port)
diff --git a/listener/tcp_packet.go b/listener/tcp_packet.go
index e466d6e..73d0007 100644
--- a/listener/tcp_packet.go
+++ b/listener/tcp_packet.go
@@ -36,7 +36,7 @@ type TCPPacket struct {
 	Data []byte
 }
 
-func NewTCPPacket(b []byte) (p *TCPPacket) {
+func ParseTCPPacket(b []byte) (p *TCPPacket) {
 	p = &TCPPacket{Data: b}
 	p.ParseFast()