[linux-block.git] / fs / smb / server / ksmbd_work.c

// SPDX-License-Identifier: GPL-2.0-or-later
/*
 *   Copyright (C) 2019 Samsung Electronics Co., Ltd.
 */

#include <linux/list.h>
#include <linux/mm.h>
#include <linux/slab.h>
#include <linux/workqueue.h>

#include "server.h"
#include "connection.h"
#include "ksmbd_work.h"
#include "mgmt/ksmbd_ida.h"

static struct kmem_cache *work_cache;
static struct workqueue_struct *ksmbd_wq;

struct ksmbd_work *ksmbd_alloc_work_struct(void)
{
	struct ksmbd_work *work = kmem_cache_zalloc(work_cache, GFP_KERNEL);

	if (work) {
		work->compound_fid = KSMBD_NO_FID;
		work->compound_pfid = KSMBD_NO_FID;
		INIT_LIST_HEAD(&work->request_entry);
		INIT_LIST_HEAD(&work->async_request_entry);
		INIT_LIST_HEAD(&work->fp_entry);
		INIT_LIST_HEAD(&work->interim_entry);
		INIT_LIST_HEAD(&work->aux_read_list);
		work->iov_alloc_cnt = 4;
		work->iov = kcalloc(work->iov_alloc_cnt, sizeof(struct kvec),
				    GFP_KERNEL);
		if (!work->iov) {
			kmem_cache_free(work_cache, work);
			work = NULL;
		}
	}
	return work;
}

void ksmbd_free_work_struct(struct ksmbd_work *work)
{
	struct aux_read *ar, *tmp;

	WARN_ON(work->saved_cred != NULL);

	kvfree(work->response_buf);

	list_for_each_entry_safe(ar, tmp, &work->aux_read_list, entry) {
		kvfree(ar->buf);
		list_del(&ar->entry);
		kfree(ar);
	}

	kfree(work->tr_buf);
	kvfree(work->request_buf);
	kfree(work->iov);
	if (work->async_id)
		ksmbd_release_id(&work->conn->async_ida, work->async_id);
	kmem_cache_free(work_cache, work);
}

void ksmbd_work_pool_destroy(void)
{
	kmem_cache_destroy(work_cache);
}

int ksmbd_work_pool_init(void)
{
	work_cache = kmem_cache_create("ksmbd_work_cache",
				       sizeof(struct ksmbd_work), 0,
				       SLAB_HWCACHE_ALIGN, NULL);
	if (!work_cache)
		return -ENOMEM;
	return 0;
}

int ksmbd_workqueue_init(void)
{
	ksmbd_wq = alloc_workqueue("ksmbd-io", 0, 0);
	if (!ksmbd_wq)
		return -ENOMEM;
	return 0;
}

void ksmbd_workqueue_destroy(void)
{
	destroy_workqueue(ksmbd_wq);
	ksmbd_wq = NULL;
}

bool ksmbd_queue_work(struct ksmbd_work *work)
{
	return queue_work(ksmbd_wq, &work->work);
}

static inline void __ksmbd_iov_pin(struct ksmbd_work *work, void *ib,
				   unsigned int ib_len)
{
	work->iov[++work->iov_idx].iov_base = ib;
	work->iov[work->iov_idx].iov_len = ib_len;
	work->iov_cnt++;
}

static int __ksmbd_iov_pin_rsp(struct ksmbd_work *work, void *ib, int len,
			       void *aux_buf, unsigned int aux_size)
{
	struct aux_read *ar;
	int need_iov_cnt = 1;

	if (aux_size) {
		need_iov_cnt++;
		ar = kmalloc(sizeof(struct aux_read), GFP_KERNEL);
		if (!ar)
			return -ENOMEM;
	}

	if (work->iov_alloc_cnt < work->iov_cnt + need_iov_cnt) {
		struct kvec *new;

		work->iov_alloc_cnt += 4;
		new = krealloc(work->iov,
			       sizeof(struct kvec) * work->iov_alloc_cnt,
			       GFP_KERNEL | __GFP_ZERO);
		if (!new)
			return -ENOMEM;
		work->iov = new;
	}

	/* Plus rfc_length size on first iov */
	if (!work->iov_idx) {
		work->iov[work->iov_idx].iov_base = work->response_buf;
		*(__be32 *)work->iov[0].iov_base = 0;
		work->iov[work->iov_idx].iov_len = 4;
		work->iov_cnt++;
	}

	__ksmbd_iov_pin(work, ib, len);
	inc_rfc1001_len(work->iov[0].iov_base, len);

	if (aux_size) {
		__ksmbd_iov_pin(work, aux_buf, aux_size);
		inc_rfc1001_len(work->iov[0].iov_base, aux_size);

		ar->buf = aux_buf;
		list_add(&ar->entry, &work->aux_read_list);
	}

	return 0;
}

int ksmbd_iov_pin_rsp(struct ksmbd_work *work, void *ib, int len)
{
	return __ksmbd_iov_pin_rsp(work, ib, len, NULL, 0);
}

int ksmbd_iov_pin_rsp_read(struct ksmbd_work *work, void *ib, int len,
			   void *aux_buf, unsigned int aux_size)
{
	return __ksmbd_iov_pin_rsp(work, ib, len, aux_buf, aux_size);
}

int allocate_interim_rsp_buf(struct ksmbd_work *work)
{
	work->response_buf = kzalloc(MAX_CIFS_SMALL_BUFFER_SIZE, GFP_KERNEL);
	if (!work->response_buf)
		return -ENOMEM;
	work->response_sz = MAX_CIFS_SMALL_BUFFER_SIZE;
	return 0;
}
Commit	Line	Data
0626e664 NJ	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"
0626e664	14	#include "mgmt/ksmbd_ida.h"
0626e664 NJ	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);
e2b76ab8 NJ	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	}
0626e664 NJ	38	}
	39	return work;
	40	}
	41
	42	void ksmbd_free_work_struct(struct ksmbd_work *work)
	43	{
e2b76ab8 NJ	44	struct aux_read ar, tmp;
e2b76ab8 NJ	45
0626e664	46	WARN_ON(work->saved_cred != NULL);
0626e664	47
c30f4eb8	48	kvfree(work->response_buf);
e2b76ab8 NJ	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
79f6b11a NJ	56	kfree(work->tr_buf);
79f6b11a NJ	57	kvfree(work->request_buf);
e2b76ab8	58	kfree(work->iov);
0626e664	59	if (work->async_id)
d40012a8	60	ksmbd_release_id(&work->conn->async_ida, work->async_id);
0626e664 NJ	61	kmem_cache_free(work_cache, work);
	62	}
	63
	64	void ksmbd_work_pool_destroy(void)
	65	{
	66	kmem_cache_destroy(work_cache);
	67	}
	68
	69	int ksmbd_work_pool_init(void)
	70	{
	71	work_cache = kmem_cache_create("ksmbd_work_cache",
070fb21e NJ	72	sizeof(struct ksmbd_work), 0,
070fb21e NJ	73	SLAB_HWCACHE_ALIGN, NULL);
0626e664 NJ	74	if (!work_cache)
	75	return -ENOMEM;
	76	return 0;
	77	}
	78
	79	int ksmbd_workqueue_init(void)
	80	{
	81	ksmbd_wq = alloc_workqueue("ksmbd-io", 0, 0);
	82	if (!ksmbd_wq)
	83	return -ENOMEM;
	84	return 0;
	85	}
	86
	87	void ksmbd_workqueue_destroy(void)
	88	{
0626e664 NJ	89	destroy_workqueue(ksmbd_wq);
	90	ksmbd_wq = NULL;
	91	}
	92
	93	bool ksmbd_queue_work(struct ksmbd_work *work)
	94	{
	95	return queue_work(ksmbd_wq, &work->work);
	96	}
e2b76ab8	97
1819a904 NJ	98	static inline void __ksmbd_iov_pin(struct ksmbd_work work, void ib,
1819a904 NJ	99	unsigned int ib_len)
e2b76ab8	100	{
1819a904 NJ	101	work->iov[++work->iov_idx].iov_base = ib;
	102	work->iov[work->iov_idx].iov_len = ib_len;
	103	work->iov_cnt++;
	104	}
	105
	106	static int __ksmbd_iov_pin_rsp(struct ksmbd_work work, void ib, int len,
	107	void *aux_buf, unsigned int aux_size)
	108	{
	109	struct aux_read *ar;
	110	int need_iov_cnt = 1;
e2b76ab8	111
1819a904 NJ	112	if (aux_size) {
	113	need_iov_cnt++;
	114	ar = kmalloc(sizeof(struct aux_read), GFP_KERNEL);
	115	if (!ar)
	116	return -ENOMEM;
	117	}
	118
	119	if (work->iov_alloc_cnt < work->iov_cnt + need_iov_cnt) {
e2b76ab8 NJ	120	struct kvec *new;
	121
	122	work->iov_alloc_cnt += 4;
	123	new = krealloc(work->iov,
	124	sizeof(struct kvec) * work->iov_alloc_cnt,
	125	GFP_KERNEL \| __GFP_ZERO);
	126	if (!new)
	127	return -ENOMEM;
	128	work->iov = new;
	129	}
	130
e2b76ab8 NJ	131	/* Plus rfc_length size on first iov */
	132	if (!work->iov_idx) {
	133	work->iov[work->iov_idx].iov_base = work->response_buf;
	134	(__be32 )work->iov[0].iov_base = 0;
	135	work->iov[work->iov_idx].iov_len = 4;
	136	work->iov_cnt++;
	137	}
	138
1819a904	139	__ksmbd_iov_pin(work, ib, len);
e2b76ab8 NJ	140	inc_rfc1001_len(work->iov[0].iov_base, len);
	141
	142	if (aux_size) {
1819a904	143	__ksmbd_iov_pin(work, aux_buf, aux_size);
e2b76ab8 NJ	144	inc_rfc1001_len(work->iov[0].iov_base, aux_size);
e2b76ab8 NJ	145
e2b76ab8 NJ	146	ar->buf = aux_buf;
	147	list_add(&ar->entry, &work->aux_read_list);
	148	}
	149
	150	return 0;
	151	}
	152
	153	int ksmbd_iov_pin_rsp(struct ksmbd_work work, void ib, int len)
	154	{
	155	return __ksmbd_iov_pin_rsp(work, ib, len, NULL, 0);
	156	}
	157
	158	int ksmbd_iov_pin_rsp_read(struct ksmbd_work work, void ib, int len,
	159	void *aux_buf, unsigned int aux_size)
	160	{
	161	return __ksmbd_iov_pin_rsp(work, ib, len, aux_buf, aux_size);
	162	}
	163
041bba44	164	int allocate_interim_rsp_buf(struct ksmbd_work *work)
e2b76ab8	165	{
041bba44 NJ	166	work->response_buf = kzalloc(MAX_CIFS_SMALL_BUFFER_SIZE, GFP_KERNEL);
	167	if (!work->response_buf)
	168	return -ENOMEM;
	169	work->response_sz = MAX_CIFS_SMALL_BUFFER_SIZE;
	170	return 0;
e2b76ab8	171	}