mirror of
https://github.com/samba-team/samba.git
synced 2024-12-29 11:21:54 +03:00
768a5a23e0
Thanks to Moskvin for testing
(This used to be commit d3c31aa36c
)
2876 lines
58 KiB
C
2876 lines
58 KiB
C
/*
|
|
Unix SMB/CIFS implementation.
|
|
Samba utility functions
|
|
|
|
Copyright (C) Andrew Tridgell 1992-2001
|
|
Copyright (C) Simo Sorce 2001-2002
|
|
Copyright (C) Martin Pool 2003
|
|
Copyright (C) James Peach 2006
|
|
Copyright (C) Jeremy Allison 1992-2007
|
|
|
|
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 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
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.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "includes.h"
|
|
|
|
char toupper_ascii_fast_table[128] = {
|
|
0x0, 0x1, 0x2, 0x3, 0x4, 0x5, 0x6, 0x7, 0x8, 0x9, 0xa, 0xb, 0xc, 0xd, 0xe, 0xf,
|
|
0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, 0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f,
|
|
0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27, 0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f,
|
|
0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39, 0x3a, 0x3b, 0x3c, 0x3d, 0x3e, 0x3f,
|
|
0x40, 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48, 0x49, 0x4a, 0x4b, 0x4c, 0x4d, 0x4e, 0x4f,
|
|
0x50, 0x51, 0x52, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, 0x59, 0x5a, 0x5b, 0x5c, 0x5d, 0x5e, 0x5f,
|
|
0x60, 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48, 0x49, 0x4a, 0x4b, 0x4c, 0x4d, 0x4e, 0x4f,
|
|
0x50, 0x51, 0x52, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, 0x59, 0x5a, 0x7b, 0x7c, 0x7d, 0x7e, 0x7f
|
|
};
|
|
|
|
/**
|
|
* @file
|
|
* @brief String utilities.
|
|
**/
|
|
|
|
static bool next_token_internal_talloc(TALLOC_CTX *ctx,
|
|
const char **ptr,
|
|
char **pp_buff,
|
|
const char *sep,
|
|
bool ltrim)
|
|
{
|
|
char *s;
|
|
char *saved_s;
|
|
char *pbuf;
|
|
bool quoted;
|
|
size_t len=1;
|
|
|
|
*pp_buff = NULL;
|
|
if (!ptr) {
|
|
return(false);
|
|
}
|
|
|
|
s = (char *)*ptr;
|
|
|
|
/* default to simple separators */
|
|
if (!sep) {
|
|
sep = " \t\n\r";
|
|
}
|
|
|
|
/* find the first non sep char, if left-trimming is requested */
|
|
if (ltrim) {
|
|
while (*s && strchr_m(sep,*s)) {
|
|
s++;
|
|
}
|
|
}
|
|
|
|
/* nothing left? */
|
|
if (!*s) {
|
|
return false;
|
|
}
|
|
|
|
/* When restarting we need to go from here. */
|
|
saved_s = s;
|
|
|
|
/* Work out the length needed. */
|
|
for (quoted = false; *s &&
|
|
(quoted || !strchr_m(sep,*s)); s++) {
|
|
if (*s == '\"') {
|
|
quoted = !quoted;
|
|
} else {
|
|
len++;
|
|
}
|
|
}
|
|
|
|
/* We started with len = 1 so we have space for the nul. */
|
|
*pp_buff = TALLOC_ARRAY(ctx, char, len);
|
|
if (!*pp_buff) {
|
|
return false;
|
|
}
|
|
|
|
/* copy over the token */
|
|
pbuf = *pp_buff;
|
|
s = saved_s;
|
|
for (quoted = false; *s &&
|
|
(quoted || !strchr_m(sep,*s)); s++) {
|
|
if ( *s == '\"' ) {
|
|
quoted = !quoted;
|
|
} else {
|
|
*pbuf++ = *s;
|
|
}
|
|
}
|
|
|
|
*ptr = (*s) ? s+1 : s;
|
|
*pbuf = 0;
|
|
|
|
return true;
|
|
}
|
|
|
|
#if 0
|
|
/*
|
|
* Get the next token from a string, return false if none found. Handles
|
|
* double-quotes. This version trims leading separator characters before
|
|
* looking for a token.
|
|
*/
|
|
bool next_token(const char **ptr, char *buff, const char *sep, size_t bufsize)
|
|
{
|
|
return next_token_internal(ptr, buff, sep, bufsize, true);
|
|
}
|
|
#endif
|
|
|
|
bool next_token_talloc(TALLOC_CTX *ctx,
|
|
const char **ptr,
|
|
char **pp_buff,
|
|
const char *sep)
|
|
{
|
|
return next_token_internal_talloc(ctx, ptr, pp_buff, sep, true);
|
|
}
|
|
|
|
/*
|
|
* Get the next token from a string, return false if none found. Handles
|
|
* double-quotes. This version does not trim leading separator characters
|
|
* before looking for a token.
|
|
*/
|
|
|
|
bool next_token_no_ltrim_talloc(TALLOC_CTX *ctx,
|
|
const char **ptr,
|
|
char **pp_buff,
|
|
const char *sep)
|
|
{
|
|
return next_token_internal_talloc(ctx, ptr, pp_buff, sep, false);
|
|
}
|
|
|
|
/**
|
|
* Case insensitive string compararison.
|
|
*
|
|
* iconv does not directly give us a way to compare strings in
|
|
* arbitrary unix character sets -- all we can is convert and then
|
|
* compare. This is expensive.
|
|
*
|
|
* As an optimization, we do a first pass that considers only the
|
|
* prefix of the strings that is entirely 7-bit. Within this, we
|
|
* check whether they have the same value.
|
|
*
|
|
* Hopefully this will often give the answer without needing to copy.
|
|
* In particular it should speed comparisons to literal ascii strings
|
|
* or comparisons of strings that are "obviously" different.
|
|
*
|
|
* If we find a non-ascii character we fall back to converting via
|
|
* iconv.
|
|
*
|
|
* This should never be slower than convering the whole thing, and
|
|
* often faster.
|
|
*
|
|
* A different optimization would be to compare for bitwise equality
|
|
* in the binary encoding. (It would be possible thought hairy to do
|
|
* both simultaneously.) But in that case if they turn out to be
|
|
* different, we'd need to restart the whole thing.
|
|
*
|
|
* Even better is to implement strcasecmp for each encoding and use a
|
|
* function pointer.
|
|
**/
|
|
int StrCaseCmp(const char *s, const char *t)
|
|
{
|
|
|
|
const char *ps, *pt;
|
|
size_t size;
|
|
smb_ucs2_t *buffer_s, *buffer_t;
|
|
int ret;
|
|
|
|
for (ps = s, pt = t; ; ps++, pt++) {
|
|
char us, ut;
|
|
|
|
if (!*ps && !*pt)
|
|
return 0; /* both ended */
|
|
else if (!*ps)
|
|
return -1; /* s is a prefix */
|
|
else if (!*pt)
|
|
return +1; /* t is a prefix */
|
|
else if ((*ps & 0x80) || (*pt & 0x80))
|
|
/* not ascii anymore, do it the hard way
|
|
* from here on in */
|
|
break;
|
|
|
|
us = toupper_ascii_fast(*ps);
|
|
ut = toupper_ascii_fast(*pt);
|
|
if (us == ut)
|
|
continue;
|
|
else if (us < ut)
|
|
return -1;
|
|
else if (us > ut)
|
|
return +1;
|
|
}
|
|
|
|
size = push_ucs2_allocate(&buffer_s, ps);
|
|
if (size == (size_t)-1) {
|
|
return strcmp(ps, pt);
|
|
/* Not quite the right answer, but finding the right one
|
|
under this failure case is expensive, and it's pretty
|
|
close */
|
|
}
|
|
|
|
size = push_ucs2_allocate(&buffer_t, pt);
|
|
if (size == (size_t)-1) {
|
|
SAFE_FREE(buffer_s);
|
|
return strcmp(ps, pt);
|
|
/* Not quite the right answer, but finding the right one
|
|
under this failure case is expensive, and it's pretty
|
|
close */
|
|
}
|
|
|
|
ret = strcasecmp_w(buffer_s, buffer_t);
|
|
SAFE_FREE(buffer_s);
|
|
SAFE_FREE(buffer_t);
|
|
return ret;
|
|
}
|
|
|
|
|
|
/**
|
|
Case insensitive string compararison, length limited.
|
|
**/
|
|
int StrnCaseCmp(const char *s, const char *t, size_t len)
|
|
{
|
|
size_t n = 0;
|
|
const char *ps, *pt;
|
|
size_t size;
|
|
smb_ucs2_t *buffer_s, *buffer_t;
|
|
int ret;
|
|
|
|
for (ps = s, pt = t; n < len ; ps++, pt++, n++) {
|
|
char us, ut;
|
|
|
|
if (!*ps && !*pt)
|
|
return 0; /* both ended */
|
|
else if (!*ps)
|
|
return -1; /* s is a prefix */
|
|
else if (!*pt)
|
|
return +1; /* t is a prefix */
|
|
else if ((*ps & 0x80) || (*pt & 0x80))
|
|
/* not ascii anymore, do it the
|
|
* hard way from here on in */
|
|
break;
|
|
|
|
us = toupper_ascii_fast(*ps);
|
|
ut = toupper_ascii_fast(*pt);
|
|
if (us == ut)
|
|
continue;
|
|
else if (us < ut)
|
|
return -1;
|
|
else if (us > ut)
|
|
return +1;
|
|
}
|
|
|
|
if (n == len) {
|
|
return 0;
|
|
}
|
|
|
|
size = push_ucs2_allocate(&buffer_s, ps);
|
|
if (size == (size_t)-1) {
|
|
return strncmp(ps, pt, len-n);
|
|
/* Not quite the right answer, but finding the right one
|
|
under this failure case is expensive,
|
|
and it's pretty close */
|
|
}
|
|
|
|
size = push_ucs2_allocate(&buffer_t, pt);
|
|
if (size == (size_t)-1) {
|
|
SAFE_FREE(buffer_s);
|
|
return strncmp(ps, pt, len-n);
|
|
/* Not quite the right answer, but finding the right one
|
|
under this failure case is expensive,
|
|
and it's pretty close */
|
|
}
|
|
|
|
ret = strncasecmp_w(buffer_s, buffer_t, len-n);
|
|
SAFE_FREE(buffer_s);
|
|
SAFE_FREE(buffer_t);
|
|
return ret;
|
|
}
|
|
|
|
/**
|
|
* Compare 2 strings.
|
|
*
|
|
* @note The comparison is case-insensitive.
|
|
**/
|
|
bool strequal(const char *s1, const char *s2)
|
|
{
|
|
if (s1 == s2)
|
|
return(true);
|
|
if (!s1 || !s2)
|
|
return(false);
|
|
|
|
return(StrCaseCmp(s1,s2)==0);
|
|
}
|
|
|
|
/**
|
|
* Compare 2 strings up to and including the nth char.
|
|
*
|
|
* @note The comparison is case-insensitive.
|
|
**/
|
|
bool strnequal(const char *s1,const char *s2,size_t n)
|
|
{
|
|
if (s1 == s2)
|
|
return(true);
|
|
if (!s1 || !s2 || !n)
|
|
return(false);
|
|
|
|
return(StrnCaseCmp(s1,s2,n)==0);
|
|
}
|
|
|
|
/**
|
|
Compare 2 strings (case sensitive).
|
|
**/
|
|
|
|
bool strcsequal(const char *s1,const char *s2)
|
|
{
|
|
if (s1 == s2)
|
|
return(true);
|
|
if (!s1 || !s2)
|
|
return(false);
|
|
|
|
return(strcmp(s1,s2)==0);
|
|
}
|
|
|
|
/**
|
|
Do a case-insensitive, whitespace-ignoring string compare.
|
|
**/
|
|
|
|
int strwicmp(const char *psz1, const char *psz2)
|
|
{
|
|
/* if BOTH strings are NULL, return TRUE, if ONE is NULL return */
|
|
/* appropriate value. */
|
|
if (psz1 == psz2)
|
|
return (0);
|
|
else if (psz1 == NULL)
|
|
return (-1);
|
|
else if (psz2 == NULL)
|
|
return (1);
|
|
|
|
/* sync the strings on first non-whitespace */
|
|
while (1) {
|
|
while (isspace((int)*psz1))
|
|
psz1++;
|
|
while (isspace((int)*psz2))
|
|
psz2++;
|
|
if (toupper_ascii(*psz1) != toupper_ascii(*psz2) ||
|
|
*psz1 == '\0' || *psz2 == '\0')
|
|
break;
|
|
psz1++;
|
|
psz2++;
|
|
}
|
|
return (*psz1 - *psz2);
|
|
}
|
|
|
|
/**
|
|
Convert a string to "normal" form.
|
|
**/
|
|
|
|
void strnorm(char *s, int case_default)
|
|
{
|
|
if (case_default == CASE_UPPER)
|
|
strupper_m(s);
|
|
else
|
|
strlower_m(s);
|
|
}
|
|
|
|
/**
|
|
Check if a string is in "normal" case.
|
|
**/
|
|
|
|
bool strisnormal(const char *s, int case_default)
|
|
{
|
|
if (case_default == CASE_UPPER)
|
|
return(!strhaslower(s));
|
|
|
|
return(!strhasupper(s));
|
|
}
|
|
|
|
|
|
/**
|
|
String replace.
|
|
NOTE: oldc and newc must be 7 bit characters
|
|
**/
|
|
void string_replace( char *s, char oldc, char newc )
|
|
{
|
|
char *p;
|
|
|
|
/* this is quite a common operation, so we want it to be
|
|
fast. We optimise for the ascii case, knowing that all our
|
|
supported multi-byte character sets are ascii-compatible
|
|
(ie. they match for the first 128 chars) */
|
|
|
|
for (p = s; *p; p++) {
|
|
if (*p & 0x80) /* mb string - slow path. */
|
|
break;
|
|
if (*p == oldc) {
|
|
*p = newc;
|
|
}
|
|
}
|
|
|
|
if (!*p)
|
|
return;
|
|
|
|
/* Slow (mb) path. */
|
|
#ifdef BROKEN_UNICODE_COMPOSE_CHARACTERS
|
|
/* With compose characters we must restart from the beginning. JRA. */
|
|
p = s;
|
|
#endif
|
|
|
|
while (*p) {
|
|
size_t c_size;
|
|
next_codepoint(p, &c_size);
|
|
|
|
if (c_size == 1) {
|
|
if (*p == oldc) {
|
|
*p = newc;
|
|
}
|
|
}
|
|
p += c_size;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Skip past some strings in a buffer - old version - no checks.
|
|
* **/
|
|
|
|
char *push_skip_string(char *buf)
|
|
{
|
|
buf += strlen(buf) + 1;
|
|
return(buf);
|
|
}
|
|
|
|
/**
|
|
Skip past a string in a buffer. Buffer may not be
|
|
null terminated. end_ptr points to the first byte after
|
|
then end of the buffer.
|
|
**/
|
|
|
|
char *skip_string(const char *base, size_t len, char *buf)
|
|
{
|
|
const char *end_ptr = base + len;
|
|
|
|
if (end_ptr < base || !base || !buf || buf >= end_ptr) {
|
|
return NULL;
|
|
}
|
|
|
|
/* Skip the string */
|
|
while (*buf) {
|
|
buf++;
|
|
if (buf >= end_ptr) {
|
|
return NULL;
|
|
}
|
|
}
|
|
/* Skip the '\0' */
|
|
buf++;
|
|
return buf;
|
|
}
|
|
|
|
/**
|
|
Count the number of characters in a string. Normally this will
|
|
be the same as the number of bytes in a string for single byte strings,
|
|
but will be different for multibyte.
|
|
**/
|
|
|
|
size_t str_charnum(const char *s)
|
|
{
|
|
size_t ret;
|
|
smb_ucs2_t *tmpbuf2 = NULL;
|
|
if (push_ucs2_allocate(&tmpbuf2, s) == (size_t)-1) {
|
|
return 0;
|
|
}
|
|
ret = strlen_w(tmpbuf2);
|
|
SAFE_FREE(tmpbuf2);
|
|
return ret;
|
|
}
|
|
|
|
/**
|
|
Count the number of characters in a string. Normally this will
|
|
be the same as the number of bytes in a string for single byte strings,
|
|
but will be different for multibyte.
|
|
**/
|
|
|
|
size_t str_ascii_charnum(const char *s)
|
|
{
|
|
size_t ret;
|
|
char *tmpbuf2 = NULL;
|
|
if (push_ascii_allocate(&tmpbuf2, s) == (size_t)-1) {
|
|
return 0;
|
|
}
|
|
ret = strlen(tmpbuf2);
|
|
SAFE_FREE(tmpbuf2);
|
|
return ret;
|
|
}
|
|
|
|
bool trim_char(char *s,char cfront,char cback)
|
|
{
|
|
bool ret = false;
|
|
char *ep;
|
|
char *fp = s;
|
|
|
|
/* Ignore null or empty strings. */
|
|
if (!s || (s[0] == '\0'))
|
|
return false;
|
|
|
|
if (cfront) {
|
|
while (*fp && *fp == cfront)
|
|
fp++;
|
|
if (!*fp) {
|
|
/* We ate the string. */
|
|
s[0] = '\0';
|
|
return true;
|
|
}
|
|
if (fp != s)
|
|
ret = true;
|
|
}
|
|
|
|
ep = fp + strlen(fp) - 1;
|
|
if (cback) {
|
|
/* Attempt ascii only. Bail for mb strings. */
|
|
while ((ep >= fp) && (*ep == cback)) {
|
|
ret = true;
|
|
if ((ep > fp) && (((unsigned char)ep[-1]) & 0x80)) {
|
|
/* Could be mb... bail back to tim_string. */
|
|
char fs[2], bs[2];
|
|
if (cfront) {
|
|
fs[0] = cfront;
|
|
fs[1] = '\0';
|
|
}
|
|
bs[0] = cback;
|
|
bs[1] = '\0';
|
|
return trim_string(s, cfront ? fs : NULL, bs);
|
|
} else {
|
|
ep--;
|
|
}
|
|
}
|
|
if (ep < fp) {
|
|
/* We ate the string. */
|
|
s[0] = '\0';
|
|
return true;
|
|
}
|
|
}
|
|
|
|
ep[1] = '\0';
|
|
memmove(s, fp, ep-fp+2);
|
|
return ret;
|
|
}
|
|
|
|
/**
|
|
Trim the specified elements off the front and back of a string.
|
|
**/
|
|
|
|
bool trim_string(char *s,const char *front,const char *back)
|
|
{
|
|
bool ret = false;
|
|
size_t front_len;
|
|
size_t back_len;
|
|
size_t len;
|
|
|
|
/* Ignore null or empty strings. */
|
|
if (!s || (s[0] == '\0'))
|
|
return false;
|
|
|
|
front_len = front? strlen(front) : 0;
|
|
back_len = back? strlen(back) : 0;
|
|
|
|
len = strlen(s);
|
|
|
|
if (front_len) {
|
|
while (len && strncmp(s, front, front_len)==0) {
|
|
/* Must use memmove here as src & dest can
|
|
* easily overlap. Found by valgrind. JRA. */
|
|
memmove(s, s+front_len, (len-front_len)+1);
|
|
len -= front_len;
|
|
ret=true;
|
|
}
|
|
}
|
|
|
|
if (back_len) {
|
|
while ((len >= back_len) &&
|
|
strncmp(s+len-back_len,back,back_len)==0) {
|
|
s[len-back_len]='\0';
|
|
len -= back_len;
|
|
ret=true;
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
/**
|
|
Does a string have any uppercase chars in it?
|
|
**/
|
|
|
|
bool strhasupper(const char *s)
|
|
{
|
|
smb_ucs2_t *tmp, *p;
|
|
bool ret;
|
|
|
|
if (push_ucs2_allocate(&tmp, s) == -1) {
|
|
return false;
|
|
}
|
|
|
|
for(p = tmp; *p != 0; p++) {
|
|
if(isupper_w(*p)) {
|
|
break;
|
|
}
|
|
}
|
|
|
|
ret = (*p != 0);
|
|
SAFE_FREE(tmp);
|
|
return ret;
|
|
}
|
|
|
|
/**
|
|
Does a string have any lowercase chars in it?
|
|
**/
|
|
|
|
bool strhaslower(const char *s)
|
|
{
|
|
smb_ucs2_t *tmp, *p;
|
|
bool ret;
|
|
|
|
if (push_ucs2_allocate(&tmp, s) == -1) {
|
|
return false;
|
|
}
|
|
|
|
for(p = tmp; *p != 0; p++) {
|
|
if(islower_w(*p)) {
|
|
break;
|
|
}
|
|
}
|
|
|
|
ret = (*p != 0);
|
|
SAFE_FREE(tmp);
|
|
return ret;
|
|
}
|
|
|
|
/**
|
|
Find the number of 'c' chars in a string
|
|
**/
|
|
|
|
size_t count_chars(const char *s,char c)
|
|
{
|
|
smb_ucs2_t *ptr;
|
|
int count;
|
|
smb_ucs2_t *alloc_tmpbuf = NULL;
|
|
|
|
if (push_ucs2_allocate(&alloc_tmpbuf, s) == (size_t)-1) {
|
|
return 0;
|
|
}
|
|
|
|
for(count=0,ptr=alloc_tmpbuf;*ptr;ptr++)
|
|
if(*ptr==UCS2_CHAR(c))
|
|
count++;
|
|
|
|
SAFE_FREE(alloc_tmpbuf);
|
|
return(count);
|
|
}
|
|
|
|
/**
|
|
Safe string copy into a known length string. maxlength does not
|
|
include the terminating zero.
|
|
**/
|
|
|
|
char *safe_strcpy_fn(const char *fn,
|
|
int line,
|
|
char *dest,
|
|
const char *src,
|
|
size_t maxlength)
|
|
{
|
|
size_t len;
|
|
|
|
if (!dest) {
|
|
DEBUG(0,("ERROR: NULL dest in safe_strcpy, "
|
|
"called from [%s][%d]\n", fn, line));
|
|
return NULL;
|
|
}
|
|
|
|
#ifdef DEVELOPER
|
|
clobber_region(fn,line,dest, maxlength+1);
|
|
#endif
|
|
|
|
if (!src) {
|
|
*dest = 0;
|
|
return dest;
|
|
}
|
|
|
|
len = strnlen(src, maxlength+1);
|
|
|
|
if (len > maxlength) {
|
|
DEBUG(0,("ERROR: string overflow by "
|
|
"%lu (%lu - %lu) in safe_strcpy [%.50s]\n",
|
|
(unsigned long)(len-maxlength), (unsigned long)len,
|
|
(unsigned long)maxlength, src));
|
|
len = maxlength;
|
|
}
|
|
|
|
memmove(dest, src, len);
|
|
dest[len] = 0;
|
|
return dest;
|
|
}
|
|
|
|
/**
|
|
Safe string cat into a string. maxlength does not
|
|
include the terminating zero.
|
|
**/
|
|
char *safe_strcat_fn(const char *fn,
|
|
int line,
|
|
char *dest,
|
|
const char *src,
|
|
size_t maxlength)
|
|
{
|
|
size_t src_len, dest_len;
|
|
|
|
if (!dest) {
|
|
DEBUG(0,("ERROR: NULL dest in safe_strcat, "
|
|
"called from [%s][%d]\n", fn, line));
|
|
return NULL;
|
|
}
|
|
|
|
if (!src)
|
|
return dest;
|
|
|
|
src_len = strnlen(src, maxlength + 1);
|
|
dest_len = strnlen(dest, maxlength + 1);
|
|
|
|
#ifdef DEVELOPER
|
|
clobber_region(fn, line, dest + dest_len, maxlength + 1 - dest_len);
|
|
#endif
|
|
|
|
if (src_len + dest_len > maxlength) {
|
|
DEBUG(0,("ERROR: string overflow by %d "
|
|
"in safe_strcat [%.50s]\n",
|
|
(int)(src_len + dest_len - maxlength), src));
|
|
if (maxlength > dest_len) {
|
|
memcpy(&dest[dest_len], src, maxlength - dest_len);
|
|
}
|
|
dest[maxlength] = 0;
|
|
return NULL;
|
|
}
|
|
|
|
memcpy(&dest[dest_len], src, src_len);
|
|
dest[dest_len + src_len] = 0;
|
|
return dest;
|
|
}
|
|
|
|
/**
|
|
Paranoid strcpy into a buffer of given length (includes terminating
|
|
zero. Strips out all but 'a-Z0-9' and the character in other_safe_chars
|
|
and replaces with '_'. Deliberately does *NOT* check for multibyte
|
|
characters. Don't change it !
|
|
**/
|
|
|
|
char *alpha_strcpy_fn(const char *fn,
|
|
int line,
|
|
char *dest,
|
|
const char *src,
|
|
const char *other_safe_chars,
|
|
size_t maxlength)
|
|
{
|
|
size_t len, i;
|
|
|
|
#ifdef DEVELOPER
|
|
clobber_region(fn, line, dest, maxlength);
|
|
#endif
|
|
|
|
if (!dest) {
|
|
DEBUG(0,("ERROR: NULL dest in alpha_strcpy, "
|
|
"called from [%s][%d]\n", fn, line));
|
|
return NULL;
|
|
}
|
|
|
|
if (!src) {
|
|
*dest = 0;
|
|
return dest;
|
|
}
|
|
|
|
len = strlen(src);
|
|
if (len >= maxlength)
|
|
len = maxlength - 1;
|
|
|
|
if (!other_safe_chars)
|
|
other_safe_chars = "";
|
|
|
|
for(i = 0; i < len; i++) {
|
|
int val = (src[i] & 0xff);
|
|
if (isupper_ascii(val) || islower_ascii(val) ||
|
|
isdigit(val) || strchr_m(other_safe_chars, val))
|
|
dest[i] = src[i];
|
|
else
|
|
dest[i] = '_';
|
|
}
|
|
|
|
dest[i] = '\0';
|
|
|
|
return dest;
|
|
}
|
|
|
|
/**
|
|
Like strncpy but always null terminates. Make sure there is room!
|
|
The variable n should always be one less than the available size.
|
|
**/
|
|
char *StrnCpy_fn(const char *fn, int line,char *dest,const char *src,size_t n)
|
|
{
|
|
char *d = dest;
|
|
|
|
#ifdef DEVELOPER
|
|
clobber_region(fn, line, dest, n+1);
|
|
#endif
|
|
|
|
if (!dest) {
|
|
DEBUG(0,("ERROR: NULL dest in StrnCpy, "
|
|
"called from [%s][%d]\n", fn, line));
|
|
return(NULL);
|
|
}
|
|
|
|
if (!src) {
|
|
*dest = 0;
|
|
return(dest);
|
|
}
|
|
|
|
while (n-- && (*d = *src)) {
|
|
d++;
|
|
src++;
|
|
}
|
|
|
|
*d = 0;
|
|
return(dest);
|
|
}
|
|
|
|
#if 0
|
|
/**
|
|
Like strncpy but copies up to the character marker. always null terminates.
|
|
returns a pointer to the character marker in the source string (src).
|
|
**/
|
|
|
|
static char *strncpyn(char *dest, const char *src, size_t n, char c)
|
|
{
|
|
char *p;
|
|
size_t str_len;
|
|
|
|
#ifdef DEVELOPER
|
|
clobber_region(dest, n+1);
|
|
#endif
|
|
p = strchr_m(src, c);
|
|
if (p == NULL) {
|
|
DEBUG(5, ("strncpyn: separator character (%c) not found\n", c));
|
|
return NULL;
|
|
}
|
|
|
|
str_len = PTR_DIFF(p, src);
|
|
strncpy(dest, src, MIN(n, str_len));
|
|
dest[str_len] = '\0';
|
|
|
|
return p;
|
|
}
|
|
#endif
|
|
|
|
/**
|
|
Routine to get hex characters and turn them into a 16 byte array.
|
|
the array can be variable length, and any non-hex-numeric
|
|
characters are skipped. "0xnn" or "0Xnn" is specially catered
|
|
for.
|
|
|
|
valid examples: "0A5D15"; "0x15, 0x49, 0xa2"; "59\ta9\te3\n"
|
|
|
|
**/
|
|
|
|
size_t strhex_to_str(char *buf, size_t buf_len, const char *strhex, size_t strhex_len)
|
|
{
|
|
size_t i;
|
|
size_t num_chars = 0;
|
|
unsigned char lonybble, hinybble;
|
|
const char *hexchars = "0123456789ABCDEF";
|
|
char *p1 = NULL, *p2 = NULL;
|
|
|
|
for (i = 0; i < strhex_len && strhex[i] != 0; i++) {
|
|
if (strnequal(hexchars, "0x", 2)) {
|
|
i++; /* skip two chars */
|
|
continue;
|
|
}
|
|
|
|
if (!(p1 = strchr_m(hexchars, toupper_ascii(strhex[i]))))
|
|
break;
|
|
|
|
i++; /* next hex digit */
|
|
|
|
if (!(p2 = strchr_m(hexchars, toupper_ascii(strhex[i]))))
|
|
break;
|
|
|
|
/* get the two nybbles */
|
|
hinybble = PTR_DIFF(p1, hexchars);
|
|
lonybble = PTR_DIFF(p2, hexchars);
|
|
|
|
if (num_chars >= buf_len) {
|
|
break;
|
|
}
|
|
buf[num_chars] = (hinybble << 4) | lonybble;
|
|
num_chars++;
|
|
|
|
p1 = NULL;
|
|
p2 = NULL;
|
|
}
|
|
return num_chars;
|
|
}
|
|
|
|
DATA_BLOB strhex_to_data_blob(TALLOC_CTX *mem_ctx, const char *strhex)
|
|
{
|
|
DATA_BLOB ret_blob;
|
|
|
|
if (mem_ctx != NULL)
|
|
ret_blob = data_blob_talloc(mem_ctx, NULL, strlen(strhex)/2+1);
|
|
else
|
|
ret_blob = data_blob(NULL, strlen(strhex)/2+1);
|
|
|
|
ret_blob.length = strhex_to_str((char*)ret_blob.data,
|
|
ret_blob.length,
|
|
strhex,
|
|
strlen(strhex));
|
|
|
|
return ret_blob;
|
|
}
|
|
|
|
/**
|
|
* Routine to print a buffer as HEX digits, into an allocated string.
|
|
*/
|
|
|
|
char *hex_encode(TALLOC_CTX *mem_ctx, const unsigned char *buff_in, size_t len)
|
|
{
|
|
int i;
|
|
char *hex_buffer;
|
|
|
|
hex_buffer = TALLOC_ARRAY(mem_ctx, char, (len*2)+1);
|
|
|
|
for (i = 0; i < len; i++)
|
|
slprintf(&hex_buffer[i*2], 3, "%02X", buff_in[i]);
|
|
|
|
return hex_buffer;
|
|
}
|
|
|
|
/**
|
|
Check if a string is part of a list.
|
|
**/
|
|
|
|
bool in_list(const char *s, const char *list, bool casesensitive)
|
|
{
|
|
char *tok = NULL;
|
|
bool ret = false;
|
|
TALLOC_CTX *frame;
|
|
|
|
if (!list) {
|
|
return false;
|
|
}
|
|
|
|
frame = talloc_stackframe();
|
|
while (next_token_talloc(frame, &list, &tok,LIST_SEP)) {
|
|
if (casesensitive) {
|
|
if (strcmp(tok,s) == 0) {
|
|
ret = true;
|
|
break;
|
|
}
|
|
} else {
|
|
if (StrCaseCmp(tok,s) == 0) {
|
|
ret = true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
TALLOC_FREE(frame);
|
|
return ret;
|
|
}
|
|
|
|
/* this is used to prevent lots of mallocs of size 1 */
|
|
static const char null_string[] = "";
|
|
|
|
/**
|
|
Set a string value, allocing the space for the string
|
|
**/
|
|
|
|
static bool string_init(char **dest,const char *src)
|
|
{
|
|
size_t l;
|
|
|
|
if (!src)
|
|
src = "";
|
|
|
|
l = strlen(src);
|
|
|
|
if (l == 0) {
|
|
*dest = CONST_DISCARD(char*, null_string);
|
|
} else {
|
|
(*dest) = SMB_STRDUP(src);
|
|
if ((*dest) == NULL) {
|
|
DEBUG(0,("Out of memory in string_init\n"));
|
|
return false;
|
|
}
|
|
}
|
|
return(true);
|
|
}
|
|
|
|
/**
|
|
Free a string value.
|
|
**/
|
|
|
|
void string_free(char **s)
|
|
{
|
|
if (!s || !(*s))
|
|
return;
|
|
if (*s == null_string)
|
|
*s = NULL;
|
|
SAFE_FREE(*s);
|
|
}
|
|
|
|
/**
|
|
Set a string value, deallocating any existing space, and allocing the space
|
|
for the string
|
|
**/
|
|
|
|
bool string_set(char **dest,const char *src)
|
|
{
|
|
string_free(dest);
|
|
return(string_init(dest,src));
|
|
}
|
|
|
|
/**
|
|
Substitute a string for a pattern in another string. Make sure there is
|
|
enough room!
|
|
|
|
This routine looks for pattern in s and replaces it with
|
|
insert. It may do multiple replacements or just one.
|
|
|
|
Any of " ; ' $ or ` in the insert string are replaced with _
|
|
if len==0 then the string cannot be extended. This is different from the old
|
|
use of len==0 which was for no length checks to be done.
|
|
**/
|
|
|
|
void string_sub2(char *s,const char *pattern, const char *insert, size_t len,
|
|
bool remove_unsafe_characters, bool replace_once,
|
|
bool allow_trailing_dollar)
|
|
{
|
|
char *p;
|
|
ssize_t ls,lp,li, i;
|
|
|
|
if (!insert || !pattern || !*pattern || !s)
|
|
return;
|
|
|
|
ls = (ssize_t)strlen(s);
|
|
lp = (ssize_t)strlen(pattern);
|
|
li = (ssize_t)strlen(insert);
|
|
|
|
if (len == 0)
|
|
len = ls + 1; /* len is number of *bytes* */
|
|
|
|
while (lp <= ls && (p = strstr_m(s,pattern))) {
|
|
if (ls + (li-lp) >= len) {
|
|
DEBUG(0,("ERROR: string overflow by "
|
|
"%d in string_sub(%.50s, %d)\n",
|
|
(int)(ls + (li-lp) - len),
|
|
pattern, (int)len));
|
|
break;
|
|
}
|
|
if (li != lp) {
|
|
memmove(p+li,p+lp,strlen(p+lp)+1);
|
|
}
|
|
for (i=0;i<li;i++) {
|
|
switch (insert[i]) {
|
|
case '`':
|
|
case '"':
|
|
case '\'':
|
|
case ';':
|
|
case '$':
|
|
/* allow a trailing $
|
|
* (as in machine accounts) */
|
|
if (allow_trailing_dollar && (i == li - 1 )) {
|
|
p[i] = insert[i];
|
|
break;
|
|
}
|
|
case '%':
|
|
case '\r':
|
|
case '\n':
|
|
if ( remove_unsafe_characters ) {
|
|
p[i] = '_';
|
|
/* yes this break should be here
|
|
* since we want to fall throw if
|
|
* not replacing unsafe chars */
|
|
break;
|
|
}
|
|
default:
|
|
p[i] = insert[i];
|
|
}
|
|
}
|
|
s = p + li;
|
|
ls += (li-lp);
|
|
|
|
if (replace_once)
|
|
break;
|
|
}
|
|
}
|
|
|
|
void string_sub_once(char *s, const char *pattern,
|
|
const char *insert, size_t len)
|
|
{
|
|
string_sub2( s, pattern, insert, len, true, true, false );
|
|
}
|
|
|
|
void string_sub(char *s,const char *pattern, const char *insert, size_t len)
|
|
{
|
|
string_sub2( s, pattern, insert, len, true, false, false );
|
|
}
|
|
|
|
void fstring_sub(char *s,const char *pattern,const char *insert)
|
|
{
|
|
string_sub(s, pattern, insert, sizeof(fstring));
|
|
}
|
|
|
|
/**
|
|
Similar to string_sub2, but it will accept only allocated strings
|
|
and may realloc them so pay attention at what you pass on no
|
|
pointers inside strings, no const may be passed
|
|
as string.
|
|
**/
|
|
|
|
char *realloc_string_sub2(char *string,
|
|
const char *pattern,
|
|
const char *insert,
|
|
bool remove_unsafe_characters,
|
|
bool allow_trailing_dollar)
|
|
{
|
|
char *p, *in;
|
|
char *s;
|
|
ssize_t ls,lp,li,ld, i;
|
|
|
|
if (!insert || !pattern || !*pattern || !string || !*string)
|
|
return NULL;
|
|
|
|
s = string;
|
|
|
|
in = SMB_STRDUP(insert);
|
|
if (!in) {
|
|
DEBUG(0, ("realloc_string_sub: out of memory!\n"));
|
|
return NULL;
|
|
}
|
|
ls = (ssize_t)strlen(s);
|
|
lp = (ssize_t)strlen(pattern);
|
|
li = (ssize_t)strlen(insert);
|
|
ld = li - lp;
|
|
for (i=0;i<li;i++) {
|
|
switch (in[i]) {
|
|
case '`':
|
|
case '"':
|
|
case '\'':
|
|
case ';':
|
|
case '$':
|
|
/* allow a trailing $
|
|
* (as in machine accounts) */
|
|
if (allow_trailing_dollar && (i == li - 1 )) {
|
|
break;
|
|
}
|
|
case '%':
|
|
case '\r':
|
|
case '\n':
|
|
if ( remove_unsafe_characters ) {
|
|
in[i] = '_';
|
|
break;
|
|
}
|
|
default:
|
|
/* ok */
|
|
break;
|
|
}
|
|
}
|
|
|
|
while ((p = strstr_m(s,pattern))) {
|
|
if (ld > 0) {
|
|
int offset = PTR_DIFF(s,string);
|
|
string = (char *)SMB_REALLOC(string, ls + ld + 1);
|
|
if (!string) {
|
|
DEBUG(0, ("realloc_string_sub: "
|
|
"out of memory!\n"));
|
|
SAFE_FREE(in);
|
|
return NULL;
|
|
}
|
|
p = string + offset + (p - s);
|
|
}
|
|
if (li != lp) {
|
|
memmove(p+li,p+lp,strlen(p+lp)+1);
|
|
}
|
|
memcpy(p, in, li);
|
|
s = p + li;
|
|
ls += ld;
|
|
}
|
|
SAFE_FREE(in);
|
|
return string;
|
|
}
|
|
|
|
char *realloc_string_sub(char *string,
|
|
const char *pattern,
|
|
const char *insert)
|
|
{
|
|
return realloc_string_sub2(string, pattern, insert, true, false);
|
|
}
|
|
|
|
/*
|
|
* Internal guts of talloc_string_sub and talloc_all_string_sub.
|
|
* talloc version of string_sub2.
|
|
*/
|
|
|
|
char *talloc_string_sub2(TALLOC_CTX *mem_ctx, const char *src,
|
|
const char *pattern,
|
|
const char *insert,
|
|
bool remove_unsafe_characters,
|
|
bool replace_once,
|
|
bool allow_trailing_dollar)
|
|
{
|
|
char *p, *in;
|
|
char *s;
|
|
char *string;
|
|
ssize_t ls,lp,li,ld, i;
|
|
|
|
if (!insert || !pattern || !*pattern || !src || !*src) {
|
|
return NULL;
|
|
}
|
|
|
|
string = talloc_strdup(mem_ctx, src);
|
|
if (string == NULL) {
|
|
DEBUG(0, ("talloc_string_sub2: "
|
|
"talloc_strdup failed\n"));
|
|
return NULL;
|
|
}
|
|
|
|
s = string;
|
|
|
|
in = SMB_STRDUP(insert);
|
|
if (!in) {
|
|
DEBUG(0, ("talloc_string_sub2: ENOMEM\n"));
|
|
return NULL;
|
|
}
|
|
ls = (ssize_t)strlen(s);
|
|
lp = (ssize_t)strlen(pattern);
|
|
li = (ssize_t)strlen(insert);
|
|
ld = li - lp;
|
|
|
|
for (i=0;i<li;i++) {
|
|
switch (in[i]) {
|
|
case '`':
|
|
case '"':
|
|
case '\'':
|
|
case ';':
|
|
case '$':
|
|
/* allow a trailing $
|
|
* (as in machine accounts) */
|
|
if (allow_trailing_dollar && (i == li - 1 )) {
|
|
break;
|
|
}
|
|
case '%':
|
|
case '\r':
|
|
case '\n':
|
|
if (remove_unsafe_characters) {
|
|
in[i] = '_';
|
|
break;
|
|
}
|
|
default:
|
|
/* ok */
|
|
break;
|
|
}
|
|
}
|
|
|
|
while ((p = strstr_m(s,pattern))) {
|
|
if (ld > 0) {
|
|
int offset = PTR_DIFF(s,string);
|
|
string = (char *)TALLOC_REALLOC(mem_ctx, string,
|
|
ls + ld + 1);
|
|
if (!string) {
|
|
DEBUG(0, ("talloc_string_sub: out of "
|
|
"memory!\n"));
|
|
SAFE_FREE(in);
|
|
return NULL;
|
|
}
|
|
p = string + offset + (p - s);
|
|
}
|
|
if (li != lp) {
|
|
memmove(p+li,p+lp,strlen(p+lp)+1);
|
|
}
|
|
memcpy(p, in, li);
|
|
s = p + li;
|
|
ls += ld;
|
|
|
|
if (replace_once) {
|
|
break;
|
|
}
|
|
}
|
|
SAFE_FREE(in);
|
|
return string;
|
|
}
|
|
|
|
/* Same as string_sub, but returns a talloc'ed string */
|
|
|
|
char *talloc_string_sub(TALLOC_CTX *mem_ctx,
|
|
const char *src,
|
|
const char *pattern,
|
|
const char *insert)
|
|
{
|
|
return talloc_string_sub2(mem_ctx, src, pattern, insert,
|
|
true, false, false);
|
|
}
|
|
|
|
/**
|
|
Similar to string_sub() but allows for any character to be substituted.
|
|
Use with caution!
|
|
if len==0 then the string cannot be extended. This is different from the old
|
|
use of len==0 which was for no length checks to be done.
|
|
**/
|
|
|
|
void all_string_sub(char *s,const char *pattern,const char *insert, size_t len)
|
|
{
|
|
char *p;
|
|
ssize_t ls,lp,li;
|
|
|
|
if (!insert || !pattern || !s)
|
|
return;
|
|
|
|
ls = (ssize_t)strlen(s);
|
|
lp = (ssize_t)strlen(pattern);
|
|
li = (ssize_t)strlen(insert);
|
|
|
|
if (!*pattern)
|
|
return;
|
|
|
|
if (len == 0)
|
|
len = ls + 1; /* len is number of *bytes* */
|
|
|
|
while (lp <= ls && (p = strstr_m(s,pattern))) {
|
|
if (ls + (li-lp) >= len) {
|
|
DEBUG(0,("ERROR: string overflow by "
|
|
"%d in all_string_sub(%.50s, %d)\n",
|
|
(int)(ls + (li-lp) - len),
|
|
pattern, (int)len));
|
|
break;
|
|
}
|
|
if (li != lp) {
|
|
memmove(p+li,p+lp,strlen(p+lp)+1);
|
|
}
|
|
memcpy(p, insert, li);
|
|
s = p + li;
|
|
ls += (li-lp);
|
|
}
|
|
}
|
|
|
|
char *talloc_all_string_sub(TALLOC_CTX *ctx,
|
|
const char *src,
|
|
const char *pattern,
|
|
const char *insert)
|
|
{
|
|
return talloc_string_sub2(ctx, src, pattern, insert,
|
|
false, false, false);
|
|
}
|
|
|
|
/**
|
|
Write an octal as a string.
|
|
**/
|
|
|
|
char *octal_string(int i)
|
|
{
|
|
char *result;
|
|
if (i == -1) {
|
|
result = talloc_strdup(talloc_tos(), "-1");
|
|
}
|
|
else {
|
|
result = talloc_asprintf(talloc_tos(), "0%o", i);
|
|
}
|
|
SMB_ASSERT(result != NULL);
|
|
return result;
|
|
}
|
|
|
|
|
|
/**
|
|
Truncate a string at a specified length.
|
|
**/
|
|
|
|
char *string_truncate(char *s, unsigned int length)
|
|
{
|
|
if (s && strlen(s) > length)
|
|
s[length] = 0;
|
|
return s;
|
|
}
|
|
|
|
/**
|
|
Strchr and strrchr_m are very hard to do on general multi-byte strings.
|
|
We convert via ucs2 for now.
|
|
**/
|
|
|
|
char *strchr_m(const char *src, char c)
|
|
{
|
|
smb_ucs2_t *ws = NULL;
|
|
char *s2 = NULL;
|
|
smb_ucs2_t *p;
|
|
const char *s;
|
|
char *ret;
|
|
|
|
/* characters below 0x3F are guaranteed to not appear in
|
|
non-initial position in multi-byte charsets */
|
|
if ((c & 0xC0) == 0) {
|
|
return strchr(src, c);
|
|
}
|
|
|
|
/* this is quite a common operation, so we want it to be
|
|
fast. We optimise for the ascii case, knowing that all our
|
|
supported multi-byte character sets are ascii-compatible
|
|
(ie. they match for the first 128 chars) */
|
|
|
|
for (s = src; *s && !(((unsigned char)s[0]) & 0x80); s++) {
|
|
if (*s == c)
|
|
return (char *)s;
|
|
}
|
|
|
|
if (!*s)
|
|
return NULL;
|
|
|
|
#ifdef BROKEN_UNICODE_COMPOSE_CHARACTERS
|
|
/* With compose characters we must restart from the beginning. JRA. */
|
|
s = src;
|
|
#endif
|
|
|
|
if (push_ucs2_allocate(&ws, s)==(size_t)-1) {
|
|
/* Wrong answer, but what can we do... */
|
|
return strchr(src, c);
|
|
}
|
|
p = strchr_w(ws, UCS2_CHAR(c));
|
|
if (!p) {
|
|
SAFE_FREE(ws);
|
|
return NULL;
|
|
}
|
|
*p = 0;
|
|
if (pull_ucs2_allocate(&s2, ws)==(size_t)-1) {
|
|
SAFE_FREE(ws);
|
|
/* Wrong answer, but what can we do... */
|
|
return strchr(src, c);
|
|
}
|
|
ret = (char *)(s+strlen(s2));
|
|
SAFE_FREE(ws);
|
|
SAFE_FREE(s2);
|
|
return ret;
|
|
}
|
|
|
|
char *strrchr_m(const char *s, char c)
|
|
{
|
|
/* characters below 0x3F are guaranteed to not appear in
|
|
non-initial position in multi-byte charsets */
|
|
if ((c & 0xC0) == 0) {
|
|
return strrchr(s, c);
|
|
}
|
|
|
|
/* this is quite a common operation, so we want it to be
|
|
fast. We optimise for the ascii case, knowing that all our
|
|
supported multi-byte character sets are ascii-compatible
|
|
(ie. they match for the first 128 chars). Also, in Samba
|
|
we only search for ascii characters in 'c' and that
|
|
in all mb character sets with a compound character
|
|
containing c, if 'c' is not a match at position
|
|
p, then p[-1] > 0x7f. JRA. */
|
|
|
|
{
|
|
size_t len = strlen(s);
|
|
const char *cp = s;
|
|
bool got_mb = false;
|
|
|
|
if (len == 0)
|
|
return NULL;
|
|
cp += (len - 1);
|
|
do {
|
|
if (c == *cp) {
|
|
/* Could be a match. Part of a multibyte ? */
|
|
if ((cp > s) &&
|
|
(((unsigned char)cp[-1]) & 0x80)) {
|
|
/* Yep - go slow :-( */
|
|
got_mb = true;
|
|
break;
|
|
}
|
|
/* No - we have a match ! */
|
|
return (char *)cp;
|
|
}
|
|
} while (cp-- != s);
|
|
if (!got_mb)
|
|
return NULL;
|
|
}
|
|
|
|
/* String contained a non-ascii char. Slow path. */
|
|
{
|
|
smb_ucs2_t *ws = NULL;
|
|
char *s2 = NULL;
|
|
smb_ucs2_t *p;
|
|
char *ret;
|
|
|
|
if (push_ucs2_allocate(&ws,s)==(size_t)-1) {
|
|
/* Wrong answer, but what can we do. */
|
|
return strrchr(s, c);
|
|
}
|
|
p = strrchr_w(ws, UCS2_CHAR(c));
|
|
if (!p) {
|
|
SAFE_FREE(ws);
|
|
return NULL;
|
|
}
|
|
*p = 0;
|
|
if (pull_ucs2_allocate(&s2,ws)==(size_t)-1) {
|
|
SAFE_FREE(ws);
|
|
/* Wrong answer, but what can we do. */
|
|
return strrchr(s, c);
|
|
}
|
|
ret = (char *)(s+strlen(s2));
|
|
SAFE_FREE(ws);
|
|
SAFE_FREE(s2);
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
/***********************************************************************
|
|
Return the equivalent of doing strrchr 'n' times - always going
|
|
backwards.
|
|
***********************************************************************/
|
|
|
|
char *strnrchr_m(const char *s, char c, unsigned int n)
|
|
{
|
|
smb_ucs2_t *ws = NULL;
|
|
char *s2 = NULL;
|
|
smb_ucs2_t *p;
|
|
char *ret;
|
|
|
|
if (push_ucs2_allocate(&ws,s)==(size_t)-1) {
|
|
/* Too hard to try and get right. */
|
|
return NULL;
|
|
}
|
|
p = strnrchr_w(ws, UCS2_CHAR(c), n);
|
|
if (!p) {
|
|
SAFE_FREE(ws);
|
|
return NULL;
|
|
}
|
|
*p = 0;
|
|
if (pull_ucs2_allocate(&s2,ws)==(size_t)-1) {
|
|
SAFE_FREE(ws);
|
|
/* Too hard to try and get right. */
|
|
return NULL;
|
|
}
|
|
ret = (char *)(s+strlen(s2));
|
|
SAFE_FREE(ws);
|
|
SAFE_FREE(s2);
|
|
return ret;
|
|
}
|
|
|
|
/***********************************************************************
|
|
strstr_m - We convert via ucs2 for now.
|
|
***********************************************************************/
|
|
|
|
char *strstr_m(const char *src, const char *findstr)
|
|
{
|
|
smb_ucs2_t *p;
|
|
smb_ucs2_t *src_w, *find_w;
|
|
const char *s;
|
|
char *s2;
|
|
char *retp;
|
|
|
|
size_t findstr_len = 0;
|
|
|
|
/* for correctness */
|
|
if (!findstr[0]) {
|
|
return (char*)src;
|
|
}
|
|
|
|
/* Samba does single character findstr calls a *lot*. */
|
|
if (findstr[1] == '\0')
|
|
return strchr_m(src, *findstr);
|
|
|
|
/* We optimise for the ascii case, knowing that all our
|
|
supported multi-byte character sets are ascii-compatible
|
|
(ie. they match for the first 128 chars) */
|
|
|
|
for (s = src; *s && !(((unsigned char)s[0]) & 0x80); s++) {
|
|
if (*s == *findstr) {
|
|
if (!findstr_len)
|
|
findstr_len = strlen(findstr);
|
|
|
|
if (strncmp(s, findstr, findstr_len) == 0) {
|
|
return (char *)s;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!*s)
|
|
return NULL;
|
|
|
|
#if 1 /* def BROKEN_UNICODE_COMPOSE_CHARACTERS */
|
|
/* 'make check' fails unless we do this */
|
|
|
|
/* With compose characters we must restart from the beginning. JRA. */
|
|
s = src;
|
|
#endif
|
|
|
|
if (push_ucs2_allocate(&src_w, src) == (size_t)-1) {
|
|
DEBUG(0,("strstr_m: src malloc fail\n"));
|
|
return NULL;
|
|
}
|
|
|
|
if (push_ucs2_allocate(&find_w, findstr) == (size_t)-1) {
|
|
SAFE_FREE(src_w);
|
|
DEBUG(0,("strstr_m: find malloc fail\n"));
|
|
return NULL;
|
|
}
|
|
|
|
p = strstr_w(src_w, find_w);
|
|
|
|
if (!p) {
|
|
SAFE_FREE(src_w);
|
|
SAFE_FREE(find_w);
|
|
return NULL;
|
|
}
|
|
|
|
*p = 0;
|
|
if (pull_ucs2_allocate(&s2, src_w) == (size_t)-1) {
|
|
SAFE_FREE(src_w);
|
|
SAFE_FREE(find_w);
|
|
DEBUG(0,("strstr_m: dest malloc fail\n"));
|
|
return NULL;
|
|
}
|
|
retp = (char *)(s+strlen(s2));
|
|
SAFE_FREE(src_w);
|
|
SAFE_FREE(find_w);
|
|
SAFE_FREE(s2);
|
|
return retp;
|
|
}
|
|
|
|
/**
|
|
Convert a string to lower case.
|
|
**/
|
|
|
|
void strlower_m(char *s)
|
|
{
|
|
size_t len;
|
|
int errno_save;
|
|
|
|
/* this is quite a common operation, so we want it to be
|
|
fast. We optimise for the ascii case, knowing that all our
|
|
supported multi-byte character sets are ascii-compatible
|
|
(ie. they match for the first 128 chars) */
|
|
|
|
while (*s && !(((unsigned char)s[0]) & 0x80)) {
|
|
*s = tolower_ascii((unsigned char)*s);
|
|
s++;
|
|
}
|
|
|
|
if (!*s)
|
|
return;
|
|
|
|
/* I assume that lowercased string takes the same number of bytes
|
|
* as source string even in UTF-8 encoding. (VIV) */
|
|
len = strlen(s) + 1;
|
|
errno_save = errno;
|
|
errno = 0;
|
|
unix_strlower(s,len,s,len);
|
|
/* Catch mb conversion errors that may not terminate. */
|
|
if (errno)
|
|
s[len-1] = '\0';
|
|
errno = errno_save;
|
|
}
|
|
|
|
/**
|
|
Convert a string to upper case.
|
|
**/
|
|
|
|
void strupper_m(char *s)
|
|
{
|
|
size_t len;
|
|
int errno_save;
|
|
|
|
/* this is quite a common operation, so we want it to be
|
|
fast. We optimise for the ascii case, knowing that all our
|
|
supported multi-byte character sets are ascii-compatible
|
|
(ie. they match for the first 128 chars) */
|
|
|
|
while (*s && !(((unsigned char)s[0]) & 0x80)) {
|
|
*s = toupper_ascii_fast((unsigned char)*s);
|
|
s++;
|
|
}
|
|
|
|
if (!*s)
|
|
return;
|
|
|
|
/* I assume that lowercased string takes the same number of bytes
|
|
* as source string even in multibyte encoding. (VIV) */
|
|
len = strlen(s) + 1;
|
|
errno_save = errno;
|
|
errno = 0;
|
|
unix_strupper(s,len,s,len);
|
|
/* Catch mb conversion errors that may not terminate. */
|
|
if (errno)
|
|
s[len-1] = '\0';
|
|
errno = errno_save;
|
|
}
|
|
|
|
/**
|
|
Count the number of UCS2 characters in a string. Normally this will
|
|
be the same as the number of bytes in a string for single byte strings,
|
|
but will be different for multibyte.
|
|
**/
|
|
|
|
size_t strlen_m(const char *s)
|
|
{
|
|
size_t count = 0;
|
|
|
|
if (!s) {
|
|
return 0;
|
|
}
|
|
|
|
while (*s && !(((uint8_t)*s) & 0x80)) {
|
|
s++;
|
|
count++;
|
|
}
|
|
|
|
if (!*s) {
|
|
return count;
|
|
}
|
|
|
|
while (*s) {
|
|
size_t c_size;
|
|
codepoint_t c = next_codepoint(s, &c_size);
|
|
if (c < 0x10000) {
|
|
/* Unicode char fits into 16 bits. */
|
|
count += 1;
|
|
} else {
|
|
/* Double-width unicode char - 32 bits. */
|
|
count += 2;
|
|
}
|
|
s += c_size;
|
|
}
|
|
|
|
return count;
|
|
}
|
|
|
|
/**
|
|
Count the number of UCS2 characters in a string including the null
|
|
terminator.
|
|
**/
|
|
|
|
size_t strlen_m_term(const char *s)
|
|
{
|
|
if (!s) {
|
|
return 0;
|
|
}
|
|
return strlen_m(s) + 1;
|
|
}
|
|
|
|
/*
|
|
* Weird helper routine for the winreg pipe: If nothing is around, return 0,
|
|
* if a string is there, include the terminator.
|
|
*/
|
|
|
|
size_t strlen_m_term_null(const char *s)
|
|
{
|
|
size_t len;
|
|
if (!s) {
|
|
return 0;
|
|
}
|
|
len = strlen_m(s);
|
|
if (len == 0) {
|
|
return 0;
|
|
}
|
|
|
|
return len+1;
|
|
}
|
|
/**
|
|
Return a RFC2254 binary string representation of a buffer.
|
|
Used in LDAP filters.
|
|
Caller must free.
|
|
**/
|
|
|
|
char *binary_string_rfc2254(char *buf, int len)
|
|
{
|
|
char *s;
|
|
int i, j;
|
|
const char *hex = "0123456789ABCDEF";
|
|
s = (char *)SMB_MALLOC(len * 3 + 1);
|
|
if (!s)
|
|
return NULL;
|
|
for (j=i=0;i<len;i++) {
|
|
s[j] = '\\';
|
|
s[j+1] = hex[((unsigned char)buf[i]) >> 4];
|
|
s[j+2] = hex[((unsigned char)buf[i]) & 0xF];
|
|
j += 3;
|
|
}
|
|
s[j] = 0;
|
|
return s;
|
|
}
|
|
|
|
char *binary_string(char *buf, int len)
|
|
{
|
|
char *s;
|
|
int i, j;
|
|
const char *hex = "0123456789ABCDEF";
|
|
s = (char *)SMB_MALLOC(len * 2 + 1);
|
|
if (!s)
|
|
return NULL;
|
|
for (j=i=0;i<len;i++) {
|
|
s[j] = hex[((unsigned char)buf[i]) >> 4];
|
|
s[j+1] = hex[((unsigned char)buf[i]) & 0xF];
|
|
j += 2;
|
|
}
|
|
s[j] = 0;
|
|
return s;
|
|
}
|
|
|
|
/**
|
|
Just a typesafety wrapper for snprintf into a fstring.
|
|
**/
|
|
|
|
int fstr_sprintf(fstring s, const char *fmt, ...)
|
|
{
|
|
va_list ap;
|
|
int ret;
|
|
|
|
va_start(ap, fmt);
|
|
ret = vsnprintf(s, FSTRING_LEN, fmt, ap);
|
|
va_end(ap);
|
|
return ret;
|
|
}
|
|
|
|
/**
|
|
List of Strings manipulation functions
|
|
**/
|
|
|
|
#define S_LIST_ABS 16 /* List Allocation Block Size */
|
|
|
|
char **str_list_make(TALLOC_CTX *mem_ctx, const char *string, const char *sep)
|
|
{
|
|
char **list;
|
|
const char *str;
|
|
char *s;
|
|
int num, lsize;
|
|
char *tok;
|
|
|
|
if (!string || !*string)
|
|
return NULL;
|
|
|
|
list = TALLOC_ARRAY(mem_ctx, char *, S_LIST_ABS+1);
|
|
if (list == NULL) {
|
|
return NULL;
|
|
}
|
|
lsize = S_LIST_ABS;
|
|
|
|
s = talloc_strdup(list, string);
|
|
if (s == NULL) {
|
|
DEBUG(0,("str_list_make: Unable to allocate memory"));
|
|
TALLOC_FREE(list);
|
|
return NULL;
|
|
}
|
|
if (!sep) sep = LIST_SEP;
|
|
|
|
num = 0;
|
|
str = s;
|
|
|
|
while (next_token_talloc(list, &str, &tok, sep)) {
|
|
|
|
if (num == lsize) {
|
|
char **tmp;
|
|
|
|
lsize += S_LIST_ABS;
|
|
|
|
tmp = TALLOC_REALLOC_ARRAY(mem_ctx, list, char *,
|
|
lsize + 1);
|
|
if (tmp == NULL) {
|
|
DEBUG(0,("str_list_make: "
|
|
"Unable to allocate memory"));
|
|
TALLOC_FREE(list);
|
|
return NULL;
|
|
}
|
|
|
|
list = tmp;
|
|
|
|
memset (&list[num], 0,
|
|
((sizeof(char**)) * (S_LIST_ABS +1)));
|
|
}
|
|
|
|
list[num] = tok;
|
|
num += 1;
|
|
}
|
|
|
|
list[num] = NULL;
|
|
|
|
TALLOC_FREE(s);
|
|
return list;
|
|
}
|
|
|
|
bool str_list_copy(TALLOC_CTX *mem_ctx, char ***dest, const char **src)
|
|
{
|
|
char **list;
|
|
int i, num;
|
|
|
|
*dest = NULL;
|
|
if (!src)
|
|
return false;
|
|
|
|
num = 0;
|
|
while (src[num] != NULL) {
|
|
num += 1;
|
|
}
|
|
|
|
list = TALLOC_ARRAY(mem_ctx, char *, num+1);
|
|
if (list == NULL) {
|
|
return false;
|
|
}
|
|
|
|
for (i=0; i<num; i++) {
|
|
list[i] = talloc_strdup(list, src[i]);
|
|
if (list[i] == NULL) {
|
|
TALLOC_FREE(list);
|
|
return false;
|
|
}
|
|
}
|
|
list[i] = NULL;
|
|
*dest = list;
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* Return true if all the elements of the list match exactly.
|
|
**/
|
|
bool str_list_compare(char **list1, char **list2)
|
|
{
|
|
int num;
|
|
|
|
if (!list1 || !list2)
|
|
return (list1 == list2);
|
|
|
|
for (num = 0; list1[num]; num++) {
|
|
if (!list2[num])
|
|
return false;
|
|
if (!strcsequal(list1[num], list2[num]))
|
|
return false;
|
|
}
|
|
if (list2[num])
|
|
return false; /* if list2 has more elements than list1 fail */
|
|
|
|
return true;
|
|
}
|
|
|
|
/******************************************************************************
|
|
*****************************************************************************/
|
|
|
|
int str_list_count( const char **list )
|
|
{
|
|
int i = 0;
|
|
|
|
if ( ! list )
|
|
return 0;
|
|
|
|
/* count the number of list members */
|
|
|
|
for ( i=0; *list; i++, list++ );
|
|
|
|
return i;
|
|
}
|
|
|
|
/******************************************************************************
|
|
version of standard_sub_basic() for string lists; uses talloc_sub_basic()
|
|
for the work
|
|
*****************************************************************************/
|
|
|
|
bool str_list_sub_basic( char **list, const char *smb_name,
|
|
const char *domain_name )
|
|
{
|
|
TALLOC_CTX *ctx = list;
|
|
char *s, *tmpstr;
|
|
|
|
while ( *list ) {
|
|
s = *list;
|
|
tmpstr = talloc_sub_basic(ctx, smb_name, domain_name, s);
|
|
if ( !tmpstr ) {
|
|
DEBUG(0,("str_list_sub_basic: "
|
|
"alloc_sub_basic() return NULL!\n"));
|
|
return false;
|
|
}
|
|
|
|
TALLOC_FREE(*list);
|
|
*list = tmpstr;
|
|
|
|
list++;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
/******************************************************************************
|
|
substritute a specific pattern in a string list
|
|
*****************************************************************************/
|
|
|
|
bool str_list_substitute(char **list, const char *pattern, const char *insert)
|
|
{
|
|
char *p, *s, *t;
|
|
ssize_t ls, lp, li, ld, i, d;
|
|
|
|
if (!list)
|
|
return false;
|
|
if (!pattern)
|
|
return false;
|
|
if (!insert)
|
|
return false;
|
|
|
|
lp = (ssize_t)strlen(pattern);
|
|
li = (ssize_t)strlen(insert);
|
|
ld = li -lp;
|
|
|
|
while (*list) {
|
|
s = *list;
|
|
ls = (ssize_t)strlen(s);
|
|
|
|
while ((p = strstr_m(s, pattern))) {
|
|
t = *list;
|
|
d = p -t;
|
|
if (ld) {
|
|
t = (char *) SMB_MALLOC(ls +ld +1);
|
|
if (!t) {
|
|
DEBUG(0,("str_list_substitute: "
|
|
"Unable to allocate memory"));
|
|
return false;
|
|
}
|
|
memcpy(t, *list, d);
|
|
memcpy(t +d +li, p +lp, ls -d -lp +1);
|
|
SAFE_FREE(*list);
|
|
*list = t;
|
|
ls += ld;
|
|
s = t +d +li;
|
|
}
|
|
|
|
for (i = 0; i < li; i++) {
|
|
switch (insert[i]) {
|
|
case '`':
|
|
case '"':
|
|
case '\'':
|
|
case ';':
|
|
case '$':
|
|
case '%':
|
|
case '\r':
|
|
case '\n':
|
|
t[d +i] = '_';
|
|
break;
|
|
default:
|
|
t[d +i] = insert[i];
|
|
}
|
|
}
|
|
}
|
|
|
|
list++;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
#define IPSTR_LIST_SEP ","
|
|
#define IPSTR_LIST_CHAR ','
|
|
|
|
/**
|
|
* Add ip string representation to ipstr list. Used also
|
|
* as part of @function ipstr_list_make
|
|
*
|
|
* @param ipstr_list pointer to string containing ip list;
|
|
* MUST BE already allocated and IS reallocated if necessary
|
|
* @param ipstr_size pointer to current size of ipstr_list (might be changed
|
|
* as a result of reallocation)
|
|
* @param ip IP address which is to be added to list
|
|
* @return pointer to string appended with new ip and possibly
|
|
* reallocated to new length
|
|
**/
|
|
|
|
static char *ipstr_list_add(char **ipstr_list, const struct ip_service *service)
|
|
{
|
|
char *new_ipstr = NULL;
|
|
char addr_buf[INET6_ADDRSTRLEN];
|
|
int ret;
|
|
|
|
/* arguments checking */
|
|
if (!ipstr_list || !service) {
|
|
return NULL;
|
|
}
|
|
|
|
print_sockaddr(addr_buf,
|
|
sizeof(addr_buf),
|
|
&service->ss);
|
|
|
|
/* attempt to convert ip to a string and append colon separator to it */
|
|
if (*ipstr_list) {
|
|
if (service->ss.ss_family == AF_INET) {
|
|
/* IPv4 */
|
|
ret = asprintf(&new_ipstr, "%s%s%s:%d", *ipstr_list,
|
|
IPSTR_LIST_SEP, addr_buf,
|
|
service->port);
|
|
} else {
|
|
/* IPv6 */
|
|
ret = asprintf(&new_ipstr, "%s%s[%s]:%d", *ipstr_list,
|
|
IPSTR_LIST_SEP, addr_buf,
|
|
service->port);
|
|
}
|
|
SAFE_FREE(*ipstr_list);
|
|
} else {
|
|
if (service->ss.ss_family == AF_INET) {
|
|
/* IPv4 */
|
|
ret = asprintf(&new_ipstr, "%s:%d", addr_buf,
|
|
service->port);
|
|
} else {
|
|
/* IPv6 */
|
|
ret = asprintf(&new_ipstr, "[%s]:%d", addr_buf,
|
|
service->port);
|
|
}
|
|
}
|
|
if (ret == -1) {
|
|
return NULL;
|
|
}
|
|
*ipstr_list = new_ipstr;
|
|
return *ipstr_list;
|
|
}
|
|
|
|
/**
|
|
* Allocate and initialise an ipstr list using ip adresses
|
|
* passed as arguments.
|
|
*
|
|
* @param ipstr_list pointer to string meant to be allocated and set
|
|
* @param ip_list array of ip addresses to place in the list
|
|
* @param ip_count number of addresses stored in ip_list
|
|
* @return pointer to allocated ip string
|
|
**/
|
|
|
|
char *ipstr_list_make(char **ipstr_list,
|
|
const struct ip_service *ip_list,
|
|
int ip_count)
|
|
{
|
|
int i;
|
|
|
|
/* arguments checking */
|
|
if (!ip_list || !ipstr_list) {
|
|
return 0;
|
|
}
|
|
|
|
*ipstr_list = NULL;
|
|
|
|
/* process ip addresses given as arguments */
|
|
for (i = 0; i < ip_count; i++) {
|
|
*ipstr_list = ipstr_list_add(ipstr_list, &ip_list[i]);
|
|
}
|
|
|
|
return (*ipstr_list);
|
|
}
|
|
|
|
|
|
/**
|
|
* Parse given ip string list into array of ip addresses
|
|
* (as ip_service structures)
|
|
* e.g. [IPv6]:port,192.168.1.100:389,192.168.1.78, ...
|
|
*
|
|
* @param ipstr ip string list to be parsed
|
|
* @param ip_list pointer to array of ip addresses which is
|
|
* allocated by this function and must be freed by caller
|
|
* @return number of successfully parsed addresses
|
|
**/
|
|
|
|
int ipstr_list_parse(const char *ipstr_list, struct ip_service **ip_list)
|
|
{
|
|
TALLOC_CTX *frame;
|
|
char *token_str = NULL;
|
|
size_t count;
|
|
int i;
|
|
|
|
if (!ipstr_list || !ip_list)
|
|
return 0;
|
|
|
|
count = count_chars(ipstr_list, IPSTR_LIST_CHAR) + 1;
|
|
if ( (*ip_list = SMB_MALLOC_ARRAY(struct ip_service, count)) == NULL ) {
|
|
DEBUG(0,("ipstr_list_parse: malloc failed for %lu entries\n",
|
|
(unsigned long)count));
|
|
return 0;
|
|
}
|
|
|
|
frame = talloc_stackframe();
|
|
for ( i=0; next_token_talloc(frame, &ipstr_list, &token_str,
|
|
IPSTR_LIST_SEP) && i<count; i++ ) {
|
|
char *s = token_str;
|
|
char *p = strrchr(token_str, ':');
|
|
|
|
if (p) {
|
|
*p = 0;
|
|
(*ip_list)[i].port = atoi(p+1);
|
|
}
|
|
|
|
/* convert single token to ip address */
|
|
if (token_str[0] == '[') {
|
|
/* IPv6 address. */
|
|
s++;
|
|
p = strchr(token_str, ']');
|
|
if (!p) {
|
|
continue;
|
|
}
|
|
*p = '\0';
|
|
}
|
|
if (!interpret_string_addr(&(*ip_list)[i].ss,
|
|
s,
|
|
AI_NUMERICHOST)) {
|
|
continue;
|
|
}
|
|
}
|
|
TALLOC_FREE(frame);
|
|
return count;
|
|
}
|
|
|
|
/**
|
|
* Safely free ip string list
|
|
*
|
|
* @param ipstr_list ip string list to be freed
|
|
**/
|
|
|
|
void ipstr_list_free(char* ipstr_list)
|
|
{
|
|
SAFE_FREE(ipstr_list);
|
|
}
|
|
|
|
/**
|
|
Unescape a URL encoded string, in place.
|
|
**/
|
|
|
|
void rfc1738_unescape(char *buf)
|
|
{
|
|
char *p=buf;
|
|
|
|
while (p && *p && (p=strchr_m(p,'%'))) {
|
|
int c1 = p[1];
|
|
int c2 = p[2];
|
|
|
|
if (c1 >= '0' && c1 <= '9')
|
|
c1 = c1 - '0';
|
|
else if (c1 >= 'A' && c1 <= 'F')
|
|
c1 = 10 + c1 - 'A';
|
|
else if (c1 >= 'a' && c1 <= 'f')
|
|
c1 = 10 + c1 - 'a';
|
|
else {p++; continue;}
|
|
|
|
if (c2 >= '0' && c2 <= '9')
|
|
c2 = c2 - '0';
|
|
else if (c2 >= 'A' && c2 <= 'F')
|
|
c2 = 10 + c2 - 'A';
|
|
else if (c2 >= 'a' && c2 <= 'f')
|
|
c2 = 10 + c2 - 'a';
|
|
else {p++; continue;}
|
|
|
|
*p = (c1<<4) | c2;
|
|
|
|
memmove(p+1, p+3, strlen(p+3)+1);
|
|
p++;
|
|
}
|
|
}
|
|
|
|
static const char b64[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
|
|
|
|
/**
|
|
* Decode a base64 string into a DATA_BLOB - simple and slow algorithm
|
|
**/
|
|
DATA_BLOB base64_decode_data_blob(const char *s)
|
|
{
|
|
int bit_offset, byte_offset, idx, i, n;
|
|
DATA_BLOB decoded = data_blob(s, strlen(s)+1);
|
|
unsigned char *d = decoded.data;
|
|
char *p;
|
|
|
|
n=i=0;
|
|
|
|
while (*s && (p=strchr_m(b64,*s))) {
|
|
idx = (int)(p - b64);
|
|
byte_offset = (i*6)/8;
|
|
bit_offset = (i*6)%8;
|
|
d[byte_offset] &= ~((1<<(8-bit_offset))-1);
|
|
if (bit_offset < 3) {
|
|
d[byte_offset] |= (idx << (2-bit_offset));
|
|
n = byte_offset+1;
|
|
} else {
|
|
d[byte_offset] |= (idx >> (bit_offset-2));
|
|
d[byte_offset+1] = 0;
|
|
d[byte_offset+1] |= (idx << (8-(bit_offset-2))) & 0xFF;
|
|
n = byte_offset+2;
|
|
}
|
|
s++; i++;
|
|
}
|
|
|
|
if ((n > 0) && (*s == '=')) {
|
|
n -= 1;
|
|
}
|
|
|
|
/* fix up length */
|
|
decoded.length = n;
|
|
return decoded;
|
|
}
|
|
|
|
/**
|
|
* Decode a base64 string in-place - wrapper for the above
|
|
**/
|
|
void base64_decode_inplace(char *s)
|
|
{
|
|
DATA_BLOB decoded = base64_decode_data_blob(s);
|
|
|
|
if ( decoded.length != 0 ) {
|
|
memcpy(s, decoded.data, decoded.length);
|
|
|
|
/* null terminate */
|
|
s[decoded.length] = '\0';
|
|
} else {
|
|
*s = '\0';
|
|
}
|
|
|
|
data_blob_free(&decoded);
|
|
}
|
|
|
|
/**
|
|
* Encode a base64 string into a talloc()ed string caller to free.
|
|
*
|
|
* From SQUID: adopted from http://ftp.sunet.se/pub2/gnu/vm/base64-encode.c
|
|
* with adjustments
|
|
**/
|
|
|
|
char *base64_encode_data_blob(TALLOC_CTX *mem_ctx, DATA_BLOB data)
|
|
{
|
|
int bits = 0;
|
|
int char_count = 0;
|
|
size_t out_cnt, len, output_len;
|
|
char *result;
|
|
|
|
if (!data.length || !data.data)
|
|
return NULL;
|
|
|
|
out_cnt = 0;
|
|
len = data.length;
|
|
output_len = data.length * 2;
|
|
result = TALLOC_ARRAY(mem_ctx, char, output_len); /* get us plenty of space */
|
|
SMB_ASSERT(result != NULL);
|
|
|
|
while (len-- && out_cnt < (data.length * 2) - 5) {
|
|
int c = (unsigned char) *(data.data++);
|
|
bits += c;
|
|
char_count++;
|
|
if (char_count == 3) {
|
|
result[out_cnt++] = b64[bits >> 18];
|
|
result[out_cnt++] = b64[(bits >> 12) & 0x3f];
|
|
result[out_cnt++] = b64[(bits >> 6) & 0x3f];
|
|
result[out_cnt++] = b64[bits & 0x3f];
|
|
bits = 0;
|
|
char_count = 0;
|
|
} else {
|
|
bits <<= 8;
|
|
}
|
|
}
|
|
if (char_count != 0) {
|
|
bits <<= 16 - (8 * char_count);
|
|
result[out_cnt++] = b64[bits >> 18];
|
|
result[out_cnt++] = b64[(bits >> 12) & 0x3f];
|
|
if (char_count == 1) {
|
|
result[out_cnt++] = '=';
|
|
result[out_cnt++] = '=';
|
|
} else {
|
|
result[out_cnt++] = b64[(bits >> 6) & 0x3f];
|
|
result[out_cnt++] = '=';
|
|
}
|
|
}
|
|
result[out_cnt] = '\0'; /* terminate */
|
|
return result;
|
|
}
|
|
|
|
/* read a SMB_BIG_UINT from a string */
|
|
SMB_BIG_UINT STR_TO_SMB_BIG_UINT(const char *nptr, const char **entptr)
|
|
{
|
|
|
|
SMB_BIG_UINT val = -1;
|
|
const char *p = nptr;
|
|
|
|
if (!p) {
|
|
if (entptr) {
|
|
*entptr = p;
|
|
}
|
|
return val;
|
|
}
|
|
|
|
while (*p && isspace(*p))
|
|
p++;
|
|
|
|
#ifdef LARGE_SMB_OFF_T
|
|
sscanf(p,"%llu",&val);
|
|
#else /* LARGE_SMB_OFF_T */
|
|
sscanf(p,"%lu",&val);
|
|
#endif /* LARGE_SMB_OFF_T */
|
|
if (entptr) {
|
|
while (*p && isdigit(*p))
|
|
p++;
|
|
*entptr = p;
|
|
}
|
|
|
|
return val;
|
|
}
|
|
|
|
/* Convert a size specification to a count of bytes. We accept the following
|
|
* suffixes:
|
|
* bytes if there is no suffix
|
|
* kK kibibytes
|
|
* mM mebibytes
|
|
* gG gibibytes
|
|
* tT tibibytes
|
|
* pP whatever the ISO name for petabytes is
|
|
*
|
|
* Returns 0 if the string can't be converted.
|
|
*/
|
|
SMB_OFF_T conv_str_size(const char * str)
|
|
{
|
|
SMB_OFF_T lval;
|
|
char * end;
|
|
|
|
if (str == NULL || *str == '\0') {
|
|
return 0;
|
|
}
|
|
|
|
#ifdef HAVE_STRTOULL
|
|
if (sizeof(SMB_OFF_T) == 8) {
|
|
lval = strtoull(str, &end, 10 /* base */);
|
|
} else {
|
|
lval = strtoul(str, &end, 10 /* base */);
|
|
}
|
|
#else
|
|
lval = strtoul(str, &end, 10 /* base */);
|
|
#endif
|
|
|
|
if (end == NULL || end == str) {
|
|
return 0;
|
|
}
|
|
|
|
if (*end) {
|
|
SMB_OFF_T lval_orig = lval;
|
|
|
|
if (strwicmp(end, "K") == 0) {
|
|
lval *= (SMB_OFF_T)1024;
|
|
} else if (strwicmp(end, "M") == 0) {
|
|
lval *= ((SMB_OFF_T)1024 * (SMB_OFF_T)1024);
|
|
} else if (strwicmp(end, "G") == 0) {
|
|
lval *= ((SMB_OFF_T)1024 * (SMB_OFF_T)1024 *
|
|
(SMB_OFF_T)1024);
|
|
} else if (strwicmp(end, "T") == 0) {
|
|
lval *= ((SMB_OFF_T)1024 * (SMB_OFF_T)1024 *
|
|
(SMB_OFF_T)1024 * (SMB_OFF_T)1024);
|
|
} else if (strwicmp(end, "P") == 0) {
|
|
lval *= ((SMB_OFF_T)1024 * (SMB_OFF_T)1024 *
|
|
(SMB_OFF_T)1024 * (SMB_OFF_T)1024 *
|
|
(SMB_OFF_T)1024);
|
|
} else {
|
|
return 0;
|
|
}
|
|
|
|
/* Primitive attempt to detect wrapping on platforms with
|
|
* 4-byte SMB_OFF_T. It's better to let the caller handle
|
|
* a failure than some random number.
|
|
*/
|
|
if (lval_orig <= lval) {
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
return lval;
|
|
}
|
|
|
|
void string_append(char **left, const char *right)
|
|
{
|
|
int new_len = strlen(right) + 1;
|
|
|
|
if (*left == NULL) {
|
|
*left = (char *)SMB_MALLOC(new_len);
|
|
*left[0] = '\0';
|
|
} else {
|
|
new_len += strlen(*left);
|
|
*left = (char *)SMB_REALLOC(*left, new_len);
|
|
}
|
|
|
|
if (*left == NULL) {
|
|
return;
|
|
}
|
|
|
|
safe_strcat(*left, right, new_len-1);
|
|
}
|
|
|
|
bool add_string_to_array(TALLOC_CTX *mem_ctx,
|
|
const char *str, const char ***strings,
|
|
int *num)
|
|
{
|
|
char *dup_str = talloc_strdup(mem_ctx, str);
|
|
|
|
*strings = TALLOC_REALLOC_ARRAY(mem_ctx, *strings,
|
|
const char *, (*num)+1);
|
|
|
|
if ((*strings == NULL) || (dup_str == NULL)) {
|
|
*num = 0;
|
|
return false;
|
|
}
|
|
|
|
(*strings)[*num] = dup_str;
|
|
*num += 1;
|
|
return true;
|
|
}
|
|
|
|
/* Append an sprintf'ed string. Double buffer size on demand. Usable without
|
|
* error checking in between. The indiation that something weird happened is
|
|
* string==NULL */
|
|
|
|
void sprintf_append(TALLOC_CTX *mem_ctx, char **string, ssize_t *len,
|
|
size_t *bufsize, const char *fmt, ...)
|
|
{
|
|
va_list ap;
|
|
char *newstr;
|
|
int ret;
|
|
bool increased;
|
|
|
|
/* len<0 is an internal marker that something failed */
|
|
if (*len < 0)
|
|
goto error;
|
|
|
|
if (*string == NULL) {
|
|
if (*bufsize == 0)
|
|
*bufsize = 128;
|
|
|
|
*string = TALLOC_ARRAY(mem_ctx, char, *bufsize);
|
|
if (*string == NULL)
|
|
goto error;
|
|
}
|
|
|
|
va_start(ap, fmt);
|
|
ret = vasprintf(&newstr, fmt, ap);
|
|
va_end(ap);
|
|
|
|
if (ret < 0)
|
|
goto error;
|
|
|
|
increased = false;
|
|
|
|
while ((*len)+ret >= *bufsize) {
|
|
increased = true;
|
|
*bufsize *= 2;
|
|
if (*bufsize >= (1024*1024*256))
|
|
goto error;
|
|
}
|
|
|
|
if (increased) {
|
|
*string = TALLOC_REALLOC_ARRAY(mem_ctx, *string, char,
|
|
*bufsize);
|
|
if (*string == NULL) {
|
|
goto error;
|
|
}
|
|
}
|
|
|
|
StrnCpy((*string)+(*len), newstr, ret);
|
|
(*len) += ret;
|
|
free(newstr);
|
|
return;
|
|
|
|
error:
|
|
*len = -1;
|
|
*string = NULL;
|
|
}
|
|
|
|
/*
|
|
* asprintf into a string and strupper_m it after that.
|
|
*/
|
|
|
|
int asprintf_strupper_m(char **strp, const char *fmt, ...)
|
|
{
|
|
va_list ap;
|
|
char *result;
|
|
int ret;
|
|
|
|
va_start(ap, fmt);
|
|
ret = vasprintf(&result, fmt, ap);
|
|
va_end(ap);
|
|
|
|
if (ret == -1)
|
|
return -1;
|
|
|
|
strupper_m(result);
|
|
*strp = result;
|
|
return ret;
|
|
}
|
|
|
|
char *talloc_asprintf_strupper_m(TALLOC_CTX *t, const char *fmt, ...)
|
|
{
|
|
va_list ap;
|
|
char *ret;
|
|
|
|
va_start(ap, fmt);
|
|
ret = talloc_vasprintf(t, fmt, ap);
|
|
va_end(ap);
|
|
|
|
if (ret == NULL) {
|
|
return NULL;
|
|
}
|
|
strupper_m(ret);
|
|
return ret;
|
|
}
|
|
|
|
/*
|
|
Returns the substring from src between the first occurrence of
|
|
the char "front" and the first occurence of the char "back".
|
|
Mallocs the return string which must be freed. Not for use
|
|
with wide character strings.
|
|
*/
|
|
char *sstring_sub(const char *src, char front, char back)
|
|
{
|
|
char *temp1, *temp2, *temp3;
|
|
ptrdiff_t len;
|
|
|
|
temp1 = strchr(src, front);
|
|
if (temp1 == NULL) return NULL;
|
|
temp2 = strchr(src, back);
|
|
if (temp2 == NULL) return NULL;
|
|
len = temp2 - temp1;
|
|
if (len <= 0) return NULL;
|
|
temp3 = (char*)SMB_MALLOC(len);
|
|
if (temp3 == NULL) {
|
|
DEBUG(1,("Malloc failure in sstring_sub\n"));
|
|
return NULL;
|
|
}
|
|
memcpy(temp3, temp1+1, len-1);
|
|
temp3[len-1] = '\0';
|
|
return temp3;
|
|
}
|
|
|
|
/********************************************************************
|
|
Check a string for any occurrences of a specified list of invalid
|
|
characters.
|
|
********************************************************************/
|
|
|
|
bool validate_net_name( const char *name,
|
|
const char *invalid_chars,
|
|
int max_len)
|
|
{
|
|
int i;
|
|
|
|
for ( i=0; i<max_len && name[i]; i++ ) {
|
|
/* fail if strchr_m() finds one of the invalid characters */
|
|
if ( name[i] && strchr_m( invalid_chars, name[i] ) ) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
/**
|
|
return the number of bytes occupied by a buffer in ASCII format
|
|
the result includes the null termination
|
|
limited by 'n' bytes
|
|
**/
|
|
size_t ascii_len_n(const char *src, size_t n)
|
|
{
|
|
size_t len;
|
|
|
|
len = strnlen(src, n);
|
|
if (len+1 <= n) {
|
|
len += 1;
|
|
}
|
|
|
|
return len;
|
|
}
|
|
|
|
/**
|
|
return the number of bytes occupied by a buffer in CH_UTF16 format
|
|
the result includes the null termination
|
|
**/
|
|
size_t utf16_len(const void *buf)
|
|
{
|
|
size_t len;
|
|
|
|
for (len = 0; SVAL(buf,len); len += 2) ;
|
|
|
|
return len + 2;
|
|
}
|
|
|
|
/**
|
|
return the number of bytes occupied by a buffer in CH_UTF16 format
|
|
the result includes the null termination
|
|
limited by 'n' bytes
|
|
**/
|
|
size_t utf16_len_n(const void *src, size_t n)
|
|
{
|
|
size_t len;
|
|
|
|
for (len = 0; (len+2 < n) && SVAL(src, len); len += 2) ;
|
|
|
|
if (len+2 <= n) {
|
|
len += 2;
|
|
}
|
|
|
|
return len;
|
|
}
|
|
|
|
/*******************************************************************
|
|
Add a shell escape character '\' to any character not in a known list
|
|
of characters. UNIX charset format.
|
|
*******************************************************************/
|
|
|
|
#define INCLUDE_LIST "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz_/ \t.,"
|
|
#define INSIDE_DQUOTE_LIST "$`\n\"\\"
|
|
|
|
char *escape_shell_string(const char *src)
|
|
{
|
|
size_t srclen = strlen(src);
|
|
char *ret = SMB_MALLOC_ARRAY(char, (srclen * 2) + 1);
|
|
char *dest = ret;
|
|
bool in_s_quote = false;
|
|
bool in_d_quote = false;
|
|
bool next_escaped = false;
|
|
|
|
if (!ret) {
|
|
return NULL;
|
|
}
|
|
|
|
while (*src) {
|
|
size_t c_size;
|
|
codepoint_t c = next_codepoint(src, &c_size);
|
|
|
|
if (c == INVALID_CODEPOINT) {
|
|
SAFE_FREE(ret);
|
|
return NULL;
|
|
}
|
|
|
|
if (c_size > 1) {
|
|
memcpy(dest, src, c_size);
|
|
src += c_size;
|
|
dest += c_size;
|
|
next_escaped = false;
|
|
continue;
|
|
}
|
|
|
|
/*
|
|
* Deal with backslash escaped state.
|
|
* This only lasts for one character.
|
|
*/
|
|
|
|
if (next_escaped) {
|
|
*dest++ = *src++;
|
|
next_escaped = false;
|
|
continue;
|
|
}
|
|
|
|
/*
|
|
* Deal with single quote state. The
|
|
* only thing we care about is exiting
|
|
* this state.
|
|
*/
|
|
|
|
if (in_s_quote) {
|
|
if (*src == '\'') {
|
|
in_s_quote = false;
|
|
}
|
|
*dest++ = *src++;
|
|
continue;
|
|
}
|
|
|
|
/*
|
|
* Deal with double quote state. The most
|
|
* complex state. We must cope with \, meaning
|
|
* possibly escape next char (depending what it
|
|
* is), ", meaning exit this state, and possibly
|
|
* add an \ escape to any unprotected character
|
|
* (listed in INSIDE_DQUOTE_LIST).
|
|
*/
|
|
|
|
if (in_d_quote) {
|
|
if (*src == '\\') {
|
|
/*
|
|
* Next character might be escaped.
|
|
* We have to peek. Inside double
|
|
* quotes only INSIDE_DQUOTE_LIST
|
|
* characters are escaped by a \.
|
|
*/
|
|
|
|
char nextchar;
|
|
|
|
c = next_codepoint(&src[1], &c_size);
|
|
if (c == INVALID_CODEPOINT) {
|
|
SAFE_FREE(ret);
|
|
return NULL;
|
|
}
|
|
if (c_size > 1) {
|
|
/*
|
|
* Don't escape the next char.
|
|
* Just copy the \.
|
|
*/
|
|
*dest++ = *src++;
|
|
continue;
|
|
}
|
|
|
|
nextchar = src[1];
|
|
|
|
if (nextchar && strchr(INSIDE_DQUOTE_LIST,
|
|
(int)nextchar)) {
|
|
next_escaped = true;
|
|
}
|
|
*dest++ = *src++;
|
|
continue;
|
|
}
|
|
|
|
if (*src == '\"') {
|
|
/* Exit double quote state. */
|
|
in_d_quote = false;
|
|
*dest++ = *src++;
|
|
continue;
|
|
}
|
|
|
|
/*
|
|
* We know the character isn't \ or ",
|
|
* so escape it if it's any of the other
|
|
* possible unprotected characters.
|
|
*/
|
|
|
|
if (strchr(INSIDE_DQUOTE_LIST, (int)*src)) {
|
|
*dest++ = '\\';
|
|
}
|
|
*dest++ = *src++;
|
|
continue;
|
|
}
|
|
|
|
/*
|
|
* From here to the end of the loop we're
|
|
* not in the single or double quote state.
|
|
*/
|
|
|
|
if (*src == '\\') {
|
|
/* Next character must be escaped. */
|
|
next_escaped = true;
|
|
*dest++ = *src++;
|
|
continue;
|
|
}
|
|
|
|
if (*src == '\'') {
|
|
/* Go into single quote state. */
|
|
in_s_quote = true;
|
|
*dest++ = *src++;
|
|
continue;
|
|
}
|
|
|
|
if (*src == '\"') {
|
|
/* Go into double quote state. */
|
|
in_d_quote = true;
|
|
*dest++ = *src++;
|
|
continue;
|
|
}
|
|
|
|
/* Check if we need to escape the character. */
|
|
|
|
if (!strchr(INCLUDE_LIST, (int)*src)) {
|
|
*dest++ = '\\';
|
|
}
|
|
*dest++ = *src++;
|
|
}
|
|
*dest++ = '\0';
|
|
return ret;
|
|
}
|