chars.c 20.7 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
/* $Id$ */
/**************************************************************************
 *   chars.c                                                              *
 *                                                                        *
 *   Copyright (C) 2005 Chris Allegretta                                  *
 *   This program is free software; you can redistribute it and/or modify *
 *   it under the terms of the GNU General Public License as published by *
 *   the Free Software Foundation; either version 2, or (at your option)  *
 *   any later version.                                                   *
 *                                                                        *
11
12
13
14
 *   This program is distributed in the hope that it will be useful, but  *
 *   WITHOUT ANY WARRANTY; without even the implied warranty of           *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU    *
 *   General Public License for more details.                             *
15
16
17
 *                                                                        *
 *   You should have received a copy of the GNU General Public License    *
 *   along with this program; if not, write to the Free Software          *
18
19
 *   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA            *
 *   02110-1301, USA.                                                     *
20
21
22
23
24
25
26
 *                                                                        *
 **************************************************************************/

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

27
#include <string.h>
28
29
30
#include <ctype.h>
#include "proto.h"

31
#ifdef ENABLE_UTF8
32
#ifdef HAVE_WCHAR_H
33
34
#include <wchar.h>
#endif
35
#ifdef HAVE_WCTYPE_H
36
37
#include <wctype.h>
#endif
38
39
40
41
42
43

static const wchar_t bad_wchar = 0xFFFD;
	/* If we get an invalid multibyte sequence, we treat it as
	 * Unicode FFFD (Replacement Character), unless we're
	 * determining if it's a control character or searching for a
	 * match to it. */
44
45
static const char *bad_mbchar = "\xEF\xBF\xBD";
static const int bad_mbchar_len = 3;
46
#endif
47

48
49
#ifndef HAVE_ISBLANK
/* This function is equivalent to isblank(). */
50
bool nisblank(int c)
51
{
52
    return isspace(c) && (c == '\t' || !is_cntrl_char(c));
53
}
54
#endif
55

56
#if !defined(HAVE_ISWBLANK) && defined(ENABLE_UTF8)
57
/* This function is equivalent to iswblank(). */
58
bool niswblank(wchar_t wc)
59
{
60
    return iswspace(wc) && (wc == '\t' || !is_cntrl_wchar(wc));
61
}
62
#endif
63

64
65
66
67
68
69
70
/* Return TRUE if the value of c is in byte range, and FALSE
 * otherwise. */
bool is_byte(int c)
{
    return ((unsigned int)c == (unsigned char)c);
}

71
72
73
74
75
/* This function is equivalent to isalnum() for multibyte characters. */
bool is_alnum_mbchar(const char *c)
{
    assert(c != NULL);

76
#ifdef ENABLE_UTF8
77
    if (ISSET(USE_UTF8)) {
78
79
	wchar_t wc;

80
	if (mbtowc(&wc, c, MB_CUR_MAX) < 0) {
81
	    mbtowc(NULL, NULL, 0);
82
	    wc = bad_wchar;
83
84
85
86
87
88
89
90
	}

	return iswalnum(wc);
    } else
#endif
	return isalnum((unsigned char)*c);
}

91
92
93
94
95
/* This function is equivalent to isblank() for multibyte characters. */
bool is_blank_mbchar(const char *c)
{
    assert(c != NULL);

96
#ifdef ENABLE_UTF8
97
    if (ISSET(USE_UTF8)) {
98
99
	wchar_t wc;

100
	if (mbtowc(&wc, c, MB_CUR_MAX) < 0) {
101
	    mbtowc(NULL, NULL, 0);
102
	    wc = bad_wchar;
103
104
	}

105
	return iswblank(wc);
106
107
    } else
#endif
108
	return isblank((unsigned char)*c);
109
110
111
}

/* This function is equivalent to iscntrl(), except in that it also
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
112
 * handles high-bit control characters. */
113
bool is_cntrl_char(int c)
114
{
115
116
    return (-128 <= c && c < -96) || (0 <= c && c < 32) ||
	(127 <= c && c < 160);
117
118
}

119
#ifdef ENABLE_UTF8
120
121
122
/* This function is equivalent to iscntrl() for wide characters, except
 * in that it also handles wide control characters with their high bits
 * set. */
123
bool is_cntrl_wchar(wchar_t wc)
124
{
125
    return (0 <= wc && wc < 32) || (127 <= wc && wc < 160);
126
127
128
}
#endif

129
130
131
132
133
134
135
/* This function is equivalent to iscntrl() for multibyte characters,
 * except in that it also handles multibyte control characters with
 * their high bits set. */
bool is_cntrl_mbchar(const char *c)
{
    assert(c != NULL);

136
#ifdef ENABLE_UTF8
137
    if (ISSET(USE_UTF8)) {
138
139
	wchar_t wc;

140
	if (mbtowc(&wc, c, MB_CUR_MAX) < 0) {
141
	    mbtowc(NULL, NULL, 0);
142
	    wc = bad_wchar;
143
144
145
146
147
148
149
150
	}

	return is_cntrl_wchar(wc);
    } else
#endif
	return is_cntrl_char((unsigned char)*c);
}

151
152
/* This function is equivalent to ispunct() for multibyte characters. */
bool is_punct_mbchar(const char *c)
153
154
155
{
    assert(c != NULL);

156
#ifdef ENABLE_UTF8
157
    if (ISSET(USE_UTF8)) {
158
159
160
	wchar_t wc;
	int c_mb_len = mbtowc(&wc, c, MB_CUR_MAX);

161
	if (c_mb_len < 0) {
162
	    mbtowc(NULL, NULL, 0);
163
	    wc = bad_wchar;
164
165
	}

166
	return iswpunct(wc);
167
168
    } else
#endif
169
170
171
172
	return ispunct((unsigned char)*c);
}

/* Return TRUE for a multibyte character found in a word (currently only
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
173
 * an alphanumeric or punctuation character, and only the latter if
174
175
176
177
178
179
180
 * allow_punct is TRUE) and FALSE otherwise. */
bool is_word_mbchar(const char *c, bool allow_punct)
{
    assert(c != NULL);

    return is_alnum_mbchar(c) || (allow_punct ? is_punct_mbchar(c) :
	FALSE);
181
182
}

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
183
/* c is a control character.  It displays as ^@, ^?, or ^[ch], where ch
184
 * is (c + 64).  We return that character. */
185
char control_rep(char c)
186
{
187
188
    assert(is_cntrl_char(c));

189
190
191
192
193
194
195
196
197
    /* Treat newlines embedded in a line as encoded nulls. */
    if (c == '\n')
	return '@';
    else if (c == NANO_CONTROL_8)
	return '?';
    else
	return c + 64;
}

198
#ifdef ENABLE_UTF8
199
/* c is a wide control character.  It displays as ^@, ^?, or ^[ch],
200
 * where ch is (c + 64).  We return that wide character. */
201
202
wchar_t control_wrep(wchar_t wc)
{
203
204
    assert(is_cntrl_wchar(wc));

205
206
207
208
209
210
211
212
213
214
215
    /* Treat newlines embedded in a line as encoded nulls. */
    if (wc == '\n')
	return '@';
    else if (wc == NANO_CONTROL_8)
	return '?';
    else
	return wc + 64;
}
#endif

/* c is a multibyte control character.  It displays as ^@, ^?, or ^[ch],
216
 * where ch is (c + 64).  We return that multibyte character. */
217
218
char *control_mbrep(const char *c, char *crep, int *crep_len)
{
219
    assert(c != NULL && crep != NULL && crep_len != NULL);
220

221
#ifdef ENABLE_UTF8
222
    if (ISSET(USE_UTF8)) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
223
	wchar_t wc;
224

225
	if (mbtowc(&wc, c, MB_CUR_MAX) < 0) {
226
	    mbtowc(NULL, NULL, 0);
227
228
229
230
231
232
233
234
235
	    crep = (char *)bad_mbchar;
	    *crep_len = bad_mbchar_len;
	} else {
	    *crep_len = wctomb(crep, control_wrep(wc));

	    if (*crep_len < 0) {
		wctomb(NULL, 0);
		*crep_len = 0;
	    }
236
237
238
239
	}
    } else {
#endif
	*crep_len = 1;
240
	*crep = control_rep(*c);
241
#ifdef ENABLE_UTF8
242
243
    }
#endif
244
245

    return crep;
246
247
}

248
249
250
251
252
253
254
255
256
257
/* c is a multibyte non-control character.  We return that multibyte
 * character. */
char *mbrep(const char *c, char *crep, int *crep_len)
{
    assert(c != NULL && crep != NULL && crep_len != NULL);

#ifdef ENABLE_UTF8
    if (ISSET(USE_UTF8)) {
	wchar_t wc;

258
259
	/* Reject invalid Unicode characters. */
	if (mbtowc(&wc, c, MB_CUR_MAX) < 0 || !is_valid_unicode(wc)) {
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
	    mbtowc(NULL, NULL, 0);
	    crep = (char *)bad_mbchar;
	    *crep_len = bad_mbchar_len;
	} else {
	    *crep_len = wctomb(crep, wc);

	    if (*crep_len < 0) {
		wctomb(NULL, 0);
		*crep_len = 0;
	    }
	}
    } else {
#endif
	*crep_len = 1;
	*crep = *c;
#ifdef ENABLE_UTF8
    }
#endif

    return crep;
}

282
283
284
285
286
/* This function is equivalent to wcwidth() for multibyte characters. */
int mbwidth(const char *c)
{
    assert(c != NULL);

287
#ifdef ENABLE_UTF8
288
    if (ISSET(USE_UTF8)) {
289
	wchar_t wc;
290
	int width;
291

292
	if (mbtowc(&wc, c, MB_CUR_MAX) < 0) {
293
	    mbtowc(NULL, NULL, 0);
294
	    wc = bad_wchar;
295
296
297
	}

	width = wcwidth(wc);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
298

299
300
301
302
303
304
305
306
307
308
309
310
	if (width == -1)
	    width++;

	return width;
    } else
#endif
	return 1;
}

/* Return the maximum width in bytes of a multibyte character. */
int mb_cur_max(void)
{
311
    return
312
#ifdef ENABLE_UTF8
313
	ISSET(USE_UTF8) ? MB_CUR_MAX :
314
#endif
315
	1;
316
317
}

318
319
/* Convert the Unicode value in chr to a multibyte character with the
 * same wide character value as chr, if possible.  If the conversion
320
321
322
 * succeeds, return the (dynamically allocated) multibyte character and
 * its length.  Otherwise, return an undefined (dynamically allocated)
 * multibyte character and a length of zero. */
323
char *make_mbchar(long chr, int *chr_mb_len)
324
{
325
326
    char *chr_mb;

327
    assert(chr_mb_len != NULL);
328

329
#ifdef ENABLE_UTF8
330
    if (ISSET(USE_UTF8)) {
331
	chr_mb = charalloc(MB_CUR_MAX);
332
	*chr_mb_len = wctomb(chr_mb, (wchar_t)chr);
333

334
335
	/* Reject invalid Unicode characters. */
	if (*chr_mb_len < 0 || !is_valid_unicode((wchar_t)chr)) {
336
337
	    wctomb(NULL, 0);
	    *chr_mb_len = 0;
338
339
340
341
	}
    } else {
#endif
	*chr_mb_len = 1;
342
	chr_mb = mallocstrncpy(NULL, (char *)&chr, 1);
343
#ifdef ENABLE_UTF8
344
345
346
347
348
349
350
351
    }
#endif

    return chr_mb;
}

/* Parse a multibyte character from buf.  Return the number of bytes
 * used.  If chr isn't NULL, store the multibyte character in it.  If
352
353
354
 * col isn't NULL, store the new display width in it.  If *buf is '\t',
 * we expect col to have the current display width. */
int parse_mbchar(const char *buf, char *chr, size_t *col)
355
356
357
358
359
{
    int buf_mb_len;

    assert(buf != NULL);

360
#ifdef ENABLE_UTF8
361
    if (ISSET(USE_UTF8)) {
362
363
364
	/* Get the number of bytes in the multibyte character. */
	buf_mb_len = mblen(buf, MB_CUR_MAX);

365
366
367
	/* If buf contains an invalid multibyte character, set bad_chr
	 * to TRUE and interpret buf's first byte. */
	if (buf_mb_len < 0) {
368
	    mblen(NULL, 0);
369
	    buf_mb_len = 1;
370
371
	} else if (buf_mb_len == 0)
	    buf_mb_len++;
372
373
374
375

	/* Save the multibyte character in chr. */
	if (chr != NULL) {
	    int i;
376

377
378
379
380
381
382
383
384
385
386
387
388
389
	    for (i = 0; i < buf_mb_len; i++)
		chr[i] = buf[i];
	}

	/* Save the column width of the wide character in col. */
	if (col != NULL) {
	    /* If we have a tab, get its width in columns using the
	     * current value of col. */
	    if (*buf == '\t')
		*col += tabsize - *col % tabsize;
	    /* If we have a control character, get its width using one
	     * column for the "^" that will be displayed in front of it,
	     * and the width in columns of its visible equivalent as
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
390
	     * returned by control_mbrep(). */
391
	    else if (is_cntrl_mbchar(buf)) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
392
		char *ctrl_buf_mb = charalloc(MB_CUR_MAX);
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
		int ctrl_buf_mb_len;

		(*col)++;

		ctrl_buf_mb = control_mbrep(buf, ctrl_buf_mb,
			&ctrl_buf_mb_len);

		*col += mbwidth(ctrl_buf_mb);

		free(ctrl_buf_mb);
	    /* If we have a normal character, get its width in columns
	     * normally. */
	    } else
		*col += mbwidth(buf);
	}
    } else {
#endif
	/* Get the number of bytes in the byte character. */
	buf_mb_len = 1;

	/* Save the byte character in chr. */
	if (chr != NULL)
	    *chr = *buf;

	if (col != NULL) {
	    /* If we have a tab, get its width in columns using the
	     * current value of col. */
	    if (*buf == '\t')
		*col += tabsize - *col % tabsize;
	    /* If we have a control character, it's two columns wide:
	     * one column for the "^" that will be displayed in front of
	     * it, and one column for its visible equivalent as returned
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
425
	     * by control_mbrep(). */
426
427
428
429
430
431
	    else if (is_cntrl_char((unsigned char)*buf))
		*col += 2;
	    /* If we have a normal character, it's one column wide. */
	    else
		(*col)++;
	}
432
#ifdef ENABLE_UTF8
433
434
435
436
437
    }
#endif

    return buf_mb_len;
}
438
439
440
441
442
443
444

/* Return the index in buf of the beginning of the multibyte character
 * before the one at pos. */
size_t move_mbleft(const char *buf, size_t pos)
{
    size_t pos_prev = pos;

445
    assert(buf != NULL && pos <= strlen(buf));
446
447
448
449

    /* There is no library function to move backward one multibyte
     * character.  Here is the naive, O(pos) way to do it. */
    while (TRUE) {
450
	int buf_mb_len = parse_mbchar(buf + pos - pos_prev, NULL, NULL);
451

452
	if (pos_prev <= (size_t)buf_mb_len)
453
454
455
456
457
458
459
460
461
462
463
464
	    break;

	pos_prev -= buf_mb_len;
    }

    return pos - pos_prev;
}

/* Return the index in buf of the beginning of the multibyte character
 * after the one at pos. */
size_t move_mbright(const char *buf, size_t pos)
{
465
    return pos + parse_mbchar(buf + pos, NULL, NULL);
466
}
467
468
469
470
471

#ifndef HAVE_STRCASECMP
/* This function is equivalent to strcasecmp(). */
int nstrcasecmp(const char *s1, const char *s2)
{
472
    return strncasecmp(s1, s2, (size_t)-1);
473
474
475
476
477
478
}
#endif

/* This function is equivalent to strcasecmp() for multibyte strings. */
int mbstrcasecmp(const char *s1, const char *s2)
{
479
    return mbstrncasecmp(s1, s2, (size_t)-1);
480
481
482
483
484
485
486
487
488
489
490
491
492
493
}

#ifndef HAVE_STRNCASECMP
/* This function is equivalent to strncasecmp(). */
int nstrncasecmp(const char *s1, const char *s2, size_t n)
{
    assert(s1 != NULL && s2 != NULL);

    for (; n > 0 && *s1 != '\0' && *s2 != '\0'; n--, s1++, s2++) {
	if (tolower(*s1) != tolower(*s2))
	    break;
    }

    if (n > 0)
494
	return tolower(*s1) - tolower(*s2);
495
496
497
498
499
500
501
502
503
    else
	return 0;
}
#endif

/* This function is equivalent to strncasecmp() for multibyte
 * strings. */
int mbstrncasecmp(const char *s1, const char *s2, size_t n)
{
504
#ifdef ENABLE_UTF8
505
    if (ISSET(USE_UTF8)) {
506
507
	char *s1_mb = charalloc(MB_CUR_MAX);
	char *s2_mb = charalloc(MB_CUR_MAX);
508
509
	wchar_t ws1, ws2;

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
510
511
	assert(s1 != NULL && s2 != NULL);

512
	while (n > 0 && *s1 != '\0' && *s2 != '\0') {
513
	    bool bad_s1_mb = FALSE, bad_s2_mb = FALSE;
514
515
	    int s1_mb_len, s2_mb_len;

516
	    s1_mb_len = parse_mbchar(s1, s1_mb, NULL);
517

518
	    if (mbtowc(&ws1, s1_mb, s1_mb_len) < 0) {
519
520
		mbtowc(NULL, NULL, 0);
		ws1 = (unsigned char)*s1_mb;
521
		bad_s1_mb = TRUE;
522
523
	    }

524
	    s2_mb_len = parse_mbchar(s2, s2_mb, NULL);
525

526
	    if (mbtowc(&ws2, s2_mb, s2_mb_len) < 0) {
527
528
		mbtowc(NULL, NULL, 0);
		ws2 = (unsigned char)*s2_mb;
529
		bad_s2_mb = TRUE;
530
531
	    }

532
533
	    if (n == 0 || bad_s1_mb != bad_s2_mb ||
		towlower(ws1) != towlower(ws2))
534
535
536
537
		break;

	    s1 += s1_mb_len;
	    s2 += s2_mb_len;
538
	    n--;
539
540
541
542
543
	}

	free(s1_mb);
	free(s2_mb);

544
	return towlower(ws1) - towlower(ws2);
545
546
    } else
#endif
547
	return strncasecmp(s1, s2, n);
548
549
550
551
552
553
554
555
556
557
}

#ifndef HAVE_STRCASESTR
/* This function is equivalent to strcasestr().  It was adapted from
 * mutt's mutt_stristr() function. */
const char *nstrcasestr(const char *haystack, const char *needle)
{
    assert(haystack != NULL && needle != NULL);

    for (; *haystack != '\0'; haystack++) {
558
	const char *r = haystack, *q = needle;
559

560
	for (; tolower(*r) == tolower(*q) && *q != '\0'; r++, q++)
561
562
563
564
565
566
567
568
569
570
	    ;

	if (*q == '\0')
	    return haystack;
    }

    return NULL;
}
#endif

571
572
573
/* This function is equivalent to strcasestr() for multibyte strings. */
const char *mbstrcasestr(const char *haystack, const char *needle)
{
574
#ifdef ENABLE_UTF8
575
    if (ISSET(USE_UTF8)) {
576
	char *r_mb = charalloc(MB_CUR_MAX);
577
	char *q_mb = charalloc(MB_CUR_MAX);
578
	wchar_t wr, wq;
579
580
581
582
583
	bool found_needle = FALSE;

	assert(haystack != NULL && needle != NULL);

	while (*haystack != '\0') {
584
585
	    const char *r = haystack, *q = needle;
	    int r_mb_len, q_mb_len;
586
587

	    while (*q != '\0') {
588
589
		bool bad_r_mb = FALSE, bad_q_mb = FALSE;

590
		r_mb_len = parse_mbchar(r, r_mb, NULL);
591

592
		if (mbtowc(&wr, r_mb, r_mb_len) < 0) {
593
		    mbtowc(NULL, NULL, 0);
594
		    wr = (unsigned char)*r;
595
		    bad_r_mb = TRUE;
596
597
		}

598
		q_mb_len = parse_mbchar(q, q_mb, NULL);
599

600
		if (mbtowc(&wq, q_mb, q_mb_len) < 0) {
601
602
		    mbtowc(NULL, NULL, 0);
		    wq = (unsigned char)*q;
603
		    bad_q_mb = TRUE;
604
605
		}

606
607
		if (bad_r_mb != bad_q_mb ||
			towlower(wr) != towlower(wq))
608
609
		    break;

610
		r += r_mb_len;
611
612
613
614
615
616
617
618
		q += q_mb_len;
	    }

	    if (*q == '\0') {
		found_needle = TRUE;
		break;
	    }

619
	    haystack += move_mbright(haystack, 0);
620
621
	}

622
	free(r_mb);
623
624
	free(q_mb);

625
	return found_needle ? haystack : NULL;
626
627
    } else
#endif
628
	return strcasestr(haystack, needle);
629
630
}

631
#if !defined(NANO_SMALL) || !defined(DISABLE_TABCOMP)
632
/* This function is equivalent to strstr(), except in that it scans the
633
 * string in reverse, starting at rev_start. */
634
635
636
637
638
639
640
641
const char *revstrstr(const char *haystack, const char *needle, const
	char *rev_start)
{
    assert(haystack != NULL && needle != NULL && rev_start != NULL);

    for (; rev_start >= haystack; rev_start--) {
	const char *r, *q;

642
	for (r = rev_start, q = needle; *r == *q && *q != '\0'; r++, q++)
643
644
645
646
647
648
649
650
	    ;

	if (*q == '\0')
	    return rev_start;
    }

    return NULL;
}
651
#endif /* !NANO_SMALL || !DISABLE_TABCOMP */
652

653
#ifndef NANO_SMALL
654
/* This function is equivalent to strcasestr(), except in that it scans
655
 * the string in reverse, starting at rev_start. */
656
657
658
659
660
661
662
663
const char *revstrcasestr(const char *haystack, const char *needle,
	const char *rev_start)
{
    assert(haystack != NULL && needle != NULL && rev_start != NULL);

    for (; rev_start >= haystack; rev_start--) {
	const char *r = rev_start, *q = needle;

664
	for (; tolower(*r) == tolower(*q) && *q != '\0'; r++, q++)
665
666
667
668
669
670
671
672
	    ;

	if (*q == '\0')
	    return rev_start;
    }

    return NULL;
}
673
674
675
676
677
678
679

/* This function is equivalent to strcasestr() for multibyte strings,
 * except in that it scans the string in reverse, starting at
 * rev_start. */
const char *mbrevstrcasestr(const char *haystack, const char *needle,
	const char *rev_start)
{
680
#ifdef ENABLE_UTF8
681
    if (ISSET(USE_UTF8)) {
682
683
684
685
686
687
688
689
690
691
692
693
	char *r_mb = charalloc(MB_CUR_MAX);
	char *q_mb = charalloc(MB_CUR_MAX);
	wchar_t wr, wq;
	bool begin_line = FALSE, found_needle = FALSE;

	assert(haystack != NULL && needle != NULL && rev_start != NULL);

	while (!begin_line) {
	    const char *r = rev_start, *q = needle;
	    int r_mb_len, q_mb_len;

	    while (*q != '\0') {
694
695
		bool bad_r_mb = FALSE, bad_q_mb = FALSE;

696
		r_mb_len = parse_mbchar(r, r_mb, NULL);
697

698
		if (mbtowc(&wr, r_mb, r_mb_len) < 0) {
699
700
		    mbtowc(NULL, NULL, 0);
		    wr = (unsigned char)*r;
701
		    bad_r_mb = TRUE;
702
703
		}

704
		q_mb_len = parse_mbchar(q, q_mb, NULL);
705

706
		if (mbtowc(&wq, q_mb, q_mb_len) < 0) {
707
708
		    mbtowc(NULL, NULL, 0);
		    wq = (unsigned char)*q;
709
		    bad_q_mb = TRUE;
710
711
		}

712
713
		if (bad_r_mb != bad_q_mb ||
			towlower(wr) != towlower(wq))
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
		    break;

		r += r_mb_len;
		q += q_mb_len;
	    }

	    if (*q == '\0') {
		found_needle = TRUE;
		break;
	    }

	    if (rev_start == haystack)
		begin_line = TRUE;
	    else
		rev_start = haystack + move_mbleft(haystack, rev_start -
			haystack);
	}

	free(r_mb);
	free(q_mb);

735
	return found_needle ? rev_start : NULL;
736
737
738
739
    } else
#endif
	return revstrcasestr(haystack, needle, rev_start);
}
740
#endif /* !NANO_SMALL */
741

742
743
744
745
746
747
/* This function is equivalent to strlen() for multibyte strings. */
size_t mbstrlen(const char *s)
{
    return mbstrnlen(s, (size_t)-1);
}

748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
#ifndef HAVE_STRNLEN
/* This function is equivalent to strnlen(). */
size_t nstrnlen(const char *s, size_t maxlen)
{
    size_t n = 0;

    assert(s != NULL);

    for (; maxlen > 0 && *s != '\0'; maxlen--, n++, s++)
	;

    return n;
}
#endif

/* This function is equivalent to strnlen() for multibyte strings. */
size_t mbstrnlen(const char *s, size_t maxlen)
{
    assert(s != NULL);

768
#ifdef ENABLE_UTF8
769
    if (ISSET(USE_UTF8)) {
770
771
772
773
	size_t n = 0;
	int s_mb_len;

	while (*s != '\0') {
774
	    s_mb_len = parse_mbchar(s, NULL, NULL);
775

776
	    if (maxlen == 0)
777
778
		break;

779
	    maxlen--;
780
781
	    s += s_mb_len;
	    n++;
782
783
	}

784
	return n;
785
786
    } else
#endif
787
	return strnlen(s, maxlen);
788
}
789
790

#ifndef DISABLE_JUSTIFY
791
792
793
794
795
796
797
798
799
800
801
802
803
/* This function is equivalent to strchr() for multibyte strings. */
char *mbstrchr(const char *s, char *c)
{
    assert(s != NULL && c != NULL);

#ifdef ENABLE_UTF8
    if (ISSET(USE_UTF8)) {
	bool bad_c_mb = FALSE, bad_s_mb = FALSE;
	char *s_mb = charalloc(MB_CUR_MAX);
	const char *q = s;
	wchar_t ws, wc;
	int c_mb_len = mbtowc(&wc, c, MB_CUR_MAX);

804
	if (c_mb_len < 0) {
805
806
807
808
809
810
	    mbtowc(NULL, NULL, 0);
	    wc = (unsigned char)*c;
	    bad_c_mb = TRUE;
	}

	while (*s != '\0') {
811
	    int s_mb_len = parse_mbchar(s, s_mb, NULL);
812

813
	    if (mbtowc(&ws, s_mb, s_mb_len) < 0) {
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
		mbtowc(NULL, NULL, 0);
		ws = (unsigned char)*s;
		bad_s_mb = TRUE;
	    }

	    if (bad_s_mb == bad_c_mb && ws == wc)
		break;

	    s += s_mb_len;
	    q += s_mb_len;
	}

	free(s_mb);

	if (ws != wc)
	    q = NULL;

	return (char *)q;
    } else
#endif
	return strchr(s, *c);
}

837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
#ifdef ENABLE_NANORC
/* Return TRUE if the string s contains one or more blank characters,
 * and FALSE otherwise. */
bool has_blank_chars(const char *s)
{
    assert(s != NULL);

    for (; *s != '\0'; s++) {
	if (isblank(*s))
	    return TRUE;
    }

    return FALSE;
}

/* Return TRUE if the multibyte string s contains one or more blank
 * multibyte characters, and FALSE otherwise. */
bool has_blank_mbchars(const char *s)
{
856
    assert(s != NULL);
857

858
#ifdef ENABLE_UTF8
859
    if (ISSET(USE_UTF8)) {
860
861
862
863
864
865
	char *chr_mb = charalloc(MB_CUR_MAX);
	bool retval = FALSE;

	while (*s != '\0') {
	    int chr_mb_len;

866
	    chr_mb_len = parse_mbchar(s, chr_mb, NULL);
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882

	    if (is_blank_mbchar(chr_mb)) {
		retval = TRUE;
		break;
	    }

	    s += chr_mb_len;
	}

	free(chr_mb);

	return retval;
    } else
#endif
	return has_blank_chars(s);
}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
883
#endif /* ENABLE_NANORC */
884
#endif /* !DISABLE_JUSTIFY */
885

886
887
888
889
890
#ifdef ENABLE_UTF8
/* Return TRUE if wc is valid Unicode (i.e, it's not negative or in the
 * ranges D800-DFFF or FFFE-FFFF), and FALSE otherwise. */
bool is_valid_unicode(wchar_t wc)
{
891
892
    return (0 <= wc && (wc <= 0xD7FF || 0xE000 <= wc) && (wc <=
	0xFFFD || 0x10000 <= wc));
893
894
895
}
#endif

896
897
898
899
900
901
902
903
#ifdef ENABLE_NANORC
/* Check if the string s is a valid multibyte string.  Return TRUE if it
 * is, and FALSE otherwise. */
bool is_valid_mbstring(const char *s)
{
    assert(s != NULL);

    return 
904
#ifdef ENABLE_UTF8
905
	ISSET(USE_UTF8) ?
906
	(mbstowcs(NULL, s, 0) != (size_t)-1) :
907
908
909
910
911
#endif

	TRUE;
}
#endif /* ENABLE_NANORC */