sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-08-24 21:49:29 +03:00
Code Activity

s4-smbtorture: add RPC torture test for winreg_QueryMultipleValues2.

Browse Source 
Guenther
This commit is contained in:
Günther Deschner
2010-06-29 12:14:02 +02:00
parent c02cbaae34
commit 7ae7750c40
1 changed files with 41 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
source4/torture/rpc/winreg.c
View File

@ -1645,6 +1645,45 @@ static bool test_QueryMultipleValues(struct dcerpc_binding_handle *b,
return true;
}
static bool test_QueryMultipleValues2(struct dcerpc_binding_handle *b,
struct torture_context *tctx,
struct policy_handle *handle,
const char *valuename)
{
struct winreg_QueryMultipleValues2 r;
uint32_t offered = 0, needed;
ZERO_STRUCT(r);
r.in.key_handle = handle;
r.in.values_in = r.out.values_out = talloc_zero_array(tctx, struct QueryMultipleValue, 1);
r.in.values_in[0].ve_valuename = talloc(tctx, struct winreg_ValNameBuf);
r.in.values_in[0].ve_valuename->name = valuename;
/* size needs to be set manually for winreg_ValNameBuf */
r.in.values_in[0].ve_valuename->size = strlen_m_term(valuename)*2;
r.in.num_values = 1;
r.in.offered = &offered;
r.out.needed = &needed;
torture_assert_ntstatus_ok(tctx,
dcerpc_winreg_QueryMultipleValues2_r(b, tctx, &r),
"QueryMultipleValues2 failed");
if (W_ERROR_EQUAL(r.out.result, WERR_MORE_DATA)) {
*r.in.offered = *r.out.needed;
r.in.buffer = r.out.buffer = talloc_zero_array(tctx, uint8_t, *r.in.offered);
torture_assert_ntstatus_ok(tctx,
dcerpc_winreg_QueryMultipleValues2_r(b, tctx, &r),
"QueryMultipleValues2 failed");
}
torture_assert_werr_ok(tctx, r.out.result,
"QueryMultipleValues2 failed");
return true;
}
static bool test_QueryValue(struct dcerpc_binding_handle *b,
struct torture_context *tctx,
struct policy_handle *handle,
@ -1813,6 +1852,8 @@ static bool test_EnumValue(struct dcerpc_binding_handle *b,
r.out.name->name);
ret &= test_QueryMultipleValues(b, tctx, handle,
r.out.name->name);
ret &= test_QueryMultipleValues2(b, tctx, handle,
r.out.name->name);
}
r.in.enum_index++;