/*
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;
}
