bstr.h

   1 /*
   2  * This file is part of MPlayer.
   3  *
   4  * MPlayer is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * MPlayer is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  */
  18
  19 #ifndef MPLAYER_BSTR_H
  20 #define MPLAYER_BSTR_H
  21
  22 #include <stdint.h>
  23 #include <stddef.h>
  24 #include <string.h>
  25 #include <stdbool.h>
  26
  27 #include "talloc.h"
  28
  29 /* NOTE: 'len' is size_t, but most string-handling functions below assume
  30  * that input size has been sanity checked and len fits in an int.
  31  */
  32 struct bstr {
  33     unsigned char *start;
  34     size_t len;
  35 };
  36
  37 // demux_rtp.cpp (live555) C++ compilation workaround
  38 #ifndef __cplusplus
  39 // If str.start is NULL, return NULL.
  40 static inline char *bstrdup0(void *talloc_ctx, struct bstr str)
  41 {
  42     return talloc_strndup(talloc_ctx, (char *)str.start, str.len);
  43 }
  44
  45 // Return start = NULL iff that is true for the original.
  46 static inline struct bstr bstrdup(void *talloc_ctx, struct bstr str)
  47 {
  48     struct bstr r = { NULL, str.len };
  49     if (str.start)
  50         r.start = talloc_memdup(talloc_ctx, str.start, str.len);
  51     return r;
  52 }
  53
  54 static inline struct bstr bstr(const unsigned char *s)
  55 {
  56     return (struct bstr){(unsigned char *)s, s ? strlen(s) : 0};
  57 }
  58
  59 int bstrcmp(struct bstr str1, struct bstr str2);
  60 int bstrcasecmp(struct bstr str1, struct bstr str2);
  61 int bstrchr(struct bstr str, int c);
  62 int bstrrchr(struct bstr str, int c);
  63 int bstrcspn(struct bstr str, const char *reject);
  64
  65 int bstr_find(struct bstr haystack, struct bstr needle);
  66 struct bstr *bstr_splitlines(void *talloc_ctx, struct bstr str);
  67 struct bstr bstr_lstrip(struct bstr str);
  68 struct bstr bstr_strip(struct bstr str);
  69 struct bstr bstr_split(struct bstr str, const char *sep, struct bstr *rest);
  70 struct bstr bstr_splice(struct bstr str, int start, int end);
  71 long long bstrtoll(struct bstr str, struct bstr *rest, int base);
  72 double bstrtod(struct bstr str, struct bstr *rest);
  73 void bstr_lower(struct bstr str);
  74 int bstr_sscanf(struct bstr str, const char *format, ...);
  75
  76 // Decode the UTF-8 code point at the start of the string,, and return the
  77 // character.
  78 // After calling this function, *out_next will point to the next character.
  79 // out_next can be NULL.
  80 // On error, -1 is returned, and *out_next is not modified.
  81 int bstr_decode_utf8(struct bstr str, struct bstr *out_next);
  82
  83 // Return the length of the UTF-8 sequence that starts with the given byte.
  84 // Given a string char *s, the next UTF-8 code point is to be expected at
  85 //      s + bstr_parse_utf8_code_length(s[0])
  86 // On error, -1 is returned. On success, it returns a value in the range [1, 4].
  87 int bstr_parse_utf8_code_length(unsigned char b);
  88
  89 // If s starts with prefix, return true and return the rest of the string in s.
  90 bool bstr_eatstart(struct bstr *s, struct bstr prefix);
  91
  92 static inline struct bstr bstr_cut(struct bstr str, int n)
  93 {
  94     if (n > str.len)
  95         n = str.len;
  96     return (struct bstr){str.start + n, str.len - n};
  97 }
  98
  99 static inline bool bstr_startswith(struct bstr str, struct bstr prefix)
 100 {
 101     if (str.len < prefix.len)
 102         return false;
 103     return !memcmp(str.start, prefix.start, prefix.len);
 104 }
 105
 106 static inline bool bstr_startswith0(struct bstr str, const char *prefix)
 107 {
 108     return bstr_startswith(str, bstr(prefix));
 109 }
 110
 111 static inline bool bstr_endswith(struct bstr str, struct bstr suffix)
 112 {
 113     if (str.len < suffix.len)
 114         return false;
 115     return !memcmp(str.start + str.len - suffix.len, suffix.start, suffix.len);
 116 }
 117
 118 static inline bool bstr_endswith0(struct bstr str, const char *suffix)
 119 {
 120     return bstr_endswith(str, bstr(suffix));
 121 }
 122
 123 static inline int bstrcmp0(struct bstr str1, const char *str2)
 124 {
 125     return bstrcmp(str1, bstr(str2));
 126 }
 127
 128 static inline int bstrcasecmp0(struct bstr str1, const char *str2)
 129 {
 130     return bstrcasecmp(str1, bstr(str2));
 131 }
 132
 133 static inline int bstr_find0(struct bstr haystack, const char *needle)
 134 {
 135     return bstr_find(haystack, bstr(needle));
 136 }
 137
 138 #endif
 139
 140 // create a pair (not single value!) for "%.*s" printf syntax
 141 #define BSTR_P(bstr) (int)((bstr).len), (bstr).start
 142
 143 #define WHITESPACE " \f\n\r\t\v"
 144
 145 #endif /* MPLAYER_BSTR_H */