int PacketPeerStream::get_available_packet_count() const { _poll_buffer(); uint32_t remaining = ring_buffer.data_left(); int ofs=0; int count=0; while(remaining>=4) { uint8_t lbuf[4]; ring_buffer.copy(lbuf,ofs,4); uint32_t len = decode_uint32(lbuf); remaining-=4; ofs+=4; if (len>remaining) break; remaining-=len; ofs+=len; count++; } return count; }
Error PacketPeerStream::get_packet(const uint8_t **r_buffer, int &r_buffer_size) const { ERR_FAIL_COND_V(peer.is_null(), ERR_UNCONFIGURED); _poll_buffer(); int remaining = ring_buffer.data_left(); ERR_FAIL_COND_V(remaining < 4, ERR_UNAVAILABLE); uint8_t lbuf[4]; ring_buffer.copy(lbuf, 0, 4); remaining -= 4; uint32_t len = decode_uint32(lbuf); ERR_FAIL_COND_V(remaining < (int)len, ERR_UNAVAILABLE); ring_buffer.read(lbuf, 4); //get rid of first 4 bytes ring_buffer.read(&temp_buffer[0], len); // read packet *r_buffer = &temp_buffer[0]; r_buffer_size = len; return OK; }
Error PacketPeerStream::put_packet(const uint8_t *p_buffer,int p_buffer_size) { ERR_FAIL_COND_V(peer.is_null(),ERR_UNCONFIGURED); Error err = _poll_buffer(); //won't hurt to poll here too if (err) return err; if (p_buffer_size==0) return OK; ERR_FAIL_COND_V( p_buffer_size<0, ERR_INVALID_PARAMETER); ERR_FAIL_COND_V( p_buffer_size+4 > temp_buffer.size(), ERR_INVALID_PARAMETER ); encode_uint32(p_buffer_size,&temp_buffer[0]); uint8_t *dst=&temp_buffer[4]; for(int i=0;i<p_buffer_size;i++) dst[i]=p_buffer[i]; return peer->put_data(&temp_buffer[0],p_buffer_size+4); }