deserialize.move 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141
  1. module pyth::deserialize {
  2. use wormhole::bytes::{Self};
  3. use wormhole::cursor::{take_rest, Cursor};
  4. use pyth::i64::{Self, I64};
  5. #[test_only]
  6. use wormhole::cursor::{Self};
  7. public fun deserialize_vector(cur: &mut Cursor<u8>, n: u64): vector<u8> {
  8. bytes::take_bytes(cur, n)
  9. }
  10. public fun deserialize_u8(cur: &mut Cursor<u8>): u8 {
  11. bytes::take_u8(cur)
  12. }
  13. public fun deserialize_u16(cur: &mut Cursor<u8>): u16 {
  14. bytes::take_u16_be(cur)
  15. }
  16. public fun deserialize_u32(cur: &mut Cursor<u8>): u32 {
  17. bytes::take_u32_be(cur)
  18. }
  19. public fun deserialize_i32(cur: &mut Cursor<u8>): I64 {
  20. let deserialized = deserialize_u32(cur);
  21. // If negative, pad the value
  22. let negative = (deserialized >> 31) == 1;
  23. if (negative) {
  24. let padded = (0xFFFFFFFF << 32) + (deserialized as u64);
  25. i64::from_u64((padded as u64))
  26. } else {
  27. i64::from_u64((deserialized as u64))
  28. }
  29. }
  30. public fun deserialize_u64(cur: &mut Cursor<u8>): u64 {
  31. bytes::take_u64_be(cur)
  32. }
  33. public fun deserialize_i64(cur: &mut Cursor<u8>): I64 {
  34. i64::from_u64(deserialize_u64(cur))
  35. }
  36. #[test]
  37. fun test_deserialize_u8() {
  38. let input = x"48258963";
  39. let cursor = cursor::new(input);
  40. let result = deserialize_u8(&mut cursor);
  41. assert!(result == 0x48, 1);
  42. let rest = take_rest(cursor);
  43. assert!(rest == x"258963", 1);
  44. }
  45. #[test]
  46. fun test_deserialize_u16() {
  47. let input = x"48258963";
  48. let cursor = cursor::new(input);
  49. let result = deserialize_u16(&mut cursor);
  50. assert!(result == 0x4825, 1);
  51. let rest = take_rest(cursor);
  52. assert!(rest == x"8963", 1);
  53. }
  54. #[test]
  55. fun test_deserialize_u32() {
  56. let input = x"4825896349741695";
  57. let cursor = cursor::new(input);
  58. let result = deserialize_u32(&mut cursor);
  59. assert!(result == 0x48258963, 1);
  60. let rest = take_rest(cursor);
  61. assert!(rest == x"49741695", 1);
  62. }
  63. #[test]
  64. fun test_deserialize_i32_positive() {
  65. let input = x"4825896349741695";
  66. let cursor = cursor::new(input);
  67. let result = deserialize_i32(&mut cursor);
  68. assert!(result == i64::from_u64(0x48258963), 1);
  69. let rest = take_rest(cursor);
  70. assert!(rest == x"49741695", 1);
  71. }
  72. #[test]
  73. fun test_deserialize_i32_negative() {
  74. let input = x"FFFFFDC349741695";
  75. let cursor = cursor::new(input);
  76. let result = deserialize_i32(&mut cursor);
  77. assert!(result == i64::from_u64(0xFFFFFFFFFFFFFDC3), 1);
  78. let rest = take_rest(cursor);
  79. assert!(rest == x"49741695", 1);
  80. }
  81. #[test]
  82. fun test_deserialize_u64() {
  83. let input = x"48258963497416957497253486";
  84. let cursor = cursor::new(input);
  85. let result = deserialize_u64(&mut cursor);
  86. assert!(result == 0x4825896349741695, 1);
  87. let rest = take_rest(cursor);
  88. assert!(rest == x"7497253486", 1);
  89. }
  90. #[test]
  91. fun test_deserialize_i64_positive() {
  92. let input = x"48258963497416957497253486";
  93. let cursor = cursor::new(input);
  94. let result = deserialize_i64(&mut cursor);
  95. assert!(result == i64::from_u64(0x4825896349741695), 1);
  96. let rest = take_rest(cursor);
  97. assert!(rest == x"7497253486", 1);
  98. }
  99. #[test]
  100. fun test_deserialize_i64_negative() {
  101. let input = x"FFFFFFFFFFFFFDC37497253486";
  102. let cursor = cursor::new(input);
  103. let result = deserialize_i64(&mut cursor);
  104. assert!(result == i64::from_u64(0xFFFFFFFFFFFFFDC3), 1);
  105. let rest = take_rest(cursor);
  106. assert!(rest == x"7497253486", 1);
  107. }
  108. }