diff options
| -rw-r--r-- | source4/libcli/security/privilege.c | 11 | ||||
| -rw-r--r-- | source4/rpc_server/lsa/dcesrv_lsa.c | 6 |
2 files changed, 14 insertions, 3 deletions
diff --git a/source4/libcli/security/privilege.c b/source4/libcli/security/privilege.c index c21e1cbb6f..9fd7192883 100644 --- a/source4/libcli/security/privilege.c +++ b/source4/libcli/security/privilege.c @@ -235,6 +235,17 @@ enum sec_privilege sec_privilege_from_mask(uint64_t mask) return -1; } +/* + map a privilege name to a privilege id. Return -1 if not found +*/ +enum sec_privilege sec_privilege_from_index(int idx) +{ + if (idx >= 0 && idx<ARRAY_SIZE(privilege_names)) { + return privilege_names[idx].privilege; + } + return -1; +} + /* return a privilege mask given a privilege id diff --git a/source4/rpc_server/lsa/dcesrv_lsa.c b/source4/rpc_server/lsa/dcesrv_lsa.c index 51b6339a12..988acda10b 100644 --- a/source4/rpc_server/lsa/dcesrv_lsa.c +++ b/source4/rpc_server/lsa/dcesrv_lsa.c @@ -291,6 +291,7 @@ static NTSTATUS dcesrv_lsa_EnumPrivs(struct dcesrv_call_state *dce_call, TALLOC_ struct dcesrv_handle *h; struct lsa_policy_state *state; uint32_t i; + enum sec_privilege priv; const char *privname; DCESRV_PULL_HANDLE(h, r->in.handle, LSA_HANDLE_POLICY); @@ -298,12 +299,11 @@ static NTSTATUS dcesrv_lsa_EnumPrivs(struct dcesrv_call_state *dce_call, TALLOC_ state = h->data; i = *r->in.resume_handle; - if (i == 0) i = 1; - while ((privname = sec_privilege_name(i)) && + while (((priv = sec_privilege_from_index(i)) != -1) && r->out.privs->count < r->in.max_count) { struct lsa_PrivEntry *e; - + privname = sec_privilege_name(priv); r->out.privs->privs = talloc_realloc(r->out.privs, r->out.privs->privs, struct lsa_PrivEntry, |