1 // SPDX-License-Identifier: GPL-2.0-only
4 * Phillip Lougher <phillip@squashfs.org.uk>
7 #include <linux/kernel.h>
8 #include <linux/slab.h>
9 #include <linux/pagemap.h>
10 #include "squashfs_fs_sb.h"
11 #include "decompressor.h"
12 #include "page_actor.h"
15 * This file contains implementations of page_actor for decompressing into
16 * an intermediate buffer, and for decompressing directly into the
19 * Calling code should avoid sleeping between calls to squashfs_first_page()
20 * and squashfs_finish_page().
23 /* Implementation of page_actor for decompressing into intermediate buffer */
24 static void *cache_first_page(struct squashfs_page_actor *actor)
27 return actor->buffer[0];
30 static void *cache_next_page(struct squashfs_page_actor *actor)
32 if (actor->next_page == actor->pages)
35 return actor->buffer[actor->next_page++];
38 static void cache_finish_page(struct squashfs_page_actor *actor)
43 struct squashfs_page_actor *squashfs_page_actor_init(void **buffer,
44 int pages, int length)
46 struct squashfs_page_actor *actor = kmalloc(sizeof(*actor), GFP_KERNEL);
51 actor->length = length ? : pages * PAGE_SIZE;
52 actor->buffer = buffer;
55 actor->squashfs_first_page = cache_first_page;
56 actor->squashfs_next_page = cache_next_page;
57 actor->squashfs_finish_page = cache_finish_page;
61 /* Implementation of page_actor for decompressing directly into page cache. */
62 static void *handle_next_page(struct squashfs_page_actor *actor)
64 int max_pages = (actor->length + PAGE_SIZE - 1) >> PAGE_SHIFT;
66 if (actor->returned_pages == max_pages)
69 if ((actor->next_page == actor->pages) ||
70 (actor->next_index != actor->page[actor->next_page]->index)) {
71 if (actor->alloc_buffer) {
72 void *tmp_buffer = kmalloc(PAGE_SIZE, GFP_KERNEL);
75 actor->tmp_buffer = tmp_buffer;
77 actor->returned_pages++;
83 actor->returned_pages++;
84 return ERR_PTR(-ENOMEM);
88 actor->returned_pages++;
89 return actor->pageaddr = kmap_local_page(actor->page[actor->next_page++]);
92 static void *direct_first_page(struct squashfs_page_actor *actor)
94 return handle_next_page(actor);
97 static void *direct_next_page(struct squashfs_page_actor *actor)
100 kunmap_local(actor->pageaddr);
102 kfree(actor->tmp_buffer);
103 actor->pageaddr = actor->tmp_buffer = NULL;
105 return handle_next_page(actor);
108 static void direct_finish_page(struct squashfs_page_actor *actor)
111 kunmap_local(actor->pageaddr);
113 kfree(actor->tmp_buffer);
116 struct squashfs_page_actor *squashfs_page_actor_init_special(struct squashfs_sb_info *msblk,
117 struct page **page, int pages, int length)
119 struct squashfs_page_actor *actor = kmalloc(sizeof(*actor), GFP_KERNEL);
124 actor->length = length ? : pages * PAGE_SIZE;
126 actor->pages = pages;
127 actor->next_page = 0;
128 actor->returned_pages = 0;
129 actor->next_index = page[0]->index & ~((1 << (msblk->block_log - PAGE_SHIFT)) - 1);
130 actor->pageaddr = NULL;
131 actor->tmp_buffer = NULL;
132 actor->alloc_buffer = msblk->decompressor->alloc_buffer;
133 actor->squashfs_first_page = direct_first_page;
134 actor->squashfs_next_page = direct_next_page;
135 actor->squashfs_finish_page = direct_finish_page;