15274f052SJens Axboe /* 25274f052SJens Axboe * "splice": joining two ropes together by interweaving their strands. 35274f052SJens Axboe * 45274f052SJens Axboe * This is the "extended pipe" functionality, where a pipe is used as 55274f052SJens Axboe * an arbitrary in-memory buffer. Think of a pipe as a small kernel 65274f052SJens Axboe * buffer that you can use to transfer data from one end to the other. 75274f052SJens Axboe * 85274f052SJens Axboe * The traditional unix read/write is extended with a "splice()" operation 95274f052SJens Axboe * that transfers data buffers to or from a pipe buffer. 105274f052SJens Axboe * 115274f052SJens Axboe * Named by Larry McVoy, original implementation from Linus, extended by 12c2058e06SJens Axboe * Jens to support splicing to files, network, direct splicing, etc and 13c2058e06SJens Axboe * fixing lots of bugs. 145274f052SJens Axboe * 150fe23479SJens Axboe * Copyright (C) 2005-2006 Jens Axboe <axboe@kernel.dk> 16c2058e06SJens Axboe * Copyright (C) 2005-2006 Linus Torvalds <torvalds@osdl.org> 17c2058e06SJens Axboe * Copyright (C) 2006 Ingo Molnar <mingo@elte.hu> 185274f052SJens Axboe * 195274f052SJens Axboe */ 20be297968SChristoph Hellwig #include <linux/bvec.h> 215274f052SJens Axboe #include <linux/fs.h> 225274f052SJens Axboe #include <linux/file.h> 235274f052SJens Axboe #include <linux/pagemap.h> 24d6b29d7cSJens Axboe #include <linux/splice.h> 2508e552c6SKAMEZAWA Hiroyuki #include <linux/memcontrol.h> 265274f052SJens Axboe #include <linux/mm_inline.h> 275abc97aaSJens Axboe #include <linux/swap.h> 284f6f0bd2SJens Axboe #include <linux/writeback.h> 29630d9c47SPaul Gortmaker #include <linux/export.h> 304f6f0bd2SJens Axboe #include <linux/syscalls.h> 31912d35f8SJens Axboe #include <linux/uio.h> 3229ce2058SJames Morris #include <linux/security.h> 335a0e3ad6STejun Heo #include <linux/gfp.h> 3435f9c09fSEric Dumazet #include <linux/socket.h> 3576b021d0SAl Viro #include <linux/compat.h> 36174cd4b1SIngo Molnar #include <linux/sched/signal.h> 37174cd4b1SIngo Molnar 3806ae43f3SAl Viro #include "internal.h" 395274f052SJens Axboe 4083f9135bSJens Axboe /* 4183f9135bSJens Axboe * Attempt to steal a page from a pipe buffer. This should perhaps go into 4283f9135bSJens Axboe * a vm helper function, it's already simplified quite a bit by the 4383f9135bSJens Axboe * addition of remove_mapping(). If success is returned, the caller may 4483f9135bSJens Axboe * attempt to reuse this page for another destination. 4583f9135bSJens Axboe */ 4676ad4d11SJens Axboe static int page_cache_pipe_buf_steal(struct pipe_inode_info *pipe, 475abc97aaSJens Axboe struct pipe_buffer *buf) 485abc97aaSJens Axboe { 495abc97aaSJens Axboe struct page *page = buf->page; 509e94cd4fSJens Axboe struct address_space *mapping; 515abc97aaSJens Axboe 529e0267c2SJens Axboe lock_page(page); 539e0267c2SJens Axboe 549e94cd4fSJens Axboe mapping = page_mapping(page); 559e94cd4fSJens Axboe if (mapping) { 565abc97aaSJens Axboe WARN_ON(!PageUptodate(page)); 575abc97aaSJens Axboe 58ad8d6f0aSJens Axboe /* 599e94cd4fSJens Axboe * At least for ext2 with nobh option, we need to wait on 609e94cd4fSJens Axboe * writeback completing on this page, since we'll remove it 619e94cd4fSJens Axboe * from the pagecache. Otherwise truncate wont wait on the 629e94cd4fSJens Axboe * page, allowing the disk blocks to be reused by someone else 639e94cd4fSJens Axboe * before we actually wrote our data to them. fs corruption 649e94cd4fSJens Axboe * ensues. 65ad8d6f0aSJens Axboe */ 66ad8d6f0aSJens Axboe wait_on_page_writeback(page); 67ad8d6f0aSJens Axboe 68266cf658SDavid Howells if (page_has_private(page) && 69266cf658SDavid Howells !try_to_release_page(page, GFP_KERNEL)) 70ca39d651SJens Axboe goto out_unlock; 714f6f0bd2SJens Axboe 729e94cd4fSJens Axboe /* 739e94cd4fSJens Axboe * If we succeeded in removing the mapping, set LRU flag 749e94cd4fSJens Axboe * and return good. 759e94cd4fSJens Axboe */ 769e94cd4fSJens Axboe if (remove_mapping(mapping, page)) { 771432873aSJens Axboe buf->flags |= PIPE_BUF_FLAG_LRU; 785abc97aaSJens Axboe return 0; 795abc97aaSJens Axboe } 809e94cd4fSJens Axboe } 819e94cd4fSJens Axboe 829e94cd4fSJens Axboe /* 839e94cd4fSJens Axboe * Raced with truncate or failed to remove page from current 849e94cd4fSJens Axboe * address space, unlock and return failure. 859e94cd4fSJens Axboe */ 86ca39d651SJens Axboe out_unlock: 879e94cd4fSJens Axboe unlock_page(page); 889e94cd4fSJens Axboe return 1; 899e94cd4fSJens Axboe } 905abc97aaSJens Axboe 9176ad4d11SJens Axboe static void page_cache_pipe_buf_release(struct pipe_inode_info *pipe, 925274f052SJens Axboe struct pipe_buffer *buf) 935274f052SJens Axboe { 9409cbfeafSKirill A. Shutemov put_page(buf->page); 951432873aSJens Axboe buf->flags &= ~PIPE_BUF_FLAG_LRU; 965274f052SJens Axboe } 975274f052SJens Axboe 980845718dSJens Axboe /* 990845718dSJens Axboe * Check whether the contents of buf is OK to access. Since the content 1000845718dSJens Axboe * is a page cache page, IO may be in flight. 1010845718dSJens Axboe */ 102cac36bb0SJens Axboe static int page_cache_pipe_buf_confirm(struct pipe_inode_info *pipe, 1035274f052SJens Axboe struct pipe_buffer *buf) 1045274f052SJens Axboe { 1055274f052SJens Axboe struct page *page = buf->page; 10649d0b21bSJens Axboe int err; 1075274f052SJens Axboe 1085274f052SJens Axboe if (!PageUptodate(page)) { 10949d0b21bSJens Axboe lock_page(page); 1105274f052SJens Axboe 11149d0b21bSJens Axboe /* 11249d0b21bSJens Axboe * Page got truncated/unhashed. This will cause a 0-byte 11373d62d83SIngo Molnar * splice, if this is the first page. 11449d0b21bSJens Axboe */ 1155274f052SJens Axboe if (!page->mapping) { 11649d0b21bSJens Axboe err = -ENODATA; 11749d0b21bSJens Axboe goto error; 1185274f052SJens Axboe } 1195274f052SJens Axboe 12049d0b21bSJens Axboe /* 12173d62d83SIngo Molnar * Uh oh, read-error from disk. 12249d0b21bSJens Axboe */ 12349d0b21bSJens Axboe if (!PageUptodate(page)) { 12449d0b21bSJens Axboe err = -EIO; 12549d0b21bSJens Axboe goto error; 12649d0b21bSJens Axboe } 12749d0b21bSJens Axboe 12849d0b21bSJens Axboe /* 129f84d7519SJens Axboe * Page is ok afterall, we are done. 13049d0b21bSJens Axboe */ 13149d0b21bSJens Axboe unlock_page(page); 13249d0b21bSJens Axboe } 13349d0b21bSJens Axboe 134f84d7519SJens Axboe return 0; 13549d0b21bSJens Axboe error: 13649d0b21bSJens Axboe unlock_page(page); 137f84d7519SJens Axboe return err; 13870524490SJens Axboe } 13970524490SJens Axboe 140708e3508SHugh Dickins const struct pipe_buf_operations page_cache_pipe_buf_ops = { 141cac36bb0SJens Axboe .confirm = page_cache_pipe_buf_confirm, 1425274f052SJens Axboe .release = page_cache_pipe_buf_release, 1435abc97aaSJens Axboe .steal = page_cache_pipe_buf_steal, 144f84d7519SJens Axboe .get = generic_pipe_buf_get, 1455274f052SJens Axboe }; 1465274f052SJens Axboe 147912d35f8SJens Axboe static int user_page_pipe_buf_steal(struct pipe_inode_info *pipe, 148912d35f8SJens Axboe struct pipe_buffer *buf) 149912d35f8SJens Axboe { 1507afa6fd0SJens Axboe if (!(buf->flags & PIPE_BUF_FLAG_GIFT)) 151912d35f8SJens Axboe return 1; 1527afa6fd0SJens Axboe 1531432873aSJens Axboe buf->flags |= PIPE_BUF_FLAG_LRU; 154330ab716SJens Axboe return generic_pipe_buf_steal(pipe, buf); 155912d35f8SJens Axboe } 156912d35f8SJens Axboe 157d4c3cca9SEric Dumazet static const struct pipe_buf_operations user_page_pipe_buf_ops = { 158cac36bb0SJens Axboe .confirm = generic_pipe_buf_confirm, 159912d35f8SJens Axboe .release = page_cache_pipe_buf_release, 160912d35f8SJens Axboe .steal = user_page_pipe_buf_steal, 161f84d7519SJens Axboe .get = generic_pipe_buf_get, 162912d35f8SJens Axboe }; 163912d35f8SJens Axboe 164825cdcb1SNamhyung Kim static void wakeup_pipe_readers(struct pipe_inode_info *pipe) 165825cdcb1SNamhyung Kim { 166825cdcb1SNamhyung Kim smp_mb(); 167825cdcb1SNamhyung Kim if (waitqueue_active(&pipe->wait)) 168825cdcb1SNamhyung Kim wake_up_interruptible(&pipe->wait); 169825cdcb1SNamhyung Kim kill_fasync(&pipe->fasync_readers, SIGIO, POLL_IN); 170825cdcb1SNamhyung Kim } 171825cdcb1SNamhyung Kim 172932cc6d4SJens Axboe /** 173932cc6d4SJens Axboe * splice_to_pipe - fill passed data into a pipe 174932cc6d4SJens Axboe * @pipe: pipe to fill 175932cc6d4SJens Axboe * @spd: data to fill 176932cc6d4SJens Axboe * 177932cc6d4SJens Axboe * Description: 17879685b8dSRandy Dunlap * @spd contains a map of pages and len/offset tuples, along with 179932cc6d4SJens Axboe * the struct pipe_buf_operations associated with these pages. This 180932cc6d4SJens Axboe * function will link that data to the pipe. 181932cc6d4SJens Axboe * 18283f9135bSJens Axboe */ 183d6b29d7cSJens Axboe ssize_t splice_to_pipe(struct pipe_inode_info *pipe, 184912d35f8SJens Axboe struct splice_pipe_desc *spd) 1855274f052SJens Axboe { 18600de00bdSJens Axboe unsigned int spd_pages = spd->nr_pages; 1878924feffSAl Viro int ret = 0, page_nr = 0; 1885274f052SJens Axboe 189d6785d91SRabin Vincent if (!spd_pages) 190d6785d91SRabin Vincent return 0; 191d6785d91SRabin Vincent 1928924feffSAl Viro if (unlikely(!pipe->readers)) { 1935274f052SJens Axboe send_sig(SIGPIPE, current, 0); 1945274f052SJens Axboe ret = -EPIPE; 1958924feffSAl Viro goto out; 1965274f052SJens Axboe } 1975274f052SJens Axboe 1988924feffSAl Viro while (pipe->nrbufs < pipe->buffers) { 19935f3d14dSJens Axboe int newbuf = (pipe->curbuf + pipe->nrbufs) & (pipe->buffers - 1); 2003a326a2cSIngo Molnar struct pipe_buffer *buf = pipe->bufs + newbuf; 2015274f052SJens Axboe 202912d35f8SJens Axboe buf->page = spd->pages[page_nr]; 203912d35f8SJens Axboe buf->offset = spd->partial[page_nr].offset; 204912d35f8SJens Axboe buf->len = spd->partial[page_nr].len; 205497f9625SJens Axboe buf->private = spd->partial[page_nr].private; 206912d35f8SJens Axboe buf->ops = spd->ops; 2075a81e6a1SMiklos Szeredi buf->flags = 0; 2087afa6fd0SJens Axboe 2096f767b04SJens Axboe pipe->nrbufs++; 210912d35f8SJens Axboe page_nr++; 211912d35f8SJens Axboe ret += buf->len; 212912d35f8SJens Axboe 213912d35f8SJens Axboe if (!--spd->nr_pages) 2145274f052SJens Axboe break; 2155274f052SJens Axboe } 2165274f052SJens Axboe 21729e35094SLinus Torvalds if (!ret) 21829e35094SLinus Torvalds ret = -EAGAIN; 21929e35094SLinus Torvalds 2208924feffSAl Viro out: 22100de00bdSJens Axboe while (page_nr < spd_pages) 222bbdfc2f7SJens Axboe spd->spd_release(spd, page_nr++); 2235274f052SJens Axboe 2245274f052SJens Axboe return ret; 2255274f052SJens Axboe } 2262b514574SHannes Frederic Sowa EXPORT_SYMBOL_GPL(splice_to_pipe); 2275274f052SJens Axboe 22879fddc4eSAl Viro ssize_t add_to_pipe(struct pipe_inode_info *pipe, struct pipe_buffer *buf) 22979fddc4eSAl Viro { 23079fddc4eSAl Viro int ret; 23179fddc4eSAl Viro 23279fddc4eSAl Viro if (unlikely(!pipe->readers)) { 23379fddc4eSAl Viro send_sig(SIGPIPE, current, 0); 23479fddc4eSAl Viro ret = -EPIPE; 23579fddc4eSAl Viro } else if (pipe->nrbufs == pipe->buffers) { 23679fddc4eSAl Viro ret = -EAGAIN; 23779fddc4eSAl Viro } else { 23879fddc4eSAl Viro int newbuf = (pipe->curbuf + pipe->nrbufs) & (pipe->buffers - 1); 23979fddc4eSAl Viro pipe->bufs[newbuf] = *buf; 24079fddc4eSAl Viro pipe->nrbufs++; 24179fddc4eSAl Viro return buf->len; 24279fddc4eSAl Viro } 243a779638cSMiklos Szeredi pipe_buf_release(pipe, buf); 24479fddc4eSAl Viro return ret; 24579fddc4eSAl Viro } 24679fddc4eSAl Viro EXPORT_SYMBOL(add_to_pipe); 24779fddc4eSAl Viro 24835f3d14dSJens Axboe /* 24935f3d14dSJens Axboe * Check if we need to grow the arrays holding pages and partial page 25035f3d14dSJens Axboe * descriptions. 25135f3d14dSJens Axboe */ 252047fe360SEric Dumazet int splice_grow_spd(const struct pipe_inode_info *pipe, struct splice_pipe_desc *spd) 25335f3d14dSJens Axboe { 2546aa7de05SMark Rutland unsigned int buffers = READ_ONCE(pipe->buffers); 255047fe360SEric Dumazet 256047fe360SEric Dumazet spd->nr_pages_max = buffers; 257047fe360SEric Dumazet if (buffers <= PIPE_DEF_BUFFERS) 25835f3d14dSJens Axboe return 0; 25935f3d14dSJens Axboe 2606da2ec56SKees Cook spd->pages = kmalloc_array(buffers, sizeof(struct page *), GFP_KERNEL); 2616da2ec56SKees Cook spd->partial = kmalloc_array(buffers, sizeof(struct partial_page), 2626da2ec56SKees Cook GFP_KERNEL); 26335f3d14dSJens Axboe 26435f3d14dSJens Axboe if (spd->pages && spd->partial) 26535f3d14dSJens Axboe return 0; 26635f3d14dSJens Axboe 26735f3d14dSJens Axboe kfree(spd->pages); 26835f3d14dSJens Axboe kfree(spd->partial); 26935f3d14dSJens Axboe return -ENOMEM; 27035f3d14dSJens Axboe } 27135f3d14dSJens Axboe 272047fe360SEric Dumazet void splice_shrink_spd(struct splice_pipe_desc *spd) 27335f3d14dSJens Axboe { 274047fe360SEric Dumazet if (spd->nr_pages_max <= PIPE_DEF_BUFFERS) 27535f3d14dSJens Axboe return; 27635f3d14dSJens Axboe 27735f3d14dSJens Axboe kfree(spd->pages); 27835f3d14dSJens Axboe kfree(spd->partial); 27935f3d14dSJens Axboe } 28035f3d14dSJens Axboe 28183f9135bSJens Axboe /** 28283f9135bSJens Axboe * generic_file_splice_read - splice data from file to a pipe 28383f9135bSJens Axboe * @in: file to splice from 284932cc6d4SJens Axboe * @ppos: position in @in 28583f9135bSJens Axboe * @pipe: pipe to splice to 28683f9135bSJens Axboe * @len: number of bytes to splice 28783f9135bSJens Axboe * @flags: splice modifier flags 28883f9135bSJens Axboe * 289932cc6d4SJens Axboe * Description: 290932cc6d4SJens Axboe * Will read pages from given file and fill them into a pipe. Can be 29182c156f8SAl Viro * used as long as it has more or less sane ->read_iter(). 292932cc6d4SJens Axboe * 29383f9135bSJens Axboe */ 294cbb7e577SJens Axboe ssize_t generic_file_splice_read(struct file *in, loff_t *ppos, 295cbb7e577SJens Axboe struct pipe_inode_info *pipe, size_t len, 296cbb7e577SJens Axboe unsigned int flags) 2975274f052SJens Axboe { 29882c156f8SAl Viro struct iov_iter to; 29982c156f8SAl Viro struct kiocb kiocb; 30082c156f8SAl Viro int idx, ret; 301be64f884SBoaz Harrosh 302aa563d7bSDavid Howells iov_iter_pipe(&to, READ, pipe, len); 30382c156f8SAl Viro idx = to.idx; 30482c156f8SAl Viro init_sync_kiocb(&kiocb, in); 30582c156f8SAl Viro kiocb.ki_pos = *ppos; 306bb7462b6SMiklos Szeredi ret = call_read_iter(in, &kiocb, &to); 307723590edSMiklos Szeredi if (ret > 0) { 30882c156f8SAl Viro *ppos = kiocb.ki_pos; 309723590edSMiklos Szeredi file_accessed(in); 31082c156f8SAl Viro } else if (ret < 0) { 31182c156f8SAl Viro to.idx = idx; 31282c156f8SAl Viro to.iov_offset = 0; 31382c156f8SAl Viro iov_iter_advance(&to, 0); /* to free what was emitted */ 31482c156f8SAl Viro /* 31582c156f8SAl Viro * callers of ->splice_read() expect -EAGAIN on 31682c156f8SAl Viro * "can't put anything in there", rather than -EFAULT. 31782c156f8SAl Viro */ 31882c156f8SAl Viro if (ret == -EFAULT) 31982c156f8SAl Viro ret = -EAGAIN; 320723590edSMiklos Szeredi } 3215274f052SJens Axboe 3225274f052SJens Axboe return ret; 3235274f052SJens Axboe } 324059a8f37SJens Axboe EXPORT_SYMBOL(generic_file_splice_read); 325059a8f37SJens Axboe 326241699cdSAl Viro const struct pipe_buf_operations default_pipe_buf_ops = { 3276818173bSMiklos Szeredi .confirm = generic_pipe_buf_confirm, 3286818173bSMiklos Szeredi .release = generic_pipe_buf_release, 3296818173bSMiklos Szeredi .steal = generic_pipe_buf_steal, 3306818173bSMiklos Szeredi .get = generic_pipe_buf_get, 3316818173bSMiklos Szeredi }; 3326818173bSMiklos Szeredi 333b9872226SJann Horn int generic_pipe_buf_nosteal(struct pipe_inode_info *pipe, 33428a625cbSMiklos Szeredi struct pipe_buffer *buf) 33528a625cbSMiklos Szeredi { 33628a625cbSMiklos Szeredi return 1; 33728a625cbSMiklos Szeredi } 33828a625cbSMiklos Szeredi 33928a625cbSMiklos Szeredi /* Pipe buffer operations for a socket and similar. */ 34028a625cbSMiklos Szeredi const struct pipe_buf_operations nosteal_pipe_buf_ops = { 34128a625cbSMiklos Szeredi .confirm = generic_pipe_buf_confirm, 34228a625cbSMiklos Szeredi .release = generic_pipe_buf_release, 34328a625cbSMiklos Szeredi .steal = generic_pipe_buf_nosteal, 34428a625cbSMiklos Szeredi .get = generic_pipe_buf_get, 34528a625cbSMiklos Szeredi }; 34628a625cbSMiklos Szeredi EXPORT_SYMBOL(nosteal_pipe_buf_ops); 34728a625cbSMiklos Szeredi 348523ac9afSAl Viro static ssize_t kernel_readv(struct file *file, const struct kvec *vec, 3496818173bSMiklos Szeredi unsigned long vlen, loff_t offset) 3506818173bSMiklos Szeredi { 3516818173bSMiklos Szeredi mm_segment_t old_fs; 3526818173bSMiklos Szeredi loff_t pos = offset; 3536818173bSMiklos Szeredi ssize_t res; 3546818173bSMiklos Szeredi 3556818173bSMiklos Szeredi old_fs = get_fs(); 356736706beSLinus Torvalds set_fs(KERNEL_DS); 3576818173bSMiklos Szeredi /* The cast to a user pointer is valid due to the set_fs() */ 358793b80efSChristoph Hellwig res = vfs_readv(file, (const struct iovec __user *)vec, vlen, &pos, 0); 3596818173bSMiklos Szeredi set_fs(old_fs); 3606818173bSMiklos Szeredi 3616818173bSMiklos Szeredi return res; 3626818173bSMiklos Szeredi } 3636818173bSMiklos Szeredi 36482c156f8SAl Viro static ssize_t default_file_splice_read(struct file *in, loff_t *ppos, 3656818173bSMiklos Szeredi struct pipe_inode_info *pipe, size_t len, 3666818173bSMiklos Szeredi unsigned int flags) 3676818173bSMiklos Szeredi { 368523ac9afSAl Viro struct kvec *vec, __vec[PIPE_DEF_BUFFERS]; 369523ac9afSAl Viro struct iov_iter to; 370523ac9afSAl Viro struct page **pages; 3716818173bSMiklos Szeredi unsigned int nr_pages; 37213c0f52bSAl Viro size_t offset, base, copied = 0; 3736818173bSMiklos Szeredi ssize_t res; 3746818173bSMiklos Szeredi int i; 3756818173bSMiklos Szeredi 376523ac9afSAl Viro if (pipe->nrbufs == pipe->buffers) 377523ac9afSAl Viro return -EAGAIN; 378523ac9afSAl Viro 379523ac9afSAl Viro /* 380523ac9afSAl Viro * Try to keep page boundaries matching to source pagecache ones - 381523ac9afSAl Viro * it probably won't be much help, but... 382523ac9afSAl Viro */ 383523ac9afSAl Viro offset = *ppos & ~PAGE_MASK; 384523ac9afSAl Viro 385aa563d7bSDavid Howells iov_iter_pipe(&to, READ, pipe, len + offset); 386523ac9afSAl Viro 38713c0f52bSAl Viro res = iov_iter_get_pages_alloc(&to, &pages, len + offset, &base); 388523ac9afSAl Viro if (res <= 0) 38935f3d14dSJens Axboe return -ENOMEM; 39035f3d14dSJens Axboe 39113c0f52bSAl Viro nr_pages = DIV_ROUND_UP(res + base, PAGE_SIZE); 392523ac9afSAl Viro 39335f3d14dSJens Axboe vec = __vec; 394523ac9afSAl Viro if (nr_pages > PIPE_DEF_BUFFERS) { 3956da2ec56SKees Cook vec = kmalloc_array(nr_pages, sizeof(struct kvec), GFP_KERNEL); 396523ac9afSAl Viro if (unlikely(!vec)) { 397523ac9afSAl Viro res = -ENOMEM; 398523ac9afSAl Viro goto out; 399523ac9afSAl Viro } 40035f3d14dSJens Axboe } 40135f3d14dSJens Axboe 402523ac9afSAl Viro pipe->bufs[to.idx].offset = offset; 403523ac9afSAl Viro pipe->bufs[to.idx].len -= offset; 4046818173bSMiklos Szeredi 405523ac9afSAl Viro for (i = 0; i < nr_pages; i++) { 406523ac9afSAl Viro size_t this_len = min_t(size_t, len, PAGE_SIZE - offset); 407523ac9afSAl Viro vec[i].iov_base = page_address(pages[i]) + offset; 4086818173bSMiklos Szeredi vec[i].iov_len = this_len; 4096818173bSMiklos Szeredi len -= this_len; 4106818173bSMiklos Szeredi offset = 0; 4116818173bSMiklos Szeredi } 4126818173bSMiklos Szeredi 413523ac9afSAl Viro res = kernel_readv(in, vec, nr_pages, *ppos); 414523ac9afSAl Viro if (res > 0) { 415523ac9afSAl Viro copied = res; 4166818173bSMiklos Szeredi *ppos += res; 417523ac9afSAl Viro } 4186818173bSMiklos Szeredi 41935f3d14dSJens Axboe if (vec != __vec) 42035f3d14dSJens Axboe kfree(vec); 421523ac9afSAl Viro out: 422523ac9afSAl Viro for (i = 0; i < nr_pages; i++) 423523ac9afSAl Viro put_page(pages[i]); 424523ac9afSAl Viro kvfree(pages); 425523ac9afSAl Viro iov_iter_advance(&to, copied); /* truncates and discards */ 4266818173bSMiklos Szeredi return res; 4276818173bSMiklos Szeredi } 4286818173bSMiklos Szeredi 4295274f052SJens Axboe /* 4304f6f0bd2SJens Axboe * Send 'sd->len' bytes to socket from 'sd->file' at position 'sd->pos' 431016b661eSJens Axboe * using sendpage(). Return the number of bytes sent. 4325274f052SJens Axboe */ 43376ad4d11SJens Axboe static int pipe_to_sendpage(struct pipe_inode_info *pipe, 4345274f052SJens Axboe struct pipe_buffer *buf, struct splice_desc *sd) 4355274f052SJens Axboe { 4366a14b90bSJens Axboe struct file *file = sd->u.file; 4375274f052SJens Axboe loff_t pos = sd->pos; 438a8adbe37SMichał Mirosław int more; 4395274f052SJens Axboe 44072c2d531SAl Viro if (!likely(file->f_op->sendpage)) 441a8adbe37SMichał Mirosław return -EINVAL; 442a8adbe37SMichał Mirosław 44335f9c09fSEric Dumazet more = (sd->flags & SPLICE_F_MORE) ? MSG_MORE : 0; 444ae62ca7bSEric Dumazet 445ae62ca7bSEric Dumazet if (sd->len < sd->total_len && pipe->nrbufs > 1) 44635f9c09fSEric Dumazet more |= MSG_SENDPAGE_NOTLAST; 447ae62ca7bSEric Dumazet 448a8adbe37SMichał Mirosław return file->f_op->sendpage(file, buf->page, buf->offset, 449f84d7519SJens Axboe sd->len, &pos, more); 4505274f052SJens Axboe } 4515274f052SJens Axboe 452b3c2d2ddSMiklos Szeredi static void wakeup_pipe_writers(struct pipe_inode_info *pipe) 453b3c2d2ddSMiklos Szeredi { 454b3c2d2ddSMiklos Szeredi smp_mb(); 455b3c2d2ddSMiklos Szeredi if (waitqueue_active(&pipe->wait)) 456b3c2d2ddSMiklos Szeredi wake_up_interruptible(&pipe->wait); 457b3c2d2ddSMiklos Szeredi kill_fasync(&pipe->fasync_writers, SIGIO, POLL_OUT); 458b3c2d2ddSMiklos Szeredi } 459b3c2d2ddSMiklos Szeredi 460b3c2d2ddSMiklos Szeredi /** 461b3c2d2ddSMiklos Szeredi * splice_from_pipe_feed - feed available data from a pipe to a file 462b3c2d2ddSMiklos Szeredi * @pipe: pipe to splice from 463b3c2d2ddSMiklos Szeredi * @sd: information to @actor 464b3c2d2ddSMiklos Szeredi * @actor: handler that splices the data 465b3c2d2ddSMiklos Szeredi * 466b3c2d2ddSMiklos Szeredi * Description: 467b3c2d2ddSMiklos Szeredi * This function loops over the pipe and calls @actor to do the 468b3c2d2ddSMiklos Szeredi * actual moving of a single struct pipe_buffer to the desired 469b3c2d2ddSMiklos Szeredi * destination. It returns when there's no more buffers left in 470b3c2d2ddSMiklos Szeredi * the pipe or if the requested number of bytes (@sd->total_len) 471b3c2d2ddSMiklos Szeredi * have been copied. It returns a positive number (one) if the 472b3c2d2ddSMiklos Szeredi * pipe needs to be filled with more data, zero if the required 473b3c2d2ddSMiklos Szeredi * number of bytes have been copied and -errno on error. 474b3c2d2ddSMiklos Szeredi * 475b3c2d2ddSMiklos Szeredi * This, together with splice_from_pipe_{begin,end,next}, may be 476b3c2d2ddSMiklos Szeredi * used to implement the functionality of __splice_from_pipe() when 477b3c2d2ddSMiklos Szeredi * locking is required around copying the pipe buffers to the 478b3c2d2ddSMiklos Szeredi * destination. 479b3c2d2ddSMiklos Szeredi */ 48096f9bc8fSAl Viro static int splice_from_pipe_feed(struct pipe_inode_info *pipe, struct splice_desc *sd, 481b3c2d2ddSMiklos Szeredi splice_actor *actor) 482b3c2d2ddSMiklos Szeredi { 483b3c2d2ddSMiklos Szeredi int ret; 484b3c2d2ddSMiklos Szeredi 485b3c2d2ddSMiklos Szeredi while (pipe->nrbufs) { 486b3c2d2ddSMiklos Szeredi struct pipe_buffer *buf = pipe->bufs + pipe->curbuf; 487b3c2d2ddSMiklos Szeredi 488b3c2d2ddSMiklos Szeredi sd->len = buf->len; 489b3c2d2ddSMiklos Szeredi if (sd->len > sd->total_len) 490b3c2d2ddSMiklos Szeredi sd->len = sd->total_len; 491b3c2d2ddSMiklos Szeredi 492fba597dbSMiklos Szeredi ret = pipe_buf_confirm(pipe, buf); 493a8adbe37SMichał Mirosław if (unlikely(ret)) { 494b3c2d2ddSMiklos Szeredi if (ret == -ENODATA) 495b3c2d2ddSMiklos Szeredi ret = 0; 496b3c2d2ddSMiklos Szeredi return ret; 497b3c2d2ddSMiklos Szeredi } 498a8adbe37SMichał Mirosław 499a8adbe37SMichał Mirosław ret = actor(pipe, buf, sd); 500a8adbe37SMichał Mirosław if (ret <= 0) 501a8adbe37SMichał Mirosław return ret; 502a8adbe37SMichał Mirosław 503b3c2d2ddSMiklos Szeredi buf->offset += ret; 504b3c2d2ddSMiklos Szeredi buf->len -= ret; 505b3c2d2ddSMiklos Szeredi 506b3c2d2ddSMiklos Szeredi sd->num_spliced += ret; 507b3c2d2ddSMiklos Szeredi sd->len -= ret; 508b3c2d2ddSMiklos Szeredi sd->pos += ret; 509b3c2d2ddSMiklos Szeredi sd->total_len -= ret; 510b3c2d2ddSMiklos Szeredi 511b3c2d2ddSMiklos Szeredi if (!buf->len) { 512a779638cSMiklos Szeredi pipe_buf_release(pipe, buf); 51335f3d14dSJens Axboe pipe->curbuf = (pipe->curbuf + 1) & (pipe->buffers - 1); 514b3c2d2ddSMiklos Szeredi pipe->nrbufs--; 5156447a3cfSAl Viro if (pipe->files) 516b3c2d2ddSMiklos Szeredi sd->need_wakeup = true; 517b3c2d2ddSMiklos Szeredi } 518b3c2d2ddSMiklos Szeredi 519b3c2d2ddSMiklos Szeredi if (!sd->total_len) 520b3c2d2ddSMiklos Szeredi return 0; 521b3c2d2ddSMiklos Szeredi } 522b3c2d2ddSMiklos Szeredi 523b3c2d2ddSMiklos Szeredi return 1; 524b3c2d2ddSMiklos Szeredi } 525b3c2d2ddSMiklos Szeredi 526b3c2d2ddSMiklos Szeredi /** 527b3c2d2ddSMiklos Szeredi * splice_from_pipe_next - wait for some data to splice from 528b3c2d2ddSMiklos Szeredi * @pipe: pipe to splice from 529b3c2d2ddSMiklos Szeredi * @sd: information about the splice operation 530b3c2d2ddSMiklos Szeredi * 531b3c2d2ddSMiklos Szeredi * Description: 532b3c2d2ddSMiklos Szeredi * This function will wait for some data and return a positive 533b3c2d2ddSMiklos Szeredi * value (one) if pipe buffers are available. It will return zero 534b3c2d2ddSMiklos Szeredi * or -errno if no more data needs to be spliced. 535b3c2d2ddSMiklos Szeredi */ 53696f9bc8fSAl Viro static int splice_from_pipe_next(struct pipe_inode_info *pipe, struct splice_desc *sd) 537b3c2d2ddSMiklos Szeredi { 538c725bfceSJan Kara /* 539c725bfceSJan Kara * Check for signal early to make process killable when there are 540c725bfceSJan Kara * always buffers available 541c725bfceSJan Kara */ 542c725bfceSJan Kara if (signal_pending(current)) 543c725bfceSJan Kara return -ERESTARTSYS; 544c725bfceSJan Kara 545b3c2d2ddSMiklos Szeredi while (!pipe->nrbufs) { 546b3c2d2ddSMiklos Szeredi if (!pipe->writers) 547b3c2d2ddSMiklos Szeredi return 0; 548b3c2d2ddSMiklos Szeredi 549b3c2d2ddSMiklos Szeredi if (!pipe->waiting_writers && sd->num_spliced) 550b3c2d2ddSMiklos Szeredi return 0; 551b3c2d2ddSMiklos Szeredi 552b3c2d2ddSMiklos Szeredi if (sd->flags & SPLICE_F_NONBLOCK) 553b3c2d2ddSMiklos Szeredi return -EAGAIN; 554b3c2d2ddSMiklos Szeredi 555b3c2d2ddSMiklos Szeredi if (signal_pending(current)) 556b3c2d2ddSMiklos Szeredi return -ERESTARTSYS; 557b3c2d2ddSMiklos Szeredi 558b3c2d2ddSMiklos Szeredi if (sd->need_wakeup) { 559b3c2d2ddSMiklos Szeredi wakeup_pipe_writers(pipe); 560b3c2d2ddSMiklos Szeredi sd->need_wakeup = false; 561b3c2d2ddSMiklos Szeredi } 562b3c2d2ddSMiklos Szeredi 563b3c2d2ddSMiklos Szeredi pipe_wait(pipe); 564b3c2d2ddSMiklos Szeredi } 565b3c2d2ddSMiklos Szeredi 566b3c2d2ddSMiklos Szeredi return 1; 567b3c2d2ddSMiklos Szeredi } 568b3c2d2ddSMiklos Szeredi 569b3c2d2ddSMiklos Szeredi /** 570b3c2d2ddSMiklos Szeredi * splice_from_pipe_begin - start splicing from pipe 571b80901bbSRandy Dunlap * @sd: information about the splice operation 572b3c2d2ddSMiklos Szeredi * 573b3c2d2ddSMiklos Szeredi * Description: 574b3c2d2ddSMiklos Szeredi * This function should be called before a loop containing 575b3c2d2ddSMiklos Szeredi * splice_from_pipe_next() and splice_from_pipe_feed() to 576b3c2d2ddSMiklos Szeredi * initialize the necessary fields of @sd. 577b3c2d2ddSMiklos Szeredi */ 57896f9bc8fSAl Viro static void splice_from_pipe_begin(struct splice_desc *sd) 579b3c2d2ddSMiklos Szeredi { 580b3c2d2ddSMiklos Szeredi sd->num_spliced = 0; 581b3c2d2ddSMiklos Szeredi sd->need_wakeup = false; 582b3c2d2ddSMiklos Szeredi } 583b3c2d2ddSMiklos Szeredi 584b3c2d2ddSMiklos Szeredi /** 585b3c2d2ddSMiklos Szeredi * splice_from_pipe_end - finish splicing from pipe 586b3c2d2ddSMiklos Szeredi * @pipe: pipe to splice from 587b3c2d2ddSMiklos Szeredi * @sd: information about the splice operation 588b3c2d2ddSMiklos Szeredi * 589b3c2d2ddSMiklos Szeredi * Description: 590b3c2d2ddSMiklos Szeredi * This function will wake up pipe writers if necessary. It should 591b3c2d2ddSMiklos Szeredi * be called after a loop containing splice_from_pipe_next() and 592b3c2d2ddSMiklos Szeredi * splice_from_pipe_feed(). 593b3c2d2ddSMiklos Szeredi */ 59496f9bc8fSAl Viro static void splice_from_pipe_end(struct pipe_inode_info *pipe, struct splice_desc *sd) 595b3c2d2ddSMiklos Szeredi { 596b3c2d2ddSMiklos Szeredi if (sd->need_wakeup) 597b3c2d2ddSMiklos Szeredi wakeup_pipe_writers(pipe); 598b3c2d2ddSMiklos Szeredi } 599b3c2d2ddSMiklos Szeredi 600932cc6d4SJens Axboe /** 601932cc6d4SJens Axboe * __splice_from_pipe - splice data from a pipe to given actor 602932cc6d4SJens Axboe * @pipe: pipe to splice from 603932cc6d4SJens Axboe * @sd: information to @actor 604932cc6d4SJens Axboe * @actor: handler that splices the data 605932cc6d4SJens Axboe * 606932cc6d4SJens Axboe * Description: 607932cc6d4SJens Axboe * This function does little more than loop over the pipe and call 608932cc6d4SJens Axboe * @actor to do the actual moving of a single struct pipe_buffer to 609932cc6d4SJens Axboe * the desired destination. See pipe_to_file, pipe_to_sendpage, or 610932cc6d4SJens Axboe * pipe_to_user. 611932cc6d4SJens Axboe * 61283f9135bSJens Axboe */ 613c66ab6faSJens Axboe ssize_t __splice_from_pipe(struct pipe_inode_info *pipe, struct splice_desc *sd, 614c66ab6faSJens Axboe splice_actor *actor) 6155274f052SJens Axboe { 616b3c2d2ddSMiklos Szeredi int ret; 6175274f052SJens Axboe 618b3c2d2ddSMiklos Szeredi splice_from_pipe_begin(sd); 619b3c2d2ddSMiklos Szeredi do { 620c2489e07SJan Kara cond_resched(); 621b3c2d2ddSMiklos Szeredi ret = splice_from_pipe_next(pipe, sd); 622b3c2d2ddSMiklos Szeredi if (ret > 0) 623b3c2d2ddSMiklos Szeredi ret = splice_from_pipe_feed(pipe, sd, actor); 624b3c2d2ddSMiklos Szeredi } while (ret > 0); 625b3c2d2ddSMiklos Szeredi splice_from_pipe_end(pipe, sd); 6265274f052SJens Axboe 627b3c2d2ddSMiklos Szeredi return sd->num_spliced ? sd->num_spliced : ret; 6285274f052SJens Axboe } 62940bee44eSMark Fasheh EXPORT_SYMBOL(__splice_from_pipe); 6305274f052SJens Axboe 631932cc6d4SJens Axboe /** 632932cc6d4SJens Axboe * splice_from_pipe - splice data from a pipe to a file 633932cc6d4SJens Axboe * @pipe: pipe to splice from 634932cc6d4SJens Axboe * @out: file to splice to 635932cc6d4SJens Axboe * @ppos: position in @out 636932cc6d4SJens Axboe * @len: how many bytes to splice 637932cc6d4SJens Axboe * @flags: splice modifier flags 638932cc6d4SJens Axboe * @actor: handler that splices the data 639932cc6d4SJens Axboe * 640932cc6d4SJens Axboe * Description: 6412933970bSMiklos Szeredi * See __splice_from_pipe. This function locks the pipe inode, 642932cc6d4SJens Axboe * otherwise it's identical to __splice_from_pipe(). 643932cc6d4SJens Axboe * 644932cc6d4SJens Axboe */ 6456da61809SMark Fasheh ssize_t splice_from_pipe(struct pipe_inode_info *pipe, struct file *out, 6466da61809SMark Fasheh loff_t *ppos, size_t len, unsigned int flags, 6476da61809SMark Fasheh splice_actor *actor) 6486da61809SMark Fasheh { 6496da61809SMark Fasheh ssize_t ret; 650c66ab6faSJens Axboe struct splice_desc sd = { 651c66ab6faSJens Axboe .total_len = len, 652c66ab6faSJens Axboe .flags = flags, 653c66ab6faSJens Axboe .pos = *ppos, 6546a14b90bSJens Axboe .u.file = out, 655c66ab6faSJens Axboe }; 6566da61809SMark Fasheh 65761e0d47cSMiklos Szeredi pipe_lock(pipe); 658c66ab6faSJens Axboe ret = __splice_from_pipe(pipe, &sd, actor); 65961e0d47cSMiklos Szeredi pipe_unlock(pipe); 6606da61809SMark Fasheh 6616da61809SMark Fasheh return ret; 6626da61809SMark Fasheh } 6636da61809SMark Fasheh 6646da61809SMark Fasheh /** 6658d020765SAl Viro * iter_file_splice_write - splice data from a pipe to a file 6668d020765SAl Viro * @pipe: pipe info 6678d020765SAl Viro * @out: file to write to 6688d020765SAl Viro * @ppos: position in @out 6698d020765SAl Viro * @len: number of bytes to splice 6708d020765SAl Viro * @flags: splice modifier flags 6718d020765SAl Viro * 6728d020765SAl Viro * Description: 6738d020765SAl Viro * Will either move or copy pages (determined by @flags options) from 6748d020765SAl Viro * the given pipe inode to the given file. 6758d020765SAl Viro * This one is ->write_iter-based. 6768d020765SAl Viro * 6778d020765SAl Viro */ 6788d020765SAl Viro ssize_t 6798d020765SAl Viro iter_file_splice_write(struct pipe_inode_info *pipe, struct file *out, 6808d020765SAl Viro loff_t *ppos, size_t len, unsigned int flags) 6818d020765SAl Viro { 6828d020765SAl Viro struct splice_desc sd = { 6838d020765SAl Viro .total_len = len, 6848d020765SAl Viro .flags = flags, 6858d020765SAl Viro .pos = *ppos, 6868d020765SAl Viro .u.file = out, 6878d020765SAl Viro }; 6888d020765SAl Viro int nbufs = pipe->buffers; 6898d020765SAl Viro struct bio_vec *array = kcalloc(nbufs, sizeof(struct bio_vec), 6908d020765SAl Viro GFP_KERNEL); 6918d020765SAl Viro ssize_t ret; 6928d020765SAl Viro 6938d020765SAl Viro if (unlikely(!array)) 6948d020765SAl Viro return -ENOMEM; 6958d020765SAl Viro 6968d020765SAl Viro pipe_lock(pipe); 6978d020765SAl Viro 6988d020765SAl Viro splice_from_pipe_begin(&sd); 6998d020765SAl Viro while (sd.total_len) { 7008d020765SAl Viro struct iov_iter from; 7018d020765SAl Viro size_t left; 7028d020765SAl Viro int n, idx; 7038d020765SAl Viro 7048d020765SAl Viro ret = splice_from_pipe_next(pipe, &sd); 7058d020765SAl Viro if (ret <= 0) 7068d020765SAl Viro break; 7078d020765SAl Viro 7088d020765SAl Viro if (unlikely(nbufs < pipe->buffers)) { 7098d020765SAl Viro kfree(array); 7108d020765SAl Viro nbufs = pipe->buffers; 7118d020765SAl Viro array = kcalloc(nbufs, sizeof(struct bio_vec), 7128d020765SAl Viro GFP_KERNEL); 7138d020765SAl Viro if (!array) { 7148d020765SAl Viro ret = -ENOMEM; 7158d020765SAl Viro break; 7168d020765SAl Viro } 7178d020765SAl Viro } 7188d020765SAl Viro 7198d020765SAl Viro /* build the vector */ 7208d020765SAl Viro left = sd.total_len; 7218d020765SAl Viro for (n = 0, idx = pipe->curbuf; left && n < pipe->nrbufs; n++, idx++) { 7228d020765SAl Viro struct pipe_buffer *buf = pipe->bufs + idx; 7238d020765SAl Viro size_t this_len = buf->len; 7248d020765SAl Viro 7258d020765SAl Viro if (this_len > left) 7268d020765SAl Viro this_len = left; 7278d020765SAl Viro 7288d020765SAl Viro if (idx == pipe->buffers - 1) 7298d020765SAl Viro idx = -1; 7308d020765SAl Viro 731fba597dbSMiklos Szeredi ret = pipe_buf_confirm(pipe, buf); 7328d020765SAl Viro if (unlikely(ret)) { 7338d020765SAl Viro if (ret == -ENODATA) 7348d020765SAl Viro ret = 0; 7358d020765SAl Viro goto done; 7368d020765SAl Viro } 7378d020765SAl Viro 7388d020765SAl Viro array[n].bv_page = buf->page; 7398d020765SAl Viro array[n].bv_len = this_len; 7408d020765SAl Viro array[n].bv_offset = buf->offset; 7418d020765SAl Viro left -= this_len; 7428d020765SAl Viro } 7438d020765SAl Viro 744aa563d7bSDavid Howells iov_iter_bvec(&from, WRITE, array, n, sd.total_len - left); 745abbb6589SChristoph Hellwig ret = vfs_iter_write(out, &from, &sd.pos, 0); 7468d020765SAl Viro if (ret <= 0) 7478d020765SAl Viro break; 7488d020765SAl Viro 7498d020765SAl Viro sd.num_spliced += ret; 7508d020765SAl Viro sd.total_len -= ret; 751dbe4e192SChristoph Hellwig *ppos = sd.pos; 7528d020765SAl Viro 7538d020765SAl Viro /* dismiss the fully eaten buffers, adjust the partial one */ 7548d020765SAl Viro while (ret) { 7558d020765SAl Viro struct pipe_buffer *buf = pipe->bufs + pipe->curbuf; 7568d020765SAl Viro if (ret >= buf->len) { 7578d020765SAl Viro ret -= buf->len; 7588d020765SAl Viro buf->len = 0; 759a779638cSMiklos Szeredi pipe_buf_release(pipe, buf); 7608d020765SAl Viro pipe->curbuf = (pipe->curbuf + 1) & (pipe->buffers - 1); 7618d020765SAl Viro pipe->nrbufs--; 7628d020765SAl Viro if (pipe->files) 7638d020765SAl Viro sd.need_wakeup = true; 7648d020765SAl Viro } else { 7658d020765SAl Viro buf->offset += ret; 7668d020765SAl Viro buf->len -= ret; 7678d020765SAl Viro ret = 0; 7688d020765SAl Viro } 7698d020765SAl Viro } 7708d020765SAl Viro } 7718d020765SAl Viro done: 7728d020765SAl Viro kfree(array); 7738d020765SAl Viro splice_from_pipe_end(pipe, &sd); 7748d020765SAl Viro 7758d020765SAl Viro pipe_unlock(pipe); 7768d020765SAl Viro 7778d020765SAl Viro if (sd.num_spliced) 7788d020765SAl Viro ret = sd.num_spliced; 7798d020765SAl Viro 7808d020765SAl Viro return ret; 7818d020765SAl Viro } 7828d020765SAl Viro 7838d020765SAl Viro EXPORT_SYMBOL(iter_file_splice_write); 7848d020765SAl Viro 785b2858d7dSMiklos Szeredi static int write_pipe_buf(struct pipe_inode_info *pipe, struct pipe_buffer *buf, 786b2858d7dSMiklos Szeredi struct splice_desc *sd) 7870b0a47f5SMiklos Szeredi { 788b2858d7dSMiklos Szeredi int ret; 789b2858d7dSMiklos Szeredi void *data; 79006ae43f3SAl Viro loff_t tmp = sd->pos; 791b2858d7dSMiklos Szeredi 792fbb32750SAl Viro data = kmap(buf->page); 79306ae43f3SAl Viro ret = __kernel_write(sd->u.file, data + buf->offset, sd->len, &tmp); 794fbb32750SAl Viro kunmap(buf->page); 795b2858d7dSMiklos Szeredi 796b2858d7dSMiklos Szeredi return ret; 7970b0a47f5SMiklos Szeredi } 7980b0a47f5SMiklos Szeredi 7990b0a47f5SMiklos Szeredi static ssize_t default_file_splice_write(struct pipe_inode_info *pipe, 8000b0a47f5SMiklos Szeredi struct file *out, loff_t *ppos, 8010b0a47f5SMiklos Szeredi size_t len, unsigned int flags) 8020b0a47f5SMiklos Szeredi { 803b2858d7dSMiklos Szeredi ssize_t ret; 8040b0a47f5SMiklos Szeredi 805b2858d7dSMiklos Szeredi ret = splice_from_pipe(pipe, out, ppos, len, flags, write_pipe_buf); 806b2858d7dSMiklos Szeredi if (ret > 0) 807b2858d7dSMiklos Szeredi *ppos += ret; 8080b0a47f5SMiklos Szeredi 809b2858d7dSMiklos Szeredi return ret; 8100b0a47f5SMiklos Szeredi } 8110b0a47f5SMiklos Szeredi 81283f9135bSJens Axboe /** 81383f9135bSJens Axboe * generic_splice_sendpage - splice data from a pipe to a socket 814932cc6d4SJens Axboe * @pipe: pipe to splice from 81583f9135bSJens Axboe * @out: socket to write to 816932cc6d4SJens Axboe * @ppos: position in @out 81783f9135bSJens Axboe * @len: number of bytes to splice 81883f9135bSJens Axboe * @flags: splice modifier flags 81983f9135bSJens Axboe * 820932cc6d4SJens Axboe * Description: 82183f9135bSJens Axboe * Will send @len bytes from the pipe to a network socket. No data copying 82283f9135bSJens Axboe * is involved. 82383f9135bSJens Axboe * 82483f9135bSJens Axboe */ 8253a326a2cSIngo Molnar ssize_t generic_splice_sendpage(struct pipe_inode_info *pipe, struct file *out, 826cbb7e577SJens Axboe loff_t *ppos, size_t len, unsigned int flags) 8275274f052SJens Axboe { 82800522fb4SJens Axboe return splice_from_pipe(pipe, out, ppos, len, flags, pipe_to_sendpage); 8295274f052SJens Axboe } 8305274f052SJens Axboe 831059a8f37SJens Axboe EXPORT_SYMBOL(generic_splice_sendpage); 832a0f06780SJeff Garzik 83383f9135bSJens Axboe /* 83483f9135bSJens Axboe * Attempt to initiate a splice from pipe to file. 83583f9135bSJens Axboe */ 8363a326a2cSIngo Molnar static long do_splice_from(struct pipe_inode_info *pipe, struct file *out, 837cbb7e577SJens Axboe loff_t *ppos, size_t len, unsigned int flags) 8385274f052SJens Axboe { 8390b0a47f5SMiklos Szeredi ssize_t (*splice_write)(struct pipe_inode_info *, struct file *, 8400b0a47f5SMiklos Szeredi loff_t *, size_t, unsigned int); 8415274f052SJens Axboe 84272c2d531SAl Viro if (out->f_op->splice_write) 8430b0a47f5SMiklos Szeredi splice_write = out->f_op->splice_write; 844cc56f7deSChangli Gao else 8450b0a47f5SMiklos Szeredi splice_write = default_file_splice_write; 8460b0a47f5SMiklos Szeredi 847500368f7SAl Viro return splice_write(pipe, out, ppos, len, flags); 8485274f052SJens Axboe } 8495274f052SJens Axboe 85083f9135bSJens Axboe /* 85183f9135bSJens Axboe * Attempt to initiate a splice from a file to a pipe. 85283f9135bSJens Axboe */ 853cbb7e577SJens Axboe static long do_splice_to(struct file *in, loff_t *ppos, 854cbb7e577SJens Axboe struct pipe_inode_info *pipe, size_t len, 855cbb7e577SJens Axboe unsigned int flags) 8565274f052SJens Axboe { 8576818173bSMiklos Szeredi ssize_t (*splice_read)(struct file *, loff_t *, 8586818173bSMiklos Szeredi struct pipe_inode_info *, size_t, unsigned int); 8595274f052SJens Axboe int ret; 8605274f052SJens Axboe 86149570e9bSJens Axboe if (unlikely(!(in->f_mode & FMODE_READ))) 8625274f052SJens Axboe return -EBADF; 8635274f052SJens Axboe 864cbb7e577SJens Axboe ret = rw_verify_area(READ, in, ppos, len); 8655274f052SJens Axboe if (unlikely(ret < 0)) 8665274f052SJens Axboe return ret; 8675274f052SJens Axboe 86803cc0789SAl Viro if (unlikely(len > MAX_RW_COUNT)) 86903cc0789SAl Viro len = MAX_RW_COUNT; 87003cc0789SAl Viro 87172c2d531SAl Viro if (in->f_op->splice_read) 8726818173bSMiklos Szeredi splice_read = in->f_op->splice_read; 873cc56f7deSChangli Gao else 8746818173bSMiklos Szeredi splice_read = default_file_splice_read; 8756818173bSMiklos Szeredi 8766818173bSMiklos Szeredi return splice_read(in, ppos, pipe, len, flags); 8775274f052SJens Axboe } 8785274f052SJens Axboe 879932cc6d4SJens Axboe /** 880932cc6d4SJens Axboe * splice_direct_to_actor - splices data directly between two non-pipes 881932cc6d4SJens Axboe * @in: file to splice from 882932cc6d4SJens Axboe * @sd: actor information on where to splice to 883932cc6d4SJens Axboe * @actor: handles the data splicing 884932cc6d4SJens Axboe * 885932cc6d4SJens Axboe * Description: 886932cc6d4SJens Axboe * This is a special case helper to splice directly between two 887932cc6d4SJens Axboe * points, without requiring an explicit pipe. Internally an allocated 888932cc6d4SJens Axboe * pipe is cached in the process, and reused during the lifetime of 889932cc6d4SJens Axboe * that process. 890932cc6d4SJens Axboe * 891c66ab6faSJens Axboe */ 892c66ab6faSJens Axboe ssize_t splice_direct_to_actor(struct file *in, struct splice_desc *sd, 893c66ab6faSJens Axboe splice_direct_actor *actor) 894b92ce558SJens Axboe { 895b92ce558SJens Axboe struct pipe_inode_info *pipe; 896b92ce558SJens Axboe long ret, bytes; 897b92ce558SJens Axboe umode_t i_mode; 898c66ab6faSJens Axboe size_t len; 8990ff28d9fSChristophe Leroy int i, flags, more; 900b92ce558SJens Axboe 901b92ce558SJens Axboe /* 902b92ce558SJens Axboe * We require the input being a regular file, as we don't want to 903b92ce558SJens Axboe * randomly drop data for eg socket -> socket splicing. Use the 904b92ce558SJens Axboe * piped splicing for that! 905b92ce558SJens Axboe */ 906496ad9aaSAl Viro i_mode = file_inode(in)->i_mode; 907b92ce558SJens Axboe if (unlikely(!S_ISREG(i_mode) && !S_ISBLK(i_mode))) 908b92ce558SJens Axboe return -EINVAL; 909b92ce558SJens Axboe 910b92ce558SJens Axboe /* 911b92ce558SJens Axboe * neither in nor out is a pipe, setup an internal pipe attached to 912b92ce558SJens Axboe * 'out' and transfer the wanted data from 'in' to 'out' through that 913b92ce558SJens Axboe */ 914b92ce558SJens Axboe pipe = current->splice_pipe; 91549570e9bSJens Axboe if (unlikely(!pipe)) { 9167bee130eSAl Viro pipe = alloc_pipe_info(); 917b92ce558SJens Axboe if (!pipe) 918b92ce558SJens Axboe return -ENOMEM; 919b92ce558SJens Axboe 920b92ce558SJens Axboe /* 921b92ce558SJens Axboe * We don't have an immediate reader, but we'll read the stuff 92200522fb4SJens Axboe * out of the pipe right after the splice_to_pipe(). So set 923b92ce558SJens Axboe * PIPE_READERS appropriately. 924b92ce558SJens Axboe */ 925b92ce558SJens Axboe pipe->readers = 1; 926b92ce558SJens Axboe 927b92ce558SJens Axboe current->splice_pipe = pipe; 928b92ce558SJens Axboe } 929b92ce558SJens Axboe 930b92ce558SJens Axboe /* 93173d62d83SIngo Molnar * Do the splice. 932b92ce558SJens Axboe */ 933b92ce558SJens Axboe ret = 0; 934b92ce558SJens Axboe bytes = 0; 935c66ab6faSJens Axboe len = sd->total_len; 936c66ab6faSJens Axboe flags = sd->flags; 937c66ab6faSJens Axboe 938c66ab6faSJens Axboe /* 939c66ab6faSJens Axboe * Don't block on output, we have to drain the direct pipe. 940c66ab6faSJens Axboe */ 941c66ab6faSJens Axboe sd->flags &= ~SPLICE_F_NONBLOCK; 9420ff28d9fSChristophe Leroy more = sd->flags & SPLICE_F_MORE; 943b92ce558SJens Axboe 94417614445SDarrick J. Wong WARN_ON_ONCE(pipe->nrbufs != 0); 94517614445SDarrick J. Wong 946b92ce558SJens Axboe while (len) { 94751a92c0fSJens Axboe size_t read_len; 948a82c53a0STom Zanussi loff_t pos = sd->pos, prev_pos = pos; 949b92ce558SJens Axboe 95017614445SDarrick J. Wong /* Don't try to read more the pipe has space for. */ 95117614445SDarrick J. Wong read_len = min_t(size_t, len, 95217614445SDarrick J. Wong (pipe->buffers - pipe->nrbufs) << PAGE_SHIFT); 95317614445SDarrick J. Wong ret = do_splice_to(in, &pos, pipe, read_len, flags); 95451a92c0fSJens Axboe if (unlikely(ret <= 0)) 955b92ce558SJens Axboe goto out_release; 956b92ce558SJens Axboe 957b92ce558SJens Axboe read_len = ret; 958c66ab6faSJens Axboe sd->total_len = read_len; 959b92ce558SJens Axboe 960b92ce558SJens Axboe /* 9610ff28d9fSChristophe Leroy * If more data is pending, set SPLICE_F_MORE 9620ff28d9fSChristophe Leroy * If this is the last data and SPLICE_F_MORE was not set 9630ff28d9fSChristophe Leroy * initially, clears it. 9640ff28d9fSChristophe Leroy */ 9650ff28d9fSChristophe Leroy if (read_len < len) 9660ff28d9fSChristophe Leroy sd->flags |= SPLICE_F_MORE; 9670ff28d9fSChristophe Leroy else if (!more) 9680ff28d9fSChristophe Leroy sd->flags &= ~SPLICE_F_MORE; 9690ff28d9fSChristophe Leroy /* 970b92ce558SJens Axboe * NOTE: nonblocking mode only applies to the input. We 971b92ce558SJens Axboe * must not do the output in nonblocking mode as then we 972b92ce558SJens Axboe * could get stuck data in the internal pipe: 973b92ce558SJens Axboe */ 974c66ab6faSJens Axboe ret = actor(pipe, sd); 975a82c53a0STom Zanussi if (unlikely(ret <= 0)) { 976a82c53a0STom Zanussi sd->pos = prev_pos; 977b92ce558SJens Axboe goto out_release; 978a82c53a0STom Zanussi } 979b92ce558SJens Axboe 980b92ce558SJens Axboe bytes += ret; 981b92ce558SJens Axboe len -= ret; 982bcd4f3acSJens Axboe sd->pos = pos; 983b92ce558SJens Axboe 984a82c53a0STom Zanussi if (ret < read_len) { 985a82c53a0STom Zanussi sd->pos = prev_pos + ret; 98651a92c0fSJens Axboe goto out_release; 987b92ce558SJens Axboe } 988a82c53a0STom Zanussi } 989b92ce558SJens Axboe 9909e97198dSJens Axboe done: 991b92ce558SJens Axboe pipe->nrbufs = pipe->curbuf = 0; 9929e97198dSJens Axboe file_accessed(in); 993b92ce558SJens Axboe return bytes; 994b92ce558SJens Axboe 995b92ce558SJens Axboe out_release: 996b92ce558SJens Axboe /* 997b92ce558SJens Axboe * If we did an incomplete transfer we must release 998b92ce558SJens Axboe * the pipe buffers in question: 999b92ce558SJens Axboe */ 100035f3d14dSJens Axboe for (i = 0; i < pipe->buffers; i++) { 1001b92ce558SJens Axboe struct pipe_buffer *buf = pipe->bufs + i; 1002b92ce558SJens Axboe 1003a779638cSMiklos Szeredi if (buf->ops) 1004a779638cSMiklos Szeredi pipe_buf_release(pipe, buf); 1005b92ce558SJens Axboe } 1006b92ce558SJens Axboe 10079e97198dSJens Axboe if (!bytes) 10089e97198dSJens Axboe bytes = ret; 1009b92ce558SJens Axboe 10109e97198dSJens Axboe goto done; 1011c66ab6faSJens Axboe } 1012c66ab6faSJens Axboe EXPORT_SYMBOL(splice_direct_to_actor); 1013c66ab6faSJens Axboe 1014c66ab6faSJens Axboe static int direct_splice_actor(struct pipe_inode_info *pipe, 1015c66ab6faSJens Axboe struct splice_desc *sd) 1016c66ab6faSJens Axboe { 10176a14b90bSJens Axboe struct file *file = sd->u.file; 1018c66ab6faSJens Axboe 10197995bd28SAl Viro return do_splice_from(pipe, file, sd->opos, sd->total_len, 10202cb4b05eSChangli Gao sd->flags); 1021c66ab6faSJens Axboe } 1022c66ab6faSJens Axboe 1023932cc6d4SJens Axboe /** 1024932cc6d4SJens Axboe * do_splice_direct - splices data directly between two files 1025932cc6d4SJens Axboe * @in: file to splice from 1026932cc6d4SJens Axboe * @ppos: input file offset 1027932cc6d4SJens Axboe * @out: file to splice to 1028acdb37c3SRandy Dunlap * @opos: output file offset 1029932cc6d4SJens Axboe * @len: number of bytes to splice 1030932cc6d4SJens Axboe * @flags: splice modifier flags 1031932cc6d4SJens Axboe * 1032932cc6d4SJens Axboe * Description: 1033932cc6d4SJens Axboe * For use by do_sendfile(). splice can easily emulate sendfile, but 1034932cc6d4SJens Axboe * doing it in the application would incur an extra system call 1035932cc6d4SJens Axboe * (splice in + splice out, as compared to just sendfile()). So this helper 1036932cc6d4SJens Axboe * can splice directly through a process-private pipe. 1037932cc6d4SJens Axboe * 1038932cc6d4SJens Axboe */ 1039c66ab6faSJens Axboe long do_splice_direct(struct file *in, loff_t *ppos, struct file *out, 10407995bd28SAl Viro loff_t *opos, size_t len, unsigned int flags) 1041c66ab6faSJens Axboe { 1042c66ab6faSJens Axboe struct splice_desc sd = { 1043c66ab6faSJens Axboe .len = len, 1044c66ab6faSJens Axboe .total_len = len, 1045c66ab6faSJens Axboe .flags = flags, 1046c66ab6faSJens Axboe .pos = *ppos, 10476a14b90bSJens Axboe .u.file = out, 10487995bd28SAl Viro .opos = opos, 1049c66ab6faSJens Axboe }; 105051a92c0fSJens Axboe long ret; 1051c66ab6faSJens Axboe 105218c67cb9SAl Viro if (unlikely(!(out->f_mode & FMODE_WRITE))) 105318c67cb9SAl Viro return -EBADF; 105418c67cb9SAl Viro 105518c67cb9SAl Viro if (unlikely(out->f_flags & O_APPEND)) 105618c67cb9SAl Viro return -EINVAL; 105718c67cb9SAl Viro 105818c67cb9SAl Viro ret = rw_verify_area(WRITE, out, opos, len); 105918c67cb9SAl Viro if (unlikely(ret < 0)) 106018c67cb9SAl Viro return ret; 106118c67cb9SAl Viro 1062c66ab6faSJens Axboe ret = splice_direct_to_actor(in, &sd, direct_splice_actor); 106351a92c0fSJens Axboe if (ret > 0) 1064a82c53a0STom Zanussi *ppos = sd.pos; 106551a92c0fSJens Axboe 1066c66ab6faSJens Axboe return ret; 1067b92ce558SJens Axboe } 10681c118596SMiklos Szeredi EXPORT_SYMBOL(do_splice_direct); 1069b92ce558SJens Axboe 10708924feffSAl Viro static int wait_for_space(struct pipe_inode_info *pipe, unsigned flags) 10718924feffSAl Viro { 107252bce911SLinus Torvalds for (;;) { 107352bce911SLinus Torvalds if (unlikely(!pipe->readers)) { 107452bce911SLinus Torvalds send_sig(SIGPIPE, current, 0); 107552bce911SLinus Torvalds return -EPIPE; 107652bce911SLinus Torvalds } 107752bce911SLinus Torvalds if (pipe->nrbufs != pipe->buffers) 107852bce911SLinus Torvalds return 0; 10798924feffSAl Viro if (flags & SPLICE_F_NONBLOCK) 10808924feffSAl Viro return -EAGAIN; 10818924feffSAl Viro if (signal_pending(current)) 10828924feffSAl Viro return -ERESTARTSYS; 10838924feffSAl Viro pipe->waiting_writers++; 10848924feffSAl Viro pipe_wait(pipe); 10858924feffSAl Viro pipe->waiting_writers--; 10868924feffSAl Viro } 10878924feffSAl Viro } 10888924feffSAl Viro 10897c77f0b3SMiklos Szeredi static int splice_pipe_to_pipe(struct pipe_inode_info *ipipe, 10907c77f0b3SMiklos Szeredi struct pipe_inode_info *opipe, 10917c77f0b3SMiklos Szeredi size_t len, unsigned int flags); 1092ddac0d39SJens Axboe 1093ddac0d39SJens Axboe /* 109483f9135bSJens Axboe * Determine where to splice to/from. 109583f9135bSJens Axboe */ 1096529565dcSIngo Molnar static long do_splice(struct file *in, loff_t __user *off_in, 1097529565dcSIngo Molnar struct file *out, loff_t __user *off_out, 1098529565dcSIngo Molnar size_t len, unsigned int flags) 10995274f052SJens Axboe { 11007c77f0b3SMiklos Szeredi struct pipe_inode_info *ipipe; 11017c77f0b3SMiklos Szeredi struct pipe_inode_info *opipe; 11027995bd28SAl Viro loff_t offset; 1103a4514ebdSJens Axboe long ret; 11045274f052SJens Axboe 110571993e62SLinus Torvalds ipipe = get_pipe_info(in); 110671993e62SLinus Torvalds opipe = get_pipe_info(out); 11077c77f0b3SMiklos Szeredi 11087c77f0b3SMiklos Szeredi if (ipipe && opipe) { 11097c77f0b3SMiklos Szeredi if (off_in || off_out) 11107c77f0b3SMiklos Szeredi return -ESPIPE; 11117c77f0b3SMiklos Szeredi 11127c77f0b3SMiklos Szeredi if (!(in->f_mode & FMODE_READ)) 11137c77f0b3SMiklos Szeredi return -EBADF; 11147c77f0b3SMiklos Szeredi 11157c77f0b3SMiklos Szeredi if (!(out->f_mode & FMODE_WRITE)) 11167c77f0b3SMiklos Szeredi return -EBADF; 11177c77f0b3SMiklos Szeredi 11187c77f0b3SMiklos Szeredi /* Splicing to self would be fun, but... */ 11197c77f0b3SMiklos Szeredi if (ipipe == opipe) 11207c77f0b3SMiklos Szeredi return -EINVAL; 11217c77f0b3SMiklos Szeredi 1122ee5e0011SSlavomir Kaslev if ((in->f_flags | out->f_flags) & O_NONBLOCK) 1123ee5e0011SSlavomir Kaslev flags |= SPLICE_F_NONBLOCK; 1124ee5e0011SSlavomir Kaslev 11257c77f0b3SMiklos Szeredi return splice_pipe_to_pipe(ipipe, opipe, len, flags); 11267c77f0b3SMiklos Szeredi } 11277c77f0b3SMiklos Szeredi 11287c77f0b3SMiklos Szeredi if (ipipe) { 1129529565dcSIngo Molnar if (off_in) 1130529565dcSIngo Molnar return -ESPIPE; 1131b92ce558SJens Axboe if (off_out) { 113219c9a49bSChangli Gao if (!(out->f_mode & FMODE_PWRITE)) 1133b92ce558SJens Axboe return -EINVAL; 1134cbb7e577SJens Axboe if (copy_from_user(&offset, off_out, sizeof(loff_t))) 1135b92ce558SJens Axboe return -EFAULT; 11367995bd28SAl Viro } else { 11377995bd28SAl Viro offset = out->f_pos; 11387995bd28SAl Viro } 1139529565dcSIngo Molnar 114018c67cb9SAl Viro if (unlikely(!(out->f_mode & FMODE_WRITE))) 114118c67cb9SAl Viro return -EBADF; 114218c67cb9SAl Viro 114318c67cb9SAl Viro if (unlikely(out->f_flags & O_APPEND)) 114418c67cb9SAl Viro return -EINVAL; 114518c67cb9SAl Viro 114618c67cb9SAl Viro ret = rw_verify_area(WRITE, out, &offset, len); 114718c67cb9SAl Viro if (unlikely(ret < 0)) 114818c67cb9SAl Viro return ret; 114918c67cb9SAl Viro 1150ee5e0011SSlavomir Kaslev if (in->f_flags & O_NONBLOCK) 1151ee5e0011SSlavomir Kaslev flags |= SPLICE_F_NONBLOCK; 1152ee5e0011SSlavomir Kaslev 1153500368f7SAl Viro file_start_write(out); 11547995bd28SAl Viro ret = do_splice_from(ipipe, out, &offset, len, flags); 1155500368f7SAl Viro file_end_write(out); 1156a4514ebdSJens Axboe 11577995bd28SAl Viro if (!off_out) 11587995bd28SAl Viro out->f_pos = offset; 11597995bd28SAl Viro else if (copy_to_user(off_out, &offset, sizeof(loff_t))) 1160a4514ebdSJens Axboe ret = -EFAULT; 1161a4514ebdSJens Axboe 1162a4514ebdSJens Axboe return ret; 1163529565dcSIngo Molnar } 11645274f052SJens Axboe 11657c77f0b3SMiklos Szeredi if (opipe) { 1166529565dcSIngo Molnar if (off_out) 1167529565dcSIngo Molnar return -ESPIPE; 1168b92ce558SJens Axboe if (off_in) { 116919c9a49bSChangli Gao if (!(in->f_mode & FMODE_PREAD)) 1170b92ce558SJens Axboe return -EINVAL; 1171cbb7e577SJens Axboe if (copy_from_user(&offset, off_in, sizeof(loff_t))) 1172b92ce558SJens Axboe return -EFAULT; 11737995bd28SAl Viro } else { 11747995bd28SAl Viro offset = in->f_pos; 11757995bd28SAl Viro } 1176529565dcSIngo Molnar 1177ee5e0011SSlavomir Kaslev if (out->f_flags & O_NONBLOCK) 1178ee5e0011SSlavomir Kaslev flags |= SPLICE_F_NONBLOCK; 1179ee5e0011SSlavomir Kaslev 11808924feffSAl Viro pipe_lock(opipe); 11818924feffSAl Viro ret = wait_for_space(opipe, flags); 11828924feffSAl Viro if (!ret) 11837995bd28SAl Viro ret = do_splice_to(in, &offset, opipe, len, flags); 11848924feffSAl Viro pipe_unlock(opipe); 11858924feffSAl Viro if (ret > 0) 11868924feffSAl Viro wakeup_pipe_readers(opipe); 11877995bd28SAl Viro if (!off_in) 11887995bd28SAl Viro in->f_pos = offset; 11897995bd28SAl Viro else if (copy_to_user(off_in, &offset, sizeof(loff_t))) 1190a4514ebdSJens Axboe ret = -EFAULT; 1191a4514ebdSJens Axboe 1192a4514ebdSJens Axboe return ret; 1193529565dcSIngo Molnar } 11945274f052SJens Axboe 11955274f052SJens Axboe return -EINVAL; 11965274f052SJens Axboe } 11975274f052SJens Axboe 119879fddc4eSAl Viro static int iter_to_pipe(struct iov_iter *from, 119979fddc4eSAl Viro struct pipe_inode_info *pipe, 120079fddc4eSAl Viro unsigned flags) 1201912d35f8SJens Axboe { 120279fddc4eSAl Viro struct pipe_buffer buf = { 120379fddc4eSAl Viro .ops = &user_page_pipe_buf_ops, 120479fddc4eSAl Viro .flags = flags 120579fddc4eSAl Viro }; 120679fddc4eSAl Viro size_t total = 0; 120779fddc4eSAl Viro int ret = 0; 120879fddc4eSAl Viro bool failed = false; 120979fddc4eSAl Viro 121079fddc4eSAl Viro while (iov_iter_count(from) && !failed) { 121179fddc4eSAl Viro struct page *pages[16]; 1212db85a9ebSAl Viro ssize_t copied; 1213db85a9ebSAl Viro size_t start; 121479fddc4eSAl Viro int n; 1215912d35f8SJens Axboe 121679fddc4eSAl Viro copied = iov_iter_get_pages(from, pages, ~0UL, 16, &start); 121779fddc4eSAl Viro if (copied <= 0) { 121879fddc4eSAl Viro ret = copied; 121979fddc4eSAl Viro break; 122079fddc4eSAl Viro } 1221912d35f8SJens Axboe 122279fddc4eSAl Viro for (n = 0; copied; n++, start = 0) { 1223db85a9ebSAl Viro int size = min_t(int, copied, PAGE_SIZE - start); 122479fddc4eSAl Viro if (!failed) { 122579fddc4eSAl Viro buf.page = pages[n]; 122679fddc4eSAl Viro buf.offset = start; 122779fddc4eSAl Viro buf.len = size; 122879fddc4eSAl Viro ret = add_to_pipe(pipe, &buf); 122979fddc4eSAl Viro if (unlikely(ret < 0)) { 123079fddc4eSAl Viro failed = true; 123179fddc4eSAl Viro } else { 123279fddc4eSAl Viro iov_iter_advance(from, ret); 123379fddc4eSAl Viro total += ret; 123479fddc4eSAl Viro } 123579fddc4eSAl Viro } else { 123679fddc4eSAl Viro put_page(pages[n]); 123779fddc4eSAl Viro } 1238db85a9ebSAl Viro copied -= size; 1239912d35f8SJens Axboe } 1240912d35f8SJens Axboe } 124179fddc4eSAl Viro return total ? total : ret; 1242912d35f8SJens Axboe } 1243912d35f8SJens Axboe 12446a14b90bSJens Axboe static int pipe_to_user(struct pipe_inode_info *pipe, struct pipe_buffer *buf, 12456a14b90bSJens Axboe struct splice_desc *sd) 12466a14b90bSJens Axboe { 12476130f531SAl Viro int n = copy_page_to_iter(buf->page, buf->offset, sd->len, sd->u.data); 12486130f531SAl Viro return n == sd->len ? n : -EFAULT; 12496a14b90bSJens Axboe } 12506a14b90bSJens Axboe 12516a14b90bSJens Axboe /* 12526a14b90bSJens Axboe * For lack of a better implementation, implement vmsplice() to userspace 12536a14b90bSJens Axboe * as a simple copy of the pipes pages to the user iov. 12546a14b90bSJens Axboe */ 125587a3002aSAl Viro static long vmsplice_to_user(struct file *file, struct iov_iter *iter, 125687a3002aSAl Viro unsigned int flags) 12576a14b90bSJens Axboe { 125887a3002aSAl Viro struct pipe_inode_info *pipe = get_pipe_info(file); 125987a3002aSAl Viro struct splice_desc sd = { 126087a3002aSAl Viro .total_len = iov_iter_count(iter), 126187a3002aSAl Viro .flags = flags, 126287a3002aSAl Viro .u.data = iter 126387a3002aSAl Viro }; 126487a3002aSAl Viro long ret = 0; 12656a14b90bSJens Axboe 12666a14b90bSJens Axboe if (!pipe) 12676a14b90bSJens Axboe return -EBADF; 12686a14b90bSJens Axboe 1269345995faSAl Viro if (sd.total_len) { 12706130f531SAl Viro pipe_lock(pipe); 12716130f531SAl Viro ret = __splice_from_pipe(pipe, &sd, pipe_to_user); 127261e0d47cSMiklos Szeredi pipe_unlock(pipe); 1273345995faSAl Viro } 12746a14b90bSJens Axboe 12756a14b90bSJens Axboe return ret; 12766a14b90bSJens Axboe } 12776a14b90bSJens Axboe 1278912d35f8SJens Axboe /* 1279912d35f8SJens Axboe * vmsplice splices a user address range into a pipe. It can be thought of 1280912d35f8SJens Axboe * as splice-from-memory, where the regular splice is splice-from-file (or 1281912d35f8SJens Axboe * to file). In both cases the output is a pipe, naturally. 1282912d35f8SJens Axboe */ 128387a3002aSAl Viro static long vmsplice_to_pipe(struct file *file, struct iov_iter *iter, 128487a3002aSAl Viro unsigned int flags) 1285912d35f8SJens Axboe { 1286ddac0d39SJens Axboe struct pipe_inode_info *pipe; 128787a3002aSAl Viro long ret = 0; 128879fddc4eSAl Viro unsigned buf_flag = 0; 128979fddc4eSAl Viro 129079fddc4eSAl Viro if (flags & SPLICE_F_GIFT) 129179fddc4eSAl Viro buf_flag = PIPE_BUF_FLAG_GIFT; 1292912d35f8SJens Axboe 129371993e62SLinus Torvalds pipe = get_pipe_info(file); 1294ddac0d39SJens Axboe if (!pipe) 1295912d35f8SJens Axboe return -EBADF; 1296912d35f8SJens Axboe 12978924feffSAl Viro pipe_lock(pipe); 12988924feffSAl Viro ret = wait_for_space(pipe, flags); 129979fddc4eSAl Viro if (!ret) 130087a3002aSAl Viro ret = iter_to_pipe(iter, pipe, buf_flag); 13018924feffSAl Viro pipe_unlock(pipe); 13028924feffSAl Viro if (ret > 0) 13038924feffSAl Viro wakeup_pipe_readers(pipe); 130435f3d14dSJens Axboe return ret; 1305912d35f8SJens Axboe } 1306912d35f8SJens Axboe 130787a3002aSAl Viro static int vmsplice_type(struct fd f, int *type) 130887a3002aSAl Viro { 130987a3002aSAl Viro if (!f.file) 131087a3002aSAl Viro return -EBADF; 131187a3002aSAl Viro if (f.file->f_mode & FMODE_WRITE) { 131287a3002aSAl Viro *type = WRITE; 131387a3002aSAl Viro } else if (f.file->f_mode & FMODE_READ) { 131487a3002aSAl Viro *type = READ; 131587a3002aSAl Viro } else { 131687a3002aSAl Viro fdput(f); 131787a3002aSAl Viro return -EBADF; 131887a3002aSAl Viro } 131987a3002aSAl Viro return 0; 132087a3002aSAl Viro } 132187a3002aSAl Viro 13226a14b90bSJens Axboe /* 13236a14b90bSJens Axboe * Note that vmsplice only really supports true splicing _from_ user memory 13246a14b90bSJens Axboe * to a pipe, not the other way around. Splicing from user memory is a simple 13256a14b90bSJens Axboe * operation that can be supported without any funky alignment restrictions 13266a14b90bSJens Axboe * or nasty vm tricks. We simply map in the user memory and fill them into 13276a14b90bSJens Axboe * a pipe. The reverse isn't quite as easy, though. There are two possible 13286a14b90bSJens Axboe * solutions for that: 13296a14b90bSJens Axboe * 13306a14b90bSJens Axboe * - memcpy() the data internally, at which point we might as well just 13316a14b90bSJens Axboe * do a regular read() on the buffer anyway. 13326a14b90bSJens Axboe * - Lots of nasty vm tricks, that are neither fast nor flexible (it 13336a14b90bSJens Axboe * has restriction limitations on both ends of the pipe). 13346a14b90bSJens Axboe * 13356a14b90bSJens Axboe * Currently we punt and implement it as a normal copy, see pipe_to_user(). 13366a14b90bSJens Axboe * 13376a14b90bSJens Axboe */ 133887a3002aSAl Viro static long do_vmsplice(struct file *f, struct iov_iter *iter, unsigned int flags) 1339912d35f8SJens Axboe { 13403d6ea290SAl Viro if (unlikely(flags & ~SPLICE_F_ALL)) 13413d6ea290SAl Viro return -EINVAL; 134287a3002aSAl Viro 134387a3002aSAl Viro if (!iov_iter_count(iter)) 13446a14b90bSJens Axboe return 0; 13456a14b90bSJens Axboe 134687a3002aSAl Viro if (iov_iter_rw(iter) == WRITE) 134787a3002aSAl Viro return vmsplice_to_pipe(f, iter, flags); 134887a3002aSAl Viro else 134987a3002aSAl Viro return vmsplice_to_user(f, iter, flags); 1350912d35f8SJens Axboe } 1351912d35f8SJens Axboe 135287a3002aSAl Viro SYSCALL_DEFINE4(vmsplice, int, fd, const struct iovec __user *, uiov, 135330cfe4efSDominik Brodowski unsigned long, nr_segs, unsigned int, flags) 135430cfe4efSDominik Brodowski { 135587a3002aSAl Viro struct iovec iovstack[UIO_FASTIOV]; 135687a3002aSAl Viro struct iovec *iov = iovstack; 135787a3002aSAl Viro struct iov_iter iter; 135887a3002aSAl Viro long error; 135987a3002aSAl Viro struct fd f; 136087a3002aSAl Viro int type; 136187a3002aSAl Viro 136287a3002aSAl Viro f = fdget(fd); 136387a3002aSAl Viro error = vmsplice_type(f, &type); 136487a3002aSAl Viro if (error) 136587a3002aSAl Viro return error; 136687a3002aSAl Viro 136787a3002aSAl Viro error = import_iovec(type, uiov, nr_segs, 136887a3002aSAl Viro ARRAY_SIZE(iovstack), &iov, &iter); 136987a3002aSAl Viro if (!error) { 137087a3002aSAl Viro error = do_vmsplice(f.file, &iter, flags); 137187a3002aSAl Viro kfree(iov); 137287a3002aSAl Viro } 137387a3002aSAl Viro fdput(f); 137487a3002aSAl Viro return error; 137530cfe4efSDominik Brodowski } 137630cfe4efSDominik Brodowski 137776b021d0SAl Viro #ifdef CONFIG_COMPAT 137876b021d0SAl Viro COMPAT_SYSCALL_DEFINE4(vmsplice, int, fd, const struct compat_iovec __user *, iov32, 137976b021d0SAl Viro unsigned int, nr_segs, unsigned int, flags) 138076b021d0SAl Viro { 138187a3002aSAl Viro struct iovec iovstack[UIO_FASTIOV]; 138287a3002aSAl Viro struct iovec *iov = iovstack; 138387a3002aSAl Viro struct iov_iter iter; 138487a3002aSAl Viro long error; 138587a3002aSAl Viro struct fd f; 138687a3002aSAl Viro int type; 138787a3002aSAl Viro 138887a3002aSAl Viro f = fdget(fd); 138987a3002aSAl Viro error = vmsplice_type(f, &type); 139087a3002aSAl Viro if (error) 139187a3002aSAl Viro return error; 139287a3002aSAl Viro 139387a3002aSAl Viro error = compat_import_iovec(type, iov32, nr_segs, 139487a3002aSAl Viro ARRAY_SIZE(iovstack), &iov, &iter); 139587a3002aSAl Viro if (!error) { 139687a3002aSAl Viro error = do_vmsplice(f.file, &iter, flags); 139787a3002aSAl Viro kfree(iov); 139876b021d0SAl Viro } 139987a3002aSAl Viro fdput(f); 140087a3002aSAl Viro return error; 140176b021d0SAl Viro } 140276b021d0SAl Viro #endif 140376b021d0SAl Viro 1404836f92adSHeiko Carstens SYSCALL_DEFINE6(splice, int, fd_in, loff_t __user *, off_in, 1405836f92adSHeiko Carstens int, fd_out, loff_t __user *, off_out, 1406836f92adSHeiko Carstens size_t, len, unsigned int, flags) 14075274f052SJens Axboe { 14082903ff01SAl Viro struct fd in, out; 14095274f052SJens Axboe long error; 14105274f052SJens Axboe 14115274f052SJens Axboe if (unlikely(!len)) 14125274f052SJens Axboe return 0; 14135274f052SJens Axboe 14143d6ea290SAl Viro if (unlikely(flags & ~SPLICE_F_ALL)) 14153d6ea290SAl Viro return -EINVAL; 14163d6ea290SAl Viro 14175274f052SJens Axboe error = -EBADF; 14182903ff01SAl Viro in = fdget(fd_in); 14192903ff01SAl Viro if (in.file) { 14202903ff01SAl Viro if (in.file->f_mode & FMODE_READ) { 14212903ff01SAl Viro out = fdget(fd_out); 14222903ff01SAl Viro if (out.file) { 14232903ff01SAl Viro if (out.file->f_mode & FMODE_WRITE) 14242903ff01SAl Viro error = do_splice(in.file, off_in, 14252903ff01SAl Viro out.file, off_out, 1426529565dcSIngo Molnar len, flags); 14272903ff01SAl Viro fdput(out); 14285274f052SJens Axboe } 14295274f052SJens Axboe } 14302903ff01SAl Viro fdput(in); 14315274f052SJens Axboe } 14325274f052SJens Axboe return error; 14335274f052SJens Axboe } 143470524490SJens Axboe 143570524490SJens Axboe /* 1436aadd06e5SJens Axboe * Make sure there's data to read. Wait for input if we can, otherwise 1437aadd06e5SJens Axboe * return an appropriate error. 1438aadd06e5SJens Axboe */ 14397c77f0b3SMiklos Szeredi static int ipipe_prep(struct pipe_inode_info *pipe, unsigned int flags) 1440aadd06e5SJens Axboe { 1441aadd06e5SJens Axboe int ret; 1442aadd06e5SJens Axboe 1443aadd06e5SJens Axboe /* 1444aadd06e5SJens Axboe * Check ->nrbufs without the inode lock first. This function 1445aadd06e5SJens Axboe * is speculative anyways, so missing one is ok. 1446aadd06e5SJens Axboe */ 1447aadd06e5SJens Axboe if (pipe->nrbufs) 1448aadd06e5SJens Axboe return 0; 1449aadd06e5SJens Axboe 1450aadd06e5SJens Axboe ret = 0; 145161e0d47cSMiklos Szeredi pipe_lock(pipe); 1452aadd06e5SJens Axboe 1453aadd06e5SJens Axboe while (!pipe->nrbufs) { 1454aadd06e5SJens Axboe if (signal_pending(current)) { 1455aadd06e5SJens Axboe ret = -ERESTARTSYS; 1456aadd06e5SJens Axboe break; 1457aadd06e5SJens Axboe } 1458aadd06e5SJens Axboe if (!pipe->writers) 1459aadd06e5SJens Axboe break; 1460aadd06e5SJens Axboe if (!pipe->waiting_writers) { 1461aadd06e5SJens Axboe if (flags & SPLICE_F_NONBLOCK) { 1462aadd06e5SJens Axboe ret = -EAGAIN; 1463aadd06e5SJens Axboe break; 1464aadd06e5SJens Axboe } 1465aadd06e5SJens Axboe } 1466aadd06e5SJens Axboe pipe_wait(pipe); 1467aadd06e5SJens Axboe } 1468aadd06e5SJens Axboe 146961e0d47cSMiklos Szeredi pipe_unlock(pipe); 1470aadd06e5SJens Axboe return ret; 1471aadd06e5SJens Axboe } 1472aadd06e5SJens Axboe 1473aadd06e5SJens Axboe /* 1474aadd06e5SJens Axboe * Make sure there's writeable room. Wait for room if we can, otherwise 1475aadd06e5SJens Axboe * return an appropriate error. 1476aadd06e5SJens Axboe */ 14777c77f0b3SMiklos Szeredi static int opipe_prep(struct pipe_inode_info *pipe, unsigned int flags) 1478aadd06e5SJens Axboe { 1479aadd06e5SJens Axboe int ret; 1480aadd06e5SJens Axboe 1481aadd06e5SJens Axboe /* 1482aadd06e5SJens Axboe * Check ->nrbufs without the inode lock first. This function 1483aadd06e5SJens Axboe * is speculative anyways, so missing one is ok. 1484aadd06e5SJens Axboe */ 148535f3d14dSJens Axboe if (pipe->nrbufs < pipe->buffers) 1486aadd06e5SJens Axboe return 0; 1487aadd06e5SJens Axboe 1488aadd06e5SJens Axboe ret = 0; 148961e0d47cSMiklos Szeredi pipe_lock(pipe); 1490aadd06e5SJens Axboe 149135f3d14dSJens Axboe while (pipe->nrbufs >= pipe->buffers) { 1492aadd06e5SJens Axboe if (!pipe->readers) { 1493aadd06e5SJens Axboe send_sig(SIGPIPE, current, 0); 1494aadd06e5SJens Axboe ret = -EPIPE; 1495aadd06e5SJens Axboe break; 1496aadd06e5SJens Axboe } 1497aadd06e5SJens Axboe if (flags & SPLICE_F_NONBLOCK) { 1498aadd06e5SJens Axboe ret = -EAGAIN; 1499aadd06e5SJens Axboe break; 1500aadd06e5SJens Axboe } 1501aadd06e5SJens Axboe if (signal_pending(current)) { 1502aadd06e5SJens Axboe ret = -ERESTARTSYS; 1503aadd06e5SJens Axboe break; 1504aadd06e5SJens Axboe } 1505aadd06e5SJens Axboe pipe->waiting_writers++; 1506aadd06e5SJens Axboe pipe_wait(pipe); 1507aadd06e5SJens Axboe pipe->waiting_writers--; 1508aadd06e5SJens Axboe } 1509aadd06e5SJens Axboe 151061e0d47cSMiklos Szeredi pipe_unlock(pipe); 1511aadd06e5SJens Axboe return ret; 1512aadd06e5SJens Axboe } 1513aadd06e5SJens Axboe 1514aadd06e5SJens Axboe /* 15157c77f0b3SMiklos Szeredi * Splice contents of ipipe to opipe. 15167c77f0b3SMiklos Szeredi */ 15177c77f0b3SMiklos Szeredi static int splice_pipe_to_pipe(struct pipe_inode_info *ipipe, 15187c77f0b3SMiklos Szeredi struct pipe_inode_info *opipe, 15197c77f0b3SMiklos Szeredi size_t len, unsigned int flags) 15207c77f0b3SMiklos Szeredi { 15217c77f0b3SMiklos Szeredi struct pipe_buffer *ibuf, *obuf; 15227c77f0b3SMiklos Szeredi int ret = 0, nbuf; 15237c77f0b3SMiklos Szeredi bool input_wakeup = false; 15247c77f0b3SMiklos Szeredi 15257c77f0b3SMiklos Szeredi 15267c77f0b3SMiklos Szeredi retry: 15277c77f0b3SMiklos Szeredi ret = ipipe_prep(ipipe, flags); 15287c77f0b3SMiklos Szeredi if (ret) 15297c77f0b3SMiklos Szeredi return ret; 15307c77f0b3SMiklos Szeredi 15317c77f0b3SMiklos Szeredi ret = opipe_prep(opipe, flags); 15327c77f0b3SMiklos Szeredi if (ret) 15337c77f0b3SMiklos Szeredi return ret; 15347c77f0b3SMiklos Szeredi 15357c77f0b3SMiklos Szeredi /* 15367c77f0b3SMiklos Szeredi * Potential ABBA deadlock, work around it by ordering lock 15377c77f0b3SMiklos Szeredi * grabbing by pipe info address. Otherwise two different processes 15387c77f0b3SMiklos Szeredi * could deadlock (one doing tee from A -> B, the other from B -> A). 15397c77f0b3SMiklos Szeredi */ 15407c77f0b3SMiklos Szeredi pipe_double_lock(ipipe, opipe); 15417c77f0b3SMiklos Szeredi 15427c77f0b3SMiklos Szeredi do { 15437c77f0b3SMiklos Szeredi if (!opipe->readers) { 15447c77f0b3SMiklos Szeredi send_sig(SIGPIPE, current, 0); 15457c77f0b3SMiklos Szeredi if (!ret) 15467c77f0b3SMiklos Szeredi ret = -EPIPE; 15477c77f0b3SMiklos Szeredi break; 15487c77f0b3SMiklos Szeredi } 15497c77f0b3SMiklos Szeredi 15507c77f0b3SMiklos Szeredi if (!ipipe->nrbufs && !ipipe->writers) 15517c77f0b3SMiklos Szeredi break; 15527c77f0b3SMiklos Szeredi 15537c77f0b3SMiklos Szeredi /* 15547c77f0b3SMiklos Szeredi * Cannot make any progress, because either the input 15557c77f0b3SMiklos Szeredi * pipe is empty or the output pipe is full. 15567c77f0b3SMiklos Szeredi */ 155735f3d14dSJens Axboe if (!ipipe->nrbufs || opipe->nrbufs >= opipe->buffers) { 15587c77f0b3SMiklos Szeredi /* Already processed some buffers, break */ 15597c77f0b3SMiklos Szeredi if (ret) 15607c77f0b3SMiklos Szeredi break; 15617c77f0b3SMiklos Szeredi 15627c77f0b3SMiklos Szeredi if (flags & SPLICE_F_NONBLOCK) { 15637c77f0b3SMiklos Szeredi ret = -EAGAIN; 15647c77f0b3SMiklos Szeredi break; 15657c77f0b3SMiklos Szeredi } 15667c77f0b3SMiklos Szeredi 15677c77f0b3SMiklos Szeredi /* 15687c77f0b3SMiklos Szeredi * We raced with another reader/writer and haven't 15697c77f0b3SMiklos Szeredi * managed to process any buffers. A zero return 15707c77f0b3SMiklos Szeredi * value means EOF, so retry instead. 15717c77f0b3SMiklos Szeredi */ 15727c77f0b3SMiklos Szeredi pipe_unlock(ipipe); 15737c77f0b3SMiklos Szeredi pipe_unlock(opipe); 15747c77f0b3SMiklos Szeredi goto retry; 15757c77f0b3SMiklos Szeredi } 15767c77f0b3SMiklos Szeredi 15777c77f0b3SMiklos Szeredi ibuf = ipipe->bufs + ipipe->curbuf; 157835f3d14dSJens Axboe nbuf = (opipe->curbuf + opipe->nrbufs) & (opipe->buffers - 1); 15797c77f0b3SMiklos Szeredi obuf = opipe->bufs + nbuf; 15807c77f0b3SMiklos Szeredi 15817c77f0b3SMiklos Szeredi if (len >= ibuf->len) { 15827c77f0b3SMiklos Szeredi /* 15837c77f0b3SMiklos Szeredi * Simply move the whole buffer from ipipe to opipe 15847c77f0b3SMiklos Szeredi */ 15857c77f0b3SMiklos Szeredi *obuf = *ibuf; 15867c77f0b3SMiklos Szeredi ibuf->ops = NULL; 15877c77f0b3SMiklos Szeredi opipe->nrbufs++; 158835f3d14dSJens Axboe ipipe->curbuf = (ipipe->curbuf + 1) & (ipipe->buffers - 1); 15897c77f0b3SMiklos Szeredi ipipe->nrbufs--; 15907c77f0b3SMiklos Szeredi input_wakeup = true; 15917c77f0b3SMiklos Szeredi } else { 15927c77f0b3SMiklos Szeredi /* 15937c77f0b3SMiklos Szeredi * Get a reference to this pipe buffer, 15947c77f0b3SMiklos Szeredi * so we can copy the contents over. 15957c77f0b3SMiklos Szeredi */ 15967bf2d1dfSMiklos Szeredi pipe_buf_get(ipipe, ibuf); 15977c77f0b3SMiklos Szeredi *obuf = *ibuf; 15987c77f0b3SMiklos Szeredi 15997c77f0b3SMiklos Szeredi /* 16007c77f0b3SMiklos Szeredi * Don't inherit the gift flag, we need to 16017c77f0b3SMiklos Szeredi * prevent multiple steals of this page. 16027c77f0b3SMiklos Szeredi */ 16037c77f0b3SMiklos Szeredi obuf->flags &= ~PIPE_BUF_FLAG_GIFT; 16047c77f0b3SMiklos Szeredi 1605a0ce2f0aSJann Horn pipe_buf_mark_unmergeable(obuf); 1606a0ce2f0aSJann Horn 16077c77f0b3SMiklos Szeredi obuf->len = len; 16087c77f0b3SMiklos Szeredi opipe->nrbufs++; 16097c77f0b3SMiklos Szeredi ibuf->offset += obuf->len; 16107c77f0b3SMiklos Szeredi ibuf->len -= obuf->len; 16117c77f0b3SMiklos Szeredi } 16127c77f0b3SMiklos Szeredi ret += obuf->len; 16137c77f0b3SMiklos Szeredi len -= obuf->len; 16147c77f0b3SMiklos Szeredi } while (len); 16157c77f0b3SMiklos Szeredi 16167c77f0b3SMiklos Szeredi pipe_unlock(ipipe); 16177c77f0b3SMiklos Szeredi pipe_unlock(opipe); 16187c77f0b3SMiklos Szeredi 16197c77f0b3SMiklos Szeredi /* 16207c77f0b3SMiklos Szeredi * If we put data in the output pipe, wakeup any potential readers. 16217c77f0b3SMiklos Szeredi */ 1622825cdcb1SNamhyung Kim if (ret > 0) 1623825cdcb1SNamhyung Kim wakeup_pipe_readers(opipe); 1624825cdcb1SNamhyung Kim 16257c77f0b3SMiklos Szeredi if (input_wakeup) 16267c77f0b3SMiklos Szeredi wakeup_pipe_writers(ipipe); 16277c77f0b3SMiklos Szeredi 16287c77f0b3SMiklos Szeredi return ret; 16297c77f0b3SMiklos Szeredi } 16307c77f0b3SMiklos Szeredi 16317c77f0b3SMiklos Szeredi /* 163270524490SJens Axboe * Link contents of ipipe to opipe. 163370524490SJens Axboe */ 163470524490SJens Axboe static int link_pipe(struct pipe_inode_info *ipipe, 163570524490SJens Axboe struct pipe_inode_info *opipe, 163670524490SJens Axboe size_t len, unsigned int flags) 163770524490SJens Axboe { 163870524490SJens Axboe struct pipe_buffer *ibuf, *obuf; 1639aadd06e5SJens Axboe int ret = 0, i = 0, nbuf; 164070524490SJens Axboe 164170524490SJens Axboe /* 164270524490SJens Axboe * Potential ABBA deadlock, work around it by ordering lock 164361e0d47cSMiklos Szeredi * grabbing by pipe info address. Otherwise two different processes 164470524490SJens Axboe * could deadlock (one doing tee from A -> B, the other from B -> A). 164570524490SJens Axboe */ 164661e0d47cSMiklos Szeredi pipe_double_lock(ipipe, opipe); 164770524490SJens Axboe 1648aadd06e5SJens Axboe do { 164970524490SJens Axboe if (!opipe->readers) { 165070524490SJens Axboe send_sig(SIGPIPE, current, 0); 165170524490SJens Axboe if (!ret) 165270524490SJens Axboe ret = -EPIPE; 165370524490SJens Axboe break; 165470524490SJens Axboe } 165570524490SJens Axboe 165670524490SJens Axboe /* 1657aadd06e5SJens Axboe * If we have iterated all input buffers or ran out of 1658aadd06e5SJens Axboe * output room, break. 165970524490SJens Axboe */ 166035f3d14dSJens Axboe if (i >= ipipe->nrbufs || opipe->nrbufs >= opipe->buffers) 1661aadd06e5SJens Axboe break; 1662aadd06e5SJens Axboe 166335f3d14dSJens Axboe ibuf = ipipe->bufs + ((ipipe->curbuf + i) & (ipipe->buffers-1)); 166435f3d14dSJens Axboe nbuf = (opipe->curbuf + opipe->nrbufs) & (opipe->buffers - 1); 166570524490SJens Axboe 166670524490SJens Axboe /* 166770524490SJens Axboe * Get a reference to this pipe buffer, 166870524490SJens Axboe * so we can copy the contents over. 166970524490SJens Axboe */ 16707bf2d1dfSMiklos Szeredi pipe_buf_get(ipipe, ibuf); 167170524490SJens Axboe 167270524490SJens Axboe obuf = opipe->bufs + nbuf; 167370524490SJens Axboe *obuf = *ibuf; 167470524490SJens Axboe 16757afa6fd0SJens Axboe /* 16767afa6fd0SJens Axboe * Don't inherit the gift flag, we need to 16777afa6fd0SJens Axboe * prevent multiple steals of this page. 16787afa6fd0SJens Axboe */ 16797afa6fd0SJens Axboe obuf->flags &= ~PIPE_BUF_FLAG_GIFT; 16807afa6fd0SJens Axboe 1681a0ce2f0aSJann Horn pipe_buf_mark_unmergeable(obuf); 1682a0ce2f0aSJann Horn 168370524490SJens Axboe if (obuf->len > len) 168470524490SJens Axboe obuf->len = len; 168570524490SJens Axboe 168670524490SJens Axboe opipe->nrbufs++; 168770524490SJens Axboe ret += obuf->len; 168870524490SJens Axboe len -= obuf->len; 1689aadd06e5SJens Axboe i++; 1690aadd06e5SJens Axboe } while (len); 169170524490SJens Axboe 169202cf01aeSJens Axboe /* 169302cf01aeSJens Axboe * return EAGAIN if we have the potential of some data in the 169402cf01aeSJens Axboe * future, otherwise just return 0 169502cf01aeSJens Axboe */ 169602cf01aeSJens Axboe if (!ret && ipipe->waiting_writers && (flags & SPLICE_F_NONBLOCK)) 169702cf01aeSJens Axboe ret = -EAGAIN; 169802cf01aeSJens Axboe 169961e0d47cSMiklos Szeredi pipe_unlock(ipipe); 170061e0d47cSMiklos Szeredi pipe_unlock(opipe); 170170524490SJens Axboe 1702aadd06e5SJens Axboe /* 1703aadd06e5SJens Axboe * If we put data in the output pipe, wakeup any potential readers. 1704aadd06e5SJens Axboe */ 1705825cdcb1SNamhyung Kim if (ret > 0) 1706825cdcb1SNamhyung Kim wakeup_pipe_readers(opipe); 170770524490SJens Axboe 170870524490SJens Axboe return ret; 170970524490SJens Axboe } 171070524490SJens Axboe 171170524490SJens Axboe /* 171270524490SJens Axboe * This is a tee(1) implementation that works on pipes. It doesn't copy 171370524490SJens Axboe * any data, it simply references the 'in' pages on the 'out' pipe. 171470524490SJens Axboe * The 'flags' used are the SPLICE_F_* variants, currently the only 171570524490SJens Axboe * applicable one is SPLICE_F_NONBLOCK. 171670524490SJens Axboe */ 171770524490SJens Axboe static long do_tee(struct file *in, struct file *out, size_t len, 171870524490SJens Axboe unsigned int flags) 171970524490SJens Axboe { 172071993e62SLinus Torvalds struct pipe_inode_info *ipipe = get_pipe_info(in); 172171993e62SLinus Torvalds struct pipe_inode_info *opipe = get_pipe_info(out); 1722aadd06e5SJens Axboe int ret = -EINVAL; 172370524490SJens Axboe 172470524490SJens Axboe /* 1725aadd06e5SJens Axboe * Duplicate the contents of ipipe to opipe without actually 1726aadd06e5SJens Axboe * copying the data. 172770524490SJens Axboe */ 1728aadd06e5SJens Axboe if (ipipe && opipe && ipipe != opipe) { 1729ee5e0011SSlavomir Kaslev if ((in->f_flags | out->f_flags) & O_NONBLOCK) 1730ee5e0011SSlavomir Kaslev flags |= SPLICE_F_NONBLOCK; 1731ee5e0011SSlavomir Kaslev 1732aadd06e5SJens Axboe /* 1733aadd06e5SJens Axboe * Keep going, unless we encounter an error. The ipipe/opipe 1734aadd06e5SJens Axboe * ordering doesn't really matter. 1735aadd06e5SJens Axboe */ 17367c77f0b3SMiklos Szeredi ret = ipipe_prep(ipipe, flags); 1737aadd06e5SJens Axboe if (!ret) { 17387c77f0b3SMiklos Szeredi ret = opipe_prep(opipe, flags); 173902cf01aeSJens Axboe if (!ret) 1740aadd06e5SJens Axboe ret = link_pipe(ipipe, opipe, len, flags); 1741aadd06e5SJens Axboe } 1742aadd06e5SJens Axboe } 174370524490SJens Axboe 1744aadd06e5SJens Axboe return ret; 174570524490SJens Axboe } 174670524490SJens Axboe 1747836f92adSHeiko Carstens SYSCALL_DEFINE4(tee, int, fdin, int, fdout, size_t, len, unsigned int, flags) 174870524490SJens Axboe { 17492903ff01SAl Viro struct fd in; 17502903ff01SAl Viro int error; 175170524490SJens Axboe 17523d6ea290SAl Viro if (unlikely(flags & ~SPLICE_F_ALL)) 17533d6ea290SAl Viro return -EINVAL; 17543d6ea290SAl Viro 175570524490SJens Axboe if (unlikely(!len)) 175670524490SJens Axboe return 0; 175770524490SJens Axboe 175870524490SJens Axboe error = -EBADF; 17592903ff01SAl Viro in = fdget(fdin); 17602903ff01SAl Viro if (in.file) { 17612903ff01SAl Viro if (in.file->f_mode & FMODE_READ) { 17622903ff01SAl Viro struct fd out = fdget(fdout); 17632903ff01SAl Viro if (out.file) { 17642903ff01SAl Viro if (out.file->f_mode & FMODE_WRITE) 17652903ff01SAl Viro error = do_tee(in.file, out.file, 17662903ff01SAl Viro len, flags); 17672903ff01SAl Viro fdput(out); 176870524490SJens Axboe } 176970524490SJens Axboe } 17702903ff01SAl Viro fdput(in); 177170524490SJens Axboe } 177270524490SJens Axboe 177370524490SJens Axboe return error; 177470524490SJens Axboe } 1775