splice.c 35 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555
  1. /*
  2. * "splice": joining two ropes together by interweaving their strands.
  3. *
  4. * This is the "extended pipe" functionality, where a pipe is used as
  5. * an arbitrary in-memory buffer. Think of a pipe as a small kernel
  6. * buffer that you can use to transfer data from one end to the other.
  7. *
  8. * The traditional unix read/write is extended with a "splice()" operation
  9. * that transfers data buffers to or from a pipe buffer.
  10. *
  11. * Named by Larry McVoy, original implementation from Linus, extended by
  12. * Jens to support splicing to files, network, direct splicing, etc and
  13. * fixing lots of bugs.
  14. *
  15. * Copyright (C) 2005-2006 Jens Axboe <axboe@kernel.dk>
  16. * Copyright (C) 2005-2006 Linus Torvalds <torvalds@osdl.org>
  17. * Copyright (C) 2006 Ingo Molnar <mingo@elte.hu>
  18. *
  19. */
  20. #include <linux/fs.h>
  21. #include <linux/file.h>
  22. #include <linux/pagemap.h>
  23. #include <linux/pipe_fs_i.h>
  24. #include <linux/mm_inline.h>
  25. #include <linux/swap.h>
  26. #include <linux/writeback.h>
  27. #include <linux/buffer_head.h>
  28. #include <linux/module.h>
  29. #include <linux/syscalls.h>
  30. #include <linux/uio.h>
  31. struct partial_page {
  32. unsigned int offset;
  33. unsigned int len;
  34. };
  35. /*
  36. * Passed to splice_to_pipe
  37. */
  38. struct splice_pipe_desc {
  39. struct page **pages; /* page map */
  40. struct partial_page *partial; /* pages[] may not be contig */
  41. int nr_pages; /* number of pages in map */
  42. unsigned int flags; /* splice flags */
  43. const struct pipe_buf_operations *ops;/* ops associated with output pipe */
  44. };
  45. /*
  46. * Attempt to steal a page from a pipe buffer. This should perhaps go into
  47. * a vm helper function, it's already simplified quite a bit by the
  48. * addition of remove_mapping(). If success is returned, the caller may
  49. * attempt to reuse this page for another destination.
  50. */
  51. static int page_cache_pipe_buf_steal(struct pipe_inode_info *pipe,
  52. struct pipe_buffer *buf)
  53. {
  54. struct page *page = buf->page;
  55. struct address_space *mapping;
  56. lock_page(page);
  57. mapping = page_mapping(page);
  58. if (mapping) {
  59. WARN_ON(!PageUptodate(page));
  60. /*
  61. * At least for ext2 with nobh option, we need to wait on
  62. * writeback completing on this page, since we'll remove it
  63. * from the pagecache. Otherwise truncate wont wait on the
  64. * page, allowing the disk blocks to be reused by someone else
  65. * before we actually wrote our data to them. fs corruption
  66. * ensues.
  67. */
  68. wait_on_page_writeback(page);
  69. if (PagePrivate(page))
  70. try_to_release_page(page, GFP_KERNEL);
  71. /*
  72. * If we succeeded in removing the mapping, set LRU flag
  73. * and return good.
  74. */
  75. if (remove_mapping(mapping, page)) {
  76. buf->flags |= PIPE_BUF_FLAG_LRU;
  77. return 0;
  78. }
  79. }
  80. /*
  81. * Raced with truncate or failed to remove page from current
  82. * address space, unlock and return failure.
  83. */
  84. unlock_page(page);
  85. return 1;
  86. }
  87. static void page_cache_pipe_buf_release(struct pipe_inode_info *pipe,
  88. struct pipe_buffer *buf)
  89. {
  90. page_cache_release(buf->page);
  91. buf->flags &= ~PIPE_BUF_FLAG_LRU;
  92. }
  93. static int page_cache_pipe_buf_pin(struct pipe_inode_info *pipe,
  94. struct pipe_buffer *buf)
  95. {
  96. struct page *page = buf->page;
  97. int err;
  98. if (!PageUptodate(page)) {
  99. lock_page(page);
  100. /*
  101. * Page got truncated/unhashed. This will cause a 0-byte
  102. * splice, if this is the first page.
  103. */
  104. if (!page->mapping) {
  105. err = -ENODATA;
  106. goto error;
  107. }
  108. /*
  109. * Uh oh, read-error from disk.
  110. */
  111. if (!PageUptodate(page)) {
  112. err = -EIO;
  113. goto error;
  114. }
  115. /*
  116. * Page is ok afterall, we are done.
  117. */
  118. unlock_page(page);
  119. }
  120. return 0;
  121. error:
  122. unlock_page(page);
  123. return err;
  124. }
  125. static const struct pipe_buf_operations page_cache_pipe_buf_ops = {
  126. .can_merge = 0,
  127. .map = generic_pipe_buf_map,
  128. .unmap = generic_pipe_buf_unmap,
  129. .pin = page_cache_pipe_buf_pin,
  130. .release = page_cache_pipe_buf_release,
  131. .steal = page_cache_pipe_buf_steal,
  132. .get = generic_pipe_buf_get,
  133. };
  134. static int user_page_pipe_buf_steal(struct pipe_inode_info *pipe,
  135. struct pipe_buffer *buf)
  136. {
  137. if (!(buf->flags & PIPE_BUF_FLAG_GIFT))
  138. return 1;
  139. buf->flags |= PIPE_BUF_FLAG_LRU;
  140. return generic_pipe_buf_steal(pipe, buf);
  141. }
  142. static const struct pipe_buf_operations user_page_pipe_buf_ops = {
  143. .can_merge = 0,
  144. .map = generic_pipe_buf_map,
  145. .unmap = generic_pipe_buf_unmap,
  146. .pin = generic_pipe_buf_pin,
  147. .release = page_cache_pipe_buf_release,
  148. .steal = user_page_pipe_buf_steal,
  149. .get = generic_pipe_buf_get,
  150. };
  151. /*
  152. * Pipe output worker. This sets up our pipe format with the page cache
  153. * pipe buffer operations. Otherwise very similar to the regular pipe_writev().
  154. */
  155. static ssize_t splice_to_pipe(struct pipe_inode_info *pipe,
  156. struct splice_pipe_desc *spd)
  157. {
  158. int ret, do_wakeup, page_nr;
  159. ret = 0;
  160. do_wakeup = 0;
  161. page_nr = 0;
  162. if (pipe->inode)
  163. mutex_lock(&pipe->inode->i_mutex);
  164. for (;;) {
  165. if (!pipe->readers) {
  166. send_sig(SIGPIPE, current, 0);
  167. if (!ret)
  168. ret = -EPIPE;
  169. break;
  170. }
  171. if (pipe->nrbufs < PIPE_BUFFERS) {
  172. int newbuf = (pipe->curbuf + pipe->nrbufs) & (PIPE_BUFFERS - 1);
  173. struct pipe_buffer *buf = pipe->bufs + newbuf;
  174. buf->page = spd->pages[page_nr];
  175. buf->offset = spd->partial[page_nr].offset;
  176. buf->len = spd->partial[page_nr].len;
  177. buf->ops = spd->ops;
  178. if (spd->flags & SPLICE_F_GIFT)
  179. buf->flags |= PIPE_BUF_FLAG_GIFT;
  180. pipe->nrbufs++;
  181. page_nr++;
  182. ret += buf->len;
  183. if (pipe->inode)
  184. do_wakeup = 1;
  185. if (!--spd->nr_pages)
  186. break;
  187. if (pipe->nrbufs < PIPE_BUFFERS)
  188. continue;
  189. break;
  190. }
  191. if (spd->flags & SPLICE_F_NONBLOCK) {
  192. if (!ret)
  193. ret = -EAGAIN;
  194. break;
  195. }
  196. if (signal_pending(current)) {
  197. if (!ret)
  198. ret = -ERESTARTSYS;
  199. break;
  200. }
  201. if (do_wakeup) {
  202. smp_mb();
  203. if (waitqueue_active(&pipe->wait))
  204. wake_up_interruptible_sync(&pipe->wait);
  205. kill_fasync(&pipe->fasync_readers, SIGIO, POLL_IN);
  206. do_wakeup = 0;
  207. }
  208. pipe->waiting_writers++;
  209. pipe_wait(pipe);
  210. pipe->waiting_writers--;
  211. }
  212. if (pipe->inode)
  213. mutex_unlock(&pipe->inode->i_mutex);
  214. if (do_wakeup) {
  215. smp_mb();
  216. if (waitqueue_active(&pipe->wait))
  217. wake_up_interruptible(&pipe->wait);
  218. kill_fasync(&pipe->fasync_readers, SIGIO, POLL_IN);
  219. }
  220. while (page_nr < spd->nr_pages)
  221. page_cache_release(spd->pages[page_nr++]);
  222. return ret;
  223. }
  224. static int
  225. __generic_file_splice_read(struct file *in, loff_t *ppos,
  226. struct pipe_inode_info *pipe, size_t len,
  227. unsigned int flags)
  228. {
  229. struct address_space *mapping = in->f_mapping;
  230. unsigned int loff, nr_pages;
  231. struct page *pages[PIPE_BUFFERS];
  232. struct partial_page partial[PIPE_BUFFERS];
  233. struct page *page;
  234. pgoff_t index, end_index;
  235. loff_t isize;
  236. size_t total_len;
  237. int error, page_nr;
  238. struct splice_pipe_desc spd = {
  239. .pages = pages,
  240. .partial = partial,
  241. .flags = flags,
  242. .ops = &page_cache_pipe_buf_ops,
  243. };
  244. index = *ppos >> PAGE_CACHE_SHIFT;
  245. loff = *ppos & ~PAGE_CACHE_MASK;
  246. nr_pages = (len + loff + PAGE_CACHE_SIZE - 1) >> PAGE_CACHE_SHIFT;
  247. if (nr_pages > PIPE_BUFFERS)
  248. nr_pages = PIPE_BUFFERS;
  249. /*
  250. * Don't try to 2nd guess the read-ahead logic, call into
  251. * page_cache_readahead() like the page cache reads would do.
  252. */
  253. page_cache_readahead(mapping, &in->f_ra, in, index, nr_pages);
  254. /*
  255. * Now fill in the holes:
  256. */
  257. error = 0;
  258. total_len = 0;
  259. /*
  260. * Lookup the (hopefully) full range of pages we need.
  261. */
  262. spd.nr_pages = find_get_pages_contig(mapping, index, nr_pages, pages);
  263. /*
  264. * If find_get_pages_contig() returned fewer pages than we needed,
  265. * allocate the rest.
  266. */
  267. index += spd.nr_pages;
  268. while (spd.nr_pages < nr_pages) {
  269. /*
  270. * Page could be there, find_get_pages_contig() breaks on
  271. * the first hole.
  272. */
  273. page = find_get_page(mapping, index);
  274. if (!page) {
  275. /*
  276. * Make sure the read-ahead engine is notified
  277. * about this failure.
  278. */
  279. handle_ra_miss(mapping, &in->f_ra, index);
  280. /*
  281. * page didn't exist, allocate one.
  282. */
  283. page = page_cache_alloc_cold(mapping);
  284. if (!page)
  285. break;
  286. error = add_to_page_cache_lru(page, mapping, index,
  287. GFP_KERNEL);
  288. if (unlikely(error)) {
  289. page_cache_release(page);
  290. if (error == -EEXIST)
  291. continue;
  292. break;
  293. }
  294. /*
  295. * add_to_page_cache() locks the page, unlock it
  296. * to avoid convoluting the logic below even more.
  297. */
  298. unlock_page(page);
  299. }
  300. pages[spd.nr_pages++] = page;
  301. index++;
  302. }
  303. /*
  304. * Now loop over the map and see if we need to start IO on any
  305. * pages, fill in the partial map, etc.
  306. */
  307. index = *ppos >> PAGE_CACHE_SHIFT;
  308. nr_pages = spd.nr_pages;
  309. spd.nr_pages = 0;
  310. for (page_nr = 0; page_nr < nr_pages; page_nr++) {
  311. unsigned int this_len;
  312. if (!len)
  313. break;
  314. /*
  315. * this_len is the max we'll use from this page
  316. */
  317. this_len = min_t(unsigned long, len, PAGE_CACHE_SIZE - loff);
  318. page = pages[page_nr];
  319. /*
  320. * If the page isn't uptodate, we may need to start io on it
  321. */
  322. if (!PageUptodate(page)) {
  323. /*
  324. * If in nonblock mode then dont block on waiting
  325. * for an in-flight io page
  326. */
  327. if (flags & SPLICE_F_NONBLOCK) {
  328. if (TestSetPageLocked(page))
  329. break;
  330. } else
  331. lock_page(page);
  332. /*
  333. * page was truncated, stop here. if this isn't the
  334. * first page, we'll just complete what we already
  335. * added
  336. */
  337. if (!page->mapping) {
  338. unlock_page(page);
  339. break;
  340. }
  341. /*
  342. * page was already under io and is now done, great
  343. */
  344. if (PageUptodate(page)) {
  345. unlock_page(page);
  346. goto fill_it;
  347. }
  348. /*
  349. * need to read in the page
  350. */
  351. error = mapping->a_ops->readpage(in, page);
  352. if (unlikely(error)) {
  353. /*
  354. * We really should re-lookup the page here,
  355. * but it complicates things a lot. Instead
  356. * lets just do what we already stored, and
  357. * we'll get it the next time we are called.
  358. */
  359. if (error == AOP_TRUNCATED_PAGE)
  360. error = 0;
  361. break;
  362. }
  363. /*
  364. * i_size must be checked after ->readpage().
  365. */
  366. isize = i_size_read(mapping->host);
  367. end_index = (isize - 1) >> PAGE_CACHE_SHIFT;
  368. if (unlikely(!isize || index > end_index))
  369. break;
  370. /*
  371. * if this is the last page, see if we need to shrink
  372. * the length and stop
  373. */
  374. if (end_index == index) {
  375. loff = PAGE_CACHE_SIZE - (isize & ~PAGE_CACHE_MASK);
  376. if (total_len + loff > isize)
  377. break;
  378. /*
  379. * force quit after adding this page
  380. */
  381. len = this_len;
  382. this_len = min(this_len, loff);
  383. loff = 0;
  384. }
  385. }
  386. fill_it:
  387. partial[page_nr].offset = loff;
  388. partial[page_nr].len = this_len;
  389. len -= this_len;
  390. total_len += this_len;
  391. loff = 0;
  392. spd.nr_pages++;
  393. index++;
  394. }
  395. /*
  396. * Release any pages at the end, if we quit early. 'i' is how far
  397. * we got, 'nr_pages' is how many pages are in the map.
  398. */
  399. while (page_nr < nr_pages)
  400. page_cache_release(pages[page_nr++]);
  401. if (spd.nr_pages)
  402. return splice_to_pipe(pipe, &spd);
  403. return error;
  404. }
  405. /**
  406. * generic_file_splice_read - splice data from file to a pipe
  407. * @in: file to splice from
  408. * @pipe: pipe to splice to
  409. * @len: number of bytes to splice
  410. * @flags: splice modifier flags
  411. *
  412. * Will read pages from given file and fill them into a pipe.
  413. */
  414. ssize_t generic_file_splice_read(struct file *in, loff_t *ppos,
  415. struct pipe_inode_info *pipe, size_t len,
  416. unsigned int flags)
  417. {
  418. ssize_t spliced;
  419. int ret;
  420. loff_t isize, left;
  421. isize = i_size_read(in->f_mapping->host);
  422. if (unlikely(*ppos >= isize))
  423. return 0;
  424. left = isize - *ppos;
  425. if (unlikely(left < len))
  426. len = left;
  427. ret = 0;
  428. spliced = 0;
  429. while (len) {
  430. ret = __generic_file_splice_read(in, ppos, pipe, len, flags);
  431. if (ret < 0)
  432. break;
  433. else if (!ret) {
  434. if (spliced)
  435. break;
  436. if (flags & SPLICE_F_NONBLOCK) {
  437. ret = -EAGAIN;
  438. break;
  439. }
  440. }
  441. *ppos += ret;
  442. len -= ret;
  443. spliced += ret;
  444. }
  445. if (spliced)
  446. return spliced;
  447. return ret;
  448. }
  449. EXPORT_SYMBOL(generic_file_splice_read);
  450. /*
  451. * Send 'sd->len' bytes to socket from 'sd->file' at position 'sd->pos'
  452. * using sendpage(). Return the number of bytes sent.
  453. */
  454. static int pipe_to_sendpage(struct pipe_inode_info *pipe,
  455. struct pipe_buffer *buf, struct splice_desc *sd)
  456. {
  457. struct file *file = sd->file;
  458. loff_t pos = sd->pos;
  459. int ret, more;
  460. ret = buf->ops->pin(pipe, buf);
  461. if (!ret) {
  462. more = (sd->flags & SPLICE_F_MORE) || sd->len < sd->total_len;
  463. ret = file->f_op->sendpage(file, buf->page, buf->offset,
  464. sd->len, &pos, more);
  465. }
  466. return ret;
  467. }
  468. /*
  469. * This is a little more tricky than the file -> pipe splicing. There are
  470. * basically three cases:
  471. *
  472. * - Destination page already exists in the address space and there
  473. * are users of it. For that case we have no other option that
  474. * copying the data. Tough luck.
  475. * - Destination page already exists in the address space, but there
  476. * are no users of it. Make sure it's uptodate, then drop it. Fall
  477. * through to last case.
  478. * - Destination page does not exist, we can add the pipe page to
  479. * the page cache and avoid the copy.
  480. *
  481. * If asked to move pages to the output file (SPLICE_F_MOVE is set in
  482. * sd->flags), we attempt to migrate pages from the pipe to the output
  483. * file address space page cache. This is possible if no one else has
  484. * the pipe page referenced outside of the pipe and page cache. If
  485. * SPLICE_F_MOVE isn't set, or we cannot move the page, we simply create
  486. * a new page in the output file page cache and fill/dirty that.
  487. */
  488. static int pipe_to_file(struct pipe_inode_info *pipe, struct pipe_buffer *buf,
  489. struct splice_desc *sd)
  490. {
  491. struct file *file = sd->file;
  492. struct address_space *mapping = file->f_mapping;
  493. unsigned int offset, this_len;
  494. struct page *page;
  495. pgoff_t index;
  496. int ret;
  497. /*
  498. * make sure the data in this buffer is uptodate
  499. */
  500. ret = buf->ops->pin(pipe, buf);
  501. if (unlikely(ret))
  502. return ret;
  503. index = sd->pos >> PAGE_CACHE_SHIFT;
  504. offset = sd->pos & ~PAGE_CACHE_MASK;
  505. this_len = sd->len;
  506. if (this_len + offset > PAGE_CACHE_SIZE)
  507. this_len = PAGE_CACHE_SIZE - offset;
  508. find_page:
  509. page = find_lock_page(mapping, index);
  510. if (!page) {
  511. ret = -ENOMEM;
  512. page = page_cache_alloc_cold(mapping);
  513. if (unlikely(!page))
  514. goto out_ret;
  515. /*
  516. * This will also lock the page
  517. */
  518. ret = add_to_page_cache_lru(page, mapping, index,
  519. GFP_KERNEL);
  520. if (unlikely(ret))
  521. goto out;
  522. }
  523. ret = mapping->a_ops->prepare_write(file, page, offset, offset+this_len);
  524. if (unlikely(ret)) {
  525. loff_t isize = i_size_read(mapping->host);
  526. if (ret != AOP_TRUNCATED_PAGE)
  527. unlock_page(page);
  528. page_cache_release(page);
  529. if (ret == AOP_TRUNCATED_PAGE)
  530. goto find_page;
  531. /*
  532. * prepare_write() may have instantiated a few blocks
  533. * outside i_size. Trim these off again.
  534. */
  535. if (sd->pos + this_len > isize)
  536. vmtruncate(mapping->host, isize);
  537. goto out_ret;
  538. }
  539. if (buf->page != page) {
  540. /*
  541. * Careful, ->map() uses KM_USER0!
  542. */
  543. char *src = buf->ops->map(pipe, buf, 1);
  544. char *dst = kmap_atomic(page, KM_USER1);
  545. memcpy(dst + offset, src + buf->offset, this_len);
  546. flush_dcache_page(page);
  547. kunmap_atomic(dst, KM_USER1);
  548. buf->ops->unmap(pipe, buf, src);
  549. }
  550. ret = mapping->a_ops->commit_write(file, page, offset, offset+this_len);
  551. if (ret) {
  552. if (ret == AOP_TRUNCATED_PAGE) {
  553. page_cache_release(page);
  554. goto find_page;
  555. }
  556. if (ret < 0)
  557. goto out;
  558. /*
  559. * Partial write has happened, so 'ret' already initialized by
  560. * number of bytes written, Where is nothing we have to do here.
  561. */
  562. } else
  563. ret = this_len;
  564. /*
  565. * Return the number of bytes written and mark page as
  566. * accessed, we are now done!
  567. */
  568. mark_page_accessed(page);
  569. balance_dirty_pages_ratelimited(mapping);
  570. out:
  571. page_cache_release(page);
  572. unlock_page(page);
  573. out_ret:
  574. return ret;
  575. }
  576. /*
  577. * Pipe input worker. Most of this logic works like a regular pipe, the
  578. * key here is the 'actor' worker passed in that actually moves the data
  579. * to the wanted destination. See pipe_to_file/pipe_to_sendpage above.
  580. */
  581. ssize_t __splice_from_pipe(struct pipe_inode_info *pipe,
  582. struct file *out, loff_t *ppos, size_t len,
  583. unsigned int flags, splice_actor *actor)
  584. {
  585. int ret, do_wakeup, err;
  586. struct splice_desc sd;
  587. ret = 0;
  588. do_wakeup = 0;
  589. sd.total_len = len;
  590. sd.flags = flags;
  591. sd.file = out;
  592. sd.pos = *ppos;
  593. for (;;) {
  594. if (pipe->nrbufs) {
  595. struct pipe_buffer *buf = pipe->bufs + pipe->curbuf;
  596. const struct pipe_buf_operations *ops = buf->ops;
  597. sd.len = buf->len;
  598. if (sd.len > sd.total_len)
  599. sd.len = sd.total_len;
  600. err = actor(pipe, buf, &sd);
  601. if (err <= 0) {
  602. if (!ret && err != -ENODATA)
  603. ret = err;
  604. break;
  605. }
  606. ret += err;
  607. buf->offset += err;
  608. buf->len -= err;
  609. sd.len -= err;
  610. sd.pos += err;
  611. sd.total_len -= err;
  612. if (sd.len)
  613. continue;
  614. if (!buf->len) {
  615. buf->ops = NULL;
  616. ops->release(pipe, buf);
  617. pipe->curbuf = (pipe->curbuf + 1) & (PIPE_BUFFERS - 1);
  618. pipe->nrbufs--;
  619. if (pipe->inode)
  620. do_wakeup = 1;
  621. }
  622. if (!sd.total_len)
  623. break;
  624. }
  625. if (pipe->nrbufs)
  626. continue;
  627. if (!pipe->writers)
  628. break;
  629. if (!pipe->waiting_writers) {
  630. if (ret)
  631. break;
  632. }
  633. if (flags & SPLICE_F_NONBLOCK) {
  634. if (!ret)
  635. ret = -EAGAIN;
  636. break;
  637. }
  638. if (signal_pending(current)) {
  639. if (!ret)
  640. ret = -ERESTARTSYS;
  641. break;
  642. }
  643. if (do_wakeup) {
  644. smp_mb();
  645. if (waitqueue_active(&pipe->wait))
  646. wake_up_interruptible_sync(&pipe->wait);
  647. kill_fasync(&pipe->fasync_writers, SIGIO, POLL_OUT);
  648. do_wakeup = 0;
  649. }
  650. pipe_wait(pipe);
  651. }
  652. if (do_wakeup) {
  653. smp_mb();
  654. if (waitqueue_active(&pipe->wait))
  655. wake_up_interruptible(&pipe->wait);
  656. kill_fasync(&pipe->fasync_writers, SIGIO, POLL_OUT);
  657. }
  658. return ret;
  659. }
  660. EXPORT_SYMBOL(__splice_from_pipe);
  661. ssize_t splice_from_pipe(struct pipe_inode_info *pipe, struct file *out,
  662. loff_t *ppos, size_t len, unsigned int flags,
  663. splice_actor *actor)
  664. {
  665. ssize_t ret;
  666. struct inode *inode = out->f_mapping->host;
  667. /*
  668. * The actor worker might be calling ->prepare_write and
  669. * ->commit_write. Most of the time, these expect i_mutex to
  670. * be held. Since this may result in an ABBA deadlock with
  671. * pipe->inode, we have to order lock acquiry here.
  672. */
  673. inode_double_lock(inode, pipe->inode);
  674. ret = __splice_from_pipe(pipe, out, ppos, len, flags, actor);
  675. inode_double_unlock(inode, pipe->inode);
  676. return ret;
  677. }
  678. /**
  679. * generic_file_splice_write_nolock - generic_file_splice_write without mutexes
  680. * @pipe: pipe info
  681. * @out: file to write to
  682. * @len: number of bytes to splice
  683. * @flags: splice modifier flags
  684. *
  685. * Will either move or copy pages (determined by @flags options) from
  686. * the given pipe inode to the given file. The caller is responsible
  687. * for acquiring i_mutex on both inodes.
  688. *
  689. */
  690. ssize_t
  691. generic_file_splice_write_nolock(struct pipe_inode_info *pipe, struct file *out,
  692. loff_t *ppos, size_t len, unsigned int flags)
  693. {
  694. struct address_space *mapping = out->f_mapping;
  695. struct inode *inode = mapping->host;
  696. ssize_t ret;
  697. int err;
  698. err = remove_suid(out->f_path.dentry);
  699. if (unlikely(err))
  700. return err;
  701. ret = __splice_from_pipe(pipe, out, ppos, len, flags, pipe_to_file);
  702. if (ret > 0) {
  703. *ppos += ret;
  704. /*
  705. * If file or inode is SYNC and we actually wrote some data,
  706. * sync it.
  707. */
  708. if (unlikely((out->f_flags & O_SYNC) || IS_SYNC(inode))) {
  709. err = generic_osync_inode(inode, mapping,
  710. OSYNC_METADATA|OSYNC_DATA);
  711. if (err)
  712. ret = err;
  713. }
  714. }
  715. return ret;
  716. }
  717. EXPORT_SYMBOL(generic_file_splice_write_nolock);
  718. /**
  719. * generic_file_splice_write - splice data from a pipe to a file
  720. * @pipe: pipe info
  721. * @out: file to write to
  722. * @len: number of bytes to splice
  723. * @flags: splice modifier flags
  724. *
  725. * Will either move or copy pages (determined by @flags options) from
  726. * the given pipe inode to the given file.
  727. *
  728. */
  729. ssize_t
  730. generic_file_splice_write(struct pipe_inode_info *pipe, struct file *out,
  731. loff_t *ppos, size_t len, unsigned int flags)
  732. {
  733. struct address_space *mapping = out->f_mapping;
  734. struct inode *inode = mapping->host;
  735. ssize_t ret;
  736. int err;
  737. err = should_remove_suid(out->f_path.dentry);
  738. if (unlikely(err)) {
  739. mutex_lock(&inode->i_mutex);
  740. err = __remove_suid(out->f_path.dentry, err);
  741. mutex_unlock(&inode->i_mutex);
  742. if (err)
  743. return err;
  744. }
  745. ret = splice_from_pipe(pipe, out, ppos, len, flags, pipe_to_file);
  746. if (ret > 0) {
  747. *ppos += ret;
  748. /*
  749. * If file or inode is SYNC and we actually wrote some data,
  750. * sync it.
  751. */
  752. if (unlikely((out->f_flags & O_SYNC) || IS_SYNC(inode))) {
  753. mutex_lock(&inode->i_mutex);
  754. err = generic_osync_inode(inode, mapping,
  755. OSYNC_METADATA|OSYNC_DATA);
  756. mutex_unlock(&inode->i_mutex);
  757. if (err)
  758. ret = err;
  759. }
  760. }
  761. return ret;
  762. }
  763. EXPORT_SYMBOL(generic_file_splice_write);
  764. /**
  765. * generic_splice_sendpage - splice data from a pipe to a socket
  766. * @inode: pipe inode
  767. * @out: socket to write to
  768. * @len: number of bytes to splice
  769. * @flags: splice modifier flags
  770. *
  771. * Will send @len bytes from the pipe to a network socket. No data copying
  772. * is involved.
  773. *
  774. */
  775. ssize_t generic_splice_sendpage(struct pipe_inode_info *pipe, struct file *out,
  776. loff_t *ppos, size_t len, unsigned int flags)
  777. {
  778. return splice_from_pipe(pipe, out, ppos, len, flags, pipe_to_sendpage);
  779. }
  780. EXPORT_SYMBOL(generic_splice_sendpage);
  781. /*
  782. * Attempt to initiate a splice from pipe to file.
  783. */
  784. static long do_splice_from(struct pipe_inode_info *pipe, struct file *out,
  785. loff_t *ppos, size_t len, unsigned int flags)
  786. {
  787. int ret;
  788. if (unlikely(!out->f_op || !out->f_op->splice_write))
  789. return -EINVAL;
  790. if (unlikely(!(out->f_mode & FMODE_WRITE)))
  791. return -EBADF;
  792. ret = rw_verify_area(WRITE, out, ppos, len);
  793. if (unlikely(ret < 0))
  794. return ret;
  795. return out->f_op->splice_write(pipe, out, ppos, len, flags);
  796. }
  797. /*
  798. * Attempt to initiate a splice from a file to a pipe.
  799. */
  800. static long do_splice_to(struct file *in, loff_t *ppos,
  801. struct pipe_inode_info *pipe, size_t len,
  802. unsigned int flags)
  803. {
  804. int ret;
  805. if (unlikely(!in->f_op || !in->f_op->splice_read))
  806. return -EINVAL;
  807. if (unlikely(!(in->f_mode & FMODE_READ)))
  808. return -EBADF;
  809. ret = rw_verify_area(READ, in, ppos, len);
  810. if (unlikely(ret < 0))
  811. return ret;
  812. return in->f_op->splice_read(in, ppos, pipe, len, flags);
  813. }
  814. long do_splice_direct(struct file *in, loff_t *ppos, struct file *out,
  815. size_t len, unsigned int flags)
  816. {
  817. struct pipe_inode_info *pipe;
  818. long ret, bytes;
  819. loff_t out_off;
  820. umode_t i_mode;
  821. int i;
  822. /*
  823. * We require the input being a regular file, as we don't want to
  824. * randomly drop data for eg socket -> socket splicing. Use the
  825. * piped splicing for that!
  826. */
  827. i_mode = in->f_path.dentry->d_inode->i_mode;
  828. if (unlikely(!S_ISREG(i_mode) && !S_ISBLK(i_mode)))
  829. return -EINVAL;
  830. /*
  831. * neither in nor out is a pipe, setup an internal pipe attached to
  832. * 'out' and transfer the wanted data from 'in' to 'out' through that
  833. */
  834. pipe = current->splice_pipe;
  835. if (unlikely(!pipe)) {
  836. pipe = alloc_pipe_info(NULL);
  837. if (!pipe)
  838. return -ENOMEM;
  839. /*
  840. * We don't have an immediate reader, but we'll read the stuff
  841. * out of the pipe right after the splice_to_pipe(). So set
  842. * PIPE_READERS appropriately.
  843. */
  844. pipe->readers = 1;
  845. current->splice_pipe = pipe;
  846. }
  847. /*
  848. * Do the splice.
  849. */
  850. ret = 0;
  851. bytes = 0;
  852. out_off = 0;
  853. while (len) {
  854. size_t read_len, max_read_len;
  855. /*
  856. * Do at most PIPE_BUFFERS pages worth of transfer:
  857. */
  858. max_read_len = min(len, (size_t)(PIPE_BUFFERS*PAGE_SIZE));
  859. ret = do_splice_to(in, ppos, pipe, max_read_len, flags);
  860. if (unlikely(ret < 0))
  861. goto out_release;
  862. read_len = ret;
  863. /*
  864. * NOTE: nonblocking mode only applies to the input. We
  865. * must not do the output in nonblocking mode as then we
  866. * could get stuck data in the internal pipe:
  867. */
  868. ret = do_splice_from(pipe, out, &out_off, read_len,
  869. flags & ~SPLICE_F_NONBLOCK);
  870. if (unlikely(ret < 0))
  871. goto out_release;
  872. bytes += ret;
  873. len -= ret;
  874. /*
  875. * In nonblocking mode, if we got back a short read then
  876. * that was due to either an IO error or due to the
  877. * pagecache entry not being there. In the IO error case
  878. * the _next_ splice attempt will produce a clean IO error
  879. * return value (not a short read), so in both cases it's
  880. * correct to break out of the loop here:
  881. */
  882. if ((flags & SPLICE_F_NONBLOCK) && (read_len < max_read_len))
  883. break;
  884. }
  885. pipe->nrbufs = pipe->curbuf = 0;
  886. return bytes;
  887. out_release:
  888. /*
  889. * If we did an incomplete transfer we must release
  890. * the pipe buffers in question:
  891. */
  892. for (i = 0; i < PIPE_BUFFERS; i++) {
  893. struct pipe_buffer *buf = pipe->bufs + i;
  894. if (buf->ops) {
  895. buf->ops->release(pipe, buf);
  896. buf->ops = NULL;
  897. }
  898. }
  899. pipe->nrbufs = pipe->curbuf = 0;
  900. /*
  901. * If we transferred some data, return the number of bytes:
  902. */
  903. if (bytes > 0)
  904. return bytes;
  905. return ret;
  906. }
  907. EXPORT_SYMBOL(do_splice_direct);
  908. /*
  909. * After the inode slimming patch, i_pipe/i_bdev/i_cdev share the same
  910. * location, so checking ->i_pipe is not enough to verify that this is a
  911. * pipe.
  912. */
  913. static inline struct pipe_inode_info *pipe_info(struct inode *inode)
  914. {
  915. if (S_ISFIFO(inode->i_mode))
  916. return inode->i_pipe;
  917. return NULL;
  918. }
  919. /*
  920. * Determine where to splice to/from.
  921. */
  922. static long do_splice(struct file *in, loff_t __user *off_in,
  923. struct file *out, loff_t __user *off_out,
  924. size_t len, unsigned int flags)
  925. {
  926. struct pipe_inode_info *pipe;
  927. loff_t offset, *off;
  928. long ret;
  929. pipe = pipe_info(in->f_path.dentry->d_inode);
  930. if (pipe) {
  931. if (off_in)
  932. return -ESPIPE;
  933. if (off_out) {
  934. if (out->f_op->llseek == no_llseek)
  935. return -EINVAL;
  936. if (copy_from_user(&offset, off_out, sizeof(loff_t)))
  937. return -EFAULT;
  938. off = &offset;
  939. } else
  940. off = &out->f_pos;
  941. ret = do_splice_from(pipe, out, off, len, flags);
  942. if (off_out && copy_to_user(off_out, off, sizeof(loff_t)))
  943. ret = -EFAULT;
  944. return ret;
  945. }
  946. pipe = pipe_info(out->f_path.dentry->d_inode);
  947. if (pipe) {
  948. if (off_out)
  949. return -ESPIPE;
  950. if (off_in) {
  951. if (in->f_op->llseek == no_llseek)
  952. return -EINVAL;
  953. if (copy_from_user(&offset, off_in, sizeof(loff_t)))
  954. return -EFAULT;
  955. off = &offset;
  956. } else
  957. off = &in->f_pos;
  958. ret = do_splice_to(in, off, pipe, len, flags);
  959. if (off_in && copy_to_user(off_in, off, sizeof(loff_t)))
  960. ret = -EFAULT;
  961. return ret;
  962. }
  963. return -EINVAL;
  964. }
  965. /*
  966. * Map an iov into an array of pages and offset/length tupples. With the
  967. * partial_page structure, we can map several non-contiguous ranges into
  968. * our ones pages[] map instead of splitting that operation into pieces.
  969. * Could easily be exported as a generic helper for other users, in which
  970. * case one would probably want to add a 'max_nr_pages' parameter as well.
  971. */
  972. static int get_iovec_page_array(const struct iovec __user *iov,
  973. unsigned int nr_vecs, struct page **pages,
  974. struct partial_page *partial, int aligned)
  975. {
  976. int buffers = 0, error = 0;
  977. /*
  978. * It's ok to take the mmap_sem for reading, even
  979. * across a "get_user()".
  980. */
  981. down_read(&current->mm->mmap_sem);
  982. while (nr_vecs) {
  983. unsigned long off, npages;
  984. void __user *base;
  985. size_t len;
  986. int i;
  987. /*
  988. * Get user address base and length for this iovec.
  989. */
  990. error = get_user(base, &iov->iov_base);
  991. if (unlikely(error))
  992. break;
  993. error = get_user(len, &iov->iov_len);
  994. if (unlikely(error))
  995. break;
  996. /*
  997. * Sanity check this iovec. 0 read succeeds.
  998. */
  999. if (unlikely(!len))
  1000. break;
  1001. error = -EFAULT;
  1002. if (unlikely(!base))
  1003. break;
  1004. /*
  1005. * Get this base offset and number of pages, then map
  1006. * in the user pages.
  1007. */
  1008. off = (unsigned long) base & ~PAGE_MASK;
  1009. /*
  1010. * If asked for alignment, the offset must be zero and the
  1011. * length a multiple of the PAGE_SIZE.
  1012. */
  1013. error = -EINVAL;
  1014. if (aligned && (off || len & ~PAGE_MASK))
  1015. break;
  1016. npages = (off + len + PAGE_SIZE - 1) >> PAGE_SHIFT;
  1017. if (npages > PIPE_BUFFERS - buffers)
  1018. npages = PIPE_BUFFERS - buffers;
  1019. error = get_user_pages(current, current->mm,
  1020. (unsigned long) base, npages, 0, 0,
  1021. &pages[buffers], NULL);
  1022. if (unlikely(error <= 0))
  1023. break;
  1024. /*
  1025. * Fill this contiguous range into the partial page map.
  1026. */
  1027. for (i = 0; i < error; i++) {
  1028. const int plen = min_t(size_t, len, PAGE_SIZE - off);
  1029. partial[buffers].offset = off;
  1030. partial[buffers].len = plen;
  1031. off = 0;
  1032. len -= plen;
  1033. buffers++;
  1034. }
  1035. /*
  1036. * We didn't complete this iov, stop here since it probably
  1037. * means we have to move some of this into a pipe to
  1038. * be able to continue.
  1039. */
  1040. if (len)
  1041. break;
  1042. /*
  1043. * Don't continue if we mapped fewer pages than we asked for,
  1044. * or if we mapped the max number of pages that we have
  1045. * room for.
  1046. */
  1047. if (error < npages || buffers == PIPE_BUFFERS)
  1048. break;
  1049. nr_vecs--;
  1050. iov++;
  1051. }
  1052. up_read(&current->mm->mmap_sem);
  1053. if (buffers)
  1054. return buffers;
  1055. return error;
  1056. }
  1057. /*
  1058. * vmsplice splices a user address range into a pipe. It can be thought of
  1059. * as splice-from-memory, where the regular splice is splice-from-file (or
  1060. * to file). In both cases the output is a pipe, naturally.
  1061. *
  1062. * Note that vmsplice only supports splicing _from_ user memory to a pipe,
  1063. * not the other way around. Splicing from user memory is a simple operation
  1064. * that can be supported without any funky alignment restrictions or nasty
  1065. * vm tricks. We simply map in the user memory and fill them into a pipe.
  1066. * The reverse isn't quite as easy, though. There are two possible solutions
  1067. * for that:
  1068. *
  1069. * - memcpy() the data internally, at which point we might as well just
  1070. * do a regular read() on the buffer anyway.
  1071. * - Lots of nasty vm tricks, that are neither fast nor flexible (it
  1072. * has restriction limitations on both ends of the pipe).
  1073. *
  1074. * Alas, it isn't here.
  1075. *
  1076. */
  1077. static long do_vmsplice(struct file *file, const struct iovec __user *iov,
  1078. unsigned long nr_segs, unsigned int flags)
  1079. {
  1080. struct pipe_inode_info *pipe;
  1081. struct page *pages[PIPE_BUFFERS];
  1082. struct partial_page partial[PIPE_BUFFERS];
  1083. struct splice_pipe_desc spd = {
  1084. .pages = pages,
  1085. .partial = partial,
  1086. .flags = flags,
  1087. .ops = &user_page_pipe_buf_ops,
  1088. };
  1089. pipe = pipe_info(file->f_path.dentry->d_inode);
  1090. if (!pipe)
  1091. return -EBADF;
  1092. if (unlikely(nr_segs > UIO_MAXIOV))
  1093. return -EINVAL;
  1094. else if (unlikely(!nr_segs))
  1095. return 0;
  1096. spd.nr_pages = get_iovec_page_array(iov, nr_segs, pages, partial,
  1097. flags & SPLICE_F_GIFT);
  1098. if (spd.nr_pages <= 0)
  1099. return spd.nr_pages;
  1100. return splice_to_pipe(pipe, &spd);
  1101. }
  1102. asmlinkage long sys_vmsplice(int fd, const struct iovec __user *iov,
  1103. unsigned long nr_segs, unsigned int flags)
  1104. {
  1105. struct file *file;
  1106. long error;
  1107. int fput;
  1108. error = -EBADF;
  1109. file = fget_light(fd, &fput);
  1110. if (file) {
  1111. if (file->f_mode & FMODE_WRITE)
  1112. error = do_vmsplice(file, iov, nr_segs, flags);
  1113. fput_light(file, fput);
  1114. }
  1115. return error;
  1116. }
  1117. asmlinkage long sys_splice(int fd_in, loff_t __user *off_in,
  1118. int fd_out, loff_t __user *off_out,
  1119. size_t len, unsigned int flags)
  1120. {
  1121. long error;
  1122. struct file *in, *out;
  1123. int fput_in, fput_out;
  1124. if (unlikely(!len))
  1125. return 0;
  1126. error = -EBADF;
  1127. in = fget_light(fd_in, &fput_in);
  1128. if (in) {
  1129. if (in->f_mode & FMODE_READ) {
  1130. out = fget_light(fd_out, &fput_out);
  1131. if (out) {
  1132. if (out->f_mode & FMODE_WRITE)
  1133. error = do_splice(in, off_in,
  1134. out, off_out,
  1135. len, flags);
  1136. fput_light(out, fput_out);
  1137. }
  1138. }
  1139. fput_light(in, fput_in);
  1140. }
  1141. return error;
  1142. }
  1143. /*
  1144. * Make sure there's data to read. Wait for input if we can, otherwise
  1145. * return an appropriate error.
  1146. */
  1147. static int link_ipipe_prep(struct pipe_inode_info *pipe, unsigned int flags)
  1148. {
  1149. int ret;
  1150. /*
  1151. * Check ->nrbufs without the inode lock first. This function
  1152. * is speculative anyways, so missing one is ok.
  1153. */
  1154. if (pipe->nrbufs)
  1155. return 0;
  1156. ret = 0;
  1157. mutex_lock(&pipe->inode->i_mutex);
  1158. while (!pipe->nrbufs) {
  1159. if (signal_pending(current)) {
  1160. ret = -ERESTARTSYS;
  1161. break;
  1162. }
  1163. if (!pipe->writers)
  1164. break;
  1165. if (!pipe->waiting_writers) {
  1166. if (flags & SPLICE_F_NONBLOCK) {
  1167. ret = -EAGAIN;
  1168. break;
  1169. }
  1170. }
  1171. pipe_wait(pipe);
  1172. }
  1173. mutex_unlock(&pipe->inode->i_mutex);
  1174. return ret;
  1175. }
  1176. /*
  1177. * Make sure there's writeable room. Wait for room if we can, otherwise
  1178. * return an appropriate error.
  1179. */
  1180. static int link_opipe_prep(struct pipe_inode_info *pipe, unsigned int flags)
  1181. {
  1182. int ret;
  1183. /*
  1184. * Check ->nrbufs without the inode lock first. This function
  1185. * is speculative anyways, so missing one is ok.
  1186. */
  1187. if (pipe->nrbufs < PIPE_BUFFERS)
  1188. return 0;
  1189. ret = 0;
  1190. mutex_lock(&pipe->inode->i_mutex);
  1191. while (pipe->nrbufs >= PIPE_BUFFERS) {
  1192. if (!pipe->readers) {
  1193. send_sig(SIGPIPE, current, 0);
  1194. ret = -EPIPE;
  1195. break;
  1196. }
  1197. if (flags & SPLICE_F_NONBLOCK) {
  1198. ret = -EAGAIN;
  1199. break;
  1200. }
  1201. if (signal_pending(current)) {
  1202. ret = -ERESTARTSYS;
  1203. break;
  1204. }
  1205. pipe->waiting_writers++;
  1206. pipe_wait(pipe);
  1207. pipe->waiting_writers--;
  1208. }
  1209. mutex_unlock(&pipe->inode->i_mutex);
  1210. return ret;
  1211. }
  1212. /*
  1213. * Link contents of ipipe to opipe.
  1214. */
  1215. static int link_pipe(struct pipe_inode_info *ipipe,
  1216. struct pipe_inode_info *opipe,
  1217. size_t len, unsigned int flags)
  1218. {
  1219. struct pipe_buffer *ibuf, *obuf;
  1220. int ret = 0, i = 0, nbuf;
  1221. /*
  1222. * Potential ABBA deadlock, work around it by ordering lock
  1223. * grabbing by inode address. Otherwise two different processes
  1224. * could deadlock (one doing tee from A -> B, the other from B -> A).
  1225. */
  1226. inode_double_lock(ipipe->inode, opipe->inode);
  1227. do {
  1228. if (!opipe->readers) {
  1229. send_sig(SIGPIPE, current, 0);
  1230. if (!ret)
  1231. ret = -EPIPE;
  1232. break;
  1233. }
  1234. /*
  1235. * If we have iterated all input buffers or ran out of
  1236. * output room, break.
  1237. */
  1238. if (i >= ipipe->nrbufs || opipe->nrbufs >= PIPE_BUFFERS)
  1239. break;
  1240. ibuf = ipipe->bufs + ((ipipe->curbuf + i) & (PIPE_BUFFERS - 1));
  1241. nbuf = (opipe->curbuf + opipe->nrbufs) & (PIPE_BUFFERS - 1);
  1242. /*
  1243. * Get a reference to this pipe buffer,
  1244. * so we can copy the contents over.
  1245. */
  1246. ibuf->ops->get(ipipe, ibuf);
  1247. obuf = opipe->bufs + nbuf;
  1248. *obuf = *ibuf;
  1249. /*
  1250. * Don't inherit the gift flag, we need to
  1251. * prevent multiple steals of this page.
  1252. */
  1253. obuf->flags &= ~PIPE_BUF_FLAG_GIFT;
  1254. if (obuf->len > len)
  1255. obuf->len = len;
  1256. opipe->nrbufs++;
  1257. ret += obuf->len;
  1258. len -= obuf->len;
  1259. i++;
  1260. } while (len);
  1261. inode_double_unlock(ipipe->inode, opipe->inode);
  1262. /*
  1263. * If we put data in the output pipe, wakeup any potential readers.
  1264. */
  1265. if (ret > 0) {
  1266. smp_mb();
  1267. if (waitqueue_active(&opipe->wait))
  1268. wake_up_interruptible(&opipe->wait);
  1269. kill_fasync(&opipe->fasync_readers, SIGIO, POLL_IN);
  1270. }
  1271. return ret;
  1272. }
  1273. /*
  1274. * This is a tee(1) implementation that works on pipes. It doesn't copy
  1275. * any data, it simply references the 'in' pages on the 'out' pipe.
  1276. * The 'flags' used are the SPLICE_F_* variants, currently the only
  1277. * applicable one is SPLICE_F_NONBLOCK.
  1278. */
  1279. static long do_tee(struct file *in, struct file *out, size_t len,
  1280. unsigned int flags)
  1281. {
  1282. struct pipe_inode_info *ipipe = pipe_info(in->f_path.dentry->d_inode);
  1283. struct pipe_inode_info *opipe = pipe_info(out->f_path.dentry->d_inode);
  1284. int ret = -EINVAL;
  1285. /*
  1286. * Duplicate the contents of ipipe to opipe without actually
  1287. * copying the data.
  1288. */
  1289. if (ipipe && opipe && ipipe != opipe) {
  1290. /*
  1291. * Keep going, unless we encounter an error. The ipipe/opipe
  1292. * ordering doesn't really matter.
  1293. */
  1294. ret = link_ipipe_prep(ipipe, flags);
  1295. if (!ret) {
  1296. ret = link_opipe_prep(opipe, flags);
  1297. if (!ret) {
  1298. ret = link_pipe(ipipe, opipe, len, flags);
  1299. if (!ret && (flags & SPLICE_F_NONBLOCK))
  1300. ret = -EAGAIN;
  1301. }
  1302. }
  1303. }
  1304. return ret;
  1305. }
  1306. asmlinkage long sys_tee(int fdin, int fdout, size_t len, unsigned int flags)
  1307. {
  1308. struct file *in;
  1309. int error, fput_in;
  1310. if (unlikely(!len))
  1311. return 0;
  1312. error = -EBADF;
  1313. in = fget_light(fdin, &fput_in);
  1314. if (in) {
  1315. if (in->f_mode & FMODE_READ) {
  1316. int fput_out;
  1317. struct file *out = fget_light(fdout, &fput_out);
  1318. if (out) {
  1319. if (out->f_mode & FMODE_WRITE)
  1320. error = do_tee(in, out, len, flags);
  1321. fput_light(out, fput_out);
  1322. }
  1323. }
  1324. fput_light(in, fput_in);
  1325. }
  1326. return error;
  1327. }