diff options
Diffstat (limited to 'arch/s390/lib/uaccess_pt.c')
-rw-r--r-- | arch/s390/lib/uaccess_pt.c | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/arch/s390/lib/uaccess_pt.c b/arch/s390/lib/uaccess_pt.c index a70ee84c0241..c1aaf22c326b 100644 --- a/arch/s390/lib/uaccess_pt.c +++ b/arch/s390/lib/uaccess_pt.c | |||
@@ -172,6 +172,8 @@ static size_t strnlen_user_pt(size_t count, const char __user *src) | |||
172 | unsigned long offset, done, len, kaddr; | 172 | unsigned long offset, done, len, kaddr; |
173 | size_t len_str; | 173 | size_t len_str; |
174 | 174 | ||
175 | if (unlikely(!count)) | ||
176 | return 0; | ||
175 | if (segment_eq(get_fs(), KERNEL_DS)) | 177 | if (segment_eq(get_fs(), KERNEL_DS)) |
176 | return strnlen((const char __kernel __force *) src, count) + 1; | 178 | return strnlen((const char __kernel __force *) src, count) + 1; |
177 | done = 0; | 179 | done = 0; |
@@ -202,6 +204,8 @@ static size_t strncpy_from_user_pt(size_t count, const char __user *src, | |||
202 | { | 204 | { |
203 | size_t n = strnlen_user_pt(count, src); | 205 | size_t n = strnlen_user_pt(count, src); |
204 | 206 | ||
207 | if (unlikely(!count)) | ||
208 | return 0; | ||
205 | if (!n) | 209 | if (!n) |
206 | return -EFAULT; | 210 | return -EFAULT; |
207 | if (n > count) | 211 | if (n > count) |