seek.c 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104
  1. // SPDX-License-Identifier: GPL-2.0
  2. /*
  3. * Copyright (C) 2017 Red Hat, Inc.
  4. * Copyright (c) 2018-2021 Christoph Hellwig.
  5. */
  6. #include <linux/module.h>
  7. #include <linux/compiler.h>
  8. #include <linux/fs.h>
  9. #include <linux/iomap.h>
  10. #include <linux/pagemap.h>
  11. #include <linux/pagevec.h>
  12. static loff_t iomap_seek_hole_iter(const struct iomap_iter *iter,
  13. loff_t *hole_pos)
  14. {
  15. loff_t length = iomap_length(iter);
  16. switch (iter->iomap.type) {
  17. case IOMAP_UNWRITTEN:
  18. *hole_pos = mapping_seek_hole_data(iter->inode->i_mapping,
  19. iter->pos, iter->pos + length, SEEK_HOLE);
  20. if (*hole_pos == iter->pos + length)
  21. return length;
  22. return 0;
  23. case IOMAP_HOLE:
  24. *hole_pos = iter->pos;
  25. return 0;
  26. default:
  27. return length;
  28. }
  29. }
  30. loff_t
  31. iomap_seek_hole(struct inode *inode, loff_t pos, const struct iomap_ops *ops)
  32. {
  33. loff_t size = i_size_read(inode);
  34. struct iomap_iter iter = {
  35. .inode = inode,
  36. .pos = pos,
  37. .flags = IOMAP_REPORT,
  38. };
  39. int ret;
  40. /* Nothing to be found before or beyond the end of the file. */
  41. if (pos < 0 || pos >= size)
  42. return -ENXIO;
  43. iter.len = size - pos;
  44. while ((ret = iomap_iter(&iter, ops)) > 0)
  45. iter.processed = iomap_seek_hole_iter(&iter, &pos);
  46. if (ret < 0)
  47. return ret;
  48. if (iter.len) /* found hole before EOF */
  49. return pos;
  50. return size;
  51. }
  52. EXPORT_SYMBOL_GPL(iomap_seek_hole);
  53. static loff_t iomap_seek_data_iter(const struct iomap_iter *iter,
  54. loff_t *hole_pos)
  55. {
  56. loff_t length = iomap_length(iter);
  57. switch (iter->iomap.type) {
  58. case IOMAP_HOLE:
  59. return length;
  60. case IOMAP_UNWRITTEN:
  61. *hole_pos = mapping_seek_hole_data(iter->inode->i_mapping,
  62. iter->pos, iter->pos + length, SEEK_DATA);
  63. if (*hole_pos < 0)
  64. return length;
  65. return 0;
  66. default:
  67. *hole_pos = iter->pos;
  68. return 0;
  69. }
  70. }
  71. loff_t
  72. iomap_seek_data(struct inode *inode, loff_t pos, const struct iomap_ops *ops)
  73. {
  74. loff_t size = i_size_read(inode);
  75. struct iomap_iter iter = {
  76. .inode = inode,
  77. .pos = pos,
  78. .flags = IOMAP_REPORT,
  79. };
  80. int ret;
  81. /* Nothing to be found before or beyond the end of the file. */
  82. if (pos < 0 || pos >= size)
  83. return -ENXIO;
  84. iter.len = size - pos;
  85. while ((ret = iomap_iter(&iter, ops)) > 0)
  86. iter.processed = iomap_seek_data_iter(&iter, &pos);
  87. if (ret < 0)
  88. return ret;
  89. if (iter.len) /* found data before EOF */
  90. return pos;
  91. /* We've reached the end of the file without finding data */
  92. return -ENXIO;
  93. }
  94. EXPORT_SYMBOL_GPL(iomap_seek_data);