6 * Copyright (c) 2002 James Morris <jmorris@intercode.com.au>
7 * 2002 Adam J. Richter <adam@yggdrasil.com>
8 * 2004 Jean-Luc Cooke <jlcooke@certainkey.com>
10 * This program is free software; you can redistribute it and/or modify it
11 * under the terms of the GNU General Public License as published by the Free
12 * Software Foundation; either version 2 of the License, or (at your option)
17 #include <crypto/scatterwalk.h>
18 #include <linux/kernel.h>
20 #include <linux/module.h>
21 #include <linux/pagemap.h>
22 #include <linux/highmem.h>
23 #include <linux/scatterlist.h>
25 static inline void memcpy_dir(void *buf, void *sgdata, size_t nbytes, int out)
27 void *src = out ? buf : sgdata;
28 void *dst = out ? sgdata : buf;
30 memcpy(dst, src, nbytes);
33 void scatterwalk_start(struct scatter_walk *walk, struct scatterlist *sg)
39 walk->offset = sg->offset;
41 EXPORT_SYMBOL_GPL(scatterwalk_start);
43 void *scatterwalk_map(struct scatter_walk *walk)
45 return kmap_atomic(scatterwalk_page(walk)) +
46 offset_in_page(walk->offset);
48 EXPORT_SYMBOL_GPL(scatterwalk_map);
50 static void scatterwalk_pagedone(struct scatter_walk *walk, int out,
56 page = sg_page(walk->sg) + ((walk->offset - 1) >> PAGE_SHIFT);
58 flush_dcache_page(page);
62 walk->offset += PAGE_SIZE - 1;
63 walk->offset &= PAGE_MASK;
64 if (walk->offset >= walk->sg->offset + walk->sg->length)
65 scatterwalk_start(walk, sg_next(walk->sg));
69 void scatterwalk_done(struct scatter_walk *walk, int out, int more)
71 if (!(scatterwalk_pagelen(walk) & (PAGE_SIZE - 1)) || !more)
72 scatterwalk_pagedone(walk, out, more);
74 EXPORT_SYMBOL_GPL(scatterwalk_done);
76 void scatterwalk_copychunks(void *buf, struct scatter_walk *walk,
77 size_t nbytes, int out)
80 unsigned int len_this_page = scatterwalk_pagelen(walk);
83 if (len_this_page > nbytes)
84 len_this_page = nbytes;
86 vaddr = scatterwalk_map(walk);
87 memcpy_dir(buf, vaddr, len_this_page, out);
88 scatterwalk_unmap(vaddr);
90 scatterwalk_advance(walk, len_this_page);
92 if (nbytes == len_this_page)
96 nbytes -= len_this_page;
98 scatterwalk_pagedone(walk, out, 1);
101 EXPORT_SYMBOL_GPL(scatterwalk_copychunks);
103 void scatterwalk_map_and_copy(void *buf, struct scatterlist *sg,
104 unsigned int start, unsigned int nbytes, int out)
106 struct scatter_walk walk;
107 struct scatterlist tmp[2];
112 sg = scatterwalk_ffwd(tmp, sg, start);
114 if (sg_page(sg) == virt_to_page(buf) &&
115 sg->offset == offset_in_page(buf))
118 scatterwalk_start(&walk, sg);
119 scatterwalk_copychunks(buf, &walk, nbytes, out);
120 scatterwalk_done(&walk, out, 0);
122 EXPORT_SYMBOL_GPL(scatterwalk_map_and_copy);
124 int scatterwalk_bytes_sglen(struct scatterlist *sg, int num_bytes)
126 int offset = 0, n = 0;
128 /* num_bytes is too small */
129 if (num_bytes < sg->length)
133 offset += sg->length;
137 /* num_bytes is too large */
138 if (unlikely(!sg && (num_bytes < offset)))
140 } while (sg && (num_bytes > offset));
144 EXPORT_SYMBOL_GPL(scatterwalk_bytes_sglen);
146 struct scatterlist *scatterwalk_ffwd(struct scatterlist dst[2],
147 struct scatterlist *src,
154 if (src->length > len)
161 sg_set_page(dst, sg_page(src), src->length - len, src->offset + len);
162 scatterwalk_crypto_chain(dst, sg_next(src), 0, 2);
166 EXPORT_SYMBOL_GPL(scatterwalk_ffwd);