util.go 744 B

123456789101112131415161718192021222324252627282930313233
  1. package srtp
  2. import "bytes"
  3. // Grow the buffer size to the given number of bytes.
  4. func growBufferSize(buf []byte, size int) []byte {
  5. if size <= cap(buf) {
  6. return buf[:size]
  7. }
  8. buf2 := make([]byte, size)
  9. copy(buf2, buf)
  10. return buf2
  11. }
  12. // Check if buffers match, if not allocate a new buffer and return it
  13. func allocateIfMismatch(dst, src []byte) []byte {
  14. if dst == nil {
  15. dst = make([]byte, len(src))
  16. copy(dst, src)
  17. } else if !bytes.Equal(dst, src) { // bytes.Equal returns on ref equality, no optimization needed
  18. extraNeeded := len(src) - len(dst)
  19. if extraNeeded > 0 {
  20. dst = append(dst, make([]byte, extraNeeded)...)
  21. } else if extraNeeded < 0 {
  22. dst = dst[:len(dst)+extraNeeded]
  23. }
  24. copy(dst, src)
  25. }
  26. return dst
  27. }