include/linux/sunrpc/xdr.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 /*
   3  * XDR standard data types and function declarations
   4  *
   5  * Copyright (C) 1995-1997 Olaf Kirch <okir@monad.swb.de>
   6  *
   7  * Based on:
   8  *   RFC 4506 "XDR: External Data Representation Standard", May 2006
   9  */
  10
  11 #ifndef _SUNRPC_XDR_H_
  12 #define _SUNRPC_XDR_H_
  13
  14 #include <linux/uio.h>
  15 #include <asm/byteorder.h>
  16 #include <asm/unaligned.h>
  17 #include <linux/scatterlist.h>
  18
  19 struct bio_vec;
  20 struct rpc_rqst;
  21
  22 /*
  23  * Buffer adjustment
  24  */
  25 #define XDR_QUADLEN(l)          (((l) + 3) >> 2)
  26
  27 /*
  28  * Generic opaque `network object.'
  29  */
  30 #define XDR_MAX_NETOBJ          1024
  31 struct xdr_netobj {
  32         unsigned int            len;
  33         u8 *                    data;
  34 };
  35
  36 /*
  37  * Basic structure for transmission/reception of a client XDR message.
  38  * Features a header (for a linear buffer containing RPC headers
  39  * and the data payload for short messages), and then an array of
  40  * pages.
  41  * The tail iovec allows you to append data after the page array. Its
  42  * main interest is for appending padding to the pages in order to
  43  * satisfy the int_32-alignment requirements in RFC1832.
  44  *
  45  * For the future, we might want to string several of these together
  46  * in a list if anybody wants to make use of NFSv4 COMPOUND
  47  * operations and/or has a need for scatter/gather involving pages.
  48  */
  49 struct xdr_buf {
  50         struct kvec     head[1],        /* RPC header + non-page data */
  51                         tail[1];        /* Appended after page data */
  52
  53         struct bio_vec  *bvec;
  54         struct page **  pages;          /* Array of pages */
  55         unsigned int    page_base,      /* Start of page data */
  56                         page_len,       /* Length of page data */
  57                         flags;          /* Flags for data disposition */
  58 #define XDRBUF_READ             0x01            /* target of file read */
  59 #define XDRBUF_WRITE            0x02            /* source of file write */
  60 #define XDRBUF_SPARSE_PAGES     0x04            /* Page array is sparse */
  61
  62         unsigned int    buflen,         /* Total length of storage buffer */
  63                         len;            /* Length of XDR encoded message */
  64 };
  65
  66 static inline void
  67 xdr_buf_init(struct xdr_buf *buf, void *start, size_t len)
  68 {
  69         buf->head[0].iov_base = start;
  70         buf->head[0].iov_len = len;
  71         buf->tail[0].iov_len = 0;
  72         buf->pages = NULL;
  73         buf->page_len = 0;
  74         buf->flags = 0;
  75         buf->len = 0;
  76         buf->buflen = len;
  77 }
  78
  79 /*
  80  * pre-xdr'ed macros.
  81  */
  82
  83 #define xdr_zero        cpu_to_be32(0)
  84 #define xdr_one         cpu_to_be32(1)
  85 #define xdr_two         cpu_to_be32(2)
  86
  87 #define rpc_auth_null   cpu_to_be32(RPC_AUTH_NULL)
  88 #define rpc_auth_unix   cpu_to_be32(RPC_AUTH_UNIX)
  89 #define rpc_auth_short  cpu_to_be32(RPC_AUTH_SHORT)
  90 #define rpc_auth_gss    cpu_to_be32(RPC_AUTH_GSS)
  91
  92 #define rpc_call        cpu_to_be32(RPC_CALL)
  93 #define rpc_reply       cpu_to_be32(RPC_REPLY)
  94
  95 #define rpc_msg_accepted        cpu_to_be32(RPC_MSG_ACCEPTED)
  96
  97 #define rpc_success             cpu_to_be32(RPC_SUCCESS)
  98 #define rpc_prog_unavail        cpu_to_be32(RPC_PROG_UNAVAIL)
  99 #define rpc_prog_mismatch       cpu_to_be32(RPC_PROG_MISMATCH)
 100 #define rpc_proc_unavail        cpu_to_be32(RPC_PROC_UNAVAIL)
 101 #define rpc_garbage_args        cpu_to_be32(RPC_GARBAGE_ARGS)
 102 #define rpc_system_err          cpu_to_be32(RPC_SYSTEM_ERR)
 103 #define rpc_drop_reply          cpu_to_be32(RPC_DROP_REPLY)
 104
 105 #define rpc_mismatch            cpu_to_be32(RPC_MISMATCH)
 106 #define rpc_auth_error          cpu_to_be32(RPC_AUTH_ERROR)
 107
 108 #define rpc_auth_ok             cpu_to_be32(RPC_AUTH_OK)
 109 #define rpc_autherr_badcred     cpu_to_be32(RPC_AUTH_BADCRED)
 110 #define rpc_autherr_rejectedcred cpu_to_be32(RPC_AUTH_REJECTEDCRED)
 111 #define rpc_autherr_badverf     cpu_to_be32(RPC_AUTH_BADVERF)
 112 #define rpc_autherr_rejectedverf cpu_to_be32(RPC_AUTH_REJECTEDVERF)
 113 #define rpc_autherr_tooweak     cpu_to_be32(RPC_AUTH_TOOWEAK)
 114 #define rpcsec_gsserr_credproblem       cpu_to_be32(RPCSEC_GSS_CREDPROBLEM)
 115 #define rpcsec_gsserr_ctxproblem        cpu_to_be32(RPCSEC_GSS_CTXPROBLEM)
 116
 117 /*
 118  * Miscellaneous XDR helper functions
 119  */
 120 __be32 *xdr_encode_opaque_fixed(__be32 *p, const void *ptr, unsigned int len);
 121 __be32 *xdr_encode_opaque(__be32 *p, const void *ptr, unsigned int len);
 122 __be32 *xdr_encode_string(__be32 *p, const char *s);
 123 __be32 *xdr_decode_string_inplace(__be32 *p, char **sp, unsigned int *lenp,
 124                         unsigned int maxlen);
 125 __be32 *xdr_encode_netobj(__be32 *p, const struct xdr_netobj *);
 126 __be32 *xdr_decode_netobj(__be32 *p, struct xdr_netobj *);
 127
 128 void    xdr_inline_pages(struct xdr_buf *, unsigned int,
 129                          struct page **, unsigned int, unsigned int);
 130 void    xdr_terminate_string(struct xdr_buf *, const u32);
 131 size_t  xdr_buf_pagecount(struct xdr_buf *buf);
 132 int     xdr_alloc_bvec(struct xdr_buf *buf, gfp_t gfp);
 133 void    xdr_free_bvec(struct xdr_buf *buf);
 134
 135 static inline __be32 *xdr_encode_array(__be32 *p, const void *s, unsigned int len)
 136 {
 137         return xdr_encode_opaque(p, s, len);
 138 }
 139
 140 /*
 141  * Decode 64bit quantities (NFSv3 support)
 142  */
 143 static inline __be32 *
 144 xdr_encode_hyper(__be32 *p, __u64 val)
 145 {
 146         put_unaligned_be64(val, p);
 147         return p + 2;
 148 }
 149
 150 static inline __be32 *
 151 xdr_decode_hyper(__be32 *p, __u64 *valp)
 152 {
 153         *valp = get_unaligned_be64(p);
 154         return p + 2;
 155 }
 156
 157 static inline __be32 *
 158 xdr_decode_opaque_fixed(__be32 *p, void *ptr, unsigned int len)
 159 {
 160         memcpy(ptr, p, len);
 161         return p + XDR_QUADLEN(len);
 162 }
 163
 164 static inline void xdr_netobj_dup(struct xdr_netobj *dst,
 165                                   struct xdr_netobj *src, gfp_t gfp_mask)
 166 {
 167         dst->data = kmemdup(src->data, src->len, gfp_mask);
 168         dst->len = src->len;
 169 }
 170
 171 /*
 172  * Adjust kvec to reflect end of xdr'ed data (RPC client XDR)
 173  */
 174 static inline int
 175 xdr_adjust_iovec(struct kvec *iov, __be32 *p)
 176 {
 177         return iov->iov_len = ((u8 *) p - (u8 *) iov->iov_base);
 178 }
 179
 180 /*
 181  * XDR buffer helper functions
 182  */
 183 extern void xdr_shift_buf(struct xdr_buf *, size_t);
 184 extern void xdr_buf_from_iov(struct kvec *, struct xdr_buf *);
 185 extern int xdr_buf_subsegment(struct xdr_buf *, struct xdr_buf *, unsigned int, unsigned int);
 186 extern void xdr_buf_trim(struct xdr_buf *, unsigned int);
 187 extern int read_bytes_from_xdr_buf(struct xdr_buf *, unsigned int, void *, unsigned int);
 188 extern int write_bytes_to_xdr_buf(struct xdr_buf *, unsigned int, void *, unsigned int);
 189
 190 extern int xdr_encode_word(struct xdr_buf *, unsigned int, u32);
 191 extern int xdr_decode_word(struct xdr_buf *, unsigned int, u32 *);
 192
 193 struct xdr_array2_desc;
 194 typedef int (*xdr_xcode_elem_t)(struct xdr_array2_desc *desc, void *elem);
 195 struct xdr_array2_desc {
 196         unsigned int elem_size;
 197         unsigned int array_len;
 198         unsigned int array_maxlen;
 199         xdr_xcode_elem_t xcode;
 200 };
 201
 202 extern int xdr_decode_array2(struct xdr_buf *buf, unsigned int base,
 203                              struct xdr_array2_desc *desc);
 204 extern int xdr_encode_array2(struct xdr_buf *buf, unsigned int base,
 205                              struct xdr_array2_desc *desc);
 206 extern void _copy_from_pages(char *p, struct page **pages, size_t pgbase,
 207                              size_t len);
 208
 209 /*
 210  * Provide some simple tools for XDR buffer overflow-checking etc.
 211  */
 212 struct xdr_stream {
 213         __be32 *p;              /* start of available buffer */
 214         struct xdr_buf *buf;    /* XDR buffer to read/write */
 215
 216         __be32 *end;            /* end of available buffer space */
 217         struct kvec *iov;       /* pointer to the current kvec */
 218         struct kvec scratch;    /* Scratch buffer */
 219         struct page **page_ptr; /* pointer to the current page */
 220         unsigned int nwords;    /* Remaining decode buffer length */
 221
 222         struct rpc_rqst *rqst;  /* For debugging */
 223 };
 224
 225 /*
 226  * These are the xdr_stream style generic XDR encode and decode functions.
 227  */
 228 typedef void    (*kxdreproc_t)(struct rpc_rqst *rqstp, struct xdr_stream *xdr,
 229                 const void *obj);
 230 typedef int     (*kxdrdproc_t)(struct rpc_rqst *rqstp, struct xdr_stream *xdr,
 231                 void *obj);
 232
 233 extern void xdr_init_encode(struct xdr_stream *xdr, struct xdr_buf *buf,
 234                             __be32 *p, struct rpc_rqst *rqst);
 235 extern __be32 *xdr_reserve_space(struct xdr_stream *xdr, size_t nbytes);
 236 extern int xdr_reserve_space_vec(struct xdr_stream *xdr, struct kvec *vec,
 237                 size_t nbytes);
 238 extern void xdr_commit_encode(struct xdr_stream *xdr);
 239 extern void xdr_truncate_encode(struct xdr_stream *xdr, size_t len);
 240 extern int xdr_restrict_buflen(struct xdr_stream *xdr, int newbuflen);
 241 extern void xdr_write_pages(struct xdr_stream *xdr, struct page **pages,
 242                 unsigned int base, unsigned int len);
 243 extern unsigned int xdr_stream_pos(const struct xdr_stream *xdr);
 244 extern unsigned int xdr_page_pos(const struct xdr_stream *xdr);
 245 extern void xdr_init_decode(struct xdr_stream *xdr, struct xdr_buf *buf,
 246                             __be32 *p, struct rpc_rqst *rqst);
 247 extern void xdr_init_decode_pages(struct xdr_stream *xdr, struct xdr_buf *buf,
 248                 struct page **pages, unsigned int len);
 249 extern void xdr_set_scratch_buffer(struct xdr_stream *xdr, void *buf, size_t buflen);
 250 extern __be32 *xdr_inline_decode(struct xdr_stream *xdr, size_t nbytes);
 251 extern unsigned int xdr_read_pages(struct xdr_stream *xdr, unsigned int len);
 252 extern void xdr_enter_page(struct xdr_stream *xdr, unsigned int len);
 253 extern int xdr_process_buf(struct xdr_buf *buf, unsigned int offset, unsigned int len, int (*actor)(struct scatterlist *, void *), void *data);
 254 extern uint64_t xdr_align_data(struct xdr_stream *, uint64_t, uint32_t);
 255 extern uint64_t xdr_expand_hole(struct xdr_stream *, uint64_t, uint64_t);
 256
 257 /**
 258  * xdr_stream_remaining - Return the number of bytes remaining in the stream
 259  * @xdr: pointer to struct xdr_stream
 260  *
 261  * Return value:
 262  *   Number of bytes remaining in @xdr before xdr->end
 263  */
 264 static inline size_t
 265 xdr_stream_remaining(const struct xdr_stream *xdr)
 266 {
 267         return xdr->nwords << 2;
 268 }
 269
 270 ssize_t xdr_stream_decode_opaque(struct xdr_stream *xdr, void *ptr,
 271                 size_t size);
 272 ssize_t xdr_stream_decode_opaque_dup(struct xdr_stream *xdr, void **ptr,
 273                 size_t maxlen, gfp_t gfp_flags);
 274 ssize_t xdr_stream_decode_string(struct xdr_stream *xdr, char *str,
 275                 size_t size);
 276 ssize_t xdr_stream_decode_string_dup(struct xdr_stream *xdr, char **str,
 277                 size_t maxlen, gfp_t gfp_flags);
 278 /**
 279  * xdr_align_size - Calculate padded size of an object
 280  * @n: Size of an object being XDR encoded (in bytes)
 281  *
 282  * Return value:
 283  *   Size (in bytes) of the object including xdr padding
 284  */
 285 static inline size_t
 286 xdr_align_size(size_t n)
 287 {
 288         const size_t mask = sizeof(__u32) - 1;
 289
 290         return (n + mask) & ~mask;
 291 }
 292
 293 /**
 294  * xdr_pad_size - Calculate size of an object's pad
 295  * @n: Size of an object being XDR encoded (in bytes)
 296  *
 297  * This implementation avoids the need for conditional
 298  * branches or modulo division.
 299  *
 300  * Return value:
 301  *   Size (in bytes) of the needed XDR pad
 302  */
 303 static inline size_t xdr_pad_size(size_t n)
 304 {
 305         return xdr_align_size(n) - n;
 306 }
 307
 308 /**
 309  * xdr_stream_encode_item_present - Encode a "present" list item
 310  * @xdr: pointer to xdr_stream
 311  *
 312  * Return values:
 313  *   On success, returns length in bytes of XDR buffer consumed
 314  *   %-EMSGSIZE on XDR buffer overflow
 315  */
 316 static inline ssize_t xdr_stream_encode_item_present(struct xdr_stream *xdr)
 317 {
 318         const size_t len = sizeof(__be32);
 319         __be32 *p = xdr_reserve_space(xdr, len);
 320
 321         if (unlikely(!p))
 322                 return -EMSGSIZE;
 323         *p = xdr_one;
 324         return len;
 325 }
 326
 327 /**
 328  * xdr_stream_encode_item_absent - Encode a "not present" list item
 329  * @xdr: pointer to xdr_stream
 330  *
 331  * Return values:
 332  *   On success, returns length in bytes of XDR buffer consumed
 333  *   %-EMSGSIZE on XDR buffer overflow
 334  */
 335 static inline int xdr_stream_encode_item_absent(struct xdr_stream *xdr)
 336 {
 337         const size_t len = sizeof(__be32);
 338         __be32 *p = xdr_reserve_space(xdr, len);
 339
 340         if (unlikely(!p))
 341                 return -EMSGSIZE;
 342         *p = xdr_zero;
 343         return len;
 344 }
 345
 346 /**
 347  * xdr_stream_encode_u32 - Encode a 32-bit integer
 348  * @xdr: pointer to xdr_stream
 349  * @n: integer to encode
 350  *
 351  * Return values:
 352  *   On success, returns length in bytes of XDR buffer consumed
 353  *   %-EMSGSIZE on XDR buffer overflow
 354  */
 355 static inline ssize_t
 356 xdr_stream_encode_u32(struct xdr_stream *xdr, __u32 n)
 357 {
 358         const size_t len = sizeof(n);
 359         __be32 *p = xdr_reserve_space(xdr, len);
 360
 361         if (unlikely(!p))
 362                 return -EMSGSIZE;
 363         *p = cpu_to_be32(n);
 364         return len;
 365 }
 366
 367 /**
 368  * xdr_stream_encode_u64 - Encode a 64-bit integer
 369  * @xdr: pointer to xdr_stream
 370  * @n: 64-bit integer to encode
 371  *
 372  * Return values:
 373  *   On success, returns length in bytes of XDR buffer consumed
 374  *   %-EMSGSIZE on XDR buffer overflow
 375  */
 376 static inline ssize_t
 377 xdr_stream_encode_u64(struct xdr_stream *xdr, __u64 n)
 378 {
 379         const size_t len = sizeof(n);
 380         __be32 *p = xdr_reserve_space(xdr, len);
 381
 382         if (unlikely(!p))
 383                 return -EMSGSIZE;
 384         xdr_encode_hyper(p, n);
 385         return len;
 386 }
 387
 388 /**
 389  * xdr_stream_encode_opaque_inline - Encode opaque xdr data
 390  * @xdr: pointer to xdr_stream
 391  * @ptr: pointer to void pointer
 392  * @len: size of object
 393  *
 394  * Return values:
 395  *   On success, returns length in bytes of XDR buffer consumed
 396  *   %-EMSGSIZE on XDR buffer overflow
 397  */
 398 static inline ssize_t
 399 xdr_stream_encode_opaque_inline(struct xdr_stream *xdr, void **ptr, size_t len)
 400 {
 401         size_t count = sizeof(__u32) + xdr_align_size(len);
 402         __be32 *p = xdr_reserve_space(xdr, count);
 403
 404         if (unlikely(!p)) {
 405                 *ptr = NULL;
 406                 return -EMSGSIZE;
 407         }
 408         xdr_encode_opaque(p, NULL, len);
 409         *ptr = ++p;
 410         return count;
 411 }
 412
 413 /**
 414  * xdr_stream_encode_opaque_fixed - Encode fixed length opaque xdr data
 415  * @xdr: pointer to xdr_stream
 416  * @ptr: pointer to opaque data object
 417  * @len: size of object pointed to by @ptr
 418  *
 419  * Return values:
 420  *   On success, returns length in bytes of XDR buffer consumed
 421  *   %-EMSGSIZE on XDR buffer overflow
 422  */
 423 static inline ssize_t
 424 xdr_stream_encode_opaque_fixed(struct xdr_stream *xdr, const void *ptr, size_t len)
 425 {
 426         __be32 *p = xdr_reserve_space(xdr, len);
 427
 428         if (unlikely(!p))
 429                 return -EMSGSIZE;
 430         xdr_encode_opaque_fixed(p, ptr, len);
 431         return xdr_align_size(len);
 432 }
 433
 434 /**
 435  * xdr_stream_encode_opaque - Encode variable length opaque xdr data
 436  * @xdr: pointer to xdr_stream
 437  * @ptr: pointer to opaque data object
 438  * @len: size of object pointed to by @ptr
 439  *
 440  * Return values:
 441  *   On success, returns length in bytes of XDR buffer consumed
 442  *   %-EMSGSIZE on XDR buffer overflow
 443  */
 444 static inline ssize_t
 445 xdr_stream_encode_opaque(struct xdr_stream *xdr, const void *ptr, size_t len)
 446 {
 447         size_t count = sizeof(__u32) + xdr_align_size(len);
 448         __be32 *p = xdr_reserve_space(xdr, count);
 449
 450         if (unlikely(!p))
 451                 return -EMSGSIZE;
 452         xdr_encode_opaque(p, ptr, len);
 453         return count;
 454 }
 455
 456 /**
 457  * xdr_stream_encode_uint32_array - Encode variable length array of integers
 458  * @xdr: pointer to xdr_stream
 459  * @array: array of integers
 460  * @array_size: number of elements in @array
 461  *
 462  * Return values:
 463  *   On success, returns length in bytes of XDR buffer consumed
 464  *   %-EMSGSIZE on XDR buffer overflow
 465  */
 466 static inline ssize_t
 467 xdr_stream_encode_uint32_array(struct xdr_stream *xdr,
 468                 const __u32 *array, size_t array_size)
 469 {
 470         ssize_t ret = (array_size+1) * sizeof(__u32);
 471         __be32 *p = xdr_reserve_space(xdr, ret);
 472
 473         if (unlikely(!p))
 474                 return -EMSGSIZE;
 475         *p++ = cpu_to_be32(array_size);
 476         for (; array_size > 0; p++, array++, array_size--)
 477                 *p = cpu_to_be32p(array);
 478         return ret;
 479 }
 480
 481 /**
 482  * xdr_item_is_absent - symbolically handle XDR discriminators
 483  * @p: pointer to undecoded discriminator
 484  *
 485  * Return values:
 486  *   %true if the following XDR item is absent
 487  *   %false if the following XDR item is present
 488  */
 489 static inline bool xdr_item_is_absent(const __be32 *p)
 490 {
 491         return *p == xdr_zero;
 492 }
 493
 494 /**
 495  * xdr_item_is_present - symbolically handle XDR discriminators
 496  * @p: pointer to undecoded discriminator
 497  *
 498  * Return values:
 499  *   %true if the following XDR item is present
 500  *   %false if the following XDR item is absent
 501  */
 502 static inline bool xdr_item_is_present(const __be32 *p)
 503 {
 504         return *p != xdr_zero;
 505 }
 506
 507 /**
 508  * xdr_stream_decode_u32 - Decode a 32-bit integer
 509  * @xdr: pointer to xdr_stream
 510  * @ptr: location to store integer
 511  *
 512  * Return values:
 513  *   %0 on success
 514  *   %-EBADMSG on XDR buffer overflow
 515  */
 516 static inline ssize_t
 517 xdr_stream_decode_u32(struct xdr_stream *xdr, __u32 *ptr)
 518 {
 519         const size_t count = sizeof(*ptr);
 520         __be32 *p = xdr_inline_decode(xdr, count);
 521
 522         if (unlikely(!p))
 523                 return -EBADMSG;
 524         *ptr = be32_to_cpup(p);
 525         return 0;
 526 }
 527
 528 /**
 529  * xdr_stream_decode_opaque_fixed - Decode fixed length opaque xdr data
 530  * @xdr: pointer to xdr_stream
 531  * @ptr: location to store data
 532  * @len: size of buffer pointed to by @ptr
 533  *
 534  * Return values:
 535  *   On success, returns size of object stored in @ptr
 536  *   %-EBADMSG on XDR buffer overflow
 537  */
 538 static inline ssize_t
 539 xdr_stream_decode_opaque_fixed(struct xdr_stream *xdr, void *ptr, size_t len)
 540 {
 541         __be32 *p = xdr_inline_decode(xdr, len);
 542
 543         if (unlikely(!p))
 544                 return -EBADMSG;
 545         xdr_decode_opaque_fixed(p, ptr, len);
 546         return len;
 547 }
 548
 549 /**
 550  * xdr_stream_decode_opaque_inline - Decode variable length opaque xdr data
 551  * @xdr: pointer to xdr_stream
 552  * @ptr: location to store pointer to opaque data
 553  * @maxlen: maximum acceptable object size
 554  *
 555  * Note: the pointer stored in @ptr cannot be assumed valid after the XDR
 556  * buffer has been destroyed, or even after calling xdr_inline_decode()
 557  * on @xdr. It is therefore expected that the object it points to should
 558  * be processed immediately.
 559  *
 560  * Return values:
 561  *   On success, returns size of object stored in *@ptr
 562  *   %-EBADMSG on XDR buffer overflow
 563  *   %-EMSGSIZE if the size of the object would exceed @maxlen
 564  */
 565 static inline ssize_t
 566 xdr_stream_decode_opaque_inline(struct xdr_stream *xdr, void **ptr, size_t maxlen)
 567 {
 568         __be32 *p;
 569         __u32 len;
 570
 571         *ptr = NULL;
 572         if (unlikely(xdr_stream_decode_u32(xdr, &len) < 0))
 573                 return -EBADMSG;
 574         if (len != 0) {
 575                 p = xdr_inline_decode(xdr, len);
 576                 if (unlikely(!p))
 577                         return -EBADMSG;
 578                 if (unlikely(len > maxlen))
 579                         return -EMSGSIZE;
 580                 *ptr = p;
 581         }
 582         return len;
 583 }
 584
 585 /**
 586  * xdr_stream_decode_uint32_array - Decode variable length array of integers
 587  * @xdr: pointer to xdr_stream
 588  * @array: location to store the integer array or NULL
 589  * @array_size: number of elements to store
 590  *
 591  * Return values:
 592  *   On success, returns number of elements stored in @array
 593  *   %-EBADMSG on XDR buffer overflow
 594  *   %-EMSGSIZE if the size of the array exceeds @array_size
 595  */
 596 static inline ssize_t
 597 xdr_stream_decode_uint32_array(struct xdr_stream *xdr,
 598                 __u32 *array, size_t array_size)
 599 {
 600         __be32 *p;
 601         __u32 len;
 602         ssize_t retval;
 603
 604         if (unlikely(xdr_stream_decode_u32(xdr, &len) < 0))
 605                 return -EBADMSG;
 606         if (len > SIZE_MAX / sizeof(*p))
 607                 return -EBADMSG;
 608         p = xdr_inline_decode(xdr, len * sizeof(*p));
 609         if (unlikely(!p))
 610                 return -EBADMSG;
 611         if (array == NULL)
 612                 return len;
 613         if (len <= array_size) {
 614                 if (len < array_size)
 615                         memset(array+len, 0, (array_size-len)*sizeof(*array));
 616                 array_size = len;
 617                 retval = len;
 618         } else
 619                 retval = -EMSGSIZE;
 620         for (; array_size > 0; p++, array++, array_size--)
 621                 *array = be32_to_cpup(p);
 622         return retval;
 623 }
 624
 625 #endif /* _SUNRPC_XDR_H_ */