source: branches/samba-3.3.x/source/librpc/ndr/ndr_string.c

Last change on this file was 206, checked in by Herwig Bauernfeind, 16 years ago

Import Samba 3.3 branch at 3.0.0 level (psmedley's port)
File size: 19.1 KB
Line 
1/*
2 Unix SMB/CIFS implementation.
3
4 routines for marshalling/unmarshalling string types
5
6 Copyright (C) Andrew Tridgell 2003
7
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3 of the License, or
11 (at your option) any later version.
12
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
17
18 You should have received a copy of the GNU General Public License
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
20*/
21
22#include "includes.h"
23#include "librpc/ndr/libndr.h"
24
25/**
26 pull a general string from the wire
27*/
28_PUBLIC_ enum ndr_err_code ndr_pull_string(struct ndr_pull *ndr, int ndr_flags, const char **s)
29{
30 char *as=NULL;
31 uint32_t len1, ofs, len2;
32 uint16_t len3;
33 size_t converted_size;
34 int chset = CH_UTF16;
35 unsigned byte_mul = 2;
36 unsigned flags = ndr->flags;
37 unsigned c_len_term = 0;
38
39 if (!(ndr_flags & NDR_SCALARS)) {
40 return NDR_ERR_SUCCESS;
41 }
42
43 if (NDR_BE(ndr)) {
44 chset = CH_UTF16BE;
45 }
46
47 if (flags & LIBNDR_FLAG_STR_ASCII) {
48 chset = CH_DOS;
49 byte_mul = 1;
50 flags &= ~LIBNDR_FLAG_STR_ASCII;
51 }
52
53 if (flags & LIBNDR_FLAG_STR_UTF8) {
54 chset = CH_UTF8;
55 byte_mul = 1;
56 flags &= ~LIBNDR_FLAG_STR_UTF8;
57 }
58
59 flags &= ~LIBNDR_FLAG_STR_CONFORMANT;
60 if (flags & LIBNDR_FLAG_STR_CHARLEN) {
61 c_len_term = 1;
62 flags &= ~LIBNDR_FLAG_STR_CHARLEN;
63 }
64
65 switch (flags & LIBNDR_STRING_FLAGS) {
66 case LIBNDR_FLAG_STR_LEN4|LIBNDR_FLAG_STR_SIZE4:
67 case LIBNDR_FLAG_STR_LEN4|LIBNDR_FLAG_STR_SIZE4|LIBNDR_FLAG_STR_NOTERM:
68 NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &len1));
69 NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &ofs));
70 if (ofs != 0) {
71 return ndr_pull_error(ndr, NDR_ERR_STRING, "non-zero array offset with string flags 0x%x\n",
72 ndr->flags & LIBNDR_STRING_FLAGS);
73 }
74 NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &len2));
75 if (len2 > len1) {
76 return ndr_pull_error(ndr, NDR_ERR_STRING,
77 "Bad string lengths len1=%u ofs=%u len2=%u\n",
78 len1, ofs, len2);
79 }
80 NDR_PULL_NEED_BYTES(ndr, (len2 + c_len_term)*byte_mul);
81 if (len2 == 0) {
82 as = talloc_strdup(ndr->current_mem_ctx, "");
83 } else {
84 if (!convert_string_talloc(ndr->current_mem_ctx, chset,
85 CH_UNIX,
86 ndr->data+ndr->offset,
87 (len2 + c_len_term)*byte_mul,
88 (void **)(void *)&as,
89 &converted_size, false))
90 {
91 return ndr_pull_error(ndr, NDR_ERR_CHARCNV,
92 "Bad char conversion");
93 }
94 }
95 NDR_CHECK(ndr_pull_advance(ndr, (len2 + c_len_term)*byte_mul));
96
97 if (len1 != len2) {
98 DEBUG(6,("len1[%u] != len2[%u] '%s'\n", len1, len2, as));
99 }
100
101 /* this is a way of detecting if a string is sent with the wrong
102 termination */
103 if (ndr->flags & LIBNDR_FLAG_STR_NOTERM) {
104 if (strlen(as) < (len2 + c_len_term)) {
105 DEBUG(6,("short string '%s'\n", as));
106 }
107 } else {
108 if (strlen(as) == (len2 + c_len_term)) {
109 DEBUG(6,("long string '%s'\n", as));
110 }
111 }
112 *s = as;
113 break;
114
115 case LIBNDR_FLAG_STR_SIZE4:
116 case LIBNDR_FLAG_STR_SIZE4|LIBNDR_FLAG_STR_NOTERM:
117 NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &len1));
118 NDR_PULL_NEED_BYTES(ndr, (len1 + c_len_term)*byte_mul);
119 if (len1 == 0) {
120 as = talloc_strdup(ndr->current_mem_ctx, "");
121 } else {
122 if (!convert_string_talloc(ndr->current_mem_ctx, chset,
123 CH_UNIX,
124 ndr->data+ndr->offset,
125 (len1 + c_len_term)*byte_mul,
126 (void **)(void *)&as,
127 &converted_size, false))
128 {
129 return ndr_pull_error(ndr, NDR_ERR_CHARCNV,
130 "Bad char conversion");
131 }
132 }
133 NDR_CHECK(ndr_pull_advance(ndr, (len1 + c_len_term)*byte_mul));
134
135 /* this is a way of detecting if a string is sent with the wrong
136 termination */
137 if (ndr->flags & LIBNDR_FLAG_STR_NOTERM) {
138 if (strlen(as) < (len1 + c_len_term)) {
139 DEBUG(6,("short string '%s'\n", as));
140 }
141 } else {
142 if (strlen(as) == (len1 + c_len_term)) {
143 DEBUG(6,("long string '%s'\n", as));
144 }
145 }
146 *s = as;
147 break;
148
149 case LIBNDR_FLAG_STR_LEN4:
150 case LIBNDR_FLAG_STR_LEN4|LIBNDR_FLAG_STR_NOTERM:
151 NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &ofs));
152 if (ofs != 0) {
153 return ndr_pull_error(ndr, NDR_ERR_STRING, "non-zero array offset with string flags 0x%x\n",
154 ndr->flags & LIBNDR_STRING_FLAGS);
155 }
156 NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &len1));
157 NDR_PULL_NEED_BYTES(ndr, (len1 + c_len_term)*byte_mul);
158 if (len1 == 0) {
159 as = talloc_strdup(ndr->current_mem_ctx, "");
160 } else {
161 if (!convert_string_talloc(ndr->current_mem_ctx, chset,
162 CH_UNIX,
163 ndr->data+ndr->offset,
164 (len1 + c_len_term)*byte_mul,
165 (void **)(void *)&as,
166 &converted_size, false))
167 {
168 return ndr_pull_error(ndr, NDR_ERR_CHARCNV,
169 "Bad char conversion");
170 }
171 }
172 NDR_CHECK(ndr_pull_advance(ndr, (len1 + c_len_term)*byte_mul));
173
174 /* this is a way of detecting if a string is sent with the wrong
175 termination */
176 if (ndr->flags & LIBNDR_FLAG_STR_NOTERM) {
177 if (strlen(as) < (len1 + c_len_term)) {
178 DEBUG(6,("short string '%s'\n", as));
179 }
180 } else {
181 if (strlen(as) == (len1 + c_len_term)) {
182 DEBUG(6,("long string '%s'\n", as));
183 }
184 }
185 *s = as;
186 break;
187
188
189 case LIBNDR_FLAG_STR_SIZE2:
190 case LIBNDR_FLAG_STR_SIZE2|LIBNDR_FLAG_STR_NOTERM:
191 NDR_CHECK(ndr_pull_uint16(ndr, NDR_SCALARS, &len3));
192 NDR_PULL_NEED_BYTES(ndr, (len3 + c_len_term)*byte_mul);
193 if (len3 == 0) {
194 as = talloc_strdup(ndr->current_mem_ctx, "");
195 } else {
196 if (!convert_string_talloc(ndr->current_mem_ctx, chset,
197 CH_UNIX,
198 ndr->data+ndr->offset,
199 (len3 + c_len_term)*byte_mul,
200 (void **)(void *)&as,
201 &converted_size, false))
202 {
203 return ndr_pull_error(ndr, NDR_ERR_CHARCNV,
204 "Bad char conversion");
205 }
206 }
207 NDR_CHECK(ndr_pull_advance(ndr, (len3 + c_len_term)*byte_mul));
208
209 /* this is a way of detecting if a string is sent with the wrong
210 termination */
211 if (ndr->flags & LIBNDR_FLAG_STR_NOTERM) {
212 if (strlen(as) < (len3 + c_len_term)) {
213 DEBUG(6,("short string '%s'\n", as));
214 }
215 } else {
216 if (strlen(as) == (len3 + c_len_term)) {
217 DEBUG(6,("long string '%s'\n", as));
218 }
219 }
220 *s = as;
221 break;
222
223 case LIBNDR_FLAG_STR_SIZE2|LIBNDR_FLAG_STR_NOTERM|LIBNDR_FLAG_STR_BYTESIZE:
224 NDR_CHECK(ndr_pull_uint16(ndr, NDR_SCALARS, &len3));
225 NDR_PULL_NEED_BYTES(ndr, len3);
226 if (len3 == 0) {
227 as = talloc_strdup(ndr->current_mem_ctx, "");
228 } else {
229 if (!convert_string_talloc(ndr->current_mem_ctx, chset,
230 CH_UNIX,
231 ndr->data+ndr->offset, len3,
232 (void **)(void *)&as,
233 &converted_size, false))
234 {
235 return ndr_pull_error(ndr, NDR_ERR_CHARCNV,
236 "Bad char conversion");
237 }
238 }
239 NDR_CHECK(ndr_pull_advance(ndr, len3));
240 *s = as;
241 break;
242
243 case LIBNDR_FLAG_STR_NULLTERM:
244 if (byte_mul == 1) {
245 len1 = ascii_len_n((const char *)(ndr->data+ndr->offset), ndr->data_size - ndr->offset);
246 } else {
247 len1 = utf16_len_n(ndr->data+ndr->offset, ndr->data_size - ndr->offset);
248 }
249 if (!convert_string_talloc(ndr->current_mem_ctx, chset, CH_UNIX,
250 ndr->data+ndr->offset, len1,
251 (void **)(void *)&as,
252 &converted_size, false))
253 {
254 return ndr_pull_error(ndr, NDR_ERR_CHARCNV,
255 "Bad char conversion");
256 }
257 NDR_CHECK(ndr_pull_advance(ndr, len1));
258 *s = as;
259 break;
260
261 case LIBNDR_FLAG_STR_FIXLEN15:
262 case LIBNDR_FLAG_STR_FIXLEN32:
263 len1 = (flags & LIBNDR_FLAG_STR_FIXLEN32)?32:15;
264 NDR_PULL_NEED_BYTES(ndr, len1*byte_mul);
265 if (!convert_string_talloc(ndr->current_mem_ctx, chset, CH_UNIX,
266 ndr->data+ndr->offset, len1*byte_mul,
267 (void **)(void *)&as,
268 &converted_size, false))
269 {
270 return ndr_pull_error(ndr, NDR_ERR_CHARCNV,
271 "Bad char conversion");
272 }
273 NDR_CHECK(ndr_pull_advance(ndr, len1*byte_mul));
274 *s = as;
275 break;
276
277 case LIBNDR_FLAG_STR_NOTERM:
278 if (!(ndr->flags & LIBNDR_FLAG_REMAINING)) {
279 return ndr_pull_error(ndr, NDR_ERR_STRING, "Bad string flags 0x%x (missing NDR_REMAINING)\n",
280 ndr->flags & LIBNDR_STRING_FLAGS);
281 }
282
283 len1 = ndr->data_size - ndr->offset;
284
285 NDR_PULL_NEED_BYTES(ndr, len1);
286 if (len1 == 0) {
287 as = talloc_strdup(ndr->current_mem_ctx, "");
288 } else {
289 if (!convert_string_talloc(ndr->current_mem_ctx, chset,
290 CH_UNIX,
291 ndr->data+ndr->offset, len1,
292 (void **)(void *)&as,
293 &converted_size, false))
294 {
295 return ndr_pull_error(ndr, NDR_ERR_CHARCNV,
296 "Bad char conversion");
297 }
298 }
299 NDR_CHECK(ndr_pull_advance(ndr, len1));
300
301 *s = as;
302 break;
303
304 default:
305 return ndr_pull_error(ndr, NDR_ERR_STRING, "Bad string flags 0x%x\n",
306 ndr->flags & LIBNDR_STRING_FLAGS);
307 }
308
309 return NDR_ERR_SUCCESS;
310}
311
312
313/**
314 push a general string onto the wire
315*/
316_PUBLIC_ enum ndr_err_code ndr_push_string(struct ndr_push *ndr, int ndr_flags, const char *s)
317{
318 ssize_t s_len, c_len;
319 size_t d_len;
320 int chset = CH_UTF16;
321 unsigned flags = ndr->flags;
322 unsigned byte_mul = 2;
323 uint8_t *dest = NULL;
324
325 if (!(ndr_flags & NDR_SCALARS)) {
326 return NDR_ERR_SUCCESS;
327 }
328
329 if (NDR_BE(ndr)) {
330 chset = CH_UTF16BE;
331 }
332
333 s_len = s?strlen(s):0;
334
335 if (flags & LIBNDR_FLAG_STR_ASCII) {
336 chset = CH_DOS;
337 byte_mul = 1;
338 flags &= ~LIBNDR_FLAG_STR_ASCII;
339 }
340
341 if (flags & LIBNDR_FLAG_STR_UTF8) {
342 chset = CH_UTF8;
343 byte_mul = 1;
344 flags &= ~LIBNDR_FLAG_STR_UTF8;
345 }
346
347 flags &= ~LIBNDR_FLAG_STR_CONFORMANT;
348
349 if (!(flags &
350 (LIBNDR_FLAG_STR_NOTERM |
351 LIBNDR_FLAG_STR_FIXLEN15 |
352 LIBNDR_FLAG_STR_FIXLEN32))) {
353 s_len++;
354 }
355 if (!convert_string_talloc(ndr, CH_UNIX, chset, s, s_len,
356 (void **)(void *)&dest, &d_len, false))
357 {
358 return ndr_push_error(ndr, NDR_ERR_CHARCNV,
359 "Bad char conversion");
360 }
361
362 if (flags & LIBNDR_FLAG_STR_BYTESIZE) {
363 c_len = d_len;
364 flags &= ~LIBNDR_FLAG_STR_BYTESIZE;
365 } else if (flags & LIBNDR_FLAG_STR_CHARLEN) {
366 c_len = (d_len / byte_mul)-1;
367 flags &= ~LIBNDR_FLAG_STR_CHARLEN;
368 } else {
369 c_len = d_len / byte_mul;
370 }
371
372 switch ((flags & LIBNDR_STRING_FLAGS) & ~LIBNDR_FLAG_STR_NOTERM) {
373 case LIBNDR_FLAG_STR_LEN4|LIBNDR_FLAG_STR_SIZE4:
374 NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, c_len));
375 NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, 0));
376 NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, c_len));
377 NDR_CHECK(ndr_push_bytes(ndr, dest, d_len));
378 break;
379
380 case LIBNDR_FLAG_STR_LEN4:
381 NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, 0));
382 NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, c_len));
383 NDR_CHECK(ndr_push_bytes(ndr, dest, d_len));
384 break;
385
386 case LIBNDR_FLAG_STR_SIZE4:
387 NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, c_len));
388 NDR_CHECK(ndr_push_bytes(ndr, dest, d_len));
389 break;
390
391 case LIBNDR_FLAG_STR_SIZE2:
392 NDR_CHECK(ndr_push_uint16(ndr, NDR_SCALARS, c_len));
393 NDR_CHECK(ndr_push_bytes(ndr, dest, d_len));
394 break;
395
396 case LIBNDR_FLAG_STR_NULLTERM:
397 NDR_CHECK(ndr_push_bytes(ndr, dest, d_len));
398 break;
399
400 case LIBNDR_FLAG_STR_FIXLEN15:
401 case LIBNDR_FLAG_STR_FIXLEN32: {
402 ssize_t fix_len = (flags & LIBNDR_FLAG_STR_FIXLEN32)?32:15;
403 uint32_t pad_len = fix_len - d_len;
404 if (d_len > fix_len) {
405 return ndr_push_error(ndr, NDR_ERR_CHARCNV,
406 "Bad char conversion");
407 }
408 NDR_CHECK(ndr_push_bytes(ndr, dest, d_len));
409 if (pad_len != 0) {
410 NDR_CHECK(ndr_push_zero(ndr, pad_len));
411 }
412 break;
413 }
414
415 default:
416 if (ndr->flags & LIBNDR_FLAG_REMAINING) {
417 NDR_CHECK(ndr_push_bytes(ndr, dest, d_len));
418 break;
419 }
420
421 return ndr_push_error(ndr, NDR_ERR_STRING, "Bad string flags 0x%x\n",
422 ndr->flags & LIBNDR_STRING_FLAGS);
423 }
424
425 talloc_free(dest);
426
427 return NDR_ERR_SUCCESS;
428}
429
430/**
431 push a general string onto the wire
432*/
433_PUBLIC_ size_t ndr_string_array_size(struct ndr_push *ndr, const char *s)
434{
435 size_t c_len;
436 unsigned flags = ndr->flags;
437 unsigned byte_mul = 2;
438 unsigned c_len_term = 1;
439
440 if (flags & LIBNDR_FLAG_STR_FIXLEN32) {
441 return 32;
442 }
443 if (flags & LIBNDR_FLAG_STR_FIXLEN15) {
444 return 15;
445 }
446
447 c_len = s?strlen_m(s):0;
448
449 if (flags & (LIBNDR_FLAG_STR_ASCII|LIBNDR_FLAG_STR_UTF8)) {
450 byte_mul = 1;
451 }
452
453 if (flags & LIBNDR_FLAG_STR_NOTERM) {
454 c_len_term = 0;
455 }
456
457 c_len = c_len + c_len_term;
458
459 if (flags & LIBNDR_FLAG_STR_BYTESIZE) {
460 c_len = c_len * byte_mul;
461 }
462
463 return c_len;
464}
465
466_PUBLIC_ void ndr_print_string(struct ndr_print *ndr, const char *name, const char *s)
467{
468 if (s) {
469 ndr->print(ndr, "%-25s: '%s'", name, s);
470 } else {
471 ndr->print(ndr, "%-25s: NULL", name);
472 }
473}
474
475_PUBLIC_ uint32_t ndr_size_string(int ret, const char * const* string, int flags)
476{
477 /* FIXME: Is this correct for all strings ? */
478 if(!(*string)) return ret;
479 return ret+strlen(*string)+1;
480}
481
482/**
483 pull a general string array from the wire
484*/
485_PUBLIC_ enum ndr_err_code ndr_pull_string_array(struct ndr_pull *ndr, int ndr_flags, const char ***_a)
486{
487 const char **a = *_a;
488 uint32_t count;
489 unsigned flags = ndr->flags;
490 unsigned saved_flags = ndr->flags;
491
492 if (!(ndr_flags & NDR_SCALARS)) {
493 return NDR_ERR_SUCCESS;
494 }
495
496 switch (flags & LIBNDR_STRING_FLAGS) {
497 case LIBNDR_FLAG_STR_NULLTERM:
498 /*
499 * here the strings are null terminated
500 * but also the array is null terminated
501 */
502 for (count = 0;; count++) {
503 TALLOC_CTX *tmp_ctx;
504 const char *s = NULL;
505 a = talloc_realloc(ndr->current_mem_ctx, a, const char *, count + 2);
506 NDR_ERR_HAVE_NO_MEMORY(a);
507 a[count] = NULL;
508 a[count+1] = NULL;
509
510 tmp_ctx = ndr->current_mem_ctx;
511 ndr->current_mem_ctx = a;
512 NDR_CHECK(ndr_pull_string(ndr, ndr_flags, &s));
513 ndr->current_mem_ctx = tmp_ctx;
514 if (strcmp("", s)==0) {
515 a[count] = NULL;
516 break;
517 } else {
518 a[count] = s;
519 }
520 }
521
522 *_a =a;
523 break;
524
525 case LIBNDR_FLAG_STR_NOTERM:
526 if (!(ndr->flags & LIBNDR_FLAG_REMAINING)) {
527 return ndr_pull_error(ndr, NDR_ERR_STRING, "Bad string flags 0x%x (missing NDR_REMAINING)\n",
528 ndr->flags & LIBNDR_STRING_FLAGS);
529 }
530 /*
531 * here the strings are not null terminated
532 * but serarated by a null terminator
533 *
534 * which means the same as:
535 * very string is null terminated exept the last
536 * string is terminated by the end of the buffer
537 *
538 * as LIBNDR_FLAG_STR_NULLTERM also end at the end
539 * of the buffer, we can pull each string with this flag
540 */
541 ndr->flags &= ~(LIBNDR_FLAG_STR_NOTERM|LIBNDR_FLAG_REMAINING);
542 ndr->flags |= LIBNDR_FLAG_STR_NULLTERM;
543
544 for (count = 0; ((ndr->data_size - ndr->offset) > 0); count++) {
545 TALLOC_CTX *tmp_ctx;
546 const char *s = NULL;
547 a = talloc_realloc(ndr->current_mem_ctx, a, const char *, count + 2);
548 NDR_ERR_HAVE_NO_MEMORY(a);
549 a[count] = NULL;
550 a[count+1] = NULL;
551
552 tmp_ctx = ndr->current_mem_ctx;
553 ndr->current_mem_ctx = a;
554 NDR_CHECK(ndr_pull_string(ndr, ndr_flags, &s));
555 ndr->current_mem_ctx = tmp_ctx;
556 a[count] = s;
557 }
558
559 *_a =a;
560 break;
561
562 default:
563 return ndr_pull_error(ndr, NDR_ERR_STRING, "Bad string flags 0x%x\n",
564 ndr->flags & LIBNDR_STRING_FLAGS);
565 }
566
567 ndr->flags = saved_flags;
568 return NDR_ERR_SUCCESS;
569}
570
571/**
572 push a general string array onto the wire
573*/
574_PUBLIC_ enum ndr_err_code ndr_push_string_array(struct ndr_push *ndr, int ndr_flags, const char **a)
575{
576 uint32_t count;
577 unsigned flags = ndr->flags;
578 unsigned saved_flags = ndr->flags;
579
580 if (!(ndr_flags & NDR_SCALARS)) {
581 return NDR_ERR_SUCCESS;
582 }
583
584 switch (flags & LIBNDR_STRING_FLAGS) {
585 case LIBNDR_FLAG_STR_NULLTERM:
586 for (count = 0; a && a[count]; count++) {
587 NDR_CHECK(ndr_push_string(ndr, ndr_flags, a[count]));
588 }
589
590 NDR_CHECK(ndr_push_string(ndr, ndr_flags, ""));
591 break;
592
593 case LIBNDR_FLAG_STR_NOTERM:
594 if (!(ndr->flags & LIBNDR_FLAG_REMAINING)) {
595 return ndr_push_error(ndr, NDR_ERR_STRING, "Bad string flags 0x%x (missing NDR_REMAINING)\n",
596 ndr->flags & LIBNDR_STRING_FLAGS);
597 }
598
599 for (count = 0; a && a[count]; count++) {
600 if (count > 0) {
601 ndr->flags &= ~(LIBNDR_FLAG_STR_NOTERM|LIBNDR_FLAG_REMAINING);
602 ndr->flags |= LIBNDR_FLAG_STR_NULLTERM;
603 NDR_CHECK(ndr_push_string(ndr, ndr_flags, ""));
604 ndr->flags = saved_flags;
605 }
606 NDR_CHECK(ndr_push_string(ndr, ndr_flags, a[count]));
607 }
608
609 break;
610
611 default:
612 return ndr_push_error(ndr, NDR_ERR_STRING, "Bad string flags 0x%x\n",
613 ndr->flags & LIBNDR_STRING_FLAGS);
614 }
615
616 ndr->flags = saved_flags;
617 return NDR_ERR_SUCCESS;
618}
619
620_PUBLIC_ void ndr_print_string_array(struct ndr_print *ndr, const char *name, const char **a)
621{
622 uint32_t count;
623 uint32_t i;
624
625 for (count = 0; a && a[count]; count++) {}
626
627 ndr->print(ndr, "%s: ARRAY(%d)", name, count);
628 ndr->depth++;
629 for (i=0;i<count;i++) {
630 char *idx=NULL;
631 if (asprintf(&idx, "[%d]", i) != -1) {
632 ndr_print_string(ndr, idx, a[i]);
633 free(idx);
634 }
635 }
636 ndr->depth--;
637}
638
639/**
640 * Return number of elements in a string including the last (zeroed) element
641 */
642_PUBLIC_ uint32_t ndr_string_length(const void *_var, uint32_t element_size)
643{
644 uint32_t i;
645 uint8_t zero[4] = {0,0,0,0};
646 const char *var = (const char *)_var;
647
648 for (i = 0; memcmp(var+i*element_size,zero,element_size) != 0; i++);
649
650 return i+1;
651}
652
653_PUBLIC_ enum ndr_err_code ndr_check_string_terminator(struct ndr_pull *ndr, uint32_t count, uint32_t element_size)
654{
655 uint32_t i;
656 struct ndr_pull_save save_offset;
657
658 ndr_pull_save(ndr, &save_offset);
659 ndr_pull_advance(ndr, (count - 1) * element_size);
660 NDR_PULL_NEED_BYTES(ndr, element_size);
661
662 for (i = 0; i < element_size; i++) {
663 if (ndr->data[ndr->offset+i] != 0) {
664 ndr_pull_restore(ndr, &save_offset);
665
666 return ndr_pull_error(ndr, NDR_ERR_ARRAY_SIZE, "String terminator not present or outside string boundaries");
667 }
668 }
669
670 ndr_pull_restore(ndr, &save_offset);
671
672 return NDR_ERR_SUCCESS;
673}
674
675_PUBLIC_ enum ndr_err_code ndr_pull_charset(struct ndr_pull *ndr, int ndr_flags, const char **var, uint32_t length, uint8_t byte_mul, charset_t chset)
676{
677 size_t converted_size;
678
679 if (length == 0) {
680 *var = talloc_strdup(ndr->current_mem_ctx, "");
681 return NDR_ERR_SUCCESS;
682 }
683
684 if (NDR_BE(ndr) && chset == CH_UTF16) {
685 chset = CH_UTF16BE;
686 }
687
688 NDR_PULL_NEED_BYTES(ndr, length*byte_mul);
689
690 if (!convert_string_talloc(ndr->current_mem_ctx, chset, CH_UNIX,
691 ndr->data+ndr->offset, length*byte_mul,
692 discard_const_p(void *, var),
693 &converted_size, false))
694 {
695 return ndr_pull_error(ndr, NDR_ERR_CHARCNV,
696 "Bad char conversion");
697 }
698 NDR_CHECK(ndr_pull_advance(ndr, length*byte_mul));
699
700 return NDR_ERR_SUCCESS;
701}
702
703_PUBLIC_ enum ndr_err_code ndr_push_charset(struct ndr_push *ndr, int ndr_flags, const char *var, uint32_t length, uint8_t byte_mul, charset_t chset)
704{
705 ssize_t ret, required;
706
707 if (NDR_BE(ndr) && chset == CH_UTF16) {
708 chset = CH_UTF16BE;
709 }
710
711 required = byte_mul * length;
712
713 NDR_PUSH_NEED_BYTES(ndr, required);
714 ret = convert_string(CH_UNIX, chset,
715 var, strlen(var),
716 ndr->data+ndr->offset, required, false);
717 if (ret == -1) {
718 return ndr_push_error(ndr, NDR_ERR_CHARCNV,
719 "Bad char conversion");
720 }
721
722 /* Make sure the remaining part of the string is filled with zeroes */
723 if (ret < required) {
724 memset(ndr->data+ndr->offset+ret, 0, required-ret);
725 }
726
727 ndr->offset += required;
728
729 return NDR_ERR_SUCCESS;
730}
731
732/* Return number of elements in a string in the specified charset */
733_PUBLIC_ uint32_t ndr_charset_length(const void *var, charset_t chset)
734{
735 /* FIXME: Treat special chars special here, taking chset into account */
736 /* Also include 0 byte */
737 return strlen((const char *)var)+1;
738}
Note: See TracBrowser for help on using the repository browser.