Generics.swift 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. //
  2. // CryptoSwift
  3. //
  4. // Copyright (C) 2014-2017 Marcin Krzyżanowski <marcin@krzyzanowskim.com>
  5. // This software is provided 'as-is', without any express or implied warranty.
  6. //
  7. // In no event will the authors be held liable for any damages arising from the use of this software.
  8. //
  9. // Permission is granted to anyone to use this software for any purpose,including commercial applications, and to alter it and redistribute it freely, subject to the following restrictions:
  10. //
  11. // - The origin of this software must not be misrepresented; you must not claim that you wrote the original software. If you use this software in a product, an acknowledgment in the product documentation is required.
  12. // - Altered source versions must be plainly marked as such, and must not be misrepresented as being the original software.
  13. // - This notice may not be removed or altered from any source or binary distribution.
  14. //
  15. /// Array of bytes. Caution: don't use directly because generic is slow.
  16. ///
  17. /// - parameter value: integer value
  18. /// - parameter length: length of output array. By default size of value type
  19. ///
  20. /// - returns: Array of bytes
  21. @_specialize(where T == Int)
  22. @_specialize(where T == UInt)
  23. @_specialize(where T == UInt8)
  24. @_specialize(where T == UInt16)
  25. @_specialize(where T == UInt32)
  26. @_specialize(where T == UInt64)
  27. func arrayOfBytes<T: FixedWidthInteger>(value: T, length totalBytes: Int = MemoryLayout<T>.size) -> Array<UInt8> {
  28. let valuePointer = UnsafeMutablePointer<T>.allocate(capacity: 1)
  29. valuePointer.pointee = value
  30. let bytesPointer = UnsafeMutablePointer<UInt8>(OpaquePointer(valuePointer))
  31. var bytes = Array<UInt8>(repeating: 0, count: totalBytes)
  32. for j in 0..<min(MemoryLayout<T>.size, totalBytes) {
  33. bytes[totalBytes - 1 - j] = (bytesPointer + j).pointee
  34. }
  35. valuePointer.deinitialize(count: 1)
  36. valuePointer.deallocate()
  37. return bytes
  38. }