usr.bin/sort/bwstring.h

   1 /*      $FreeBSD$       */
   2
   3 /*-
   4  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
   5  *
   6  * Copyright (C) 2009 Gabor Kovesdan <gabor@FreeBSD.org>
   7  * Copyright (C) 2012 Oleg Moskalenko <mom040267@gmail.com>
   8  * All rights reserved.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  29  * SUCH DAMAGE.
  30  */
  31
  32 #if !defined(__BWSTRING_H__)
  33 #define __BWSTRING_H__
  34
  35 #include <stdbool.h>
  36 #include <stdio.h>
  37 #include <errno.h>
  38 #include <sysexits.h>
  39 #include <wchar.h>
  40
  41 #include "sort.h"
  42 #include "mem.h"
  43
  44 extern bool byte_sort;
  45
  46 /* wchar_t is of 4 bytes: */
  47 #define SIZEOF_WCHAR_STRING(LEN) ((LEN)*sizeof(wchar_t))
  48
  49 struct wstr {
  50         size_t len;
  51         wchar_t str[];
  52 };
  53
  54 struct cstr {
  55         size_t len;
  56         char str[];
  57 };
  58
  59 /*
  60  * Binary "wide" string
  61  */
  62 struct bwstring
  63 {
  64         union {
  65                 struct wstr wdata;
  66                 struct cstr cdata;
  67         };
  68 };
  69
  70 typedef void *bwstring_iterator;
  71
  72 #define BWSLEN(s) ((mb_cur_max == 1) ? (s)->cdata.len : (s)->wdata.len)
  73 struct bwstring *bwsalloc(size_t sz);
  74
  75 size_t bwsrawlen(const struct bwstring *bws);
  76 const void* bwsrawdata(const struct bwstring *bws);
  77 void bws_setlen(struct bwstring *bws, size_t newlen);
  78 size_t bws_memsize(const struct bwstring *bws);
  79 double bwstod(struct bwstring *s0, bool *empty);
  80 int bws_month_score(const struct bwstring *s0);
  81
  82 struct bwstring *ignore_leading_blanks(struct bwstring *str);
  83 struct bwstring *ignore_nonprinting(struct bwstring *str);
  84 struct bwstring *dictionary_order(struct bwstring *str);
  85 struct bwstring *ignore_case(struct bwstring *str);
  86
  87 void bwsprintf(FILE*, struct bwstring*, const char *prefix, const char *suffix);
  88 void bws_disorder_warnx(struct bwstring *s, const char *fn, size_t pos);
  89
  90 struct bwstring *bwsdup(const struct bwstring *s);
  91 struct bwstring *bwssbdup(const wchar_t *str, size_t size);
  92 struct bwstring *bwscsbdup(const unsigned char *str, size_t size);
  93 void bwsfree(const struct bwstring *s);
  94 struct bwstring *bwsnocpy(struct bwstring *dst, const struct bwstring *src, size_t offset, size_t size);
  95 int bwscmp(const struct bwstring *bws1, const struct bwstring *bws2, size_t offset);
  96 int bwsncmp(const struct bwstring *bws1, const struct bwstring *bws2, size_t offset, size_t len);
  97 int bwscoll(const struct bwstring *bws1, const struct bwstring *bws2, size_t offset);
  98 size_t bwsfwrite(struct bwstring *bws, FILE *f, bool zero_ended);
  99
 100 static inline bwstring_iterator
 101 bws_begin(struct bwstring *bws)
 102 {
 103
 104         return ((bwstring_iterator)bws->wdata.str);
 105 }
 106
 107 static inline bwstring_iterator
 108 bws_end(struct bwstring *bws)
 109 {
 110
 111         return ((mb_cur_max == 1) ?
 112             (bwstring_iterator) (bws->cdata.str + bws->cdata.len) :
 113             (bwstring_iterator) (bws->wdata.str + bws->wdata.len));
 114 }
 115
 116 static inline bwstring_iterator
 117 bws_iterator_inc(bwstring_iterator iter, size_t pos)
 118 {
 119
 120         if (mb_cur_max == 1)
 121                 return ((unsigned char *) iter) + pos;
 122         else
 123                 return ((wchar_t*) iter) + pos;
 124 }
 125
 126 static inline wchar_t
 127 bws_get_iter_value(bwstring_iterator iter)
 128 {
 129
 130         if (mb_cur_max == 1)
 131                 return *((unsigned char *) iter);
 132         else
 133                 return *((wchar_t*) iter);
 134 }
 135
 136 int
 137 bws_iterator_cmp(bwstring_iterator iter1, bwstring_iterator iter2, size_t len);
 138
 139 #define BWS_GET(bws, pos) ((mb_cur_max == 1) ? (bws->cdata.str[(pos)]) : bws->wdata.str[(pos)])
 140
 141 void initialise_months(void);
 142
 143 #endif /* __BWSTRING_H__ */