Line data    Source code

       1             : // SPDX-License-Identifier: GPL-2.0
       2             : /*
       3             :  * Range add and subtract
       4             :  */
       5             : #include <linux/init.h>
       6             : #include <linux/minmax.h>
       7             : #include <linux/printk.h>
       8             : #include <linux/sort.h>
       9             : #include <linux/string.h>
      10             : #include <linux/range.h>
      11             : 
      12           0 : int add_range(struct range *range, int az, int nr_range, u64 start, u64 end)
      13             : {
      14           0 :         if (start >= end)
      15             :                 return nr_range;
      16             : 
      17             :         /* Out of slots: */
      18           0 :         if (nr_range >= az)
      19             :                 return nr_range;
      20             : 
      21           0 :         range[nr_range].start = start;
      22           0 :         range[nr_range].end = end;
      23             : 
      24           0 :         nr_range++;
      25             : 
      26           0 :         return nr_range;
      27             : }
      28             : 
      29           0 : int add_range_with_merge(struct range *range, int az, int nr_range,
      30             :                      u64 start, u64 end)
      31             : {
      32             :         int i;
      33             : 
      34           0 :         if (start >= end)
      35             :                 return nr_range;
      36             : 
      37             :         /* get new start/end: */
      38           0 :         for (i = 0; i < nr_range; i++) {
      39             :                 u64 common_start, common_end;
      40             : 
      41           0 :                 if (!range[i].end)
      42           0 :                         continue;
      43             : 
      44           0 :                 common_start = max(range[i].start, start);
      45           0 :                 common_end = min(range[i].end, end);
      46           0 :                 if (common_start > common_end)
      47           0 :                         continue;
      48             : 
      49             :                 /* new start/end, will add it back at last */
      50           0 :                 start = min(range[i].start, start);
      51           0 :                 end = max(range[i].end, end);
      52             : 
      53           0 :                 memmove(&range[i], &range[i + 1],
      54             :                         (nr_range - (i + 1)) * sizeof(range[i]));
      55           0 :                 range[nr_range - 1].start = 0;
      56           0 :                 range[nr_range - 1].end   = 0;
      57           0 :                 nr_range--;
      58           0 :                 i--;
      59             :         }
      60             : 
      61             :         /* Need to add it: */
      62             :         return add_range(range, az, nr_range, start, end);
      63             : }
      64             : 
      65           0 : void subtract_range(struct range *range, int az, u64 start, u64 end)
      66             : {
      67             :         int i, j;
      68             : 
      69           0 :         if (start >= end)
      70             :                 return;
      71             : 
      72           0 :         for (j = 0; j < az; j++) {
      73           0 :                 if (!range[j].end)
      74           0 :                         continue;
      75             : 
      76           0 :                 if (start <= range[j].start && end >= range[j].end) {
      77           0 :                         range[j].start = 0;
      78           0 :                         range[j].end = 0;
      79           0 :                         continue;
      80             :                 }
      81             : 
      82           0 :                 if (start <= range[j].start && end < range[j].end &&
      83             :                     range[j].start < end) {
      84           0 :                         range[j].start = end;
      85           0 :                         continue;
      86             :                 }
      87             : 
      88             : 
      89           0 :                 if (start > range[j].start && end >= range[j].end &&
      90             :                     range[j].end > start) {
      91           0 :                         range[j].end = start;
      92           0 :                         continue;
      93             :                 }
      94             : 
      95           0 :                 if (start > range[j].start && end < range[j].end) {
      96             :                         /* Find the new spare: */
      97           0 :                         for (i = 0; i < az; i++) {
      98           0 :                                 if (range[i].end == 0)
      99             :                                         break;
     100             :                         }
     101           0 :                         if (i < az) {
     102           0 :                                 range[i].end = range[j].end;
     103           0 :                                 range[i].start = end;
     104             :                         } else {
     105           0 :                                 pr_err("%s: run out of slot in ranges\n",
     106             :                                         __func__);
     107             :                         }
     108           0 :                         range[j].end = start;
     109           0 :                         continue;
     110             :                 }
     111             :         }
     112             : }
     113             : 
     114           0 : static int cmp_range(const void *x1, const void *x2)
     115             : {
     116           0 :         const struct range *r1 = x1;
     117           0 :         const struct range *r2 = x2;
     118             : 
     119           0 :         if (r1->start < r2->start)
     120             :                 return -1;
     121           0 :         if (r1->start > r2->start)
     122             :                 return 1;
     123           0 :         return 0;
     124             : }
     125             : 
     126           0 : int clean_sort_range(struct range *range, int az)
     127             : {
     128           0 :         int i, j, k = az - 1, nr_range = az;
     129             : 
     130           0 :         for (i = 0; i < k; i++) {
     131           0 :                 if (range[i].end)
     132           0 :                         continue;
     133           0 :                 for (j = k; j > i; j--) {
     134           0 :                         if (range[j].end) {
     135             :                                 k = j;
     136             :                                 break;
     137             :                         }
     138             :                 }
     139           0 :                 if (j == i)
     140             :                         break;
     141           0 :                 range[i].start = range[k].start;
     142           0 :                 range[i].end   = range[k].end;
     143           0 :                 range[k].start = 0;
     144           0 :                 range[k].end   = 0;
     145           0 :                 k--;
     146             :         }
     147             :         /* count it */
     148           0 :         for (i = 0; i < az; i++) {
     149           0 :                 if (!range[i].end) {
     150             :                         nr_range = i;
     151             :                         break;
     152             :                 }
     153             :         }
     154             : 
     155             :         /* sort them */
     156           0 :         sort(range, nr_range, sizeof(struct range), cmp_range, NULL);
     157             : 
     158           0 :         return nr_range;
     159             : }
     160             : 
     161           0 : void sort_range(struct range *range, int nr_range)
     162             : {
     163             :         /* sort them */
     164           0 :         sort(range, nr_range, sizeof(struct range), cmp_range, NULL);
     165           0 : }