#include "includes.h"
#define ALLOC_CHECK(ptr, err, label, str) do { if ((ptr) == NULL) { DEBUG(0, ("%s: out of memory!\n", str)); err = NT_STATUS_NO_MEMORY; goto label; } } while(0)
#define NTSTATUS_CHECK(err, label, str1, str2) do { if (!NT_STATUS_IS_OK(err)) { DEBUG(0, ("%s: %s failed!\n", str1, str2)); } } while(0)
#if 0
NTSTATUS is_mapped_group(BOOL *mapped, const DOM_SID *sid)
{
NTSTATUS result;
gid_t id;
result = idmap_get_gid_from_sid(&id, sid, False);
if (NT_STATUS_IS_OK(result)) {
*mapped = gid_is_in_winbind_range(id);
} else {
*mapped = False;
}
return result;
}
#endif
NTSTATUS dupalloc_luid_attr(TALLOC_CTX *mem_ctx, LUID_ATTR **new_la, LUID_ATTR *old_la)
{
NTSTATUS ret;
if ( !old_la )
return NT_STATUS_OK;
*new_la = TALLOC_P(mem_ctx, LUID_ATTR);
ALLOC_CHECK(new_la, ret, done, "dupalloc_luid_attr");
(*new_la)->luid.high = old_la->luid.high;
(*new_la)->luid.low = old_la->luid.low;
(*new_la)->attr = old_la->attr;
ret = NT_STATUS_OK;
done:
return ret;
}
NTSTATUS init_privilege(PRIVILEGE_SET **priv_set)
{
NTSTATUS ret;
TALLOC_CTX *mem_ctx = talloc_init("privilege set");
ALLOC_CHECK(mem_ctx, ret, done, "init_privilege");
*priv_set = TALLOC_ZERO_P(mem_ctx, PRIVILEGE_SET);
ALLOC_CHECK(*priv_set, ret, done, "init_privilege");
(*priv_set)->mem_ctx = mem_ctx;
ret = NT_STATUS_OK;
done:
return ret;
}
NTSTATUS init_priv_with_ctx(TALLOC_CTX *mem_ctx, PRIVILEGE_SET **priv_set)
{
NTSTATUS ret;
*priv_set = TALLOC_ZERO_P(mem_ctx, PRIVILEGE_SET);
ALLOC_CHECK(*priv_set, ret, done, "init_privilege");
(*priv_set)->mem_ctx = mem_ctx;
(*priv_set)->ext_ctx = True;
ret = NT_STATUS_OK;
done:
return ret;
}
void reset_privilege(PRIVILEGE_SET *priv_set)
{
priv_set->count = 0;
priv_set->control = 0;
priv_set->set = NULL;
}
void destroy_privilege(PRIVILEGE_SET **priv_set)
{
reset_privilege(*priv_set);
if (!((*priv_set)->ext_ctx))
talloc_destroy((*priv_set)->mem_ctx);
*priv_set = NULL;
}
NTSTATUS add_privilege(PRIVILEGE_SET *priv_set, LUID_ATTR set)
{
NTSTATUS ret;
LUID_ATTR *new_set;
if (NT_STATUS_IS_OK(check_priv_in_privilege(priv_set, set)))
return NT_STATUS_UNSUCCESSFUL;
new_set = TALLOC_REALLOC_ARRAY(priv_set->mem_ctx, priv_set->set, LUID_ATTR, priv_set->count + 1);
ALLOC_CHECK(new_set, ret, done, "add_privilege");
new_set[priv_set->count].luid.high = set.luid.high;
new_set[priv_set->count].luid.low = set.luid.low;
new_set[priv_set->count].attr = set.attr;
priv_set->count++;
priv_set->set = new_set;
ret = NT_STATUS_OK;
done:
return ret;
}
NTSTATUS add_all_privilege(PRIVILEGE_SET *priv_set)
{
NTSTATUS result = NT_STATUS_OK;
LUID_ATTR set;
set.attr = 0;
set.luid.high = 0;
set.luid.low = SE_PRIV_ADD_USERS;
result = add_privilege(priv_set, set);
NTSTATUS_CHECK(result, done, "add_all_privilege", "add_privilege");
set.luid.low = SE_PRIV_ADD_MACHINES;
result = add_privilege(priv_set, set);
NTSTATUS_CHECK(result, done, "add_all_privilege", "add_privilege");
set.luid.low = SE_PRIV_PRINT_OPERATOR;
result = add_privilege(priv_set, set);
NTSTATUS_CHECK(result, done, "add_all_privilege", "add_privilege");
return result;
}
NTSTATUS check_empty_privilege(PRIVILEGE_SET *priv_set)
{
if (!priv_set)
return NT_STATUS_INVALID_PARAMETER;
if (priv_set->count == 0)
return NT_STATUS_OK;
return NT_STATUS_UNSUCCESSFUL;
}
NTSTATUS check_priv_in_privilege(PRIVILEGE_SET *priv_set, LUID_ATTR set)
{
int i;
if (!priv_set)
return NT_STATUS_INVALID_PARAMETER;
if (NT_STATUS_IS_OK(check_empty_privilege(priv_set)))
return NT_STATUS_UNSUCCESSFUL;
for (i = 0; i < priv_set->count; i++) {
LUID_ATTR *cur_set;
cur_set = &priv_set->set[i];
if ( (cur_set->luid.low == set.luid.low) &&
(cur_set->luid.high == set.luid.high) ) {
return NT_STATUS_OK;
}
}
return NT_STATUS_UNSUCCESSFUL;
}
NTSTATUS remove_privilege(PRIVILEGE_SET *priv_set, LUID_ATTR set)
{
NTSTATUS ret;
LUID_ATTR *new_set;
LUID_ATTR *old_set;
int i,j;
if (!priv_set)
return NT_STATUS_INVALID_PARAMETER;
if (!NT_STATUS_IS_OK(check_priv_in_privilege(priv_set, set)))
return NT_STATUS_UNSUCCESSFUL;
if (priv_set->count == 1) {
reset_privilege(priv_set);
return NT_STATUS_OK;
}
old_set = priv_set->set;
new_set = TALLOC_ARRAY(priv_set->mem_ctx, LUID_ATTR, priv_set->count - 1);
ALLOC_CHECK(new_set, ret, done, "remove_privilege");
for (i=0, j=0; i < priv_set->count; i++) {
if ( (old_set[i].luid.low == set.luid.low) &&
(old_set[i].luid.high == set.luid.high) ) {
continue;
}
new_set[j].luid.low = old_set[i].luid.low;
new_set[j].luid.high = old_set[i].luid.high;
new_set[j].attr = old_set[i].attr;
j++;
}
if (j != priv_set->count - 1) {
DEBUG(0,("remove_privilege: mismatch ! difference is not -1\n"));
DEBUGADD(0,("old count:%d, new count:%d\n", priv_set->count, j));
return NT_STATUS_INTERNAL_ERROR;
}
priv_set->count--;
priv_set->set = new_set;
ret = NT_STATUS_OK;
done:
return ret;
}
NTSTATUS dup_priv_set(PRIVILEGE_SET *new_priv_set, PRIVILEGE_SET *priv_set)
{
NTSTATUS ret;
LUID_ATTR *new_set;
LUID_ATTR *old_set;
int i;
if (!new_priv_set || !priv_set)
return NT_STATUS_INVALID_PARAMETER;
if (priv_set->count == 0) {
return NT_STATUS_OK;
}
old_set = priv_set->set;
new_set = TALLOC_ARRAY(new_priv_set->mem_ctx, LUID_ATTR, priv_set->count - 1);
ALLOC_CHECK(new_set, ret, done, "dup_priv_set");
for (i=0; i < priv_set->count; i++) {
new_set[i].luid.low = old_set[i].luid.low;
new_set[i].luid.high = old_set[i].luid.high;
new_set[i].attr = old_set[i].attr;
}
new_priv_set->count = priv_set->count;
new_priv_set->control = priv_set->control;
new_priv_set->set = new_set;
ret = NT_STATUS_OK;
done:
return ret;
}