Merge tag 'icc-6.7-rc5' of git://git.kernel.org/pub/scm/linux/kernel/git/djakov/icc...
[linux-2.6-microblaze.git] / fs / smb / server / ksmbd_work.c
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  *   Copyright (C) 2019 Samsung Electronics Co., Ltd.
4  */
5
6 #include <linux/list.h>
7 #include <linux/mm.h>
8 #include <linux/slab.h>
9 #include <linux/workqueue.h>
10
11 #include "server.h"
12 #include "connection.h"
13 #include "ksmbd_work.h"
14 #include "mgmt/ksmbd_ida.h"
15
16 static struct kmem_cache *work_cache;
17 static struct workqueue_struct *ksmbd_wq;
18
19 struct ksmbd_work *ksmbd_alloc_work_struct(void)
20 {
21         struct ksmbd_work *work = kmem_cache_zalloc(work_cache, GFP_KERNEL);
22
23         if (work) {
24                 work->compound_fid = KSMBD_NO_FID;
25                 work->compound_pfid = KSMBD_NO_FID;
26                 INIT_LIST_HEAD(&work->request_entry);
27                 INIT_LIST_HEAD(&work->async_request_entry);
28                 INIT_LIST_HEAD(&work->fp_entry);
29                 INIT_LIST_HEAD(&work->interim_entry);
30                 INIT_LIST_HEAD(&work->aux_read_list);
31                 work->iov_alloc_cnt = 4;
32                 work->iov = kcalloc(work->iov_alloc_cnt, sizeof(struct kvec),
33                                     GFP_KERNEL);
34                 if (!work->iov) {
35                         kmem_cache_free(work_cache, work);
36                         work = NULL;
37                 }
38         }
39         return work;
40 }
41
42 void ksmbd_free_work_struct(struct ksmbd_work *work)
43 {
44         struct aux_read *ar, *tmp;
45
46         WARN_ON(work->saved_cred != NULL);
47
48         kvfree(work->response_buf);
49
50         list_for_each_entry_safe(ar, tmp, &work->aux_read_list, entry) {
51                 kvfree(ar->buf);
52                 list_del(&ar->entry);
53                 kfree(ar);
54         }
55
56         kfree(work->tr_buf);
57         kvfree(work->request_buf);
58         kfree(work->iov);
59         if (!list_empty(&work->interim_entry))
60                 list_del(&work->interim_entry);
61
62         if (work->async_id)
63                 ksmbd_release_id(&work->conn->async_ida, work->async_id);
64         kmem_cache_free(work_cache, work);
65 }
66
67 void ksmbd_work_pool_destroy(void)
68 {
69         kmem_cache_destroy(work_cache);
70 }
71
72 int ksmbd_work_pool_init(void)
73 {
74         work_cache = kmem_cache_create("ksmbd_work_cache",
75                                        sizeof(struct ksmbd_work), 0,
76                                        SLAB_HWCACHE_ALIGN, NULL);
77         if (!work_cache)
78                 return -ENOMEM;
79         return 0;
80 }
81
82 int ksmbd_workqueue_init(void)
83 {
84         ksmbd_wq = alloc_workqueue("ksmbd-io", 0, 0);
85         if (!ksmbd_wq)
86                 return -ENOMEM;
87         return 0;
88 }
89
90 void ksmbd_workqueue_destroy(void)
91 {
92         destroy_workqueue(ksmbd_wq);
93         ksmbd_wq = NULL;
94 }
95
96 bool ksmbd_queue_work(struct ksmbd_work *work)
97 {
98         return queue_work(ksmbd_wq, &work->work);
99 }
100
101 static inline void __ksmbd_iov_pin(struct ksmbd_work *work, void *ib,
102                                    unsigned int ib_len)
103 {
104         work->iov[++work->iov_idx].iov_base = ib;
105         work->iov[work->iov_idx].iov_len = ib_len;
106         work->iov_cnt++;
107 }
108
109 static int __ksmbd_iov_pin_rsp(struct ksmbd_work *work, void *ib, int len,
110                                void *aux_buf, unsigned int aux_size)
111 {
112         struct aux_read *ar = NULL;
113         int need_iov_cnt = 1;
114
115         if (aux_size) {
116                 need_iov_cnt++;
117                 ar = kmalloc(sizeof(struct aux_read), GFP_KERNEL);
118                 if (!ar)
119                         return -ENOMEM;
120         }
121
122         if (work->iov_alloc_cnt < work->iov_cnt + need_iov_cnt) {
123                 struct kvec *new;
124
125                 work->iov_alloc_cnt += 4;
126                 new = krealloc(work->iov,
127                                sizeof(struct kvec) * work->iov_alloc_cnt,
128                                GFP_KERNEL | __GFP_ZERO);
129                 if (!new) {
130                         kfree(ar);
131                         work->iov_alloc_cnt -= 4;
132                         return -ENOMEM;
133                 }
134                 work->iov = new;
135         }
136
137         /* Plus rfc_length size on first iov */
138         if (!work->iov_idx) {
139                 work->iov[work->iov_idx].iov_base = work->response_buf;
140                 *(__be32 *)work->iov[0].iov_base = 0;
141                 work->iov[work->iov_idx].iov_len = 4;
142                 work->iov_cnt++;
143         }
144
145         __ksmbd_iov_pin(work, ib, len);
146         inc_rfc1001_len(work->iov[0].iov_base, len);
147
148         if (aux_size) {
149                 __ksmbd_iov_pin(work, aux_buf, aux_size);
150                 inc_rfc1001_len(work->iov[0].iov_base, aux_size);
151
152                 ar->buf = aux_buf;
153                 list_add(&ar->entry, &work->aux_read_list);
154         }
155
156         return 0;
157 }
158
159 int ksmbd_iov_pin_rsp(struct ksmbd_work *work, void *ib, int len)
160 {
161         return __ksmbd_iov_pin_rsp(work, ib, len, NULL, 0);
162 }
163
164 int ksmbd_iov_pin_rsp_read(struct ksmbd_work *work, void *ib, int len,
165                            void *aux_buf, unsigned int aux_size)
166 {
167         return __ksmbd_iov_pin_rsp(work, ib, len, aux_buf, aux_size);
168 }
169
170 int allocate_interim_rsp_buf(struct ksmbd_work *work)
171 {
172         work->response_buf = kzalloc(MAX_CIFS_SMALL_BUFFER_SIZE, GFP_KERNEL);
173         if (!work->response_buf)
174                 return -ENOMEM;
175         work->response_sz = MAX_CIFS_SMALL_BUFFER_SIZE;
176         return 0;
177 }