Merge tag 'dma-mapping-5.13' of git://git.infradead.org/users/hch/dma-mapping
[platform/kernel/linux-rpi.git] / drivers / thunderbolt / property.c
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Thunderbolt XDomain property support
4  *
5  * Copyright (C) 2017, Intel Corporation
6  * Authors: Michael Jamet <michael.jamet@intel.com>
7  *          Mika Westerberg <mika.westerberg@linux.intel.com>
8  */
9
10 #include <linux/err.h>
11 #include <linux/slab.h>
12 #include <linux/string.h>
13 #include <linux/uuid.h>
14 #include <linux/thunderbolt.h>
15
16 struct tb_property_entry {
17         u32 key_hi;
18         u32 key_lo;
19         u16 length;
20         u8 reserved;
21         u8 type;
22         u32 value;
23 };
24
25 struct tb_property_rootdir_entry {
26         u32 magic;
27         u32 length;
28         struct tb_property_entry entries[];
29 };
30
31 struct tb_property_dir_entry {
32         u32 uuid[4];
33         struct tb_property_entry entries[];
34 };
35
36 #define TB_PROPERTY_ROOTDIR_MAGIC       0x55584401
37
38 static struct tb_property_dir *__tb_property_parse_dir(const u32 *block,
39         size_t block_len, unsigned int dir_offset, size_t dir_len,
40         bool is_root);
41
42 static inline void parse_dwdata(void *dst, const void *src, size_t dwords)
43 {
44         be32_to_cpu_array(dst, src, dwords);
45 }
46
47 static inline void format_dwdata(void *dst, const void *src, size_t dwords)
48 {
49         cpu_to_be32_array(dst, src, dwords);
50 }
51
52 static bool tb_property_entry_valid(const struct tb_property_entry *entry,
53                                   size_t block_len)
54 {
55         switch (entry->type) {
56         case TB_PROPERTY_TYPE_DIRECTORY:
57         case TB_PROPERTY_TYPE_DATA:
58         case TB_PROPERTY_TYPE_TEXT:
59                 if (entry->length > block_len)
60                         return false;
61                 if (entry->value + entry->length > block_len)
62                         return false;
63                 break;
64
65         case TB_PROPERTY_TYPE_VALUE:
66                 if (entry->length != 1)
67                         return false;
68                 break;
69         }
70
71         return true;
72 }
73
74 static bool tb_property_key_valid(const char *key)
75 {
76         return key && strlen(key) <= TB_PROPERTY_KEY_SIZE;
77 }
78
79 static struct tb_property *
80 tb_property_alloc(const char *key, enum tb_property_type type)
81 {
82         struct tb_property *property;
83
84         property = kzalloc(sizeof(*property), GFP_KERNEL);
85         if (!property)
86                 return NULL;
87
88         strcpy(property->key, key);
89         property->type = type;
90         INIT_LIST_HEAD(&property->list);
91
92         return property;
93 }
94
95 static struct tb_property *tb_property_parse(const u32 *block, size_t block_len,
96                                         const struct tb_property_entry *entry)
97 {
98         char key[TB_PROPERTY_KEY_SIZE + 1];
99         struct tb_property *property;
100         struct tb_property_dir *dir;
101
102         if (!tb_property_entry_valid(entry, block_len))
103                 return NULL;
104
105         parse_dwdata(key, entry, 2);
106         key[TB_PROPERTY_KEY_SIZE] = '\0';
107
108         property = tb_property_alloc(key, entry->type);
109         if (!property)
110                 return NULL;
111
112         property->length = entry->length;
113
114         switch (property->type) {
115         case TB_PROPERTY_TYPE_DIRECTORY:
116                 dir = __tb_property_parse_dir(block, block_len, entry->value,
117                                               entry->length, false);
118                 if (!dir) {
119                         kfree(property);
120                         return NULL;
121                 }
122                 property->value.dir = dir;
123                 break;
124
125         case TB_PROPERTY_TYPE_DATA:
126                 property->value.data = kcalloc(property->length, sizeof(u32),
127                                                GFP_KERNEL);
128                 if (!property->value.data) {
129                         kfree(property);
130                         return NULL;
131                 }
132                 parse_dwdata(property->value.data, block + entry->value,
133                              entry->length);
134                 break;
135
136         case TB_PROPERTY_TYPE_TEXT:
137                 property->value.text = kcalloc(property->length, sizeof(u32),
138                                                GFP_KERNEL);
139                 if (!property->value.text) {
140                         kfree(property);
141                         return NULL;
142                 }
143                 parse_dwdata(property->value.text, block + entry->value,
144                              entry->length);
145                 /* Force null termination */
146                 property->value.text[property->length * 4 - 1] = '\0';
147                 break;
148
149         case TB_PROPERTY_TYPE_VALUE:
150                 property->value.immediate = entry->value;
151                 break;
152
153         default:
154                 property->type = TB_PROPERTY_TYPE_UNKNOWN;
155                 break;
156         }
157
158         return property;
159 }
160
161 static struct tb_property_dir *__tb_property_parse_dir(const u32 *block,
162         size_t block_len, unsigned int dir_offset, size_t dir_len, bool is_root)
163 {
164         const struct tb_property_entry *entries;
165         size_t i, content_len, nentries;
166         unsigned int content_offset;
167         struct tb_property_dir *dir;
168
169         dir = kzalloc(sizeof(*dir), GFP_KERNEL);
170         if (!dir)
171                 return NULL;
172
173         if (is_root) {
174                 content_offset = dir_offset + 2;
175                 content_len = dir_len;
176         } else {
177                 dir->uuid = kmemdup(&block[dir_offset], sizeof(*dir->uuid),
178                                     GFP_KERNEL);
179                 if (!dir->uuid) {
180                         tb_property_free_dir(dir);
181                         return NULL;
182                 }
183                 content_offset = dir_offset + 4;
184                 content_len = dir_len - 4; /* Length includes UUID */
185         }
186
187         entries = (const struct tb_property_entry *)&block[content_offset];
188         nentries = content_len / (sizeof(*entries) / 4);
189
190         INIT_LIST_HEAD(&dir->properties);
191
192         for (i = 0; i < nentries; i++) {
193                 struct tb_property *property;
194
195                 property = tb_property_parse(block, block_len, &entries[i]);
196                 if (!property) {
197                         tb_property_free_dir(dir);
198                         return NULL;
199                 }
200
201                 list_add_tail(&property->list, &dir->properties);
202         }
203
204         return dir;
205 }
206
207 /**
208  * tb_property_parse_dir() - Parses properties from given property block
209  * @block: Property block to parse
210  * @block_len: Number of dword elements in the property block
211  *
212  * This function parses the XDomain properties data block into format that
213  * can be traversed using the helper functions provided by this module.
214  * Upon success returns the parsed directory. In case of error returns
215  * %NULL. The resulting &struct tb_property_dir needs to be released by
216  * calling tb_property_free_dir() when not needed anymore.
217  *
218  * The @block is expected to be root directory.
219  */
220 struct tb_property_dir *tb_property_parse_dir(const u32 *block,
221                                               size_t block_len)
222 {
223         const struct tb_property_rootdir_entry *rootdir =
224                 (const struct tb_property_rootdir_entry *)block;
225
226         if (rootdir->magic != TB_PROPERTY_ROOTDIR_MAGIC)
227                 return NULL;
228         if (rootdir->length > block_len)
229                 return NULL;
230
231         return __tb_property_parse_dir(block, block_len, 0, rootdir->length,
232                                        true);
233 }
234
235 /**
236  * tb_property_create_dir() - Creates new property directory
237  * @uuid: UUID used to identify the particular directory
238  *
239  * Creates new, empty property directory. If @uuid is %NULL then the
240  * directory is assumed to be root directory.
241  */
242 struct tb_property_dir *tb_property_create_dir(const uuid_t *uuid)
243 {
244         struct tb_property_dir *dir;
245
246         dir = kzalloc(sizeof(*dir), GFP_KERNEL);
247         if (!dir)
248                 return NULL;
249
250         INIT_LIST_HEAD(&dir->properties);
251         if (uuid) {
252                 dir->uuid = kmemdup(uuid, sizeof(*dir->uuid), GFP_KERNEL);
253                 if (!dir->uuid) {
254                         kfree(dir);
255                         return NULL;
256                 }
257         }
258
259         return dir;
260 }
261 EXPORT_SYMBOL_GPL(tb_property_create_dir);
262
263 static void tb_property_free(struct tb_property *property)
264 {
265         switch (property->type) {
266         case TB_PROPERTY_TYPE_DIRECTORY:
267                 tb_property_free_dir(property->value.dir);
268                 break;
269
270         case TB_PROPERTY_TYPE_DATA:
271                 kfree(property->value.data);
272                 break;
273
274         case TB_PROPERTY_TYPE_TEXT:
275                 kfree(property->value.text);
276                 break;
277
278         default:
279                 break;
280         }
281
282         kfree(property);
283 }
284
285 /**
286  * tb_property_free_dir() - Release memory allocated for property directory
287  * @dir: Directory to release
288  *
289  * This will release all the memory the directory occupies including all
290  * descendants. It is OK to pass %NULL @dir, then the function does
291  * nothing.
292  */
293 void tb_property_free_dir(struct tb_property_dir *dir)
294 {
295         struct tb_property *property, *tmp;
296
297         if (!dir)
298                 return;
299
300         list_for_each_entry_safe(property, tmp, &dir->properties, list) {
301                 list_del(&property->list);
302                 tb_property_free(property);
303         }
304         kfree(dir->uuid);
305         kfree(dir);
306 }
307 EXPORT_SYMBOL_GPL(tb_property_free_dir);
308
309 static size_t tb_property_dir_length(const struct tb_property_dir *dir,
310                                      bool recurse, size_t *data_len)
311 {
312         const struct tb_property *property;
313         size_t len = 0;
314
315         if (dir->uuid)
316                 len += sizeof(*dir->uuid) / 4;
317         else
318                 len += sizeof(struct tb_property_rootdir_entry) / 4;
319
320         list_for_each_entry(property, &dir->properties, list) {
321                 len += sizeof(struct tb_property_entry) / 4;
322
323                 switch (property->type) {
324                 case TB_PROPERTY_TYPE_DIRECTORY:
325                         if (recurse) {
326                                 len += tb_property_dir_length(
327                                         property->value.dir, recurse, data_len);
328                         }
329                         /* Reserve dword padding after each directory */
330                         if (data_len)
331                                 *data_len += 1;
332                         break;
333
334                 case TB_PROPERTY_TYPE_DATA:
335                 case TB_PROPERTY_TYPE_TEXT:
336                         if (data_len)
337                                 *data_len += property->length;
338                         break;
339
340                 default:
341                         break;
342                 }
343         }
344
345         return len;
346 }
347
348 static ssize_t __tb_property_format_dir(const struct tb_property_dir *dir,
349         u32 *block, unsigned int start_offset, size_t block_len)
350 {
351         unsigned int data_offset, dir_end;
352         const struct tb_property *property;
353         struct tb_property_entry *entry;
354         size_t dir_len, data_len = 0;
355         int ret;
356
357         /*
358          * The structure of property block looks like following. Leaf
359          * data/text is included right after the directory and each
360          * directory follows each other (even nested ones).
361          *
362          * +----------+ <-- start_offset
363          * |  header  | <-- root directory header
364          * +----------+ ---
365          * |  entry 0 | -^--------------------.
366          * +----------+  |                    |
367          * |  entry 1 | -|--------------------|--.
368          * +----------+  |                    |  |
369          * |  entry 2 | -|-----------------.  |  |
370          * +----------+  |                 |  |  |
371          * :          :  |  dir_len        |  |  |
372          * .          .  |                 |  |  |
373          * :          :  |                 |  |  |
374          * +----------+  |                 |  |  |
375          * |  entry n |  v                 |  |  |
376          * +----------+ <-- data_offset    |  |  |
377          * |  data 0  | <------------------|--'  |
378          * +----------+                    |     |
379          * |  data 1  | <------------------|-----'
380          * +----------+                    |
381          * | 00000000 | padding            |
382          * +----------+ <-- dir_end <------'
383          * |   UUID   | <-- directory UUID (child directory)
384          * +----------+
385          * |  entry 0 |
386          * +----------+
387          * |  entry 1 |
388          * +----------+
389          * :          :
390          * .          .
391          * :          :
392          * +----------+
393          * |  entry n |
394          * +----------+
395          * |  data 0  |
396          * +----------+
397          *
398          * We use dir_end to hold pointer to the end of the directory. It
399          * will increase as we add directories and each directory should be
400          * added starting from previous dir_end.
401          */
402         dir_len = tb_property_dir_length(dir, false, &data_len);
403         data_offset = start_offset + dir_len;
404         dir_end = start_offset + data_len + dir_len;
405
406         if (data_offset > dir_end)
407                 return -EINVAL;
408         if (dir_end > block_len)
409                 return -EINVAL;
410
411         /* Write headers first */
412         if (dir->uuid) {
413                 struct tb_property_dir_entry *pe;
414
415                 pe = (struct tb_property_dir_entry *)&block[start_offset];
416                 memcpy(pe->uuid, dir->uuid, sizeof(pe->uuid));
417                 entry = pe->entries;
418         } else {
419                 struct tb_property_rootdir_entry *re;
420
421                 re = (struct tb_property_rootdir_entry *)&block[start_offset];
422                 re->magic = TB_PROPERTY_ROOTDIR_MAGIC;
423                 re->length = dir_len - sizeof(*re) / 4;
424                 entry = re->entries;
425         }
426
427         list_for_each_entry(property, &dir->properties, list) {
428                 const struct tb_property_dir *child;
429
430                 format_dwdata(entry, property->key, 2);
431                 entry->type = property->type;
432
433                 switch (property->type) {
434                 case TB_PROPERTY_TYPE_DIRECTORY:
435                         child = property->value.dir;
436                         ret = __tb_property_format_dir(child, block, dir_end,
437                                                        block_len);
438                         if (ret < 0)
439                                 return ret;
440                         entry->length = tb_property_dir_length(child, false,
441                                                                NULL);
442                         entry->value = dir_end;
443                         dir_end = ret;
444                         break;
445
446                 case TB_PROPERTY_TYPE_DATA:
447                         format_dwdata(&block[data_offset], property->value.data,
448                                       property->length);
449                         entry->length = property->length;
450                         entry->value = data_offset;
451                         data_offset += entry->length;
452                         break;
453
454                 case TB_PROPERTY_TYPE_TEXT:
455                         format_dwdata(&block[data_offset], property->value.text,
456                                       property->length);
457                         entry->length = property->length;
458                         entry->value = data_offset;
459                         data_offset += entry->length;
460                         break;
461
462                 case TB_PROPERTY_TYPE_VALUE:
463                         entry->length = property->length;
464                         entry->value = property->value.immediate;
465                         break;
466
467                 default:
468                         break;
469                 }
470
471                 entry++;
472         }
473
474         return dir_end;
475 }
476
477 /**
478  * tb_property_format_dir() - Formats directory to the packed XDomain format
479  * @dir: Directory to format
480  * @block: Property block where the packed data is placed
481  * @block_len: Length of the property block
482  *
483  * This function formats the directory to the packed format that can be
484  * then send over the thunderbolt fabric to receiving host. Returns %0 in
485  * case of success and negative errno on faulure. Passing %NULL in @block
486  * returns number of entries the block takes.
487  */
488 ssize_t tb_property_format_dir(const struct tb_property_dir *dir, u32 *block,
489                                size_t block_len)
490 {
491         ssize_t ret;
492
493         if (!block) {
494                 size_t dir_len, data_len = 0;
495
496                 dir_len = tb_property_dir_length(dir, true, &data_len);
497                 return dir_len + data_len;
498         }
499
500         ret = __tb_property_format_dir(dir, block, 0, block_len);
501         return ret < 0 ? ret : 0;
502 }
503
504 /**
505  * tb_property_copy_dir() - Take a deep copy of directory
506  * @dir: Directory to copy
507  *
508  * This function takes a deep copy of @dir and returns back the copy. In
509  * case of error returns %NULL. The resulting directory needs to be
510  * released by calling tb_property_free_dir().
511  */
512 struct tb_property_dir *tb_property_copy_dir(const struct tb_property_dir *dir)
513 {
514         struct tb_property *property, *p = NULL;
515         struct tb_property_dir *d;
516
517         if (!dir)
518                 return NULL;
519
520         d = tb_property_create_dir(dir->uuid);
521         if (!d)
522                 return NULL;
523
524         list_for_each_entry(property, &dir->properties, list) {
525                 struct tb_property *p;
526
527                 p = tb_property_alloc(property->key, property->type);
528                 if (!p)
529                         goto err_free;
530
531                 p->length = property->length;
532
533                 switch (property->type) {
534                 case TB_PROPERTY_TYPE_DIRECTORY:
535                         p->value.dir = tb_property_copy_dir(property->value.dir);
536                         if (!p->value.dir)
537                                 goto err_free;
538                         break;
539
540                 case TB_PROPERTY_TYPE_DATA:
541                         p->value.data = kmemdup(property->value.data,
542                                                 property->length * 4,
543                                                 GFP_KERNEL);
544                         if (!p->value.data)
545                                 goto err_free;
546                         break;
547
548                 case TB_PROPERTY_TYPE_TEXT:
549                         p->value.text = kzalloc(p->length * 4, GFP_KERNEL);
550                         if (!p->value.text)
551                                 goto err_free;
552                         strcpy(p->value.text, property->value.text);
553                         break;
554
555                 case TB_PROPERTY_TYPE_VALUE:
556                         p->value.immediate = property->value.immediate;
557                         break;
558
559                 default:
560                         break;
561                 }
562
563                 list_add_tail(&p->list, &d->properties);
564         }
565
566         return d;
567
568 err_free:
569         kfree(p);
570         tb_property_free_dir(d);
571
572         return NULL;
573 }
574
575 /**
576  * tb_property_add_immediate() - Add immediate property to directory
577  * @parent: Directory to add the property
578  * @key: Key for the property
579  * @value: Immediate value to store with the property
580  */
581 int tb_property_add_immediate(struct tb_property_dir *parent, const char *key,
582                               u32 value)
583 {
584         struct tb_property *property;
585
586         if (!tb_property_key_valid(key))
587                 return -EINVAL;
588
589         property = tb_property_alloc(key, TB_PROPERTY_TYPE_VALUE);
590         if (!property)
591                 return -ENOMEM;
592
593         property->length = 1;
594         property->value.immediate = value;
595
596         list_add_tail(&property->list, &parent->properties);
597         return 0;
598 }
599 EXPORT_SYMBOL_GPL(tb_property_add_immediate);
600
601 /**
602  * tb_property_add_data() - Adds arbitrary data property to directory
603  * @parent: Directory to add the property
604  * @key: Key for the property
605  * @buf: Data buffer to add
606  * @buflen: Number of bytes in the data buffer
607  *
608  * Function takes a copy of @buf and adds it to the directory.
609  */
610 int tb_property_add_data(struct tb_property_dir *parent, const char *key,
611                          const void *buf, size_t buflen)
612 {
613         /* Need to pad to dword boundary */
614         size_t size = round_up(buflen, 4);
615         struct tb_property *property;
616
617         if (!tb_property_key_valid(key))
618                 return -EINVAL;
619
620         property = tb_property_alloc(key, TB_PROPERTY_TYPE_DATA);
621         if (!property)
622                 return -ENOMEM;
623
624         property->length = size / 4;
625         property->value.data = kzalloc(size, GFP_KERNEL);
626         if (!property->value.data) {
627                 kfree(property);
628                 return -ENOMEM;
629         }
630
631         memcpy(property->value.data, buf, buflen);
632
633         list_add_tail(&property->list, &parent->properties);
634         return 0;
635 }
636 EXPORT_SYMBOL_GPL(tb_property_add_data);
637
638 /**
639  * tb_property_add_text() - Adds string property to directory
640  * @parent: Directory to add the property
641  * @key: Key for the property
642  * @text: String to add
643  *
644  * Function takes a copy of @text and adds it to the directory.
645  */
646 int tb_property_add_text(struct tb_property_dir *parent, const char *key,
647                          const char *text)
648 {
649         /* Need to pad to dword boundary */
650         size_t size = round_up(strlen(text) + 1, 4);
651         struct tb_property *property;
652
653         if (!tb_property_key_valid(key))
654                 return -EINVAL;
655
656         property = tb_property_alloc(key, TB_PROPERTY_TYPE_TEXT);
657         if (!property)
658                 return -ENOMEM;
659
660         property->length = size / 4;
661         property->value.text = kzalloc(size, GFP_KERNEL);
662         if (!property->value.text) {
663                 kfree(property);
664                 return -ENOMEM;
665         }
666
667         strcpy(property->value.text, text);
668
669         list_add_tail(&property->list, &parent->properties);
670         return 0;
671 }
672 EXPORT_SYMBOL_GPL(tb_property_add_text);
673
674 /**
675  * tb_property_add_dir() - Adds a directory to the parent directory
676  * @parent: Directory to add the property
677  * @key: Key for the property
678  * @dir: Directory to add
679  */
680 int tb_property_add_dir(struct tb_property_dir *parent, const char *key,
681                         struct tb_property_dir *dir)
682 {
683         struct tb_property *property;
684
685         if (!tb_property_key_valid(key))
686                 return -EINVAL;
687
688         property = tb_property_alloc(key, TB_PROPERTY_TYPE_DIRECTORY);
689         if (!property)
690                 return -ENOMEM;
691
692         property->value.dir = dir;
693
694         list_add_tail(&property->list, &parent->properties);
695         return 0;
696 }
697 EXPORT_SYMBOL_GPL(tb_property_add_dir);
698
699 /**
700  * tb_property_remove() - Removes property from a parent directory
701  * @property: Property to remove
702  *
703  * Note memory for @property is released as well so it is not allowed to
704  * touch the object after call to this function.
705  */
706 void tb_property_remove(struct tb_property *property)
707 {
708         list_del(&property->list);
709         kfree(property);
710 }
711 EXPORT_SYMBOL_GPL(tb_property_remove);
712
713 /**
714  * tb_property_find() - Find a property from a directory
715  * @dir: Directory where the property is searched
716  * @key: Key to look for
717  * @type: Type of the property
718  *
719  * Finds and returns property from the given directory. Does not recurse
720  * into sub-directories. Returns %NULL if the property was not found.
721  */
722 struct tb_property *tb_property_find(struct tb_property_dir *dir,
723         const char *key, enum tb_property_type type)
724 {
725         struct tb_property *property;
726
727         list_for_each_entry(property, &dir->properties, list) {
728                 if (property->type == type && !strcmp(property->key, key))
729                         return property;
730         }
731
732         return NULL;
733 }
734 EXPORT_SYMBOL_GPL(tb_property_find);
735
736 /**
737  * tb_property_get_next() - Get next property from directory
738  * @dir: Directory holding properties
739  * @prev: Previous property in the directory (%NULL returns the first)
740  */
741 struct tb_property *tb_property_get_next(struct tb_property_dir *dir,
742                                          struct tb_property *prev)
743 {
744         if (prev) {
745                 if (list_is_last(&prev->list, &dir->properties))
746                         return NULL;
747                 return list_next_entry(prev, list);
748         }
749         return list_first_entry_or_null(&dir->properties, struct tb_property,
750                                         list);
751 }
752 EXPORT_SYMBOL_GPL(tb_property_get_next);