decode.h 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196
  1. #ifndef __CEPH_DECODE_H
  2. #define __CEPH_DECODE_H
  3. #include <asm/unaligned.h>
  4. #include <linux/time.h>
  5. #include "types.h"
  6. /*
  7. * in all cases,
  8. * void **p pointer to position pointer
  9. * void *end pointer to end of buffer (last byte + 1)
  10. */
  11. static inline u64 ceph_decode_64(void **p)
  12. {
  13. u64 v = get_unaligned_le64(*p);
  14. *p += sizeof(u64);
  15. return v;
  16. }
  17. static inline u32 ceph_decode_32(void **p)
  18. {
  19. u32 v = get_unaligned_le32(*p);
  20. *p += sizeof(u32);
  21. return v;
  22. }
  23. static inline u16 ceph_decode_16(void **p)
  24. {
  25. u16 v = get_unaligned_le16(*p);
  26. *p += sizeof(u16);
  27. return v;
  28. }
  29. static inline u8 ceph_decode_8(void **p)
  30. {
  31. u8 v = *(u8 *)*p;
  32. (*p)++;
  33. return v;
  34. }
  35. static inline void ceph_decode_copy(void **p, void *pv, size_t n)
  36. {
  37. memcpy(pv, *p, n);
  38. *p += n;
  39. }
  40. /*
  41. * bounds check input.
  42. */
  43. #define ceph_decode_need(p, end, n, bad) \
  44. do { \
  45. if (unlikely(*(p) + (n) > (end))) \
  46. goto bad; \
  47. } while (0)
  48. #define ceph_decode_64_safe(p, end, v, bad) \
  49. do { \
  50. ceph_decode_need(p, end, sizeof(u64), bad); \
  51. v = ceph_decode_64(p); \
  52. } while (0)
  53. #define ceph_decode_32_safe(p, end, v, bad) \
  54. do { \
  55. ceph_decode_need(p, end, sizeof(u32), bad); \
  56. v = ceph_decode_32(p); \
  57. } while (0)
  58. #define ceph_decode_16_safe(p, end, v, bad) \
  59. do { \
  60. ceph_decode_need(p, end, sizeof(u16), bad); \
  61. v = ceph_decode_16(p); \
  62. } while (0)
  63. #define ceph_decode_8_safe(p, end, v, bad) \
  64. do { \
  65. ceph_decode_need(p, end, sizeof(u8), bad); \
  66. v = ceph_decode_8(p); \
  67. } while (0)
  68. #define ceph_decode_copy_safe(p, end, pv, n, bad) \
  69. do { \
  70. ceph_decode_need(p, end, n, bad); \
  71. ceph_decode_copy(p, pv, n); \
  72. } while (0)
  73. /*
  74. * struct ceph_timespec <-> struct timespec
  75. */
  76. static inline void ceph_decode_timespec(struct timespec *ts,
  77. const struct ceph_timespec *tv)
  78. {
  79. ts->tv_sec = le32_to_cpu(tv->tv_sec);
  80. ts->tv_nsec = le32_to_cpu(tv->tv_nsec);
  81. }
  82. static inline void ceph_encode_timespec(struct ceph_timespec *tv,
  83. const struct timespec *ts)
  84. {
  85. tv->tv_sec = cpu_to_le32(ts->tv_sec);
  86. tv->tv_nsec = cpu_to_le32(ts->tv_nsec);
  87. }
  88. /*
  89. * sockaddr_storage <-> ceph_sockaddr
  90. */
  91. static inline void ceph_encode_addr(struct ceph_entity_addr *a)
  92. {
  93. __be16 ss_family = htons(a->in_addr.ss_family);
  94. a->in_addr.ss_family = *(__u16 *)&ss_family;
  95. }
  96. static inline void ceph_decode_addr(struct ceph_entity_addr *a)
  97. {
  98. __be16 ss_family = *(__be16 *)&a->in_addr.ss_family;
  99. a->in_addr.ss_family = ntohs(ss_family);
  100. WARN_ON(a->in_addr.ss_family == 512);
  101. }
  102. /*
  103. * encoders
  104. */
  105. static inline void ceph_encode_64(void **p, u64 v)
  106. {
  107. put_unaligned_le64(v, (__le64 *)*p);
  108. *p += sizeof(u64);
  109. }
  110. static inline void ceph_encode_32(void **p, u32 v)
  111. {
  112. put_unaligned_le32(v, (__le32 *)*p);
  113. *p += sizeof(u32);
  114. }
  115. static inline void ceph_encode_16(void **p, u16 v)
  116. {
  117. put_unaligned_le16(v, (__le16 *)*p);
  118. *p += sizeof(u16);
  119. }
  120. static inline void ceph_encode_8(void **p, u8 v)
  121. {
  122. *(u8 *)*p = v;
  123. (*p)++;
  124. }
  125. static inline void ceph_encode_copy(void **p, const void *s, int len)
  126. {
  127. memcpy(*p, s, len);
  128. *p += len;
  129. }
  130. /*
  131. * filepath, string encoders
  132. */
  133. static inline void ceph_encode_filepath(void **p, void *end,
  134. u64 ino, const char *path)
  135. {
  136. u32 len = path ? strlen(path) : 0;
  137. BUG_ON(*p + sizeof(ino) + sizeof(len) + len > end);
  138. ceph_encode_8(p, 1);
  139. ceph_encode_64(p, ino);
  140. ceph_encode_32(p, len);
  141. if (len)
  142. memcpy(*p, path, len);
  143. *p += len;
  144. }
  145. static inline void ceph_encode_string(void **p, void *end,
  146. const char *s, u32 len)
  147. {
  148. BUG_ON(*p + sizeof(len) + len > end);
  149. ceph_encode_32(p, len);
  150. if (len)
  151. memcpy(*p, s, len);
  152. *p += len;
  153. }
  154. #define ceph_encode_need(p, end, n, bad) \
  155. do { \
  156. if (unlikely(*(p) + (n) > (end))) \
  157. goto bad; \
  158. } while (0)
  159. #define ceph_encode_64_safe(p, end, v, bad) \
  160. do { \
  161. ceph_encode_need(p, end, sizeof(u64), bad); \
  162. ceph_encode_64(p, v); \
  163. } while (0)
  164. #define ceph_encode_32_safe(p, end, v, bad) \
  165. do { \
  166. ceph_encode_need(p, end, sizeof(u32), bad); \
  167. ceph_encode_32(p, v); \
  168. } while (0)
  169. #define ceph_encode_16_safe(p, end, v, bad) \
  170. do { \
  171. ceph_encode_need(p, end, sizeof(u16), bad); \
  172. ceph_encode_16(p, v); \
  173. } while (0)
  174. #define ceph_encode_copy_safe(p, end, pv, n, bad) \
  175. do { \
  176. ceph_encode_need(p, end, n, bad); \
  177. ceph_encode_copy(p, pv, n); \
  178. } while (0)
  179. #endif