source/librpc/ndr/ndr_basic.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    routines for marshalling/unmarshalling basic 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 "system/network.h"
  24 #include "librpc/ndr/libndr.h"
  25
  26 #define NDR_SVAL(ndr, ofs) (NDR_BE(ndr)?RSVAL(ndr->data,ofs):SVAL(ndr->data,ofs))
  27 #define NDR_IVAL(ndr, ofs) (NDR_BE(ndr)?RIVAL(ndr->data,ofs):IVAL(ndr->data,ofs))
  28 #define NDR_IVALS(ndr, ofs) (NDR_BE(ndr)?RIVALS(ndr->data,ofs):IVALS(ndr->data,ofs))
  29 #define NDR_SSVAL(ndr, ofs, v) do { if (NDR_BE(ndr))  { RSSVAL(ndr->data,ofs,v); } else SSVAL(ndr->data,ofs,v); } while (0)
  30 #define NDR_SIVAL(ndr, ofs, v) do { if (NDR_BE(ndr))  { RSIVAL(ndr->data,ofs,v); } else SIVAL(ndr->data,ofs,v); } while (0)
  31 #define NDR_SIVALS(ndr, ofs, v) do { if (NDR_BE(ndr))  { RSIVALS(ndr->data,ofs,v); } else SIVALS(ndr->data,ofs,v); } while (0)
  32
  33
  34 /*
  35   check for data leaks from the server by looking for non-zero pad bytes
  36   these could also indicate that real structure elements have been
  37   mistaken for padding in the IDL
  38 */
  39 _PUBLIC_ void ndr_check_padding(struct ndr_pull *ndr, size_t n)
  40 {
  41         size_t ofs2 = (ndr->offset + (n-1)) & ~(n-1);
  42         int i;
  43         for (i=ndr->offset;i<ofs2;i++) {
  44                 if (ndr->data[i] != 0) {
  45                         break;
  46                 }
  47         }
  48         if (i<ofs2) {
  49                 DEBUG(0,("WARNING: Non-zero padding to %d: ", (int)n));
  50                 for (i=ndr->offset;i<ofs2;i++) {
  51                         DEBUG(0,("%02x ", ndr->data[i]));
  52                 }
  53                 DEBUG(0,("\n"));
  54         }
  55
  56 }
  57
  58 /*
  59   parse a int8_t
  60 */
  61 _PUBLIC_ enum ndr_err_code ndr_pull_int8(struct ndr_pull *ndr, int ndr_flags, int8_t *v)
  62 {
  63         NDR_PULL_NEED_BYTES(ndr, 1);
  64         *v = (int8_t)CVAL(ndr->data, ndr->offset);
  65         ndr->offset += 1;
  66         return NDR_ERR_SUCCESS;
  67 }
  68
  69 /*
  70   parse a uint8_t
  71 */
  72 _PUBLIC_ enum ndr_err_code ndr_pull_uint8(struct ndr_pull *ndr, int ndr_flags, uint8_t *v)
  73 {
  74         NDR_PULL_NEED_BYTES(ndr, 1);
  75         *v = CVAL(ndr->data, ndr->offset);
  76         ndr->offset += 1;
  77         return NDR_ERR_SUCCESS;
  78 }
  79
  80 /*
  81   parse a int16_t
  82 */
  83 _PUBLIC_ enum ndr_err_code ndr_pull_int16(struct ndr_pull *ndr, int ndr_flags, int16_t *v)
  84 {
  85         NDR_PULL_ALIGN(ndr, 2);
  86         NDR_PULL_NEED_BYTES(ndr, 2);
  87         *v = (uint16_t)NDR_SVAL(ndr, ndr->offset);
  88         ndr->offset += 2;
  89         return NDR_ERR_SUCCESS;
  90 }
  91
  92 /*
  93   parse a uint16_t
  94 */
  95 _PUBLIC_ enum ndr_err_code ndr_pull_uint16(struct ndr_pull *ndr, int ndr_flags, uint16_t *v)
  96 {
  97         NDR_PULL_ALIGN(ndr, 2);
  98         NDR_PULL_NEED_BYTES(ndr, 2);
  99         *v = NDR_SVAL(ndr, ndr->offset);
 100         ndr->offset += 2;
 101         return NDR_ERR_SUCCESS;
 102 }
 103
 104 /*
 105   parse a int32_t
 106 */
 107 _PUBLIC_ enum ndr_err_code ndr_pull_int32(struct ndr_pull *ndr, int ndr_flags, int32_t *v)
 108 {
 109         NDR_PULL_ALIGN(ndr, 4);
 110         NDR_PULL_NEED_BYTES(ndr, 4);
 111         *v = NDR_IVALS(ndr, ndr->offset);
 112         ndr->offset += 4;
 113         return NDR_ERR_SUCCESS;
 114 }
 115
 116 /*
 117   parse a uint32_t
 118 */
 119 _PUBLIC_ enum ndr_err_code ndr_pull_uint32(struct ndr_pull *ndr, int ndr_flags, uint32_t *v)
 120 {
 121         NDR_PULL_ALIGN(ndr, 4);
 122         NDR_PULL_NEED_BYTES(ndr, 4);
 123         *v = NDR_IVAL(ndr, ndr->offset);
 124         ndr->offset += 4;
 125         return NDR_ERR_SUCCESS;
 126 }
 127
 128 /*
 129   parse a pointer referent identifier
 130 */
 131 _PUBLIC_ enum ndr_err_code ndr_pull_generic_ptr(struct ndr_pull *ndr, uint32_t *v)
 132 {
 133         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, v));
 134         if (*v != 0) {
 135                 ndr->ptr_count++;
 136         }
 137         return NDR_ERR_SUCCESS;
 138 }
 139
 140 /*
 141   parse a ref pointer referent identifier
 142 */
 143 _PUBLIC_ enum ndr_err_code ndr_pull_ref_ptr(struct ndr_pull *ndr, uint32_t *v)
 144 {
 145         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, v));
 146         /* ref pointers always point to data */
 147         *v = 1;
 148         return NDR_ERR_SUCCESS;
 149 }
 150
 151 /*
 152   parse a udlong
 153 */
 154 _PUBLIC_ enum ndr_err_code ndr_pull_udlong(struct ndr_pull *ndr, int ndr_flags, uint64_t *v)
 155 {
 156         NDR_PULL_ALIGN(ndr, 4);
 157         NDR_PULL_NEED_BYTES(ndr, 8);
 158         *v = NDR_IVAL(ndr, ndr->offset);
 159         *v |= (uint64_t)(NDR_IVAL(ndr, ndr->offset+4)) << 32;
 160         ndr->offset += 8;
 161         return NDR_ERR_SUCCESS;
 162 }
 163
 164 /*
 165   parse a udlongr
 166 */
 167 _PUBLIC_ enum ndr_err_code ndr_pull_udlongr(struct ndr_pull *ndr, int ndr_flags, uint64_t *v)
 168 {
 169         NDR_PULL_ALIGN(ndr, 4);
 170         NDR_PULL_NEED_BYTES(ndr, 8);
 171         *v = ((uint64_t)NDR_IVAL(ndr, ndr->offset)) << 32;
 172         *v |= NDR_IVAL(ndr, ndr->offset+4);
 173         ndr->offset += 8;
 174         return NDR_ERR_SUCCESS;
 175 }
 176
 177 /*
 178   parse a dlong
 179 */
 180 _PUBLIC_ enum ndr_err_code ndr_pull_dlong(struct ndr_pull *ndr, int ndr_flags, int64_t *v)
 181 {
 182         return ndr_pull_udlong(ndr, ndr_flags, (uint64_t *)v);
 183 }
 184
 185 /*
 186   parse a hyper
 187 */
 188 _PUBLIC_ enum ndr_err_code ndr_pull_hyper(struct ndr_pull *ndr, int ndr_flags, uint64_t *v)
 189 {
 190         NDR_PULL_ALIGN(ndr, 8);
 191         return ndr_pull_udlong(ndr, ndr_flags, v);
 192 }
 193
 194 /*
 195   parse a pointer
 196 */
 197 _PUBLIC_ enum ndr_err_code ndr_pull_pointer(struct ndr_pull *ndr, int ndr_flags, void* *v)
 198 {
 199         uintptr_t h;
 200         NDR_PULL_ALIGN(ndr, sizeof(h));
 201         NDR_PULL_NEED_BYTES(ndr, sizeof(h));
 202         memcpy(&h, ndr->data+ndr->offset, sizeof(h));
 203         ndr->offset += sizeof(h);
 204         *v = (void *)h;
 205         return NDR_ERR_SUCCESS;
 206 }
 207
 208 /*
 209   pull a NTSTATUS
 210 */
 211 _PUBLIC_ enum ndr_err_code ndr_pull_NTSTATUS(struct ndr_pull *ndr, int ndr_flags, NTSTATUS *status)
 212 {
 213         uint32_t v;
 214         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &v));
 215         *status = NT_STATUS(v);
 216         return NDR_ERR_SUCCESS;
 217 }
 218
 219 /*
 220   push a NTSTATUS
 221 */
 222 _PUBLIC_ enum ndr_err_code ndr_push_NTSTATUS(struct ndr_push *ndr, int ndr_flags, NTSTATUS status)
 223 {
 224         return ndr_push_uint32(ndr, ndr_flags, NT_STATUS_V(status));
 225 }
 226
 227 _PUBLIC_ void ndr_print_NTSTATUS(struct ndr_print *ndr, const char *name, NTSTATUS r)
 228 {
 229         ndr->print(ndr, "%-25s: %s", name, nt_errstr(r));
 230 }
 231
 232 /*
 233   pull a WERROR
 234 */
 235 _PUBLIC_ enum ndr_err_code ndr_pull_WERROR(struct ndr_pull *ndr, int ndr_flags, WERROR *status)
 236 {
 237         uint32_t v;
 238         NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &v));
 239         *status = W_ERROR(v);
 240         return NDR_ERR_SUCCESS;
 241 }
 242
 243 /*
 244   push a WERROR
 245 */
 246 _PUBLIC_ enum ndr_err_code ndr_push_WERROR(struct ndr_push *ndr, int ndr_flags, WERROR status)
 247 {
 248         return ndr_push_uint32(ndr, NDR_SCALARS, W_ERROR_V(status));
 249 }
 250
 251 _PUBLIC_ void ndr_print_WERROR(struct ndr_print *ndr, const char *name, WERROR r)
 252 {
 253         ndr->print(ndr, "%-25s: %s", name, win_errstr(r));
 254 }
 255
 256 /*
 257   parse a set of bytes
 258 */
 259 _PUBLIC_ enum ndr_err_code ndr_pull_bytes(struct ndr_pull *ndr, uint8_t *data, uint32_t n)
 260 {
 261         NDR_PULL_NEED_BYTES(ndr, n);
 262         memcpy(data, ndr->data + ndr->offset, n);
 263         ndr->offset += n;
 264         return NDR_ERR_SUCCESS;
 265 }
 266
 267 /*
 268   pull an array of uint8
 269 */
 270 _PUBLIC_ enum ndr_err_code ndr_pull_array_uint8(struct ndr_pull *ndr, int ndr_flags, uint8_t *data, uint32_t n)
 271 {
 272         if (!(ndr_flags & NDR_SCALARS)) {
 273                 return NDR_ERR_SUCCESS;
 274         }
 275         return ndr_pull_bytes(ndr, data, n);
 276 }
 277
 278 /*
 279   push a int8_t
 280 */
 281 _PUBLIC_ enum ndr_err_code ndr_push_int8(struct ndr_push *ndr, int ndr_flags, int8_t v)
 282 {
 283         NDR_PUSH_NEED_BYTES(ndr, 1);
 284         SCVAL(ndr->data, ndr->offset, (uint8_t)v);
 285         ndr->offset += 1;
 286         return NDR_ERR_SUCCESS;
 287 }
 288
 289 /*
 290   push a uint8_t
 291 */
 292 _PUBLIC_ enum ndr_err_code ndr_push_uint8(struct ndr_push *ndr, int ndr_flags, uint8_t v)
 293 {
 294         NDR_PUSH_NEED_BYTES(ndr, 1);
 295         SCVAL(ndr->data, ndr->offset, v);
 296         ndr->offset += 1;
 297         return NDR_ERR_SUCCESS;
 298 }
 299
 300 /*
 301   push a int16_t
 302 */
 303 _PUBLIC_ enum ndr_err_code ndr_push_int16(struct ndr_push *ndr, int ndr_flags, int16_t v)
 304 {
 305         NDR_PUSH_ALIGN(ndr, 2);
 306         NDR_PUSH_NEED_BYTES(ndr, 2);
 307         NDR_SSVAL(ndr, ndr->offset, (uint16_t)v);
 308         ndr->offset += 2;
 309         return NDR_ERR_SUCCESS;
 310 }
 311
 312 /*
 313   push a uint16_t
 314 */
 315 _PUBLIC_ enum ndr_err_code ndr_push_uint16(struct ndr_push *ndr, int ndr_flags, uint16_t v)
 316 {
 317         NDR_PUSH_ALIGN(ndr, 2);
 318         NDR_PUSH_NEED_BYTES(ndr, 2);
 319         NDR_SSVAL(ndr, ndr->offset, v);
 320         ndr->offset += 2;
 321         return NDR_ERR_SUCCESS;
 322 }
 323
 324 /*
 325   push a int32_t
 326 */
 327 _PUBLIC_ enum ndr_err_code ndr_push_int32(struct ndr_push *ndr, int ndr_flags, int32_t v)
 328 {
 329         NDR_PUSH_ALIGN(ndr, 4);
 330         NDR_PUSH_NEED_BYTES(ndr, 4);
 331         NDR_SIVALS(ndr, ndr->offset, v);
 332         ndr->offset += 4;
 333         return NDR_ERR_SUCCESS;
 334 }
 335
 336 /*
 337   push a uint32_t
 338 */
 339 _PUBLIC_ enum ndr_err_code ndr_push_uint32(struct ndr_push *ndr, int ndr_flags, uint32_t v)
 340 {
 341         NDR_PUSH_ALIGN(ndr, 4);
 342         NDR_PUSH_NEED_BYTES(ndr, 4);
 343         NDR_SIVAL(ndr, ndr->offset, v);
 344         ndr->offset += 4;
 345         return NDR_ERR_SUCCESS;
 346 }
 347
 348 /*
 349   push a udlong
 350 */
 351 _PUBLIC_ enum ndr_err_code ndr_push_udlong(struct ndr_push *ndr, int ndr_flags, uint64_t v)
 352 {
 353         NDR_PUSH_ALIGN(ndr, 4);
 354         NDR_PUSH_NEED_BYTES(ndr, 8);
 355         NDR_SIVAL(ndr, ndr->offset, (v & 0xFFFFFFFF));
 356         NDR_SIVAL(ndr, ndr->offset+4, (v>>32));
 357         ndr->offset += 8;
 358         return NDR_ERR_SUCCESS;
 359 }
 360
 361 /*
 362   push a udlongr
 363 */
 364 _PUBLIC_ enum ndr_err_code ndr_push_udlongr(struct ndr_push *ndr, int ndr_flags, uint64_t v)
 365 {
 366         NDR_PUSH_ALIGN(ndr, 4);
 367         NDR_PUSH_NEED_BYTES(ndr, 8);
 368         NDR_SIVAL(ndr, ndr->offset, (v>>32));
 369         NDR_SIVAL(ndr, ndr->offset+4, (v & 0xFFFFFFFF));
 370         ndr->offset += 8;
 371         return NDR_ERR_SUCCESS;
 372 }
 373
 374 /*
 375   push a dlong
 376 */
 377 _PUBLIC_ enum ndr_err_code ndr_push_dlong(struct ndr_push *ndr, int ndr_flags, int64_t v)
 378 {
 379         return ndr_push_udlong(ndr, NDR_SCALARS, (uint64_t)v);
 380 }
 381
 382 /*
 383   push a hyper
 384 */
 385 _PUBLIC_ enum ndr_err_code ndr_push_hyper(struct ndr_push *ndr, int ndr_flags, uint64_t v)
 386 {
 387         NDR_PUSH_ALIGN(ndr, 8);
 388         return ndr_push_udlong(ndr, NDR_SCALARS, v);
 389 }
 390
 391 /*
 392   push a pointer
 393 */
 394 _PUBLIC_ enum ndr_err_code ndr_push_pointer(struct ndr_push *ndr, int ndr_flags, void* v)
 395 {
 396         uintptr_t h = (intptr_t)v;
 397         NDR_PUSH_ALIGN(ndr, sizeof(h));
 398         NDR_PUSH_NEED_BYTES(ndr, sizeof(h));
 399         memcpy(ndr->data+ndr->offset, &h, sizeof(h));
 400         ndr->offset += sizeof(h);
 401         return NDR_ERR_SUCCESS;
 402 }
 403
 404 _PUBLIC_ enum ndr_err_code ndr_push_align(struct ndr_push *ndr, size_t size)
 405 {
 406         NDR_PUSH_ALIGN(ndr, size);
 407         return NDR_ERR_SUCCESS;
 408 }
 409
 410 _PUBLIC_ enum ndr_err_code ndr_pull_align(struct ndr_pull *ndr, size_t size)
 411 {
 412         NDR_PULL_ALIGN(ndr, size);
 413         return NDR_ERR_SUCCESS;
 414 }
 415
 416 /*
 417   push some bytes
 418 */
 419 _PUBLIC_ enum ndr_err_code ndr_push_bytes(struct ndr_push *ndr, const uint8_t *data, uint32_t n)
 420 {
 421         NDR_PUSH_NEED_BYTES(ndr, n);
 422         memcpy(ndr->data + ndr->offset, data, n);
 423         ndr->offset += n;
 424         return NDR_ERR_SUCCESS;
 425 }
 426
 427 /*
 428   push some zero bytes
 429 */
 430 _PUBLIC_ enum ndr_err_code ndr_push_zero(struct ndr_push *ndr, uint32_t n)
 431 {
 432         NDR_PUSH_NEED_BYTES(ndr, n);
 433         memset(ndr->data + ndr->offset, 0, n);
 434         ndr->offset += n;
 435         return NDR_ERR_SUCCESS;
 436 }
 437
 438 /*
 439   push an array of uint8
 440 */
 441 _PUBLIC_ enum ndr_err_code ndr_push_array_uint8(struct ndr_push *ndr, int ndr_flags, const uint8_t *data, uint32_t n)
 442 {
 443         if (!(ndr_flags & NDR_SCALARS)) {
 444                 return NDR_ERR_SUCCESS;
 445         }
 446         return ndr_push_bytes(ndr, data, n);
 447 }
 448
 449 /*
 450   save the current position
 451  */
 452 _PUBLIC_ void ndr_push_save(struct ndr_push *ndr, struct ndr_push_save *save)
 453 {
 454         save->offset = ndr->offset;
 455 }
 456
 457 /*
 458   restore the position
 459  */
 460 _PUBLIC_ void ndr_push_restore(struct ndr_push *ndr, struct ndr_push_save *save)
 461 {
 462         ndr->offset = save->offset;
 463 }
 464
 465 /*
 466   push a unique non-zero value if a pointer is non-NULL, otherwise 0
 467 */
 468 _PUBLIC_ enum ndr_err_code ndr_push_unique_ptr(struct ndr_push *ndr, const void *p)
 469 {
 470         uint32_t ptr = 0;
 471         if (p) {
 472                 ptr = ndr->ptr_count * 4;
 473                 ptr |= 0x00020000;
 474                 ndr->ptr_count++;
 475         }
 476         return ndr_push_uint32(ndr, NDR_SCALARS, ptr);
 477 }
 478
 479 /*
 480   push a 'simple' full non-zero value if a pointer is non-NULL, otherwise 0
 481 */
 482 _PUBLIC_ enum ndr_err_code ndr_push_full_ptr(struct ndr_push *ndr, const void *p)
 483 {
 484         uint32_t ptr = 0;
 485         if (p) {
 486                 /* Check if the pointer already exists and has an id */
 487                 ptr = ndr_token_peek(&ndr->full_ptr_list, p);
 488                 if (ptr == 0) {
 489                         ndr->ptr_count++;
 490                         ptr = ndr->ptr_count;
 491                         ndr_token_store(ndr, &ndr->full_ptr_list, p, ptr);
 492                 }
 493         }
 494         return ndr_push_uint32(ndr, NDR_SCALARS, ptr);
 495 }
 496
 497 /*
 498   push always a 0, if a pointer is NULL it's a fatal error
 499 */
 500 _PUBLIC_ enum ndr_err_code ndr_push_ref_ptr(struct ndr_push *ndr)
 501 {
 502         return ndr_push_uint32(ndr, NDR_SCALARS, 0xAEF1AEF1);
 503 }
 504
 505
 506 /*
 507   push a NTTIME
 508 */
 509 _PUBLIC_ enum ndr_err_code ndr_push_NTTIME(struct ndr_push *ndr, int ndr_flags, NTTIME t)
 510 {
 511         NDR_CHECK(ndr_push_udlong(ndr, ndr_flags, t));
 512         return NDR_ERR_SUCCESS;
 513 }
 514
 515 /*
 516   pull a NTTIME
 517 */
 518 _PUBLIC_ enum ndr_err_code ndr_pull_NTTIME(struct ndr_pull *ndr, int ndr_flags, NTTIME *t)
 519 {
 520         NDR_CHECK(ndr_pull_udlong(ndr, ndr_flags, t));
 521         return NDR_ERR_SUCCESS;
 522 }
 523
 524 /*
 525   push a NTTIME
 526 */
 527 _PUBLIC_ enum ndr_err_code ndr_push_NTTIME_1sec(struct ndr_push *ndr, int ndr_flags, NTTIME t)
 528 {
 529         t /= 10000000;
 530         NDR_CHECK(ndr_push_hyper(ndr, ndr_flags, t));
 531         return NDR_ERR_SUCCESS;
 532 }
 533
 534 /*
 535   pull a NTTIME_1sec
 536 */
 537 _PUBLIC_ enum ndr_err_code ndr_pull_NTTIME_1sec(struct ndr_pull *ndr, int ndr_flags, NTTIME *t)
 538 {
 539         NDR_CHECK(ndr_pull_hyper(ndr, ndr_flags, t));
 540         (*t) *= 10000000;
 541         return NDR_ERR_SUCCESS;
 542 }
 543
 544 /*
 545   pull a NTTIME_hyper
 546 */
 547 _PUBLIC_ enum ndr_err_code ndr_pull_NTTIME_hyper(struct ndr_pull *ndr, int ndr_flags, NTTIME *t)
 548 {
 549         NDR_CHECK(ndr_pull_hyper(ndr, ndr_flags, t));
 550         return NDR_ERR_SUCCESS;
 551 }
 552
 553 /*
 554   push a NTTIME_hyper
 555 */
 556 _PUBLIC_ enum ndr_err_code ndr_push_NTTIME_hyper(struct ndr_push *ndr, int ndr_flags, NTTIME t)
 557 {
 558         NDR_CHECK(ndr_push_hyper(ndr, ndr_flags, t));
 559         return NDR_ERR_SUCCESS;
 560 }
 561
 562 /*
 563   push a time_t
 564 */
 565 _PUBLIC_ enum ndr_err_code ndr_push_time_t(struct ndr_push *ndr, int ndr_flags, time_t t)
 566 {
 567         return ndr_push_uint32(ndr, ndr_flags, t);
 568 }
 569
 570 /*
 571   pull a time_t
 572 */
 573 _PUBLIC_ enum ndr_err_code ndr_pull_time_t(struct ndr_pull *ndr, int ndr_flags, time_t *t)
 574 {
 575         uint32_t tt;
 576         NDR_CHECK(ndr_pull_uint32(ndr, ndr_flags, &tt));
 577         *t = tt;
 578         return NDR_ERR_SUCCESS;
 579 }
 580
 581
 582 /*
 583   pull a ipv4address
 584 */
 585 _PUBLIC_ enum ndr_err_code ndr_pull_ipv4address(struct ndr_pull *ndr, int ndr_flags, const char **address)
 586 {
 587         struct in_addr in;
 588         NDR_CHECK(ndr_pull_uint32(ndr, ndr_flags, &in.s_addr));
 589         in.s_addr = htonl(in.s_addr);
 590         *address = talloc_strdup(ndr->current_mem_ctx, inet_ntoa(in));
 591         NDR_ERR_HAVE_NO_MEMORY(*address);
 592         return NDR_ERR_SUCCESS;
 593 }
 594
 595 /*
 596   push a ipv4address
 597 */
 598 _PUBLIC_ enum ndr_err_code ndr_push_ipv4address(struct ndr_push *ndr, int ndr_flags, const char *address)
 599 {
 600         uint32_t addr;
 601         if (!is_ipaddress(address)) {
 602                 return ndr_push_error(ndr, NDR_ERR_IPV4ADDRESS,
 603                                       "Invalid IPv4 address: '%s'",
 604                                       address);
 605         }
 606         addr = inet_addr(address);
 607         NDR_CHECK(ndr_push_uint32(ndr, ndr_flags, htonl(addr)));
 608         return NDR_ERR_SUCCESS;
 609 }
 610
 611 /*
 612   print a ipv4address
 613 */
 614 _PUBLIC_ void ndr_print_ipv4address(struct ndr_print *ndr, const char *name,
 615                            const char *address)
 616 {
 617         ndr->print(ndr, "%-25s: %s", name, address);
 618 }
 619
 620
 621 _PUBLIC_ void ndr_print_struct(struct ndr_print *ndr, const char *name, const char *type)
 622 {
 623         ndr->print(ndr, "%s: struct %s", name, type);
 624 }
 625
 626 _PUBLIC_ void ndr_print_enum(struct ndr_print *ndr, const char *name, const char *type,
 627                     const char *val, uint32_t value)
 628 {
 629         if (ndr->flags & LIBNDR_PRINT_ARRAY_HEX) {
 630                 ndr->print(ndr, "%-25s: %s (0x%X)", name, val?val:"UNKNOWN_ENUM_VALUE", value);
 631         } else {
 632                 ndr->print(ndr, "%-25s: %s (%d)", name, val?val:"UNKNOWN_ENUM_VALUE", value);
 633         }
 634 }
 635
 636 _PUBLIC_ void ndr_print_bitmap_flag(struct ndr_print *ndr, size_t size, const char *flag_name, uint32_t flag, uint32_t value)
 637 {
 638         /* this is an attempt to support multi-bit bitmap masks */
 639         value &= flag;
 640
 641         while (!(flag & 1)) {
 642                 flag >>= 1;
 643                 value >>= 1;
 644         }
 645         if (flag == 1) {
 646                 ndr->print(ndr, "   %d: %-25s", value, flag_name);
 647         } else {
 648                 ndr->print(ndr, "0x%02x: %-25s (%d)", value, flag_name, value);
 649         }
 650 }
 651
 652 _PUBLIC_ void ndr_print_int8(struct ndr_print *ndr, const char *name, int8_t v)
 653 {
 654         ndr->print(ndr, "%-25s: %d", name, v);
 655 }
 656
 657 _PUBLIC_ void ndr_print_uint8(struct ndr_print *ndr, const char *name, uint8_t v)
 658 {
 659         ndr->print(ndr, "%-25s: 0x%02x (%u)", name, v, v);
 660 }
 661
 662 _PUBLIC_ void ndr_print_int16(struct ndr_print *ndr, const char *name, int16_t v)
 663 {
 664         ndr->print(ndr, "%-25s: %d", name, v);
 665 }
 666
 667 _PUBLIC_ void ndr_print_uint16(struct ndr_print *ndr, const char *name, uint16_t v)
 668 {
 669         ndr->print(ndr, "%-25s: 0x%04x (%u)", name, v, v);
 670 }
 671
 672 _PUBLIC_ void ndr_print_int32(struct ndr_print *ndr, const char *name, int32_t v)
 673 {
 674         ndr->print(ndr, "%-25s: %d", name, v);
 675 }
 676
 677 _PUBLIC_ void ndr_print_uint32(struct ndr_print *ndr, const char *name, uint32_t v)
 678 {
 679         ndr->print(ndr, "%-25s: 0x%08x (%u)", name, v, v);
 680 }
 681
 682 _PUBLIC_ void ndr_print_udlong(struct ndr_print *ndr, const char *name, uint64_t v)
 683 {
 684         ndr->print(ndr, "%-25s: 0x%016llx (%llu)", name, (unsigned long long)v, (unsigned long long)v);
 685 }
 686
 687 _PUBLIC_ void ndr_print_udlongr(struct ndr_print *ndr, const char *name, uint64_t v)
 688 {
 689         ndr_print_udlong(ndr, name, v);
 690 }
 691
 692 _PUBLIC_ void ndr_print_dlong(struct ndr_print *ndr, const char *name, int64_t v)
 693 {
 694         ndr->print(ndr, "%-25s: 0x%016llx (%lld)", name, (unsigned long long)v, (long long)v);
 695 }
 696
 697 _PUBLIC_ void ndr_print_hyper(struct ndr_print *ndr, const char *name, uint64_t v)
 698 {
 699         ndr_print_dlong(ndr, name, v);
 700 }
 701
 702 _PUBLIC_ void ndr_print_pointer(struct ndr_print *ndr, const char *name, void *v)
 703 {
 704         ndr->print(ndr, "%-25s: %p", name, v);
 705 }
 706
 707 _PUBLIC_ void ndr_print_ptr(struct ndr_print *ndr, const char *name, const void *p)
 708 {
 709         if (p) {
 710                 ndr->print(ndr, "%-25s: *", name);
 711         } else {
 712                 ndr->print(ndr, "%-25s: NULL", name);
 713         }
 714 }
 715
 716 _PUBLIC_ void ndr_print_NTTIME(struct ndr_print *ndr, const char *name, NTTIME t)
 717 {
 718         ndr->print(ndr, "%-25s: %s", name, nt_time_string(ndr, t));
 719 }
 720
 721 _PUBLIC_ void ndr_print_NTTIME_1sec(struct ndr_print *ndr, const char *name, NTTIME t)
 722 {
 723         /* this is a standard NTTIME here
 724          * as it's already converted in the pull/push code
 725          */
 726         ndr_print_NTTIME(ndr, name, t);
 727 }
 728
 729 _PUBLIC_ void ndr_print_NTTIME_hyper(struct ndr_print *ndr, const char *name, NTTIME t)
 730 {
 731         ndr_print_NTTIME(ndr, name, t);
 732 }
 733
 734 _PUBLIC_ void ndr_print_time_t(struct ndr_print *ndr, const char *name, time_t t)
 735 {
 736         if (t == (time_t)-1 || t == 0) {
 737                 ndr->print(ndr, "%-25s: (time_t)%d", name, (int)t);
 738         } else {
 739                 ndr->print(ndr, "%-25s: %s", name, timestring(ndr, t));
 740         }
 741 }
 742
 743 _PUBLIC_ void ndr_print_union(struct ndr_print *ndr, const char *name, int level, const char *type)
 744 {
 745         if (ndr->flags & LIBNDR_PRINT_ARRAY_HEX) {
 746                 ndr->print(ndr, "%-25s: union %s(case 0x%X)", name, type, level);
 747         } else {
 748                 ndr->print(ndr, "%-25s: union %s(case %d)", name, type, level);
 749         }
 750 }
 751
 752 _PUBLIC_ void ndr_print_bad_level(struct ndr_print *ndr, const char *name, uint16_t level)
 753 {
 754         ndr->print(ndr, "UNKNOWN LEVEL %u", level);
 755 }
 756
 757 _PUBLIC_ void ndr_print_array_uint8(struct ndr_print *ndr, const char *name,
 758                            const uint8_t *data, uint32_t count)
 759 {
 760         int i;
 761
 762         if (count <= 600 && (ndr->flags & LIBNDR_PRINT_ARRAY_HEX)) {
 763                 char s[1202];
 764                 for (i=0;i<count;i++) {
 765                         snprintf(&s[i*2], 3, "%02x", data[i]);
 766                 }
 767                 s[i*2] = 0;
 768                 ndr->print(ndr, "%-25s: %s", name, s);
 769                 return;
 770         }
 771
 772         ndr->print(ndr, "%s: ARRAY(%d)", name, count);
 773         ndr->depth++;
 774         for (i=0;i<count;i++) {
 775                 char *idx=NULL;
 776                 if (asprintf(&idx, "[%d]", i) != -1) {
 777                         ndr_print_uint8(ndr, idx, data[i]);
 778                         free(idx);
 779                 }
 780         }
 781         ndr->depth--;
 782 }
 783
 784 _PUBLIC_ void ndr_print_DATA_BLOB(struct ndr_print *ndr, const char *name, DATA_BLOB r)
 785 {
 786         ndr->print(ndr, "%-25s: DATA_BLOB length=%u", name, (unsigned)r.length);
 787         if (r.length) {
 788                 dump_data(10, r.data, r.length);
 789         }
 790 }
 791
 792
 793 /*
 794   push a DATA_BLOB onto the wire.
 795 */
 796 _PUBLIC_ enum ndr_err_code ndr_push_DATA_BLOB(struct ndr_push *ndr, int ndr_flags, DATA_BLOB blob)
 797 {
 798         if (ndr->flags & LIBNDR_ALIGN_FLAGS) {
 799                 if (ndr->flags & LIBNDR_FLAG_ALIGN2) {
 800                         blob.length = NDR_ALIGN(ndr, 2);
 801                 } else if (ndr->flags & LIBNDR_FLAG_ALIGN4) {
 802                         blob.length = NDR_ALIGN(ndr, 4);
 803                 } else if (ndr->flags & LIBNDR_FLAG_ALIGN8) {
 804                         blob.length = NDR_ALIGN(ndr, 8);
 805                 }
 806                 NDR_PUSH_ALLOC_SIZE(ndr, blob.data, blob.length);
 807                 data_blob_clear(&blob);
 808         } else if (!(ndr->flags & LIBNDR_FLAG_REMAINING)) {
 809                 NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, blob.length));
 810         }
 811         NDR_CHECK(ndr_push_bytes(ndr, blob.data, blob.length));
 812         return NDR_ERR_SUCCESS;
 813 }
 814
 815 /*
 816   pull a DATA_BLOB from the wire.
 817 */
 818 _PUBLIC_ enum ndr_err_code ndr_pull_DATA_BLOB(struct ndr_pull *ndr, int ndr_flags, DATA_BLOB *blob)
 819 {
 820         uint32_t length = 0;
 821
 822         if (ndr->flags & LIBNDR_ALIGN_FLAGS) {
 823                 if (ndr->flags & LIBNDR_FLAG_ALIGN2) {
 824                         length = NDR_ALIGN(ndr, 2);
 825                 } else if (ndr->flags & LIBNDR_FLAG_ALIGN4) {
 826                         length = NDR_ALIGN(ndr, 4);
 827                 } else if (ndr->flags & LIBNDR_FLAG_ALIGN8) {
 828                         length = NDR_ALIGN(ndr, 8);
 829                 }
 830                 if (ndr->data_size - ndr->offset < length) {
 831                         length = ndr->data_size - ndr->offset;
 832                 }
 833         } else if (ndr->flags & LIBNDR_FLAG_REMAINING) {
 834                 length = ndr->data_size - ndr->offset;
 835         } else {
 836                 NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &length));
 837         }
 838         NDR_PULL_NEED_BYTES(ndr, length);
 839         *blob = data_blob_talloc(ndr->current_mem_ctx, ndr->data+ndr->offset, length);
 840         ndr->offset += length;
 841         return NDR_ERR_SUCCESS;
 842 }
 843
 844 _PUBLIC_ uint32_t ndr_size_DATA_BLOB(int ret, const DATA_BLOB *data, int flags)
 845 {
 846         if (!data) return ret;
 847         return ret + data->length;
 848 }
 849
 850 _PUBLIC_ void ndr_print_bool(struct ndr_print *ndr, const char *name, const bool b)
 851 {
 852         ndr->print(ndr, "%-25s: %s", name, b?"true":"false");
 853 }
 854
 855 _PUBLIC_ void ndr_print_sockaddr_storage(struct ndr_print *ndr, const char *name, const struct sockaddr_storage *ss)
 856 {
 857         char addr[INET6_ADDRSTRLEN];
 858         ndr->print(ndr, "%-25s: %s", name, print_sockaddr(addr, sizeof(addr), ss));
 859 }