librpc/ndr/ndr_ntprinting.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    routines for marshalling/unmarshalling special ntprinting structures
   5
   6    Copyright (C) Guenther Deschner 2010
   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/gen_ndr/ndr_ntprinting.h"
  24
  25 _PUBLIC_ enum ndr_err_code ndr_pull_ntprinting_printer(struct ndr_pull *ndr, int ndr_flags, struct ntprinting_printer *r)
  26 {
  27         uint32_t _ptr_devmode;
  28         TALLOC_CTX *_mem_save_devmode_0;
  29         {
  30                 uint32_t _flags_save_STRUCT = ndr->flags;
  31                 ndr_set_flags(&ndr->flags, LIBNDR_FLAG_NOALIGN);
  32                 if (ndr_flags & NDR_SCALARS) {
  33                         NDR_CHECK(ndr_pull_align(ndr, 5));
  34                         NDR_CHECK(ndr_pull_ntprinting_printer_info(ndr, NDR_SCALARS, &r->info));
  35                         NDR_CHECK(ndr_pull_generic_ptr(ndr, &_ptr_devmode));
  36                         if (_ptr_devmode) {
  37                                 NDR_PULL_ALLOC(ndr, r->devmode);
  38                         } else {
  39                                 r->devmode = NULL;
  40                         }
  41                 }
  42                 if (ndr_flags & NDR_BUFFERS) {
  43                         if (r->devmode) {
  44                                 _mem_save_devmode_0 = NDR_PULL_GET_MEM_CTX(ndr);
  45                                 NDR_PULL_SET_MEM_CTX(ndr, r->devmode, 0);
  46                                 NDR_CHECK(ndr_pull_ntprinting_devicemode(ndr, NDR_SCALARS|NDR_BUFFERS, r->devmode));
  47                                 NDR_PULL_SET_MEM_CTX(ndr, _mem_save_devmode_0, 0);
  48                         }
  49                 }
  50                 if (ndr_flags & NDR_SCALARS) {
  51                         r->count = 0;
  52                         NDR_PULL_ALLOC_N(ndr, r->printer_data, r->count);
  53                         while (ndr->offset + 4 <= ndr->data_size) {
  54                                 uint32_t ptr = 0;
  55                                 ptr = IVAL(ndr->data, ndr->offset);
  56                                 if (ptr == 0) {
  57                                         break;
  58                                 }
  59                                 r->printer_data = talloc_realloc(ndr, r->printer_data, struct ntprinting_printer_data, r->count + 1);
  60                                 NDR_ERR_HAVE_NO_MEMORY(r->printer_data);
  61                                 NDR_CHECK(ndr_pull_ntprinting_printer_data(ndr, NDR_SCALARS, &r->printer_data[r->count]));
  62                                 r->count++;
  63                         }
  64                         NDR_CHECK(ndr_pull_trailer_align(ndr, 5));
  65                 }
  66                 ndr->flags = _flags_save_STRUCT;
  67         }
  68         return NDR_ERR_SUCCESS;
  69 }