lib/memcoll.c

   1 /* Locale-specific memory comparison.
   2
   3    Copyright (C) 1999, 2002-2004, 2006, 2009-2010 Free Software Foundation,
   4    Inc.
   5
   6    This program is free software: you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
  18
  19 /* Contributed by Paul Eggert <eggert@twinsun.com>.  */
  20
  21 #include <config.h>
  22
  23 #include "memcoll.h"
  24
  25 #include <errno.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28
  29 /* Ensure strcoll operates on the entire input strings, in case they contain
  30    NUL bytes. */
  31
  32 static inline int
  33 strcoll_loop (char const *s1, size_t s1len, const char *s2, size_t s2len)
  34 {
  35   int diff;
  36
  37   while (! (errno = 0, (diff = strcoll (s1, s2)) || errno))
  38     {
  39       /* strcoll found no difference, but perhaps it was fooled by NUL
  40          characters in the data.  Work around this problem by advancing
  41          past the NUL chars.  */
  42       size_t size1 = strlen (s1) + 1;
  43       size_t size2 = strlen (s2) + 1;
  44       s1 += size1;
  45       s2 += size2;
  46       s1len -= size1;
  47       s2len -= size2;
  48
  49       if (s1len == 0)
  50         {
  51           if (s2len != 0)
  52             diff = -1;
  53           break;
  54         }
  55       else if (s2len == 0)
  56         {
  57           diff = 1;
  58           break;
  59         }
  60     }
  61
  62   return diff;
  63 }
  64
  65 /* Compare S1 (with length S1LEN) and S2 (with length S2LEN) according
  66    to the LC_COLLATE locale.  S1 and S2 do not overlap, and are not
  67    adjacent.  Perhaps temporarily modify the bytes after S1 and S2,
  68    but restore their original contents before returning.  Set errno to an
  69    error number if there is an error, and to zero otherwise.  */
  70 int
  71 memcoll (char *s1, size_t s1len, char *s2, size_t s2len)
  72 {
  73   int diff;
  74
  75   /* strcoll is slow on many platforms, so check for the common case
  76      where the arguments are bytewise equal.  Otherwise, walk through
  77      the buffers using strcoll on each substring.  */
  78
  79   if (s1len == s2len && memcmp (s1, s2, s1len) == 0)
  80     {
  81       errno = 0;
  82       diff = 0;
  83     }
  84   else
  85     {
  86       char n1 = s1[s1len];
  87       char n2 = s2[s2len];
  88
  89       s1[s1len++] = '\0';
  90       s2[s2len++] = '\0';
  91
  92       diff = strcoll_loop (s1, s1len, s2, s2len);
  93
  94       s1[s1len - 1] = n1;
  95       s2[s2len - 1] = n2;
  96     }
  97
  98   return diff;
  99 }
 100
 101 /* Compare S1 (with length S1LEN) and S2 (with length S2LEN) according
 102    to the LC_COLLATE locale.  S1 and S2 must both end in a null byte.
 103    Set errno to an error number if there is an error, and to zero
 104    otherwise.  */
 105 int
 106 memcoll0 (char const *s1, size_t s1len, const char *s2, size_t s2len)
 107 {
 108   if (s1len == s2len && memcmp (s1, s2, s1len) == 0)
 109     {
 110       errno = 0;
 111       return 0;
 112     }
 113   else
 114     return strcoll_loop (s1, s1len, s2, s2len);
 115 }