decode.h 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194
  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. a->in_addr.ss_family = htons(a->in_addr.ss_family);
  94. }
  95. static inline void ceph_decode_addr(struct ceph_entity_addr *a)
  96. {
  97. a->in_addr.ss_family = ntohs(a->in_addr.ss_family);
  98. WARN_ON(a->in_addr.ss_family == 512);
  99. }
  100. /*
  101. * encoders
  102. */
  103. static inline void ceph_encode_64(void **p, u64 v)
  104. {
  105. put_unaligned_le64(v, (__le64 *)*p);
  106. *p += sizeof(u64);
  107. }
  108. static inline void ceph_encode_32(void **p, u32 v)
  109. {
  110. put_unaligned_le32(v, (__le32 *)*p);
  111. *p += sizeof(u32);
  112. }
  113. static inline void ceph_encode_16(void **p, u16 v)
  114. {
  115. put_unaligned_le16(v, (__le16 *)*p);
  116. *p += sizeof(u16);
  117. }
  118. static inline void ceph_encode_8(void **p, u8 v)
  119. {
  120. *(u8 *)*p = v;
  121. (*p)++;
  122. }
  123. static inline void ceph_encode_copy(void **p, const void *s, int len)
  124. {
  125. memcpy(*p, s, len);
  126. *p += len;
  127. }
  128. /*
  129. * filepath, string encoders
  130. */
  131. static inline void ceph_encode_filepath(void **p, void *end,
  132. u64 ino, const char *path)
  133. {
  134. u32 len = path ? strlen(path) : 0;
  135. BUG_ON(*p + sizeof(ino) + sizeof(len) + len > end);
  136. ceph_encode_8(p, 1);
  137. ceph_encode_64(p, ino);
  138. ceph_encode_32(p, len);
  139. if (len)
  140. memcpy(*p, path, len);
  141. *p += len;
  142. }
  143. static inline void ceph_encode_string(void **p, void *end,
  144. const char *s, u32 len)
  145. {
  146. BUG_ON(*p + sizeof(len) + len > end);
  147. ceph_encode_32(p, len);
  148. if (len)
  149. memcpy(*p, s, len);
  150. *p += len;
  151. }
  152. #define ceph_encode_need(p, end, n, bad) \
  153. do { \
  154. if (unlikely(*(p) + (n) > (end))) \
  155. goto bad; \
  156. } while (0)
  157. #define ceph_encode_64_safe(p, end, v, bad) \
  158. do { \
  159. ceph_encode_need(p, end, sizeof(u64), bad); \
  160. ceph_encode_64(p, v); \
  161. } while (0)
  162. #define ceph_encode_32_safe(p, end, v, bad) \
  163. do { \
  164. ceph_encode_need(p, end, sizeof(u32), bad); \
  165. ceph_encode_32(p, v); \
  166. } while (0)
  167. #define ceph_encode_16_safe(p, end, v, bad) \
  168. do { \
  169. ceph_encode_need(p, end, sizeof(u16), bad); \
  170. ceph_encode_16(p, v); \
  171. } while (0)
  172. #define ceph_encode_copy_safe(p, end, pv, n, bad) \
  173. do { \
  174. ceph_encode_need(p, end, n, bad); \
  175. ceph_encode_copy(p, pv, n); \
  176. } while (0)
  177. #endif