diff options
-rw-r--r-- | net/sunrpc/sched.c | 22 |
1 files changed, 15 insertions, 7 deletions
diff --git a/net/sunrpc/sched.c b/net/sunrpc/sched.c index 99014516b73c..b28a0b037f4f 100644 --- a/net/sunrpc/sched.c +++ b/net/sunrpc/sched.c | |||
@@ -736,6 +736,11 @@ static void rpc_async_schedule(struct work_struct *work) | |||
736 | __rpc_execute(container_of(work, struct rpc_task, u.tk_work)); | 736 | __rpc_execute(container_of(work, struct rpc_task, u.tk_work)); |
737 | } | 737 | } |
738 | 738 | ||
739 | struct rpc_buffer { | ||
740 | size_t len; | ||
741 | char data[]; | ||
742 | }; | ||
743 | |||
739 | /** | 744 | /** |
740 | * rpc_malloc - allocate an RPC buffer | 745 | * rpc_malloc - allocate an RPC buffer |
741 | * @task: RPC task that will use this buffer | 746 | * @task: RPC task that will use this buffer |
@@ -754,18 +759,18 @@ static void rpc_async_schedule(struct work_struct *work) | |||
754 | */ | 759 | */ |
755 | void *rpc_malloc(struct rpc_task *task, size_t size) | 760 | void *rpc_malloc(struct rpc_task *task, size_t size) |
756 | { | 761 | { |
757 | size_t *buf; | 762 | struct rpc_buffer *buf; |
758 | gfp_t gfp = RPC_IS_SWAPPER(task) ? GFP_ATOMIC : GFP_NOWAIT; | 763 | gfp_t gfp = RPC_IS_SWAPPER(task) ? GFP_ATOMIC : GFP_NOWAIT; |
759 | 764 | ||
760 | size += sizeof(size_t); | 765 | size += sizeof(struct rpc_buffer); |
761 | if (size <= RPC_BUFFER_MAXSIZE) | 766 | if (size <= RPC_BUFFER_MAXSIZE) |
762 | buf = mempool_alloc(rpc_buffer_mempool, gfp); | 767 | buf = mempool_alloc(rpc_buffer_mempool, gfp); |
763 | else | 768 | else |
764 | buf = kmalloc(size, gfp); | 769 | buf = kmalloc(size, gfp); |
765 | *buf = size; | 770 | buf->len = size; |
766 | dprintk("RPC: %5u allocated buffer of size %zu at %p\n", | 771 | dprintk("RPC: %5u allocated buffer of size %zu at %p\n", |
767 | task->tk_pid, size, buf); | 772 | task->tk_pid, size, buf); |
768 | return ++buf; | 773 | return &buf->data; |
769 | } | 774 | } |
770 | 775 | ||
771 | /** | 776 | /** |
@@ -775,15 +780,18 @@ void *rpc_malloc(struct rpc_task *task, size_t size) | |||
775 | */ | 780 | */ |
776 | void rpc_free(void *buffer) | 781 | void rpc_free(void *buffer) |
777 | { | 782 | { |
778 | size_t size, *buf = buffer; | 783 | size_t size; |
784 | struct rpc_buffer *buf; | ||
779 | 785 | ||
780 | if (!buffer) | 786 | if (!buffer) |
781 | return; | 787 | return; |
782 | size = *buf; | 788 | |
783 | buf--; | 789 | buf = container_of(buffer, struct rpc_buffer, data); |
790 | size = buf->len; | ||
784 | 791 | ||
785 | dprintk("RPC: freeing buffer of size %zu at %p\n", | 792 | dprintk("RPC: freeing buffer of size %zu at %p\n", |
786 | size, buf); | 793 | size, buf); |
794 | |||
787 | if (size <= RPC_BUFFER_MAXSIZE) | 795 | if (size <= RPC_BUFFER_MAXSIZE) |
788 | mempool_free(buf, rpc_buffer_mempool); | 796 | mempool_free(buf, rpc_buffer_mempool); |
789 | else | 797 | else |