Blame view

fs/splice.c 46.5 KB
5274f052e   Jens Axboe   [PATCH] Introduce...
1
2
3
4
5
6
7
8
9
10
11
  /*
   * "splice": joining two ropes together by interweaving their strands.
   *
   * This is the "extended pipe" functionality, where a pipe is used as
   * an arbitrary in-memory buffer. Think of a pipe as a small kernel
   * buffer that you can use to transfer data from one end to the other.
   *
   * The traditional unix read/write is extended with a "splice()" operation
   * that transfers data buffers to or from a pipe buffer.
   *
   * Named by Larry McVoy, original implementation from Linus, extended by
c2058e061   Jens Axboe   [PATCH] splice: a...
12
13
   * Jens to support splicing to files, network, direct splicing, etc and
   * fixing lots of bugs.
5274f052e   Jens Axboe   [PATCH] Introduce...
14
   *
0fe234795   Jens Axboe   [PATCH] Update ax...
15
   * Copyright (C) 2005-2006 Jens Axboe <axboe@kernel.dk>
c2058e061   Jens Axboe   [PATCH] splice: a...
16
17
   * Copyright (C) 2005-2006 Linus Torvalds <torvalds@osdl.org>
   * Copyright (C) 2006 Ingo Molnar <mingo@elte.hu>
5274f052e   Jens Axboe   [PATCH] Introduce...
18
19
20
21
22
   *
   */
  #include <linux/fs.h>
  #include <linux/file.h>
  #include <linux/pagemap.h>
d6b29d7ce   Jens Axboe   splice: divorce t...
23
  #include <linux/splice.h>
08e552c69   KAMEZAWA Hiroyuki   memcg: synchroniz...
24
  #include <linux/memcontrol.h>
5274f052e   Jens Axboe   [PATCH] Introduce...
25
  #include <linux/mm_inline.h>
5abc97aa2   Jens Axboe   [PATCH] splice: a...
26
  #include <linux/swap.h>
4f6f0bd2f   Jens Axboe   [PATCH] splice: i...
27
28
  #include <linux/writeback.h>
  #include <linux/buffer_head.h>
a0f067802   Jeff Garzik   [PATCH] splice ex...
29
  #include <linux/module.h>
4f6f0bd2f   Jens Axboe   [PATCH] splice: i...
30
  #include <linux/syscalls.h>
912d35f86   Jens Axboe   [PATCH] Add suppo...
31
  #include <linux/uio.h>
29ce20586   James Morris   security: revalid...
32
  #include <linux/security.h>
5a0e3ad6a   Tejun Heo   include cleanup: ...
33
  #include <linux/gfp.h>
5274f052e   Jens Axboe   [PATCH] Introduce...
34

83f9135bd   Jens Axboe   [PATCH] splice: a...
35
36
37
38
39
40
  /*
   * Attempt to steal a page from a pipe buffer. This should perhaps go into
   * a vm helper function, it's already simplified quite a bit by the
   * addition of remove_mapping(). If success is returned, the caller may
   * attempt to reuse this page for another destination.
   */
76ad4d111   Jens Axboe   [PATCH] splice: r...
41
  static int page_cache_pipe_buf_steal(struct pipe_inode_info *pipe,
5abc97aa2   Jens Axboe   [PATCH] splice: a...
42
43
44
  				     struct pipe_buffer *buf)
  {
  	struct page *page = buf->page;
9e94cd4fd   Jens Axboe   [PATCH] splice: r...
45
  	struct address_space *mapping;
5abc97aa2   Jens Axboe   [PATCH] splice: a...
46

9e0267c26   Jens Axboe   [PATCH] splice: f...
47
  	lock_page(page);
9e94cd4fd   Jens Axboe   [PATCH] splice: r...
48
49
50
  	mapping = page_mapping(page);
  	if (mapping) {
  		WARN_ON(!PageUptodate(page));
5abc97aa2   Jens Axboe   [PATCH] splice: a...
51

9e94cd4fd   Jens Axboe   [PATCH] splice: r...
52
53
54
55
56
57
58
59
60
  		/*
  		 * At least for ext2 with nobh option, we need to wait on
  		 * writeback completing on this page, since we'll remove it
  		 * from the pagecache.  Otherwise truncate wont wait on the
  		 * page, allowing the disk blocks to be reused by someone else
  		 * before we actually wrote our data to them. fs corruption
  		 * ensues.
  		 */
  		wait_on_page_writeback(page);
ad8d6f0a7   Jens Axboe   [PATCH] splice: p...
61

266cf658e   David Howells   FS-Cache: Recruit...
62
63
  		if (page_has_private(page) &&
  		    !try_to_release_page(page, GFP_KERNEL))
ca39d651d   Jens Axboe   splice: handle tr...
64
  			goto out_unlock;
4f6f0bd2f   Jens Axboe   [PATCH] splice: i...
65

9e94cd4fd   Jens Axboe   [PATCH] splice: r...
66
67
68
69
70
71
72
73
  		/*
  		 * If we succeeded in removing the mapping, set LRU flag
  		 * and return good.
  		 */
  		if (remove_mapping(mapping, page)) {
  			buf->flags |= PIPE_BUF_FLAG_LRU;
  			return 0;
  		}
9e0267c26   Jens Axboe   [PATCH] splice: f...
74
  	}
5abc97aa2   Jens Axboe   [PATCH] splice: a...
75

9e94cd4fd   Jens Axboe   [PATCH] splice: r...
76
77
78
79
  	/*
  	 * Raced with truncate or failed to remove page from current
  	 * address space, unlock and return failure.
  	 */
ca39d651d   Jens Axboe   splice: handle tr...
80
  out_unlock:
9e94cd4fd   Jens Axboe   [PATCH] splice: r...
81
82
  	unlock_page(page);
  	return 1;
5abc97aa2   Jens Axboe   [PATCH] splice: a...
83
  }
76ad4d111   Jens Axboe   [PATCH] splice: r...
84
  static void page_cache_pipe_buf_release(struct pipe_inode_info *pipe,
5274f052e   Jens Axboe   [PATCH] Introduce...
85
86
87
  					struct pipe_buffer *buf)
  {
  	page_cache_release(buf->page);
1432873af   Jens Axboe   [PATCH] splice: L...
88
  	buf->flags &= ~PIPE_BUF_FLAG_LRU;
5274f052e   Jens Axboe   [PATCH] Introduce...
89
  }
0845718da   Jens Axboe   pipe: add documen...
90
91
92
93
  /*
   * Check whether the contents of buf is OK to access. Since the content
   * is a page cache page, IO may be in flight.
   */
cac36bb06   Jens Axboe   pipe: change the ...
94
95
  static int page_cache_pipe_buf_confirm(struct pipe_inode_info *pipe,
  				       struct pipe_buffer *buf)
5274f052e   Jens Axboe   [PATCH] Introduce...
96
97
  {
  	struct page *page = buf->page;
49d0b21be   Jens Axboe   [PATCH] splice: o...
98
  	int err;
5274f052e   Jens Axboe   [PATCH] Introduce...
99
100
  
  	if (!PageUptodate(page)) {
49d0b21be   Jens Axboe   [PATCH] splice: o...
101
102
103
104
  		lock_page(page);
  
  		/*
  		 * Page got truncated/unhashed. This will cause a 0-byte
73d62d83e   Ingo Molnar   [PATCH] splice: c...
105
  		 * splice, if this is the first page.
49d0b21be   Jens Axboe   [PATCH] splice: o...
106
107
108
109
110
  		 */
  		if (!page->mapping) {
  			err = -ENODATA;
  			goto error;
  		}
5274f052e   Jens Axboe   [PATCH] Introduce...
111

49d0b21be   Jens Axboe   [PATCH] splice: o...
112
  		/*
73d62d83e   Ingo Molnar   [PATCH] splice: c...
113
  		 * Uh oh, read-error from disk.
49d0b21be   Jens Axboe   [PATCH] splice: o...
114
115
116
117
118
119
120
  		 */
  		if (!PageUptodate(page)) {
  			err = -EIO;
  			goto error;
  		}
  
  		/*
f84d75199   Jens Axboe   [PATCH] pipe: int...
121
  		 * Page is ok afterall, we are done.
49d0b21be   Jens Axboe   [PATCH] splice: o...
122
  		 */
5274f052e   Jens Axboe   [PATCH] Introduce...
123
  		unlock_page(page);
5274f052e   Jens Axboe   [PATCH] Introduce...
124
  	}
f84d75199   Jens Axboe   [PATCH] pipe: int...
125
  	return 0;
49d0b21be   Jens Axboe   [PATCH] splice: o...
126
127
  error:
  	unlock_page(page);
f84d75199   Jens Axboe   [PATCH] pipe: int...
128
  	return err;
70524490e   Jens Axboe   [PATCH] splice: a...
129
  }
d4c3cca94   Eric Dumazet   [PATCH] constify ...
130
  static const struct pipe_buf_operations page_cache_pipe_buf_ops = {
5274f052e   Jens Axboe   [PATCH] Introduce...
131
  	.can_merge = 0,
f84d75199   Jens Axboe   [PATCH] pipe: int...
132
133
  	.map = generic_pipe_buf_map,
  	.unmap = generic_pipe_buf_unmap,
cac36bb06   Jens Axboe   pipe: change the ...
134
  	.confirm = page_cache_pipe_buf_confirm,
5274f052e   Jens Axboe   [PATCH] Introduce...
135
  	.release = page_cache_pipe_buf_release,
5abc97aa2   Jens Axboe   [PATCH] splice: a...
136
  	.steal = page_cache_pipe_buf_steal,
f84d75199   Jens Axboe   [PATCH] pipe: int...
137
  	.get = generic_pipe_buf_get,
5274f052e   Jens Axboe   [PATCH] Introduce...
138
  };
912d35f86   Jens Axboe   [PATCH] Add suppo...
139
140
141
  static int user_page_pipe_buf_steal(struct pipe_inode_info *pipe,
  				    struct pipe_buffer *buf)
  {
7afa6fd03   Jens Axboe   [PATCH] vmsplice:...
142
143
  	if (!(buf->flags & PIPE_BUF_FLAG_GIFT))
  		return 1;
1432873af   Jens Axboe   [PATCH] splice: L...
144
  	buf->flags |= PIPE_BUF_FLAG_LRU;
330ab7161   Jens Axboe   [PATCH] vmsplice:...
145
  	return generic_pipe_buf_steal(pipe, buf);
912d35f86   Jens Axboe   [PATCH] Add suppo...
146
  }
d4c3cca94   Eric Dumazet   [PATCH] constify ...
147
  static const struct pipe_buf_operations user_page_pipe_buf_ops = {
912d35f86   Jens Axboe   [PATCH] Add suppo...
148
  	.can_merge = 0,
f84d75199   Jens Axboe   [PATCH] pipe: int...
149
150
  	.map = generic_pipe_buf_map,
  	.unmap = generic_pipe_buf_unmap,
cac36bb06   Jens Axboe   pipe: change the ...
151
  	.confirm = generic_pipe_buf_confirm,
912d35f86   Jens Axboe   [PATCH] Add suppo...
152
153
  	.release = page_cache_pipe_buf_release,
  	.steal = user_page_pipe_buf_steal,
f84d75199   Jens Axboe   [PATCH] pipe: int...
154
  	.get = generic_pipe_buf_get,
912d35f86   Jens Axboe   [PATCH] Add suppo...
155
  };
932cc6d4f   Jens Axboe   splice: completel...
156
157
158
159
160
161
  /**
   * splice_to_pipe - fill passed data into a pipe
   * @pipe:	pipe to fill
   * @spd:	data to fill
   *
   * Description:
79685b8de   Randy Dunlap   docbook: add pipe...
162
   *    @spd contains a map of pages and len/offset tuples, along with
932cc6d4f   Jens Axboe   splice: completel...
163
164
165
   *    the struct pipe_buf_operations associated with these pages. This
   *    function will link that data to the pipe.
   *
83f9135bd   Jens Axboe   [PATCH] splice: a...
166
   */
d6b29d7ce   Jens Axboe   splice: divorce t...
167
168
  ssize_t splice_to_pipe(struct pipe_inode_info *pipe,
  		       struct splice_pipe_desc *spd)
5274f052e   Jens Axboe   [PATCH] Introduce...
169
  {
00de00bda   Jens Axboe   splice: fix leak ...
170
  	unsigned int spd_pages = spd->nr_pages;
912d35f86   Jens Axboe   [PATCH] Add suppo...
171
  	int ret, do_wakeup, page_nr;
5274f052e   Jens Axboe   [PATCH] Introduce...
172
173
174
  
  	ret = 0;
  	do_wakeup = 0;
912d35f86   Jens Axboe   [PATCH] Add suppo...
175
  	page_nr = 0;
5274f052e   Jens Axboe   [PATCH] Introduce...
176

61e0d47c3   Miklos Szeredi   splice: add helpe...
177
  	pipe_lock(pipe);
5274f052e   Jens Axboe   [PATCH] Introduce...
178

5274f052e   Jens Axboe   [PATCH] Introduce...
179
  	for (;;) {
3a326a2ce   Ingo Molnar   [PATCH] introduce...
180
  		if (!pipe->readers) {
5274f052e   Jens Axboe   [PATCH] Introduce...
181
182
183
184
185
  			send_sig(SIGPIPE, current, 0);
  			if (!ret)
  				ret = -EPIPE;
  			break;
  		}
35f3d14db   Jens Axboe   pipe: add support...
186
187
  		if (pipe->nrbufs < pipe->buffers) {
  			int newbuf = (pipe->curbuf + pipe->nrbufs) & (pipe->buffers - 1);
3a326a2ce   Ingo Molnar   [PATCH] introduce...
188
  			struct pipe_buffer *buf = pipe->bufs + newbuf;
5274f052e   Jens Axboe   [PATCH] Introduce...
189

912d35f86   Jens Axboe   [PATCH] Add suppo...
190
191
192
  			buf->page = spd->pages[page_nr];
  			buf->offset = spd->partial[page_nr].offset;
  			buf->len = spd->partial[page_nr].len;
497f9625c   Jens Axboe   pipe: allow passi...
193
  			buf->private = spd->partial[page_nr].private;
912d35f86   Jens Axboe   [PATCH] Add suppo...
194
  			buf->ops = spd->ops;
7afa6fd03   Jens Axboe   [PATCH] vmsplice:...
195
196
  			if (spd->flags & SPLICE_F_GIFT)
  				buf->flags |= PIPE_BUF_FLAG_GIFT;
6f767b042   Jens Axboe   [PATCH] splice: s...
197
  			pipe->nrbufs++;
912d35f86   Jens Axboe   [PATCH] Add suppo...
198
199
  			page_nr++;
  			ret += buf->len;
6f767b042   Jens Axboe   [PATCH] splice: s...
200
201
  			if (pipe->inode)
  				do_wakeup = 1;
5274f052e   Jens Axboe   [PATCH] Introduce...
202

912d35f86   Jens Axboe   [PATCH] Add suppo...
203
  			if (!--spd->nr_pages)
5274f052e   Jens Axboe   [PATCH] Introduce...
204
  				break;
35f3d14db   Jens Axboe   pipe: add support...
205
  			if (pipe->nrbufs < pipe->buffers)
5274f052e   Jens Axboe   [PATCH] Introduce...
206
207
208
209
  				continue;
  
  			break;
  		}
912d35f86   Jens Axboe   [PATCH] Add suppo...
210
  		if (spd->flags & SPLICE_F_NONBLOCK) {
29e350944   Linus Torvalds   splice: add SPLIC...
211
212
213
214
  			if (!ret)
  				ret = -EAGAIN;
  			break;
  		}
5274f052e   Jens Axboe   [PATCH] Introduce...
215
216
217
218
219
220
221
  		if (signal_pending(current)) {
  			if (!ret)
  				ret = -ERESTARTSYS;
  			break;
  		}
  
  		if (do_wakeup) {
c0bd1f650   Jens Axboe   [PATCH] splice: o...
222
  			smp_mb();
3a326a2ce   Ingo Molnar   [PATCH] introduce...
223
224
225
  			if (waitqueue_active(&pipe->wait))
  				wake_up_interruptible_sync(&pipe->wait);
  			kill_fasync(&pipe->fasync_readers, SIGIO, POLL_IN);
5274f052e   Jens Axboe   [PATCH] Introduce...
226
227
  			do_wakeup = 0;
  		}
3a326a2ce   Ingo Molnar   [PATCH] introduce...
228
229
230
  		pipe->waiting_writers++;
  		pipe_wait(pipe);
  		pipe->waiting_writers--;
5274f052e   Jens Axboe   [PATCH] Introduce...
231
  	}
61e0d47c3   Miklos Szeredi   splice: add helpe...
232
  	pipe_unlock(pipe);
5274f052e   Jens Axboe   [PATCH] Introduce...
233

61e0d47c3   Miklos Szeredi   splice: add helpe...
234
235
236
237
238
  	if (do_wakeup) {
  		smp_mb();
  		if (waitqueue_active(&pipe->wait))
  			wake_up_interruptible(&pipe->wait);
  		kill_fasync(&pipe->fasync_readers, SIGIO, POLL_IN);
5274f052e   Jens Axboe   [PATCH] Introduce...
239
  	}
00de00bda   Jens Axboe   splice: fix leak ...
240
  	while (page_nr < spd_pages)
bbdfc2f70   Jens Axboe   [SPLICE]: Don't a...
241
  		spd->spd_release(spd, page_nr++);
5274f052e   Jens Axboe   [PATCH] Introduce...
242
243
244
  
  	return ret;
  }
bbdfc2f70   Jens Axboe   [SPLICE]: Don't a...
245
246
247
248
  static void spd_release_page(struct splice_pipe_desc *spd, unsigned int i)
  {
  	page_cache_release(spd->pages[i]);
  }
35f3d14db   Jens Axboe   pipe: add support...
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
  /*
   * Check if we need to grow the arrays holding pages and partial page
   * descriptions.
   */
  int splice_grow_spd(struct pipe_inode_info *pipe, struct splice_pipe_desc *spd)
  {
  	if (pipe->buffers <= PIPE_DEF_BUFFERS)
  		return 0;
  
  	spd->pages = kmalloc(pipe->buffers * sizeof(struct page *), GFP_KERNEL);
  	spd->partial = kmalloc(pipe->buffers * sizeof(struct partial_page), GFP_KERNEL);
  
  	if (spd->pages && spd->partial)
  		return 0;
  
  	kfree(spd->pages);
  	kfree(spd->partial);
  	return -ENOMEM;
  }
  
  void splice_shrink_spd(struct pipe_inode_info *pipe,
  		       struct splice_pipe_desc *spd)
  {
  	if (pipe->buffers <= PIPE_DEF_BUFFERS)
  		return;
  
  	kfree(spd->pages);
  	kfree(spd->partial);
  }
3a326a2ce   Ingo Molnar   [PATCH] introduce...
278
  static int
cbb7e577e   Jens Axboe   [PATCH] splice: p...
279
280
281
  __generic_file_splice_read(struct file *in, loff_t *ppos,
  			   struct pipe_inode_info *pipe, size_t len,
  			   unsigned int flags)
5274f052e   Jens Axboe   [PATCH] Introduce...
282
283
  {
  	struct address_space *mapping = in->f_mapping;
d8983910a   Fengguang Wu   readahead: pass r...
284
  	unsigned int loff, nr_pages, req_pages;
35f3d14db   Jens Axboe   pipe: add support...
285
286
  	struct page *pages[PIPE_DEF_BUFFERS];
  	struct partial_page partial[PIPE_DEF_BUFFERS];
5274f052e   Jens Axboe   [PATCH] Introduce...
287
  	struct page *page;
91ad66ef4   Jens Axboe   [PATCH] splice: c...
288
289
  	pgoff_t index, end_index;
  	loff_t isize;
eb20796bf   Jens Axboe   [PATCH] splice: m...
290
  	int error, page_nr;
912d35f86   Jens Axboe   [PATCH] Add suppo...
291
292
293
294
295
  	struct splice_pipe_desc spd = {
  		.pages = pages,
  		.partial = partial,
  		.flags = flags,
  		.ops = &page_cache_pipe_buf_ops,
bbdfc2f70   Jens Axboe   [SPLICE]: Don't a...
296
  		.spd_release = spd_release_page,
912d35f86   Jens Axboe   [PATCH] Add suppo...
297
  	};
5274f052e   Jens Axboe   [PATCH] Introduce...
298

35f3d14db   Jens Axboe   pipe: add support...
299
300
  	if (splice_grow_spd(pipe, &spd))
  		return -ENOMEM;
cbb7e577e   Jens Axboe   [PATCH] splice: p...
301
  	index = *ppos >> PAGE_CACHE_SHIFT;
912d35f86   Jens Axboe   [PATCH] Add suppo...
302
  	loff = *ppos & ~PAGE_CACHE_MASK;
d8983910a   Fengguang Wu   readahead: pass r...
303
  	req_pages = (len + loff + PAGE_CACHE_SIZE - 1) >> PAGE_CACHE_SHIFT;
35f3d14db   Jens Axboe   pipe: add support...
304
  	nr_pages = min(req_pages, pipe->buffers);
5274f052e   Jens Axboe   [PATCH] Introduce...
305
306
  
  	/*
eb20796bf   Jens Axboe   [PATCH] splice: m...
307
308
  	 * Lookup the (hopefully) full range of pages we need.
  	 */
35f3d14db   Jens Axboe   pipe: add support...
309
  	spd.nr_pages = find_get_pages_contig(mapping, index, nr_pages, spd.pages);
431a4820b   Fengguang Wu   readahead: move s...
310
  	index += spd.nr_pages;
82aa5d618   Jens Axboe   [PATCH] splice: f...
311

eb20796bf   Jens Axboe   [PATCH] splice: m...
312
313
  	/*
  	 * If find_get_pages_contig() returned fewer pages than we needed,
431a4820b   Fengguang Wu   readahead: move s...
314
  	 * readahead/allocate the rest and fill in the holes.
eb20796bf   Jens Axboe   [PATCH] splice: m...
315
  	 */
431a4820b   Fengguang Wu   readahead: move s...
316
  	if (spd.nr_pages < nr_pages)
cf914a7d6   Rusty Russell   readahead: split ...
317
318
  		page_cache_sync_readahead(mapping, &in->f_ra, in,
  				index, req_pages - spd.nr_pages);
431a4820b   Fengguang Wu   readahead: move s...
319

932cc6d4f   Jens Axboe   splice: completel...
320
  	error = 0;
eb20796bf   Jens Axboe   [PATCH] splice: m...
321
  	while (spd.nr_pages < nr_pages) {
82aa5d618   Jens Axboe   [PATCH] splice: f...
322
  		/*
eb20796bf   Jens Axboe   [PATCH] splice: m...
323
324
  		 * Page could be there, find_get_pages_contig() breaks on
  		 * the first hole.
5274f052e   Jens Axboe   [PATCH] Introduce...
325
  		 */
7480a9043   Jens Axboe   [PATCH] splice: s...
326
327
  		page = find_get_page(mapping, index);
  		if (!page) {
e27dedd84   Jens Axboe   [PATCH] splice: c...
328
  			/*
eb20796bf   Jens Axboe   [PATCH] splice: m...
329
  			 * page didn't exist, allocate one.
7480a9043   Jens Axboe   [PATCH] splice: s...
330
331
332
333
334
335
  			 */
  			page = page_cache_alloc_cold(mapping);
  			if (!page)
  				break;
  
  			error = add_to_page_cache_lru(page, mapping, index,
0ae0b5d05   Nick Piggin   fs/splice.c: fix ...
336
  						GFP_KERNEL);
7480a9043   Jens Axboe   [PATCH] splice: s...
337
338
  			if (unlikely(error)) {
  				page_cache_release(page);
a0548871e   Jens Axboe   [PATCH] splice: r...
339
340
  				if (error == -EEXIST)
  					continue;
7480a9043   Jens Axboe   [PATCH] splice: s...
341
342
  				break;
  			}
eb20796bf   Jens Axboe   [PATCH] splice: m...
343
344
345
346
347
  			/*
  			 * add_to_page_cache() locks the page, unlock it
  			 * to avoid convoluting the logic below even more.
  			 */
  			unlock_page(page);
7480a9043   Jens Axboe   [PATCH] splice: s...
348
  		}
35f3d14db   Jens Axboe   pipe: add support...
349
  		spd.pages[spd.nr_pages++] = page;
eb20796bf   Jens Axboe   [PATCH] splice: m...
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
  		index++;
  	}
  
  	/*
  	 * Now loop over the map and see if we need to start IO on any
  	 * pages, fill in the partial map, etc.
  	 */
  	index = *ppos >> PAGE_CACHE_SHIFT;
  	nr_pages = spd.nr_pages;
  	spd.nr_pages = 0;
  	for (page_nr = 0; page_nr < nr_pages; page_nr++) {
  		unsigned int this_len;
  
  		if (!len)
  			break;
  
  		/*
  		 * this_len is the max we'll use from this page
  		 */
  		this_len = min_t(unsigned long, len, PAGE_CACHE_SIZE - loff);
35f3d14db   Jens Axboe   pipe: add support...
370
  		page = spd.pages[page_nr];
eb20796bf   Jens Axboe   [PATCH] splice: m...
371

a08a166fe   Fengguang Wu   readahead: conver...
372
  		if (PageReadahead(page))
cf914a7d6   Rusty Russell   readahead: split ...
373
  			page_cache_async_readahead(mapping, &in->f_ra, in,
d8983910a   Fengguang Wu   readahead: pass r...
374
  					page, index, req_pages - page_nr);
a08a166fe   Fengguang Wu   readahead: conver...
375

7480a9043   Jens Axboe   [PATCH] splice: s...
376
377
378
379
  		/*
  		 * If the page isn't uptodate, we may need to start io on it
  		 */
  		if (!PageUptodate(page)) {
6965031d3   Miklos Szeredi   splice: fix misus...
380
  			lock_page(page);
7480a9043   Jens Axboe   [PATCH] splice: s...
381
382
  
  			/*
32502b841   Miklos Szeredi   splice: fix gener...
383
384
385
386
  			 * Page was truncated, or invalidated by the
  			 * filesystem.  Redo the find/create, but this time the
  			 * page is kept locked, so there's no chance of another
  			 * race with truncate/invalidate.
7480a9043   Jens Axboe   [PATCH] splice: s...
387
388
389
  			 */
  			if (!page->mapping) {
  				unlock_page(page);
32502b841   Miklos Szeredi   splice: fix gener...
390
391
392
393
394
395
396
  				page = find_or_create_page(mapping, index,
  						mapping_gfp_mask(mapping));
  
  				if (!page) {
  					error = -ENOMEM;
  					break;
  				}
35f3d14db   Jens Axboe   pipe: add support...
397
398
  				page_cache_release(spd.pages[page_nr]);
  				spd.pages[page_nr] = page;
7480a9043   Jens Axboe   [PATCH] splice: s...
399
400
401
402
403
404
405
406
  			}
  			/*
  			 * page was already under io and is now done, great
  			 */
  			if (PageUptodate(page)) {
  				unlock_page(page);
  				goto fill_it;
  			}
5274f052e   Jens Axboe   [PATCH] Introduce...
407

7480a9043   Jens Axboe   [PATCH] splice: s...
408
409
410
411
  			/*
  			 * need to read in the page
  			 */
  			error = mapping->a_ops->readpage(in, page);
5274f052e   Jens Axboe   [PATCH] Introduce...
412
  			if (unlikely(error)) {
eb20796bf   Jens Axboe   [PATCH] splice: m...
413
414
415
416
417
418
  				/*
  				 * We really should re-lookup the page here,
  				 * but it complicates things a lot. Instead
  				 * lets just do what we already stored, and
  				 * we'll get it the next time we are called.
  				 */
7480a9043   Jens Axboe   [PATCH] splice: s...
419
  				if (error == AOP_TRUNCATED_PAGE)
eb20796bf   Jens Axboe   [PATCH] splice: m...
420
  					error = 0;
5274f052e   Jens Axboe   [PATCH] Introduce...
421
422
  				break;
  			}
620a324b7   Jens Axboe   splice: __generic...
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
  		}
  fill_it:
  		/*
  		 * i_size must be checked after PageUptodate.
  		 */
  		isize = i_size_read(mapping->host);
  		end_index = (isize - 1) >> PAGE_CACHE_SHIFT;
  		if (unlikely(!isize || index > end_index))
  			break;
  
  		/*
  		 * if this is the last page, see if we need to shrink
  		 * the length and stop
  		 */
  		if (end_index == index) {
  			unsigned int plen;
91ad66ef4   Jens Axboe   [PATCH] splice: c...
439
440
  
  			/*
620a324b7   Jens Axboe   splice: __generic...
441
  			 * max good bytes in this page
91ad66ef4   Jens Axboe   [PATCH] splice: c...
442
  			 */
620a324b7   Jens Axboe   splice: __generic...
443
444
  			plen = ((isize - 1) & ~PAGE_CACHE_MASK) + 1;
  			if (plen <= loff)
91ad66ef4   Jens Axboe   [PATCH] splice: c...
445
  				break;
91ad66ef4   Jens Axboe   [PATCH] splice: c...
446
447
  
  			/*
620a324b7   Jens Axboe   splice: __generic...
448
  			 * force quit after adding this page
91ad66ef4   Jens Axboe   [PATCH] splice: c...
449
  			 */
620a324b7   Jens Axboe   splice: __generic...
450
451
  			this_len = min(this_len, plen - loff);
  			len = this_len;
5274f052e   Jens Axboe   [PATCH] Introduce...
452
  		}
620a324b7   Jens Axboe   splice: __generic...
453

35f3d14db   Jens Axboe   pipe: add support...
454
455
  		spd.partial[page_nr].offset = loff;
  		spd.partial[page_nr].len = this_len;
82aa5d618   Jens Axboe   [PATCH] splice: f...
456
  		len -= this_len;
91ad66ef4   Jens Axboe   [PATCH] splice: c...
457
  		loff = 0;
eb20796bf   Jens Axboe   [PATCH] splice: m...
458
459
  		spd.nr_pages++;
  		index++;
5274f052e   Jens Axboe   [PATCH] Introduce...
460
  	}
eb20796bf   Jens Axboe   [PATCH] splice: m...
461
  	/*
475ecade6   Hugh Dickins   splice: __generic...
462
  	 * Release any pages at the end, if we quit early. 'page_nr' is how far
eb20796bf   Jens Axboe   [PATCH] splice: m...
463
464
465
  	 * we got, 'nr_pages' is how many pages are in the map.
  	 */
  	while (page_nr < nr_pages)
35f3d14db   Jens Axboe   pipe: add support...
466
  		page_cache_release(spd.pages[page_nr++]);
f4e6b498d   Fengguang Wu   readahead: combin...
467
  	in->f_ra.prev_pos = (loff_t)index << PAGE_CACHE_SHIFT;
eb20796bf   Jens Axboe   [PATCH] splice: m...
468

912d35f86   Jens Axboe   [PATCH] Add suppo...
469
  	if (spd.nr_pages)
35f3d14db   Jens Axboe   pipe: add support...
470
  		error = splice_to_pipe(pipe, &spd);
5274f052e   Jens Axboe   [PATCH] Introduce...
471

35f3d14db   Jens Axboe   pipe: add support...
472
  	splice_shrink_spd(pipe, &spd);
7480a9043   Jens Axboe   [PATCH] splice: s...
473
  	return error;
5274f052e   Jens Axboe   [PATCH] Introduce...
474
  }
83f9135bd   Jens Axboe   [PATCH] splice: a...
475
476
477
  /**
   * generic_file_splice_read - splice data from file to a pipe
   * @in:		file to splice from
932cc6d4f   Jens Axboe   splice: completel...
478
   * @ppos:	position in @in
83f9135bd   Jens Axboe   [PATCH] splice: a...
479
480
481
482
   * @pipe:	pipe to splice to
   * @len:	number of bytes to splice
   * @flags:	splice modifier flags
   *
932cc6d4f   Jens Axboe   splice: completel...
483
484
485
486
487
   * Description:
   *    Will read pages from given file and fill them into a pipe. Can be
   *    used as long as the address_space operations for the source implements
   *    a readpage() hook.
   *
83f9135bd   Jens Axboe   [PATCH] splice: a...
488
   */
cbb7e577e   Jens Axboe   [PATCH] splice: p...
489
490
491
  ssize_t generic_file_splice_read(struct file *in, loff_t *ppos,
  				 struct pipe_inode_info *pipe, size_t len,
  				 unsigned int flags)
5274f052e   Jens Axboe   [PATCH] Introduce...
492
  {
d366d3988   Jens Axboe   splice: move inod...
493
  	loff_t isize, left;
8191ecd1d   Jens Axboe   splice: fix infin...
494
  	int ret;
d366d3988   Jens Axboe   splice: move inod...
495
496
497
498
499
500
501
502
  
  	isize = i_size_read(in->f_mapping->host);
  	if (unlikely(*ppos >= isize))
  		return 0;
  
  	left = isize - *ppos;
  	if (unlikely(left < len))
  		len = left;
5274f052e   Jens Axboe   [PATCH] Introduce...
503

8191ecd1d   Jens Axboe   splice: fix infin...
504
  	ret = __generic_file_splice_read(in, ppos, pipe, len, flags);
723590ed5   Miklos Szeredi   splice: update mt...
505
  	if (ret > 0) {
cbb7e577e   Jens Axboe   [PATCH] splice: p...
506
  		*ppos += ret;
723590ed5   Miklos Szeredi   splice: update mt...
507
508
  		file_accessed(in);
  	}
5274f052e   Jens Axboe   [PATCH] Introduce...
509
510
511
  
  	return ret;
  }
059a8f373   Jens Axboe   [PATCH] splice: e...
512
  EXPORT_SYMBOL(generic_file_splice_read);
6818173bd   Miklos Szeredi   splice: implement...
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
  static const struct pipe_buf_operations default_pipe_buf_ops = {
  	.can_merge = 0,
  	.map = generic_pipe_buf_map,
  	.unmap = generic_pipe_buf_unmap,
  	.confirm = generic_pipe_buf_confirm,
  	.release = generic_pipe_buf_release,
  	.steal = generic_pipe_buf_steal,
  	.get = generic_pipe_buf_get,
  };
  
  static ssize_t kernel_readv(struct file *file, const struct iovec *vec,
  			    unsigned long vlen, loff_t offset)
  {
  	mm_segment_t old_fs;
  	loff_t pos = offset;
  	ssize_t res;
  
  	old_fs = get_fs();
  	set_fs(get_ds());
  	/* The cast to a user pointer is valid due to the set_fs() */
  	res = vfs_readv(file, (const struct iovec __user *)vec, vlen, &pos);
  	set_fs(old_fs);
  
  	return res;
  }
b2858d7d1   Miklos Szeredi   splice: fix kmaps...
538
539
  static ssize_t kernel_write(struct file *file, const char *buf, size_t count,
  			    loff_t pos)
0b0a47f5c   Miklos Szeredi   splice: implement...
540
541
542
543
544
545
546
  {
  	mm_segment_t old_fs;
  	ssize_t res;
  
  	old_fs = get_fs();
  	set_fs(get_ds());
  	/* The cast to a user pointer is valid due to the set_fs() */
b2858d7d1   Miklos Szeredi   splice: fix kmaps...
547
  	res = vfs_write(file, (const char __user *)buf, count, &pos);
0b0a47f5c   Miklos Szeredi   splice: implement...
548
549
550
551
  	set_fs(old_fs);
  
  	return res;
  }
6818173bd   Miklos Szeredi   splice: implement...
552
553
554
555
556
557
558
  ssize_t default_file_splice_read(struct file *in, loff_t *ppos,
  				 struct pipe_inode_info *pipe, size_t len,
  				 unsigned int flags)
  {
  	unsigned int nr_pages;
  	unsigned int nr_freed;
  	size_t offset;
35f3d14db   Jens Axboe   pipe: add support...
559
560
561
  	struct page *pages[PIPE_DEF_BUFFERS];
  	struct partial_page partial[PIPE_DEF_BUFFERS];
  	struct iovec *vec, __vec[PIPE_DEF_BUFFERS];
6818173bd   Miklos Szeredi   splice: implement...
562
563
564
565
566
567
568
569
570
571
572
  	ssize_t res;
  	size_t this_len;
  	int error;
  	int i;
  	struct splice_pipe_desc spd = {
  		.pages = pages,
  		.partial = partial,
  		.flags = flags,
  		.ops = &default_pipe_buf_ops,
  		.spd_release = spd_release_page,
  	};
35f3d14db   Jens Axboe   pipe: add support...
573
574
575
576
577
578
579
580
581
582
  	if (splice_grow_spd(pipe, &spd))
  		return -ENOMEM;
  
  	res = -ENOMEM;
  	vec = __vec;
  	if (pipe->buffers > PIPE_DEF_BUFFERS) {
  		vec = kmalloc(pipe->buffers * sizeof(struct iovec), GFP_KERNEL);
  		if (!vec)
  			goto shrink_ret;
  	}
6818173bd   Miklos Szeredi   splice: implement...
583
584
  	offset = *ppos & ~PAGE_CACHE_MASK;
  	nr_pages = (len + offset + PAGE_CACHE_SIZE - 1) >> PAGE_CACHE_SHIFT;
35f3d14db   Jens Axboe   pipe: add support...
585
  	for (i = 0; i < nr_pages && i < pipe->buffers && len; i++) {
6818173bd   Miklos Szeredi   splice: implement...
586
  		struct page *page;
4f2312285   Jens Axboe   splice: fix repea...
587
  		page = alloc_page(GFP_USER);
6818173bd   Miklos Szeredi   splice: implement...
588
589
590
591
592
  		error = -ENOMEM;
  		if (!page)
  			goto err;
  
  		this_len = min_t(size_t, len, PAGE_CACHE_SIZE - offset);
4f2312285   Jens Axboe   splice: fix repea...
593
  		vec[i].iov_base = (void __user *) page_address(page);
6818173bd   Miklos Szeredi   splice: implement...
594
  		vec[i].iov_len = this_len;
35f3d14db   Jens Axboe   pipe: add support...
595
  		spd.pages[i] = page;
6818173bd   Miklos Szeredi   splice: implement...
596
597
598
599
600
601
  		spd.nr_pages++;
  		len -= this_len;
  		offset = 0;
  	}
  
  	res = kernel_readv(in, vec, spd.nr_pages, *ppos);
77f6bf57b   Andrew Morton   splice: fix error...
602
603
  	if (res < 0) {
  		error = res;
6818173bd   Miklos Szeredi   splice: implement...
604
  		goto err;
77f6bf57b   Andrew Morton   splice: fix error...
605
  	}
6818173bd   Miklos Szeredi   splice: implement...
606
607
608
609
610
611
612
  
  	error = 0;
  	if (!res)
  		goto err;
  
  	nr_freed = 0;
  	for (i = 0; i < spd.nr_pages; i++) {
6818173bd   Miklos Szeredi   splice: implement...
613
  		this_len = min_t(size_t, vec[i].iov_len, res);
35f3d14db   Jens Axboe   pipe: add support...
614
615
  		spd.partial[i].offset = 0;
  		spd.partial[i].len = this_len;
6818173bd   Miklos Szeredi   splice: implement...
616
  		if (!this_len) {
35f3d14db   Jens Axboe   pipe: add support...
617
618
  			__free_page(spd.pages[i]);
  			spd.pages[i] = NULL;
6818173bd   Miklos Szeredi   splice: implement...
619
620
621
622
623
624
625
626
627
  			nr_freed++;
  		}
  		res -= this_len;
  	}
  	spd.nr_pages -= nr_freed;
  
  	res = splice_to_pipe(pipe, &spd);
  	if (res > 0)
  		*ppos += res;
35f3d14db   Jens Axboe   pipe: add support...
628
629
630
631
  shrink_ret:
  	if (vec != __vec)
  		kfree(vec);
  	splice_shrink_spd(pipe, &spd);
6818173bd   Miklos Szeredi   splice: implement...
632
633
634
  	return res;
  
  err:
4f2312285   Jens Axboe   splice: fix repea...
635
  	for (i = 0; i < spd.nr_pages; i++)
35f3d14db   Jens Axboe   pipe: add support...
636
  		__free_page(spd.pages[i]);
4f2312285   Jens Axboe   splice: fix repea...
637

35f3d14db   Jens Axboe   pipe: add support...
638
639
  	res = error;
  	goto shrink_ret;
6818173bd   Miklos Szeredi   splice: implement...
640
641
  }
  EXPORT_SYMBOL(default_file_splice_read);
5274f052e   Jens Axboe   [PATCH] Introduce...
642
  /*
4f6f0bd2f   Jens Axboe   [PATCH] splice: i...
643
   * Send 'sd->len' bytes to socket from 'sd->file' at position 'sd->pos'
016b661e2   Jens Axboe   [PATCH] splice: f...
644
   * using sendpage(). Return the number of bytes sent.
5274f052e   Jens Axboe   [PATCH] Introduce...
645
   */
76ad4d111   Jens Axboe   [PATCH] splice: r...
646
  static int pipe_to_sendpage(struct pipe_inode_info *pipe,
5274f052e   Jens Axboe   [PATCH] Introduce...
647
648
  			    struct pipe_buffer *buf, struct splice_desc *sd)
  {
6a14b90bb   Jens Axboe   vmsplice: add vms...
649
  	struct file *file = sd->u.file;
5274f052e   Jens Axboe   [PATCH] Introduce...
650
  	loff_t pos = sd->pos;
a8adbe378   MichaÅ‚ MirosÅ‚aw   fs/splice: Pull b...
651
  	int more;
5274f052e   Jens Axboe   [PATCH] Introduce...
652

a8adbe378   MichaÅ‚ MirosÅ‚aw   fs/splice: Pull b...
653
654
655
656
657
658
  	if (!likely(file->f_op && file->f_op->sendpage))
  		return -EINVAL;
  
  	more = (sd->flags & SPLICE_F_MORE) || sd->len < sd->total_len;
  	return file->f_op->sendpage(file, buf->page, buf->offset,
  				    sd->len, &pos, more);
5274f052e   Jens Axboe   [PATCH] Introduce...
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
  }
  
  /*
   * This is a little more tricky than the file -> pipe splicing. There are
   * basically three cases:
   *
   *	- Destination page already exists in the address space and there
   *	  are users of it. For that case we have no other option that
   *	  copying the data. Tough luck.
   *	- Destination page already exists in the address space, but there
   *	  are no users of it. Make sure it's uptodate, then drop it. Fall
   *	  through to last case.
   *	- Destination page does not exist, we can add the pipe page to
   *	  the page cache and avoid the copy.
   *
83f9135bd   Jens Axboe   [PATCH] splice: a...
674
675
676
677
678
679
   * If asked to move pages to the output file (SPLICE_F_MOVE is set in
   * sd->flags), we attempt to migrate pages from the pipe to the output
   * file address space page cache. This is possible if no one else has
   * the pipe page referenced outside of the pipe and page cache. If
   * SPLICE_F_MOVE isn't set, or we cannot move the page, we simply create
   * a new page in the output file page cache and fill/dirty that.
5274f052e   Jens Axboe   [PATCH] Introduce...
680
   */
328eaaba4   Miklos Szeredi   ocfs2: fix i_mute...
681
682
  int pipe_to_file(struct pipe_inode_info *pipe, struct pipe_buffer *buf,
  		 struct splice_desc *sd)
5274f052e   Jens Axboe   [PATCH] Introduce...
683
  {
6a14b90bb   Jens Axboe   vmsplice: add vms...
684
  	struct file *file = sd->u.file;
5274f052e   Jens Axboe   [PATCH] Introduce...
685
  	struct address_space *mapping = file->f_mapping;
016b661e2   Jens Axboe   [PATCH] splice: f...
686
  	unsigned int offset, this_len;
5274f052e   Jens Axboe   [PATCH] Introduce...
687
  	struct page *page;
afddba49d   Nick Piggin   fs: introduce wri...
688
  	void *fsdata;
3e7ee3e7b   Jens Axboe   [PATCH] splice: f...
689
  	int ret;
5274f052e   Jens Axboe   [PATCH] Introduce...
690

5274f052e   Jens Axboe   [PATCH] Introduce...
691
  	offset = sd->pos & ~PAGE_CACHE_MASK;
016b661e2   Jens Axboe   [PATCH] splice: f...
692
693
694
  	this_len = sd->len;
  	if (this_len + offset > PAGE_CACHE_SIZE)
  		this_len = PAGE_CACHE_SIZE - offset;
afddba49d   Nick Piggin   fs: introduce wri...
695
696
697
698
  	ret = pagecache_write_begin(file, mapping, sd->pos, this_len,
  				AOP_FLAG_UNINTERRUPTIBLE, &page, &fsdata);
  	if (unlikely(ret))
  		goto out;
5274f052e   Jens Axboe   [PATCH] Introduce...
699

0568b409c   Jens Axboe   [PATCH] splice: f...
700
  	if (buf->page != page) {
f84d75199   Jens Axboe   [PATCH] pipe: int...
701
702
703
  		/*
  		 * Careful, ->map() uses KM_USER0!
  		 */
76ad4d111   Jens Axboe   [PATCH] splice: r...
704
  		char *src = buf->ops->map(pipe, buf, 1);
f84d75199   Jens Axboe   [PATCH] pipe: int...
705
  		char *dst = kmap_atomic(page, KM_USER1);
5abc97aa2   Jens Axboe   [PATCH] splice: a...
706

016b661e2   Jens Axboe   [PATCH] splice: f...
707
  		memcpy(dst + offset, src + buf->offset, this_len);
5abc97aa2   Jens Axboe   [PATCH] splice: a...
708
  		flush_dcache_page(page);
f84d75199   Jens Axboe   [PATCH] pipe: int...
709
  		kunmap_atomic(dst, KM_USER1);
76ad4d111   Jens Axboe   [PATCH] splice: r...
710
  		buf->ops->unmap(pipe, buf, src);
5abc97aa2   Jens Axboe   [PATCH] splice: a...
711
  	}
afddba49d   Nick Piggin   fs: introduce wri...
712
713
  	ret = pagecache_write_end(file, mapping, sd->pos, this_len, this_len,
  				page, fsdata);
5274f052e   Jens Axboe   [PATCH] Introduce...
714
  out:
5274f052e   Jens Axboe   [PATCH] Introduce...
715
716
  	return ret;
  }
328eaaba4   Miklos Szeredi   ocfs2: fix i_mute...
717
  EXPORT_SYMBOL(pipe_to_file);
5274f052e   Jens Axboe   [PATCH] Introduce...
718

b3c2d2ddd   Miklos Szeredi   splice: split up ...
719
720
721
722
723
724
725
  static void wakeup_pipe_writers(struct pipe_inode_info *pipe)
  {
  	smp_mb();
  	if (waitqueue_active(&pipe->wait))
  		wake_up_interruptible(&pipe->wait);
  	kill_fasync(&pipe->fasync_writers, SIGIO, POLL_OUT);
  }
932cc6d4f   Jens Axboe   splice: completel...
726
  /**
b3c2d2ddd   Miklos Szeredi   splice: split up ...
727
   * splice_from_pipe_feed - feed available data from a pipe to a file
932cc6d4f   Jens Axboe   splice: completel...
728
729
730
731
732
   * @pipe:	pipe to splice from
   * @sd:		information to @actor
   * @actor:	handler that splices the data
   *
   * Description:
b3c2d2ddd   Miklos Szeredi   splice: split up ...
733
734
735
736
737
738
739
   *    This function loops over the pipe and calls @actor to do the
   *    actual moving of a single struct pipe_buffer to the desired
   *    destination.  It returns when there's no more buffers left in
   *    the pipe or if the requested number of bytes (@sd->total_len)
   *    have been copied.  It returns a positive number (one) if the
   *    pipe needs to be filled with more data, zero if the required
   *    number of bytes have been copied and -errno on error.
932cc6d4f   Jens Axboe   splice: completel...
740
   *
b3c2d2ddd   Miklos Szeredi   splice: split up ...
741
742
743
744
   *    This, together with splice_from_pipe_{begin,end,next}, may be
   *    used to implement the functionality of __splice_from_pipe() when
   *    locking is required around copying the pipe buffers to the
   *    destination.
83f9135bd   Jens Axboe   [PATCH] splice: a...
745
   */
b3c2d2ddd   Miklos Szeredi   splice: split up ...
746
747
  int splice_from_pipe_feed(struct pipe_inode_info *pipe, struct splice_desc *sd,
  			  splice_actor *actor)
5274f052e   Jens Axboe   [PATCH] Introduce...
748
  {
b3c2d2ddd   Miklos Szeredi   splice: split up ...
749
  	int ret;
5274f052e   Jens Axboe   [PATCH] Introduce...
750

b3c2d2ddd   Miklos Szeredi   splice: split up ...
751
752
753
  	while (pipe->nrbufs) {
  		struct pipe_buffer *buf = pipe->bufs + pipe->curbuf;
  		const struct pipe_buf_operations *ops = buf->ops;
5274f052e   Jens Axboe   [PATCH] Introduce...
754

b3c2d2ddd   Miklos Szeredi   splice: split up ...
755
756
757
  		sd->len = buf->len;
  		if (sd->len > sd->total_len)
  			sd->len = sd->total_len;
5274f052e   Jens Axboe   [PATCH] Introduce...
758

a8adbe378   MichaÅ‚ MirosÅ‚aw   fs/splice: Pull b...
759
760
  		ret = buf->ops->confirm(pipe, buf);
  		if (unlikely(ret)) {
b3c2d2ddd   Miklos Szeredi   splice: split up ...
761
762
763
764
  			if (ret == -ENODATA)
  				ret = 0;
  			return ret;
  		}
a8adbe378   MichaÅ‚ MirosÅ‚aw   fs/splice: Pull b...
765
766
767
768
  
  		ret = actor(pipe, buf, sd);
  		if (ret <= 0)
  			return ret;
b3c2d2ddd   Miklos Szeredi   splice: split up ...
769
770
771
772
773
774
775
776
777
778
779
  		buf->offset += ret;
  		buf->len -= ret;
  
  		sd->num_spliced += ret;
  		sd->len -= ret;
  		sd->pos += ret;
  		sd->total_len -= ret;
  
  		if (!buf->len) {
  			buf->ops = NULL;
  			ops->release(pipe, buf);
35f3d14db   Jens Axboe   pipe: add support...
780
  			pipe->curbuf = (pipe->curbuf + 1) & (pipe->buffers - 1);
b3c2d2ddd   Miklos Szeredi   splice: split up ...
781
782
783
784
  			pipe->nrbufs--;
  			if (pipe->inode)
  				sd->need_wakeup = true;
  		}
5274f052e   Jens Axboe   [PATCH] Introduce...
785

b3c2d2ddd   Miklos Szeredi   splice: split up ...
786
787
788
  		if (!sd->total_len)
  			return 0;
  	}
5274f052e   Jens Axboe   [PATCH] Introduce...
789

b3c2d2ddd   Miklos Szeredi   splice: split up ...
790
791
792
  	return 1;
  }
  EXPORT_SYMBOL(splice_from_pipe_feed);
5274f052e   Jens Axboe   [PATCH] Introduce...
793

b3c2d2ddd   Miklos Szeredi   splice: split up ...
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
  /**
   * splice_from_pipe_next - wait for some data to splice from
   * @pipe:	pipe to splice from
   * @sd:		information about the splice operation
   *
   * Description:
   *    This function will wait for some data and return a positive
   *    value (one) if pipe buffers are available.  It will return zero
   *    or -errno if no more data needs to be spliced.
   */
  int splice_from_pipe_next(struct pipe_inode_info *pipe, struct splice_desc *sd)
  {
  	while (!pipe->nrbufs) {
  		if (!pipe->writers)
  			return 0;
016b661e2   Jens Axboe   [PATCH] splice: f...
809

b3c2d2ddd   Miklos Szeredi   splice: split up ...
810
811
  		if (!pipe->waiting_writers && sd->num_spliced)
  			return 0;
73d62d83e   Ingo Molnar   [PATCH] splice: c...
812

b3c2d2ddd   Miklos Szeredi   splice: split up ...
813
814
  		if (sd->flags & SPLICE_F_NONBLOCK)
  			return -EAGAIN;
5274f052e   Jens Axboe   [PATCH] Introduce...
815

b3c2d2ddd   Miklos Szeredi   splice: split up ...
816
817
  		if (signal_pending(current))
  			return -ERESTARTSYS;
5274f052e   Jens Axboe   [PATCH] Introduce...
818

b3c2d2ddd   Miklos Szeredi   splice: split up ...
819
820
821
  		if (sd->need_wakeup) {
  			wakeup_pipe_writers(pipe);
  			sd->need_wakeup = false;
5274f052e   Jens Axboe   [PATCH] Introduce...
822
  		}
b3c2d2ddd   Miklos Szeredi   splice: split up ...
823
824
  		pipe_wait(pipe);
  	}
29e350944   Linus Torvalds   splice: add SPLIC...
825

b3c2d2ddd   Miklos Szeredi   splice: split up ...
826
827
828
  	return 1;
  }
  EXPORT_SYMBOL(splice_from_pipe_next);
5274f052e   Jens Axboe   [PATCH] Introduce...
829

b3c2d2ddd   Miklos Szeredi   splice: split up ...
830
831
  /**
   * splice_from_pipe_begin - start splicing from pipe
b80901bbf   Randy Dunlap   splice: fix new k...
832
   * @sd:		information about the splice operation
b3c2d2ddd   Miklos Szeredi   splice: split up ...
833
834
835
836
837
838
839
840
841
842
843
844
   *
   * Description:
   *    This function should be called before a loop containing
   *    splice_from_pipe_next() and splice_from_pipe_feed() to
   *    initialize the necessary fields of @sd.
   */
  void splice_from_pipe_begin(struct splice_desc *sd)
  {
  	sd->num_spliced = 0;
  	sd->need_wakeup = false;
  }
  EXPORT_SYMBOL(splice_from_pipe_begin);
5274f052e   Jens Axboe   [PATCH] Introduce...
845

b3c2d2ddd   Miklos Szeredi   splice: split up ...
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
  /**
   * splice_from_pipe_end - finish splicing from pipe
   * @pipe:	pipe to splice from
   * @sd:		information about the splice operation
   *
   * Description:
   *    This function will wake up pipe writers if necessary.  It should
   *    be called after a loop containing splice_from_pipe_next() and
   *    splice_from_pipe_feed().
   */
  void splice_from_pipe_end(struct pipe_inode_info *pipe, struct splice_desc *sd)
  {
  	if (sd->need_wakeup)
  		wakeup_pipe_writers(pipe);
  }
  EXPORT_SYMBOL(splice_from_pipe_end);
5274f052e   Jens Axboe   [PATCH] Introduce...
862

b3c2d2ddd   Miklos Szeredi   splice: split up ...
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
  /**
   * __splice_from_pipe - splice data from a pipe to given actor
   * @pipe:	pipe to splice from
   * @sd:		information to @actor
   * @actor:	handler that splices the data
   *
   * Description:
   *    This function does little more than loop over the pipe and call
   *    @actor to do the actual moving of a single struct pipe_buffer to
   *    the desired destination. See pipe_to_file, pipe_to_sendpage, or
   *    pipe_to_user.
   *
   */
  ssize_t __splice_from_pipe(struct pipe_inode_info *pipe, struct splice_desc *sd,
  			   splice_actor *actor)
  {
  	int ret;
5274f052e   Jens Axboe   [PATCH] Introduce...
880

b3c2d2ddd   Miklos Szeredi   splice: split up ...
881
882
883
884
885
886
887
888
889
  	splice_from_pipe_begin(sd);
  	do {
  		ret = splice_from_pipe_next(pipe, sd);
  		if (ret > 0)
  			ret = splice_from_pipe_feed(pipe, sd, actor);
  	} while (ret > 0);
  	splice_from_pipe_end(pipe, sd);
  
  	return sd->num_spliced ? sd->num_spliced : ret;
5274f052e   Jens Axboe   [PATCH] Introduce...
890
  }
40bee44ea   Mark Fasheh   Export __splice_f...
891
  EXPORT_SYMBOL(__splice_from_pipe);
5274f052e   Jens Axboe   [PATCH] Introduce...
892

932cc6d4f   Jens Axboe   splice: completel...
893
894
895
896
897
898
899
900
901
902
  /**
   * splice_from_pipe - splice data from a pipe to a file
   * @pipe:	pipe to splice from
   * @out:	file to splice to
   * @ppos:	position in @out
   * @len:	how many bytes to splice
   * @flags:	splice modifier flags
   * @actor:	handler that splices the data
   *
   * Description:
2933970b9   Miklos Szeredi   splice: remove i_...
903
   *    See __splice_from_pipe. This function locks the pipe inode,
932cc6d4f   Jens Axboe   splice: completel...
904
905
906
   *    otherwise it's identical to __splice_from_pipe().
   *
   */
6da618098   Mark Fasheh   [PATCH] Introduce...
907
908
909
910
911
  ssize_t splice_from_pipe(struct pipe_inode_info *pipe, struct file *out,
  			 loff_t *ppos, size_t len, unsigned int flags,
  			 splice_actor *actor)
  {
  	ssize_t ret;
c66ab6fa7   Jens Axboe   splice: abstract ...
912
913
914
915
  	struct splice_desc sd = {
  		.total_len = len,
  		.flags = flags,
  		.pos = *ppos,
6a14b90bb   Jens Axboe   vmsplice: add vms...
916
  		.u.file = out,
c66ab6fa7   Jens Axboe   splice: abstract ...
917
  	};
6da618098   Mark Fasheh   [PATCH] Introduce...
918

61e0d47c3   Miklos Szeredi   splice: add helpe...
919
  	pipe_lock(pipe);
c66ab6fa7   Jens Axboe   splice: abstract ...
920
  	ret = __splice_from_pipe(pipe, &sd, actor);
61e0d47c3   Miklos Szeredi   splice: add helpe...
921
  	pipe_unlock(pipe);
6da618098   Mark Fasheh   [PATCH] Introduce...
922
923
924
925
926
  
  	return ret;
  }
  
  /**
83f9135bd   Jens Axboe   [PATCH] splice: a...
927
   * generic_file_splice_write - splice data from a pipe to a file
3a326a2ce   Ingo Molnar   [PATCH] introduce...
928
   * @pipe:	pipe info
83f9135bd   Jens Axboe   [PATCH] splice: a...
929
   * @out:	file to write to
932cc6d4f   Jens Axboe   splice: completel...
930
   * @ppos:	position in @out
83f9135bd   Jens Axboe   [PATCH] splice: a...
931
932
933
   * @len:	number of bytes to splice
   * @flags:	splice modifier flags
   *
932cc6d4f   Jens Axboe   splice: completel...
934
935
936
   * Description:
   *    Will either move or copy pages (determined by @flags options) from
   *    the given pipe inode to the given file.
83f9135bd   Jens Axboe   [PATCH] splice: a...
937
938
   *
   */
3a326a2ce   Ingo Molnar   [PATCH] introduce...
939
940
  ssize_t
  generic_file_splice_write(struct pipe_inode_info *pipe, struct file *out,
cbb7e577e   Jens Axboe   [PATCH] splice: p...
941
  			  loff_t *ppos, size_t len, unsigned int flags)
5274f052e   Jens Axboe   [PATCH] Introduce...
942
  {
4f6f0bd2f   Jens Axboe   [PATCH] splice: i...
943
  	struct address_space *mapping = out->f_mapping;
8c34e2d63   Jens Axboe   [PATCH] Remove SU...
944
  	struct inode *inode = mapping->host;
7f3d4ee10   Miklos Szeredi   vfs: splice remov...
945
946
947
948
949
950
  	struct splice_desc sd = {
  		.total_len = len,
  		.flags = flags,
  		.pos = *ppos,
  		.u.file = out,
  	};
3a326a2ce   Ingo Molnar   [PATCH] introduce...
951
  	ssize_t ret;
61e0d47c3   Miklos Szeredi   splice: add helpe...
952
  	pipe_lock(pipe);
eb443e5a2   Miklos Szeredi   splice: fix i_mut...
953
954
955
956
957
958
959
960
961
  
  	splice_from_pipe_begin(&sd);
  	do {
  		ret = splice_from_pipe_next(pipe, &sd);
  		if (ret <= 0)
  			break;
  
  		mutex_lock_nested(&inode->i_mutex, I_MUTEX_CHILD);
  		ret = file_remove_suid(out);
723590ed5   Miklos Szeredi   splice: update mt...
962
963
  		if (!ret) {
  			file_update_time(out);
eb443e5a2   Miklos Szeredi   splice: fix i_mut...
964
  			ret = splice_from_pipe_feed(pipe, &sd, pipe_to_file);
723590ed5   Miklos Szeredi   splice: update mt...
965
  		}
eb443e5a2   Miklos Szeredi   splice: fix i_mut...
966
967
968
  		mutex_unlock(&inode->i_mutex);
  	} while (ret > 0);
  	splice_from_pipe_end(pipe, &sd);
61e0d47c3   Miklos Szeredi   splice: add helpe...
969
  	pipe_unlock(pipe);
eb443e5a2   Miklos Szeredi   splice: fix i_mut...
970
971
972
  
  	if (sd.num_spliced)
  		ret = sd.num_spliced;
a4514ebd8   Jens Axboe   [PATCH] splice: o...
973
  	if (ret > 0) {
17ee4f49a   Jens Axboe   splice: adjust ba...
974
  		unsigned long nr_pages;
148f948ba   Jan Kara   vfs: Introduce ne...
975
  		int err;
17ee4f49a   Jens Axboe   splice: adjust ba...
976

17ee4f49a   Jens Axboe   splice: adjust ba...
977
  		nr_pages = (ret + PAGE_CACHE_SIZE - 1) >> PAGE_CACHE_SHIFT;
a4514ebd8   Jens Axboe   [PATCH] splice: o...
978

148f948ba   Jan Kara   vfs: Introduce ne...
979
980
981
982
983
  		err = generic_write_sync(out, *ppos, ret);
  		if (err)
  			ret = err;
  		else
  			*ppos += ret;
17ee4f49a   Jens Axboe   splice: adjust ba...
984
  		balance_dirty_pages_ratelimited_nr(mapping, nr_pages);
4f6f0bd2f   Jens Axboe   [PATCH] splice: i...
985
986
987
  	}
  
  	return ret;
5274f052e   Jens Axboe   [PATCH] Introduce...
988
  }
059a8f373   Jens Axboe   [PATCH] splice: e...
989
  EXPORT_SYMBOL(generic_file_splice_write);
b2858d7d1   Miklos Szeredi   splice: fix kmaps...
990
991
  static int write_pipe_buf(struct pipe_inode_info *pipe, struct pipe_buffer *buf,
  			  struct splice_desc *sd)
0b0a47f5c   Miklos Szeredi   splice: implement...
992
  {
b2858d7d1   Miklos Szeredi   splice: fix kmaps...
993
994
  	int ret;
  	void *data;
b2858d7d1   Miklos Szeredi   splice: fix kmaps...
995
996
997
998
999
  	data = buf->ops->map(pipe, buf, 0);
  	ret = kernel_write(sd->u.file, data + buf->offset, sd->len, sd->pos);
  	buf->ops->unmap(pipe, buf, data);
  
  	return ret;
0b0a47f5c   Miklos Szeredi   splice: implement...
1000
1001
1002
1003
1004
1005
  }
  
  static ssize_t default_file_splice_write(struct pipe_inode_info *pipe,
  					 struct file *out, loff_t *ppos,
  					 size_t len, unsigned int flags)
  {
b2858d7d1   Miklos Szeredi   splice: fix kmaps...
1006
  	ssize_t ret;
0b0a47f5c   Miklos Szeredi   splice: implement...
1007

b2858d7d1   Miklos Szeredi   splice: fix kmaps...
1008
1009
1010
  	ret = splice_from_pipe(pipe, out, ppos, len, flags, write_pipe_buf);
  	if (ret > 0)
  		*ppos += ret;
0b0a47f5c   Miklos Szeredi   splice: implement...
1011

b2858d7d1   Miklos Szeredi   splice: fix kmaps...
1012
  	return ret;
0b0a47f5c   Miklos Szeredi   splice: implement...
1013
  }
83f9135bd   Jens Axboe   [PATCH] splice: a...
1014
1015
  /**
   * generic_splice_sendpage - splice data from a pipe to a socket
932cc6d4f   Jens Axboe   splice: completel...
1016
   * @pipe:	pipe to splice from
83f9135bd   Jens Axboe   [PATCH] splice: a...
1017
   * @out:	socket to write to
932cc6d4f   Jens Axboe   splice: completel...
1018
   * @ppos:	position in @out
83f9135bd   Jens Axboe   [PATCH] splice: a...
1019
1020
1021
   * @len:	number of bytes to splice
   * @flags:	splice modifier flags
   *
932cc6d4f   Jens Axboe   splice: completel...
1022
1023
1024
   * Description:
   *    Will send @len bytes from the pipe to a network socket. No data copying
   *    is involved.
83f9135bd   Jens Axboe   [PATCH] splice: a...
1025
1026
   *
   */
3a326a2ce   Ingo Molnar   [PATCH] introduce...
1027
  ssize_t generic_splice_sendpage(struct pipe_inode_info *pipe, struct file *out,
cbb7e577e   Jens Axboe   [PATCH] splice: p...
1028
  				loff_t *ppos, size_t len, unsigned int flags)
5274f052e   Jens Axboe   [PATCH] Introduce...
1029
  {
00522fb41   Jens Axboe   [PATCH] splice: r...
1030
  	return splice_from_pipe(pipe, out, ppos, len, flags, pipe_to_sendpage);
5274f052e   Jens Axboe   [PATCH] Introduce...
1031
  }
059a8f373   Jens Axboe   [PATCH] splice: e...
1032
  EXPORT_SYMBOL(generic_splice_sendpage);
a0f067802   Jeff Garzik   [PATCH] splice ex...
1033

83f9135bd   Jens Axboe   [PATCH] splice: a...
1034
1035
1036
  /*
   * Attempt to initiate a splice from pipe to file.
   */
3a326a2ce   Ingo Molnar   [PATCH] introduce...
1037
  static long do_splice_from(struct pipe_inode_info *pipe, struct file *out,
cbb7e577e   Jens Axboe   [PATCH] splice: p...
1038
  			   loff_t *ppos, size_t len, unsigned int flags)
5274f052e   Jens Axboe   [PATCH] Introduce...
1039
  {
0b0a47f5c   Miklos Szeredi   splice: implement...
1040
1041
  	ssize_t (*splice_write)(struct pipe_inode_info *, struct file *,
  				loff_t *, size_t, unsigned int);
5274f052e   Jens Axboe   [PATCH] Introduce...
1042
  	int ret;
49570e9b2   Jens Axboe   [PATCH] splice: u...
1043
  	if (unlikely(!(out->f_mode & FMODE_WRITE)))
5274f052e   Jens Axboe   [PATCH] Introduce...
1044
  		return -EBADF;
efc968d45   Linus Torvalds   Don't allow splic...
1045
1046
  	if (unlikely(out->f_flags & O_APPEND))
  		return -EINVAL;
cbb7e577e   Jens Axboe   [PATCH] splice: p...
1047
  	ret = rw_verify_area(WRITE, out, ppos, len);
5274f052e   Jens Axboe   [PATCH] Introduce...
1048
1049
  	if (unlikely(ret < 0))
  		return ret;
cc56f7de7   Changli Gao   sendfile(): check...
1050
1051
1052
  	if (out->f_op && out->f_op->splice_write)
  		splice_write = out->f_op->splice_write;
  	else
0b0a47f5c   Miklos Szeredi   splice: implement...
1053
1054
1055
  		splice_write = default_file_splice_write;
  
  	return splice_write(pipe, out, ppos, len, flags);
5274f052e   Jens Axboe   [PATCH] Introduce...
1056
  }
83f9135bd   Jens Axboe   [PATCH] splice: a...
1057
1058
1059
  /*
   * Attempt to initiate a splice from a file to a pipe.
   */
cbb7e577e   Jens Axboe   [PATCH] splice: p...
1060
1061
1062
  static long do_splice_to(struct file *in, loff_t *ppos,
  			 struct pipe_inode_info *pipe, size_t len,
  			 unsigned int flags)
5274f052e   Jens Axboe   [PATCH] Introduce...
1063
  {
6818173bd   Miklos Szeredi   splice: implement...
1064
1065
  	ssize_t (*splice_read)(struct file *, loff_t *,
  			       struct pipe_inode_info *, size_t, unsigned int);
5274f052e   Jens Axboe   [PATCH] Introduce...
1066
  	int ret;
49570e9b2   Jens Axboe   [PATCH] splice: u...
1067
  	if (unlikely(!(in->f_mode & FMODE_READ)))
5274f052e   Jens Axboe   [PATCH] Introduce...
1068
  		return -EBADF;
cbb7e577e   Jens Axboe   [PATCH] splice: p...
1069
  	ret = rw_verify_area(READ, in, ppos, len);
5274f052e   Jens Axboe   [PATCH] Introduce...
1070
1071
  	if (unlikely(ret < 0))
  		return ret;
cc56f7de7   Changli Gao   sendfile(): check...
1072
1073
1074
  	if (in->f_op && in->f_op->splice_read)
  		splice_read = in->f_op->splice_read;
  	else
6818173bd   Miklos Szeredi   splice: implement...
1075
1076
1077
  		splice_read = default_file_splice_read;
  
  	return splice_read(in, ppos, pipe, len, flags);
5274f052e   Jens Axboe   [PATCH] Introduce...
1078
  }
932cc6d4f   Jens Axboe   splice: completel...
1079
1080
1081
1082
1083
1084
1085
1086
1087
  /**
   * splice_direct_to_actor - splices data directly between two non-pipes
   * @in:		file to splice from
   * @sd:		actor information on where to splice to
   * @actor:	handles the data splicing
   *
   * Description:
   *    This is a special case helper to splice directly between two
   *    points, without requiring an explicit pipe. Internally an allocated
79685b8de   Randy Dunlap   docbook: add pipe...
1088
   *    pipe is cached in the process, and reused during the lifetime of
932cc6d4f   Jens Axboe   splice: completel...
1089
1090
   *    that process.
   *
c66ab6fa7   Jens Axboe   splice: abstract ...
1091
1092
1093
   */
  ssize_t splice_direct_to_actor(struct file *in, struct splice_desc *sd,
  			       splice_direct_actor *actor)
b92ce5589   Jens Axboe   [PATCH] splice: a...
1094
1095
1096
1097
  {
  	struct pipe_inode_info *pipe;
  	long ret, bytes;
  	umode_t i_mode;
c66ab6fa7   Jens Axboe   splice: abstract ...
1098
1099
  	size_t len;
  	int i, flags;
b92ce5589   Jens Axboe   [PATCH] splice: a...
1100
1101
1102
1103
1104
1105
  
  	/*
  	 * We require the input being a regular file, as we don't want to
  	 * randomly drop data for eg socket -> socket splicing. Use the
  	 * piped splicing for that!
  	 */
0f7fc9e4d   Josef "Jeff" Sipek   [PATCH] VFS: chan...
1106
  	i_mode = in->f_path.dentry->d_inode->i_mode;
b92ce5589   Jens Axboe   [PATCH] splice: a...
1107
1108
1109
1110
1111
1112
1113
1114
  	if (unlikely(!S_ISREG(i_mode) && !S_ISBLK(i_mode)))
  		return -EINVAL;
  
  	/*
  	 * neither in nor out is a pipe, setup an internal pipe attached to
  	 * 'out' and transfer the wanted data from 'in' to 'out' through that
  	 */
  	pipe = current->splice_pipe;
49570e9b2   Jens Axboe   [PATCH] splice: u...
1115
  	if (unlikely(!pipe)) {
b92ce5589   Jens Axboe   [PATCH] splice: a...
1116
1117
1118
1119
1120
1121
  		pipe = alloc_pipe_info(NULL);
  		if (!pipe)
  			return -ENOMEM;
  
  		/*
  		 * We don't have an immediate reader, but we'll read the stuff
00522fb41   Jens Axboe   [PATCH] splice: r...
1122
  		 * out of the pipe right after the splice_to_pipe(). So set
b92ce5589   Jens Axboe   [PATCH] splice: a...
1123
1124
1125
1126
1127
1128
1129
1130
  		 * PIPE_READERS appropriately.
  		 */
  		pipe->readers = 1;
  
  		current->splice_pipe = pipe;
  	}
  
  	/*
73d62d83e   Ingo Molnar   [PATCH] splice: c...
1131
  	 * Do the splice.
b92ce5589   Jens Axboe   [PATCH] splice: a...
1132
1133
1134
  	 */
  	ret = 0;
  	bytes = 0;
c66ab6fa7   Jens Axboe   splice: abstract ...
1135
1136
1137
1138
1139
1140
1141
  	len = sd->total_len;
  	flags = sd->flags;
  
  	/*
  	 * Don't block on output, we have to drain the direct pipe.
  	 */
  	sd->flags &= ~SPLICE_F_NONBLOCK;
b92ce5589   Jens Axboe   [PATCH] splice: a...
1142
1143
  
  	while (len) {
51a92c0f6   Jens Axboe   splice: fix offse...
1144
  		size_t read_len;
a82c53a0e   Tom Zanussi   splice: fix sendf...
1145
  		loff_t pos = sd->pos, prev_pos = pos;
b92ce5589   Jens Axboe   [PATCH] splice: a...
1146

bcd4f3acb   Jens Axboe   splice: direct sp...
1147
  		ret = do_splice_to(in, &pos, pipe, len, flags);
51a92c0f6   Jens Axboe   splice: fix offse...
1148
  		if (unlikely(ret <= 0))
b92ce5589   Jens Axboe   [PATCH] splice: a...
1149
1150
1151
  			goto out_release;
  
  		read_len = ret;
c66ab6fa7   Jens Axboe   splice: abstract ...
1152
  		sd->total_len = read_len;
b92ce5589   Jens Axboe   [PATCH] splice: a...
1153
1154
1155
1156
1157
1158
  
  		/*
  		 * NOTE: nonblocking mode only applies to the input. We
  		 * must not do the output in nonblocking mode as then we
  		 * could get stuck data in the internal pipe:
  		 */
c66ab6fa7   Jens Axboe   splice: abstract ...
1159
  		ret = actor(pipe, sd);
a82c53a0e   Tom Zanussi   splice: fix sendf...
1160
1161
  		if (unlikely(ret <= 0)) {
  			sd->pos = prev_pos;
b92ce5589   Jens Axboe   [PATCH] splice: a...
1162
  			goto out_release;
a82c53a0e   Tom Zanussi   splice: fix sendf...
1163
  		}
b92ce5589   Jens Axboe   [PATCH] splice: a...
1164
1165
1166
  
  		bytes += ret;
  		len -= ret;
bcd4f3acb   Jens Axboe   splice: direct sp...
1167
  		sd->pos = pos;
b92ce5589   Jens Axboe   [PATCH] splice: a...
1168

a82c53a0e   Tom Zanussi   splice: fix sendf...
1169
1170
  		if (ret < read_len) {
  			sd->pos = prev_pos + ret;
51a92c0f6   Jens Axboe   splice: fix offse...
1171
  			goto out_release;
a82c53a0e   Tom Zanussi   splice: fix sendf...
1172
  		}
b92ce5589   Jens Axboe   [PATCH] splice: a...
1173
  	}
9e97198db   Jens Axboe   splice: fix probl...
1174
  done:
b92ce5589   Jens Axboe   [PATCH] splice: a...
1175
  	pipe->nrbufs = pipe->curbuf = 0;
808487085   Jens Axboe   splice: always up...
1176
  	file_accessed(in);
b92ce5589   Jens Axboe   [PATCH] splice: a...
1177
1178
1179
1180
1181
1182
1183
  	return bytes;
  
  out_release:
  	/*
  	 * If we did an incomplete transfer we must release
  	 * the pipe buffers in question:
  	 */
35f3d14db   Jens Axboe   pipe: add support...
1184
  	for (i = 0; i < pipe->buffers; i++) {
b92ce5589   Jens Axboe   [PATCH] splice: a...
1185
1186
1187
1188
1189
1190
1191
  		struct pipe_buffer *buf = pipe->bufs + i;
  
  		if (buf->ops) {
  			buf->ops->release(pipe, buf);
  			buf->ops = NULL;
  		}
  	}
b92ce5589   Jens Axboe   [PATCH] splice: a...
1192

9e97198db   Jens Axboe   splice: fix probl...
1193
1194
  	if (!bytes)
  		bytes = ret;
c66ab6fa7   Jens Axboe   splice: abstract ...
1195

9e97198db   Jens Axboe   splice: fix probl...
1196
  	goto done;
c66ab6fa7   Jens Axboe   splice: abstract ...
1197
1198
1199
1200
1201
1202
  }
  EXPORT_SYMBOL(splice_direct_to_actor);
  
  static int direct_splice_actor(struct pipe_inode_info *pipe,
  			       struct splice_desc *sd)
  {
6a14b90bb   Jens Axboe   vmsplice: add vms...
1203
  	struct file *file = sd->u.file;
c66ab6fa7   Jens Axboe   splice: abstract ...
1204

2cb4b05e7   Changli Gao   splice: direct_sp...
1205
1206
  	return do_splice_from(pipe, file, &file->f_pos, sd->total_len,
  			      sd->flags);
c66ab6fa7   Jens Axboe   splice: abstract ...
1207
  }
932cc6d4f   Jens Axboe   splice: completel...
1208
1209
1210
1211
1212
1213
1214
1215
1216
1217
1218
1219
1220
1221
1222
  /**
   * do_splice_direct - splices data directly between two files
   * @in:		file to splice from
   * @ppos:	input file offset
   * @out:	file to splice to
   * @len:	number of bytes to splice
   * @flags:	splice modifier flags
   *
   * Description:
   *    For use by do_sendfile(). splice can easily emulate sendfile, but
   *    doing it in the application would incur an extra system call
   *    (splice in + splice out, as compared to just sendfile()). So this helper
   *    can splice directly through a process-private pipe.
   *
   */
c66ab6fa7   Jens Axboe   splice: abstract ...
1223
1224
1225
1226
1227
1228
1229
1230
  long do_splice_direct(struct file *in, loff_t *ppos, struct file *out,
  		      size_t len, unsigned int flags)
  {
  	struct splice_desc sd = {
  		.len		= len,
  		.total_len	= len,
  		.flags		= flags,
  		.pos		= *ppos,
6a14b90bb   Jens Axboe   vmsplice: add vms...
1231
  		.u.file		= out,
c66ab6fa7   Jens Axboe   splice: abstract ...
1232
  	};
51a92c0f6   Jens Axboe   splice: fix offse...
1233
  	long ret;
c66ab6fa7   Jens Axboe   splice: abstract ...
1234
1235
  
  	ret = splice_direct_to_actor(in, &sd, direct_splice_actor);
51a92c0f6   Jens Axboe   splice: fix offse...
1236
  	if (ret > 0)
a82c53a0e   Tom Zanussi   splice: fix sendf...
1237
  		*ppos = sd.pos;
51a92c0f6   Jens Axboe   splice: fix offse...
1238

c66ab6fa7   Jens Axboe   splice: abstract ...
1239
  	return ret;
b92ce5589   Jens Axboe   [PATCH] splice: a...
1240
  }
7c77f0b3f   Miklos Szeredi   splice: implement...
1241
1242
1243
  static int splice_pipe_to_pipe(struct pipe_inode_info *ipipe,
  			       struct pipe_inode_info *opipe,
  			       size_t len, unsigned int flags);
ddac0d39c   Jens Axboe   [PATCH] splice: f...
1244
1245
  
  /*
83f9135bd   Jens Axboe   [PATCH] splice: a...
1246
1247
   * Determine where to splice to/from.
   */
529565dcb   Ingo Molnar   [PATCH] splice: a...
1248
1249
1250
  static long do_splice(struct file *in, loff_t __user *off_in,
  		      struct file *out, loff_t __user *off_out,
  		      size_t len, unsigned int flags)
5274f052e   Jens Axboe   [PATCH] Introduce...
1251
  {
7c77f0b3f   Miklos Szeredi   splice: implement...
1252
1253
  	struct pipe_inode_info *ipipe;
  	struct pipe_inode_info *opipe;
cbb7e577e   Jens Axboe   [PATCH] splice: p...
1254
  	loff_t offset, *off;
a4514ebd8   Jens Axboe   [PATCH] splice: o...
1255
  	long ret;
5274f052e   Jens Axboe   [PATCH] Introduce...
1256

71993e62a   Linus Torvalds   Rename 'pipe_info...
1257
1258
  	ipipe = get_pipe_info(in);
  	opipe = get_pipe_info(out);
7c77f0b3f   Miklos Szeredi   splice: implement...
1259
1260
1261
1262
1263
1264
1265
1266
1267
1268
1269
1270
1271
1272
1273
1274
1275
1276
1277
  
  	if (ipipe && opipe) {
  		if (off_in || off_out)
  			return -ESPIPE;
  
  		if (!(in->f_mode & FMODE_READ))
  			return -EBADF;
  
  		if (!(out->f_mode & FMODE_WRITE))
  			return -EBADF;
  
  		/* Splicing to self would be fun, but... */
  		if (ipipe == opipe)
  			return -EINVAL;
  
  		return splice_pipe_to_pipe(ipipe, opipe, len, flags);
  	}
  
  	if (ipipe) {
529565dcb   Ingo Molnar   [PATCH] splice: a...
1278
1279
  		if (off_in)
  			return -ESPIPE;
b92ce5589   Jens Axboe   [PATCH] splice: a...
1280
  		if (off_out) {
19c9a49b4   Changli Gao   splice: check f_m...
1281
  			if (!(out->f_mode & FMODE_PWRITE))
b92ce5589   Jens Axboe   [PATCH] splice: a...
1282
  				return -EINVAL;
cbb7e577e   Jens Axboe   [PATCH] splice: p...
1283
  			if (copy_from_user(&offset, off_out, sizeof(loff_t)))
b92ce5589   Jens Axboe   [PATCH] splice: a...
1284
  				return -EFAULT;
cbb7e577e   Jens Axboe   [PATCH] splice: p...
1285
1286
1287
  			off = &offset;
  		} else
  			off = &out->f_pos;
529565dcb   Ingo Molnar   [PATCH] splice: a...
1288

7c77f0b3f   Miklos Szeredi   splice: implement...
1289
  		ret = do_splice_from(ipipe, out, off, len, flags);
a4514ebd8   Jens Axboe   [PATCH] splice: o...
1290
1291
1292
1293
1294
  
  		if (off_out && copy_to_user(off_out, off, sizeof(loff_t)))
  			ret = -EFAULT;
  
  		return ret;
529565dcb   Ingo Molnar   [PATCH] splice: a...
1295
  	}
5274f052e   Jens Axboe   [PATCH] Introduce...
1296

7c77f0b3f   Miklos Szeredi   splice: implement...
1297
  	if (opipe) {
529565dcb   Ingo Molnar   [PATCH] splice: a...
1298
1299
  		if (off_out)
  			return -ESPIPE;
b92ce5589   Jens Axboe   [PATCH] splice: a...
1300
  		if (off_in) {
19c9a49b4   Changli Gao   splice: check f_m...
1301
  			if (!(in->f_mode & FMODE_PREAD))
b92ce5589   Jens Axboe   [PATCH] splice: a...
1302
  				return -EINVAL;
cbb7e577e   Jens Axboe   [PATCH] splice: p...
1303
  			if (copy_from_user(&offset, off_in, sizeof(loff_t)))
b92ce5589   Jens Axboe   [PATCH] splice: a...
1304
  				return -EFAULT;
cbb7e577e   Jens Axboe   [PATCH] splice: p...
1305
1306
1307
  			off = &offset;
  		} else
  			off = &in->f_pos;
529565dcb   Ingo Molnar   [PATCH] splice: a...
1308

7c77f0b3f   Miklos Szeredi   splice: implement...
1309
  		ret = do_splice_to(in, off, opipe, len, flags);
a4514ebd8   Jens Axboe   [PATCH] splice: o...
1310
1311
1312
1313
1314
  
  		if (off_in && copy_to_user(off_in, off, sizeof(loff_t)))
  			ret = -EFAULT;
  
  		return ret;
529565dcb   Ingo Molnar   [PATCH] splice: a...
1315
  	}
5274f052e   Jens Axboe   [PATCH] Introduce...
1316
1317
1318
  
  	return -EINVAL;
  }
912d35f86   Jens Axboe   [PATCH] Add suppo...
1319
1320
1321
1322
1323
1324
1325
1326
1327
  /*
   * Map an iov into an array of pages and offset/length tupples. With the
   * partial_page structure, we can map several non-contiguous ranges into
   * our ones pages[] map instead of splitting that operation into pieces.
   * Could easily be exported as a generic helper for other users, in which
   * case one would probably want to add a 'max_nr_pages' parameter as well.
   */
  static int get_iovec_page_array(const struct iovec __user *iov,
  				unsigned int nr_vecs, struct page **pages,
35f3d14db   Jens Axboe   pipe: add support...
1328
1329
  				struct partial_page *partial, int aligned,
  				unsigned int pipe_buffers)
912d35f86   Jens Axboe   [PATCH] Add suppo...
1330
1331
  {
  	int buffers = 0, error = 0;
912d35f86   Jens Axboe   [PATCH] Add suppo...
1332
1333
  	while (nr_vecs) {
  		unsigned long off, npages;
757239576   Linus Torvalds   Fix possible spli...
1334
  		struct iovec entry;
912d35f86   Jens Axboe   [PATCH] Add suppo...
1335
1336
1337
  		void __user *base;
  		size_t len;
  		int i;
757239576   Linus Torvalds   Fix possible spli...
1338
  		error = -EFAULT;
bc40d73c9   Nick Piggin   splice: use get_u...
1339
  		if (copy_from_user(&entry, iov, sizeof(entry)))
912d35f86   Jens Axboe   [PATCH] Add suppo...
1340
  			break;
757239576   Linus Torvalds   Fix possible spli...
1341
1342
  		base = entry.iov_base;
  		len = entry.iov_len;
912d35f86   Jens Axboe   [PATCH] Add suppo...
1343
1344
1345
  		/*
  		 * Sanity check this iovec. 0 read succeeds.
  		 */
757239576   Linus Torvalds   Fix possible spli...
1346
  		error = 0;
912d35f86   Jens Axboe   [PATCH] Add suppo...
1347
1348
1349
  		if (unlikely(!len))
  			break;
  		error = -EFAULT;
712a30e63   Bastian Blank   splice: fix user ...
1350
  		if (!access_ok(VERIFY_READ, base, len))
912d35f86   Jens Axboe   [PATCH] Add suppo...
1351
1352
1353
1354
1355
1356
1357
  			break;
  
  		/*
  		 * Get this base offset and number of pages, then map
  		 * in the user pages.
  		 */
  		off = (unsigned long) base & ~PAGE_MASK;
7afa6fd03   Jens Axboe   [PATCH] vmsplice:...
1358
1359
1360
1361
1362
1363
1364
1365
  
  		/*
  		 * If asked for alignment, the offset must be zero and the
  		 * length a multiple of the PAGE_SIZE.
  		 */
  		error = -EINVAL;
  		if (aligned && (off || len & ~PAGE_MASK))
  			break;
912d35f86   Jens Axboe   [PATCH] Add suppo...
1366
  		npages = (off + len + PAGE_SIZE - 1) >> PAGE_SHIFT;
35f3d14db   Jens Axboe   pipe: add support...
1367
1368
  		if (npages > pipe_buffers - buffers)
  			npages = pipe_buffers - buffers;
912d35f86   Jens Axboe   [PATCH] Add suppo...
1369

bc40d73c9   Nick Piggin   splice: use get_u...
1370
1371
  		error = get_user_pages_fast((unsigned long)base, npages,
  					0, &pages[buffers]);
912d35f86   Jens Axboe   [PATCH] Add suppo...
1372
1373
1374
1375
1376
1377
1378
1379
  
  		if (unlikely(error <= 0))
  			break;
  
  		/*
  		 * Fill this contiguous range into the partial page map.
  		 */
  		for (i = 0; i < error; i++) {
7591489a8   Jens Axboe   [PATCH] vmsplice:...
1380
  			const int plen = min_t(size_t, len, PAGE_SIZE - off);
912d35f86   Jens Axboe   [PATCH] Add suppo...
1381
1382
1383
1384
1385
1386
1387
1388
1389
1390
1391
1392
1393
1394
1395
1396
1397
1398
1399
1400
1401
1402
  
  			partial[buffers].offset = off;
  			partial[buffers].len = plen;
  
  			off = 0;
  			len -= plen;
  			buffers++;
  		}
  
  		/*
  		 * We didn't complete this iov, stop here since it probably
  		 * means we have to move some of this into a pipe to
  		 * be able to continue.
  		 */
  		if (len)
  			break;
  
  		/*
  		 * Don't continue if we mapped fewer pages than we asked for,
  		 * or if we mapped the max number of pages that we have
  		 * room for.
  		 */
35f3d14db   Jens Axboe   pipe: add support...
1403
  		if (error < npages || buffers == pipe_buffers)
912d35f86   Jens Axboe   [PATCH] Add suppo...
1404
1405
1406
1407
1408
  			break;
  
  		nr_vecs--;
  		iov++;
  	}
912d35f86   Jens Axboe   [PATCH] Add suppo...
1409
1410
1411
1412
1413
  	if (buffers)
  		return buffers;
  
  	return error;
  }
6a14b90bb   Jens Axboe   vmsplice: add vms...
1414
1415
1416
1417
1418
  static int pipe_to_user(struct pipe_inode_info *pipe, struct pipe_buffer *buf,
  			struct splice_desc *sd)
  {
  	char *src;
  	int ret;
6a14b90bb   Jens Axboe   vmsplice: add vms...
1419
1420
1421
1422
1423
1424
1425
1426
1427
1428
1429
1430
1431
1432
1433
1434
1435
1436
1437
1438
1439
1440
1441
  	/*
  	 * See if we can use the atomic maps, by prefaulting in the
  	 * pages and doing an atomic copy
  	 */
  	if (!fault_in_pages_writeable(sd->u.userptr, sd->len)) {
  		src = buf->ops->map(pipe, buf, 1);
  		ret = __copy_to_user_inatomic(sd->u.userptr, src + buf->offset,
  							sd->len);
  		buf->ops->unmap(pipe, buf, src);
  		if (!ret) {
  			ret = sd->len;
  			goto out;
  		}
  	}
  
  	/*
  	 * No dice, use slow non-atomic map and copy
   	 */
  	src = buf->ops->map(pipe, buf, 0);
  
  	ret = sd->len;
  	if (copy_to_user(sd->u.userptr, src + buf->offset, sd->len))
  		ret = -EFAULT;
6866bef40   Jens Axboe   splice: fix doubl...
1442
  	buf->ops->unmap(pipe, buf, src);
6a14b90bb   Jens Axboe   vmsplice: add vms...
1443
1444
1445
  out:
  	if (ret > 0)
  		sd->u.userptr += ret;
6a14b90bb   Jens Axboe   vmsplice: add vms...
1446
1447
1448
1449
1450
1451
1452
1453
1454
1455
1456
1457
1458
1459
1460
  	return ret;
  }
  
  /*
   * For lack of a better implementation, implement vmsplice() to userspace
   * as a simple copy of the pipes pages to the user iov.
   */
  static long vmsplice_to_user(struct file *file, const struct iovec __user *iov,
  			     unsigned long nr_segs, unsigned int flags)
  {
  	struct pipe_inode_info *pipe;
  	struct splice_desc sd;
  	ssize_t size;
  	int error;
  	long ret;
71993e62a   Linus Torvalds   Rename 'pipe_info...
1461
  	pipe = get_pipe_info(file);
6a14b90bb   Jens Axboe   vmsplice: add vms...
1462
1463
  	if (!pipe)
  		return -EBADF;
61e0d47c3   Miklos Szeredi   splice: add helpe...
1464
  	pipe_lock(pipe);
6a14b90bb   Jens Axboe   vmsplice: add vms...
1465
1466
1467
1468
1469
1470
1471
1472
1473
1474
1475
1476
1477
1478
1479
1480
1481
1482
1483
1484
1485
1486
1487
1488
1489
  
  	error = ret = 0;
  	while (nr_segs) {
  		void __user *base;
  		size_t len;
  
  		/*
  		 * Get user address base and length for this iovec.
  		 */
  		error = get_user(base, &iov->iov_base);
  		if (unlikely(error))
  			break;
  		error = get_user(len, &iov->iov_len);
  		if (unlikely(error))
  			break;
  
  		/*
  		 * Sanity check this iovec. 0 read succeeds.
  		 */
  		if (unlikely(!len))
  			break;
  		if (unlikely(!base)) {
  			error = -EFAULT;
  			break;
  		}
8811930dc   Jens Axboe   splice: missing u...
1490
1491
1492
1493
  		if (unlikely(!access_ok(VERIFY_WRITE, base, len))) {
  			error = -EFAULT;
  			break;
  		}
6a14b90bb   Jens Axboe   vmsplice: add vms...
1494
1495
1496
1497
1498
1499
1500
1501
1502
1503
1504
1505
1506
1507
1508
1509
1510
1511
1512
1513
1514
1515
  		sd.len = 0;
  		sd.total_len = len;
  		sd.flags = flags;
  		sd.u.userptr = base;
  		sd.pos = 0;
  
  		size = __splice_from_pipe(pipe, &sd, pipe_to_user);
  		if (size < 0) {
  			if (!ret)
  				ret = size;
  
  			break;
  		}
  
  		ret += size;
  
  		if (size < len)
  			break;
  
  		nr_segs--;
  		iov++;
  	}
61e0d47c3   Miklos Szeredi   splice: add helpe...
1516
  	pipe_unlock(pipe);
6a14b90bb   Jens Axboe   vmsplice: add vms...
1517
1518
1519
1520
1521
1522
  
  	if (!ret)
  		ret = error;
  
  	return ret;
  }
912d35f86   Jens Axboe   [PATCH] Add suppo...
1523
1524
1525
1526
  /*
   * vmsplice splices a user address range into a pipe. It can be thought of
   * as splice-from-memory, where the regular splice is splice-from-file (or
   * to file). In both cases the output is a pipe, naturally.
912d35f86   Jens Axboe   [PATCH] Add suppo...
1527
   */
6a14b90bb   Jens Axboe   vmsplice: add vms...
1528
1529
  static long vmsplice_to_pipe(struct file *file, const struct iovec __user *iov,
  			     unsigned long nr_segs, unsigned int flags)
912d35f86   Jens Axboe   [PATCH] Add suppo...
1530
  {
ddac0d39c   Jens Axboe   [PATCH] splice: f...
1531
  	struct pipe_inode_info *pipe;
35f3d14db   Jens Axboe   pipe: add support...
1532
1533
  	struct page *pages[PIPE_DEF_BUFFERS];
  	struct partial_page partial[PIPE_DEF_BUFFERS];
912d35f86   Jens Axboe   [PATCH] Add suppo...
1534
1535
1536
1537
1538
  	struct splice_pipe_desc spd = {
  		.pages = pages,
  		.partial = partial,
  		.flags = flags,
  		.ops = &user_page_pipe_buf_ops,
bbdfc2f70   Jens Axboe   [SPLICE]: Don't a...
1539
  		.spd_release = spd_release_page,
912d35f86   Jens Axboe   [PATCH] Add suppo...
1540
  	};
35f3d14db   Jens Axboe   pipe: add support...
1541
  	long ret;
912d35f86   Jens Axboe   [PATCH] Add suppo...
1542

71993e62a   Linus Torvalds   Rename 'pipe_info...
1543
  	pipe = get_pipe_info(file);
ddac0d39c   Jens Axboe   [PATCH] splice: f...
1544
  	if (!pipe)
912d35f86   Jens Axboe   [PATCH] Add suppo...
1545
  		return -EBADF;
912d35f86   Jens Axboe   [PATCH] Add suppo...
1546

35f3d14db   Jens Axboe   pipe: add support...
1547
1548
1549
1550
1551
1552
  	if (splice_grow_spd(pipe, &spd))
  		return -ENOMEM;
  
  	spd.nr_pages = get_iovec_page_array(iov, nr_segs, spd.pages,
  					    spd.partial, flags & SPLICE_F_GIFT,
  					    pipe->buffers);
912d35f86   Jens Axboe   [PATCH] Add suppo...
1553
  	if (spd.nr_pages <= 0)
35f3d14db   Jens Axboe   pipe: add support...
1554
1555
1556
  		ret = spd.nr_pages;
  	else
  		ret = splice_to_pipe(pipe, &spd);
912d35f86   Jens Axboe   [PATCH] Add suppo...
1557

35f3d14db   Jens Axboe   pipe: add support...
1558
1559
  	splice_shrink_spd(pipe, &spd);
  	return ret;
912d35f86   Jens Axboe   [PATCH] Add suppo...
1560
  }
6a14b90bb   Jens Axboe   vmsplice: add vms...
1561
1562
1563
1564
1565
1566
1567
1568
1569
1570
1571
1572
1573
1574
1575
1576
  /*
   * Note that vmsplice only really supports true splicing _from_ user memory
   * to a pipe, not the other way around. Splicing from user memory is a simple
   * operation that can be supported without any funky alignment restrictions
   * or nasty vm tricks. We simply map in the user memory and fill them into
   * a pipe. The reverse isn't quite as easy, though. There are two possible
   * solutions for that:
   *
   *	- memcpy() the data internally, at which point we might as well just
   *	  do a regular read() on the buffer anyway.
   *	- Lots of nasty vm tricks, that are neither fast nor flexible (it
   *	  has restriction limitations on both ends of the pipe).
   *
   * Currently we punt and implement it as a normal copy, see pipe_to_user().
   *
   */
836f92adf   Heiko Carstens   [CVE-2009-0029] S...
1577
1578
  SYSCALL_DEFINE4(vmsplice, int, fd, const struct iovec __user *, iov,
  		unsigned long, nr_segs, unsigned int, flags)
912d35f86   Jens Axboe   [PATCH] Add suppo...
1579
1580
1581
1582
  {
  	struct file *file;
  	long error;
  	int fput;
6a14b90bb   Jens Axboe   vmsplice: add vms...
1583
1584
1585
1586
  	if (unlikely(nr_segs > UIO_MAXIOV))
  		return -EINVAL;
  	else if (unlikely(!nr_segs))
  		return 0;
912d35f86   Jens Axboe   [PATCH] Add suppo...
1587
1588
1589
1590
  	error = -EBADF;
  	file = fget_light(fd, &fput);
  	if (file) {
  		if (file->f_mode & FMODE_WRITE)
6a14b90bb   Jens Axboe   vmsplice: add vms...
1591
1592
1593
  			error = vmsplice_to_pipe(file, iov, nr_segs, flags);
  		else if (file->f_mode & FMODE_READ)
  			error = vmsplice_to_user(file, iov, nr_segs, flags);
912d35f86   Jens Axboe   [PATCH] Add suppo...
1594
1595
1596
1597
1598
1599
  
  		fput_light(file, fput);
  	}
  
  	return error;
  }
836f92adf   Heiko Carstens   [CVE-2009-0029] S...
1600
1601
1602
  SYSCALL_DEFINE6(splice, int, fd_in, loff_t __user *, off_in,
  		int, fd_out, loff_t __user *, off_out,
  		size_t, len, unsigned int, flags)
5274f052e   Jens Axboe   [PATCH] Introduce...
1603
1604
1605
1606
1607
1608
1609
1610
1611
  {
  	long error;
  	struct file *in, *out;
  	int fput_in, fput_out;
  
  	if (unlikely(!len))
  		return 0;
  
  	error = -EBADF;
529565dcb   Ingo Molnar   [PATCH] splice: a...
1612
  	in = fget_light(fd_in, &fput_in);
5274f052e   Jens Axboe   [PATCH] Introduce...
1613
1614
  	if (in) {
  		if (in->f_mode & FMODE_READ) {
529565dcb   Ingo Molnar   [PATCH] splice: a...
1615
  			out = fget_light(fd_out, &fput_out);
5274f052e   Jens Axboe   [PATCH] Introduce...
1616
1617
  			if (out) {
  				if (out->f_mode & FMODE_WRITE)
529565dcb   Ingo Molnar   [PATCH] splice: a...
1618
1619
1620
  					error = do_splice(in, off_in,
  							  out, off_out,
  							  len, flags);
5274f052e   Jens Axboe   [PATCH] Introduce...
1621
1622
1623
1624
1625
1626
1627
1628
1629
  				fput_light(out, fput_out);
  			}
  		}
  
  		fput_light(in, fput_in);
  	}
  
  	return error;
  }
70524490e   Jens Axboe   [PATCH] splice: a...
1630
1631
  
  /*
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1632
1633
1634
   * Make sure there's data to read. Wait for input if we can, otherwise
   * return an appropriate error.
   */
7c77f0b3f   Miklos Szeredi   splice: implement...
1635
  static int ipipe_prep(struct pipe_inode_info *pipe, unsigned int flags)
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1636
1637
1638
1639
1640
1641
1642
1643
1644
1645
1646
  {
  	int ret;
  
  	/*
  	 * Check ->nrbufs without the inode lock first. This function
  	 * is speculative anyways, so missing one is ok.
  	 */
  	if (pipe->nrbufs)
  		return 0;
  
  	ret = 0;
61e0d47c3   Miklos Szeredi   splice: add helpe...
1647
  	pipe_lock(pipe);
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1648
1649
1650
1651
1652
1653
1654
1655
1656
1657
1658
1659
1660
1661
1662
1663
  
  	while (!pipe->nrbufs) {
  		if (signal_pending(current)) {
  			ret = -ERESTARTSYS;
  			break;
  		}
  		if (!pipe->writers)
  			break;
  		if (!pipe->waiting_writers) {
  			if (flags & SPLICE_F_NONBLOCK) {
  				ret = -EAGAIN;
  				break;
  			}
  		}
  		pipe_wait(pipe);
  	}
61e0d47c3   Miklos Szeredi   splice: add helpe...
1664
  	pipe_unlock(pipe);
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1665
1666
1667
1668
1669
1670
1671
  	return ret;
  }
  
  /*
   * Make sure there's writeable room. Wait for room if we can, otherwise
   * return an appropriate error.
   */
7c77f0b3f   Miklos Szeredi   splice: implement...
1672
  static int opipe_prep(struct pipe_inode_info *pipe, unsigned int flags)
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1673
1674
1675
1676
1677
1678
1679
  {
  	int ret;
  
  	/*
  	 * Check ->nrbufs without the inode lock first. This function
  	 * is speculative anyways, so missing one is ok.
  	 */
35f3d14db   Jens Axboe   pipe: add support...
1680
  	if (pipe->nrbufs < pipe->buffers)
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1681
1682
1683
  		return 0;
  
  	ret = 0;
61e0d47c3   Miklos Szeredi   splice: add helpe...
1684
  	pipe_lock(pipe);
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1685

35f3d14db   Jens Axboe   pipe: add support...
1686
  	while (pipe->nrbufs >= pipe->buffers) {
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1687
1688
1689
1690
1691
1692
1693
1694
1695
1696
1697
1698
1699
1700
1701
1702
1703
  		if (!pipe->readers) {
  			send_sig(SIGPIPE, current, 0);
  			ret = -EPIPE;
  			break;
  		}
  		if (flags & SPLICE_F_NONBLOCK) {
  			ret = -EAGAIN;
  			break;
  		}
  		if (signal_pending(current)) {
  			ret = -ERESTARTSYS;
  			break;
  		}
  		pipe->waiting_writers++;
  		pipe_wait(pipe);
  		pipe->waiting_writers--;
  	}
61e0d47c3   Miklos Szeredi   splice: add helpe...
1704
  	pipe_unlock(pipe);
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1705
1706
1707
1708
  	return ret;
  }
  
  /*
7c77f0b3f   Miklos Szeredi   splice: implement...
1709
1710
1711
1712
1713
1714
1715
1716
1717
1718
1719
1720
1721
1722
1723
1724
1725
1726
1727
1728
1729
1730
1731
1732
1733
1734
1735
1736
1737
1738
1739
1740
1741
1742
1743
1744
1745
1746
1747
1748
1749
1750
   * Splice contents of ipipe to opipe.
   */
  static int splice_pipe_to_pipe(struct pipe_inode_info *ipipe,
  			       struct pipe_inode_info *opipe,
  			       size_t len, unsigned int flags)
  {
  	struct pipe_buffer *ibuf, *obuf;
  	int ret = 0, nbuf;
  	bool input_wakeup = false;
  
  
  retry:
  	ret = ipipe_prep(ipipe, flags);
  	if (ret)
  		return ret;
  
  	ret = opipe_prep(opipe, flags);
  	if (ret)
  		return ret;
  
  	/*
  	 * Potential ABBA deadlock, work around it by ordering lock
  	 * grabbing by pipe info address. Otherwise two different processes
  	 * could deadlock (one doing tee from A -> B, the other from B -> A).
  	 */
  	pipe_double_lock(ipipe, opipe);
  
  	do {
  		if (!opipe->readers) {
  			send_sig(SIGPIPE, current, 0);
  			if (!ret)
  				ret = -EPIPE;
  			break;
  		}
  
  		if (!ipipe->nrbufs && !ipipe->writers)
  			break;
  
  		/*
  		 * Cannot make any progress, because either the input
  		 * pipe is empty or the output pipe is full.
  		 */
35f3d14db   Jens Axboe   pipe: add support...
1751
  		if (!ipipe->nrbufs || opipe->nrbufs >= opipe->buffers) {
7c77f0b3f   Miklos Szeredi   splice: implement...
1752
1753
1754
1755
1756
1757
1758
1759
1760
1761
1762
1763
1764
1765
1766
1767
1768
1769
1770
1771
  			/* Already processed some buffers, break */
  			if (ret)
  				break;
  
  			if (flags & SPLICE_F_NONBLOCK) {
  				ret = -EAGAIN;
  				break;
  			}
  
  			/*
  			 * We raced with another reader/writer and haven't
  			 * managed to process any buffers.  A zero return
  			 * value means EOF, so retry instead.
  			 */
  			pipe_unlock(ipipe);
  			pipe_unlock(opipe);
  			goto retry;
  		}
  
  		ibuf = ipipe->bufs + ipipe->curbuf;
35f3d14db   Jens Axboe   pipe: add support...
1772
  		nbuf = (opipe->curbuf + opipe->nrbufs) & (opipe->buffers - 1);
7c77f0b3f   Miklos Szeredi   splice: implement...
1773
1774
1775
1776
1777
1778
1779
1780
1781
  		obuf = opipe->bufs + nbuf;
  
  		if (len >= ibuf->len) {
  			/*
  			 * Simply move the whole buffer from ipipe to opipe
  			 */
  			*obuf = *ibuf;
  			ibuf->ops = NULL;
  			opipe->nrbufs++;
35f3d14db   Jens Axboe   pipe: add support...
1782
  			ipipe->curbuf = (ipipe->curbuf + 1) & (ipipe->buffers - 1);
7c77f0b3f   Miklos Szeredi   splice: implement...
1783
1784
1785
1786
1787
1788
1789
1790
1791
1792
1793
1794
1795
1796
1797
1798
1799
1800
1801
1802
1803
1804
1805
1806
1807
1808
1809
1810
1811
1812
1813
1814
1815
1816
1817
1818
1819
1820
1821
1822
1823
1824
1825
1826
  			ipipe->nrbufs--;
  			input_wakeup = true;
  		} else {
  			/*
  			 * Get a reference to this pipe buffer,
  			 * so we can copy the contents over.
  			 */
  			ibuf->ops->get(ipipe, ibuf);
  			*obuf = *ibuf;
  
  			/*
  			 * Don't inherit the gift flag, we need to
  			 * prevent multiple steals of this page.
  			 */
  			obuf->flags &= ~PIPE_BUF_FLAG_GIFT;
  
  			obuf->len = len;
  			opipe->nrbufs++;
  			ibuf->offset += obuf->len;
  			ibuf->len -= obuf->len;
  		}
  		ret += obuf->len;
  		len -= obuf->len;
  	} while (len);
  
  	pipe_unlock(ipipe);
  	pipe_unlock(opipe);
  
  	/*
  	 * If we put data in the output pipe, wakeup any potential readers.
  	 */
  	if (ret > 0) {
  		smp_mb();
  		if (waitqueue_active(&opipe->wait))
  			wake_up_interruptible(&opipe->wait);
  		kill_fasync(&opipe->fasync_readers, SIGIO, POLL_IN);
  	}
  	if (input_wakeup)
  		wakeup_pipe_writers(ipipe);
  
  	return ret;
  }
  
  /*
70524490e   Jens Axboe   [PATCH] splice: a...
1827
1828
1829
1830
1831
1832
1833
   * Link contents of ipipe to opipe.
   */
  static int link_pipe(struct pipe_inode_info *ipipe,
  		     struct pipe_inode_info *opipe,
  		     size_t len, unsigned int flags)
  {
  	struct pipe_buffer *ibuf, *obuf;
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1834
  	int ret = 0, i = 0, nbuf;
70524490e   Jens Axboe   [PATCH] splice: a...
1835
1836
1837
  
  	/*
  	 * Potential ABBA deadlock, work around it by ordering lock
61e0d47c3   Miklos Szeredi   splice: add helpe...
1838
  	 * grabbing by pipe info address. Otherwise two different processes
70524490e   Jens Axboe   [PATCH] splice: a...
1839
1840
  	 * could deadlock (one doing tee from A -> B, the other from B -> A).
  	 */
61e0d47c3   Miklos Szeredi   splice: add helpe...
1841
  	pipe_double_lock(ipipe, opipe);
70524490e   Jens Axboe   [PATCH] splice: a...
1842

aadd06e5c   Jens Axboe   [PATCH] splice: f...
1843
  	do {
70524490e   Jens Axboe   [PATCH] splice: a...
1844
1845
1846
1847
1848
1849
  		if (!opipe->readers) {
  			send_sig(SIGPIPE, current, 0);
  			if (!ret)
  				ret = -EPIPE;
  			break;
  		}
70524490e   Jens Axboe   [PATCH] splice: a...
1850

aadd06e5c   Jens Axboe   [PATCH] splice: f...
1851
1852
1853
1854
  		/*
  		 * If we have iterated all input buffers or ran out of
  		 * output room, break.
  		 */
35f3d14db   Jens Axboe   pipe: add support...
1855
  		if (i >= ipipe->nrbufs || opipe->nrbufs >= opipe->buffers)
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1856
  			break;
70524490e   Jens Axboe   [PATCH] splice: a...
1857

35f3d14db   Jens Axboe   pipe: add support...
1858
1859
  		ibuf = ipipe->bufs + ((ipipe->curbuf + i) & (ipipe->buffers-1));
  		nbuf = (opipe->curbuf + opipe->nrbufs) & (opipe->buffers - 1);
70524490e   Jens Axboe   [PATCH] splice: a...
1860
1861
  
  		/*
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1862
1863
  		 * Get a reference to this pipe buffer,
  		 * so we can copy the contents over.
70524490e   Jens Axboe   [PATCH] splice: a...
1864
  		 */
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1865
1866
1867
1868
  		ibuf->ops->get(ipipe, ibuf);
  
  		obuf = opipe->bufs + nbuf;
  		*obuf = *ibuf;
2a27250e6   Jens Axboe   [PATCH] tee: link...
1869
  		/*
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1870
1871
  		 * Don't inherit the gift flag, we need to
  		 * prevent multiple steals of this page.
2a27250e6   Jens Axboe   [PATCH] tee: link...
1872
  		 */
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1873
  		obuf->flags &= ~PIPE_BUF_FLAG_GIFT;
70524490e   Jens Axboe   [PATCH] splice: a...
1874

aadd06e5c   Jens Axboe   [PATCH] splice: f...
1875
1876
  		if (obuf->len > len)
  			obuf->len = len;
70524490e   Jens Axboe   [PATCH] splice: a...
1877

aadd06e5c   Jens Axboe   [PATCH] splice: f...
1878
1879
1880
1881
1882
  		opipe->nrbufs++;
  		ret += obuf->len;
  		len -= obuf->len;
  		i++;
  	} while (len);
70524490e   Jens Axboe   [PATCH] splice: a...
1883

02cf01aea   Jens Axboe   splice: only retu...
1884
1885
1886
1887
1888
1889
  	/*
  	 * return EAGAIN if we have the potential of some data in the
  	 * future, otherwise just return 0
  	 */
  	if (!ret && ipipe->waiting_writers && (flags & SPLICE_F_NONBLOCK))
  		ret = -EAGAIN;
61e0d47c3   Miklos Szeredi   splice: add helpe...
1890
1891
  	pipe_unlock(ipipe);
  	pipe_unlock(opipe);
70524490e   Jens Axboe   [PATCH] splice: a...
1892

aadd06e5c   Jens Axboe   [PATCH] splice: f...
1893
1894
1895
1896
  	/*
  	 * If we put data in the output pipe, wakeup any potential readers.
  	 */
  	if (ret > 0) {
70524490e   Jens Axboe   [PATCH] splice: a...
1897
1898
1899
1900
1901
1902
1903
1904
1905
1906
1907
1908
1909
1910
1911
1912
1913
1914
  		smp_mb();
  		if (waitqueue_active(&opipe->wait))
  			wake_up_interruptible(&opipe->wait);
  		kill_fasync(&opipe->fasync_readers, SIGIO, POLL_IN);
  	}
  
  	return ret;
  }
  
  /*
   * This is a tee(1) implementation that works on pipes. It doesn't copy
   * any data, it simply references the 'in' pages on the 'out' pipe.
   * The 'flags' used are the SPLICE_F_* variants, currently the only
   * applicable one is SPLICE_F_NONBLOCK.
   */
  static long do_tee(struct file *in, struct file *out, size_t len,
  		   unsigned int flags)
  {
71993e62a   Linus Torvalds   Rename 'pipe_info...
1915
1916
  	struct pipe_inode_info *ipipe = get_pipe_info(in);
  	struct pipe_inode_info *opipe = get_pipe_info(out);
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1917
  	int ret = -EINVAL;
70524490e   Jens Axboe   [PATCH] splice: a...
1918
1919
  
  	/*
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1920
1921
  	 * Duplicate the contents of ipipe to opipe without actually
  	 * copying the data.
70524490e   Jens Axboe   [PATCH] splice: a...
1922
  	 */
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1923
1924
1925
1926
1927
  	if (ipipe && opipe && ipipe != opipe) {
  		/*
  		 * Keep going, unless we encounter an error. The ipipe/opipe
  		 * ordering doesn't really matter.
  		 */
7c77f0b3f   Miklos Szeredi   splice: implement...
1928
  		ret = ipipe_prep(ipipe, flags);
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1929
  		if (!ret) {
7c77f0b3f   Miklos Szeredi   splice: implement...
1930
  			ret = opipe_prep(opipe, flags);
02cf01aea   Jens Axboe   splice: only retu...
1931
  			if (!ret)
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1932
  				ret = link_pipe(ipipe, opipe, len, flags);
aadd06e5c   Jens Axboe   [PATCH] splice: f...
1933
1934
  		}
  	}
70524490e   Jens Axboe   [PATCH] splice: a...
1935

aadd06e5c   Jens Axboe   [PATCH] splice: f...
1936
  	return ret;
70524490e   Jens Axboe   [PATCH] splice: a...
1937
  }
836f92adf   Heiko Carstens   [CVE-2009-0029] S...
1938
  SYSCALL_DEFINE4(tee, int, fdin, int, fdout, size_t, len, unsigned int, flags)
70524490e   Jens Axboe   [PATCH] splice: a...
1939
1940
1941
1942
1943
1944
1945
1946
1947
1948
1949
1950
1951
1952
1953
1954
1955
1956
1957
1958
1959
1960
1961
1962
1963
  {
  	struct file *in;
  	int error, fput_in;
  
  	if (unlikely(!len))
  		return 0;
  
  	error = -EBADF;
  	in = fget_light(fdin, &fput_in);
  	if (in) {
  		if (in->f_mode & FMODE_READ) {
  			int fput_out;
  			struct file *out = fget_light(fdout, &fput_out);
  
  			if (out) {
  				if (out->f_mode & FMODE_WRITE)
  					error = do_tee(in, out, len, flags);
  				fput_light(out, fput_out);
  			}
  		}
   		fput_light(in, fput_in);
   	}
  
  	return error;
  }