Cleanup code in cmux and bench_test.

This commit is a partial cherry pick of pull request #3, for code cleanup.

bench_test.go

@@ -17,8 +17,6 @@ func (c *mockConn) Read(b []byte) (n int, err error) {
 }
 
 func BenchmarkCMuxConn(b *testing.B) {
-	b.StopTimer()
-
 	benchHTTPPayload := make([]byte, 4096)
 	copy(benchHTTPPayload, []byte("GET http://www.w3.org/ HTTP/1.1"))
 
@@ -33,7 +31,7 @@ func BenchmarkCMuxConn(b *testing.B) {
 		}
 	}()
 
-	b.StartTimer()
+	b.ResetTimer()
 
 	for i := 0; i < b.N; i++ {
 		c := &mockConn{

cmux.go

@@ -109,14 +109,12 @@ func (m *cMux) Serve() error {
 
 func (m *cMux) serve(c net.Conn) {
 	muc := newMuxConn(c)
-	matched := false
 	for _, sl := range m.sls {
 		for _, s := range sl.ss {
-			matched = s(muc.sniffer())
+			matched := s(muc.sniffer())
 			muc.reset()
 			if matched {
 				select {
-				// TODO(soheil): threre is a possiblity of having unclosed connection.
 				case sl.l.connc <- muc:
 				case <-sl.l.donec:
 					c.Close()
@@ -126,12 +124,10 @@ func (m *cMux) serve(c net.Conn) {
 		}
 	}
 
-	if !matched {
-		c.Close()
-		err := ErrNotMatched{c: c}
-		if !m.handleErr(err) {
-			m.root.Close()
-		}
+	c.Close()
+	err := ErrNotMatched{c: c}
+	if !m.handleErr(err) {
+		m.root.Close()
 	}
 }