source3/lib/util_unistr.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4    Copyright (C) Andrew Tridgell 1992-2001
   5    Copyright (C) Simo Sorce 2001
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 2 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program; if not, write to the Free Software
  19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  20 */
  21
  22 #include "includes.h"
  23
  24 #ifndef MAXUNI
  25 #define MAXUNI 1024
  26 #endif
  27
  28 /* these 3 tables define the unicode case handling.  They are loaded
  29    at startup either via mmap() or read() from the lib directory */
  30 static smb_ucs2_t *upcase_table;
  31 static smb_ucs2_t *lowcase_table;
  32 static uint8 *valid_table;
  33
  34 /**
  35  * This table says which Unicode characters are valid dos
  36  * characters.
  37  *
  38  * Each value is just a single bit.
  39  **/
  40 static uint8 doschar_table[8192]; /* 65536 characters / 8 bits/byte */
  41
  42
  43 /**
  44  * Load or generate the case handling tables.
  45  *
  46  * The case tables are defined in UCS2 and don't depend on any
  47  * configured parameters, so they never need to be reloaded.
  48  **/
  49 void load_case_tables(void)
  50 {
  51         static int initialised;
  52         int i;
  53
  54         if (initialised) return;
  55         initialised = 1;
  56
  57         upcase_table = map_file(lib_path("upcase.dat"), 0x20000);
  58         lowcase_table = map_file(lib_path("lowcase.dat"), 0x20000);
  59
  60         /* we would like Samba to limp along even if these tables are
  61            not available */
  62         if (!upcase_table) {
  63                 DEBUG(1,("creating lame upcase table\n"));
  64                 upcase_table = malloc(0x20000);
  65                 for (i=0;i<0x10000;i++) {
  66                         smb_ucs2_t v;
  67                         SSVAL(&v, 0, i);
  68                         upcase_table[v] = i;
  69                 }
  70                 for (i=0;i<256;i++) {
  71                         smb_ucs2_t v;
  72                         SSVAL(&v, 0, UCS2_CHAR(i));
  73                         upcase_table[v] = UCS2_CHAR(islower(i)?toupper(i):i);
  74                 }
  75         }
  76
  77         if (!lowcase_table) {
  78                 DEBUG(1,("creating lame lowcase table\n"));
  79                 lowcase_table = malloc(0x20000);
  80                 for (i=0;i<0x10000;i++) {
  81                         smb_ucs2_t v;
  82                         SSVAL(&v, 0, i);
  83                         lowcase_table[v] = i;
  84                 }
  85                 for (i=0;i<256;i++) {
  86                         smb_ucs2_t v;
  87                         SSVAL(&v, 0, UCS2_CHAR(i));
  88                         lowcase_table[v] = UCS2_CHAR(isupper(i)?tolower(i):i);
  89                 }
  90         }
  91 }
  92
  93 /*
  94   see if a ucs2 character can be mapped correctly to a dos character
  95   and mapped back to the same character in ucs2
  96 */
  97 int check_dos_char(smb_ucs2_t c)
  98 {
  99         lazy_initialize_conv();
 100
 101         /* Find the right byte, and right bit within the byte; return
 102          * 1 or 0 */
 103         return (doschar_table[(c & 0xffff) / 8] & (1 << (c & 7))) != 0;
 104 }
 105
 106
 107 static int check_dos_char_slowly(smb_ucs2_t c)
 108 {
 109         char buf[10];
 110         smb_ucs2_t c2 = 0;
 111         int len1, len2;
 112         len1 = convert_string(CH_UCS2, CH_DOS, &c, 2, buf, sizeof(buf));
 113         if (len1 == 0) return 0;
 114         len2 = convert_string(CH_DOS, CH_UCS2, buf, len1, &c2, 2);
 115         if (len2 != 2) return 0;
 116         return (c == c2);
 117 }
 118
 119
 120 /**
 121  * Fill out doschar table the hard way, by examining each character
 122  **/
 123 void init_doschar_table(void)
 124 {
 125         int i, j, byteval;
 126
 127         /* For each byte of packed table */
 128
 129         for (i = 0; i <= 0xffff; i += 8) {
 130                 byteval = 0;
 131                 for (j = 0; j <= 7; j++) {
 132                         smb_ucs2_t c;
 133
 134                         c = i + j;
 135
 136                         if (check_dos_char_slowly(c))
 137                                 byteval |= 1 << j;
 138                 }
 139                 doschar_table[i/8] = byteval;
 140         }
 141 }
 142
 143
 144 /**
 145  * Load the valid character map table from <tt>valid.dat</tt> or
 146  * create from the configured codepage.
 147  *
 148  * This function is called whenever the configuration is reloaded.
 149  * However, the valid character table is not changed if it's loaded
 150  * from a file, because we can't unmap files.
 151  **/
 152 void init_valid_table(void)
 153 {
 154         static int mapped_file;
 155         int i;
 156         const char *allowed = ".!#$%&'()_-@^`~";
 157         uint8 *valid_file;
 158
 159         if (mapped_file) {
 160                 /* Can't unmap files, so stick with what we have */
 161                 return;
 162         }
 163
 164         valid_file = map_file(lib_path("valid.dat"), 0x10000);
 165         if (valid_file) {
 166                 valid_table = valid_file;
 167                 mapped_file = 1;
 168                 return;
 169         }
 170
 171         /* Otherwise, we're using a dynamically created valid_table.
 172          * It might need to be regenerated if the code page changed.
 173          * We know that we're not using a mapped file, so we can
 174          * free() the old one. */
 175         if (valid_table) free(valid_table);
 176
 177         DEBUG(2,("creating default valid table\n"));
 178         valid_table = malloc(0x10000);
 179         for (i=0;i<128;i++)
 180                 valid_table[i] = isalnum(i) || strchr(allowed,i);
 181
 182         for (;i<0x10000;i++) {
 183                 smb_ucs2_t c;
 184                 SSVAL(&c, 0, i);
 185                 valid_table[i] = check_dos_char(c);
 186         }
 187 }
 188
 189
 190
 191 /*******************************************************************
 192  Write a string in (little-endian) unicode format. src is in
 193  the current DOS codepage. len is the length in bytes of the
 194  string pointed to by dst.
 195
 196  if null_terminate is True then null terminate the packet (adds 2 bytes)
 197
 198  the return value is the length in bytes consumed by the string, including the
 199  null termination if applied
 200 ********************************************************************/
 201
 202 size_t dos_PutUniCode(char *dst,const char *src, ssize_t len, BOOL null_terminate)
 203 {
 204         return push_ucs2(NULL, dst, src, len,
 205                          STR_UNICODE|STR_NOALIGN | (null_terminate?STR_TERMINATE:0));
 206 }
 207
 208
 209 /*******************************************************************
 210  Skip past a unicode string, but not more than len. Always move
 211  past a terminating zero if found.
 212 ********************************************************************/
 213
 214 char *skip_unibuf(char *src, size_t len)
 215 {
 216     char *srcend = src + len;
 217
 218     while (src < srcend && SVAL(src,0))
 219         src += 2;
 220
 221     if(!SVAL(src,0))
 222         src += 2;
 223
 224     return src;
 225 }
 226
 227 /* Copy a string from little-endian or big-endian unicode source (depending
 228  * on flags) to internal samba format destination
 229  */
 230 int rpcstr_pull(char* dest, void *src, int dest_len, int src_len, int flags)
 231 {
 232         if (!src) {
 233                 dest[0] = 0;
 234                 return 0;
 235         }
 236         if(dest_len==-1) dest_len=MAXUNI-3;
 237         return pull_ucs2(NULL, dest, src, dest_len, src_len, flags|STR_UNICODE|STR_NOALIGN);
 238 }
 239
 240 /* Copy a string from a unistr2 source to internal samba format
 241    destination.  Use this instead of direct calls to rpcstr_pull() to avoid
 242    having to determine whether the source string is null terminated. */
 243
 244 int rpcstr_pull_unistr2_fstring(char *dest, UNISTR2 *src)
 245 {
 246         return pull_ucs2(NULL, dest, src->buffer, sizeof(fstring),
 247                          src->uni_str_len * 2, 0);
 248 }
 249
 250 /* Converts a string from internal samba format to unicode
 251  */
 252 int rpcstr_push(void* dest, const char *src, int dest_len, int flags)
 253 {
 254         return push_ucs2(NULL, dest, src, dest_len, flags|STR_UNICODE|STR_NOALIGN);
 255 }
 256
 257 /*******************************************************************
 258  Return a DOS codepage version of a little-endian unicode string.
 259  len is the filename length (ignoring any terminating zero) in uin16
 260  units. Always null terminates.
 261  Hack alert: uses fixed buffer(s).
 262 ********************************************************************/
 263 char *dos_unistrn2(const uint16 *src, int len)
 264 {
 265         static char lbufs[8][MAXUNI];
 266         static int nexti;
 267         char *lbuf = lbufs[nexti];
 268         nexti = (nexti+1)%8;
 269         pull_ucs2(NULL, lbuf, src, MAXUNI-3, len*2, STR_NOALIGN);
 270         return lbuf;
 271 }
 272
 273 /*******************************************************************
 274  Convert a (little-endian) UNISTR2 structure to an ASCII string
 275 ********************************************************************/
 276 void unistr2_to_ascii(char *dest, const UNISTR2 *str, size_t maxlen)
 277 {
 278         if (str == NULL) {
 279                 *dest='\0';
 280                 return;
 281         }
 282         pull_ucs2(NULL, dest, str->buffer, maxlen, str->uni_str_len*2, STR_NOALIGN);
 283 }
 284
 285 /*******************************************************************
 286 give a static string for displaying a UNISTR2
 287 ********************************************************************/
 288 const char *unistr2_static(const UNISTR2 *str)
 289 {
 290         static pstring ret;
 291         unistr2_to_ascii(ret, str, sizeof(ret));
 292         return ret;
 293 }
 294
 295
 296 /*******************************************************************
 297  duplicate a UNISTR2 string into a null terminated char*
 298  using a talloc context
 299 ********************************************************************/
 300 char *unistr2_tdup(TALLOC_CTX *ctx, const UNISTR2 *str)
 301 {
 302         char *s;
 303         int maxlen = (str->uni_str_len+1)*4;
 304         if (!str->buffer) return NULL;
 305         s = (char *)talloc(ctx, maxlen); /* convervative */
 306         if (!s) return NULL;
 307         pull_ucs2(NULL, s, str->buffer, maxlen, str->uni_str_len*2,
 308                   STR_NOALIGN);
 309         return s;
 310 }
 311
 312
 313 /*******************************************************************
 314 Return a number stored in a buffer
 315 ********************************************************************/
 316
 317 uint32 buffer2_to_uint32(BUFFER2 *str)
 318 {
 319         if (str->buf_len == 4)
 320                 return IVAL(str->buffer, 0);
 321         else
 322                 return 0;
 323 }
 324
 325 /*******************************************************************
 326  Convert a wchar to upper case.
 327 ********************************************************************/
 328
 329 smb_ucs2_t toupper_w(smb_ucs2_t val)
 330 {
 331         return upcase_table[SVAL(&val,0)];
 332 }
 333
 334 /*******************************************************************
 335  Convert a wchar to lower case.
 336 ********************************************************************/
 337
 338 smb_ucs2_t tolower_w( smb_ucs2_t val )
 339 {
 340         return lowcase_table[SVAL(&val,0)];
 341
 342 }
 343
 344 /*******************************************************************
 345 determine if a character is lowercase
 346 ********************************************************************/
 347 BOOL islower_w(smb_ucs2_t c)
 348 {
 349         return upcase_table[SVAL(&c,0)] != c;
 350 }
 351
 352 /*******************************************************************
 353 determine if a character is uppercase
 354 ********************************************************************/
 355 BOOL isupper_w(smb_ucs2_t c)
 356 {
 357         return lowcase_table[SVAL(&c,0)] != c;
 358 }
 359
 360
 361 /*******************************************************************
 362 determine if a character is valid in a 8.3 name
 363 ********************************************************************/
 364 BOOL isvalid83_w(smb_ucs2_t c)
 365 {
 366         return valid_table[SVAL(&c,0)] != 0;
 367 }
 368
 369 /*******************************************************************
 370  Count the number of characters in a smb_ucs2_t string.
 371 ********************************************************************/
 372 size_t strlen_w(const smb_ucs2_t *src)
 373 {
 374         size_t len;
 375
 376         for(len = 0; *src++; len++) ;
 377
 378         return len;
 379 }
 380
 381 /*******************************************************************
 382  Count up to max number of characters in a smb_ucs2_t string.
 383 ********************************************************************/
 384 size_t strnlen_w(const smb_ucs2_t *src, size_t max)
 385 {
 386         size_t len;
 387
 388         for(len = 0; *src++ && (len < max); len++) ;
 389
 390         return len;
 391 }
 392
 393 /*******************************************************************
 394 wide strchr()
 395 ********************************************************************/
 396 smb_ucs2_t *strchr_w(const smb_ucs2_t *s, smb_ucs2_t c)
 397 {
 398         while (*s != 0) {
 399                 if (c == *s) return (smb_ucs2_t *)s;
 400                 s++;
 401         }
 402         if (c == *s) return (smb_ucs2_t *)s;
 403
 404         return NULL;
 405 }
 406
 407 smb_ucs2_t *strchr_wa(const smb_ucs2_t *s, char c)
 408 {
 409         return strchr_w(s, UCS2_CHAR(c));
 410 }
 411
 412 smb_ucs2_t *strrchr_w(const smb_ucs2_t *s, smb_ucs2_t c)
 413 {
 414         const smb_ucs2_t *p = s;
 415         int len = strlen_w(s);
 416         if (len == 0) return NULL;
 417         p += (len - 1);
 418         do {
 419                 if (c == *p) return (smb_ucs2_t *)p;
 420         } while (p-- != s);
 421         return NULL;
 422 }
 423
 424 /*******************************************************************
 425 wide strstr()
 426 ********************************************************************/
 427 smb_ucs2_t *strstr_w(const smb_ucs2_t *s, const smb_ucs2_t *ins)
 428 {
 429         smb_ucs2_t *r;
 430         size_t slen, inslen;
 431
 432         if (!s || !*s || !ins || !*ins) return NULL;
 433         slen = strlen_w(s);
 434         inslen = strlen_w(ins);
 435         r = (smb_ucs2_t *)s;
 436         while ((r = strchr_w(r, *ins))) {
 437                 if (strncmp_w(r, ins, inslen) == 0) return r;
 438                 r++;
 439         }
 440         return NULL;
 441 }
 442
 443 /*******************************************************************
 444  Convert a string to lower case.
 445  return True if any char is converted
 446 ********************************************************************/
 447 BOOL strlower_w(smb_ucs2_t *s)
 448 {
 449         BOOL ret = False;
 450         while (*s) {
 451                 smb_ucs2_t v = tolower_w(*s);
 452                 if (v != *s) {
 453                         *s = v;
 454                         ret = True;
 455                 }
 456                 s++;
 457         }
 458         return ret;
 459 }
 460
 461 /*******************************************************************
 462  Convert a string to upper case.
 463  return True if any char is converted
 464 ********************************************************************/
 465 BOOL strupper_w(smb_ucs2_t *s)
 466 {
 467         BOOL ret = False;
 468         while (*s) {
 469                 smb_ucs2_t v = toupper_w(*s);
 470                 if (v != *s) {
 471                         *s = v;
 472                         ret = True;
 473                 }
 474                 s++;
 475         }
 476         return ret;
 477 }
 478
 479 /*******************************************************************
 480   convert a string to "normal" form
 481 ********************************************************************/
 482 void strnorm_w(smb_ucs2_t *s)
 483 {
 484   extern int case_default;
 485   if (case_default == CASE_UPPER)
 486     strupper_w(s);
 487   else
 488     strlower_w(s);
 489 }
 490
 491 int strcmp_w(const smb_ucs2_t *a, const smb_ucs2_t *b)
 492 {
 493         while (*b && *a == *b) { a++; b++; }
 494         return (*a - *b);
 495         /* warning: if *a != *b and both are not 0 we retrun a random
 496                 greater or lesser than 0 number not realted to which
 497                 string is longer */
 498 }
 499
 500 int strncmp_w(const smb_ucs2_t *a, const smb_ucs2_t *b, size_t len)
 501 {
 502         size_t n = 0;
 503         while ((n < len) && *b && *a == *b) { a++; b++; n++;}
 504         return (len - n)?(*a - *b):0;
 505 }
 506
 507 /*******************************************************************
 508 case insensitive string comparison
 509 ********************************************************************/
 510 int strcasecmp_w(const smb_ucs2_t *a, const smb_ucs2_t *b)
 511 {
 512         while (*b && toupper_w(*a) == toupper_w(*b)) { a++; b++; }
 513         return (tolower_w(*a) - tolower_w(*b));
 514 }
 515
 516 /*******************************************************************
 517 case insensitive string comparison, lenght limited
 518 ********************************************************************/
 519 int strncasecmp_w(const smb_ucs2_t *a, const smb_ucs2_t *b, size_t len)
 520 {
 521         size_t n = 0;
 522         while ((n < len) && *b && (toupper_w(*a) == toupper_w(*b))) { a++; b++; n++; }
 523         return (len - n)?(tolower_w(*a) - tolower_w(*b)):0;
 524 }
 525
 526 /*******************************************************************
 527   compare 2 strings
 528 ********************************************************************/
 529 BOOL strequal_w(const smb_ucs2_t *s1, const smb_ucs2_t *s2)
 530 {
 531         if (s1 == s2) return(True);
 532         if (!s1 || !s2) return(False);
 533
 534         return(strcasecmp_w(s1,s2)==0);
 535 }
 536
 537 /*******************************************************************
 538   compare 2 strings up to and including the nth char.
 539   ******************************************************************/
 540 BOOL strnequal_w(const smb_ucs2_t *s1,const smb_ucs2_t *s2,size_t n)
 541 {
 542   if (s1 == s2) return(True);
 543   if (!s1 || !s2 || !n) return(False);
 544
 545   return(strncasecmp_w(s1,s2,n)==0);
 546 }
 547
 548 /*******************************************************************
 549 duplicate string
 550 ********************************************************************/
 551 smb_ucs2_t *strdup_w(const smb_ucs2_t *src)
 552 {
 553         return strndup_w(src, 0);
 554 }
 555
 556 /* if len == 0 then duplicate the whole string */
 557 smb_ucs2_t *strndup_w(const smb_ucs2_t *src, size_t len)
 558 {
 559         smb_ucs2_t *dest;
 560
 561         if (!len) len = strlen_w(src);
 562         dest = (smb_ucs2_t *)malloc((len + 1) * sizeof(smb_ucs2_t));
 563         if (!dest) {
 564                 DEBUG(0,("strdup_w: out of memory!\n"));
 565                 return NULL;
 566         }
 567
 568         memcpy(dest, src, len * sizeof(smb_ucs2_t));
 569         dest[len] = 0;
 570
 571         return dest;
 572 }
 573
 574 /*******************************************************************
 575 copy a string with max len
 576 ********************************************************************/
 577
 578 smb_ucs2_t *strncpy_w(smb_ucs2_t *dest, const smb_ucs2_t *src, const size_t max)
 579 {
 580         size_t len;
 581
 582         if (!dest || !src) return NULL;
 583
 584         for (len = 0; (src[len] != 0) && (len < max); len++)
 585                 dest[len] = src[len];
 586         while (len < max)
 587                 dest[len++] = 0;
 588
 589         return dest;
 590 }
 591
 592
 593 /*******************************************************************
 594 append a string of len bytes and add a terminator
 595 ********************************************************************/
 596
 597 smb_ucs2_t *strncat_w(smb_ucs2_t *dest, const smb_ucs2_t *src, const size_t max)
 598 {
 599         size_t start;
 600         size_t len;
 601
 602         if (!dest || !src) return NULL;
 603
 604         start = strlen_w(dest);
 605         len = strnlen_w(src, max);
 606
 607         memcpy(&dest[start], src, len*sizeof(smb_ucs2_t));
 608         dest[start+len] = 0;
 609
 610         return dest;
 611 }
 612
 613 smb_ucs2_t *strcat_w(smb_ucs2_t *dest, const smb_ucs2_t *src)
 614 {
 615         size_t start;
 616         size_t len;
 617
 618         if (!dest || !src) return NULL;
 619
 620         start = strlen_w(dest);
 621         len = strlen_w(src);
 622
 623         memcpy(&dest[start], src, len*sizeof(smb_ucs2_t));
 624         dest[start+len] = 0;
 625
 626         return dest;
 627 }
 628
 629
 630 /*******************************************************************
 631 replace any occurence of oldc with newc in unicode string
 632 ********************************************************************/
 633
 634 void string_replace_w(smb_ucs2_t *s, smb_ucs2_t oldc, smb_ucs2_t newc)
 635 {
 636         for(;*s;s++) {
 637                 if(*s==oldc) *s=newc;
 638         }
 639 }
 640
 641 /*******************************************************************
 642 trim unicode string
 643 ********************************************************************/
 644
 645 BOOL trim_string_w(smb_ucs2_t *s, const smb_ucs2_t *front,
 646                                   const smb_ucs2_t *back)
 647 {
 648         BOOL ret = False;
 649         size_t len, front_len, back_len;
 650
 651         if (!s || !*s) return False;
 652
 653         len = strlen_w(s);
 654
 655         if (front && *front) {
 656                 front_len = strlen_w(front);
 657                 while (len && strncmp_w(s, front, front_len) == 0) {
 658                         memmove(s, (s + front_len), (len - front_len + 1) * sizeof(smb_ucs2_t));
 659                         len -= front_len;
 660                         ret = True;
 661                 }
 662         }
 663
 664         if (back && *back) {
 665                 back_len = strlen_w(back);
 666                 while (len && strncmp_w((s + (len - back_len)), back, back_len) == 0) {
 667                         s[len - back_len] = 0;
 668                         len -= back_len;
 669                         ret = True;
 670                 }
 671         }
 672
 673         return ret;
 674 }
 675
 676 /*
 677   The *_wa() functions take a combination of 7 bit ascii
 678   and wide characters They are used so that you can use string
 679   functions combining C string constants with ucs2 strings
 680
 681   The char* arguments must NOT be multibyte - to be completely sure
 682   of this only pass string constants */
 683
 684
 685 void pstrcpy_wa(smb_ucs2_t *dest, const char *src)
 686 {
 687         int i;
 688         for (i=0;i<PSTRING_LEN;i++) {
 689                 dest[i] = UCS2_CHAR(src[i]);
 690                 if (src[i] == 0) return;
 691         }
 692 }
 693
 694 int strcmp_wa(const smb_ucs2_t *a, const char *b)
 695 {
 696         while (*b && *a == UCS2_CHAR(*b)) { a++; b++; }
 697         return (*a - UCS2_CHAR(*b));
 698 }
 699
 700 int strncmp_wa(const smb_ucs2_t *a, const char *b, size_t len)
 701 {
 702         size_t n = 0;
 703         while ((n < len) && *b && *a == UCS2_CHAR(*b)) { a++; b++; n++;}
 704         return (len - n)?(*a - UCS2_CHAR(*b)):0;
 705 }
 706
 707 smb_ucs2_t *strpbrk_wa(const smb_ucs2_t *s, const char *p)
 708 {
 709         while (*s != 0) {
 710                 int i;
 711                 for (i=0; p[i] && *s != UCS2_CHAR(p[i]); i++)
 712                         ;
 713                 if (p[i]) return (smb_ucs2_t *)s;
 714                 s++;
 715         }
 716         return NULL;
 717 }
 718
 719 smb_ucs2_t *strstr_wa(const smb_ucs2_t *s, const char *ins)
 720 {
 721         smb_ucs2_t *r;
 722         size_t slen, inslen;
 723
 724         if (!s || !*s || !ins || !*ins) return NULL;
 725         slen = strlen_w(s);
 726         inslen = strlen(ins);
 727         r = (smb_ucs2_t *)s;
 728         while ((r = strchr_w(r, UCS2_CHAR(*ins)))) {
 729                 if (strncmp_wa(r, ins, inslen) == 0) return r;
 730                 r++;
 731         }
 732         return NULL;
 733 }
 734
 735 /*******************************************************************
 736 copy a string with max len
 737 ********************************************************************/
 738
 739 smb_ucs2_t *strncpy_wa(smb_ucs2_t *dest, const char *src, const size_t max)
 740 {
 741         smb_ucs2_t *ucs2_src;
 742
 743         if (!dest || !src) return NULL;
 744         if (!(ucs2_src = acnv_uxu2(src)))
 745                 return NULL;
 746
 747         strncpy_w(dest, ucs2_src, max);
 748         SAFE_FREE(ucs2_src);
 749         return dest;
 750 }
 751
 752 /*******************************************************************
 753 convert and duplicate an ascii string
 754 ********************************************************************/
 755 smb_ucs2_t *strdup_wa(const char *src)
 756 {
 757         return strndup_wa(src, 0);
 758 }
 759
 760 /* if len == 0 then duplicate the whole string */
 761 smb_ucs2_t *strndup_wa(const char *src, size_t len)
 762 {
 763         smb_ucs2_t *dest, *s;
 764
 765         s = acnv_dosu2(src);
 766         if (!len) len = strlen_w(s);
 767         dest = (smb_ucs2_t *)malloc((len + 1) * sizeof(smb_ucs2_t));
 768         if (!dest) {
 769                 DEBUG(0,("strdup_w: out of memory!\n"));
 770                 SAFE_FREE(s);
 771                 return NULL;
 772         }
 773
 774         memcpy(dest, src, len * sizeof(smb_ucs2_t));
 775         dest[len] = 0;
 776
 777         SAFE_FREE(s);
 778         return dest;
 779 }
 780
 781 /*******************************************************************
 782 append a string of len bytes and add a terminator
 783 ********************************************************************/
 784
 785 smb_ucs2_t *strncat_wa(smb_ucs2_t *dest, const char *src, const size_t max)
 786 {
 787         smb_ucs2_t *ucs2_src;
 788
 789         if (!dest || !src) return NULL;
 790         if (!(ucs2_src = acnv_uxu2(src)))
 791                 return NULL;
 792
 793         strncat_w(dest, ucs2_src, max);
 794         SAFE_FREE(ucs2_src);
 795         return dest;
 796 }
 797
 798 smb_ucs2_t *strcat_wa(smb_ucs2_t *dest, const char *src)
 799 {
 800         smb_ucs2_t *ucs2_src;
 801
 802         if (!dest || !src) return NULL;
 803         if (!(ucs2_src = acnv_uxu2(src)))
 804                 return NULL;
 805
 806         strcat_w(dest, ucs2_src);
 807         SAFE_FREE(ucs2_src);
 808         return dest;
 809 }
 810
 811 BOOL trim_string_wa(smb_ucs2_t *s, const char *front,
 812                                   const char *back)
 813 {
 814         wpstring f, b;
 815
 816         if (front) push_ucs2(NULL, f, front, sizeof(wpstring) - 1, STR_TERMINATE);
 817         else *f = 0;
 818         if (back) push_ucs2(NULL, b, back, sizeof(wpstring) - 1, STR_TERMINATE);
 819         else *b = 0;
 820         return trim_string_w(s, f, b);
 821 }
 822
 823 /*******************************************************************
 824  returns the length in number of wide characters
 825  ******************************************************************/
 826 int unistrlen(uint16 *s)
 827 {
 828         int len;
 829
 830         if (!s)
 831                 return -1;
 832
 833         for (len=0; *s; s++,len++);
 834
 835         return len;
 836 }
 837
 838 /*******************************************************************
 839  Strcpy for unicode strings.  returns length (in num of wide chars)
 840 ********************************************************************/
 841
 842 int unistrcpy(uint16 *dst, uint16 *src)
 843 {
 844         int num_wchars = 0;
 845
 846         while (*src) {
 847                 *dst++ = *src++;
 848                 num_wchars++;
 849         }
 850         *dst = 0;
 851
 852         return num_wchars;
 853 }
 854
 855 /**
 856  * Samba ucs2 type to UNISTR2 conversion
 857  *
 858  * @param ctx Talloc context to create the dst strcture (if null) and the
 859  *            contents of the unicode string.
 860  * @param dst UNISTR2 destination. If equals null, then it's allocated.
 861  * @param src smb_ucs2_t source.
 862  * @param max_len maximum number of unicode characters to copy. If equals
 863  *        null, then null-termination of src is taken
 864  *
 865  * @return copied UNISTR2 destination
 866  **/
 867 UNISTR2* ucs2_to_unistr2(TALLOC_CTX *ctx, UNISTR2* dst, smb_ucs2_t* src)
 868 {
 869         size_t len;
 870
 871         if (!src) return NULL;
 872         len = strlen_w(src);
 873
 874         /* allocate UNISTR2 destination if not given */
 875         if (!dst) {
 876                 dst = (UNISTR2*) talloc(ctx, sizeof(UNISTR2));
 877                 if (!dst) return NULL;
 878         }
 879         if (!dst->buffer) {
 880                 dst->buffer = (uint16*) talloc(ctx, sizeof(uint16) * (len + 1));
 881                 if (!dst->buffer) return NULL;
 882         }
 883
 884         /* set UNISTR2 parameters */
 885         dst->uni_max_len = len + 1;
 886         dst->undoc = 0;
 887         dst->uni_str_len = len;
 888
 889         /* copy the actual unicode string */
 890         strncpy_w(dst->buffer, src, dst->uni_max_len);
 891
 892         return dst;
 893 };
 894