1 /* 2 * linux/net/sunrpc/xdr.c 3 * 4 * Generic XDR support. 5 * 6 * Copyright (C) 1995, 1996 Olaf Kirch <okir@monad.swb.de> 7 */ 8 9 #include <linux/module.h> 10 #include <linux/slab.h> 11 #include <linux/types.h> 12 #include <linux/string.h> 13 #include <linux/kernel.h> 14 #include <linux/pagemap.h> 15 #include <linux/errno.h> 16 #include <linux/sunrpc/xdr.h> 17 #include <linux/sunrpc/msg_prot.h> 18 #include <linux/bvec.h> 19 #include <trace/events/sunrpc.h> 20 21 /* 22 * XDR functions for basic NFS types 23 */ 24 __be32 * 25 xdr_encode_netobj(__be32 *p, const struct xdr_netobj *obj) 26 { 27 unsigned int quadlen = XDR_QUADLEN(obj->len); 28 29 p[quadlen] = 0; /* zero trailing bytes */ 30 *p++ = cpu_to_be32(obj->len); 31 memcpy(p, obj->data, obj->len); 32 return p + XDR_QUADLEN(obj->len); 33 } 34 EXPORT_SYMBOL_GPL(xdr_encode_netobj); 35 36 __be32 * 37 xdr_decode_netobj(__be32 *p, struct xdr_netobj *obj) 38 { 39 unsigned int len; 40 41 if ((len = be32_to_cpu(*p++)) > XDR_MAX_NETOBJ) 42 return NULL; 43 obj->len = len; 44 obj->data = (u8 *) p; 45 return p + XDR_QUADLEN(len); 46 } 47 EXPORT_SYMBOL_GPL(xdr_decode_netobj); 48 49 /** 50 * xdr_encode_opaque_fixed - Encode fixed length opaque data 51 * @p: pointer to current position in XDR buffer. 52 * @ptr: pointer to data to encode (or NULL) 53 * @nbytes: size of data. 54 * 55 * Copy the array of data of length nbytes at ptr to the XDR buffer 56 * at position p, then align to the next 32-bit boundary by padding 57 * with zero bytes (see RFC1832). 58 * Note: if ptr is NULL, only the padding is performed. 59 * 60 * Returns the updated current XDR buffer position 61 * 62 */ 63 __be32 *xdr_encode_opaque_fixed(__be32 *p, const void *ptr, unsigned int nbytes) 64 { 65 if (likely(nbytes != 0)) { 66 unsigned int quadlen = XDR_QUADLEN(nbytes); 67 unsigned int padding = (quadlen << 2) - nbytes; 68 69 if (ptr != NULL) 70 memcpy(p, ptr, nbytes); 71 if (padding != 0) 72 memset((char *)p + nbytes, 0, padding); 73 p += quadlen; 74 } 75 return p; 76 } 77 EXPORT_SYMBOL_GPL(xdr_encode_opaque_fixed); 78 79 /** 80 * xdr_encode_opaque - Encode variable length opaque data 81 * @p: pointer to current position in XDR buffer. 82 * @ptr: pointer to data to encode (or NULL) 83 * @nbytes: size of data. 84 * 85 * Returns the updated current XDR buffer position 86 */ 87 __be32 *xdr_encode_opaque(__be32 *p, const void *ptr, unsigned int nbytes) 88 { 89 *p++ = cpu_to_be32(nbytes); 90 return xdr_encode_opaque_fixed(p, ptr, nbytes); 91 } 92 EXPORT_SYMBOL_GPL(xdr_encode_opaque); 93 94 __be32 * 95 xdr_encode_string(__be32 *p, const char *string) 96 { 97 return xdr_encode_array(p, string, strlen(string)); 98 } 99 EXPORT_SYMBOL_GPL(xdr_encode_string); 100 101 __be32 * 102 xdr_decode_string_inplace(__be32 *p, char **sp, 103 unsigned int *lenp, unsigned int maxlen) 104 { 105 u32 len; 106 107 len = be32_to_cpu(*p++); 108 if (len > maxlen) 109 return NULL; 110 *lenp = len; 111 *sp = (char *) p; 112 return p + XDR_QUADLEN(len); 113 } 114 EXPORT_SYMBOL_GPL(xdr_decode_string_inplace); 115 116 /** 117 * xdr_terminate_string - '\0'-terminate a string residing in an xdr_buf 118 * @buf: XDR buffer where string resides 119 * @len: length of string, in bytes 120 * 121 */ 122 void 123 xdr_terminate_string(struct xdr_buf *buf, const u32 len) 124 { 125 char *kaddr; 126 127 kaddr = kmap_atomic(buf->pages[0]); 128 kaddr[buf->page_base + len] = '\0'; 129 kunmap_atomic(kaddr); 130 } 131 EXPORT_SYMBOL_GPL(xdr_terminate_string); 132 133 size_t 134 xdr_buf_pagecount(struct xdr_buf *buf) 135 { 136 if (!buf->page_len) 137 return 0; 138 return (buf->page_base + buf->page_len + PAGE_SIZE - 1) >> PAGE_SHIFT; 139 } 140 141 int 142 xdr_alloc_bvec(struct xdr_buf *buf, gfp_t gfp) 143 { 144 size_t i, n = xdr_buf_pagecount(buf); 145 146 if (n != 0 && buf->bvec == NULL) { 147 buf->bvec = kmalloc_array(n, sizeof(buf->bvec[0]), gfp); 148 if (!buf->bvec) 149 return -ENOMEM; 150 for (i = 0; i < n; i++) { 151 buf->bvec[i].bv_page = buf->pages[i]; 152 buf->bvec[i].bv_len = PAGE_SIZE; 153 buf->bvec[i].bv_offset = 0; 154 } 155 } 156 return 0; 157 } 158 159 void 160 xdr_free_bvec(struct xdr_buf *buf) 161 { 162 kfree(buf->bvec); 163 buf->bvec = NULL; 164 } 165 166 /** 167 * xdr_inline_pages - Prepare receive buffer for a large reply 168 * @xdr: xdr_buf into which reply will be placed 169 * @offset: expected offset where data payload will start, in bytes 170 * @pages: vector of struct page pointers 171 * @base: offset in first page where receive should start, in bytes 172 * @len: expected size of the upper layer data payload, in bytes 173 * 174 */ 175 void 176 xdr_inline_pages(struct xdr_buf *xdr, unsigned int offset, 177 struct page **pages, unsigned int base, unsigned int len) 178 { 179 struct kvec *head = xdr->head; 180 struct kvec *tail = xdr->tail; 181 char *buf = (char *)head->iov_base; 182 unsigned int buflen = head->iov_len; 183 184 head->iov_len = offset; 185 186 xdr->pages = pages; 187 xdr->page_base = base; 188 xdr->page_len = len; 189 190 tail->iov_base = buf + offset; 191 tail->iov_len = buflen - offset; 192 if ((xdr->page_len & 3) == 0) 193 tail->iov_len -= sizeof(__be32); 194 195 xdr->buflen += len; 196 } 197 EXPORT_SYMBOL_GPL(xdr_inline_pages); 198 199 /* 200 * Helper routines for doing 'memmove' like operations on a struct xdr_buf 201 */ 202 203 /** 204 * _shift_data_right_pages 205 * @pages: vector of pages containing both the source and dest memory area. 206 * @pgto_base: page vector address of destination 207 * @pgfrom_base: page vector address of source 208 * @len: number of bytes to copy 209 * 210 * Note: the addresses pgto_base and pgfrom_base are both calculated in 211 * the same way: 212 * if a memory area starts at byte 'base' in page 'pages[i]', 213 * then its address is given as (i << PAGE_SHIFT) + base 214 * Also note: pgfrom_base must be < pgto_base, but the memory areas 215 * they point to may overlap. 216 */ 217 static void 218 _shift_data_right_pages(struct page **pages, size_t pgto_base, 219 size_t pgfrom_base, size_t len) 220 { 221 struct page **pgfrom, **pgto; 222 char *vfrom, *vto; 223 size_t copy; 224 225 BUG_ON(pgto_base <= pgfrom_base); 226 227 pgto_base += len; 228 pgfrom_base += len; 229 230 pgto = pages + (pgto_base >> PAGE_SHIFT); 231 pgfrom = pages + (pgfrom_base >> PAGE_SHIFT); 232 233 pgto_base &= ~PAGE_MASK; 234 pgfrom_base &= ~PAGE_MASK; 235 236 do { 237 /* Are any pointers crossing a page boundary? */ 238 if (pgto_base == 0) { 239 pgto_base = PAGE_SIZE; 240 pgto--; 241 } 242 if (pgfrom_base == 0) { 243 pgfrom_base = PAGE_SIZE; 244 pgfrom--; 245 } 246 247 copy = len; 248 if (copy > pgto_base) 249 copy = pgto_base; 250 if (copy > pgfrom_base) 251 copy = pgfrom_base; 252 pgto_base -= copy; 253 pgfrom_base -= copy; 254 255 vto = kmap_atomic(*pgto); 256 if (*pgto != *pgfrom) { 257 vfrom = kmap_atomic(*pgfrom); 258 memcpy(vto + pgto_base, vfrom + pgfrom_base, copy); 259 kunmap_atomic(vfrom); 260 } else 261 memmove(vto + pgto_base, vto + pgfrom_base, copy); 262 flush_dcache_page(*pgto); 263 kunmap_atomic(vto); 264 265 } while ((len -= copy) != 0); 266 } 267 268 /** 269 * _copy_to_pages 270 * @pages: array of pages 271 * @pgbase: page vector address of destination 272 * @p: pointer to source data 273 * @len: length 274 * 275 * Copies data from an arbitrary memory location into an array of pages 276 * The copy is assumed to be non-overlapping. 277 */ 278 static void 279 _copy_to_pages(struct page **pages, size_t pgbase, const char *p, size_t len) 280 { 281 struct page **pgto; 282 char *vto; 283 size_t copy; 284 285 pgto = pages + (pgbase >> PAGE_SHIFT); 286 pgbase &= ~PAGE_MASK; 287 288 for (;;) { 289 copy = PAGE_SIZE - pgbase; 290 if (copy > len) 291 copy = len; 292 293 vto = kmap_atomic(*pgto); 294 memcpy(vto + pgbase, p, copy); 295 kunmap_atomic(vto); 296 297 len -= copy; 298 if (len == 0) 299 break; 300 301 pgbase += copy; 302 if (pgbase == PAGE_SIZE) { 303 flush_dcache_page(*pgto); 304 pgbase = 0; 305 pgto++; 306 } 307 p += copy; 308 } 309 flush_dcache_page(*pgto); 310 } 311 312 /** 313 * _copy_from_pages 314 * @p: pointer to destination 315 * @pages: array of pages 316 * @pgbase: offset of source data 317 * @len: length 318 * 319 * Copies data into an arbitrary memory location from an array of pages 320 * The copy is assumed to be non-overlapping. 321 */ 322 void 323 _copy_from_pages(char *p, struct page **pages, size_t pgbase, size_t len) 324 { 325 struct page **pgfrom; 326 char *vfrom; 327 size_t copy; 328 329 pgfrom = pages + (pgbase >> PAGE_SHIFT); 330 pgbase &= ~PAGE_MASK; 331 332 do { 333 copy = PAGE_SIZE - pgbase; 334 if (copy > len) 335 copy = len; 336 337 vfrom = kmap_atomic(*pgfrom); 338 memcpy(p, vfrom + pgbase, copy); 339 kunmap_atomic(vfrom); 340 341 pgbase += copy; 342 if (pgbase == PAGE_SIZE) { 343 pgbase = 0; 344 pgfrom++; 345 } 346 p += copy; 347 348 } while ((len -= copy) != 0); 349 } 350 EXPORT_SYMBOL_GPL(_copy_from_pages); 351 352 /** 353 * xdr_shrink_bufhead 354 * @buf: xdr_buf 355 * @len: bytes to remove from buf->head[0] 356 * 357 * Shrinks XDR buffer's header kvec buf->head[0] by 358 * 'len' bytes. The extra data is not lost, but is instead 359 * moved into the inlined pages and/or the tail. 360 */ 361 static unsigned int 362 xdr_shrink_bufhead(struct xdr_buf *buf, size_t len) 363 { 364 struct kvec *head, *tail; 365 size_t copy, offs; 366 unsigned int pglen = buf->page_len; 367 unsigned int result; 368 369 result = 0; 370 tail = buf->tail; 371 head = buf->head; 372 373 WARN_ON_ONCE(len > head->iov_len); 374 if (len > head->iov_len) 375 len = head->iov_len; 376 377 /* Shift the tail first */ 378 if (tail->iov_len != 0) { 379 if (tail->iov_len > len) { 380 copy = tail->iov_len - len; 381 memmove((char *)tail->iov_base + len, 382 tail->iov_base, copy); 383 result += copy; 384 } 385 /* Copy from the inlined pages into the tail */ 386 copy = len; 387 if (copy > pglen) 388 copy = pglen; 389 offs = len - copy; 390 if (offs >= tail->iov_len) 391 copy = 0; 392 else if (copy > tail->iov_len - offs) 393 copy = tail->iov_len - offs; 394 if (copy != 0) { 395 _copy_from_pages((char *)tail->iov_base + offs, 396 buf->pages, 397 buf->page_base + pglen + offs - len, 398 copy); 399 result += copy; 400 } 401 /* Do we also need to copy data from the head into the tail ? */ 402 if (len > pglen) { 403 offs = copy = len - pglen; 404 if (copy > tail->iov_len) 405 copy = tail->iov_len; 406 memcpy(tail->iov_base, 407 (char *)head->iov_base + 408 head->iov_len - offs, 409 copy); 410 result += copy; 411 } 412 } 413 /* Now handle pages */ 414 if (pglen != 0) { 415 if (pglen > len) 416 _shift_data_right_pages(buf->pages, 417 buf->page_base + len, 418 buf->page_base, 419 pglen - len); 420 copy = len; 421 if (len > pglen) 422 copy = pglen; 423 _copy_to_pages(buf->pages, buf->page_base, 424 (char *)head->iov_base + head->iov_len - len, 425 copy); 426 result += copy; 427 } 428 head->iov_len -= len; 429 buf->buflen -= len; 430 /* Have we truncated the message? */ 431 if (buf->len > buf->buflen) 432 buf->len = buf->buflen; 433 434 return result; 435 } 436 437 /** 438 * xdr_shrink_pagelen 439 * @buf: xdr_buf 440 * @len: bytes to remove from buf->pages 441 * 442 * Shrinks XDR buffer's page array buf->pages by 443 * 'len' bytes. The extra data is not lost, but is instead 444 * moved into the tail. 445 */ 446 static unsigned int 447 xdr_shrink_pagelen(struct xdr_buf *buf, size_t len) 448 { 449 struct kvec *tail; 450 size_t copy; 451 unsigned int pglen = buf->page_len; 452 unsigned int tailbuf_len; 453 unsigned int result; 454 455 result = 0; 456 tail = buf->tail; 457 BUG_ON (len > pglen); 458 459 tailbuf_len = buf->buflen - buf->head->iov_len - buf->page_len; 460 461 /* Shift the tail first */ 462 if (tailbuf_len != 0) { 463 unsigned int free_space = tailbuf_len - tail->iov_len; 464 465 if (len < free_space) 466 free_space = len; 467 tail->iov_len += free_space; 468 469 copy = len; 470 if (tail->iov_len > len) { 471 char *p = (char *)tail->iov_base + len; 472 memmove(p, tail->iov_base, tail->iov_len - len); 473 result += tail->iov_len - len; 474 } else 475 copy = tail->iov_len; 476 /* Copy from the inlined pages into the tail */ 477 _copy_from_pages((char *)tail->iov_base, 478 buf->pages, buf->page_base + pglen - len, 479 copy); 480 result += copy; 481 } 482 buf->page_len -= len; 483 buf->buflen -= len; 484 /* Have we truncated the message? */ 485 if (buf->len > buf->buflen) 486 buf->len = buf->buflen; 487 488 return result; 489 } 490 491 void 492 xdr_shift_buf(struct xdr_buf *buf, size_t len) 493 { 494 xdr_shrink_bufhead(buf, len); 495 } 496 EXPORT_SYMBOL_GPL(xdr_shift_buf); 497 498 /** 499 * xdr_stream_pos - Return the current offset from the start of the xdr_stream 500 * @xdr: pointer to struct xdr_stream 501 */ 502 unsigned int xdr_stream_pos(const struct xdr_stream *xdr) 503 { 504 return (unsigned int)(XDR_QUADLEN(xdr->buf->len) - xdr->nwords) << 2; 505 } 506 EXPORT_SYMBOL_GPL(xdr_stream_pos); 507 508 /** 509 * xdr_init_encode - Initialize a struct xdr_stream for sending data. 510 * @xdr: pointer to xdr_stream struct 511 * @buf: pointer to XDR buffer in which to encode data 512 * @p: current pointer inside XDR buffer 513 * @rqst: pointer to controlling rpc_rqst, for debugging 514 * 515 * Note: at the moment the RPC client only passes the length of our 516 * scratch buffer in the xdr_buf's header kvec. Previously this 517 * meant we needed to call xdr_adjust_iovec() after encoding the 518 * data. With the new scheme, the xdr_stream manages the details 519 * of the buffer length, and takes care of adjusting the kvec 520 * length for us. 521 */ 522 void xdr_init_encode(struct xdr_stream *xdr, struct xdr_buf *buf, __be32 *p, 523 struct rpc_rqst *rqst) 524 { 525 struct kvec *iov = buf->head; 526 int scratch_len = buf->buflen - buf->page_len - buf->tail[0].iov_len; 527 528 xdr_set_scratch_buffer(xdr, NULL, 0); 529 BUG_ON(scratch_len < 0); 530 xdr->buf = buf; 531 xdr->iov = iov; 532 xdr->p = (__be32 *)((char *)iov->iov_base + iov->iov_len); 533 xdr->end = (__be32 *)((char *)iov->iov_base + scratch_len); 534 BUG_ON(iov->iov_len > scratch_len); 535 536 if (p != xdr->p && p != NULL) { 537 size_t len; 538 539 BUG_ON(p < xdr->p || p > xdr->end); 540 len = (char *)p - (char *)xdr->p; 541 xdr->p = p; 542 buf->len += len; 543 iov->iov_len += len; 544 } 545 xdr->rqst = rqst; 546 } 547 EXPORT_SYMBOL_GPL(xdr_init_encode); 548 549 /** 550 * xdr_commit_encode - Ensure all data is written to buffer 551 * @xdr: pointer to xdr_stream 552 * 553 * We handle encoding across page boundaries by giving the caller a 554 * temporary location to write to, then later copying the data into 555 * place; xdr_commit_encode does that copying. 556 * 557 * Normally the caller doesn't need to call this directly, as the 558 * following xdr_reserve_space will do it. But an explicit call may be 559 * required at the end of encoding, or any other time when the xdr_buf 560 * data might be read. 561 */ 562 void xdr_commit_encode(struct xdr_stream *xdr) 563 { 564 int shift = xdr->scratch.iov_len; 565 void *page; 566 567 if (shift == 0) 568 return; 569 page = page_address(*xdr->page_ptr); 570 memcpy(xdr->scratch.iov_base, page, shift); 571 memmove(page, page + shift, (void *)xdr->p - page); 572 xdr->scratch.iov_len = 0; 573 } 574 EXPORT_SYMBOL_GPL(xdr_commit_encode); 575 576 static __be32 *xdr_get_next_encode_buffer(struct xdr_stream *xdr, 577 size_t nbytes) 578 { 579 __be32 *p; 580 int space_left; 581 int frag1bytes, frag2bytes; 582 583 if (nbytes > PAGE_SIZE) 584 goto out_overflow; /* Bigger buffers require special handling */ 585 if (xdr->buf->len + nbytes > xdr->buf->buflen) 586 goto out_overflow; /* Sorry, we're totally out of space */ 587 frag1bytes = (xdr->end - xdr->p) << 2; 588 frag2bytes = nbytes - frag1bytes; 589 if (xdr->iov) 590 xdr->iov->iov_len += frag1bytes; 591 else 592 xdr->buf->page_len += frag1bytes; 593 xdr->page_ptr++; 594 xdr->iov = NULL; 595 /* 596 * If the last encode didn't end exactly on a page boundary, the 597 * next one will straddle boundaries. Encode into the next 598 * page, then copy it back later in xdr_commit_encode. We use 599 * the "scratch" iov to track any temporarily unused fragment of 600 * space at the end of the previous buffer: 601 */ 602 xdr->scratch.iov_base = xdr->p; 603 xdr->scratch.iov_len = frag1bytes; 604 p = page_address(*xdr->page_ptr); 605 /* 606 * Note this is where the next encode will start after we've 607 * shifted this one back: 608 */ 609 xdr->p = (void *)p + frag2bytes; 610 space_left = xdr->buf->buflen - xdr->buf->len; 611 xdr->end = (void *)p + min_t(int, space_left, PAGE_SIZE); 612 xdr->buf->page_len += frag2bytes; 613 xdr->buf->len += nbytes; 614 return p; 615 out_overflow: 616 trace_rpc_xdr_overflow(xdr, nbytes); 617 return NULL; 618 } 619 620 /** 621 * xdr_reserve_space - Reserve buffer space for sending 622 * @xdr: pointer to xdr_stream 623 * @nbytes: number of bytes to reserve 624 * 625 * Checks that we have enough buffer space to encode 'nbytes' more 626 * bytes of data. If so, update the total xdr_buf length, and 627 * adjust the length of the current kvec. 628 */ 629 __be32 * xdr_reserve_space(struct xdr_stream *xdr, size_t nbytes) 630 { 631 __be32 *p = xdr->p; 632 __be32 *q; 633 634 xdr_commit_encode(xdr); 635 /* align nbytes on the next 32-bit boundary */ 636 nbytes += 3; 637 nbytes &= ~3; 638 q = p + (nbytes >> 2); 639 if (unlikely(q > xdr->end || q < p)) 640 return xdr_get_next_encode_buffer(xdr, nbytes); 641 xdr->p = q; 642 if (xdr->iov) 643 xdr->iov->iov_len += nbytes; 644 else 645 xdr->buf->page_len += nbytes; 646 xdr->buf->len += nbytes; 647 return p; 648 } 649 EXPORT_SYMBOL_GPL(xdr_reserve_space); 650 651 /** 652 * xdr_truncate_encode - truncate an encode buffer 653 * @xdr: pointer to xdr_stream 654 * @len: new length of buffer 655 * 656 * Truncates the xdr stream, so that xdr->buf->len == len, 657 * and xdr->p points at offset len from the start of the buffer, and 658 * head, tail, and page lengths are adjusted to correspond. 659 * 660 * If this means moving xdr->p to a different buffer, we assume that 661 * that the end pointer should be set to the end of the current page, 662 * except in the case of the head buffer when we assume the head 663 * buffer's current length represents the end of the available buffer. 664 * 665 * This is *not* safe to use on a buffer that already has inlined page 666 * cache pages (as in a zero-copy server read reply), except for the 667 * simple case of truncating from one position in the tail to another. 668 * 669 */ 670 void xdr_truncate_encode(struct xdr_stream *xdr, size_t len) 671 { 672 struct xdr_buf *buf = xdr->buf; 673 struct kvec *head = buf->head; 674 struct kvec *tail = buf->tail; 675 int fraglen; 676 int new; 677 678 if (len > buf->len) { 679 WARN_ON_ONCE(1); 680 return; 681 } 682 xdr_commit_encode(xdr); 683 684 fraglen = min_t(int, buf->len - len, tail->iov_len); 685 tail->iov_len -= fraglen; 686 buf->len -= fraglen; 687 if (tail->iov_len) { 688 xdr->p = tail->iov_base + tail->iov_len; 689 WARN_ON_ONCE(!xdr->end); 690 WARN_ON_ONCE(!xdr->iov); 691 return; 692 } 693 WARN_ON_ONCE(fraglen); 694 fraglen = min_t(int, buf->len - len, buf->page_len); 695 buf->page_len -= fraglen; 696 buf->len -= fraglen; 697 698 new = buf->page_base + buf->page_len; 699 700 xdr->page_ptr = buf->pages + (new >> PAGE_SHIFT); 701 702 if (buf->page_len) { 703 xdr->p = page_address(*xdr->page_ptr); 704 xdr->end = (void *)xdr->p + PAGE_SIZE; 705 xdr->p = (void *)xdr->p + (new % PAGE_SIZE); 706 WARN_ON_ONCE(xdr->iov); 707 return; 708 } 709 if (fraglen) 710 xdr->end = head->iov_base + head->iov_len; 711 /* (otherwise assume xdr->end is already set) */ 712 xdr->page_ptr--; 713 head->iov_len = len; 714 buf->len = len; 715 xdr->p = head->iov_base + head->iov_len; 716 xdr->iov = buf->head; 717 } 718 EXPORT_SYMBOL(xdr_truncate_encode); 719 720 /** 721 * xdr_restrict_buflen - decrease available buffer space 722 * @xdr: pointer to xdr_stream 723 * @newbuflen: new maximum number of bytes available 724 * 725 * Adjust our idea of how much space is available in the buffer. 726 * If we've already used too much space in the buffer, returns -1. 727 * If the available space is already smaller than newbuflen, returns 0 728 * and does nothing. Otherwise, adjusts xdr->buf->buflen to newbuflen 729 * and ensures xdr->end is set at most offset newbuflen from the start 730 * of the buffer. 731 */ 732 int xdr_restrict_buflen(struct xdr_stream *xdr, int newbuflen) 733 { 734 struct xdr_buf *buf = xdr->buf; 735 int left_in_this_buf = (void *)xdr->end - (void *)xdr->p; 736 int end_offset = buf->len + left_in_this_buf; 737 738 if (newbuflen < 0 || newbuflen < buf->len) 739 return -1; 740 if (newbuflen > buf->buflen) 741 return 0; 742 if (newbuflen < end_offset) 743 xdr->end = (void *)xdr->end + newbuflen - end_offset; 744 buf->buflen = newbuflen; 745 return 0; 746 } 747 EXPORT_SYMBOL(xdr_restrict_buflen); 748 749 /** 750 * xdr_write_pages - Insert a list of pages into an XDR buffer for sending 751 * @xdr: pointer to xdr_stream 752 * @pages: list of pages 753 * @base: offset of first byte 754 * @len: length of data in bytes 755 * 756 */ 757 void xdr_write_pages(struct xdr_stream *xdr, struct page **pages, unsigned int base, 758 unsigned int len) 759 { 760 struct xdr_buf *buf = xdr->buf; 761 struct kvec *iov = buf->tail; 762 buf->pages = pages; 763 buf->page_base = base; 764 buf->page_len = len; 765 766 iov->iov_base = (char *)xdr->p; 767 iov->iov_len = 0; 768 xdr->iov = iov; 769 770 if (len & 3) { 771 unsigned int pad = 4 - (len & 3); 772 773 BUG_ON(xdr->p >= xdr->end); 774 iov->iov_base = (char *)xdr->p + (len & 3); 775 iov->iov_len += pad; 776 len += pad; 777 *xdr->p++ = 0; 778 } 779 buf->buflen += len; 780 buf->len += len; 781 } 782 EXPORT_SYMBOL_GPL(xdr_write_pages); 783 784 static void xdr_set_iov(struct xdr_stream *xdr, struct kvec *iov, 785 unsigned int len) 786 { 787 if (len > iov->iov_len) 788 len = iov->iov_len; 789 xdr->p = (__be32*)iov->iov_base; 790 xdr->end = (__be32*)(iov->iov_base + len); 791 xdr->iov = iov; 792 xdr->page_ptr = NULL; 793 } 794 795 static int xdr_set_page_base(struct xdr_stream *xdr, 796 unsigned int base, unsigned int len) 797 { 798 unsigned int pgnr; 799 unsigned int maxlen; 800 unsigned int pgoff; 801 unsigned int pgend; 802 void *kaddr; 803 804 maxlen = xdr->buf->page_len; 805 if (base >= maxlen) 806 return -EINVAL; 807 maxlen -= base; 808 if (len > maxlen) 809 len = maxlen; 810 811 base += xdr->buf->page_base; 812 813 pgnr = base >> PAGE_SHIFT; 814 xdr->page_ptr = &xdr->buf->pages[pgnr]; 815 kaddr = page_address(*xdr->page_ptr); 816 817 pgoff = base & ~PAGE_MASK; 818 xdr->p = (__be32*)(kaddr + pgoff); 819 820 pgend = pgoff + len; 821 if (pgend > PAGE_SIZE) 822 pgend = PAGE_SIZE; 823 xdr->end = (__be32*)(kaddr + pgend); 824 xdr->iov = NULL; 825 return 0; 826 } 827 828 static void xdr_set_next_page(struct xdr_stream *xdr) 829 { 830 unsigned int newbase; 831 832 newbase = (1 + xdr->page_ptr - xdr->buf->pages) << PAGE_SHIFT; 833 newbase -= xdr->buf->page_base; 834 835 if (xdr_set_page_base(xdr, newbase, PAGE_SIZE) < 0) 836 xdr_set_iov(xdr, xdr->buf->tail, xdr->nwords << 2); 837 } 838 839 static bool xdr_set_next_buffer(struct xdr_stream *xdr) 840 { 841 if (xdr->page_ptr != NULL) 842 xdr_set_next_page(xdr); 843 else if (xdr->iov == xdr->buf->head) { 844 if (xdr_set_page_base(xdr, 0, PAGE_SIZE) < 0) 845 xdr_set_iov(xdr, xdr->buf->tail, xdr->nwords << 2); 846 } 847 return xdr->p != xdr->end; 848 } 849 850 /** 851 * xdr_init_decode - Initialize an xdr_stream for decoding data. 852 * @xdr: pointer to xdr_stream struct 853 * @buf: pointer to XDR buffer from which to decode data 854 * @p: current pointer inside XDR buffer 855 * @rqst: pointer to controlling rpc_rqst, for debugging 856 */ 857 void xdr_init_decode(struct xdr_stream *xdr, struct xdr_buf *buf, __be32 *p, 858 struct rpc_rqst *rqst) 859 { 860 xdr->buf = buf; 861 xdr->scratch.iov_base = NULL; 862 xdr->scratch.iov_len = 0; 863 xdr->nwords = XDR_QUADLEN(buf->len); 864 if (buf->head[0].iov_len != 0) 865 xdr_set_iov(xdr, buf->head, buf->len); 866 else if (buf->page_len != 0) 867 xdr_set_page_base(xdr, 0, buf->len); 868 else 869 xdr_set_iov(xdr, buf->head, buf->len); 870 if (p != NULL && p > xdr->p && xdr->end >= p) { 871 xdr->nwords -= p - xdr->p; 872 xdr->p = p; 873 } 874 xdr->rqst = rqst; 875 } 876 EXPORT_SYMBOL_GPL(xdr_init_decode); 877 878 /** 879 * xdr_init_decode_pages - Initialize an xdr_stream for decoding into pages 880 * @xdr: pointer to xdr_stream struct 881 * @buf: pointer to XDR buffer from which to decode data 882 * @pages: list of pages to decode into 883 * @len: length in bytes of buffer in pages 884 */ 885 void xdr_init_decode_pages(struct xdr_stream *xdr, struct xdr_buf *buf, 886 struct page **pages, unsigned int len) 887 { 888 memset(buf, 0, sizeof(*buf)); 889 buf->pages = pages; 890 buf->page_len = len; 891 buf->buflen = len; 892 buf->len = len; 893 xdr_init_decode(xdr, buf, NULL, NULL); 894 } 895 EXPORT_SYMBOL_GPL(xdr_init_decode_pages); 896 897 static __be32 * __xdr_inline_decode(struct xdr_stream *xdr, size_t nbytes) 898 { 899 unsigned int nwords = XDR_QUADLEN(nbytes); 900 __be32 *p = xdr->p; 901 __be32 *q = p + nwords; 902 903 if (unlikely(nwords > xdr->nwords || q > xdr->end || q < p)) 904 return NULL; 905 xdr->p = q; 906 xdr->nwords -= nwords; 907 return p; 908 } 909 910 /** 911 * xdr_set_scratch_buffer - Attach a scratch buffer for decoding data. 912 * @xdr: pointer to xdr_stream struct 913 * @buf: pointer to an empty buffer 914 * @buflen: size of 'buf' 915 * 916 * The scratch buffer is used when decoding from an array of pages. 917 * If an xdr_inline_decode() call spans across page boundaries, then 918 * we copy the data into the scratch buffer in order to allow linear 919 * access. 920 */ 921 void xdr_set_scratch_buffer(struct xdr_stream *xdr, void *buf, size_t buflen) 922 { 923 xdr->scratch.iov_base = buf; 924 xdr->scratch.iov_len = buflen; 925 } 926 EXPORT_SYMBOL_GPL(xdr_set_scratch_buffer); 927 928 static __be32 *xdr_copy_to_scratch(struct xdr_stream *xdr, size_t nbytes) 929 { 930 __be32 *p; 931 char *cpdest = xdr->scratch.iov_base; 932 size_t cplen = (char *)xdr->end - (char *)xdr->p; 933 934 if (nbytes > xdr->scratch.iov_len) 935 goto out_overflow; 936 p = __xdr_inline_decode(xdr, cplen); 937 if (p == NULL) 938 return NULL; 939 memcpy(cpdest, p, cplen); 940 if (!xdr_set_next_buffer(xdr)) 941 goto out_overflow; 942 cpdest += cplen; 943 nbytes -= cplen; 944 p = __xdr_inline_decode(xdr, nbytes); 945 if (p == NULL) 946 return NULL; 947 memcpy(cpdest, p, nbytes); 948 return xdr->scratch.iov_base; 949 out_overflow: 950 trace_rpc_xdr_overflow(xdr, nbytes); 951 return NULL; 952 } 953 954 /** 955 * xdr_inline_decode - Retrieve XDR data to decode 956 * @xdr: pointer to xdr_stream struct 957 * @nbytes: number of bytes of data to decode 958 * 959 * Check if the input buffer is long enough to enable us to decode 960 * 'nbytes' more bytes of data starting at the current position. 961 * If so return the current pointer, then update the current 962 * pointer position. 963 */ 964 __be32 * xdr_inline_decode(struct xdr_stream *xdr, size_t nbytes) 965 { 966 __be32 *p; 967 968 if (unlikely(nbytes == 0)) 969 return xdr->p; 970 if (xdr->p == xdr->end && !xdr_set_next_buffer(xdr)) 971 goto out_overflow; 972 p = __xdr_inline_decode(xdr, nbytes); 973 if (p != NULL) 974 return p; 975 return xdr_copy_to_scratch(xdr, nbytes); 976 out_overflow: 977 trace_rpc_xdr_overflow(xdr, nbytes); 978 return NULL; 979 } 980 EXPORT_SYMBOL_GPL(xdr_inline_decode); 981 982 static unsigned int xdr_align_pages(struct xdr_stream *xdr, unsigned int len) 983 { 984 struct xdr_buf *buf = xdr->buf; 985 struct kvec *iov; 986 unsigned int nwords = XDR_QUADLEN(len); 987 unsigned int cur = xdr_stream_pos(xdr); 988 unsigned int copied, offset; 989 990 if (xdr->nwords == 0) 991 return 0; 992 993 /* Realign pages to current pointer position */ 994 iov = buf->head; 995 if (iov->iov_len > cur) { 996 offset = iov->iov_len - cur; 997 copied = xdr_shrink_bufhead(buf, offset); 998 trace_rpc_xdr_alignment(xdr, offset, copied); 999 xdr->nwords = XDR_QUADLEN(buf->len - cur); 1000 } 1001 1002 if (nwords > xdr->nwords) { 1003 nwords = xdr->nwords; 1004 len = nwords << 2; 1005 } 1006 if (buf->page_len <= len) 1007 len = buf->page_len; 1008 else if (nwords < xdr->nwords) { 1009 /* Truncate page data and move it into the tail */ 1010 offset = buf->page_len - len; 1011 copied = xdr_shrink_pagelen(buf, offset); 1012 trace_rpc_xdr_alignment(xdr, offset, copied); 1013 xdr->nwords = XDR_QUADLEN(buf->len - cur); 1014 } 1015 return len; 1016 } 1017 1018 /** 1019 * xdr_read_pages - Ensure page-based XDR data to decode is aligned at current pointer position 1020 * @xdr: pointer to xdr_stream struct 1021 * @len: number of bytes of page data 1022 * 1023 * Moves data beyond the current pointer position from the XDR head[] buffer 1024 * into the page list. Any data that lies beyond current position + "len" 1025 * bytes is moved into the XDR tail[]. 1026 * 1027 * Returns the number of XDR encoded bytes now contained in the pages 1028 */ 1029 unsigned int xdr_read_pages(struct xdr_stream *xdr, unsigned int len) 1030 { 1031 struct xdr_buf *buf = xdr->buf; 1032 struct kvec *iov; 1033 unsigned int nwords; 1034 unsigned int end; 1035 unsigned int padding; 1036 1037 len = xdr_align_pages(xdr, len); 1038 if (len == 0) 1039 return 0; 1040 nwords = XDR_QUADLEN(len); 1041 padding = (nwords << 2) - len; 1042 xdr->iov = iov = buf->tail; 1043 /* Compute remaining message length. */ 1044 end = ((xdr->nwords - nwords) << 2) + padding; 1045 if (end > iov->iov_len) 1046 end = iov->iov_len; 1047 1048 /* 1049 * Position current pointer at beginning of tail, and 1050 * set remaining message length. 1051 */ 1052 xdr->p = (__be32 *)((char *)iov->iov_base + padding); 1053 xdr->end = (__be32 *)((char *)iov->iov_base + end); 1054 xdr->page_ptr = NULL; 1055 xdr->nwords = XDR_QUADLEN(end - padding); 1056 return len; 1057 } 1058 EXPORT_SYMBOL_GPL(xdr_read_pages); 1059 1060 /** 1061 * xdr_enter_page - decode data from the XDR page 1062 * @xdr: pointer to xdr_stream struct 1063 * @len: number of bytes of page data 1064 * 1065 * Moves data beyond the current pointer position from the XDR head[] buffer 1066 * into the page list. Any data that lies beyond current position + "len" 1067 * bytes is moved into the XDR tail[]. The current pointer is then 1068 * repositioned at the beginning of the first XDR page. 1069 */ 1070 void xdr_enter_page(struct xdr_stream *xdr, unsigned int len) 1071 { 1072 len = xdr_align_pages(xdr, len); 1073 /* 1074 * Position current pointer at beginning of tail, and 1075 * set remaining message length. 1076 */ 1077 if (len != 0) 1078 xdr_set_page_base(xdr, 0, len); 1079 } 1080 EXPORT_SYMBOL_GPL(xdr_enter_page); 1081 1082 static struct kvec empty_iov = {.iov_base = NULL, .iov_len = 0}; 1083 1084 void 1085 xdr_buf_from_iov(struct kvec *iov, struct xdr_buf *buf) 1086 { 1087 buf->head[0] = *iov; 1088 buf->tail[0] = empty_iov; 1089 buf->page_len = 0; 1090 buf->buflen = buf->len = iov->iov_len; 1091 } 1092 EXPORT_SYMBOL_GPL(xdr_buf_from_iov); 1093 1094 /** 1095 * xdr_buf_subsegment - set subbuf to a portion of buf 1096 * @buf: an xdr buffer 1097 * @subbuf: the result buffer 1098 * @base: beginning of range in bytes 1099 * @len: length of range in bytes 1100 * 1101 * sets @subbuf to an xdr buffer representing the portion of @buf of 1102 * length @len starting at offset @base. 1103 * 1104 * @buf and @subbuf may be pointers to the same struct xdr_buf. 1105 * 1106 * Returns -1 if base of length are out of bounds. 1107 */ 1108 int 1109 xdr_buf_subsegment(struct xdr_buf *buf, struct xdr_buf *subbuf, 1110 unsigned int base, unsigned int len) 1111 { 1112 subbuf->buflen = subbuf->len = len; 1113 if (base < buf->head[0].iov_len) { 1114 subbuf->head[0].iov_base = buf->head[0].iov_base + base; 1115 subbuf->head[0].iov_len = min_t(unsigned int, len, 1116 buf->head[0].iov_len - base); 1117 len -= subbuf->head[0].iov_len; 1118 base = 0; 1119 } else { 1120 base -= buf->head[0].iov_len; 1121 subbuf->head[0].iov_len = 0; 1122 } 1123 1124 if (base < buf->page_len) { 1125 subbuf->page_len = min(buf->page_len - base, len); 1126 base += buf->page_base; 1127 subbuf->page_base = base & ~PAGE_MASK; 1128 subbuf->pages = &buf->pages[base >> PAGE_SHIFT]; 1129 len -= subbuf->page_len; 1130 base = 0; 1131 } else { 1132 base -= buf->page_len; 1133 subbuf->page_len = 0; 1134 } 1135 1136 if (base < buf->tail[0].iov_len) { 1137 subbuf->tail[0].iov_base = buf->tail[0].iov_base + base; 1138 subbuf->tail[0].iov_len = min_t(unsigned int, len, 1139 buf->tail[0].iov_len - base); 1140 len -= subbuf->tail[0].iov_len; 1141 base = 0; 1142 } else { 1143 base -= buf->tail[0].iov_len; 1144 subbuf->tail[0].iov_len = 0; 1145 } 1146 1147 if (base || len) 1148 return -1; 1149 return 0; 1150 } 1151 EXPORT_SYMBOL_GPL(xdr_buf_subsegment); 1152 1153 static void __read_bytes_from_xdr_buf(struct xdr_buf *subbuf, void *obj, unsigned int len) 1154 { 1155 unsigned int this_len; 1156 1157 this_len = min_t(unsigned int, len, subbuf->head[0].iov_len); 1158 memcpy(obj, subbuf->head[0].iov_base, this_len); 1159 len -= this_len; 1160 obj += this_len; 1161 this_len = min_t(unsigned int, len, subbuf->page_len); 1162 if (this_len) 1163 _copy_from_pages(obj, subbuf->pages, subbuf->page_base, this_len); 1164 len -= this_len; 1165 obj += this_len; 1166 this_len = min_t(unsigned int, len, subbuf->tail[0].iov_len); 1167 memcpy(obj, subbuf->tail[0].iov_base, this_len); 1168 } 1169 1170 /* obj is assumed to point to allocated memory of size at least len: */ 1171 int read_bytes_from_xdr_buf(struct xdr_buf *buf, unsigned int base, void *obj, unsigned int len) 1172 { 1173 struct xdr_buf subbuf; 1174 int status; 1175 1176 status = xdr_buf_subsegment(buf, &subbuf, base, len); 1177 if (status != 0) 1178 return status; 1179 __read_bytes_from_xdr_buf(&subbuf, obj, len); 1180 return 0; 1181 } 1182 EXPORT_SYMBOL_GPL(read_bytes_from_xdr_buf); 1183 1184 static void __write_bytes_to_xdr_buf(struct xdr_buf *subbuf, void *obj, unsigned int len) 1185 { 1186 unsigned int this_len; 1187 1188 this_len = min_t(unsigned int, len, subbuf->head[0].iov_len); 1189 memcpy(subbuf->head[0].iov_base, obj, this_len); 1190 len -= this_len; 1191 obj += this_len; 1192 this_len = min_t(unsigned int, len, subbuf->page_len); 1193 if (this_len) 1194 _copy_to_pages(subbuf->pages, subbuf->page_base, obj, this_len); 1195 len -= this_len; 1196 obj += this_len; 1197 this_len = min_t(unsigned int, len, subbuf->tail[0].iov_len); 1198 memcpy(subbuf->tail[0].iov_base, obj, this_len); 1199 } 1200 1201 /* obj is assumed to point to allocated memory of size at least len: */ 1202 int write_bytes_to_xdr_buf(struct xdr_buf *buf, unsigned int base, void *obj, unsigned int len) 1203 { 1204 struct xdr_buf subbuf; 1205 int status; 1206 1207 status = xdr_buf_subsegment(buf, &subbuf, base, len); 1208 if (status != 0) 1209 return status; 1210 __write_bytes_to_xdr_buf(&subbuf, obj, len); 1211 return 0; 1212 } 1213 EXPORT_SYMBOL_GPL(write_bytes_to_xdr_buf); 1214 1215 int 1216 xdr_decode_word(struct xdr_buf *buf, unsigned int base, u32 *obj) 1217 { 1218 __be32 raw; 1219 int status; 1220 1221 status = read_bytes_from_xdr_buf(buf, base, &raw, sizeof(*obj)); 1222 if (status) 1223 return status; 1224 *obj = be32_to_cpu(raw); 1225 return 0; 1226 } 1227 EXPORT_SYMBOL_GPL(xdr_decode_word); 1228 1229 int 1230 xdr_encode_word(struct xdr_buf *buf, unsigned int base, u32 obj) 1231 { 1232 __be32 raw = cpu_to_be32(obj); 1233 1234 return write_bytes_to_xdr_buf(buf, base, &raw, sizeof(obj)); 1235 } 1236 EXPORT_SYMBOL_GPL(xdr_encode_word); 1237 1238 /* If the netobj starting offset bytes from the start of xdr_buf is contained 1239 * entirely in the head or the tail, set object to point to it; otherwise 1240 * try to find space for it at the end of the tail, copy it there, and 1241 * set obj to point to it. */ 1242 int xdr_buf_read_netobj(struct xdr_buf *buf, struct xdr_netobj *obj, unsigned int offset) 1243 { 1244 struct xdr_buf subbuf; 1245 1246 if (xdr_decode_word(buf, offset, &obj->len)) 1247 return -EFAULT; 1248 if (xdr_buf_subsegment(buf, &subbuf, offset + 4, obj->len)) 1249 return -EFAULT; 1250 1251 /* Is the obj contained entirely in the head? */ 1252 obj->data = subbuf.head[0].iov_base; 1253 if (subbuf.head[0].iov_len == obj->len) 1254 return 0; 1255 /* ..or is the obj contained entirely in the tail? */ 1256 obj->data = subbuf.tail[0].iov_base; 1257 if (subbuf.tail[0].iov_len == obj->len) 1258 return 0; 1259 1260 /* use end of tail as storage for obj: 1261 * (We don't copy to the beginning because then we'd have 1262 * to worry about doing a potentially overlapping copy. 1263 * This assumes the object is at most half the length of the 1264 * tail.) */ 1265 if (obj->len > buf->buflen - buf->len) 1266 return -ENOMEM; 1267 if (buf->tail[0].iov_len != 0) 1268 obj->data = buf->tail[0].iov_base + buf->tail[0].iov_len; 1269 else 1270 obj->data = buf->head[0].iov_base + buf->head[0].iov_len; 1271 __read_bytes_from_xdr_buf(&subbuf, obj->data, obj->len); 1272 return 0; 1273 } 1274 EXPORT_SYMBOL_GPL(xdr_buf_read_netobj); 1275 1276 /* Returns 0 on success, or else a negative error code. */ 1277 static int 1278 xdr_xcode_array2(struct xdr_buf *buf, unsigned int base, 1279 struct xdr_array2_desc *desc, int encode) 1280 { 1281 char *elem = NULL, *c; 1282 unsigned int copied = 0, todo, avail_here; 1283 struct page **ppages = NULL; 1284 int err; 1285 1286 if (encode) { 1287 if (xdr_encode_word(buf, base, desc->array_len) != 0) 1288 return -EINVAL; 1289 } else { 1290 if (xdr_decode_word(buf, base, &desc->array_len) != 0 || 1291 desc->array_len > desc->array_maxlen || 1292 (unsigned long) base + 4 + desc->array_len * 1293 desc->elem_size > buf->len) 1294 return -EINVAL; 1295 } 1296 base += 4; 1297 1298 if (!desc->xcode) 1299 return 0; 1300 1301 todo = desc->array_len * desc->elem_size; 1302 1303 /* process head */ 1304 if (todo && base < buf->head->iov_len) { 1305 c = buf->head->iov_base + base; 1306 avail_here = min_t(unsigned int, todo, 1307 buf->head->iov_len - base); 1308 todo -= avail_here; 1309 1310 while (avail_here >= desc->elem_size) { 1311 err = desc->xcode(desc, c); 1312 if (err) 1313 goto out; 1314 c += desc->elem_size; 1315 avail_here -= desc->elem_size; 1316 } 1317 if (avail_here) { 1318 if (!elem) { 1319 elem = kmalloc(desc->elem_size, GFP_KERNEL); 1320 err = -ENOMEM; 1321 if (!elem) 1322 goto out; 1323 } 1324 if (encode) { 1325 err = desc->xcode(desc, elem); 1326 if (err) 1327 goto out; 1328 memcpy(c, elem, avail_here); 1329 } else 1330 memcpy(elem, c, avail_here); 1331 copied = avail_here; 1332 } 1333 base = buf->head->iov_len; /* align to start of pages */ 1334 } 1335 1336 /* process pages array */ 1337 base -= buf->head->iov_len; 1338 if (todo && base < buf->page_len) { 1339 unsigned int avail_page; 1340 1341 avail_here = min(todo, buf->page_len - base); 1342 todo -= avail_here; 1343 1344 base += buf->page_base; 1345 ppages = buf->pages + (base >> PAGE_SHIFT); 1346 base &= ~PAGE_MASK; 1347 avail_page = min_t(unsigned int, PAGE_SIZE - base, 1348 avail_here); 1349 c = kmap(*ppages) + base; 1350 1351 while (avail_here) { 1352 avail_here -= avail_page; 1353 if (copied || avail_page < desc->elem_size) { 1354 unsigned int l = min(avail_page, 1355 desc->elem_size - copied); 1356 if (!elem) { 1357 elem = kmalloc(desc->elem_size, 1358 GFP_KERNEL); 1359 err = -ENOMEM; 1360 if (!elem) 1361 goto out; 1362 } 1363 if (encode) { 1364 if (!copied) { 1365 err = desc->xcode(desc, elem); 1366 if (err) 1367 goto out; 1368 } 1369 memcpy(c, elem + copied, l); 1370 copied += l; 1371 if (copied == desc->elem_size) 1372 copied = 0; 1373 } else { 1374 memcpy(elem + copied, c, l); 1375 copied += l; 1376 if (copied == desc->elem_size) { 1377 err = desc->xcode(desc, elem); 1378 if (err) 1379 goto out; 1380 copied = 0; 1381 } 1382 } 1383 avail_page -= l; 1384 c += l; 1385 } 1386 while (avail_page >= desc->elem_size) { 1387 err = desc->xcode(desc, c); 1388 if (err) 1389 goto out; 1390 c += desc->elem_size; 1391 avail_page -= desc->elem_size; 1392 } 1393 if (avail_page) { 1394 unsigned int l = min(avail_page, 1395 desc->elem_size - copied); 1396 if (!elem) { 1397 elem = kmalloc(desc->elem_size, 1398 GFP_KERNEL); 1399 err = -ENOMEM; 1400 if (!elem) 1401 goto out; 1402 } 1403 if (encode) { 1404 if (!copied) { 1405 err = desc->xcode(desc, elem); 1406 if (err) 1407 goto out; 1408 } 1409 memcpy(c, elem + copied, l); 1410 copied += l; 1411 if (copied == desc->elem_size) 1412 copied = 0; 1413 } else { 1414 memcpy(elem + copied, c, l); 1415 copied += l; 1416 if (copied == desc->elem_size) { 1417 err = desc->xcode(desc, elem); 1418 if (err) 1419 goto out; 1420 copied = 0; 1421 } 1422 } 1423 } 1424 if (avail_here) { 1425 kunmap(*ppages); 1426 ppages++; 1427 c = kmap(*ppages); 1428 } 1429 1430 avail_page = min(avail_here, 1431 (unsigned int) PAGE_SIZE); 1432 } 1433 base = buf->page_len; /* align to start of tail */ 1434 } 1435 1436 /* process tail */ 1437 base -= buf->page_len; 1438 if (todo) { 1439 c = buf->tail->iov_base + base; 1440 if (copied) { 1441 unsigned int l = desc->elem_size - copied; 1442 1443 if (encode) 1444 memcpy(c, elem + copied, l); 1445 else { 1446 memcpy(elem + copied, c, l); 1447 err = desc->xcode(desc, elem); 1448 if (err) 1449 goto out; 1450 } 1451 todo -= l; 1452 c += l; 1453 } 1454 while (todo) { 1455 err = desc->xcode(desc, c); 1456 if (err) 1457 goto out; 1458 c += desc->elem_size; 1459 todo -= desc->elem_size; 1460 } 1461 } 1462 err = 0; 1463 1464 out: 1465 kfree(elem); 1466 if (ppages) 1467 kunmap(*ppages); 1468 return err; 1469 } 1470 1471 int 1472 xdr_decode_array2(struct xdr_buf *buf, unsigned int base, 1473 struct xdr_array2_desc *desc) 1474 { 1475 if (base >= buf->len) 1476 return -EINVAL; 1477 1478 return xdr_xcode_array2(buf, base, desc, 0); 1479 } 1480 EXPORT_SYMBOL_GPL(xdr_decode_array2); 1481 1482 int 1483 xdr_encode_array2(struct xdr_buf *buf, unsigned int base, 1484 struct xdr_array2_desc *desc) 1485 { 1486 if ((unsigned long) base + 4 + desc->array_len * desc->elem_size > 1487 buf->head->iov_len + buf->page_len + buf->tail->iov_len) 1488 return -EINVAL; 1489 1490 return xdr_xcode_array2(buf, base, desc, 1); 1491 } 1492 EXPORT_SYMBOL_GPL(xdr_encode_array2); 1493 1494 int 1495 xdr_process_buf(struct xdr_buf *buf, unsigned int offset, unsigned int len, 1496 int (*actor)(struct scatterlist *, void *), void *data) 1497 { 1498 int i, ret = 0; 1499 unsigned int page_len, thislen, page_offset; 1500 struct scatterlist sg[1]; 1501 1502 sg_init_table(sg, 1); 1503 1504 if (offset >= buf->head[0].iov_len) { 1505 offset -= buf->head[0].iov_len; 1506 } else { 1507 thislen = buf->head[0].iov_len - offset; 1508 if (thislen > len) 1509 thislen = len; 1510 sg_set_buf(sg, buf->head[0].iov_base + offset, thislen); 1511 ret = actor(sg, data); 1512 if (ret) 1513 goto out; 1514 offset = 0; 1515 len -= thislen; 1516 } 1517 if (len == 0) 1518 goto out; 1519 1520 if (offset >= buf->page_len) { 1521 offset -= buf->page_len; 1522 } else { 1523 page_len = buf->page_len - offset; 1524 if (page_len > len) 1525 page_len = len; 1526 len -= page_len; 1527 page_offset = (offset + buf->page_base) & (PAGE_SIZE - 1); 1528 i = (offset + buf->page_base) >> PAGE_SHIFT; 1529 thislen = PAGE_SIZE - page_offset; 1530 do { 1531 if (thislen > page_len) 1532 thislen = page_len; 1533 sg_set_page(sg, buf->pages[i], thislen, page_offset); 1534 ret = actor(sg, data); 1535 if (ret) 1536 goto out; 1537 page_len -= thislen; 1538 i++; 1539 page_offset = 0; 1540 thislen = PAGE_SIZE; 1541 } while (page_len != 0); 1542 offset = 0; 1543 } 1544 if (len == 0) 1545 goto out; 1546 if (offset < buf->tail[0].iov_len) { 1547 thislen = buf->tail[0].iov_len - offset; 1548 if (thislen > len) 1549 thislen = len; 1550 sg_set_buf(sg, buf->tail[0].iov_base + offset, thislen); 1551 ret = actor(sg, data); 1552 len -= thislen; 1553 } 1554 if (len != 0) 1555 ret = -EINVAL; 1556 out: 1557 return ret; 1558 } 1559 EXPORT_SYMBOL_GPL(xdr_process_buf); 1560 1561 /** 1562 * xdr_stream_decode_opaque - Decode variable length opaque 1563 * @xdr: pointer to xdr_stream 1564 * @ptr: location to store opaque data 1565 * @size: size of storage buffer @ptr 1566 * 1567 * Return values: 1568 * On success, returns size of object stored in *@ptr 1569 * %-EBADMSG on XDR buffer overflow 1570 * %-EMSGSIZE on overflow of storage buffer @ptr 1571 */ 1572 ssize_t xdr_stream_decode_opaque(struct xdr_stream *xdr, void *ptr, size_t size) 1573 { 1574 ssize_t ret; 1575 void *p; 1576 1577 ret = xdr_stream_decode_opaque_inline(xdr, &p, size); 1578 if (ret <= 0) 1579 return ret; 1580 memcpy(ptr, p, ret); 1581 return ret; 1582 } 1583 EXPORT_SYMBOL_GPL(xdr_stream_decode_opaque); 1584 1585 /** 1586 * xdr_stream_decode_opaque_dup - Decode and duplicate variable length opaque 1587 * @xdr: pointer to xdr_stream 1588 * @ptr: location to store pointer to opaque data 1589 * @maxlen: maximum acceptable object size 1590 * @gfp_flags: GFP mask to use 1591 * 1592 * Return values: 1593 * On success, returns size of object stored in *@ptr 1594 * %-EBADMSG on XDR buffer overflow 1595 * %-EMSGSIZE if the size of the object would exceed @maxlen 1596 * %-ENOMEM on memory allocation failure 1597 */ 1598 ssize_t xdr_stream_decode_opaque_dup(struct xdr_stream *xdr, void **ptr, 1599 size_t maxlen, gfp_t gfp_flags) 1600 { 1601 ssize_t ret; 1602 void *p; 1603 1604 ret = xdr_stream_decode_opaque_inline(xdr, &p, maxlen); 1605 if (ret > 0) { 1606 *ptr = kmemdup(p, ret, gfp_flags); 1607 if (*ptr != NULL) 1608 return ret; 1609 ret = -ENOMEM; 1610 } 1611 *ptr = NULL; 1612 return ret; 1613 } 1614 EXPORT_SYMBOL_GPL(xdr_stream_decode_opaque_dup); 1615 1616 /** 1617 * xdr_stream_decode_string - Decode variable length string 1618 * @xdr: pointer to xdr_stream 1619 * @str: location to store string 1620 * @size: size of storage buffer @str 1621 * 1622 * Return values: 1623 * On success, returns length of NUL-terminated string stored in *@str 1624 * %-EBADMSG on XDR buffer overflow 1625 * %-EMSGSIZE on overflow of storage buffer @str 1626 */ 1627 ssize_t xdr_stream_decode_string(struct xdr_stream *xdr, char *str, size_t size) 1628 { 1629 ssize_t ret; 1630 void *p; 1631 1632 ret = xdr_stream_decode_opaque_inline(xdr, &p, size); 1633 if (ret > 0) { 1634 memcpy(str, p, ret); 1635 str[ret] = '\0'; 1636 return strlen(str); 1637 } 1638 *str = '\0'; 1639 return ret; 1640 } 1641 EXPORT_SYMBOL_GPL(xdr_stream_decode_string); 1642 1643 /** 1644 * xdr_stream_decode_string_dup - Decode and duplicate variable length string 1645 * @xdr: pointer to xdr_stream 1646 * @str: location to store pointer to string 1647 * @maxlen: maximum acceptable string length 1648 * @gfp_flags: GFP mask to use 1649 * 1650 * Return values: 1651 * On success, returns length of NUL-terminated string stored in *@ptr 1652 * %-EBADMSG on XDR buffer overflow 1653 * %-EMSGSIZE if the size of the string would exceed @maxlen 1654 * %-ENOMEM on memory allocation failure 1655 */ 1656 ssize_t xdr_stream_decode_string_dup(struct xdr_stream *xdr, char **str, 1657 size_t maxlen, gfp_t gfp_flags) 1658 { 1659 void *p; 1660 ssize_t ret; 1661 1662 ret = xdr_stream_decode_opaque_inline(xdr, &p, maxlen); 1663 if (ret > 0) { 1664 char *s = kmalloc(ret + 1, gfp_flags); 1665 if (s != NULL) { 1666 memcpy(s, p, ret); 1667 s[ret] = '\0'; 1668 *str = s; 1669 return strlen(s); 1670 } 1671 ret = -ENOMEM; 1672 } 1673 *str = NULL; 1674 return ret; 1675 } 1676 EXPORT_SYMBOL_GPL(xdr_stream_decode_string_dup); 1677