增加随机端口范围

main.go

@@ -2,6 +2,7 @@ package gb28181
 
 import (
 	"log"
+	"math/rand"
 	"net"
 	"net/http"
 	"strconv"
@@ -18,12 +19,14 @@ import (
 
 var Devices sync.Map
 var config = struct {
-	Serial     string
-	Realm      string
-	ListenAddr string
-	Expires    int
-	AutoInvite bool
-}{"34020000002000000001", "3402000000", "127.0.0.1:5060", 3600, true}
+	Serial       string
+	Realm        string
+	ListenAddr   string
+	Expires      int
+	AutoInvite   bool
+	MediaPortMin uint16
+	MediaPortMax uint16
+}{"34020000002000000001", "3402000000", "127.0.0.1:5060", 3600, true, 58200, 58300}
 
 func init() {
 	InstallPlugin(&PluginConfig{
@@ -55,8 +58,8 @@ func run() {
 
 		AudioEnable:      true,
 		WaitKeyFrame:     true,
-		MediaPortMin:     58200,
-		MediaPortMax:     58300,
+		MediaPortMin:     config.MediaPortMin,
+		MediaPortMax:     config.MediaPortMax,
 		MediaIdleTimeout: 30,
 	}
 	s := transaction.NewCore(config)
@@ -119,7 +122,6 @@ func run() {
 	})
 	s.Start()
 }
-
 func onPublish(channel *transaction.Channel) (port int) {
 	rtpPublisher := new(rtp.RTP_PS)
 	if !rtpPublisher.Publish("gb28181/" + channel.DeviceID) {
@@ -131,19 +133,28 @@ func onPublish(channel *transaction.Channel) (port int) {
 		}
 	}()
 	rtpPublisher.Type = "GB28181"
-	addr, err := net.ResolveUDPAddr("udp", ":0")
-	if err != nil {
-		return
+	var conn *net.UDPConn
+	var err error
+	rang := int(config.MediaPortMax - config.MediaPortMin)
+	for count := rang; count > 0; count-- {
+		randNum := rand.Intn(rang)
+		port = int(config.MediaPortMin) + randNum
+		addr, _ := net.ResolveUDPAddr("udp", ":"+strconv.Itoa(port))
+		conn, err = net.ListenUDP("udp", addr)
+		if err != nil {
+			continue
+		} else {
+			break
+		}
 	}
-	conn, err := net.ListenUDP("udp", addr)
 	if err != nil {
 		return
 	}
 	networkBuffer := 1048576
-	if err := conn.SetReadBuffer(networkBuffer); err != nil {
+	if err = conn.SetReadBuffer(networkBuffer); err != nil {
 		Printf("udp server video conn set read buffer error, %v", err)
 	}
-	if err := conn.SetWriteBuffer(networkBuffer); err != nil {
+	if err = conn.SetWriteBuffer(networkBuffer); err != nil {
 		Printf("udp server video conn set write buffer error, %v", err)
 	}
 	la := conn.LocalAddr().String()

transaction/core.go

@@ -460,7 +460,7 @@ func (c *Core) AddDevice(msg *sip.Message) *Device {
 		core:         c,
 		from:         &sip.Contact{Uri: msg.StartLine.Uri, Params: make(map[string]string)},
 		to:           msg.To,
-		addr:         msg.Via.GetSendBy(),
+		Addr:         msg.Via.GetSendBy(),
 	}
 	c.Devices.Store(msg.From.Uri.UserInfo(), v)
 	return v

transaction/device.go

@@ -36,7 +36,7 @@ type Device struct {
 	sn           int
 	from         *sip.Contact
 	to           *sip.Contact
-	addr         string
+	Addr         string
 }
 
 func (c *Core) RemoveDead() {
@@ -100,7 +100,7 @@ func (d *Device) CreateMessage(Method sip.Method) (requestMsg *sip.Message) {
 			ID:     1,
 			Method: Method,
 		}, CallID: utils.RandNumString(10),
-		Addr: d.addr,
+		Addr: d.Addr,
 	}
 	requestMsg.From.Params["tag"] = utils.RandNumString(9)
 	return