Add RTOMax settings

sctptransport.go

@@ -58,6 +58,8 @@
 
 	api *API
 	log logging.LeveledLogger
+	// maximum retransmission timeout
+	rtoMax float64
 }
 
 // NewSCTPTransport creates a new SCTPTransport.
@@ -105,12 +107,13 @@
 	if dtlsTransport == nil || dtlsTransport.conn == nil {
 		return errSCTPTransportDTLS
 	}
-
+	rtoMax := float64(r.api.settingEngine.sctp.rtoMax) / float64(time.Millisecond)
 	sctpAssociation, err := sctp.Client(sctp.Config{
 		NetConn:              dtlsTransport.conn,
 		MaxReceiveBufferSize: r.api.settingEngine.sctp.maxReceiveBufferSize,
 		LoggerFactory:        r.api.settingEngine.LoggerFactory,
+		RTOMax:               rtoMax,
 	})
 	if err != nil {
 		return err
 	}

settingengine.go

@@ -76,6 +76,7 @@ type SettingEngine struct {
 	}
 	sctp struct {
 		maxReceiveBufferSize uint32
+		rtoMax               time.Duration
 	}
 	sdpMediaLevelFingerprints                 bool
 	answeringDTLSRole                         DTLSRole
@@ -441,3 +442,9 @@ func (e *SettingEngine) SetSCTPMaxReceiveBufferSize(maxReceiveBufferSize uint32)
 func (e *SettingEngine) SetDTLSCustomerCipherSuites(customCipherSuites func() []dtls.CipherSuite) {
 	e.dtls.customCipherSuites = customCipherSuites
 }
+
+// SetSCTPRTOMax sets the maximum retransmission timeout.
+// Leave this 0 for the default timeout.
+func (e *SettingEngine) SetSCTPRTOMax(rtoMax time.Duration) {
+	e.sctp.rtoMax = rtoMax
+}

settingengine_test.go

@@ -269,3 +269,12 @@ func TestSetSCTPMaxReceiverBufferSize(t *testing.T) {
 	s.SetSCTPMaxReceiveBufferSize(expSize)
 	assert.Equal(t, expSize, s.sctp.maxReceiveBufferSize)
 }
+
+func TestSetSCTPRTOMax(t *testing.T) {
+	s := SettingEngine{}
+	assert.Equal(t, time.Duration(0), s.sctp.rtoMax)
+
+	expSize := time.Second
+	s.SetSCTPRTOMax(expSize)
+	assert.Equal(t, expSize, s.sctp.rtoMax)
+}