decode.h 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206
  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. static inline int ceph_has_room(void **p, void *end, size_t n)
  44. {
  45. return end >= *p && n <= end - *p;
  46. }
  47. #define ceph_decode_need(p, end, n, bad) \
  48. do { \
  49. if (!likely(ceph_has_room(p, end, n))) \
  50. goto bad; \
  51. } while (0)
  52. #define ceph_decode_64_safe(p, end, v, bad) \
  53. do { \
  54. ceph_decode_need(p, end, sizeof(u64), bad); \
  55. v = ceph_decode_64(p); \
  56. } while (0)
  57. #define ceph_decode_32_safe(p, end, v, bad) \
  58. do { \
  59. ceph_decode_need(p, end, sizeof(u32), bad); \
  60. v = ceph_decode_32(p); \
  61. } while (0)
  62. #define ceph_decode_16_safe(p, end, v, bad) \
  63. do { \
  64. ceph_decode_need(p, end, sizeof(u16), bad); \
  65. v = ceph_decode_16(p); \
  66. } while (0)
  67. #define ceph_decode_8_safe(p, end, v, bad) \
  68. do { \
  69. ceph_decode_need(p, end, sizeof(u8), bad); \
  70. v = ceph_decode_8(p); \
  71. } while (0)
  72. #define ceph_decode_copy_safe(p, end, pv, n, bad) \
  73. do { \
  74. ceph_decode_need(p, end, n, bad); \
  75. ceph_decode_copy(p, pv, n); \
  76. } while (0)
  77. /*
  78. * struct ceph_timespec <-> struct timespec
  79. */
  80. static inline void ceph_decode_timespec(struct timespec *ts,
  81. const struct ceph_timespec *tv)
  82. {
  83. ts->tv_sec = le32_to_cpu(tv->tv_sec);
  84. ts->tv_nsec = le32_to_cpu(tv->tv_nsec);
  85. }
  86. static inline void ceph_encode_timespec(struct ceph_timespec *tv,
  87. const struct timespec *ts)
  88. {
  89. tv->tv_sec = cpu_to_le32(ts->tv_sec);
  90. tv->tv_nsec = cpu_to_le32(ts->tv_nsec);
  91. }
  92. /*
  93. * sockaddr_storage <-> ceph_sockaddr
  94. */
  95. static inline void ceph_encode_addr(struct ceph_entity_addr *a)
  96. {
  97. __be16 ss_family = htons(a->in_addr.ss_family);
  98. a->in_addr.ss_family = *(__u16 *)&ss_family;
  99. }
  100. static inline void ceph_decode_addr(struct ceph_entity_addr *a)
  101. {
  102. __be16 ss_family = *(__be16 *)&a->in_addr.ss_family;
  103. a->in_addr.ss_family = ntohs(ss_family);
  104. WARN_ON(a->in_addr.ss_family == 512);
  105. }
  106. /*
  107. * encoders
  108. */
  109. static inline void ceph_encode_64(void **p, u64 v)
  110. {
  111. put_unaligned_le64(v, (__le64 *)*p);
  112. *p += sizeof(u64);
  113. }
  114. static inline void ceph_encode_32(void **p, u32 v)
  115. {
  116. put_unaligned_le32(v, (__le32 *)*p);
  117. *p += sizeof(u32);
  118. }
  119. static inline void ceph_encode_16(void **p, u16 v)
  120. {
  121. put_unaligned_le16(v, (__le16 *)*p);
  122. *p += sizeof(u16);
  123. }
  124. static inline void ceph_encode_8(void **p, u8 v)
  125. {
  126. *(u8 *)*p = v;
  127. (*p)++;
  128. }
  129. static inline void ceph_encode_copy(void **p, const void *s, int len)
  130. {
  131. memcpy(*p, s, len);
  132. *p += len;
  133. }
  134. /*
  135. * filepath, string encoders
  136. */
  137. static inline void ceph_encode_filepath(void **p, void *end,
  138. u64 ino, const char *path)
  139. {
  140. u32 len = path ? strlen(path) : 0;
  141. BUG_ON(*p + sizeof(ino) + sizeof(len) + len > end);
  142. ceph_encode_8(p, 1);
  143. ceph_encode_64(p, ino);
  144. ceph_encode_32(p, len);
  145. if (len)
  146. memcpy(*p, path, len);
  147. *p += len;
  148. }
  149. static inline void ceph_encode_string(void **p, void *end,
  150. const char *s, u32 len)
  151. {
  152. BUG_ON(*p + sizeof(len) + len > end);
  153. ceph_encode_32(p, len);
  154. if (len)
  155. memcpy(*p, s, len);
  156. *p += len;
  157. }
  158. #define ceph_encode_need(p, end, n, bad) \
  159. do { \
  160. if (!likely(ceph_has_room(p, end, n))) \
  161. goto bad; \
  162. } while (0)
  163. #define ceph_encode_64_safe(p, end, v, bad) \
  164. do { \
  165. ceph_encode_need(p, end, sizeof(u64), bad); \
  166. ceph_encode_64(p, v); \
  167. } while (0)
  168. #define ceph_encode_32_safe(p, end, v, bad) \
  169. do { \
  170. ceph_encode_need(p, end, sizeof(u32), bad); \
  171. ceph_encode_32(p, v); \
  172. } while (0)
  173. #define ceph_encode_16_safe(p, end, v, bad) \
  174. do { \
  175. ceph_encode_need(p, end, sizeof(u16), bad); \
  176. ceph_encode_16(p, v); \
  177. } while (0)
  178. #define ceph_encode_copy_safe(p, end, pv, n, bad) \
  179. do { \
  180. ceph_encode_need(p, end, n, bad); \
  181. ceph_encode_copy(p, pv, n); \
  182. } while (0)
  183. #define ceph_encode_string_safe(p, end, s, n, bad) \
  184. do { \
  185. ceph_encode_need(p, end, n, bad); \
  186. ceph_encode_string(p, end, s, n); \
  187. } while (0)
  188. #endif