add response for invalid music request, fixed txthinking#18

wqs8050080 · May 15, 2017 · 84ee333 · 84ee333
1 parent 34fb4ce
commit 84ee333
Show file tree

Hide file tree

Showing 7 changed files with 2,628 additions and 2 deletions.
diff --git a/bkclient.go b/bkclient.go
@@ -5,6 +5,7 @@ import (
 	"crypto/aes"
 	"crypto/rand"
 	"errors"
+	"fmt"
 	"io"
 	"log"
 	"net"
@@ -246,6 +247,7 @@ func (c *BKClient) handleHTTP(conn *net.TCPConn) error {
 		return errors.New("Invalid Request")
 	}
 	method, aoru := string(bb[0]), string(bb[1])
+	fmt.Print(string(b))
 	var addr string
 	if method == "CONNECT" {
 		addr = aoru

diff --git a/bkserver.go b/bkserver.go
@@ -1,6 +1,7 @@
 package brook
 
 import (
+	"bytes"
 	"crypto/aes"
 	"errors"
 	"io"
@@ -90,6 +91,13 @@ func (s *BKServer) handle(c *net.TCPConn) error {
 		if _, err := io.ReadFull(c, tmp); err != nil {
 			return err
 		}
+		if !bytes.Equal(tmp, s.Music.GetSong()) {
+			log.Println("Got a invalid song")
+			if _, err := c.Write(s.Music.GetResponse(tmp)); err != nil {
+				return err
+			}
+			return nil
+		}
 	}
 	cc, err := s.wrapCipherConn(c)
 	if err != nil {

diff --git a/music/chinamobile_sdc.go b/music/chinamobile_sdc.go