2007-10-17 23:10:12 +04:00
/*
2002-01-30 09:08:46 +03:00
Unix SMB / CIFS implementation .
1998-11-05 19:48:35 +03:00
Samba utility functions
2007-10-17 23:10:12 +04:00
2001-07-04 11:15:53 +04:00
Copyright ( C ) Andrew Tridgell 1992 - 2001
2002-01-25 03:35:14 +03:00
Copyright ( C ) Simo Sorce 2001 - 2002
2003-03-19 00:21:21 +03:00
Copyright ( C ) Martin Pool 2003
2006-07-11 22:01:26 +04:00
Copyright ( C ) James Peach 2006
2007-09-13 05:10:01 +04:00
Copyright ( C ) Jeremy Allison 1992 - 2007
2007-10-17 23:10:12 +04:00
1998-11-05 19:48:35 +03:00
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
2007-07-09 23:25:36 +04:00
the Free Software Foundation ; either version 3 of the License , or
1998-11-05 19:48:35 +03:00
( at your option ) any later version .
2007-10-17 23:10:12 +04:00
1998-11-05 19:48:35 +03:00
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 .
2007-10-17 23:10:12 +04:00
1998-11-05 19:48:35 +03:00
You should have received a copy of the GNU General Public License
2007-07-10 04:52:41 +04:00
along with this program . If not , see < http : //www.gnu.org/licenses/>.
1998-11-05 19:48:35 +03:00
*/
# include "includes.h"
2008-12-14 15:11:41 +03:00
const char toupper_ascii_fast_table [ 128 ] = {
2008-01-10 04:32:26 +03:00
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
} ;
2003-02-24 06:09:08 +03:00
/**
2003-03-19 00:21:21 +03:00
* 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
2007-10-17 23:10:12 +04:00
* function pointer .
2003-03-19 00:21:21 +03:00
* */
1998-11-12 09:12:19 +03:00
int StrCaseCmp ( const char * s , const char * t )
1998-11-05 19:48:35 +03:00
{
2003-03-19 00:21:21 +03:00
2005-08-13 00:12:31 +04:00
const char * ps , * pt ;
2003-07-19 04:36:43 +04:00
size_t size ;
smb_ucs2_t * buffer_s , * buffer_t ;
int ret ;
2003-03-19 00:21:21 +03:00
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 ) )
2007-10-17 23:10:12 +04:00
/* not ascii anymore, do it the hard way
* from here on in */
2003-03-19 00:21:21 +03:00
break ;
2008-01-10 04:32:26 +03:00
us = toupper_ascii_fast ( * ps ) ;
ut = toupper_ascii_fast ( * pt ) ;
2003-03-19 00:21:21 +03:00
if ( us = = ut )
continue ;
else if ( us < ut )
return - 1 ;
else if ( us > ut )
return + 1 ;
}
2009-07-17 05:13:46 +04:00
if ( ! push_ucs2_talloc ( talloc_tos ( ) , & buffer_s , ps , & size ) ) {
2007-10-17 23:10:12 +04:00
return strcmp ( ps , pt ) ;
2003-07-19 04:36:43 +04:00
/* Not quite the right answer, but finding the right one
2007-10-17 23:10:12 +04:00
under this failure case is expensive , and it ' s pretty
close */
2003-07-19 04:36:43 +04:00
}
2007-10-17 23:10:12 +04:00
2009-07-17 05:13:46 +04:00
if ( ! push_ucs2_talloc ( talloc_tos ( ) , & buffer_t , pt , & size ) ) {
2009-03-19 04:20:11 +03:00
TALLOC_FREE ( buffer_s ) ;
2007-10-17 23:10:12 +04:00
return strcmp ( ps , pt ) ;
2003-07-19 04:36:43 +04:00
/* Not quite the right answer, but finding the right one
2007-10-17 23:10:12 +04:00
under this failure case is expensive , and it ' s pretty
close */
2003-07-19 04:36:43 +04:00
}
2007-10-17 23:10:12 +04:00
2003-07-19 04:36:43 +04:00
ret = strcasecmp_w ( buffer_s , buffer_t ) ;
2009-03-19 04:20:11 +03:00
TALLOC_FREE ( buffer_s ) ;
TALLOC_FREE ( buffer_t ) ;
2003-07-19 04:36:43 +04:00
return ret ;
1998-11-05 19:48:35 +03:00
}
2003-03-19 00:21:21 +03:00
2003-02-24 06:09:08 +03:00
/**
2002-10-02 23:12:14 +04:00
Case insensitive string compararison , length limited .
2003-02-24 06:09:08 +03:00
* */
2007-09-13 05:10:01 +04:00
int StrnCaseCmp ( const char * s , const char * t , size_t len )
1998-11-05 19:48:35 +03:00
{
2007-09-13 05:10:01 +04:00
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 ;
2008-01-10 04:32:26 +03:00
us = toupper_ascii_fast ( * ps ) ;
ut = toupper_ascii_fast ( * pt ) ;
2007-09-13 05:10:01 +04:00
if ( us = = ut )
continue ;
else if ( us < ut )
return - 1 ;
else if ( us > ut )
return + 1 ;
}
if ( n = = len ) {
return 0 ;
}
2009-07-17 05:13:46 +04:00
if ( ! push_ucs2_talloc ( talloc_tos ( ) , & buffer_s , ps , & size ) ) {
2007-09-13 05:10:01 +04:00
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 */
}
2009-07-17 05:13:46 +04:00
if ( ! push_ucs2_talloc ( talloc_tos ( ) , & buffer_t , pt , & size ) ) {
2009-03-19 04:20:11 +03:00
TALLOC_FREE ( buffer_s ) ;
2007-09-13 05:10:01 +04:00
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 ) ;
2009-03-19 04:20:11 +03:00
TALLOC_FREE ( buffer_s ) ;
TALLOC_FREE ( buffer_t ) ;
2007-09-13 05:10:01 +04:00
return ret ;
1998-11-05 19:48:35 +03:00
}
2003-02-24 06:09:08 +03:00
/**
2003-03-18 01:42:01 +03:00
* Compare 2 strings .
*
* @ note The comparison is case - insensitive .
* */
2007-10-17 23:10:12 +04:00
bool strequal ( const char * s1 , const char * s2 )
1998-11-05 19:48:35 +03:00
{
2002-10-02 23:12:14 +04:00
if ( s1 = = s2 )
2007-10-17 23:10:12 +04:00
return ( true ) ;
2002-10-02 23:12:14 +04:00
if ( ! s1 | | ! s2 )
2007-10-17 23:10:12 +04:00
return ( false ) ;
2007-09-13 05:10:01 +04:00
2001-07-06 15:50:45 +04:00
return ( StrCaseCmp ( s1 , s2 ) = = 0 ) ;
1998-11-05 19:48:35 +03:00
}
2003-02-24 06:09:08 +03:00
/**
2003-03-18 01:42:01 +03:00
* Compare 2 strings up to and including the nth char .
*
* @ note The comparison is case - insensitive .
* */
2007-10-17 23:10:12 +04:00
bool strnequal ( const char * s1 , const char * s2 , size_t n )
1998-11-05 19:48:35 +03:00
{
2005-08-13 02:59:35 +04:00
if ( s1 = = s2 )
2007-10-17 23:10:12 +04:00
return ( true ) ;
2005-08-13 02:59:35 +04:00
if ( ! s1 | | ! s2 | | ! n )
2007-10-17 23:10:12 +04:00
return ( false ) ;
2007-09-13 05:10:01 +04:00
2005-08-13 02:59:35 +04:00
return ( StrnCaseCmp ( s1 , s2 , n ) = = 0 ) ;
1998-11-05 19:48:35 +03:00
}
2003-02-24 06:09:08 +03:00
/**
2002-10-02 23:12:14 +04:00
Convert a string to " normal " form .
2003-02-24 06:09:08 +03:00
* */
2002-10-02 23:12:14 +04:00
2004-05-07 22:37:47 +04:00
void strnorm ( char * s , int case_default )
1998-11-05 19:48:35 +03:00
{
2002-10-02 23:12:14 +04:00
if ( case_default = = CASE_UPPER )
2003-07-03 23:11:31 +04:00
strupper_m ( s ) ;
2002-10-02 23:12:14 +04:00
else
2003-07-03 23:11:31 +04:00
strlower_m ( s ) ;
1998-11-05 19:48:35 +03:00
}
2003-02-24 06:09:08 +03:00
/**
2007-03-31 02:25:08 +04:00
* Skip past some strings in a buffer - old version - no checks .
* * */
2002-10-02 23:12:14 +04:00
2007-04-03 00:10:21 +04:00
char * push_skip_string ( char * buf )
1998-11-05 19:48:35 +03:00
{
2007-04-03 00:10:21 +04:00
buf + = strlen ( buf ) + 1 ;
2001-07-04 11:15:53 +04:00
return ( buf ) ;
1998-11-05 19:48:35 +03:00
}
2007-03-31 02:25:08 +04:00
/**
2007-04-03 00:10:21 +04:00
Skip past a string in a buffer . Buffer may not be
2007-03-31 02:25:08 +04:00
null terminated . end_ptr points to the first byte after
then end of the buffer .
* */
2007-04-03 00:10:21 +04:00
char * skip_string ( const char * base , size_t len , char * buf )
2007-03-31 02:25:08 +04:00
{
const char * end_ptr = base + len ;
if ( end_ptr < base | | ! base | | ! buf | | buf > = end_ptr ) {
return NULL ;
}
2007-04-03 00:10:21 +04:00
/* Skip the string */
while ( * buf ) {
2007-03-31 02:25:08 +04:00
buf + + ;
2007-04-03 00:10:21 +04:00
if ( buf > = end_ptr ) {
return NULL ;
}
2007-03-31 02:25:08 +04:00
}
2007-04-03 00:10:21 +04:00
/* Skip the '\0' */
buf + + ;
2007-03-31 02:25:08 +04:00
return buf ;
}
2003-02-24 06:09:08 +03:00
/**
1998-11-05 19:48:35 +03:00
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 .
2003-02-24 06:09:08 +03:00
* */
2002-10-02 23:12:14 +04:00
1998-11-17 23:50:07 +03:00
size_t str_charnum ( const char * s )
1998-11-05 19:48:35 +03:00
{
2008-04-30 01:36:24 +04:00
size_t ret , converted_size ;
2007-09-13 05:10:01 +04:00
smb_ucs2_t * tmpbuf2 = NULL ;
2009-07-17 05:13:46 +04:00
if ( ! push_ucs2_talloc ( talloc_tos ( ) , & tmpbuf2 , s , & converted_size ) ) {
2007-09-13 05:10:01 +04:00
return 0 ;
}
ret = strlen_w ( tmpbuf2 ) ;
2009-03-19 04:20:11 +03:00
TALLOC_FREE ( tmpbuf2 ) ;
2007-09-13 05:10:01 +04:00
return ret ;
2003-01-15 21:57:41 +03:00
}
2007-10-17 23:10:12 +04:00
bool trim_char ( char * s , char cfront , char cback )
2003-09-05 23:59:55 +04:00
{
2007-10-17 23:10:12 +04:00
bool ret = false ;
2003-09-05 23:59:55 +04:00
char * ep ;
char * fp = s ;
/* Ignore null or empty strings. */
if ( ! s | | ( s [ 0 ] = = ' \0 ' ) )
2007-10-17 23:10:12 +04:00
return false ;
2003-09-05 23:59:55 +04:00
if ( cfront ) {
while ( * fp & & * fp = = cfront )
fp + + ;
if ( ! * fp ) {
/* We ate the string. */
s [ 0 ] = ' \0 ' ;
2007-10-17 23:10:12 +04:00
return true ;
2003-09-05 23:59:55 +04:00
}
if ( fp ! = s )
2007-10-17 23:10:12 +04:00
ret = true ;
2003-09-05 23:59:55 +04:00
}
ep = fp + strlen ( fp ) - 1 ;
if ( cback ) {
/* Attempt ascii only. Bail for mb strings. */
while ( ( ep > = fp ) & & ( * ep = = cback ) ) {
2007-10-17 23:10:12 +04:00
ret = true ;
2003-09-05 23:59:55 +04:00
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 ' ;
2007-10-17 23:10:12 +04:00
return true ;
2003-09-05 23:59:55 +04:00
}
}
ep [ 1 ] = ' \0 ' ;
memmove ( s , fp , ep - fp + 2 ) ;
return ret ;
}
2003-02-24 06:09:08 +03:00
/**
2002-10-02 23:12:14 +04:00
Safe string copy into a known length string . maxlength does not
include the terminating zero .
2003-02-24 06:09:08 +03:00
* */
1999-12-13 16:27:58 +03:00
2011-03-22 07:30:43 +03:00
char * safe_strcpy_fn ( char * dest ,
const char * src ,
size_t maxlength )
1998-11-05 19:48:35 +03:00
{
2001-07-04 11:15:53 +04:00
size_t len ;
1998-11-05 19:48:35 +03:00
2001-07-04 11:15:53 +04:00
if ( ! dest ) {
2011-03-22 07:30:43 +03:00
smb_panic ( " ERROR: NULL dest in safe_strcpy " ) ;
2001-07-04 11:15:53 +04:00
}
1998-11-05 19:48:35 +03:00
2001-07-04 11:15:53 +04:00
if ( ! src ) {
* dest = 0 ;
return dest ;
2007-10-17 23:10:12 +04:00
}
1998-11-05 19:48:35 +03:00
2003-03-19 23:50:56 +03:00
len = strnlen ( src , maxlength + 1 ) ;
1998-11-05 19:48:35 +03:00
2001-07-04 11:15:53 +04:00
if ( len > maxlength ) {
2007-10-17 23:10:12 +04:00
DEBUG ( 0 , ( " ERROR: string overflow by "
" %lu (%lu - %lu) in safe_strcpy [%.50s] \n " ,
( unsigned long ) ( len - maxlength ) , ( unsigned long ) len ,
2003-11-03 17:34:25 +03:00
( unsigned long ) maxlength , src ) ) ;
2001-07-04 11:15:53 +04:00
len = maxlength ;
}
2007-10-17 23:10:12 +04:00
2001-07-04 11:15:53 +04:00
memmove ( dest , src , len ) ;
dest [ len ] = 0 ;
return dest ;
2007-10-17 23:10:12 +04:00
}
1998-11-05 19:48:35 +03:00
2003-02-24 06:09:08 +03:00
/**
2002-10-02 23:12:14 +04:00
Safe string cat into a string . maxlength does not
include the terminating zero .
2003-02-24 06:09:08 +03:00
* */
2011-03-22 07:30:43 +03:00
char * safe_strcat_fn ( char * dest ,
const char * src ,
size_t maxlength )
1998-11-05 19:48:35 +03:00
{
2001-07-04 11:15:53 +04:00
size_t src_len , dest_len ;
1998-11-05 19:48:35 +03:00
2001-07-04 11:15:53 +04:00
if ( ! dest ) {
2011-03-22 07:30:43 +03:00
smb_panic ( " ERROR: NULL dest in safe_strcat " ) ;
2001-07-04 11:15:53 +04:00
}
1998-11-05 19:48:35 +03:00
2002-10-02 23:12:14 +04:00
if ( ! src )
2001-07-04 11:15:53 +04:00
return dest ;
2007-10-17 23:10:12 +04:00
2003-03-19 23:50:56 +03:00
src_len = strnlen ( src , maxlength + 1 ) ;
dest_len = strnlen ( dest , maxlength + 1 ) ;
2003-02-07 07:11:36 +03:00
2001-07-04 11:15:53 +04:00
if ( src_len + dest_len > maxlength ) {
2007-10-17 23:10:12 +04:00
DEBUG ( 0 , ( " ERROR: string overflow by %d "
" in safe_strcat [%.50s] \n " ,
2001-07-04 11:15:53 +04:00
( int ) ( src_len + dest_len - maxlength ) , src ) ) ;
2003-02-07 07:11:36 +03:00
if ( maxlength > dest_len ) {
memcpy ( & dest [ dest_len ] , src , maxlength - dest_len ) ;
}
dest [ maxlength ] = 0 ;
return NULL ;
2001-07-04 11:15:53 +04:00
}
2003-03-18 04:48:11 +03:00
2001-07-04 11:15:53 +04:00
memcpy ( & dest [ dest_len ] , src , src_len ) ;
dest [ dest_len + src_len ] = 0 ;
return dest ;
1998-11-05 19:48:35 +03:00
}
2003-02-24 06:09:08 +03:00
/**
1999-12-13 16:27:58 +03:00
Like strncpy but always null terminates . Make sure there is room !
The variable n should always be one less than the available size .
2003-02-24 06:09:08 +03:00
* */
2011-03-22 12:57:36 +03:00
char * StrnCpy ( char * dest , const char * src , size_t n )
1998-11-05 19:48:35 +03:00
{
2001-07-04 11:15:53 +04:00
char * d = dest ;
2003-03-18 04:48:11 +03:00
2003-11-18 22:15:29 +03:00
if ( ! dest ) {
2011-03-22 07:30:43 +03:00
smb_panic ( " ERROR: NULL dest in StrnCpy " ) ;
2003-11-18 22:15:29 +03:00
}
2001-07-04 11:15:53 +04:00
if ( ! src ) {
* dest = 0 ;
return ( dest ) ;
}
2007-10-17 23:10:12 +04:00
2003-03-24 12:54:13 +03:00
while ( n - - & & ( * d = * src ) ) {
d + + ;
src + + ;
}
2001-07-04 11:15:53 +04:00
* d = 0 ;
return ( dest ) ;
1998-11-05 19:48:35 +03:00
}
2003-02-24 06:09:08 +03:00
/**
2002-10-02 23:12:14 +04:00
Check if a string is part of a list .
2003-02-24 06:09:08 +03:00
* */
2002-10-02 23:12:14 +04:00
2007-10-17 23:10:12 +04:00
bool in_list ( const char * s , const char * list , bool casesensitive )
1998-11-05 19:48:35 +03:00
{
2007-12-08 04:32:32 +03:00
char * tok = NULL ;
2007-10-17 23:10:12 +04:00
bool ret = false ;
2007-12-08 04:32:32 +03:00
TALLOC_CTX * frame ;
2002-10-02 23:12:14 +04:00
2007-12-08 04:32:32 +03:00
if ( ! list ) {
2007-10-17 23:10:12 +04:00
return false ;
2007-09-13 05:10:01 +04:00
}
2007-12-08 04:32:32 +03:00
frame = talloc_stackframe ( ) ;
while ( next_token_talloc ( frame , & list , & tok , LIST_SEP ) ) {
2002-10-02 23:12:14 +04:00
if ( casesensitive ) {
2007-09-13 05:10:01 +04:00
if ( strcmp ( tok , s ) = = 0 ) {
2007-10-17 23:10:12 +04:00
ret = true ;
2007-09-13 05:10:01 +04:00
break ;
}
2002-10-02 23:12:14 +04:00
} else {
2007-09-13 05:10:01 +04:00
if ( StrCaseCmp ( tok , s ) = = 0 ) {
2007-10-17 23:10:12 +04:00
ret = true ;
2007-09-13 05:10:01 +04:00
break ;
}
2002-10-02 23:12:14 +04:00
}
}
2007-12-08 04:32:32 +03:00
TALLOC_FREE ( frame ) ;
2007-09-13 05:10:01 +04:00
return ret ;
1998-11-05 19:48:35 +03:00
}
/* this is used to prevent lots of mallocs of size 1 */
2007-11-24 19:27:54 +03:00
static const char null_string [ ] = " " ;
1998-11-05 19:48:35 +03:00
2003-02-24 06:09:08 +03:00
/**
2002-10-02 23:12:14 +04:00
Set a string value , allocing the space for the string
2003-02-24 06:09:08 +03:00
* */
2002-10-02 23:12:14 +04:00
2007-10-17 23:10:12 +04:00
static bool string_init ( char * * dest , const char * src )
1998-11-05 19:48:35 +03:00
{
2002-10-02 23:12:14 +04:00
size_t l ;
2006-04-08 21:25:31 +04:00
2007-10-17 23:10:12 +04:00
if ( ! src )
2002-10-02 23:12:14 +04:00
src = " " ;
l = strlen ( src ) ;
if ( l = = 0 ) {
2006-04-08 21:25:31 +04:00
* dest = CONST_DISCARD ( char * , null_string ) ;
2002-10-02 23:12:14 +04:00
} else {
2004-12-07 21:25:53 +03:00
( * dest ) = SMB_STRDUP ( src ) ;
2002-10-02 23:12:14 +04:00
if ( ( * dest ) = = NULL ) {
DEBUG ( 0 , ( " Out of memory in string_init \n " ) ) ;
2007-10-17 23:10:12 +04:00
return false ;
2002-10-02 23:12:14 +04:00
}
}
2007-10-17 23:10:12 +04:00
return ( true ) ;
1998-11-05 19:48:35 +03:00
}
2003-02-24 06:09:08 +03:00
/**
2002-10-02 23:12:14 +04:00
Free a string value .
2003-02-24 06:09:08 +03:00
* */
2002-10-02 23:12:14 +04:00
1998-11-05 19:48:35 +03:00
void string_free ( char * * s )
{
2002-10-02 23:12:14 +04:00
if ( ! s | | ! ( * s ) )
return ;
if ( * s = = null_string )
* s = NULL ;
SAFE_FREE ( * s ) ;
1998-11-05 19:48:35 +03:00
}
2003-02-24 06:09:08 +03:00
/**
2002-10-02 23:12:14 +04:00
Set a string value , deallocating any existing space , and allocing the space
for the string
2003-02-24 06:09:08 +03:00
* */
2002-10-02 23:12:14 +04:00
2007-10-17 23:10:12 +04:00
bool string_set ( char * * dest , const char * src )
1998-11-05 19:48:35 +03:00
{
2002-10-02 23:12:14 +04:00
string_free ( dest ) ;
return ( string_init ( dest , src ) ) ;
1998-11-05 19:48:35 +03:00
}
1999-12-13 16:27:58 +03:00
void fstring_sub ( char * s , const char * pattern , const char * insert )
{
string_sub ( s , pattern , insert , sizeof ( fstring ) ) ;
}
2003-02-24 06:09:08 +03:00
/**
2007-11-15 03:05:42 +03:00
Similar to string_sub2 , but it will accept only allocated strings
2002-10-02 23:12:14 +04:00
and may realloc them so pay attention at what you pass on no
2007-12-07 23:02:44 +03:00
pointers inside strings , no const may be passed
2002-10-02 23:12:14 +04:00
as string .
2003-02-24 06:09:08 +03:00
* */
2002-07-15 14:35:28 +04:00
2007-11-15 03:05:42 +03:00
char * realloc_string_sub2 ( char * string ,
const char * pattern ,
const char * insert ,
bool remove_unsafe_characters ,
bool allow_trailing_dollar )
2002-07-15 14:35:28 +04:00
{
char * p , * in ;
char * s ;
ssize_t ls , lp , li , ld , i ;
if ( ! insert | | ! pattern | | ! * pattern | | ! string | | ! * string )
return NULL ;
s = string ;
2004-12-07 21:25:53 +03:00
in = SMB_STRDUP ( insert ) ;
2002-07-15 14:35:28 +04:00
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 ' $ ' :
2007-11-15 03:05:42 +03:00
/* allow a trailing $
* ( as in machine accounts ) */
if ( allow_trailing_dollar & & ( i = = li - 1 ) ) {
break ;
}
2011-02-03 01:41:17 +03:00
case ' ` ' :
case ' " ' :
case ' \' ' :
case ' ; ' :
2002-07-15 14:35:28 +04:00
case ' % ' :
case ' \r ' :
case ' \n ' :
2007-11-15 03:05:42 +03:00
if ( remove_unsafe_characters ) {
in [ i ] = ' _ ' ;
break ;
}
2002-07-15 14:35:28 +04:00
default :
/* ok */
break ;
}
}
2007-10-17 23:10:12 +04:00
2004-03-09 03:17:14 +03:00
while ( ( p = strstr_m ( s , pattern ) ) ) {
2002-07-15 14:35:28 +04:00
if ( ld > 0 ) {
2003-08-26 00:42:24 +04:00
int offset = PTR_DIFF ( s , string ) ;
2006-07-30 20:36:56 +04:00
string = ( char * ) SMB_REALLOC ( string , ls + ld + 1 ) ;
r13915: Fixed a very interesting class of realloc() bugs found by Coverity.
realloc can return NULL in one of two cases - (1) the realloc failed,
(2) realloc succeeded but the new size requested was zero, in which
case this is identical to a free() call.
The error paths dealing with these two cases should be different,
but mostly weren't. Secondly the standard idiom for dealing with
realloc when you know the new size is non-zero is the following :
tmp = realloc(p, size);
if (!tmp) {
SAFE_FREE(p);
return error;
} else {
p = tmp;
}
However, there were *many* *many* places in Samba where we were
using the old (broken) idiom of :
p = realloc(p, size)
if (!p) {
return error;
}
which will leak the memory pointed to by p on realloc fail.
This commit (hopefully) fixes all these cases by moving to
a standard idiom of :
p = SMB_REALLOC(p, size)
if (!p) {
return error;
}
Where if the realloc returns null due to the realloc failing
or size == 0 we *guarentee* that the storage pointed to by p
has been freed. This allows me to remove a lot of code that
was dealing with the standard (more verbose) method that required
a tmp pointer. This is almost always what you want. When a
realloc fails you never usually want the old memory, you
want to free it and get into your error processing asap.
For the 11 remaining cases where we really do need to keep the
old pointer I have invented the new macro SMB_REALLOC_KEEP_OLD_ON_ERROR,
which can be used as follows :
tmp = SMB_REALLOC_KEEP_OLD_ON_ERROR(p, size);
if (!tmp) {
SAFE_FREE(p);
return error;
} else {
p = tmp;
}
SMB_REALLOC_KEEP_OLD_ON_ERROR guarentees never to free the
pointer p, even on size == 0 or realloc fail. All this is
done by a hidden extra argument to Realloc(), BOOL free_old_on_error
which is set appropriately by the SMB_REALLOC and SMB_REALLOC_KEEP_OLD_ON_ERROR
macros (and their array counterparts).
It remains to be seen what this will do to our Coverity bug count :-).
Jeremy.
(This used to be commit 1d710d06a214f3f1740e80e0bffd6aab44aac2b0)
2006-03-07 09:31:04 +03:00
if ( ! string ) {
2007-10-17 23:10:12 +04:00
DEBUG ( 0 , ( " realloc_string_sub: "
" out of memory! \n " ) ) ;
2002-07-15 14:35:28 +04:00
SAFE_FREE ( in ) ;
return NULL ;
}
r13915: Fixed a very interesting class of realloc() bugs found by Coverity.
realloc can return NULL in one of two cases - (1) the realloc failed,
(2) realloc succeeded but the new size requested was zero, in which
case this is identical to a free() call.
The error paths dealing with these two cases should be different,
but mostly weren't. Secondly the standard idiom for dealing with
realloc when you know the new size is non-zero is the following :
tmp = realloc(p, size);
if (!tmp) {
SAFE_FREE(p);
return error;
} else {
p = tmp;
}
However, there were *many* *many* places in Samba where we were
using the old (broken) idiom of :
p = realloc(p, size)
if (!p) {
return error;
}
which will leak the memory pointed to by p on realloc fail.
This commit (hopefully) fixes all these cases by moving to
a standard idiom of :
p = SMB_REALLOC(p, size)
if (!p) {
return error;
}
Where if the realloc returns null due to the realloc failing
or size == 0 we *guarentee* that the storage pointed to by p
has been freed. This allows me to remove a lot of code that
was dealing with the standard (more verbose) method that required
a tmp pointer. This is almost always what you want. When a
realloc fails you never usually want the old memory, you
want to free it and get into your error processing asap.
For the 11 remaining cases where we really do need to keep the
old pointer I have invented the new macro SMB_REALLOC_KEEP_OLD_ON_ERROR,
which can be used as follows :
tmp = SMB_REALLOC_KEEP_OLD_ON_ERROR(p, size);
if (!tmp) {
SAFE_FREE(p);
return error;
} else {
p = tmp;
}
SMB_REALLOC_KEEP_OLD_ON_ERROR guarentees never to free the
pointer p, even on size == 0 or realloc fail. All this is
done by a hidden extra argument to Realloc(), BOOL free_old_on_error
which is set appropriately by the SMB_REALLOC and SMB_REALLOC_KEEP_OLD_ON_ERROR
macros (and their array counterparts).
It remains to be seen what this will do to our Coverity bug count :-).
Jeremy.
(This used to be commit 1d710d06a214f3f1740e80e0bffd6aab44aac2b0)
2006-03-07 09:31:04 +03:00
p = string + offset + ( p - s ) ;
2002-07-15 14:35:28 +04:00
}
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 ;
}
2007-11-15 03:05:42 +03:00
char * realloc_string_sub ( char * string ,
const char * pattern ,
const char * insert )
{
return realloc_string_sub2 ( string , pattern , insert , true , false ) ;
}
2007-11-13 02:02:50 +03:00
/*
* Internal guts of talloc_string_sub and talloc_all_string_sub .
2007-11-20 05:56:22 +03:00
* talloc version of string_sub2 .
2007-11-13 02:02:50 +03:00
*/
2005-12-18 21:06:15 +03:00
2007-11-20 05:56:22 +03:00
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 )
2005-12-18 21:06:15 +03:00
{
char * p , * in ;
char * s ;
char * string ;
ssize_t ls , lp , li , ld , i ;
2008-07-23 19:07:56 +04:00
if ( ! insert | | ! pattern | | ! * pattern | | ! src ) {
2005-12-18 21:06:15 +03:00
return NULL ;
2007-11-13 02:02:50 +03:00
}
2005-12-18 21:06:15 +03:00
string = talloc_strdup ( mem_ctx , src ) ;
if ( string = = NULL ) {
2007-11-20 05:56:22 +03:00
DEBUG ( 0 , ( " talloc_string_sub2: "
2007-11-13 02:02:50 +03:00
" talloc_strdup failed \n " ) ) ;
2005-12-18 21:06:15 +03:00
return NULL ;
}
s = string ;
in = SMB_STRDUP ( insert ) ;
if ( ! in ) {
2007-11-20 05:56:22 +03:00
DEBUG ( 0 , ( " talloc_string_sub2: ENOMEM \n " ) ) ;
2005-12-18 21:06:15 +03:00
return NULL ;
}
ls = ( ssize_t ) strlen ( s ) ;
lp = ( ssize_t ) strlen ( pattern ) ;
li = ( ssize_t ) strlen ( insert ) ;
ld = li - lp ;
2007-11-13 02:02:50 +03:00
2007-11-20 05:56:22 +03:00
for ( i = 0 ; i < li ; i + + ) {
switch ( in [ i ] ) {
case ' $ ' :
/* allow a trailing $
* ( as in machine accounts ) */
if ( allow_trailing_dollar & & ( i = = li - 1 ) ) {
break ;
}
2011-02-03 01:41:17 +03:00
case ' ` ' :
case ' " ' :
case ' \' ' :
case ' ; ' :
2007-11-20 05:56:22 +03:00
case ' % ' :
case ' \r ' :
case ' \n ' :
if ( remove_unsafe_characters ) {
2007-11-13 02:02:50 +03:00
in [ i ] = ' _ ' ;
break ;
2007-11-20 05:56:22 +03:00
}
default :
/* ok */
break ;
2005-12-18 21:06:15 +03:00
}
}
2007-10-17 23:10:12 +04:00
2005-12-18 21:06:15 +03:00
while ( ( p = strstr_m ( s , pattern ) ) ) {
if ( ld > 0 ) {
int offset = PTR_DIFF ( s , string ) ;
2006-07-30 20:36:56 +04:00
string = ( char * ) TALLOC_REALLOC ( mem_ctx , string ,
ls + ld + 1 ) ;
r13915: Fixed a very interesting class of realloc() bugs found by Coverity.
realloc can return NULL in one of two cases - (1) the realloc failed,
(2) realloc succeeded but the new size requested was zero, in which
case this is identical to a free() call.
The error paths dealing with these two cases should be different,
but mostly weren't. Secondly the standard idiom for dealing with
realloc when you know the new size is non-zero is the following :
tmp = realloc(p, size);
if (!tmp) {
SAFE_FREE(p);
return error;
} else {
p = tmp;
}
However, there were *many* *many* places in Samba where we were
using the old (broken) idiom of :
p = realloc(p, size)
if (!p) {
return error;
}
which will leak the memory pointed to by p on realloc fail.
This commit (hopefully) fixes all these cases by moving to
a standard idiom of :
p = SMB_REALLOC(p, size)
if (!p) {
return error;
}
Where if the realloc returns null due to the realloc failing
or size == 0 we *guarentee* that the storage pointed to by p
has been freed. This allows me to remove a lot of code that
was dealing with the standard (more verbose) method that required
a tmp pointer. This is almost always what you want. When a
realloc fails you never usually want the old memory, you
want to free it and get into your error processing asap.
For the 11 remaining cases where we really do need to keep the
old pointer I have invented the new macro SMB_REALLOC_KEEP_OLD_ON_ERROR,
which can be used as follows :
tmp = SMB_REALLOC_KEEP_OLD_ON_ERROR(p, size);
if (!tmp) {
SAFE_FREE(p);
return error;
} else {
p = tmp;
}
SMB_REALLOC_KEEP_OLD_ON_ERROR guarentees never to free the
pointer p, even on size == 0 or realloc fail. All this is
done by a hidden extra argument to Realloc(), BOOL free_old_on_error
which is set appropriately by the SMB_REALLOC and SMB_REALLOC_KEEP_OLD_ON_ERROR
macros (and their array counterparts).
It remains to be seen what this will do to our Coverity bug count :-).
Jeremy.
(This used to be commit 1d710d06a214f3f1740e80e0bffd6aab44aac2b0)
2006-03-07 09:31:04 +03:00
if ( ! string ) {
2005-12-18 21:06:15 +03:00
DEBUG ( 0 , ( " talloc_string_sub: out of "
" memory! \n " ) ) ;
SAFE_FREE ( in ) ;
return NULL ;
}
r13915: Fixed a very interesting class of realloc() bugs found by Coverity.
realloc can return NULL in one of two cases - (1) the realloc failed,
(2) realloc succeeded but the new size requested was zero, in which
case this is identical to a free() call.
The error paths dealing with these two cases should be different,
but mostly weren't. Secondly the standard idiom for dealing with
realloc when you know the new size is non-zero is the following :
tmp = realloc(p, size);
if (!tmp) {
SAFE_FREE(p);
return error;
} else {
p = tmp;
}
However, there were *many* *many* places in Samba where we were
using the old (broken) idiom of :
p = realloc(p, size)
if (!p) {
return error;
}
which will leak the memory pointed to by p on realloc fail.
This commit (hopefully) fixes all these cases by moving to
a standard idiom of :
p = SMB_REALLOC(p, size)
if (!p) {
return error;
}
Where if the realloc returns null due to the realloc failing
or size == 0 we *guarentee* that the storage pointed to by p
has been freed. This allows me to remove a lot of code that
was dealing with the standard (more verbose) method that required
a tmp pointer. This is almost always what you want. When a
realloc fails you never usually want the old memory, you
want to free it and get into your error processing asap.
For the 11 remaining cases where we really do need to keep the
old pointer I have invented the new macro SMB_REALLOC_KEEP_OLD_ON_ERROR,
which can be used as follows :
tmp = SMB_REALLOC_KEEP_OLD_ON_ERROR(p, size);
if (!tmp) {
SAFE_FREE(p);
return error;
} else {
p = tmp;
}
SMB_REALLOC_KEEP_OLD_ON_ERROR guarentees never to free the
pointer p, even on size == 0 or realloc fail. All this is
done by a hidden extra argument to Realloc(), BOOL free_old_on_error
which is set appropriately by the SMB_REALLOC and SMB_REALLOC_KEEP_OLD_ON_ERROR
macros (and their array counterparts).
It remains to be seen what this will do to our Coverity bug count :-).
Jeremy.
(This used to be commit 1d710d06a214f3f1740e80e0bffd6aab44aac2b0)
2006-03-07 09:31:04 +03:00
p = string + offset + ( p - s ) ;
2005-12-18 21:06:15 +03:00
}
if ( li ! = lp ) {
memmove ( p + li , p + lp , strlen ( p + lp ) + 1 ) ;
}
memcpy ( p , in , li ) ;
s = p + li ;
ls + = ld ;
2007-11-20 05:56:22 +03:00
if ( replace_once ) {
break ;
}
2005-12-18 21:06:15 +03:00
}
SAFE_FREE ( in ) ;
return string ;
}
2007-11-13 02:02:50 +03:00
/* Same as string_sub, but returns a talloc'ed string */
2007-11-20 05:56:22 +03:00
char * talloc_string_sub ( TALLOC_CTX * mem_ctx ,
const char * src ,
const char * pattern ,
const char * insert )
2007-11-13 02:02:50 +03:00
{
2007-11-20 05:56:22 +03:00
return talloc_string_sub2 ( mem_ctx , src , pattern , insert ,
true , false , false ) ;
2007-11-13 02:02:50 +03:00
}
char * talloc_all_string_sub ( TALLOC_CTX * ctx ,
const char * src ,
const char * pattern ,
const char * insert )
{
2007-11-20 05:56:22 +03:00
return talloc_string_sub2 ( ctx , src , pattern , insert ,
false , false , false ) ;
2001-11-04 21:26:53 +03:00
}
2003-02-24 06:09:08 +03:00
/**
2002-08-17 21:00:51 +04:00
Write an octal as a string .
2003-02-24 06:09:08 +03:00
* */
2002-08-17 21:00:51 +04:00
2007-11-24 19:27:54 +03:00
char * octal_string ( int i )
1999-02-16 21:02:50 +03:00
{
2007-11-24 19:27:54 +03:00
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 ;
1999-02-16 21:02:50 +03:00
}
1999-12-13 16:27:58 +03:00
2003-02-24 06:09:08 +03:00
/**
2002-08-17 21:00:51 +04:00
Truncate a string at a specified length .
2003-02-24 06:09:08 +03:00
* */
2002-08-17 21:00:51 +04:00
2003-03-18 04:48:11 +03:00
char * string_truncate ( char * s , unsigned int length )
1999-02-16 21:02:50 +03:00
{
2002-08-17 21:00:51 +04:00
if ( s & & strlen ( s ) > length )
1999-12-13 16:27:58 +03:00
s [ length ] = 0 ;
return s ;
1999-02-16 21:02:50 +03:00
}
2001-07-04 11:15:53 +04:00
2003-07-03 00:01:51 +04:00
/***********************************************************************
Return the equivalent of doing strrchr ' n ' times - always going
backwards .
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
char * strnrchr_m ( const char * s , char c , unsigned int n )
{
2007-09-13 05:10:01 +04:00
smb_ucs2_t * ws = NULL ;
char * s2 = NULL ;
2003-07-03 00:01:51 +04:00
smb_ucs2_t * p ;
2007-09-13 05:10:01 +04:00
char * ret ;
2008-04-30 01:36:24 +04:00
size_t converted_size ;
2003-07-03 00:01:51 +04:00
2009-07-17 05:13:46 +04:00
if ( ! push_ucs2_talloc ( talloc_tos ( ) , & ws , s , & converted_size ) ) {
2007-09-13 05:10:01 +04:00
/* Too hard to try and get right. */
return NULL ;
}
2003-07-03 00:01:51 +04:00
p = strnrchr_w ( ws , UCS2_CHAR ( c ) , n ) ;
2007-09-13 05:10:01 +04:00
if ( ! p ) {
2009-03-19 04:20:11 +03:00
TALLOC_FREE ( ws ) ;
2003-07-03 00:01:51 +04:00
return NULL ;
2007-09-13 05:10:01 +04:00
}
2003-07-03 00:01:51 +04:00
* p = 0 ;
2009-07-17 05:13:46 +04:00
if ( ! pull_ucs2_talloc ( talloc_tos ( ) , & s2 , ws , & converted_size ) ) {
2009-03-19 04:20:11 +03:00
TALLOC_FREE ( ws ) ;
2007-09-13 05:10:01 +04:00
/* Too hard to try and get right. */
return NULL ;
}
ret = ( char * ) ( s + strlen ( s2 ) ) ;
2009-03-19 04:20:11 +03:00
TALLOC_FREE ( ws ) ;
TALLOC_FREE ( s2 ) ;
2007-09-13 05:10:01 +04:00
return ret ;
2003-07-03 00:01:51 +04:00
}
2011-04-14 06:42:52 +04:00
static bool unix_strlower ( const char * src , size_t srclen , char * dest , size_t destlen )
{
size_t size ;
smb_ucs2_t * buffer = NULL ;
bool ret ;
if ( ! convert_string_talloc ( talloc_tos ( ) , CH_UNIX , CH_UTF16LE , src , srclen ,
( void * * ) ( void * ) & buffer , & size ) )
{
smb_panic ( " failed to create UCS2 buffer " ) ;
}
if ( ! strlower_w ( buffer ) & & ( dest = = src ) ) {
TALLOC_FREE ( buffer ) ;
return srclen ;
}
ret = convert_string ( CH_UTF16LE , CH_UNIX , buffer , size , dest , destlen , & size ) ;
TALLOC_FREE ( buffer ) ;
return ret ;
}
2003-02-24 06:09:08 +03:00
/**
2002-08-17 21:00:51 +04:00
Convert a string to lower case .
2003-02-24 06:09:08 +03:00
* */
2002-08-17 21:00:51 +04:00
2001-07-04 11:15:53 +04:00
void strlower_m ( char * s )
{
2004-02-04 23:28:51 +03:00
size_t len ;
2004-09-01 21:39:27 +04:00
int errno_save ;
2004-02-04 23:28:51 +03:00
2002-04-11 13:56:38 +04:00
/* 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 ) */
2002-08-17 21:00:51 +04:00
2004-01-08 02:21:36 +03:00
while ( * s & & ! ( ( ( unsigned char ) s [ 0 ] ) & 0x80 ) ) {
2005-12-27 23:52:36 +03:00
* s = tolower_ascii ( ( unsigned char ) * s ) ;
2003-01-17 09:40:12 +03:00
s + + ;
}
2002-04-11 13:56:38 +04:00
2002-08-17 21:00:51 +04:00
if ( ! * s )
return ;
2002-04-11 13:56:38 +04:00
2001-07-04 11:15:53 +04:00
/* I assume that lowercased string takes the same number of bytes
* as source string even in UTF - 8 encoding . ( VIV ) */
2004-02-04 23:28:51 +03:00
len = strlen ( s ) + 1 ;
2004-09-01 21:39:27 +04:00
errno_save = errno ;
2004-02-04 23:28:51 +03:00
errno = 0 ;
2007-10-17 23:10:12 +04:00
unix_strlower ( s , len , s , len ) ;
2004-02-04 23:28:51 +03:00
/* Catch mb conversion errors that may not terminate. */
if ( errno )
s [ len - 1 ] = ' \0 ' ;
2004-09-01 21:39:27 +04:00
errno = errno_save ;
2001-07-04 11:15:53 +04:00
}
2011-04-14 06:42:52 +04:00
static bool unix_strupper ( const char * src , size_t srclen , char * dest , size_t destlen )
{
size_t size ;
smb_ucs2_t * buffer ;
bool ret ;
if ( ! push_ucs2_talloc ( talloc_tos ( ) , & buffer , src , & size ) ) {
return ( size_t ) - 1 ;
}
if ( ! strupper_w ( buffer ) & & ( dest = = src ) ) {
TALLOC_FREE ( buffer ) ;
return srclen ;
}
ret = convert_string ( CH_UTF16LE , CH_UNIX , buffer , size , dest , destlen , & size ) ;
TALLOC_FREE ( buffer ) ;
return ret ;
}
2003-02-24 06:09:08 +03:00
/**
2002-08-17 21:00:51 +04:00
Convert a string to upper case .
2003-02-24 06:09:08 +03:00
* */
2002-08-17 21:00:51 +04:00
2001-07-04 11:15:53 +04:00
void strupper_m ( char * s )
{
2004-02-04 23:28:51 +03:00
size_t len ;
2004-09-01 21:39:27 +04:00
int errno_save ;
2004-02-04 23:28:51 +03:00
2002-04-11 13:56:38 +04:00
/* 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 ) */
2002-08-17 21:00:51 +04:00
2004-01-08 02:21:36 +03:00
while ( * s & & ! ( ( ( unsigned char ) s [ 0 ] ) & 0x80 ) ) {
2008-01-10 04:32:26 +03:00
* s = toupper_ascii_fast ( ( unsigned char ) * s ) ;
2003-01-17 09:40:12 +03:00
s + + ;
}
2002-04-11 13:56:38 +04:00
2002-08-17 21:00:51 +04:00
if ( ! * s )
return ;
2002-04-11 13:56:38 +04:00
2001-07-04 11:15:53 +04:00
/* I assume that lowercased string takes the same number of bytes
* as source string even in multibyte encoding . ( VIV ) */
2004-02-04 23:28:51 +03:00
len = strlen ( s ) + 1 ;
2004-09-01 21:39:27 +04:00
errno_save = errno ;
2004-02-04 23:28:51 +03:00
errno = 0 ;
2007-10-17 23:10:12 +04:00
unix_strupper ( s , len , s , len ) ;
2004-02-04 23:28:51 +03:00
/* Catch mb conversion errors that may not terminate. */
if ( errno )
s [ len - 1 ] = ' \0 ' ;
2004-09-01 21:39:27 +04:00
errno = errno_save ;
2001-07-04 11:15:53 +04:00
}
2001-12-10 03:39:01 +03:00
2003-02-24 06:09:08 +03:00
/**
2002-10-02 23:12:14 +04:00
Just a typesafety wrapper for snprintf into a fstring .
2003-02-24 06:09:08 +03:00
* */
2002-08-17 21:00:51 +04:00
2003-04-15 23:51:17 +04:00
int fstr_sprintf ( fstring s , const char * fmt , . . . )
2002-01-15 04:37:12 +03:00
{
va_list ap ;
int ret ;
va_start ( ap , fmt ) ;
2002-01-16 05:42:07 +03:00
ret = vsnprintf ( s , FSTRING_LEN , fmt , ap ) ;
2002-01-15 04:37:12 +03:00
va_end ( ap ) ;
return ret ;
}
2003-04-15 23:51:17 +04:00
2003-02-24 06:09:08 +03:00
/**
2002-07-15 14:35:28 +04:00
List of Strings manipulation functions
2003-02-24 06:09:08 +03:00
* */
2002-07-15 14:35:28 +04:00
# define S_LIST_ABS 16 /* List Allocation Block Size */
2003-09-05 09:32:32 +04:00
/******************************************************************************
2008-04-07 12:19:25 +04:00
version of standard_sub_basic ( ) for string lists ; uses talloc_sub_basic ( )
2003-09-05 09:32:32 +04:00
for the work
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
2007-10-17 23:10:12 +04:00
bool str_list_sub_basic ( char * * list , const char * smb_name ,
2006-07-11 22:01:26 +04:00
const char * domain_name )
2003-09-05 09:32:32 +04:00
{
2008-04-07 12:19:25 +04:00
TALLOC_CTX * ctx = list ;
2003-09-05 09:32:32 +04:00
char * s , * tmpstr ;
2007-10-17 23:10:12 +04:00
2003-09-05 09:32:32 +04:00
while ( * list ) {
s = * list ;
2008-04-07 12:19:25 +04:00
tmpstr = talloc_sub_basic ( ctx , smb_name , domain_name , s ) ;
2003-09-05 09:32:32 +04:00
if ( ! tmpstr ) {
2007-10-17 23:10:12 +04:00
DEBUG ( 0 , ( " str_list_sub_basic: "
" alloc_sub_basic() return NULL! \n " ) ) ;
return false ;
2003-09-05 09:32:32 +04:00
}
2008-04-07 12:19:25 +04:00
TALLOC_FREE ( * list ) ;
2003-09-05 09:32:32 +04:00
* list = tmpstr ;
2007-10-17 23:10:12 +04:00
2003-09-05 09:32:32 +04:00
list + + ;
}
2007-10-17 23:10:12 +04:00
return true ;
2003-09-05 09:32:32 +04:00
}
/******************************************************************************
2009-09-25 01:50:05 +04:00
substitute a specific pattern in a string list
2003-09-05 09:32:32 +04:00
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
2007-10-17 23:10:12 +04:00
bool str_list_substitute ( char * * list , const char * pattern , const char * insert )
2002-07-15 14:35:28 +04:00
{
2008-08-21 21:25:02 +04:00
TALLOC_CTX * ctx = list ;
2002-07-15 14:35:28 +04:00
char * p , * s , * t ;
ssize_t ls , lp , li , ld , i , d ;
2002-08-17 21:00:51 +04:00
if ( ! list )
2007-10-17 23:10:12 +04:00
return false ;
2002-08-17 21:00:51 +04:00
if ( ! pattern )
2007-10-17 23:10:12 +04:00
return false ;
2002-08-17 21:00:51 +04:00
if ( ! insert )
2007-10-17 23:10:12 +04:00
return false ;
2002-07-15 14:35:28 +04:00
lp = ( ssize_t ) strlen ( pattern ) ;
li = ( ssize_t ) strlen ( insert ) ;
ld = li - lp ;
2007-10-17 23:10:12 +04:00
2002-08-17 21:00:51 +04:00
while ( * list ) {
2002-07-15 14:35:28 +04:00
s = * list ;
ls = ( ssize_t ) strlen ( s ) ;
2004-03-09 03:17:14 +03:00
while ( ( p = strstr_m ( s , pattern ) ) ) {
2002-07-15 14:35:28 +04:00
t = * list ;
d = p - t ;
2002-08-17 21:00:51 +04:00
if ( ld ) {
2008-08-21 21:25:02 +04:00
t = TALLOC_ARRAY ( ctx , char , ls + ld + 1 ) ;
2002-07-15 14:35:28 +04:00
if ( ! t ) {
2007-10-17 23:10:12 +04:00
DEBUG ( 0 , ( " str_list_substitute: "
" Unable to allocate memory " ) ) ;
return false ;
2002-07-15 14:35:28 +04:00
}
memcpy ( t , * list , d ) ;
memcpy ( t + d + li , p + lp , ls - d - lp + 1 ) ;
2008-08-21 21:25:02 +04:00
TALLOC_FREE ( * list ) ;
2002-07-15 14:35:28 +04:00
* list = t ;
ls + = ld ;
s = t + d + li ;
}
2007-10-17 23:10:12 +04:00
2002-07-15 14:35:28 +04:00
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 ] ;
}
2007-10-17 23:10:12 +04:00
}
2002-07-15 14:35:28 +04:00
}
2007-10-17 23:10:12 +04:00
2002-07-15 14:35:28 +04:00
list + + ;
}
2007-10-17 23:10:12 +04:00
return true ;
2002-07-15 14:35:28 +04:00
}
2002-12-12 02:54:40 +03:00
2003-01-04 11:48:15 +03:00
# define IPSTR_LIST_SEP ","
2003-06-25 21:41:05 +04:00
# define IPSTR_LIST_CHAR ','
2003-01-04 11:48:15 +03:00
/**
* 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
* */
2007-10-25 01:16:54 +04:00
static char * ipstr_list_add ( char * * ipstr_list , const struct ip_service * service )
2003-01-04 11:48:15 +03:00
{
2007-10-17 23:10:12 +04:00
char * new_ipstr = NULL ;
2007-10-25 01:16:54 +04:00
char addr_buf [ INET6_ADDRSTRLEN ] ;
2008-02-25 17:24:49 +03:00
int ret ;
2007-10-17 23:10:12 +04:00
2003-01-04 11:48:15 +03:00
/* arguments checking */
2007-10-25 01:16:54 +04:00
if ( ! ipstr_list | | ! service ) {
return NULL ;
}
2003-01-04 11:48:15 +03:00
2007-10-26 06:07:25 +04:00
print_sockaddr ( addr_buf ,
sizeof ( addr_buf ) ,
& service - > ss ) ;
2003-01-04 11:48:15 +03:00
/* attempt to convert ip to a string and append colon separator to it */
if ( * ipstr_list ) {
2007-10-25 01:16:54 +04:00
if ( service - > ss . ss_family = = AF_INET ) {
/* IPv4 */
2008-02-25 17:24:49 +03:00
ret = asprintf ( & new_ipstr , " %s%s%s:%d " , * ipstr_list ,
IPSTR_LIST_SEP , addr_buf ,
service - > port ) ;
2007-10-25 01:16:54 +04:00
} else {
/* IPv6 */
2008-02-25 17:24:49 +03:00
ret = asprintf ( & new_ipstr , " %s%s[%s]:%d " , * ipstr_list ,
IPSTR_LIST_SEP , addr_buf ,
service - > port ) ;
2007-10-25 01:16:54 +04:00
}
2003-01-04 11:48:15 +03:00
SAFE_FREE ( * ipstr_list ) ;
} else {
2007-10-25 01:16:54 +04:00
if ( service - > ss . ss_family = = AF_INET ) {
/* IPv4 */
2008-02-25 17:24:49 +03:00
ret = asprintf ( & new_ipstr , " %s:%d " , addr_buf ,
service - > port ) ;
2007-10-25 01:16:54 +04:00
} else {
/* IPv6 */
2008-02-25 17:24:49 +03:00
ret = asprintf ( & new_ipstr , " [%s]:%d " , addr_buf ,
service - > port ) ;
2007-10-25 01:16:54 +04:00
}
2003-01-04 11:48:15 +03:00
}
2008-02-25 17:24:49 +03:00
if ( ret = = - 1 ) {
return NULL ;
}
2003-01-04 11:48:15 +03:00
* 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
* */
2007-10-17 23:10:12 +04:00
char * ipstr_list_make ( char * * ipstr_list ,
2007-10-25 01:16:54 +04:00
const struct ip_service * ip_list ,
int ip_count )
2003-01-04 11:48:15 +03:00
{
int i ;
2007-10-17 23:10:12 +04:00
2003-01-04 11:48:15 +03:00
/* arguments checking */
2007-10-25 01:16:54 +04:00
if ( ! ip_list | | ! ipstr_list ) {
return 0 ;
}
2003-01-04 11:48:15 +03:00
* ipstr_list = NULL ;
2007-10-17 23:10:12 +04:00
2003-01-04 11:48:15 +03:00
/* process ip addresses given as arguments */
2007-10-25 01:16:54 +04:00
for ( i = 0 ; i < ip_count ; i + + ) {
2003-01-04 11:48:15 +03:00
* ipstr_list = ipstr_list_add ( ipstr_list , & ip_list [ i ] ) ;
2007-10-25 01:16:54 +04:00
}
2007-10-17 23:10:12 +04:00
2003-01-04 11:48:15 +03:00
return ( * ipstr_list ) ;
}
/**
* Parse given ip string list into array of ip addresses
2007-10-17 23:10:12 +04:00
* ( as ip_service structures )
2007-10-25 01:16:54 +04:00
* e . g . [ IPv6 ] : port , 192.168 .1 .100 : 389 , 192.168 .1 .78 , . . .
2003-01-04 11:48:15 +03:00
*
2007-10-17 23:10:12 +04:00
* @ param ipstr ip string list to be parsed
2003-01-04 11:48:15 +03:00
* @ param ip_list pointer to array of ip addresses which is
* allocated by this function and must be freed by caller
2008-02-08 11:28:57 +03:00
* @ return number of successfully parsed addresses
2003-01-04 11:48:15 +03:00
* */
2007-10-17 23:10:12 +04:00
2007-10-25 01:16:54 +04:00
int ipstr_list_parse ( const char * ipstr_list , struct ip_service * * ip_list )
2003-01-04 11:48:15 +03:00
{
2007-12-08 04:32:32 +03:00
TALLOC_CTX * frame ;
char * token_str = NULL ;
2003-06-25 21:41:05 +04:00
size_t count ;
int i ;
2003-01-04 11:48:15 +03:00
2007-10-17 23:10:12 +04:00
if ( ! ipstr_list | | ! ip_list )
2003-06-25 21:41:05 +04:00
return 0 ;
2007-10-17 23:10:12 +04:00
2003-06-25 21:41:05 +04:00
count = count_chars ( ipstr_list , IPSTR_LIST_CHAR ) + 1 ;
2004-12-07 21:25:53 +03:00
if ( ( * ip_list = SMB_MALLOC_ARRAY ( struct ip_service , count ) ) = = NULL ) {
2007-10-17 23:10:12 +04:00
DEBUG ( 0 , ( " ipstr_list_parse: malloc failed for %lu entries \n " ,
( unsigned long ) count ) ) ;
2003-06-25 21:41:05 +04:00
return 0 ;
}
2007-10-17 23:10:12 +04:00
2007-12-08 04:32:32 +03:00
frame = talloc_stackframe ( ) ;
for ( i = 0 ; next_token_talloc ( frame , & ipstr_list , & token_str ,
IPSTR_LIST_SEP ) & & i < count ; i + + ) {
2007-10-25 01:16:54 +04:00
char * s = token_str ;
char * p = strrchr ( token_str , ' : ' ) ;
2007-10-17 23:10:12 +04:00
2003-06-25 21:41:05 +04:00
if ( p ) {
* p = 0 ;
2007-10-25 01:16:54 +04:00
( * ip_list ) [ i ] . port = atoi ( p + 1 ) ;
2003-06-25 21:41:05 +04:00
}
2003-01-04 11:48:15 +03:00
/* convert single token to ip address */
2007-10-25 01:16:54 +04:00
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 ;
}
2003-01-04 11:48:15 +03:00
}
2007-12-08 04:32:32 +03:00
TALLOC_FREE ( frame ) ;
2003-01-04 11:48:15 +03:00
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 ) ;
}
2003-05-12 05:20:17 +04:00
/* read a SMB_BIG_UINT from a string */
2008-10-14 03:59:36 +04:00
uint64_t STR_TO_SMB_BIG_UINT ( const char * nptr , const char * * entptr )
2003-05-12 05:20:17 +04:00
{
2011-02-09 21:43:56 +03:00
uint64_t val = ( uint64_t ) - 1 ;
2003-05-12 05:20:17 +04:00
const char * p = nptr ;
2007-10-17 23:10:12 +04:00
2006-06-20 03:19:24 +04:00
if ( ! p ) {
2006-08-28 05:48:04 +04:00
if ( entptr ) {
* entptr = p ;
}
2006-06-20 03:19:24 +04:00
return val ;
}
while ( * p & & isspace ( * p ) )
2003-05-12 05:20:17 +04:00
p + + ;
2006-06-20 03:19:24 +04:00
2008-10-14 03:59:36 +04:00
sscanf ( p , " % " PRIu64 , & val ) ;
2003-05-12 05:20:17 +04:00
if ( entptr ) {
2006-06-20 03:19:24 +04:00
while ( * p & & isdigit ( * p ) )
2003-05-12 05:20:17 +04:00
p + + ;
* entptr = p ;
}
return val ;
}
2004-04-07 16:43:44 +04:00
2006-07-11 22:01:26 +04:00
/* 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 )
{
2010-03-28 15:16:58 +04:00
SMB_OFF_T lval_orig ;
2006-07-11 22:01:26 +04:00
SMB_OFF_T lval ;
char * end ;
if ( str = = NULL | | * str = = ' \0 ' ) {
return 0 ;
}
# ifdef HAVE_STRTOULL
if ( sizeof ( SMB_OFF_T ) = = 8 ) {
2010-03-28 15:16:18 +04:00
lval = strtoull ( str , & end , 10 /* base */ ) ;
2006-07-11 22:01:26 +04:00
} else {
2010-03-28 15:16:18 +04:00
lval = strtoul ( str , & end , 10 /* base */ ) ;
2006-07-11 22:01:26 +04:00
}
# else
lval = strtoul ( str , & end , 10 /* base */ ) ;
# endif
if ( end = = NULL | | end = = str ) {
return 0 ;
}
2010-03-28 15:16:58 +04:00
if ( * end = = ' \0 ' ) {
return lval ;
}
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 ;
}
2006-07-11 22:01:26 +04:00
return lval ;
}
2005-06-09 02:10:34 +04:00
/* 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 ;
2007-10-17 23:10:12 +04:00
bool increased ;
2005-06-09 02:10:34 +04:00
/* len<0 is an internal marker that something failed */
if ( * len < 0 )
goto error ;
if ( * string = = NULL ) {
if ( * bufsize = = 0 )
* bufsize = 128 ;
2006-12-16 08:02:21 +03:00
* string = TALLOC_ARRAY ( mem_ctx , char , * bufsize ) ;
2005-06-09 02:10:34 +04:00
if ( * string = = NULL )
goto error ;
}
va_start ( ap , fmt ) ;
ret = vasprintf ( & newstr , fmt , ap ) ;
va_end ( ap ) ;
if ( ret < 0 )
goto error ;
2007-10-17 23:10:12 +04:00
increased = false ;
2005-06-09 02:10:34 +04:00
while ( ( * len ) + ret > = * bufsize ) {
2007-10-17 23:10:12 +04:00
increased = true ;
2005-06-09 02:10:34 +04:00
* bufsize * = 2 ;
if ( * bufsize > = ( 1024 * 1024 * 256 ) )
goto error ;
}
if ( increased ) {
2006-12-16 08:02:21 +03:00
* string = TALLOC_REALLOC_ARRAY ( mem_ctx , * string , char ,
* bufsize ) ;
r13915: Fixed a very interesting class of realloc() bugs found by Coverity.
realloc can return NULL in one of two cases - (1) the realloc failed,
(2) realloc succeeded but the new size requested was zero, in which
case this is identical to a free() call.
The error paths dealing with these two cases should be different,
but mostly weren't. Secondly the standard idiom for dealing with
realloc when you know the new size is non-zero is the following :
tmp = realloc(p, size);
if (!tmp) {
SAFE_FREE(p);
return error;
} else {
p = tmp;
}
However, there were *many* *many* places in Samba where we were
using the old (broken) idiom of :
p = realloc(p, size)
if (!p) {
return error;
}
which will leak the memory pointed to by p on realloc fail.
This commit (hopefully) fixes all these cases by moving to
a standard idiom of :
p = SMB_REALLOC(p, size)
if (!p) {
return error;
}
Where if the realloc returns null due to the realloc failing
or size == 0 we *guarentee* that the storage pointed to by p
has been freed. This allows me to remove a lot of code that
was dealing with the standard (more verbose) method that required
a tmp pointer. This is almost always what you want. When a
realloc fails you never usually want the old memory, you
want to free it and get into your error processing asap.
For the 11 remaining cases where we really do need to keep the
old pointer I have invented the new macro SMB_REALLOC_KEEP_OLD_ON_ERROR,
which can be used as follows :
tmp = SMB_REALLOC_KEEP_OLD_ON_ERROR(p, size);
if (!tmp) {
SAFE_FREE(p);
return error;
} else {
p = tmp;
}
SMB_REALLOC_KEEP_OLD_ON_ERROR guarentees never to free the
pointer p, even on size == 0 or realloc fail. All this is
done by a hidden extra argument to Realloc(), BOOL free_old_on_error
which is set appropriately by the SMB_REALLOC and SMB_REALLOC_KEEP_OLD_ON_ERROR
macros (and their array counterparts).
It remains to be seen what this will do to our Coverity bug count :-).
Jeremy.
(This used to be commit 1d710d06a214f3f1740e80e0bffd6aab44aac2b0)
2006-03-07 09:31:04 +03:00
if ( * string = = NULL ) {
2005-06-09 02:10:34 +04:00
goto error ;
r13915: Fixed a very interesting class of realloc() bugs found by Coverity.
realloc can return NULL in one of two cases - (1) the realloc failed,
(2) realloc succeeded but the new size requested was zero, in which
case this is identical to a free() call.
The error paths dealing with these two cases should be different,
but mostly weren't. Secondly the standard idiom for dealing with
realloc when you know the new size is non-zero is the following :
tmp = realloc(p, size);
if (!tmp) {
SAFE_FREE(p);
return error;
} else {
p = tmp;
}
However, there were *many* *many* places in Samba where we were
using the old (broken) idiom of :
p = realloc(p, size)
if (!p) {
return error;
}
which will leak the memory pointed to by p on realloc fail.
This commit (hopefully) fixes all these cases by moving to
a standard idiom of :
p = SMB_REALLOC(p, size)
if (!p) {
return error;
}
Where if the realloc returns null due to the realloc failing
or size == 0 we *guarentee* that the storage pointed to by p
has been freed. This allows me to remove a lot of code that
was dealing with the standard (more verbose) method that required
a tmp pointer. This is almost always what you want. When a
realloc fails you never usually want the old memory, you
want to free it and get into your error processing asap.
For the 11 remaining cases where we really do need to keep the
old pointer I have invented the new macro SMB_REALLOC_KEEP_OLD_ON_ERROR,
which can be used as follows :
tmp = SMB_REALLOC_KEEP_OLD_ON_ERROR(p, size);
if (!tmp) {
SAFE_FREE(p);
return error;
} else {
p = tmp;
}
SMB_REALLOC_KEEP_OLD_ON_ERROR guarentees never to free the
pointer p, even on size == 0 or realloc fail. All this is
done by a hidden extra argument to Realloc(), BOOL free_old_on_error
which is set appropriately by the SMB_REALLOC and SMB_REALLOC_KEEP_OLD_ON_ERROR
macros (and their array counterparts).
It remains to be seen what this will do to our Coverity bug count :-).
Jeremy.
(This used to be commit 1d710d06a214f3f1740e80e0bffd6aab44aac2b0)
2006-03-07 09:31:04 +03:00
}
2005-06-09 02:10:34 +04:00
}
StrnCpy ( ( * string ) + ( * len ) , newstr , ret ) ;
( * len ) + = ret ;
free ( newstr ) ;
return ;
error :
* len = - 1 ;
* string = NULL ;
}
2005-07-07 01:02:43 +04:00
2007-11-24 19:27:54 +03:00
/*
* 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 ;
}
2008-06-04 03:29:22 +04:00
char * talloc_asprintf_strlower_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 ;
}
strlower_m ( ret ) ;
return ret ;
}
2005-07-07 01:02:43 +04:00
/*
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 ;
}
2006-02-04 01:19:41 +03:00
/********************************************************************
Check a string for any occurrences of a specified list of invalid
characters .
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
2007-10-17 23:10:12 +04:00
bool validate_net_name ( const char * name ,
const char * invalid_chars ,
int max_len )
2006-02-04 01:19:41 +03:00
{
int i ;
2010-11-10 02:07:49 +03:00
if ( ! name ) {
return false ;
}
2006-02-04 01:19:41 +03:00
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 ] ) ) {
2007-10-17 23:10:12 +04:00
return false ;
2006-02-04 01:19:41 +03:00
}
}
2007-10-17 23:10:12 +04:00
return true ;
2006-02-04 01:19:41 +03:00
}
2006-07-11 22:01:26 +04:00
2007-05-14 18:23:51 +04:00
/*******************************************************************
Add a shell escape character ' \ ' to any character not in a known list
of characters . UNIX charset format .
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
2007-06-05 16:58:18 +04:00
# define INCLUDE_LIST "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz_ / \t.,"
2007-05-14 18:23:51 +04:00
# define INSIDE_DQUOTE_LIST "$`\n\"\\"
char * escape_shell_string ( const char * src )
{
size_t srclen = strlen ( src ) ;
2007-06-16 15:48:11 +04:00
char * ret = SMB_MALLOC_ARRAY ( char , ( srclen * 2 ) + 1 ) ;
2007-05-14 18:23:51 +04:00
char * dest = ret ;
2007-10-17 23:10:12 +04:00
bool in_s_quote = false ;
bool in_d_quote = false ;
bool next_escaped = false ;
2007-05-14 18:23:51 +04:00
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 ;
2007-10-17 23:10:12 +04:00
next_escaped = false ;
2007-05-14 18:23:51 +04:00
continue ;
}
/*
* Deal with backslash escaped state .
* This only lasts for one character .
*/
if ( next_escaped ) {
* dest + + = * src + + ;
2007-10-17 23:10:12 +04:00
next_escaped = false ;
2007-05-14 18:23:51 +04:00
continue ;
}
/*
* Deal with single quote state . The
* only thing we care about is exiting
* this state .
*/
if ( in_s_quote ) {
if ( * src = = ' \' ' ) {
2007-10-17 23:10:12 +04:00
in_s_quote = false ;
2007-05-14 18:23:51 +04:00
}
* dest + + = * src + + ;
continue ;
}
2007-10-17 23:10:12 +04:00
/*
2007-05-14 18:23:51 +04:00
* 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 = = ' \\ ' ) {
2007-10-17 23:10:12 +04:00
/*
2007-05-14 18:23:51 +04:00
* 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 ] ;
2007-10-17 23:10:12 +04:00
if ( nextchar & & strchr ( INSIDE_DQUOTE_LIST ,
( int ) nextchar ) ) {
next_escaped = true ;
2007-05-14 18:23:51 +04:00
}
* dest + + = * src + + ;
continue ;
}
if ( * src = = ' \" ' ) {
/* Exit double quote state. */
2007-10-17 23:10:12 +04:00
in_d_quote = false ;
2007-05-14 18:23:51 +04:00
* 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 ;
}
2007-10-17 23:10:12 +04:00
/*
2007-05-14 18:23:51 +04:00
* 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. */
2007-10-17 23:10:12 +04:00
next_escaped = true ;
2007-05-14 18:23:51 +04:00
* dest + + = * src + + ;
continue ;
}
if ( * src = = ' \' ' ) {
/* Go into single quote state. */
2007-10-17 23:10:12 +04:00
in_s_quote = true ;
2007-05-14 18:23:51 +04:00
* dest + + = * src + + ;
continue ;
}
if ( * src = = ' \" ' ) {
/* Go into double quote state. */
2007-10-17 23:10:12 +04:00
in_d_quote = true ;
2007-05-14 18:23:51 +04:00
* dest + + = * src + + ;
continue ;
}
/* Check if we need to escape the character. */
if ( ! strchr ( INCLUDE_LIST , ( int ) * src ) ) {
* dest + + = ' \\ ' ;
}
* dest + + = * src + + ;
}
* dest + + = ' \0 ' ;
return ret ;
}
2008-11-07 05:53:00 +03:00
/***************************************************
2009-04-30 15:37:19 +04:00
str_list_make , v3 version . The v4 version does not
look at quoted strings with embedded blanks , so
do NOT merge this function please !
2008-11-07 05:53:00 +03:00
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
2009-04-30 15:37:19 +04:00
# define S_LIST_ABS 16 /* List Allocation Block Size */
2009-09-25 02:42:36 +04:00
char * * str_list_make_v3 ( TALLOC_CTX * mem_ctx , const char * string ,
2009-09-25 01:50:05 +04:00
const char * sep )
2008-11-07 05:53:00 +03:00
{
2009-09-25 02:42:36 +04:00
char * * list ;
2009-04-30 15:37:19 +04:00
const char * str ;
2009-09-25 01:50:05 +04:00
char * s , * tok ;
2009-04-30 15:37:19 +04:00
int num , lsize ;
if ( ! string | | ! * string )
return NULL ;
2009-09-25 02:42:36 +04:00
list = TALLOC_ARRAY ( mem_ctx , char * , S_LIST_ABS + 1 ) ;
2009-04-30 15:37:19 +04:00
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 ) ;
2008-11-07 05:53:00 +03:00
return NULL ;
}
2009-04-30 15:37:19 +04:00
if ( ! sep ) sep = LIST_SEP ;
num = 0 ;
str = s ;
while ( next_token_talloc ( list , & str , & tok , sep ) ) {
if ( num = = lsize ) {
2009-09-25 02:42:36 +04:00
char * * tmp ;
2009-04-30 15:37:19 +04:00
lsize + = S_LIST_ABS ;
2009-09-25 02:42:36 +04:00
tmp = TALLOC_REALLOC_ARRAY ( mem_ctx , list , char * ,
2009-04-30 15:37:19 +04:00
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 ,
2009-09-25 02:42:36 +04:00
( ( sizeof ( char * * ) ) * ( S_LIST_ABS + 1 ) ) ) ;
2009-04-30 15:37:19 +04:00
}
list [ num ] = tok ;
num + = 1 ;
}
list [ num ] = NULL ;
TALLOC_FREE ( s ) ;
return list ;
2008-11-07 05:53:00 +03:00
}
2010-04-12 00:38:33 +04:00
char * sanitize_username ( TALLOC_CTX * mem_ctx , const char * username )
{
fstring tmp ;
alpha_strcpy ( tmp , username , " . _-$ " , sizeof ( tmp ) ) ;
return talloc_strdup ( mem_ctx , tmp ) ;
}