| 1 | /* Classify a domain name for IDNA purposes. | 
|---|
| 2 | Copyright (C) 2018-2020 Free Software Foundation, Inc. | 
|---|
| 3 | This file is part of the GNU C Library. | 
|---|
| 4 |  | 
|---|
| 5 | The GNU C Library is free software; you can redistribute it and/or | 
|---|
| 6 | modify it under the terms of the GNU Lesser General Public | 
|---|
| 7 | License as published by the Free Software Foundation; either | 
|---|
| 8 | version 2.1 of the License, or (at your option) any later version. | 
|---|
| 9 |  | 
|---|
| 10 | The GNU C Library is distributed in the hope that it will be useful, | 
|---|
| 11 | but WITHOUT ANY WARRANTY; without even the implied warranty of | 
|---|
| 12 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU | 
|---|
| 13 | Lesser General Public License for more details. | 
|---|
| 14 |  | 
|---|
| 15 | You should have received a copy of the GNU Lesser General Public | 
|---|
| 16 | License along with the GNU C Library; if not, see | 
|---|
| 17 | <https://www.gnu.org/licenses/>.  */ | 
|---|
| 18 |  | 
|---|
| 19 | #include <errno.h> | 
|---|
| 20 | #include <inet/net-internal.h> | 
|---|
| 21 | #include <stdbool.h> | 
|---|
| 22 | #include <string.h> | 
|---|
| 23 | #include <wchar.h> | 
|---|
| 24 |  | 
|---|
| 25 | enum idna_name_classification | 
|---|
| 26 | __idna_name_classify (const char *name) | 
|---|
| 27 | { | 
|---|
| 28 | mbstate_t mbs; | 
|---|
| 29 | memset (&mbs, 0, sizeof (mbs)); | 
|---|
| 30 | const char *p = name; | 
|---|
| 31 | const char *end = p + strlen (p) + 1; | 
|---|
| 32 | bool nonascii = false; | 
|---|
| 33 | bool backslash = false; | 
|---|
| 34 | while (true) | 
|---|
| 35 | { | 
|---|
| 36 | wchar_t wc; | 
|---|
| 37 | size_t result = mbrtowc (&wc, p, end - p, &mbs); | 
|---|
| 38 | if (result == 0) | 
|---|
| 39 | /* NUL terminator was reached.  */ | 
|---|
| 40 | break; | 
|---|
| 41 | else if (result == (size_t) -2) | 
|---|
| 42 | /* Incomplete trailing multi-byte character.  This is an | 
|---|
| 43 | encoding error becaue we received the full name.  */ | 
|---|
| 44 | return idna_name_encoding_error; | 
|---|
| 45 | else if (result == (size_t) -1) | 
|---|
| 46 | { | 
|---|
| 47 | /* Other error, including EILSEQ.  */ | 
|---|
| 48 | if (errno == EILSEQ) | 
|---|
| 49 | return idna_name_encoding_error; | 
|---|
| 50 | else if (errno == ENOMEM) | 
|---|
| 51 | return idna_name_memory_error; | 
|---|
| 52 | else | 
|---|
| 53 | return idna_name_error; | 
|---|
| 54 | } | 
|---|
| 55 | else | 
|---|
| 56 | { | 
|---|
| 57 | /* A wide character was decoded.  */ | 
|---|
| 58 | p += result; | 
|---|
| 59 | if (wc == L'\\') | 
|---|
| 60 | backslash = true; | 
|---|
| 61 | else if (wc > 127) | 
|---|
| 62 | nonascii = true; | 
|---|
| 63 | } | 
|---|
| 64 | } | 
|---|
| 65 |  | 
|---|
| 66 | if (nonascii) | 
|---|
| 67 | { | 
|---|
| 68 | if (backslash) | 
|---|
| 69 | return idna_name_nonascii_backslash; | 
|---|
| 70 | else | 
|---|
| 71 | return idna_name_nonascii; | 
|---|
| 72 | } | 
|---|
| 73 | else | 
|---|
| 74 | return idna_name_ascii; | 
|---|
| 75 | } | 
|---|
| 76 |  | 
|---|