/* push a relative object - stage2 start this is called during buffers processing */ _PUBLIC_ enum ndr_err_code ndr_push_relative_ptr2_start(struct ndr_push *ndr, const void *p) { if (p == NULL) { return NDR_ERR_SUCCESS; } if (!(ndr->flags & LIBNDR_FLAG_RELATIVE_REVERSE)) { uint32_t relative_offset; size_t pad; size_t align = 1; if (ndr->offset < ndr->relative_base_offset) { return ndr_push_error(ndr, NDR_ERR_BUFSIZE, "ndr_push_relative_ptr2_start ndr->offset(%u) < ndr->relative_base_offset(%u)", ndr->offset, ndr->relative_base_offset); } relative_offset = ndr->offset - ndr->relative_base_offset; if (ndr->flags & LIBNDR_FLAG_NOALIGN) { align = 1; } else if (ndr->flags & LIBNDR_FLAG_ALIGN2) { align = 2; } else if (ndr->flags & LIBNDR_FLAG_ALIGN4) { align = 4; } else if (ndr->flags & LIBNDR_FLAG_ALIGN8) { align = 8; } pad = ndr_align_size(relative_offset, align); if (pad) { NDR_CHECK(ndr_push_zero(ndr, pad)); } return ndr_push_relative_ptr2(ndr, p); } if (ndr->relative_end_offset == -1) { return ndr_push_error(ndr, NDR_ERR_RELATIVE, "ndr_push_relative_ptr2_start RELATIVE_REVERSE flag set and relative_end_offset %d", ndr->relative_end_offset); } NDR_CHECK(ndr_token_store(ndr, &ndr->relative_begin_list, p, ndr->offset)); return NDR_ERR_SUCCESS; }
_PUBLIC_ enum ndr_err_code ndr_pull_bkrp_access_check_v3(struct ndr_pull *ndr, int ndr_flags, struct bkrp_access_check_v3 *r) { if (ndr_flags & NDR_SCALARS) { size_t ofs; size_t pad; NDR_CHECK(ndr_pull_align(ndr, 4)); NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &r->magic)); NDR_CHECK(ndr_pull_uint32(ndr, NDR_SCALARS, &r->nonce_len)); NDR_PULL_ALLOC_N(ndr, r->nonce, r->nonce_len); NDR_CHECK(ndr_pull_array_uint8(ndr, NDR_SCALARS, r->nonce, r->nonce_len)); NDR_CHECK(ndr_pull_dom_sid(ndr, NDR_SCALARS, &r->sid)); ofs = ndr->offset + 64; pad = ndr_align_size(ofs, 16); NDR_CHECK(ndr_pull_advance(ndr, pad)); NDR_CHECK(ndr_pull_array_uint8(ndr, NDR_SCALARS, r->hash, 64)); NDR_CHECK(ndr_pull_trailer_align(ndr, 4)); } if (ndr_flags & NDR_BUFFERS) { } return NDR_ERR_SUCCESS; }
/* We have manual push/pull because we didn't manage to do the alignment * purely in PIDL as the padding is sized so that the whole access_check_v3 * struct size is a multiple of 16 (as specified in 2.2.2.4 of ms-bkrp.pdf) */ _PUBLIC_ enum ndr_err_code ndr_push_bkrp_access_check_v3(struct ndr_push *ndr, int ndr_flags, const struct bkrp_access_check_v3 *r) { if (ndr_flags & NDR_SCALARS) { size_t ofs; size_t pad; NDR_CHECK(ndr_push_align(ndr, 4)); NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, 0x00000001)); NDR_CHECK(ndr_push_uint32(ndr, NDR_SCALARS, r->nonce_len)); NDR_CHECK(ndr_push_array_uint8(ndr, NDR_SCALARS, r->nonce, r->nonce_len)); NDR_CHECK(ndr_push_dom_sid(ndr, NDR_SCALARS, &r->sid)); /* We articially increment the offset of 64 bytes (size of hash * comming after the pad) so that ndr_align can determine easily * the correct pad size to make the whole struct 16 bytes aligned */ ofs = ndr->offset + 64; pad = ndr_align_size(ofs, 16); NDR_CHECK(ndr_push_zero(ndr, pad)); NDR_CHECK(ndr_push_array_uint8(ndr, NDR_SCALARS, r->hash, 64)); NDR_CHECK(ndr_push_trailer_align(ndr, 4)); } if (ndr_flags & NDR_BUFFERS) { } return NDR_ERR_SUCCESS; }
/* push a relative object - stage2 end this is called during buffers processing */ _PUBLIC_ enum ndr_err_code ndr_push_relative_ptr2_end(struct ndr_push *ndr, const void *p) { uint32_t begin_offset = 0xFFFFFFFF; ssize_t len; uint32_t correct_offset = 0; uint32_t align = 1; uint32_t pad = 0; if (p == NULL) { return NDR_ERR_SUCCESS; } if (!(ndr->flags & LIBNDR_FLAG_RELATIVE_REVERSE)) { return NDR_ERR_SUCCESS; } if (ndr->flags & LIBNDR_FLAG_NO_NDR_SIZE) { /* better say more than calculation a too small buffer */ NDR_PUSH_ALIGN(ndr, 8); return NDR_ERR_SUCCESS; } if (ndr->relative_end_offset < ndr->offset) { return ndr_push_error(ndr, NDR_ERR_RELATIVE, "ndr_push_relative_ptr2_end:" "relative_end_offset %u < offset %u", ndr->relative_end_offset, ndr->offset); } NDR_CHECK(ndr_token_retrieve(&ndr->relative_begin_list, p, &begin_offset)); /* we have marshalled a buffer, see how long it was */ len = ndr->offset - begin_offset; if (len < 0) { return ndr_push_error(ndr, NDR_ERR_RELATIVE, "ndr_push_relative_ptr2_end:" "offset %u - begin_offset %u < 0", ndr->offset, begin_offset); } if (ndr->relative_end_offset < len) { return ndr_push_error(ndr, NDR_ERR_RELATIVE, "ndr_push_relative_ptr2_end:" "relative_end_offset %u < len %lld", ndr->offset, (long long)len); } /* the reversed offset is at the end of the main buffer */ correct_offset = ndr->relative_end_offset - len; /* TODO: remove this hack and let the idl use FLAG_ALIGN2 explicit */ align = 2; if (ndr->flags & LIBNDR_FLAG_ALIGN2) { align = 2; } else if (ndr->flags & LIBNDR_FLAG_ALIGN4) { align = 4; } else if (ndr->flags & LIBNDR_FLAG_ALIGN8) { align = 8; } pad = ndr_align_size(correct_offset, align); if (pad) { correct_offset += pad; correct_offset -= align; } if (correct_offset < begin_offset) { return ndr_push_error(ndr, NDR_ERR_RELATIVE, "ndr_push_relative_ptr2_end: " "correct_offset %u < begin_offset %u", correct_offset, begin_offset); } if (len > 0) { uint32_t clear_size = correct_offset - begin_offset; clear_size = MIN(clear_size, len); /* now move the marshalled buffer to the end of the main buffer */ memmove(ndr->data + correct_offset, ndr->data + begin_offset, len); if (clear_size) { /* and wipe out old buffer within the main buffer */ memset(ndr->data + begin_offset, '\0', clear_size); } } /* and set the end offset for the next buffer */ ndr->relative_end_offset = correct_offset; /* finally write the offset to the main buffer */ ndr->offset = correct_offset; NDR_CHECK(ndr_push_relative_ptr2(ndr, p)); /* restore to where we were in the main buffer */ ndr->offset = begin_offset; return NDR_ERR_SUCCESS; }