voice.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881
  1. // Discordgo - Discord bindings for Go
  2. // Available at https://github.com/bwmarrin/discordgo
  3. // Copyright 2015-2016 Bruce Marriner <bruce@sqls.net>. All rights reserved.
  4. // Use of this source code is governed by a BSD-style
  5. // license that can be found in the LICENSE file.
  6. // This file contains code related to Discord voice suppport
  7. package discordgo
  8. import (
  9. "encoding/binary"
  10. "encoding/json"
  11. "fmt"
  12. "log"
  13. "net"
  14. "runtime"
  15. "strings"
  16. "sync"
  17. "time"
  18. "github.com/gorilla/websocket"
  19. "golang.org/x/crypto/nacl/secretbox"
  20. )
  21. // ------------------------------------------------------------------------------------------------
  22. // Code related to both VoiceConnection Websocket and UDP connections.
  23. // ------------------------------------------------------------------------------------------------
  24. // A VoiceConnection struct holds all the data and functions related to a Discord Voice Connection.
  25. type VoiceConnection struct {
  26. sync.RWMutex
  27. Debug bool // If true, print extra logging -- DEPRECATED
  28. LogLevel int
  29. Ready bool // If true, voice is ready to send/receive audio
  30. UserID string
  31. GuildID string
  32. ChannelID string
  33. deaf bool
  34. mute bool
  35. speaking bool
  36. reconnecting bool // If true, voice connection is trying to reconnect
  37. OpusSend chan []byte // Chan for sending opus audio
  38. OpusRecv chan *Packet // Chan for receiving opus audio
  39. wsConn *websocket.Conn
  40. wsMutex sync.Mutex
  41. udpConn *net.UDPConn
  42. session *Session
  43. sessionID string
  44. token string
  45. endpoint string
  46. // Used to send a close signal to goroutines
  47. close chan struct{}
  48. // Used to allow blocking until connected
  49. connected chan bool
  50. // Used to pass the sessionid from onVoiceStateUpdate
  51. // sessionRecv chan string UNUSED ATM
  52. op4 voiceOP4
  53. op2 voiceOP2
  54. voiceSpeakingUpdateHandlers []VoiceSpeakingUpdateHandler
  55. }
  56. // VoiceSpeakingUpdateHandler type provides a function defination for the
  57. // VoiceSpeakingUpdate event
  58. type VoiceSpeakingUpdateHandler func(vc *VoiceConnection, vs *VoiceSpeakingUpdate)
  59. // Speaking sends a speaking notification to Discord over the voice websocket.
  60. // This must be sent as true prior to sending audio and should be set to false
  61. // once finished sending audio.
  62. // b : Send true if speaking, false if not.
  63. func (v *VoiceConnection) Speaking(b bool) (err error) {
  64. v.log(LogDebug, "called (%t)", b)
  65. type voiceSpeakingData struct {
  66. Speaking bool `json:"speaking"`
  67. Delay int `json:"delay"`
  68. }
  69. type voiceSpeakingOp struct {
  70. Op int `json:"op"` // Always 5
  71. Data voiceSpeakingData `json:"d"`
  72. }
  73. if v.wsConn == nil {
  74. return fmt.Errorf("No VoiceConnection websocket.")
  75. }
  76. data := voiceSpeakingOp{5, voiceSpeakingData{b, 0}}
  77. v.wsMutex.Lock()
  78. err = v.wsConn.WriteJSON(data)
  79. v.wsMutex.Unlock()
  80. if err != nil {
  81. v.speaking = false
  82. log.Println("Speaking() write json error:", err)
  83. return
  84. }
  85. v.speaking = b
  86. return
  87. }
  88. // ChangeChannel sends Discord a request to change channels within a Guild
  89. // !!! NOTE !!! This function may be removed in favour of just using ChannelVoiceJoin
  90. func (v *VoiceConnection) ChangeChannel(channelID string, mute, deaf bool) (err error) {
  91. v.log(LogInformational, "called")
  92. data := voiceChannelJoinOp{4, voiceChannelJoinData{&v.GuildID, &channelID, mute, deaf}}
  93. v.wsMutex.Lock()
  94. err = v.session.wsConn.WriteJSON(data)
  95. v.wsMutex.Unlock()
  96. if err != nil {
  97. return
  98. }
  99. v.ChannelID = channelID
  100. v.deaf = deaf
  101. v.mute = mute
  102. v.speaking = false
  103. return
  104. }
  105. // Disconnect disconnects from this voice channel and closes the websocket
  106. // and udp connections to Discord.
  107. // !!! NOTE !!! this function may be removed in favour of ChannelVoiceLeave
  108. func (v *VoiceConnection) Disconnect() (err error) {
  109. // Send a OP4 with a nil channel to disconnect
  110. if v.sessionID != "" {
  111. data := voiceChannelJoinOp{4, voiceChannelJoinData{&v.GuildID, nil, true, true}}
  112. v.wsMutex.Lock()
  113. err = v.session.wsConn.WriteJSON(data)
  114. v.wsMutex.Unlock()
  115. v.sessionID = ""
  116. }
  117. // Close websocket and udp connections
  118. v.Close()
  119. v.log(LogInformational, "Deleting VoiceConnection %s", v.GuildID)
  120. delete(v.session.VoiceConnections, v.GuildID)
  121. return
  122. }
  123. // Close closes the voice ws and udp connections
  124. func (v *VoiceConnection) Close() {
  125. v.log(LogInformational, "called")
  126. v.Lock()
  127. defer v.Unlock()
  128. v.Ready = false
  129. v.speaking = false
  130. v.log(LogInformational, "past lock")
  131. if v.close != nil {
  132. v.log(LogInformational, "closing v.close")
  133. close(v.close)
  134. v.close = nil
  135. }
  136. v.log(LogInformational, "past closing v.close")
  137. if v.udpConn != nil {
  138. v.log(LogInformational, "closing udp")
  139. err := v.udpConn.Close()
  140. if err != nil {
  141. log.Println("error closing udp connection: ", err)
  142. }
  143. v.udpConn = nil
  144. }
  145. v.log(LogInformational, "past closing udp")
  146. if v.wsConn != nil {
  147. v.log(LogInformational, "closing wsConn")
  148. err := v.wsConn.Close()
  149. if err != nil {
  150. log.Println("error closing websocket connection: ", err)
  151. }
  152. v.wsConn = nil
  153. }
  154. v.log(LogInformational, "all done, returning")
  155. }
  156. // AddHandler adds a Handler for VoiceSpeakingUpdate events.
  157. func (v *VoiceConnection) AddHandler(h VoiceSpeakingUpdateHandler) {
  158. v.Lock()
  159. defer v.Unlock()
  160. v.voiceSpeakingUpdateHandlers = append(v.voiceSpeakingUpdateHandlers, h)
  161. }
  162. // VoiceSpeakingUpdate is a struct for a VoiceSpeakingUpdate event.
  163. type VoiceSpeakingUpdate struct {
  164. UserID string `json:"user_id"`
  165. SSRC int `json:"ssrc"`
  166. Speaking bool `json:"speaking"`
  167. }
  168. // ------------------------------------------------------------------------------------------------
  169. // Unexported Internal Functions Below.
  170. // ------------------------------------------------------------------------------------------------
  171. // A voiceOP4 stores the data for the voice operation 4 websocket event
  172. // which provides us with the NaCl SecretBox encryption key
  173. type voiceOP4 struct {
  174. SecretKey [32]byte `json:"secret_key"`
  175. Mode string `json:"mode"`
  176. }
  177. // A voiceOP2 stores the data for the voice operation 2 websocket event
  178. // which is sort of like the voice READY packet
  179. type voiceOP2 struct {
  180. SSRC uint32 `json:"ssrc"`
  181. Port int `json:"port"`
  182. Modes []string `json:"modes"`
  183. HeartbeatInterval time.Duration `json:"heartbeat_interval"`
  184. }
  185. // WaitUntilConnected waits for the Voice Connection to
  186. // become ready, if it does not become ready it retuns an err
  187. func (v *VoiceConnection) waitUntilConnected() error {
  188. v.log(LogInformational, "called")
  189. i := 0
  190. for {
  191. if v.Ready {
  192. return nil
  193. }
  194. if i > 10 {
  195. return fmt.Errorf("Timeout waiting for voice.")
  196. }
  197. time.Sleep(1 * time.Second)
  198. i++
  199. }
  200. }
  201. // Open opens a voice connection. This should be called
  202. // after VoiceChannelJoin is used and the data VOICE websocket events
  203. // are captured.
  204. func (v *VoiceConnection) open() (err error) {
  205. v.log(LogInformational, "called")
  206. v.Lock()
  207. defer v.Unlock()
  208. // Don't open a websocket if one is already open
  209. if v.wsConn != nil {
  210. v.log(LogWarning, "refusing to overwrite non-nil websocket")
  211. return
  212. }
  213. // TODO temp? loop to wait for the SessionID
  214. i := 0
  215. for {
  216. if v.sessionID != "" {
  217. break
  218. }
  219. if i > 20 { // only loop for up to 1 second total
  220. return fmt.Errorf("Did not receive voice Session ID in time.")
  221. }
  222. time.Sleep(50 * time.Millisecond)
  223. i++
  224. }
  225. // Connect to VoiceConnection Websocket
  226. vg := fmt.Sprintf("wss://%s", strings.TrimSuffix(v.endpoint, ":80"))
  227. v.log(LogInformational, "connecting to voice endpoint %s", vg)
  228. v.wsConn, _, err = websocket.DefaultDialer.Dial(vg, nil)
  229. if err != nil {
  230. v.log(LogWarning, "error connecting to voice endpoint %s, %s", vg, err)
  231. v.log(LogDebug, "voice struct: %#v\n", v)
  232. return
  233. }
  234. type voiceHandshakeData struct {
  235. ServerID string `json:"server_id"`
  236. UserID string `json:"user_id"`
  237. SessionID string `json:"session_id"`
  238. Token string `json:"token"`
  239. }
  240. type voiceHandshakeOp struct {
  241. Op int `json:"op"` // Always 0
  242. Data voiceHandshakeData `json:"d"`
  243. }
  244. data := voiceHandshakeOp{0, voiceHandshakeData{v.GuildID, v.UserID, v.sessionID, v.token}}
  245. err = v.wsConn.WriteJSON(data)
  246. if err != nil {
  247. v.log(LogWarning, "error sending init packet, %s", err)
  248. return
  249. }
  250. v.close = make(chan struct{})
  251. go v.wsListen(v.wsConn, v.close)
  252. // add loop/check for Ready bool here?
  253. // then return false if not ready?
  254. // but then wsListen will also err.
  255. return
  256. }
  257. // wsListen listens on the voice websocket for messages and passes them
  258. // to the voice event handler. This is automatically called by the Open func
  259. func (v *VoiceConnection) wsListen(wsConn *websocket.Conn, close <-chan struct{}) {
  260. v.log(LogInformational, "called")
  261. for {
  262. _, message, err := v.wsConn.ReadMessage()
  263. if err != nil {
  264. // Detect if we have been closed manually. If a Close() has already
  265. // happened, the websocket we are listening on will be different to the
  266. // current session.
  267. v.RLock()
  268. sameConnection := v.wsConn == wsConn
  269. v.RUnlock()
  270. if sameConnection {
  271. v.log(LogError, "voice endpoint %s websocket closed unexpectantly, %s", v.endpoint, err)
  272. // Start reconnect goroutine then exit.
  273. go v.reconnect()
  274. }
  275. return
  276. }
  277. // Pass received message to voice event handler
  278. select {
  279. case <-close:
  280. return
  281. default:
  282. go v.onEvent(message)
  283. }
  284. }
  285. }
  286. // wsEvent handles any voice websocket events. This is only called by the
  287. // wsListen() function.
  288. func (v *VoiceConnection) onEvent(message []byte) {
  289. v.log(LogDebug, "received: %s", string(message))
  290. var e Event
  291. if err := json.Unmarshal(message, &e); err != nil {
  292. v.log(LogError, "unmarshall error, %s", err)
  293. return
  294. }
  295. switch e.Operation {
  296. case 2: // READY
  297. if err := json.Unmarshal(e.RawData, &v.op2); err != nil {
  298. v.log(LogError, "OP2 unmarshall error, %s, %s", err, string(e.RawData))
  299. return
  300. }
  301. // Start the voice websocket heartbeat to keep the connection alive
  302. go v.wsHeartbeat(v.wsConn, v.close, v.op2.HeartbeatInterval)
  303. // TODO monitor a chan/bool to verify this was successful
  304. // Start the UDP connection
  305. err := v.udpOpen()
  306. if err != nil {
  307. v.log(LogError, "error opening udp connection, %s", err)
  308. return
  309. }
  310. // Start the opusSender.
  311. // TODO: Should we allow 48000/960 values to be user defined?
  312. if v.OpusSend == nil {
  313. v.OpusSend = make(chan []byte, 2)
  314. }
  315. go v.opusSender(v.udpConn, v.close, v.OpusSend, 48000, 960)
  316. // Start the opusReceiver
  317. if !v.deaf {
  318. if v.OpusRecv == nil {
  319. v.OpusRecv = make(chan *Packet, 2)
  320. }
  321. go v.opusReceiver(v.udpConn, v.close, v.OpusRecv)
  322. }
  323. // Send the ready event
  324. v.connected <- true
  325. return
  326. case 3: // HEARTBEAT response
  327. // add code to use this to track latency?
  328. return
  329. case 4: // udp encryption secret key
  330. v.op4 = voiceOP4{}
  331. if err := json.Unmarshal(e.RawData, &v.op4); err != nil {
  332. v.log(LogError, "OP4 unmarshall error, %s, %s", err, string(e.RawData))
  333. return
  334. }
  335. return
  336. case 5:
  337. if len(v.voiceSpeakingUpdateHandlers) == 0 {
  338. return
  339. }
  340. voiceSpeakingUpdate := &VoiceSpeakingUpdate{}
  341. if err := json.Unmarshal(e.RawData, voiceSpeakingUpdate); err != nil {
  342. v.log(LogError, "OP5 unmarshall error, %s, %s", err, string(e.RawData))
  343. return
  344. }
  345. for _, h := range v.voiceSpeakingUpdateHandlers {
  346. h(v, voiceSpeakingUpdate)
  347. }
  348. default:
  349. v.log(LogError, "unknown voice operation, %d, %s", e.Operation, string(e.RawData))
  350. }
  351. return
  352. }
  353. type voiceHeartbeatOp struct {
  354. Op int `json:"op"` // Always 3
  355. Data int `json:"d"`
  356. }
  357. // NOTE :: When a guild voice server changes how do we shut this down
  358. // properly, so a new connection can be setup without fuss?
  359. //
  360. // wsHeartbeat sends regular heartbeats to voice Discord so it knows the client
  361. // is still connected. If you do not send these heartbeats Discord will
  362. // disconnect the websocket connection after a few seconds.
  363. func (v *VoiceConnection) wsHeartbeat(wsConn *websocket.Conn, close <-chan struct{}, i time.Duration) {
  364. if close == nil || wsConn == nil {
  365. return
  366. }
  367. var err error
  368. ticker := time.NewTicker(i * time.Millisecond)
  369. for {
  370. v.log(LogDebug, "sending heartbeat packet")
  371. v.wsMutex.Lock()
  372. err = wsConn.WriteJSON(voiceHeartbeatOp{3, int(time.Now().Unix())})
  373. v.wsMutex.Unlock()
  374. if err != nil {
  375. v.log(LogError, "error sending heartbeat to voice endpoint %s, %s", v.endpoint, err)
  376. return
  377. }
  378. select {
  379. case <-ticker.C:
  380. // continue loop and send heartbeat
  381. case <-close:
  382. return
  383. }
  384. }
  385. }
  386. // ------------------------------------------------------------------------------------------------
  387. // Code related to the VoiceConnection UDP connection
  388. // ------------------------------------------------------------------------------------------------
  389. type voiceUDPData struct {
  390. Address string `json:"address"` // Public IP of machine running this code
  391. Port uint16 `json:"port"` // UDP Port of machine running this code
  392. Mode string `json:"mode"` // always "xsalsa20_poly1305"
  393. }
  394. type voiceUDPD struct {
  395. Protocol string `json:"protocol"` // Always "udp" ?
  396. Data voiceUDPData `json:"data"`
  397. }
  398. type voiceUDPOp struct {
  399. Op int `json:"op"` // Always 1
  400. Data voiceUDPD `json:"d"`
  401. }
  402. // udpOpen opens a UDP connection to the voice server and completes the
  403. // initial required handshake. This connection is left open in the session
  404. // and can be used to send or receive audio. This should only be called
  405. // from voice.wsEvent OP2
  406. func (v *VoiceConnection) udpOpen() (err error) {
  407. v.Lock()
  408. defer v.Unlock()
  409. if v.wsConn == nil {
  410. return fmt.Errorf("nil voice websocket")
  411. }
  412. if v.udpConn != nil {
  413. return fmt.Errorf("udp connection already open")
  414. }
  415. if v.close == nil {
  416. return fmt.Errorf("nil close channel")
  417. }
  418. if v.endpoint == "" {
  419. return fmt.Errorf("empty endpoint")
  420. }
  421. host := fmt.Sprintf("%s:%d", strings.TrimSuffix(v.endpoint, ":80"), v.op2.Port)
  422. addr, err := net.ResolveUDPAddr("udp", host)
  423. if err != nil {
  424. v.log(LogWarning, "error resolving udp host %s, %s", host, err)
  425. return
  426. }
  427. v.log(LogInformational, "connecting to udp addr %s", addr.String())
  428. v.udpConn, err = net.DialUDP("udp", nil, addr)
  429. if err != nil {
  430. v.log(LogWarning, "error connecting to udp addr %s, %s", addr.String(), err)
  431. return
  432. }
  433. // Create a 70 byte array and put the SSRC code from the Op 2 VoiceConnection event
  434. // into it. Then send that over the UDP connection to Discord
  435. sb := make([]byte, 70)
  436. binary.BigEndian.PutUint32(sb, v.op2.SSRC)
  437. _, err = v.udpConn.Write(sb)
  438. if err != nil {
  439. v.log(LogWarning, "udp write error to %s, %s", addr.String(), err)
  440. return
  441. }
  442. // Create a 70 byte array and listen for the initial handshake response
  443. // from Discord. Once we get it parse the IP and PORT information out
  444. // of the response. This should be our public IP and PORT as Discord
  445. // saw us.
  446. rb := make([]byte, 70)
  447. rlen, _, err := v.udpConn.ReadFromUDP(rb)
  448. if err != nil {
  449. v.log(LogWarning, "udp read error, %s, %s", addr.String(), err)
  450. return
  451. }
  452. if rlen < 70 {
  453. v.log(LogWarning, "received udp packet too small")
  454. return fmt.Errorf("received udp packet too small")
  455. }
  456. // Loop over position 4 though 20 to grab the IP address
  457. // Should never be beyond position 20.
  458. var ip string
  459. for i := 4; i < 20; i++ {
  460. if rb[i] == 0 {
  461. break
  462. }
  463. ip += string(rb[i])
  464. }
  465. // Grab port from position 68 and 69
  466. port := binary.LittleEndian.Uint16(rb[68:70])
  467. // Take the data from above and send it back to Discord to finalize
  468. // the UDP connection handshake.
  469. data := voiceUDPOp{1, voiceUDPD{"udp", voiceUDPData{ip, port, "xsalsa20_poly1305"}}}
  470. v.wsMutex.Lock()
  471. err = v.wsConn.WriteJSON(data)
  472. v.wsMutex.Unlock()
  473. if err != nil {
  474. v.log(LogWarning, "udp write error, %#v, %s", data, err)
  475. return
  476. }
  477. // start udpKeepAlive
  478. go v.udpKeepAlive(v.udpConn, v.close, 5*time.Second)
  479. // TODO: find a way to check that it fired off okay
  480. return
  481. }
  482. // udpKeepAlive sends a udp packet to keep the udp connection open
  483. // This is still a bit of a "proof of concept"
  484. func (v *VoiceConnection) udpKeepAlive(udpConn *net.UDPConn, close <-chan struct{}, i time.Duration) {
  485. if udpConn == nil || close == nil {
  486. return
  487. }
  488. var err error
  489. var sequence uint64
  490. packet := make([]byte, 8)
  491. ticker := time.NewTicker(i)
  492. for {
  493. binary.LittleEndian.PutUint64(packet, sequence)
  494. sequence++
  495. _, err = udpConn.Write(packet)
  496. if err != nil {
  497. v.log(LogError, "write error, %s")
  498. return
  499. }
  500. select {
  501. case <-ticker.C:
  502. // continue loop and send keepalive
  503. case <-close:
  504. return
  505. }
  506. }
  507. }
  508. // opusSender will listen on the given channel and send any
  509. // pre-encoded opus audio to Discord. Supposedly.
  510. func (v *VoiceConnection) opusSender(udpConn *net.UDPConn, close <-chan struct{}, opus <-chan []byte, rate, size int) {
  511. if udpConn == nil || close == nil {
  512. return
  513. }
  514. runtime.LockOSThread()
  515. // VoiceConnection is now ready to receive audio packets
  516. // TODO: this needs reviewed as I think there must be a better way.
  517. v.Ready = true
  518. defer func() { v.Ready = false }()
  519. var sequence uint16
  520. var timestamp uint32
  521. var recvbuf []byte
  522. var ok bool
  523. udpHeader := make([]byte, 12)
  524. var nonce [24]byte
  525. // build the parts that don't change in the udpHeader
  526. udpHeader[0] = 0x80
  527. udpHeader[1] = 0x78
  528. binary.BigEndian.PutUint32(udpHeader[8:], v.op2.SSRC)
  529. // start a send loop that loops until buf chan is closed
  530. ticker := time.NewTicker(time.Millisecond * time.Duration(size/(rate/1000)))
  531. for {
  532. // Get data from chan. If chan is closed, return.
  533. select {
  534. case <-close:
  535. return
  536. case recvbuf, ok = <-opus:
  537. if !ok {
  538. return
  539. }
  540. // else, continue loop
  541. }
  542. if !v.speaking {
  543. err := v.Speaking(true)
  544. if err != nil {
  545. v.log(LogError, "error sending speaking packet, %s", err)
  546. }
  547. }
  548. // Add sequence and timestamp to udpPacket
  549. binary.BigEndian.PutUint16(udpHeader[2:], sequence)
  550. binary.BigEndian.PutUint32(udpHeader[4:], timestamp)
  551. // encrypt the opus data
  552. copy(nonce[:], udpHeader)
  553. sendbuf := secretbox.Seal(udpHeader, recvbuf, &nonce, &v.op4.SecretKey)
  554. // block here until we're exactly at the right time :)
  555. // Then send rtp audio packet to Discord over UDP
  556. select {
  557. case <-close:
  558. return
  559. case <-ticker.C:
  560. // continue
  561. }
  562. _, err := udpConn.Write(sendbuf)
  563. if err != nil {
  564. v.log(LogError, "udp write error, %s", err)
  565. v.log(LogDebug, "voice struct: %#v\n", v)
  566. return
  567. }
  568. if (sequence) == 0xFFFF {
  569. sequence = 0
  570. } else {
  571. sequence++
  572. }
  573. if (timestamp + uint32(size)) >= 0xFFFFFFFF {
  574. timestamp = 0
  575. } else {
  576. timestamp += uint32(size)
  577. }
  578. }
  579. }
  580. // A Packet contains the headers and content of a received voice packet.
  581. type Packet struct {
  582. SSRC uint32
  583. Sequence uint16
  584. Timestamp uint32
  585. Type []byte
  586. Opus []byte
  587. PCM []int16
  588. }
  589. // opusReceiver listens on the UDP socket for incoming packets
  590. // and sends them across the given channel
  591. // NOTE :: This function may change names later.
  592. func (v *VoiceConnection) opusReceiver(udpConn *net.UDPConn, close <-chan struct{}, c chan *Packet) {
  593. if udpConn == nil || close == nil {
  594. return
  595. }
  596. p := Packet{}
  597. recvbuf := make([]byte, 1024)
  598. var nonce [24]byte
  599. for {
  600. rlen, err := udpConn.Read(recvbuf)
  601. if err != nil {
  602. // Detect if we have been closed manually. If a Close() has already
  603. // happened, the udp connection we are listening on will be different
  604. // to the current session.
  605. v.RLock()
  606. sameConnection := v.udpConn == udpConn
  607. v.RUnlock()
  608. if sameConnection {
  609. v.log(LogError, "udp read error, %s, %s", v.endpoint, err)
  610. v.log(LogDebug, "voice struct: %#v\n", v)
  611. go v.reconnect()
  612. }
  613. return
  614. }
  615. select {
  616. case <-close:
  617. return
  618. default:
  619. // continue loop
  620. }
  621. // For now, skip anything except audio.
  622. if rlen < 12 || recvbuf[0] != 0x80 {
  623. continue
  624. }
  625. // build a audio packet struct
  626. p.Type = recvbuf[0:2]
  627. p.Sequence = binary.BigEndian.Uint16(recvbuf[2:4])
  628. p.Timestamp = binary.BigEndian.Uint32(recvbuf[4:8])
  629. p.SSRC = binary.BigEndian.Uint32(recvbuf[8:12])
  630. // decrypt opus data
  631. copy(nonce[:], recvbuf[0:12])
  632. p.Opus, _ = secretbox.Open(nil, recvbuf[12:rlen], &nonce, &v.op4.SecretKey)
  633. if c != nil {
  634. c <- &p
  635. }
  636. }
  637. }
  638. // Reconnect will close down a voice connection then immediately try to
  639. // reconnect to that session.
  640. // NOTE : This func is messy and a WIP while I find what works.
  641. // It will be cleaned up once a proven stable option is flushed out.
  642. // aka: this is ugly shit code, please don't judge too harshly.
  643. func (v *VoiceConnection) reconnect() {
  644. v.log(LogInformational, "called")
  645. v.Lock()
  646. if v.reconnecting {
  647. v.log(LogInformational, "already reconnecting, exiting.")
  648. v.Unlock()
  649. return
  650. }
  651. v.reconnecting = true
  652. v.Unlock()
  653. defer func() { v.reconnecting = false }()
  654. if v.session == nil {
  655. v.log(LogInformational, "cannot reconnect with nil session")
  656. v.log(LogInformational, "Deleting VoiceConnection %s", v.GuildID)
  657. delete(v.session.VoiceConnections, v.GuildID)
  658. return
  659. }
  660. // Send a OP4 with a nil channel to disconnect
  661. if v.sessionID != "" {
  662. data := voiceChannelJoinOp{4, voiceChannelJoinData{&v.GuildID, nil, true, true}}
  663. v.wsMutex.Lock()
  664. err := v.session.wsConn.WriteJSON(data)
  665. if err != nil {
  666. v.log(LogError, "error sending disconnect packet, %s", err)
  667. }
  668. v.wsMutex.Unlock()
  669. v.sessionID = ""
  670. }
  671. // Close websocket and udp connections
  672. v.Close()
  673. wait := time.Duration(1)
  674. i := 0
  675. for {
  676. if v.session == nil {
  677. v.log(LogInformational, "cannot reconnect with nil session")
  678. v.log(LogInformational, "Deleting VoiceConnection %s", v.GuildID)
  679. delete(v.session.VoiceConnections, v.GuildID)
  680. return
  681. }
  682. <-time.After(wait * time.Second)
  683. wait *= 2
  684. if wait > 600 {
  685. wait = 600
  686. }
  687. i++
  688. if v.session.DataReady == false {
  689. v.log(LogInformational, "cannot reconenct with unready session")
  690. continue
  691. }
  692. v.log(LogInformational, "trying to reconnect to voice")
  693. // Below is required because ChannelVoiceJoin checks the GuildID
  694. // to decide if we should change channels or open a new connection.
  695. // TODO: Maybe find a better method.
  696. gID := v.GuildID
  697. v.GuildID = ""
  698. v.sessionID = ""
  699. _, err := v.session.ChannelVoiceJoin(gID, v.ChannelID, v.mute, v.deaf)
  700. if err == nil {
  701. v.log(LogInformational, "successfully reconnected to voice")
  702. return
  703. }
  704. v.log(LogInformational, "error reconnecting to voice, %s", err)
  705. if i >= 10 {
  706. // NOTE: this will probably change but it's a safety net
  707. // here to prevent this goroutine from becomming abandoned.
  708. v.log(LogInformational, "timeout reconnecting, I give up.")
  709. v.log(LogInformational, "Deleting VoiceConnection %s", v.GuildID)
  710. delete(v.session.VoiceConnections, v.GuildID)
  711. return
  712. }
  713. }
  714. }