1  /* SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB */
2  /*
3   * Copyright (c) 2017, Mellanox Technologies inc.  All rights reserved.
4   */
5  
6  #ifndef _UVERBS_IOCTL_
7  #define _UVERBS_IOCTL_
8  
9  #include <rdma/uverbs_types.h>
10  #include <linux/uaccess.h>
11  #include <rdma/rdma_user_ioctl.h>
12  #include <rdma/ib_user_ioctl_verbs.h>
13  #include <rdma/ib_user_ioctl_cmds.h>
14  
15  /*
16   * =======================================
17   *	Verbs action specifications
18   * =======================================
19   */
20  
21  enum uverbs_attr_type {
22  	UVERBS_ATTR_TYPE_NA,
23  	UVERBS_ATTR_TYPE_PTR_IN,
24  	UVERBS_ATTR_TYPE_PTR_OUT,
25  	UVERBS_ATTR_TYPE_IDR,
26  	UVERBS_ATTR_TYPE_FD,
27  	UVERBS_ATTR_TYPE_RAW_FD,
28  	UVERBS_ATTR_TYPE_ENUM_IN,
29  	UVERBS_ATTR_TYPE_IDRS_ARRAY,
30  };
31  
32  enum uverbs_obj_access {
33  	UVERBS_ACCESS_READ,
34  	UVERBS_ACCESS_WRITE,
35  	UVERBS_ACCESS_NEW,
36  	UVERBS_ACCESS_DESTROY
37  };
38  
39  /* Specification of a single attribute inside the ioctl message */
40  /* good size 16 */
41  struct uverbs_attr_spec {
42  	u8 type;
43  
44  	/*
45  	 * Support extending attributes by length. Allow the user to provide
46  	 * more bytes than ptr.len, but check that everything after is zero'd
47  	 * by the user.
48  	 */
49  	u8 zero_trailing:1;
50  	/*
51  	 * Valid only for PTR_IN. Allocate and copy the data inside
52  	 * the parser
53  	 */
54  	u8 alloc_and_copy:1;
55  	u8 mandatory:1;
56  	/* True if this is from UVERBS_ATTR_UHW */
57  	u8 is_udata:1;
58  
59  	union {
60  		struct {
61  			/* Current known size to kernel */
62  			u16 len;
63  			/* User isn't allowed to provide something < min_len */
64  			u16 min_len;
65  		} ptr;
66  
67  		struct {
68  			/*
69  			 * higher bits mean the namespace and lower bits mean
70  			 * the type id within the namespace.
71  			 */
72  			u16 obj_type;
73  			u8 access;
74  		} obj;
75  
76  		struct {
77  			u8 num_elems;
78  		} enum_def;
79  	} u;
80  
81  	/* This weird split lets us remove some padding */
82  	union {
83  		struct {
84  			/*
85  			 * The enum attribute can select one of the attributes
86  			 * contained in the ids array. Currently only PTR_IN
87  			 * attributes are supported in the ids array.
88  			 */
89  			const struct uverbs_attr_spec *ids;
90  		} enum_def;
91  
92  		struct {
93  			/*
94  			 * higher bits mean the namespace and lower bits mean
95  			 * the type id within the namespace.
96  			 */
97  			u16				obj_type;
98  			u16				min_len;
99  			u16				max_len;
100  			u8				access;
101  		} objs_arr;
102  	} u2;
103  };
104  
105  /*
106   * Information about the API is loaded into a radix tree. For IOCTL we start
107   * with a tuple of:
108   *  object_id, attr_id, method_id
109   *
110   * Which is a 48 bit value, with most of the bits guaranteed to be zero. Based
111   * on the current kernel support this is compressed into 16 bit key for the
112   * radix tree. Since this compression is entirely internal to the kernel the
113   * below limits can be revised if the kernel gains additional data.
114   *
115   * With 64 leafs per node this is a 3 level radix tree.
116   *
117   * The tree encodes multiple types, and uses a scheme where OBJ_ID,0,0 returns
118   * the object slot, and OBJ_ID,METH_ID,0 and returns the method slot.
119   *
120   * This also encodes the tables for the write() and write() extended commands
121   * using the coding
122   *   OBJ_ID,UVERBS_API_METHOD_IS_WRITE,command #
123   *   OBJ_ID,UVERBS_API_METHOD_IS_WRITE_EX,command_ex #
124   * ie the WRITE path is treated as a special method type in the ioctl
125   * framework.
126   */
127  enum uapi_radix_data {
128  	UVERBS_API_NS_FLAG = 1U << UVERBS_ID_NS_SHIFT,
129  
130  	UVERBS_API_ATTR_KEY_BITS = 6,
131  	UVERBS_API_ATTR_KEY_MASK = GENMASK(UVERBS_API_ATTR_KEY_BITS - 1, 0),
132  	UVERBS_API_ATTR_BKEY_LEN = (1 << UVERBS_API_ATTR_KEY_BITS) - 1,
133  	UVERBS_API_WRITE_KEY_NUM = 1 << UVERBS_API_ATTR_KEY_BITS,
134  
135  	UVERBS_API_METHOD_KEY_BITS = 5,
136  	UVERBS_API_METHOD_KEY_SHIFT = UVERBS_API_ATTR_KEY_BITS,
137  	UVERBS_API_METHOD_KEY_NUM_CORE = 22,
138  	UVERBS_API_METHOD_IS_WRITE = 30 << UVERBS_API_METHOD_KEY_SHIFT,
139  	UVERBS_API_METHOD_IS_WRITE_EX = 31 << UVERBS_API_METHOD_KEY_SHIFT,
140  	UVERBS_API_METHOD_KEY_NUM_DRIVER =
141  		(UVERBS_API_METHOD_IS_WRITE >> UVERBS_API_METHOD_KEY_SHIFT) -
142  		UVERBS_API_METHOD_KEY_NUM_CORE,
143  	UVERBS_API_METHOD_KEY_MASK = GENMASK(
144  		UVERBS_API_METHOD_KEY_BITS + UVERBS_API_METHOD_KEY_SHIFT - 1,
145  		UVERBS_API_METHOD_KEY_SHIFT),
146  
147  	UVERBS_API_OBJ_KEY_BITS = 5,
148  	UVERBS_API_OBJ_KEY_SHIFT =
149  		UVERBS_API_METHOD_KEY_BITS + UVERBS_API_METHOD_KEY_SHIFT,
150  	UVERBS_API_OBJ_KEY_NUM_CORE = 20,
151  	UVERBS_API_OBJ_KEY_NUM_DRIVER =
152  		(1 << UVERBS_API_OBJ_KEY_BITS) - UVERBS_API_OBJ_KEY_NUM_CORE,
153  	UVERBS_API_OBJ_KEY_MASK = GENMASK(31, UVERBS_API_OBJ_KEY_SHIFT),
154  
155  	/* This id guaranteed to not exist in the radix tree */
156  	UVERBS_API_KEY_ERR = 0xFFFFFFFF,
157  };
158  
uapi_key_obj(u32 id)159  static inline __attribute_const__ u32 uapi_key_obj(u32 id)
160  {
161  	if (id & UVERBS_API_NS_FLAG) {
162  		id &= ~UVERBS_API_NS_FLAG;
163  		if (id >= UVERBS_API_OBJ_KEY_NUM_DRIVER)
164  			return UVERBS_API_KEY_ERR;
165  		id = id + UVERBS_API_OBJ_KEY_NUM_CORE;
166  	} else {
167  		if (id >= UVERBS_API_OBJ_KEY_NUM_CORE)
168  			return UVERBS_API_KEY_ERR;
169  	}
170  
171  	return id << UVERBS_API_OBJ_KEY_SHIFT;
172  }
173  
uapi_key_is_object(u32 key)174  static inline __attribute_const__ bool uapi_key_is_object(u32 key)
175  {
176  	return (key & ~UVERBS_API_OBJ_KEY_MASK) == 0;
177  }
178  
uapi_key_ioctl_method(u32 id)179  static inline __attribute_const__ u32 uapi_key_ioctl_method(u32 id)
180  {
181  	if (id & UVERBS_API_NS_FLAG) {
182  		id &= ~UVERBS_API_NS_FLAG;
183  		if (id >= UVERBS_API_METHOD_KEY_NUM_DRIVER)
184  			return UVERBS_API_KEY_ERR;
185  		id = id + UVERBS_API_METHOD_KEY_NUM_CORE;
186  	} else {
187  		id++;
188  		if (id >= UVERBS_API_METHOD_KEY_NUM_CORE)
189  			return UVERBS_API_KEY_ERR;
190  	}
191  
192  	return id << UVERBS_API_METHOD_KEY_SHIFT;
193  }
194  
uapi_key_write_method(u32 id)195  static inline __attribute_const__ u32 uapi_key_write_method(u32 id)
196  {
197  	if (id >= UVERBS_API_WRITE_KEY_NUM)
198  		return UVERBS_API_KEY_ERR;
199  	return UVERBS_API_METHOD_IS_WRITE | id;
200  }
201  
uapi_key_write_ex_method(u32 id)202  static inline __attribute_const__ u32 uapi_key_write_ex_method(u32 id)
203  {
204  	if (id >= UVERBS_API_WRITE_KEY_NUM)
205  		return UVERBS_API_KEY_ERR;
206  	return UVERBS_API_METHOD_IS_WRITE_EX | id;
207  }
208  
209  static inline __attribute_const__ u32
uapi_key_attr_to_ioctl_method(u32 attr_key)210  uapi_key_attr_to_ioctl_method(u32 attr_key)
211  {
212  	return attr_key &
213  	       (UVERBS_API_OBJ_KEY_MASK | UVERBS_API_METHOD_KEY_MASK);
214  }
215  
uapi_key_is_ioctl_method(u32 key)216  static inline __attribute_const__ bool uapi_key_is_ioctl_method(u32 key)
217  {
218  	unsigned int method = key & UVERBS_API_METHOD_KEY_MASK;
219  
220  	return method != 0 && method < UVERBS_API_METHOD_IS_WRITE &&
221  	       (key & UVERBS_API_ATTR_KEY_MASK) == 0;
222  }
223  
uapi_key_is_write_method(u32 key)224  static inline __attribute_const__ bool uapi_key_is_write_method(u32 key)
225  {
226  	return (key & UVERBS_API_METHOD_KEY_MASK) == UVERBS_API_METHOD_IS_WRITE;
227  }
228  
uapi_key_is_write_ex_method(u32 key)229  static inline __attribute_const__ bool uapi_key_is_write_ex_method(u32 key)
230  {
231  	return (key & UVERBS_API_METHOD_KEY_MASK) ==
232  	       UVERBS_API_METHOD_IS_WRITE_EX;
233  }
234  
uapi_key_attrs_start(u32 ioctl_method_key)235  static inline __attribute_const__ u32 uapi_key_attrs_start(u32 ioctl_method_key)
236  {
237  	/* 0 is the method slot itself */
238  	return ioctl_method_key + 1;
239  }
240  
uapi_key_attr(u32 id)241  static inline __attribute_const__ u32 uapi_key_attr(u32 id)
242  {
243  	/*
244  	 * The attr is designed to fit in the typical single radix tree node
245  	 * of 64 entries. Since allmost all methods have driver attributes we
246  	 * organize things so that the driver and core attributes interleave to
247  	 * reduce the length of the attributes array in typical cases.
248  	 */
249  	if (id & UVERBS_API_NS_FLAG) {
250  		id &= ~UVERBS_API_NS_FLAG;
251  		id++;
252  		if (id >= 1 << (UVERBS_API_ATTR_KEY_BITS - 1))
253  			return UVERBS_API_KEY_ERR;
254  		id = (id << 1) | 0;
255  	} else {
256  		if (id >= 1 << (UVERBS_API_ATTR_KEY_BITS - 1))
257  			return UVERBS_API_KEY_ERR;
258  		id = (id << 1) | 1;
259  	}
260  
261  	return id;
262  }
263  
264  /* Only true for ioctl methods */
uapi_key_is_attr(u32 key)265  static inline __attribute_const__ bool uapi_key_is_attr(u32 key)
266  {
267  	unsigned int method = key & UVERBS_API_METHOD_KEY_MASK;
268  
269  	return method != 0 && method < UVERBS_API_METHOD_IS_WRITE &&
270  	       (key & UVERBS_API_ATTR_KEY_MASK) != 0;
271  }
272  
273  /*
274   * This returns a value in the range [0 to UVERBS_API_ATTR_BKEY_LEN),
275   * basically it undoes the reservation of 0 in the ID numbering. attr_key
276   * must already be masked with UVERBS_API_ATTR_KEY_MASK, or be the output of
277   * uapi_key_attr().
278   */
uapi_bkey_attr(u32 attr_key)279  static inline __attribute_const__ u32 uapi_bkey_attr(u32 attr_key)
280  {
281  	return attr_key - 1;
282  }
283  
uapi_bkey_to_key_attr(u32 attr_bkey)284  static inline __attribute_const__ u32 uapi_bkey_to_key_attr(u32 attr_bkey)
285  {
286  	return attr_bkey + 1;
287  }
288  
289  /*
290   * =======================================
291   *	Verbs definitions
292   * =======================================
293   */
294  
295  struct uverbs_attr_def {
296  	u16                           id;
297  	struct uverbs_attr_spec       attr;
298  };
299  
300  struct uverbs_method_def {
301  	u16                                  id;
302  	/* Combination of bits from enum UVERBS_ACTION_FLAG_XXXX */
303  	u32				     flags;
304  	size_t				     num_attrs;
305  	const struct uverbs_attr_def * const (*attrs)[];
306  	int (*handler)(struct uverbs_attr_bundle *attrs);
307  };
308  
309  struct uverbs_object_def {
310  	u16					 id;
311  	const struct uverbs_obj_type	        *type_attrs;
312  	size_t				         num_methods;
313  	const struct uverbs_method_def * const (*methods)[];
314  };
315  
316  enum uapi_definition_kind {
317  	UAPI_DEF_END = 0,
318  	UAPI_DEF_OBJECT_START,
319  	UAPI_DEF_WRITE,
320  	UAPI_DEF_CHAIN_OBJ_TREE,
321  	UAPI_DEF_CHAIN,
322  	UAPI_DEF_IS_SUPPORTED_FUNC,
323  	UAPI_DEF_IS_SUPPORTED_DEV_FN,
324  };
325  
326  enum uapi_definition_scope {
327  	UAPI_SCOPE_OBJECT = 1,
328  	UAPI_SCOPE_METHOD = 2,
329  };
330  
331  struct uapi_definition {
332  	u8 kind;
333  	u8 scope;
334  	union {
335  		struct {
336  			u16 object_id;
337  		} object_start;
338  		struct {
339  			u16 command_num;
340  			u8 is_ex:1;
341  			u8 has_udata:1;
342  			u8 has_resp:1;
343  			u8 req_size;
344  			u8 resp_size;
345  		} write;
346  	};
347  
348  	union {
349  		bool (*func_is_supported)(struct ib_device *device);
350  		int (*func_write)(struct uverbs_attr_bundle *attrs);
351  		const struct uapi_definition *chain;
352  		const struct uverbs_object_def *chain_obj_tree;
353  		size_t needs_fn_offset;
354  	};
355  };
356  
357  /* Define things connected to object_id */
358  #define DECLARE_UVERBS_OBJECT(_object_id, ...)                                 \
359  	{                                                                      \
360  		.kind = UAPI_DEF_OBJECT_START,                                 \
361  		.object_start = { .object_id = _object_id },                   \
362  	},                                                                     \
363  		##__VA_ARGS__
364  
365  /* Use in a var_args of DECLARE_UVERBS_OBJECT */
366  #define DECLARE_UVERBS_WRITE(_command_num, _func, _cmd_desc, ...)              \
367  	{                                                                      \
368  		.kind = UAPI_DEF_WRITE,                                        \
369  		.scope = UAPI_SCOPE_OBJECT,                                    \
370  		.write = { .is_ex = 0, .command_num = _command_num },          \
371  		.func_write = _func,                                           \
372  		_cmd_desc,                                                     \
373  	},                                                                     \
374  		##__VA_ARGS__
375  
376  /* Use in a var_args of DECLARE_UVERBS_OBJECT */
377  #define DECLARE_UVERBS_WRITE_EX(_command_num, _func, _cmd_desc, ...)           \
378  	{                                                                      \
379  		.kind = UAPI_DEF_WRITE,                                        \
380  		.scope = UAPI_SCOPE_OBJECT,                                    \
381  		.write = { .is_ex = 1, .command_num = _command_num },          \
382  		.func_write = _func,                                           \
383  		_cmd_desc,                                                     \
384  	},                                                                     \
385  		##__VA_ARGS__
386  
387  /*
388   * Object is only supported if the function pointer named ibdev_fn in struct
389   * ib_device is not NULL.
390   */
391  #define UAPI_DEF_OBJ_NEEDS_FN(ibdev_fn)                                        \
392  	{                                                                      \
393  		.kind = UAPI_DEF_IS_SUPPORTED_DEV_FN,                          \
394  		.scope = UAPI_SCOPE_OBJECT,                                    \
395  		.needs_fn_offset =                                             \
396  			offsetof(struct ib_device_ops, ibdev_fn) +             \
397  			BUILD_BUG_ON_ZERO(sizeof_field(struct ib_device_ops,   \
398  						       ibdev_fn) !=            \
399  					  sizeof(void *)),                     \
400  	}
401  
402  /*
403   * Method is only supported if the function pointer named ibdev_fn in struct
404   * ib_device is not NULL.
405   */
406  #define UAPI_DEF_METHOD_NEEDS_FN(ibdev_fn)                                     \
407  	{                                                                      \
408  		.kind = UAPI_DEF_IS_SUPPORTED_DEV_FN,                          \
409  		.scope = UAPI_SCOPE_METHOD,                                    \
410  		.needs_fn_offset =                                             \
411  			offsetof(struct ib_device_ops, ibdev_fn) +             \
412  			BUILD_BUG_ON_ZERO(sizeof_field(struct ib_device_ops,   \
413  						       ibdev_fn) !=            \
414  					  sizeof(void *)),                     \
415  	}
416  
417  /* Call a function to determine if the entire object is supported or not */
418  #define UAPI_DEF_IS_OBJ_SUPPORTED(_func)                                       \
419  	{                                                                      \
420  		.kind = UAPI_DEF_IS_SUPPORTED_FUNC,                            \
421  		.scope = UAPI_SCOPE_OBJECT, .func_is_supported = _func,        \
422  	}
423  
424  /* Include another struct uapi_definition in this one */
425  #define UAPI_DEF_CHAIN(_def_var)                                               \
426  	{                                                                      \
427  		.kind = UAPI_DEF_CHAIN, .chain = _def_var,                     \
428  	}
429  
430  /* Temporary until the tree base description is replaced */
431  #define UAPI_DEF_CHAIN_OBJ_TREE(_object_enum, _object_ptr, ...)                \
432  	{                                                                      \
433  		.kind = UAPI_DEF_CHAIN_OBJ_TREE,                               \
434  		.object_start = { .object_id = _object_enum },                 \
435  		.chain_obj_tree = _object_ptr,                                 \
436  	},								       \
437  		##__VA_ARGS__
438  #define UAPI_DEF_CHAIN_OBJ_TREE_NAMED(_object_enum, ...)                       \
439  	UAPI_DEF_CHAIN_OBJ_TREE(_object_enum,				       \
440  		PTR_IF(IS_ENABLED(CONFIG_INFINIBAND_USER_ACCESS),	       \
441  		       &UVERBS_OBJECT(_object_enum)),			       \
442  		##__VA_ARGS__)
443  
444  /*
445   * =======================================
446   *	Attribute Specifications
447   * =======================================
448   */
449  
450  #define UVERBS_ATTR_SIZE(_min_len, _len)			\
451  	.u.ptr.min_len = _min_len, .u.ptr.len = _len
452  
453  #define UVERBS_ATTR_NO_DATA() UVERBS_ATTR_SIZE(0, 0)
454  
455  /*
456   * Specifies a uapi structure that cannot be extended. The user must always
457   * supply the whole structure and nothing more. The structure must be declared
458   * in a header under include/uapi/rdma.
459   */
460  #define UVERBS_ATTR_TYPE(_type)					\
461  	.u.ptr.min_len = sizeof(_type), .u.ptr.len = sizeof(_type)
462  /*
463   * Specifies a uapi structure where the user must provide at least up to
464   * member 'last'.  Anything after last and up until the end of the structure
465   * can be non-zero, anything longer than the end of the structure must be
466   * zero. The structure must be declared in a header under include/uapi/rdma.
467   */
468  #define UVERBS_ATTR_STRUCT(_type, _last)                                       \
469  	.zero_trailing = 1,                                                    \
470  	UVERBS_ATTR_SIZE(offsetofend(_type, _last), sizeof(_type))
471  /*
472   * Specifies at least min_len bytes must be passed in, but the amount can be
473   * larger, up to the protocol maximum size. No check for zeroing is done.
474   */
475  #define UVERBS_ATTR_MIN_SIZE(_min_len) UVERBS_ATTR_SIZE(_min_len, USHRT_MAX)
476  
477  /* Must be used in the '...' of any UVERBS_ATTR */
478  #define UA_ALLOC_AND_COPY .alloc_and_copy = 1
479  #define UA_MANDATORY .mandatory = 1
480  #define UA_OPTIONAL .mandatory = 0
481  
482  /*
483   * min_len must be bigger than 0 and _max_len must be smaller than 4095.  Only
484   * READ\WRITE accesses are supported.
485   */
486  #define UVERBS_ATTR_IDRS_ARR(_attr_id, _idr_type, _access, _min_len, _max_len, \
487  			     ...)                                              \
488  	(&(const struct uverbs_attr_def){                                      \
489  		.id = (_attr_id) +                                             \
490  		      BUILD_BUG_ON_ZERO((_min_len) == 0 ||                     \
491  					(_max_len) >                           \
492  						PAGE_SIZE / sizeof(void *) ||  \
493  					(_min_len) > (_max_len) ||             \
494  					(_access) == UVERBS_ACCESS_NEW ||      \
495  					(_access) == UVERBS_ACCESS_DESTROY),   \
496  		.attr = { .type = UVERBS_ATTR_TYPE_IDRS_ARRAY,                 \
497  			  .u2.objs_arr.obj_type = _idr_type,                   \
498  			  .u2.objs_arr.access = _access,                       \
499  			  .u2.objs_arr.min_len = _min_len,                     \
500  			  .u2.objs_arr.max_len = _max_len,                     \
501  			  __VA_ARGS__ } })
502  
503  /*
504   * Only for use with UVERBS_ATTR_IDR, allows any uobject type to be accepted,
505   * the user must validate the type of the uobject instead.
506   */
507  #define UVERBS_IDR_ANY_OBJECT 0xFFFF
508  
509  #define UVERBS_ATTR_IDR(_attr_id, _idr_type, _access, ...)                     \
510  	(&(const struct uverbs_attr_def){                                      \
511  		.id = _attr_id,                                                \
512  		.attr = { .type = UVERBS_ATTR_TYPE_IDR,                        \
513  			  .u.obj.obj_type = _idr_type,                         \
514  			  .u.obj.access = _access,                             \
515  			  __VA_ARGS__ } })
516  
517  #define UVERBS_ATTR_FD(_attr_id, _fd_type, _access, ...)                       \
518  	(&(const struct uverbs_attr_def){                                      \
519  		.id = (_attr_id) +                                             \
520  		      BUILD_BUG_ON_ZERO((_access) != UVERBS_ACCESS_NEW &&      \
521  					(_access) != UVERBS_ACCESS_READ),      \
522  		.attr = { .type = UVERBS_ATTR_TYPE_FD,                         \
523  			  .u.obj.obj_type = _fd_type,                          \
524  			  .u.obj.access = _access,                             \
525  			  __VA_ARGS__ } })
526  
527  #define UVERBS_ATTR_RAW_FD(_attr_id, ...)                                      \
528  	(&(const struct uverbs_attr_def){                                      \
529  		.id = (_attr_id),                                              \
530  		.attr = { .type = UVERBS_ATTR_TYPE_RAW_FD, __VA_ARGS__ } })
531  
532  #define UVERBS_ATTR_PTR_IN(_attr_id, _type, ...)                               \
533  	(&(const struct uverbs_attr_def){                                      \
534  		.id = _attr_id,                                                \
535  		.attr = { .type = UVERBS_ATTR_TYPE_PTR_IN,                     \
536  			  _type,                                               \
537  			  __VA_ARGS__ } })
538  
539  #define UVERBS_ATTR_PTR_OUT(_attr_id, _type, ...)                              \
540  	(&(const struct uverbs_attr_def){                                      \
541  		.id = _attr_id,                                                \
542  		.attr = { .type = UVERBS_ATTR_TYPE_PTR_OUT,                    \
543  			  _type,                                               \
544  			  __VA_ARGS__ } })
545  
546  /* _enum_arry should be a 'static const union uverbs_attr_spec[]' */
547  #define UVERBS_ATTR_ENUM_IN(_attr_id, _enum_arr, ...)                          \
548  	(&(const struct uverbs_attr_def){                                      \
549  		.id = _attr_id,                                                \
550  		.attr = { .type = UVERBS_ATTR_TYPE_ENUM_IN,                    \
551  			  .u2.enum_def.ids = _enum_arr,                        \
552  			  .u.enum_def.num_elems = ARRAY_SIZE(_enum_arr),       \
553  			  __VA_ARGS__ },                                       \
554  	})
555  
556  /* An input value that is a member in the enum _enum_type. */
557  #define UVERBS_ATTR_CONST_IN(_attr_id, _enum_type, ...)                        \
558  	UVERBS_ATTR_PTR_IN(                                                    \
559  		_attr_id,                                                      \
560  		UVERBS_ATTR_SIZE(                                              \
561  			sizeof(u64) + BUILD_BUG_ON_ZERO(!sizeof(_enum_type)),  \
562  			sizeof(u64)),                                          \
563  		__VA_ARGS__)
564  
565  /*
566   * An input value that is a bitwise combination of values of _enum_type.
567   * This permits the flag value to be passed as either a u32 or u64, it must
568   * be retrieved via uverbs_get_flag().
569   */
570  #define UVERBS_ATTR_FLAGS_IN(_attr_id, _enum_type, ...)                        \
571  	UVERBS_ATTR_PTR_IN(                                                    \
572  		_attr_id,                                                      \
573  		UVERBS_ATTR_SIZE(sizeof(u32) + BUILD_BUG_ON_ZERO(              \
574  						       !sizeof(_enum_type *)), \
575  				 sizeof(u64)),                                 \
576  		__VA_ARGS__)
577  
578  /*
579   * This spec is used in order to pass information to the hardware driver in a
580   * legacy way. Every verb that could get driver specific data should get this
581   * spec.
582   */
583  #define UVERBS_ATTR_UHW()                                                      \
584  	UVERBS_ATTR_PTR_IN(UVERBS_ATTR_UHW_IN,                                 \
585  			   UVERBS_ATTR_MIN_SIZE(0),			       \
586  			   UA_OPTIONAL,                                        \
587  			   .is_udata = 1),				       \
588  	UVERBS_ATTR_PTR_OUT(UVERBS_ATTR_UHW_OUT,                               \
589  			    UVERBS_ATTR_MIN_SIZE(0),			       \
590  			    UA_OPTIONAL,                                       \
591  			    .is_udata = 1)
592  
593  /* =================================================
594   *              Parsing infrastructure
595   * =================================================
596   */
597  
598  
599  struct uverbs_ptr_attr {
600  	/*
601  	 * If UVERBS_ATTR_SPEC_F_ALLOC_AND_COPY is set then the 'ptr' is
602  	 * used.
603  	 */
604  	union {
605  		void *ptr;
606  		u64 data;
607  	};
608  	u16		len;
609  	u16		uattr_idx;
610  	u8		enum_id;
611  };
612  
613  struct uverbs_obj_attr {
614  	struct ib_uobject		*uobject;
615  	const struct uverbs_api_attr	*attr_elm;
616  };
617  
618  struct uverbs_objs_arr_attr {
619  	struct ib_uobject **uobjects;
620  	u16 len;
621  };
622  
623  struct uverbs_attr {
624  	union {
625  		struct uverbs_ptr_attr	ptr_attr;
626  		struct uverbs_obj_attr	obj_attr;
627  		struct uverbs_objs_arr_attr objs_arr_attr;
628  	};
629  };
630  
631  struct uverbs_attr_bundle {
632  	struct_group_tagged(uverbs_attr_bundle_hdr, hdr,
633  		struct ib_udata driver_udata;
634  		struct ib_udata ucore;
635  		struct ib_uverbs_file *ufile;
636  		struct ib_ucontext *context;
637  		struct ib_uobject *uobject;
638  		DECLARE_BITMAP(attr_present, UVERBS_API_ATTR_BKEY_LEN);
639  	);
640  	struct uverbs_attr attrs[];
641  };
642  
uverbs_attr_is_valid(const struct uverbs_attr_bundle * attrs_bundle,unsigned int idx)643  static inline bool uverbs_attr_is_valid(const struct uverbs_attr_bundle *attrs_bundle,
644  					unsigned int idx)
645  {
646  	return test_bit(uapi_bkey_attr(uapi_key_attr(idx)),
647  			attrs_bundle->attr_present);
648  }
649  
650  /**
651   * rdma_udata_to_drv_context - Helper macro to get the driver's context out of
652   *                             ib_udata which is embedded in uverbs_attr_bundle.
653   *
654   * If udata is not NULL this cannot fail. Otherwise a NULL udata will result
655   * in a NULL ucontext pointer, as a safety precaution. Callers should be using
656   * 'udata' to determine if the driver call is in user or kernel mode, not
657   * 'ucontext'.
658   *
659   */
660  static inline struct uverbs_attr_bundle *
rdma_udata_to_uverbs_attr_bundle(struct ib_udata * udata)661  rdma_udata_to_uverbs_attr_bundle(struct ib_udata *udata)
662  {
663  	return container_of(udata, struct uverbs_attr_bundle, driver_udata);
664  }
665  
666  #define rdma_udata_to_drv_context(udata, drv_dev_struct, member)                \
667  	(udata ? container_of(rdma_udata_to_uverbs_attr_bundle(udata)->context, \
668  			      drv_dev_struct, member) : (drv_dev_struct *)NULL)
669  
670  #define IS_UVERBS_COPY_ERR(_ret)		((_ret) && (_ret) != -ENOENT)
671  
uverbs_attr_get(const struct uverbs_attr_bundle * attrs_bundle,u16 idx)672  static inline const struct uverbs_attr *uverbs_attr_get(const struct uverbs_attr_bundle *attrs_bundle,
673  							u16 idx)
674  {
675  	if (!uverbs_attr_is_valid(attrs_bundle, idx))
676  		return ERR_PTR(-ENOENT);
677  
678  	return &attrs_bundle->attrs[uapi_bkey_attr(uapi_key_attr(idx))];
679  }
680  
uverbs_attr_get_enum_id(const struct uverbs_attr_bundle * attrs_bundle,u16 idx)681  static inline int uverbs_attr_get_enum_id(const struct uverbs_attr_bundle *attrs_bundle,
682  					  u16 idx)
683  {
684  	const struct uverbs_attr *attr = uverbs_attr_get(attrs_bundle, idx);
685  
686  	if (IS_ERR(attr))
687  		return PTR_ERR(attr);
688  
689  	return attr->ptr_attr.enum_id;
690  }
691  
uverbs_attr_get_obj(const struct uverbs_attr_bundle * attrs_bundle,u16 idx)692  static inline void *uverbs_attr_get_obj(const struct uverbs_attr_bundle *attrs_bundle,
693  					u16 idx)
694  {
695  	const struct uverbs_attr *attr;
696  
697  	attr = uverbs_attr_get(attrs_bundle, idx);
698  	if (IS_ERR(attr))
699  		return ERR_CAST(attr);
700  
701  	return attr->obj_attr.uobject->object;
702  }
703  
uverbs_attr_get_uobject(const struct uverbs_attr_bundle * attrs_bundle,u16 idx)704  static inline struct ib_uobject *uverbs_attr_get_uobject(const struct uverbs_attr_bundle *attrs_bundle,
705  							 u16 idx)
706  {
707  	const struct uverbs_attr *attr = uverbs_attr_get(attrs_bundle, idx);
708  
709  	if (IS_ERR(attr))
710  		return ERR_CAST(attr);
711  
712  	return attr->obj_attr.uobject;
713  }
714  
715  static inline int
uverbs_attr_get_len(const struct uverbs_attr_bundle * attrs_bundle,u16 idx)716  uverbs_attr_get_len(const struct uverbs_attr_bundle *attrs_bundle, u16 idx)
717  {
718  	const struct uverbs_attr *attr = uverbs_attr_get(attrs_bundle, idx);
719  
720  	if (IS_ERR(attr))
721  		return PTR_ERR(attr);
722  
723  	return attr->ptr_attr.len;
724  }
725  
726  void uverbs_finalize_uobj_create(const struct uverbs_attr_bundle *attrs_bundle,
727  				 u16 idx);
728  
729  /*
730   * uverbs_attr_ptr_get_array_size() - Get array size pointer by a ptr
731   * attribute.
732   * @attrs: The attribute bundle
733   * @idx: The ID of the attribute
734   * @elem_size: The size of the element in the array
735   */
736  static inline int
uverbs_attr_ptr_get_array_size(struct uverbs_attr_bundle * attrs,u16 idx,size_t elem_size)737  uverbs_attr_ptr_get_array_size(struct uverbs_attr_bundle *attrs, u16 idx,
738  			       size_t elem_size)
739  {
740  	int size = uverbs_attr_get_len(attrs, idx);
741  
742  	if (size < 0)
743  		return size;
744  
745  	if (size % elem_size)
746  		return -EINVAL;
747  
748  	return size / elem_size;
749  }
750  
751  /**
752   * uverbs_attr_get_uobjs_arr() - Provides array's properties for attribute for
753   * UVERBS_ATTR_TYPE_IDRS_ARRAY.
754   * @arr: Returned pointer to array of pointers for uobjects or NULL if
755   *       the attribute isn't provided.
756   *
757   * Return: The array length or 0 if no attribute was provided.
758   */
uverbs_attr_get_uobjs_arr(const struct uverbs_attr_bundle * attrs_bundle,u16 attr_idx,struct ib_uobject *** arr)759  static inline int uverbs_attr_get_uobjs_arr(
760  	const struct uverbs_attr_bundle *attrs_bundle, u16 attr_idx,
761  	struct ib_uobject ***arr)
762  {
763  	const struct uverbs_attr *attr =
764  			uverbs_attr_get(attrs_bundle, attr_idx);
765  
766  	if (IS_ERR(attr)) {
767  		*arr = NULL;
768  		return 0;
769  	}
770  
771  	*arr = attr->objs_arr_attr.uobjects;
772  
773  	return attr->objs_arr_attr.len;
774  }
775  
uverbs_attr_ptr_is_inline(const struct uverbs_attr * attr)776  static inline bool uverbs_attr_ptr_is_inline(const struct uverbs_attr *attr)
777  {
778  	return attr->ptr_attr.len <= sizeof(attr->ptr_attr.data);
779  }
780  
uverbs_attr_get_alloced_ptr(const struct uverbs_attr_bundle * attrs_bundle,u16 idx)781  static inline void *uverbs_attr_get_alloced_ptr(
782  	const struct uverbs_attr_bundle *attrs_bundle, u16 idx)
783  {
784  	const struct uverbs_attr *attr = uverbs_attr_get(attrs_bundle, idx);
785  
786  	if (IS_ERR(attr))
787  		return (void *)attr;
788  
789  	return uverbs_attr_ptr_is_inline(attr) ? (void *)&attr->ptr_attr.data :
790  						 attr->ptr_attr.ptr;
791  }
792  
_uverbs_copy_from(void * to,const struct uverbs_attr_bundle * attrs_bundle,size_t idx,size_t size)793  static inline int _uverbs_copy_from(void *to,
794  				    const struct uverbs_attr_bundle *attrs_bundle,
795  				    size_t idx,
796  				    size_t size)
797  {
798  	const struct uverbs_attr *attr = uverbs_attr_get(attrs_bundle, idx);
799  
800  	if (IS_ERR(attr))
801  		return PTR_ERR(attr);
802  
803  	/*
804  	 * Validation ensures attr->ptr_attr.len >= size. If the caller is
805  	 * using UVERBS_ATTR_SPEC_F_MIN_SZ_OR_ZERO then it must call
806  	 * uverbs_copy_from_or_zero.
807  	 */
808  	if (unlikely(size < attr->ptr_attr.len))
809  		return -EINVAL;
810  
811  	if (uverbs_attr_ptr_is_inline(attr))
812  		memcpy(to, &attr->ptr_attr.data, attr->ptr_attr.len);
813  	else if (copy_from_user(to, u64_to_user_ptr(attr->ptr_attr.data),
814  				attr->ptr_attr.len))
815  		return -EFAULT;
816  
817  	return 0;
818  }
819  
_uverbs_copy_from_or_zero(void * to,const struct uverbs_attr_bundle * attrs_bundle,size_t idx,size_t size)820  static inline int _uverbs_copy_from_or_zero(void *to,
821  					    const struct uverbs_attr_bundle *attrs_bundle,
822  					    size_t idx,
823  					    size_t size)
824  {
825  	const struct uverbs_attr *attr = uverbs_attr_get(attrs_bundle, idx);
826  	size_t min_size;
827  
828  	if (IS_ERR(attr))
829  		return PTR_ERR(attr);
830  
831  	min_size = min_t(size_t, size, attr->ptr_attr.len);
832  
833  	if (uverbs_attr_ptr_is_inline(attr))
834  		memcpy(to, &attr->ptr_attr.data, min_size);
835  	else if (copy_from_user(to, u64_to_user_ptr(attr->ptr_attr.data),
836  				min_size))
837  		return -EFAULT;
838  
839  	if (size > min_size)
840  		memset(to + min_size, 0, size - min_size);
841  
842  	return 0;
843  }
844  
845  #define uverbs_copy_from(to, attrs_bundle, idx)				      \
846  	_uverbs_copy_from(to, attrs_bundle, idx, sizeof(*to))
847  
848  #define uverbs_copy_from_or_zero(to, attrs_bundle, idx)			      \
849  	_uverbs_copy_from_or_zero(to, attrs_bundle, idx, sizeof(*to))
850  
851  static inline struct ib_ucontext *
ib_uverbs_get_ucontext(const struct uverbs_attr_bundle * attrs)852  ib_uverbs_get_ucontext(const struct uverbs_attr_bundle *attrs)
853  {
854  	return ib_uverbs_get_ucontext_file(attrs->ufile);
855  }
856  
857  #if IS_ENABLED(CONFIG_INFINIBAND_USER_ACCESS)
858  int uverbs_get_flags64(u64 *to, const struct uverbs_attr_bundle *attrs_bundle,
859  		       size_t idx, u64 allowed_bits);
860  int uverbs_get_flags32(u32 *to, const struct uverbs_attr_bundle *attrs_bundle,
861  		       size_t idx, u64 allowed_bits);
862  int uverbs_copy_to(const struct uverbs_attr_bundle *attrs_bundle, size_t idx,
863  		   const void *from, size_t size);
864  __malloc void *_uverbs_alloc(struct uverbs_attr_bundle *bundle, size_t size,
865  			     gfp_t flags);
866  
uverbs_alloc(struct uverbs_attr_bundle * bundle,size_t size)867  static inline __malloc void *uverbs_alloc(struct uverbs_attr_bundle *bundle,
868  					  size_t size)
869  {
870  	return _uverbs_alloc(bundle, size, GFP_KERNEL);
871  }
872  
uverbs_zalloc(struct uverbs_attr_bundle * bundle,size_t size)873  static inline __malloc void *uverbs_zalloc(struct uverbs_attr_bundle *bundle,
874  					   size_t size)
875  {
876  	return _uverbs_alloc(bundle, size, GFP_KERNEL | __GFP_ZERO);
877  }
878  
uverbs_kcalloc(struct uverbs_attr_bundle * bundle,size_t n,size_t size)879  static inline __malloc void *uverbs_kcalloc(struct uverbs_attr_bundle *bundle,
880  					    size_t n, size_t size)
881  {
882  	size_t bytes;
883  
884  	if (unlikely(check_mul_overflow(n, size, &bytes)))
885  		return ERR_PTR(-EOVERFLOW);
886  	return uverbs_zalloc(bundle, bytes);
887  }
888  
889  int _uverbs_get_const_signed(s64 *to,
890  			     const struct uverbs_attr_bundle *attrs_bundle,
891  			     size_t idx, s64 lower_bound, u64 upper_bound,
892  			     s64 *def_val);
893  int _uverbs_get_const_unsigned(u64 *to,
894  			       const struct uverbs_attr_bundle *attrs_bundle,
895  			       size_t idx, u64 upper_bound, u64 *def_val);
896  int uverbs_copy_to_struct_or_zero(const struct uverbs_attr_bundle *bundle,
897  				  size_t idx, const void *from, size_t size);
898  #else
899  static inline int
uverbs_get_flags64(u64 * to,const struct uverbs_attr_bundle * attrs_bundle,size_t idx,u64 allowed_bits)900  uverbs_get_flags64(u64 *to, const struct uverbs_attr_bundle *attrs_bundle,
901  		   size_t idx, u64 allowed_bits)
902  {
903  	return -EINVAL;
904  }
905  static inline int
uverbs_get_flags32(u32 * to,const struct uverbs_attr_bundle * attrs_bundle,size_t idx,u64 allowed_bits)906  uverbs_get_flags32(u32 *to, const struct uverbs_attr_bundle *attrs_bundle,
907  		   size_t idx, u64 allowed_bits)
908  {
909  	return -EINVAL;
910  }
uverbs_copy_to(const struct uverbs_attr_bundle * attrs_bundle,size_t idx,const void * from,size_t size)911  static inline int uverbs_copy_to(const struct uverbs_attr_bundle *attrs_bundle,
912  				 size_t idx, const void *from, size_t size)
913  {
914  	return -EINVAL;
915  }
uverbs_alloc(struct uverbs_attr_bundle * bundle,size_t size)916  static inline __malloc void *uverbs_alloc(struct uverbs_attr_bundle *bundle,
917  					  size_t size)
918  {
919  	return ERR_PTR(-EINVAL);
920  }
uverbs_zalloc(struct uverbs_attr_bundle * bundle,size_t size)921  static inline __malloc void *uverbs_zalloc(struct uverbs_attr_bundle *bundle,
922  					   size_t size)
923  {
924  	return ERR_PTR(-EINVAL);
925  }
926  static inline int
_uverbs_get_const(s64 * to,const struct uverbs_attr_bundle * attrs_bundle,size_t idx,s64 lower_bound,u64 upper_bound,s64 * def_val)927  _uverbs_get_const(s64 *to, const struct uverbs_attr_bundle *attrs_bundle,
928  		  size_t idx, s64 lower_bound, u64 upper_bound,
929  		  s64 *def_val)
930  {
931  	return -EINVAL;
932  }
933  static inline int
uverbs_copy_to_struct_or_zero(const struct uverbs_attr_bundle * bundle,size_t idx,const void * from,size_t size)934  uverbs_copy_to_struct_or_zero(const struct uverbs_attr_bundle *bundle,
935  			      size_t idx, const void *from, size_t size)
936  {
937  	return -EINVAL;
938  }
939  static inline int
_uverbs_get_const_signed(s64 * to,const struct uverbs_attr_bundle * attrs_bundle,size_t idx,s64 lower_bound,u64 upper_bound,s64 * def_val)940  _uverbs_get_const_signed(s64 *to,
941  			 const struct uverbs_attr_bundle *attrs_bundle,
942  			 size_t idx, s64 lower_bound, u64 upper_bound,
943  			 s64 *def_val)
944  {
945  	return -EINVAL;
946  }
947  static inline int
_uverbs_get_const_unsigned(u64 * to,const struct uverbs_attr_bundle * attrs_bundle,size_t idx,u64 upper_bound,u64 * def_val)948  _uverbs_get_const_unsigned(u64 *to,
949  			   const struct uverbs_attr_bundle *attrs_bundle,
950  			   size_t idx, u64 upper_bound, u64 *def_val)
951  {
952  	return -EINVAL;
953  }
954  #endif
955  
956  #define uverbs_get_const_signed(_to, _attrs_bundle, _idx)                      \
957  	({                                                                     \
958  		s64 _val;                                                      \
959  		int _ret =                                                     \
960  			_uverbs_get_const_signed(&_val, _attrs_bundle, _idx,   \
961  					  type_min(typeof(*(_to))),            \
962  					  type_max(typeof(*(_to))), NULL);     \
963  		(*(_to)) = _val;                                               \
964  		_ret;                                                          \
965  	})
966  
967  #define uverbs_get_const_unsigned(_to, _attrs_bundle, _idx)                    \
968  	({                                                                     \
969  		u64 _val;                                                      \
970  		int _ret =                                                     \
971  			_uverbs_get_const_unsigned(&_val, _attrs_bundle, _idx, \
972  					  type_max(typeof(*(_to))), NULL);     \
973  		(*(_to)) = _val;                                               \
974  		_ret;                                                          \
975  	})
976  
977  #define uverbs_get_const_default_signed(_to, _attrs_bundle, _idx, _default)    \
978  	({                                                                     \
979  		s64 _val;                                                      \
980  		s64 _def_val = _default;                                       \
981  		int _ret =                                                     \
982  			_uverbs_get_const_signed(&_val, _attrs_bundle, _idx,   \
983  				type_min(typeof(*(_to))),                      \
984  				type_max(typeof(*(_to))), &_def_val);          \
985  		(*(_to)) = _val;                                               \
986  		_ret;                                                          \
987  	})
988  
989  #define uverbs_get_const_default_unsigned(_to, _attrs_bundle, _idx, _default)  \
990  	({                                                                     \
991  		u64 _val;                                                      \
992  		u64 _def_val = _default;                                       \
993  		int _ret =                                                     \
994  			_uverbs_get_const_unsigned(&_val, _attrs_bundle, _idx, \
995  				type_max(typeof(*(_to))), &_def_val);          \
996  		(*(_to)) = _val;                                               \
997  		_ret;                                                          \
998  	})
999  
1000  #define uverbs_get_const(_to, _attrs_bundle, _idx)                             \
1001  	(is_signed_type(typeof(*(_to))) ?                                      \
1002  		 uverbs_get_const_signed(_to, _attrs_bundle, _idx) :           \
1003  		 uverbs_get_const_unsigned(_to, _attrs_bundle, _idx))          \
1004  
1005  #define uverbs_get_const_default(_to, _attrs_bundle, _idx, _default)           \
1006  	(is_signed_type(typeof(*(_to))) ?                                      \
1007  		 uverbs_get_const_default_signed(_to, _attrs_bundle, _idx,     \
1008  						  _default) :                  \
1009  		 uverbs_get_const_default_unsigned(_to, _attrs_bundle, _idx,   \
1010  						    _default))
1011  
1012  static inline int
uverbs_get_raw_fd(int * to,const struct uverbs_attr_bundle * attrs_bundle,size_t idx)1013  uverbs_get_raw_fd(int *to, const struct uverbs_attr_bundle *attrs_bundle,
1014  		  size_t idx)
1015  {
1016  	return uverbs_get_const_signed(to, attrs_bundle, idx);
1017  }
1018  
1019  #endif
1020