git.sesse.net Git - plocate/blob - lib.cpp

   1 /* Common functions.
   2
   3 Copyright (C) 2005, 2007 Red Hat, Inc. All rights reserved.
   4 This copyrighted material is made available to anyone wishing to use, modify,
   5 copy, or redistribute it subject to the terms and conditions of the GNU General
   6 Public License v.2.
   7
   8 This program is distributed in the hope that it will be useful, but WITHOUT ANY
   9 WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
  10 PARTICULAR PURPOSE. See the GNU General Public License for more details.
  11
  12 You should have received a copy of the GNU General Public License along with
  13 this program; if not, write to the Free Software Foundation, Inc., 51 Franklin
  14 Street, Fifth Floor, Boston, MA 02110-1301, USA.
  15
  16 Author: Miloslav Trmac <mitr@redhat.com>
  17
  18 plocate modifications: Copyright (C) 2020 Steinar H. Gunderson.
  19 plocate parts and modifications are licensed under the GPLv2 or, at your option,
  20 any later version.
  21 */
  22
  23 #include "lib.h"
  24
  25 #include <algorithm>
  26 #include <arpa/inet.h>
  27 #include <assert.h>
  28 #include <errno.h>
  29 #include <limits.h>
  30 #include <stdbool.h>
  31 #include <stdint.h>
  32 #include <stdio.h>
  33 #include <stdlib.h>
  34 #include <string.h>
  35 #include <unistd.h>
  36
  37 using namespace std;
  38
  39 /* Compare two path names using the database directory order. This is not
  40    exactly strcmp () order: "a" < "a.b", so "a/z" < "a.b". */
  41 int dir_path_cmp(const string &a, const string &b)
  42 {
  43         auto [ai, bi] = mismatch(a.begin(), a.end(), b.begin(), b.end());
  44         if (ai == a.end() && bi == b.end()) {
  45                 return 0;
  46         }
  47         if (ai == a.end()) {
  48                 return -1;
  49         }
  50         if (bi == b.end()) {
  51                 return 1;
  52         }
  53         if (*ai == *bi) {
  54                 return 0;
  55         }
  56         if (*ai == '/') {
  57                 return -1;
  58         }
  59         if (*bi == '/') {
  60                 return 1;
  61         }
  62         return int((unsigned char)*ai) - int((unsigned char)*bi);
  63 }
  64
  65 /* Sort LIST using dir_path_cmp () */
  66 void string_list_dir_path_sort(vector<string> *list)
  67 {
  68         sort(list->begin(), list->end(), [](const string &a, const string &b) {
  69                 return dir_path_cmp(a, b) < 0;
  70         });
  71 }
  72
  73 /* Is PATH included in LIST?  Update *IDX to move within LIST.
  74
  75    LIST is assumed to be sorted using dir_path_cmp (), successive calls to this
  76    function are assumed to use PATH values increasing in dir_path_cmp (). */
  77 bool string_list_contains_dir_path(const vector<string> *list, size_t *idx,
  78                                    const string &path)
  79 {
  80         int cmp = 0;
  81         while (*idx < list->size() && (cmp = dir_path_cmp((*list)[*idx], path)) < 0) {
  82                 (*idx)++;
  83         }
  84         if (*idx < list->size() && cmp == 0) {
  85                 (*idx)++;
  86                 return true;
  87         }
  88         return false;
  89 }