| 1 | /* | 
|---|
| 2 | * Rtl string functions | 
|---|
| 3 | * | 
|---|
| 4 | * Copyright (C) 1996-1998 Marcus Meissner | 
|---|
| 5 | * Copyright (C) 2000      Alexandre Julliard | 
|---|
| 6 | */ | 
|---|
| 7 |  | 
|---|
| 8 | #include "config.h" | 
|---|
| 9 |  | 
|---|
| 10 | #include <stdlib.h> | 
|---|
| 11 | #include <string.h> | 
|---|
| 12 | #include <ctype.h> | 
|---|
| 13 | #include "wine/unicode.h" | 
|---|
| 14 | #include "heap.h" | 
|---|
| 15 | #include "winnls.h" | 
|---|
| 16 | #include "debugtools.h" | 
|---|
| 17 | #include "ntdll_misc.h" | 
|---|
| 18 | #include "ntddk.h" | 
|---|
| 19 |  | 
|---|
| 20 | DEFAULT_DEBUG_CHANNEL(ntdll); | 
|---|
| 21 |  | 
|---|
| 22 | /* STRING CREATION FUNCTIONS */ | 
|---|
| 23 |  | 
|---|
| 24 | /************************************************************************** | 
|---|
| 25 | *      RtlInitAnsiString   (NTDLL.@) | 
|---|
| 26 | */ | 
|---|
| 27 | void WINAPI RtlInitAnsiString( PSTRING target, LPCSTR source) | 
|---|
| 28 | { | 
|---|
| 29 | if ((target->Buffer = (LPSTR)source)) | 
|---|
| 30 | { | 
|---|
| 31 | target->Length = strlen(source); | 
|---|
| 32 | target->MaximumLength = target->Length + 1; | 
|---|
| 33 | } | 
|---|
| 34 | else target->Length = target->MaximumLength = 0; | 
|---|
| 35 | } | 
|---|
| 36 |  | 
|---|
| 37 |  | 
|---|
| 38 | /************************************************************************** | 
|---|
| 39 | *      RtlInitString   (NTDLL.@) | 
|---|
| 40 | */ | 
|---|
| 41 | void WINAPI RtlInitString( PSTRING target, LPCSTR source ) | 
|---|
| 42 | { | 
|---|
| 43 | return RtlInitAnsiString( target, source ); | 
|---|
| 44 | } | 
|---|
| 45 |  | 
|---|
| 46 |  | 
|---|
| 47 | /************************************************************************** | 
|---|
| 48 | *      RtlFreeAnsiString   (NTDLL.@) | 
|---|
| 49 | */ | 
|---|
| 50 | void WINAPI RtlFreeAnsiString( PSTRING str ) | 
|---|
| 51 | { | 
|---|
| 52 | if (str->Buffer) HeapFree( GetProcessHeap(), 0, str->Buffer ); | 
|---|
| 53 | } | 
|---|
| 54 |  | 
|---|
| 55 |  | 
|---|
| 56 | /************************************************************************** | 
|---|
| 57 | *      RtlFreeOemString   (NTDLL.@) | 
|---|
| 58 | */ | 
|---|
| 59 | void WINAPI RtlFreeOemString( PSTRING str ) | 
|---|
| 60 | { | 
|---|
| 61 | RtlFreeAnsiString( str ); | 
|---|
| 62 | } | 
|---|
| 63 |  | 
|---|
| 64 |  | 
|---|
| 65 | /************************************************************************** | 
|---|
| 66 | *      RtlCopyString   (NTDLL.@) | 
|---|
| 67 | */ | 
|---|
| 68 | void WINAPI RtlCopyString( STRING *dst, const STRING *src ) | 
|---|
| 69 | { | 
|---|
| 70 | if (src) | 
|---|
| 71 | { | 
|---|
| 72 | unsigned int len = min( src->Length, dst->MaximumLength ); | 
|---|
| 73 | memcpy( dst->Buffer, src->Buffer, len ); | 
|---|
| 74 | dst->Length = len; | 
|---|
| 75 | } | 
|---|
| 76 | else dst->Length = 0; | 
|---|
| 77 | } | 
|---|
| 78 |  | 
|---|
| 79 |  | 
|---|
| 80 | /************************************************************************** | 
|---|
| 81 | *      RtlInitUnicodeString   (NTDLL.@) | 
|---|
| 82 | */ | 
|---|
| 83 | void WINAPI RtlInitUnicodeString( PUNICODE_STRING target, LPCWSTR source ) | 
|---|
| 84 | { | 
|---|
| 85 | if ((target->Buffer = (LPWSTR)source)) | 
|---|
| 86 | { | 
|---|
| 87 | target->Length = strlenW(source) * sizeof(WCHAR); | 
|---|
| 88 | target->MaximumLength = target->Length + sizeof(WCHAR); | 
|---|
| 89 | } | 
|---|
| 90 | else target->Length = target->MaximumLength = 0; | 
|---|
| 91 | } | 
|---|
| 92 |  | 
|---|
| 93 |  | 
|---|
| 94 | /************************************************************************** | 
|---|
| 95 | *      RtlCreateUnicodeString   (NTDLL.@) | 
|---|
| 96 | */ | 
|---|
| 97 | BOOLEAN WINAPI RtlCreateUnicodeString( PUNICODE_STRING target, LPCWSTR src ) | 
|---|
| 98 | { | 
|---|
| 99 | int len = (strlenW(src) + 1) * sizeof(WCHAR); | 
|---|
| 100 | if (!(target->Buffer = HeapAlloc( GetProcessHeap(), 0, len ))) return FALSE; | 
|---|
| 101 | memcpy( target->Buffer, src, len ); | 
|---|
| 102 | target->MaximumLength = len; | 
|---|
| 103 | target->Length = len - 2; | 
|---|
| 104 | return TRUE; | 
|---|
| 105 | } | 
|---|
| 106 |  | 
|---|
| 107 |  | 
|---|
| 108 | /************************************************************************** | 
|---|
| 109 | *      RtlCreateUnicodeStringFromAsciiz   (NTDLL.@) | 
|---|
| 110 | */ | 
|---|
| 111 | BOOLEAN WINAPI RtlCreateUnicodeStringFromAsciiz( PUNICODE_STRING target, LPCSTR src ) | 
|---|
| 112 | { | 
|---|
| 113 | STRING ansi; | 
|---|
| 114 | RtlInitAnsiString( &ansi, src ); | 
|---|
| 115 | return !RtlAnsiStringToUnicodeString( target, &ansi, TRUE ); | 
|---|
| 116 | } | 
|---|
| 117 |  | 
|---|
| 118 |  | 
|---|
| 119 | /************************************************************************** | 
|---|
| 120 | *      RtlFreeUnicodeString   (NTDLL.@) | 
|---|
| 121 | */ | 
|---|
| 122 | void WINAPI RtlFreeUnicodeString( PUNICODE_STRING str ) | 
|---|
| 123 | { | 
|---|
| 124 | if (str->Buffer) HeapFree( GetProcessHeap(), 0, str->Buffer ); | 
|---|
| 125 | } | 
|---|
| 126 |  | 
|---|
| 127 |  | 
|---|
| 128 | /************************************************************************** | 
|---|
| 129 | *      RtlCopyUnicodeString   (NTDLL.@) | 
|---|
| 130 | */ | 
|---|
| 131 | void WINAPI RtlCopyUnicodeString( UNICODE_STRING *dst, const UNICODE_STRING *src ) | 
|---|
| 132 | { | 
|---|
| 133 | if (src) | 
|---|
| 134 | { | 
|---|
| 135 | unsigned int len = min( src->Length, dst->MaximumLength ); | 
|---|
| 136 | memcpy( dst->Buffer, src->Buffer, len ); | 
|---|
| 137 | dst->Length = len; | 
|---|
| 138 | /* append terminating NULL if enough space */ | 
|---|
| 139 | if (len < dst->MaximumLength) dst->Buffer[len / sizeof(WCHAR)] = 0; | 
|---|
| 140 | } | 
|---|
| 141 | else dst->Length = 0; | 
|---|
| 142 | } | 
|---|
| 143 |  | 
|---|
| 144 |  | 
|---|
| 145 | /************************************************************************** | 
|---|
| 146 | *      RtlEraseUnicodeString   (NTDLL.@) | 
|---|
| 147 | */ | 
|---|
| 148 | void WINAPI RtlEraseUnicodeString( UNICODE_STRING *str ) | 
|---|
| 149 | { | 
|---|
| 150 | if (str->Buffer) | 
|---|
| 151 | { | 
|---|
| 152 | memset( str->Buffer, 0, str->MaximumLength ); | 
|---|
| 153 | str->Length = 0; | 
|---|
| 154 | } | 
|---|
| 155 | } | 
|---|
| 156 |  | 
|---|
| 157 | /* | 
|---|
| 158 | COMPARISON FUNCTIONS | 
|---|
| 159 | */ | 
|---|
| 160 |  | 
|---|
| 161 | /****************************************************************************** | 
|---|
| 162 | *      RtlCompareString   (NTDLL.@) | 
|---|
| 163 | */ | 
|---|
| 164 | LONG WINAPI RtlCompareString( const STRING *s1, const STRING *s2, BOOLEAN CaseInsensitive ) | 
|---|
| 165 | { | 
|---|
| 166 | unsigned int len; | 
|---|
| 167 | LONG ret = 0; | 
|---|
| 168 | LPCSTR p1, p2; | 
|---|
| 169 |  | 
|---|
| 170 | len = min(s1->Length, s2->Length); | 
|---|
| 171 | p1 = s1->Buffer; | 
|---|
| 172 | p2 = s2->Buffer; | 
|---|
| 173 |  | 
|---|
| 174 | if (CaseInsensitive) | 
|---|
| 175 | { | 
|---|
| 176 | while (!ret && len--) ret = toupper(*p1++) - toupper(*p2++); | 
|---|
| 177 | } | 
|---|
| 178 | else | 
|---|
| 179 | { | 
|---|
| 180 | while (!ret && len--) ret = *p1++ - *p2++; | 
|---|
| 181 | } | 
|---|
| 182 | if (!ret) ret = s1->Length - s2->Length; | 
|---|
| 183 | return ret; | 
|---|
| 184 | } | 
|---|
| 185 |  | 
|---|
| 186 |  | 
|---|
| 187 | /****************************************************************************** | 
|---|
| 188 | *      RtlCompareUnicodeString   (NTDLL.@) | 
|---|
| 189 | */ | 
|---|
| 190 | LONG WINAPI RtlCompareUnicodeString( const UNICODE_STRING *s1, const UNICODE_STRING *s2, | 
|---|
| 191 | BOOLEAN CaseInsensitive ) | 
|---|
| 192 | { | 
|---|
| 193 | unsigned int len; | 
|---|
| 194 | LONG ret = 0; | 
|---|
| 195 | LPCWSTR p1, p2; | 
|---|
| 196 |  | 
|---|
| 197 | len = min(s1->Length, s2->Length) / sizeof(WCHAR); | 
|---|
| 198 | p1 = s1->Buffer; | 
|---|
| 199 | p2 = s2->Buffer; | 
|---|
| 200 |  | 
|---|
| 201 | if (CaseInsensitive) | 
|---|
| 202 | { | 
|---|
| 203 | while (!ret && len--) ret = toupperW(*p1++) - toupperW(*p2++); | 
|---|
| 204 | } | 
|---|
| 205 | else | 
|---|
| 206 | { | 
|---|
| 207 | while (!ret && len--) ret = *p1++ - *p2++; | 
|---|
| 208 | } | 
|---|
| 209 | if (!ret) ret = s1->Length - s2->Length; | 
|---|
| 210 | return ret; | 
|---|
| 211 | } | 
|---|
| 212 |  | 
|---|
| 213 |  | 
|---|
| 214 | /************************************************************************** | 
|---|
| 215 | *      RtlEqualString   (NTDLL.@) | 
|---|
| 216 | */ | 
|---|
| 217 | BOOLEAN WINAPI RtlEqualString( const STRING *s1, const STRING *s2, BOOLEAN CaseInsensitive ) | 
|---|
| 218 | { | 
|---|
| 219 | if (s1->Length != s2->Length) return FALSE; | 
|---|
| 220 | return !RtlCompareString( s1, s2, CaseInsensitive ); | 
|---|
| 221 | } | 
|---|
| 222 |  | 
|---|
| 223 |  | 
|---|
| 224 | /************************************************************************** | 
|---|
| 225 | *      RtlEqualUnicodeString   (NTDLL.@) | 
|---|
| 226 | */ | 
|---|
| 227 | BOOLEAN WINAPI RtlEqualUnicodeString( const UNICODE_STRING *s1, const UNICODE_STRING *s2, | 
|---|
| 228 | BOOLEAN CaseInsensitive ) | 
|---|
| 229 | { | 
|---|
| 230 | if (s1->Length != s2->Length) return FALSE; | 
|---|
| 231 | return !RtlCompareUnicodeString( s1, s2, CaseInsensitive ); | 
|---|
| 232 | } | 
|---|
| 233 |  | 
|---|
| 234 |  | 
|---|
| 235 | /************************************************************************** | 
|---|
| 236 | *      RtlPrefixString   (NTDLL.@) | 
|---|
| 237 | * | 
|---|
| 238 | * Test if s1 is a prefix in s2 | 
|---|
| 239 | */ | 
|---|
| 240 | BOOLEAN WINAPI RtlPrefixString( const STRING *s1, const STRING *s2, BOOLEAN ignore_case ) | 
|---|
| 241 | { | 
|---|
| 242 | unsigned int i; | 
|---|
| 243 |  | 
|---|
| 244 | if (s1->Length > s2->Length) return FALSE; | 
|---|
| 245 | if (ignore_case) | 
|---|
| 246 | { | 
|---|
| 247 | for (i = 0; i < s1->Length; i++) | 
|---|
| 248 | if (toupper(s1->Buffer[i]) != toupper(s2->Buffer[i])) return FALSE; | 
|---|
| 249 | } | 
|---|
| 250 | else | 
|---|
| 251 | { | 
|---|
| 252 | for (i = 0; i < s1->Length; i++) | 
|---|
| 253 | if (s1->Buffer[i] != s2->Buffer[i]) return FALSE; | 
|---|
| 254 | } | 
|---|
| 255 | return TRUE; | 
|---|
| 256 | } | 
|---|
| 257 |  | 
|---|
| 258 |  | 
|---|
| 259 | /************************************************************************** | 
|---|
| 260 | *      RtlPrefixUnicodeString   (NTDLL.@) | 
|---|
| 261 | * | 
|---|
| 262 | * Test if s1 is a prefix in s2 | 
|---|
| 263 | */ | 
|---|
| 264 | BOOLEAN WINAPI RtlPrefixUnicodeString( const UNICODE_STRING *s1, | 
|---|
| 265 | const UNICODE_STRING *s2, | 
|---|
| 266 | BOOLEAN ignore_case ) | 
|---|
| 267 | { | 
|---|
| 268 | unsigned int i; | 
|---|
| 269 |  | 
|---|
| 270 | if (s1->Length > s2->Length) return FALSE; | 
|---|
| 271 | if (ignore_case) | 
|---|
| 272 | { | 
|---|
| 273 | for (i = 0; i < s1->Length / sizeof(WCHAR); i++) | 
|---|
| 274 | if (toupper(s1->Buffer[i]) != toupper(s2->Buffer[i])) return FALSE; | 
|---|
| 275 | } | 
|---|
| 276 | else | 
|---|
| 277 | { | 
|---|
| 278 | for (i = 0; i < s1->Length / sizeof(WCHAR); i++) | 
|---|
| 279 | if (s1->Buffer[i] != s2->Buffer[i]) return FALSE; | 
|---|
| 280 | } | 
|---|
| 281 | return TRUE; | 
|---|
| 282 | } | 
|---|
| 283 |  | 
|---|
| 284 |  | 
|---|
| 285 | /* | 
|---|
| 286 | COPY BETWEEN ANSI_STRING or UNICODE_STRING | 
|---|
| 287 | there is no parameter checking, it just crashes | 
|---|
| 288 | */ | 
|---|
| 289 |  | 
|---|
| 290 |  | 
|---|
| 291 | /************************************************************************** | 
|---|
| 292 | *      RtlAnsiStringToUnicodeString   (NTDLL.@) | 
|---|
| 293 | * | 
|---|
| 294 | * NOTES: | 
|---|
| 295 | *  writes terminating 0 | 
|---|
| 296 | */ | 
|---|
| 297 | NTSTATUS WINAPI RtlAnsiStringToUnicodeString( UNICODE_STRING *uni, | 
|---|
| 298 | const STRING *ansi, | 
|---|
| 299 | BOOLEAN doalloc ) | 
|---|
| 300 | { | 
|---|
| 301 | DWORD len = MultiByteToWideChar( CP_ACP, 0, ansi->Buffer, ansi->Length, NULL, 0 ); | 
|---|
| 302 | DWORD total = (len + 1) * sizeof(WCHAR); | 
|---|
| 303 |  | 
|---|
| 304 | if (total > 0xffff) return STATUS_INVALID_PARAMETER_2; | 
|---|
| 305 | uni->Length = len * sizeof(WCHAR); | 
|---|
| 306 | if (doalloc) | 
|---|
| 307 | { | 
|---|
| 308 | uni->MaximumLength = total; | 
|---|
| 309 | if (!(uni->Buffer = HeapAlloc( GetProcessHeap(), 0, total ))) return STATUS_NO_MEMORY; | 
|---|
| 310 | } | 
|---|
| 311 | else if (total > uni->MaximumLength) return STATUS_BUFFER_OVERFLOW; | 
|---|
| 312 |  | 
|---|
| 313 | MultiByteToWideChar( CP_ACP, 0, ansi->Buffer, ansi->Length, uni->Buffer, len ); | 
|---|
| 314 | uni->Buffer[len] = 0; | 
|---|
| 315 | return STATUS_SUCCESS; | 
|---|
| 316 | } | 
|---|
| 317 |  | 
|---|
| 318 |  | 
|---|
| 319 | /************************************************************************** | 
|---|
| 320 | *      RtlOemStringToUnicodeString   (NTDLL.@) | 
|---|
| 321 | * | 
|---|
| 322 | * NOTES | 
|---|
| 323 | *  writes terminating 0 | 
|---|
| 324 | *  if resulting length > 0xffff it returns STATUS_INVALID_PARAMETER_2 | 
|---|
| 325 | */ | 
|---|
| 326 | NTSTATUS WINAPI RtlOemStringToUnicodeString( UNICODE_STRING *uni, | 
|---|
| 327 | const STRING *oem, | 
|---|
| 328 | BOOLEAN doalloc ) | 
|---|
| 329 | { | 
|---|
| 330 | DWORD len = MultiByteToWideChar( CP_OEMCP, 0, oem->Buffer, oem->Length, NULL, 0 ); | 
|---|
| 331 | DWORD total = (len + 1) * sizeof(WCHAR); | 
|---|
| 332 |  | 
|---|
| 333 | if (total > 0xffff) return STATUS_INVALID_PARAMETER_2; | 
|---|
| 334 | uni->Length = len * sizeof(WCHAR); | 
|---|
| 335 | if (doalloc) | 
|---|
| 336 | { | 
|---|
| 337 | uni->MaximumLength = total; | 
|---|
| 338 | if (!(uni->Buffer = HeapAlloc( GetProcessHeap(), 0, total ))) return STATUS_NO_MEMORY; | 
|---|
| 339 | } | 
|---|
| 340 | else if (total > uni->MaximumLength) return STATUS_BUFFER_OVERFLOW; | 
|---|
| 341 |  | 
|---|
| 342 | MultiByteToWideChar( CP_OEMCP, 0, oem->Buffer, oem->Length, uni->Buffer, len ); | 
|---|
| 343 | uni->Buffer[len] = 0; | 
|---|
| 344 | return STATUS_SUCCESS; | 
|---|
| 345 | } | 
|---|
| 346 |  | 
|---|
| 347 |  | 
|---|
| 348 | /************************************************************************** | 
|---|
| 349 | *      RtlUnicodeStringToAnsiString   (NTDLL.@) | 
|---|
| 350 | * | 
|---|
| 351 | * NOTES | 
|---|
| 352 | *  writes terminating 0 | 
|---|
| 353 | *  copies a part if the buffer is too small | 
|---|
| 354 | */ | 
|---|
| 355 | NTSTATUS WINAPI RtlUnicodeStringToAnsiString( STRING *ansi, | 
|---|
| 356 | const UNICODE_STRING *uni, | 
|---|
| 357 | BOOLEAN doalloc ) | 
|---|
| 358 | { | 
|---|
| 359 | NTSTATUS ret = STATUS_SUCCESS; | 
|---|
| 360 | DWORD len = RtlUnicodeStringToAnsiSize( uni ); | 
|---|
| 361 |  | 
|---|
| 362 | ansi->Length = len; | 
|---|
| 363 | if (doalloc) | 
|---|
| 364 | { | 
|---|
| 365 | ansi->MaximumLength = len + 1; | 
|---|
| 366 | if (!(ansi->Buffer = HeapAlloc( GetProcessHeap(), 0, len + 1 ))) return STATUS_NO_MEMORY; | 
|---|
| 367 | } | 
|---|
| 368 | else if (ansi->MaximumLength <= len) | 
|---|
| 369 | { | 
|---|
| 370 | if (!ansi->MaximumLength) return STATUS_BUFFER_OVERFLOW; | 
|---|
| 371 | ansi->Length = ansi->MaximumLength - 1; | 
|---|
| 372 | ret = STATUS_BUFFER_OVERFLOW; | 
|---|
| 373 | } | 
|---|
| 374 |  | 
|---|
| 375 | WideCharToMultiByte( CP_ACP, 0, uni->Buffer, uni->Length / sizeof(WCHAR), | 
|---|
| 376 | ansi->Buffer, ansi->Length, NULL, NULL ); | 
|---|
| 377 | ansi->Buffer[ansi->Length] = 0; | 
|---|
| 378 | return ret; | 
|---|
| 379 | } | 
|---|
| 380 |  | 
|---|
| 381 |  | 
|---|
| 382 | /************************************************************************** | 
|---|
| 383 | *      RtlUnicodeStringToOemString   (NTDLL.@) | 
|---|
| 384 | * | 
|---|
| 385 | * NOTES | 
|---|
| 386 | *   allocates uni->Length+1 | 
|---|
| 387 | *   writes terminating 0 | 
|---|
| 388 | */ | 
|---|
| 389 | NTSTATUS WINAPI RtlUnicodeStringToOemString( STRING *oem, | 
|---|
| 390 | const UNICODE_STRING *uni, | 
|---|
| 391 | BOOLEAN doalloc ) | 
|---|
| 392 | { | 
|---|
| 393 | NTSTATUS ret = STATUS_SUCCESS; | 
|---|
| 394 | DWORD len = RtlUnicodeStringToOemSize( uni ); | 
|---|
| 395 |  | 
|---|
| 396 | oem->Length = len; | 
|---|
| 397 | if (doalloc) | 
|---|
| 398 | { | 
|---|
| 399 | oem->MaximumLength = len + 1; | 
|---|
| 400 | if (!(oem->Buffer = HeapAlloc( GetProcessHeap(), 0, len + 1 ))) return STATUS_NO_MEMORY; | 
|---|
| 401 | } | 
|---|
| 402 | else if (oem->MaximumLength <= len) | 
|---|
| 403 | { | 
|---|
| 404 | if (!oem->MaximumLength) return STATUS_BUFFER_OVERFLOW; | 
|---|
| 405 | oem->Length = oem->MaximumLength - 1; | 
|---|
| 406 | ret = STATUS_BUFFER_OVERFLOW; | 
|---|
| 407 | } | 
|---|
| 408 |  | 
|---|
| 409 | WideCharToMultiByte( CP_OEMCP, 0, uni->Buffer, uni->Length / sizeof(WCHAR), | 
|---|
| 410 | oem->Buffer, oem->Length, NULL, NULL ); | 
|---|
| 411 | oem->Buffer[oem->Length] = 0; | 
|---|
| 412 | return ret; | 
|---|
| 413 | } | 
|---|
| 414 |  | 
|---|
| 415 |  | 
|---|
| 416 | /************************************************************************** | 
|---|
| 417 | *      RtlMultiByteToUnicodeN   (NTDLL.@) | 
|---|
| 418 | * | 
|---|
| 419 | * NOTES | 
|---|
| 420 | *  if unistr is too small a part is copied | 
|---|
| 421 | */ | 
|---|
| 422 | NTSTATUS WINAPI RtlMultiByteToUnicodeN( LPWSTR dst, DWORD dstlen, LPDWORD reslen, | 
|---|
| 423 | LPCSTR src, DWORD srclen ) | 
|---|
| 424 | { | 
|---|
| 425 | DWORD res = MultiByteToWideChar( CP_ACP, 0, src, srclen, dst, dstlen/sizeof(WCHAR) ); | 
|---|
| 426 | if (reslen) | 
|---|
| 427 | *reslen = res ? res * sizeof(WCHAR) : dstlen; /* overflow -> we filled up to dstlen */ | 
|---|
| 428 | return STATUS_SUCCESS; | 
|---|
| 429 | } | 
|---|
| 430 |  | 
|---|
| 431 |  | 
|---|
| 432 | /************************************************************************** | 
|---|
| 433 | *      RtlOemToUnicodeN   (NTDLL.@) | 
|---|
| 434 | */ | 
|---|
| 435 | NTSTATUS WINAPI RtlOemToUnicodeN( LPWSTR dst, DWORD dstlen, LPDWORD reslen, | 
|---|
| 436 | LPCSTR src, DWORD srclen ) | 
|---|
| 437 | { | 
|---|
| 438 | DWORD res = MultiByteToWideChar( CP_OEMCP, 0, src, srclen, dst, dstlen/sizeof(WCHAR) ); | 
|---|
| 439 | if (reslen) | 
|---|
| 440 | *reslen = res ? res * sizeof(WCHAR) : dstlen; /* overflow -> we filled up to dstlen */ | 
|---|
| 441 | return STATUS_SUCCESS; | 
|---|
| 442 | } | 
|---|
| 443 |  | 
|---|
| 444 |  | 
|---|
| 445 | /************************************************************************** | 
|---|
| 446 | *      RtlUnicodeToMultiByteN   (NTDLL.@) | 
|---|
| 447 | */ | 
|---|
| 448 | NTSTATUS WINAPI RtlUnicodeToMultiByteN( LPSTR dst, DWORD dstlen, LPDWORD reslen, | 
|---|
| 449 | LPCWSTR src, DWORD srclen ) | 
|---|
| 450 | { | 
|---|
| 451 | DWORD res = WideCharToMultiByte( CP_ACP, 0, src, srclen/sizeof(WCHAR), | 
|---|
| 452 | dst, dstlen, NULL, NULL ); | 
|---|
| 453 | if (reslen) | 
|---|
| 454 | *reslen = res ? res * sizeof(WCHAR) : dstlen; /* overflow -> we filled up to dstlen */ | 
|---|
| 455 | return STATUS_SUCCESS; | 
|---|
| 456 | } | 
|---|
| 457 |  | 
|---|
| 458 |  | 
|---|
| 459 | /************************************************************************** | 
|---|
| 460 | *      RtlUnicodeToOemN   (NTDLL.@) | 
|---|
| 461 | */ | 
|---|
| 462 | NTSTATUS WINAPI RtlUnicodeToOemN( LPSTR dst, DWORD dstlen, LPDWORD reslen, | 
|---|
| 463 | LPCWSTR src, DWORD srclen ) | 
|---|
| 464 | { | 
|---|
| 465 | DWORD res = WideCharToMultiByte( CP_OEMCP, 0, src, srclen/sizeof(WCHAR), | 
|---|
| 466 | dst, dstlen, NULL, NULL ); | 
|---|
| 467 | if (reslen) | 
|---|
| 468 | *reslen = res ? res * sizeof(WCHAR) : dstlen; /* overflow -> we filled up to dstlen */ | 
|---|
| 469 | return STATUS_SUCCESS; | 
|---|
| 470 | } | 
|---|
| 471 |  | 
|---|
| 472 |  | 
|---|
| 473 | /* | 
|---|
| 474 | CASE CONVERSIONS | 
|---|
| 475 | */ | 
|---|
| 476 |  | 
|---|
| 477 | /************************************************************************** | 
|---|
| 478 | *      RtlUpperString   (NTDLL.@) | 
|---|
| 479 | */ | 
|---|
| 480 | void WINAPI RtlUpperString( STRING *dst, const STRING *src ) | 
|---|
| 481 | { | 
|---|
| 482 | unsigned int i, len = min(src->Length, dst->MaximumLength); | 
|---|
| 483 |  | 
|---|
| 484 | for (i = 0; i < len; i++) dst->Buffer[i] = toupper(src->Buffer[i]); | 
|---|
| 485 | dst->Length = len; | 
|---|
| 486 | } | 
|---|
| 487 |  | 
|---|
| 488 |  | 
|---|
| 489 | /************************************************************************** | 
|---|
| 490 | *      RtlUpcaseUnicodeString   (NTDLL.@) | 
|---|
| 491 | * | 
|---|
| 492 | * NOTES: | 
|---|
| 493 | *  destination string is never 0-terminated because dest can be equal to src | 
|---|
| 494 | *  and src might be not 0-terminated | 
|---|
| 495 | *  dest.Length only set when success | 
|---|
| 496 | */ | 
|---|
| 497 | NTSTATUS WINAPI RtlUpcaseUnicodeString( UNICODE_STRING *dest, | 
|---|
| 498 | const UNICODE_STRING *src, | 
|---|
| 499 | BOOLEAN doalloc ) | 
|---|
| 500 | { | 
|---|
| 501 | DWORD i, len = src->Length; | 
|---|
| 502 |  | 
|---|
| 503 | if (doalloc) | 
|---|
| 504 | { | 
|---|
| 505 | dest->MaximumLength = len; | 
|---|
| 506 | if (!(dest->Buffer = HeapAlloc( GetProcessHeap(), 0, len ))) return STATUS_NO_MEMORY; | 
|---|
| 507 | } | 
|---|
| 508 | else if (len > dest->MaximumLength) return STATUS_BUFFER_OVERFLOW; | 
|---|
| 509 |  | 
|---|
| 510 | for (i = 0; i < len/sizeof(WCHAR); i++) dest->Buffer[i] = toupperW(src->Buffer[i]); | 
|---|
| 511 | dest->Length = len; | 
|---|
| 512 | return STATUS_SUCCESS; | 
|---|
| 513 | } | 
|---|
| 514 |  | 
|---|
| 515 |  | 
|---|
| 516 | /************************************************************************** | 
|---|
| 517 | *      RtlUpcaseUnicodeStringToAnsiString   (NTDLL.@) | 
|---|
| 518 | * | 
|---|
| 519 | * NOTES | 
|---|
| 520 | *  writes terminating 0 | 
|---|
| 521 | */ | 
|---|
| 522 | NTSTATUS WINAPI RtlUpcaseUnicodeStringToAnsiString( STRING *dst, | 
|---|
| 523 | const UNICODE_STRING *src, | 
|---|
| 524 | BOOLEAN doalloc ) | 
|---|
| 525 | { | 
|---|
| 526 | NTSTATUS ret; | 
|---|
| 527 | UNICODE_STRING upcase; | 
|---|
| 528 |  | 
|---|
| 529 | if (!(ret = RtlUpcaseUnicodeString( &upcase, src, TRUE ))) | 
|---|
| 530 | { | 
|---|
| 531 | ret = RtlUnicodeStringToAnsiString( dst, &upcase, doalloc ); | 
|---|
| 532 | RtlFreeUnicodeString( &upcase ); | 
|---|
| 533 | } | 
|---|
| 534 | return ret; | 
|---|
| 535 | } | 
|---|
| 536 |  | 
|---|
| 537 |  | 
|---|
| 538 | /************************************************************************** | 
|---|
| 539 | *      RtlUpcaseUnicodeStringToOemString   (NTDLL.@) | 
|---|
| 540 | * | 
|---|
| 541 | * NOTES | 
|---|
| 542 | *  writes terminating 0 | 
|---|
| 543 | */ | 
|---|
| 544 | NTSTATUS WINAPI RtlUpcaseUnicodeStringToOemString( STRING *dst, | 
|---|
| 545 | const UNICODE_STRING *src, | 
|---|
| 546 | BOOLEAN doalloc ) | 
|---|
| 547 | { | 
|---|
| 548 | NTSTATUS ret; | 
|---|
| 549 | UNICODE_STRING upcase; | 
|---|
| 550 |  | 
|---|
| 551 | if (!(ret = RtlUpcaseUnicodeString( &upcase, src, TRUE ))) | 
|---|
| 552 | { | 
|---|
| 553 | ret = RtlUnicodeStringToOemString( dst, &upcase, doalloc ); | 
|---|
| 554 | RtlFreeUnicodeString( &upcase ); | 
|---|
| 555 | } | 
|---|
| 556 | return ret; | 
|---|
| 557 | } | 
|---|
| 558 |  | 
|---|
| 559 |  | 
|---|
| 560 | /************************************************************************** | 
|---|
| 561 | *      RtlUpcaseUnicodeToMultiByteN   (NTDLL.@) | 
|---|
| 562 | */ | 
|---|
| 563 | NTSTATUS WINAPI RtlUpcaseUnicodeToMultiByteN( LPSTR dst, DWORD dstlen, LPDWORD reslen, | 
|---|
| 564 | LPCWSTR src, DWORD srclen ) | 
|---|
| 565 | { | 
|---|
| 566 | NTSTATUS ret; | 
|---|
| 567 | LPWSTR upcase; | 
|---|
| 568 | DWORD i; | 
|---|
| 569 |  | 
|---|
| 570 | if (!(upcase = HeapAlloc( GetProcessHeap(), 0, srclen ))) return STATUS_NO_MEMORY; | 
|---|
| 571 | for (i = 0; i < srclen/sizeof(WCHAR); i++) upcase[i] = toupperW(src[i]); | 
|---|
| 572 | ret = RtlUnicodeToMultiByteN( dst, dstlen, reslen, upcase, srclen ); | 
|---|
| 573 | HeapFree( GetProcessHeap(), 0, upcase ); | 
|---|
| 574 | return ret; | 
|---|
| 575 | } | 
|---|
| 576 |  | 
|---|
| 577 |  | 
|---|
| 578 | /************************************************************************** | 
|---|
| 579 | *      RtlUpcaseUnicodeToOemN   (NTDLL.@) | 
|---|
| 580 | */ | 
|---|
| 581 | NTSTATUS WINAPI RtlUpcaseUnicodeToOemN( LPSTR dst, DWORD dstlen, LPDWORD reslen, | 
|---|
| 582 | LPCWSTR src, DWORD srclen ) | 
|---|
| 583 | { | 
|---|
| 584 | NTSTATUS ret; | 
|---|
| 585 | LPWSTR upcase; | 
|---|
| 586 | DWORD i; | 
|---|
| 587 |  | 
|---|
| 588 | if (!(upcase = HeapAlloc( GetProcessHeap(), 0, srclen ))) return STATUS_NO_MEMORY; | 
|---|
| 589 | for (i = 0; i < srclen/sizeof(WCHAR); i++) upcase[i] = toupperW(src[i]); | 
|---|
| 590 | ret = RtlUnicodeToOemN( dst, dstlen, reslen, upcase, srclen ); | 
|---|
| 591 | HeapFree( GetProcessHeap(), 0, upcase ); | 
|---|
| 592 | return ret; | 
|---|
| 593 | } | 
|---|
| 594 |  | 
|---|
| 595 |  | 
|---|
| 596 | /* | 
|---|
| 597 | STRING SIZE | 
|---|
| 598 | */ | 
|---|
| 599 |  | 
|---|
| 600 | /************************************************************************** | 
|---|
| 601 | *      RtlOemStringToUnicodeSize   (NTDLL.@) | 
|---|
| 602 | * | 
|---|
| 603 | * Return the size in bytes necessary for the Unicode conversion of 'str', | 
|---|
| 604 | * including the terminating NULL. | 
|---|
| 605 | */ | 
|---|
| 606 | UINT WINAPI RtlOemStringToUnicodeSize(PSTRING str) | 
|---|
| 607 | { | 
|---|
| 608 | DWORD ret = MultiByteToWideChar( CP_OEMCP, 0, str->Buffer, str->Length, NULL, 0 ); | 
|---|
| 609 | return (ret + 1) * sizeof(WCHAR); | 
|---|
| 610 | } | 
|---|
| 611 |  | 
|---|
| 612 |  | 
|---|
| 613 | /************************************************************************** | 
|---|
| 614 | *      RtlAnsiStringToUnicodeSize   (NTDLL.@) | 
|---|
| 615 | * | 
|---|
| 616 | * Return the size in bytes necessary for the Unicode conversion of 'str', | 
|---|
| 617 | * including the terminating NULL. | 
|---|
| 618 | */ | 
|---|
| 619 | DWORD WINAPI RtlAnsiStringToUnicodeSize(PSTRING str) | 
|---|
| 620 | { | 
|---|
| 621 | DWORD ret = MultiByteToWideChar( CP_ACP, 0, str->Buffer, str->Length, NULL, 0 ); | 
|---|
| 622 | return (ret + 1) * sizeof(WCHAR); | 
|---|
| 623 | } | 
|---|
| 624 |  | 
|---|
| 625 |  | 
|---|
| 626 | /************************************************************************** | 
|---|
| 627 | *      RtlMultiByteToUnicodeSize   (NTDLL.@) | 
|---|
| 628 | * | 
|---|
| 629 | * Compute the size in bytes necessary for the Unicode conversion of 'str', | 
|---|
| 630 | * without the terminating NULL. | 
|---|
| 631 | */ | 
|---|
| 632 | NTSTATUS WINAPI RtlMultiByteToUnicodeSize( DWORD *size, LPCSTR str, UINT len ) | 
|---|
| 633 | { | 
|---|
| 634 | *size = MultiByteToWideChar( CP_ACP, 0, str, len, NULL, 0 ) * sizeof(WCHAR); | 
|---|
| 635 | return 0; | 
|---|
| 636 | } | 
|---|
| 637 |  | 
|---|
| 638 |  | 
|---|
| 639 | /************************************************************************** | 
|---|
| 640 | *      RtlUnicodeToMultiByteSize   (NTDLL.@) | 
|---|
| 641 | * | 
|---|
| 642 | * Compute the size necessary for the multibyte conversion of 'str', | 
|---|
| 643 | * without the terminating NULL. | 
|---|
| 644 | */ | 
|---|
| 645 | NTSTATUS WINAPI RtlUnicodeToMultiByteSize( DWORD *size, LPCWSTR str, UINT len ) | 
|---|
| 646 | { | 
|---|
| 647 | *size = WideCharToMultiByte( CP_ACP, 0, str, len / sizeof(WCHAR), NULL, 0, NULL, NULL ); | 
|---|
| 648 | return 0; | 
|---|
| 649 | } | 
|---|
| 650 |  | 
|---|
| 651 |  | 
|---|
| 652 | /************************************************************************** | 
|---|
| 653 | *      RtlUnicodeStringToAnsiSize   (NTDLL.@) | 
|---|
| 654 | * | 
|---|
| 655 | * Return the size in bytes necessary for the Ansi conversion of 'str', | 
|---|
| 656 | * including the terminating NULL. | 
|---|
| 657 | */ | 
|---|
| 658 | DWORD WINAPI RtlUnicodeStringToAnsiSize( const UNICODE_STRING *str ) | 
|---|
| 659 | { | 
|---|
| 660 | return WideCharToMultiByte( CP_ACP, 0, str->Buffer, str->Length / sizeof(WCHAR), | 
|---|
| 661 | NULL, 0, NULL, NULL ) + 1; | 
|---|
| 662 | } | 
|---|
| 663 |  | 
|---|
| 664 |  | 
|---|
| 665 | /************************************************************************** | 
|---|
| 666 | *      RtlUnicodeStringToOemSize   (NTDLL.@) | 
|---|
| 667 | * | 
|---|
| 668 | * Return the size in bytes necessary for the OEM conversion of 'str', | 
|---|
| 669 | * including the terminating NULL. | 
|---|
| 670 | */ | 
|---|
| 671 | DWORD WINAPI RtlUnicodeStringToOemSize( const UNICODE_STRING *str ) | 
|---|
| 672 | { | 
|---|
| 673 | return WideCharToMultiByte( CP_OEMCP, 0, str->Buffer, str->Length / sizeof(WCHAR), | 
|---|
| 674 | NULL, 0, NULL, NULL ) + 1; | 
|---|
| 675 | } | 
|---|
| 676 |  | 
|---|
| 677 |  | 
|---|
| 678 | /************************************************************************** | 
|---|
| 679 | *      RtlAppendStringToString   (NTDLL.@) | 
|---|
| 680 | */ | 
|---|
| 681 | NTSTATUS WINAPI RtlAppendStringToString( STRING *dst, const STRING *src ) | 
|---|
| 682 | { | 
|---|
| 683 | unsigned int len = src->Length + dst->Length; | 
|---|
| 684 | if (len > dst->MaximumLength) return STATUS_BUFFER_TOO_SMALL; | 
|---|
| 685 | memcpy( dst->Buffer + dst->Length, src->Buffer, src->Length ); | 
|---|
| 686 | dst->Length = len; | 
|---|
| 687 | return STATUS_SUCCESS; | 
|---|
| 688 | } | 
|---|
| 689 |  | 
|---|
| 690 |  | 
|---|
| 691 | /************************************************************************** | 
|---|
| 692 | *      RtlAppendAsciizToString   (NTDLL.@) | 
|---|
| 693 | */ | 
|---|
| 694 | NTSTATUS WINAPI RtlAppendAsciizToString( STRING *dst, LPCSTR src ) | 
|---|
| 695 | { | 
|---|
| 696 | if (src) | 
|---|
| 697 | { | 
|---|
| 698 | unsigned int srclen = strlen(src); | 
|---|
| 699 | unsigned int total  = srclen + dst->Length; | 
|---|
| 700 | if (total > dst->MaximumLength) return STATUS_BUFFER_TOO_SMALL; | 
|---|
| 701 | memcpy( dst->Buffer + dst->Length, src, srclen ); | 
|---|
| 702 | dst->Length = total; | 
|---|
| 703 | } | 
|---|
| 704 | return STATUS_SUCCESS; | 
|---|
| 705 | } | 
|---|
| 706 |  | 
|---|
| 707 |  | 
|---|
| 708 | /************************************************************************** | 
|---|
| 709 | *      RtlAppendUnicodeToString   (NTDLL.@) | 
|---|
| 710 | */ | 
|---|
| 711 | NTSTATUS WINAPI RtlAppendUnicodeToString( UNICODE_STRING *dst, LPCWSTR src ) | 
|---|
| 712 | { | 
|---|
| 713 | if (src) | 
|---|
| 714 | { | 
|---|
| 715 | unsigned int srclen = strlenW(src) * sizeof(WCHAR); | 
|---|
| 716 | unsigned int total  = srclen + dst->Length; | 
|---|
| 717 | if (total > dst->MaximumLength) return STATUS_BUFFER_TOO_SMALL; | 
|---|
| 718 | memcpy( dst->Buffer + dst->Length/sizeof(WCHAR), src, srclen ); | 
|---|
| 719 | dst->Length = total; | 
|---|
| 720 | /* append terminating NULL if enough space */ | 
|---|
| 721 | if (total < dst->MaximumLength) dst->Buffer[total / sizeof(WCHAR)] = 0; | 
|---|
| 722 | } | 
|---|
| 723 | return STATUS_SUCCESS; | 
|---|
| 724 | } | 
|---|
| 725 |  | 
|---|
| 726 |  | 
|---|
| 727 | /************************************************************************** | 
|---|
| 728 | *      RtlAppendUnicodeStringToString   (NTDLL.@) | 
|---|
| 729 | */ | 
|---|
| 730 | NTSTATUS WINAPI RtlAppendUnicodeStringToString( UNICODE_STRING *dst, const UNICODE_STRING *src ) | 
|---|
| 731 | { | 
|---|
| 732 | unsigned int len = src->Length + dst->Length; | 
|---|
| 733 | if (len > dst->MaximumLength) return STATUS_BUFFER_TOO_SMALL; | 
|---|
| 734 | memcpy( dst->Buffer + dst->Length/sizeof(WCHAR), src->Buffer, src->Length ); | 
|---|
| 735 | dst->Length = len; | 
|---|
| 736 | /* append terminating NULL if enough space */ | 
|---|
| 737 | if (len < dst->MaximumLength) dst->Buffer[len / sizeof(WCHAR)] = 0; | 
|---|
| 738 | return STATUS_SUCCESS; | 
|---|
| 739 | } | 
|---|
| 740 |  | 
|---|
| 741 |  | 
|---|
| 742 | /* | 
|---|
| 743 | MISC | 
|---|
| 744 | */ | 
|---|
| 745 |  | 
|---|
| 746 | /************************************************************************** | 
|---|
| 747 | *      RtlIsTextUnicode | 
|---|
| 748 | * | 
|---|
| 749 | *      Apply various feeble heuristics to guess whether | 
|---|
| 750 | *      the text buffer contains Unicode. | 
|---|
| 751 | *      FIXME: should implement more tests. | 
|---|
| 752 | */ | 
|---|
| 753 | DWORD WINAPI RtlIsTextUnicode( | 
|---|
| 754 | LPVOID buf, | 
|---|
| 755 | DWORD len, | 
|---|
| 756 | DWORD *pf) | 
|---|
| 757 | { | 
|---|
| 758 | LPWSTR s = buf; | 
|---|
| 759 | DWORD flags = -1, out_flags = 0; | 
|---|
| 760 |  | 
|---|
| 761 | if (!len) | 
|---|
| 762 | goto out; | 
|---|
| 763 | if (pf) | 
|---|
| 764 | flags = *pf; | 
|---|
| 765 | /* | 
|---|
| 766 | * Apply various tests to the text string. According to the | 
|---|
| 767 | * docs, each test "passed" sets the corresponding flag in | 
|---|
| 768 | * the output flags. But some of the tests are mutually | 
|---|
| 769 | * exclusive, so I don't see how you could pass all tests ... | 
|---|
| 770 | */ | 
|---|
| 771 |  | 
|---|
| 772 | /* Check for an odd length ... pass if even. */ | 
|---|
| 773 | if (!(len & 1)) | 
|---|
| 774 | out_flags |= IS_TEXT_UNICODE_ODD_LENGTH; | 
|---|
| 775 |  | 
|---|
| 776 | /* Check for the special unicode marker byte. */ | 
|---|
| 777 | if (*s == 0xFEFF) | 
|---|
| 778 | out_flags |= IS_TEXT_UNICODE_SIGNATURE; | 
|---|
| 779 |  | 
|---|
| 780 | /* | 
|---|
| 781 | * Check whether the string passed all of the tests. | 
|---|
| 782 | */ | 
|---|
| 783 | flags &= ITU_IMPLEMENTED_TESTS; | 
|---|
| 784 | if ((out_flags & flags) != flags) | 
|---|
| 785 | len = 0; | 
|---|
| 786 | out: | 
|---|
| 787 | if (pf) | 
|---|
| 788 | *pf = out_flags; | 
|---|
| 789 | return len; | 
|---|
| 790 | } | 
|---|