#include "cacusermgr.h"
#define DEFAULT_MENU_LINES 15
void create_menu(CacServerHandle *hnd, TALLOC_CTX *mem_ctx, POLICY_HND *dom_hnd) {
struct SamCreateUser cu;
struct SamCreateGroup cg;
fstring in;
fstring tmp;
if(!hnd || !mem_ctx || !dom_hnd) {
printf("No Handle to SAM.\n");
return;
}
in[0] = '\0';
while(in[0] != 'c' && in[0] != 'C' && in[0] != 'q' && in[0] != 'Q') {
printf("\n");
printf("[u] Create User\n");
printf("[g] Create Group\n");
printf("[m] Create Machine Account\n");
printf("[c] Cancel\n\n");
printf("Command: ");
mgr_getline(in);
printf("\n");
switch(in[0]) {
case 'u':
case 'U':
ZERO_STRUCT(cu);
cu.in.dom_hnd = dom_hnd;
cu.in.acb_mask = ACB_NORMAL;
printf("Enter name: ");
mgr_getline(tmp);
cu.in.name = talloc_strdup(mem_ctx, tmp);
if(!cac_SamCreateUser(hnd, mem_ctx, &cu)) {
printerr("Could not create user.", hnd->status);
}
else {
user_menu(hnd, mem_ctx, dom_hnd, cu.out.user_hnd);
}
in[0] = 'c';
break;
case 'g':
case 'G':
ZERO_STRUCT(cg);
cg.in.dom_hnd = dom_hnd;
cg.in.access = MAXIMUM_ALLOWED_ACCESS;
printf("Enter name: ");
mgr_getline(tmp);
cg.in.name = talloc_strdup(mem_ctx, tmp);
if(!cac_SamCreateGroup(hnd, mem_ctx, &cg)) {
printerr("Could not create group.", hnd->status);
}
else {
group_menu(hnd, mem_ctx, dom_hnd, cg.out.group_hnd);
}
in[0] = 'c';
break;
case 'm':
case 'M':
ZERO_STRUCT(cu);
cu.in.dom_hnd = dom_hnd;
cu.in.acb_mask = ACB_WSTRUST;
printf("Enter machine name: ");
mgr_getline(tmp);
if(tmp[strlen(tmp) - 1] != '$')
cu.in.name = talloc_asprintf(mem_ctx, "%s$", tmp);
else
cu.in.name = talloc_strdup(mem_ctx, tmp);
strlower_m(cu.in.name);
printf("Creating account: %s\n", cu.in.name);
if(!cac_SamCreateUser(hnd, mem_ctx, &cu)) {
printerr("Could not create account.", hnd->status);
}
else {
user_menu(hnd, mem_ctx, dom_hnd, cu.out.user_hnd);
}
in[0] = 'c';
break;
case 'c':
case 'C':
case 'q':
case 'Q':
break;
default:
printf("Invalid option\n");
}
}
return;
}
void main_menu(CacServerHandle *hnd, TALLOC_CTX *mem_ctx, POLICY_HND *dom_hnd) {
fstring in;
uint32 rid_type = 0;
struct SamOpenUser openu;
struct SamOpenGroup openg;
struct SamEnumUsers enumu;
struct SamEnumGroups enumg;
struct SamFlush flush;
char *name = NULL;
uint32 rid = 0;
if(!hnd || !mem_ctx || !dom_hnd) {
printf("No handle to SAM.\n");
return;
}
ZERO_STRUCT(flush);
flush.in.dom_hnd = dom_hnd;
in[0] = '\0';
while(in[0] != 'q' && in[0] != 'Q') {
printf("\n");
printf("[o] Open User or Group\n");
printf("[c] Create Account or Group\n");
printf("[u] List Users\n");
printf("[g] List Groups\n");
printf("[m] List Machine Accounts\n");
printf("[q] Quit\n\n");
printf("Command: ");
mgr_getline(in);
printf("\n");
switch(in[0]) {
case 'o':
case 'O':
printf("Enter RID or Name: ");
rid_type = rid_or_name(hnd, mem_ctx, dom_hnd, &rid, &name);
if(rid_type == CAC_USER_RID) {
ZERO_STRUCT(openu);
openu.in.dom_hnd = dom_hnd;
openu.in.rid = rid;
openu.in.access = MAXIMUM_ALLOWED_ACCESS;
if(!cac_SamOpenUser(hnd, mem_ctx, &openu))
printerr("Could not open user.", hnd->status);
else {
user_menu(hnd, mem_ctx, dom_hnd, openu.out.user_hnd);
if(!cac_SamFlush(hnd, mem_ctx, &flush)) {
printerr("Lost handle while flushing SAM.", hnd->status);
in[0] = 'q';
}
}
}
else if(rid_type == CAC_GROUP_RID) {
ZERO_STRUCT(openg);
openg.in.dom_hnd = dom_hnd;
openg.in.rid = rid;
openg.in.access = MAXIMUM_ALLOWED_ACCESS;
if(!cac_SamOpenGroup(hnd, mem_ctx, &openg))
printerr("Could not open group.", hnd->status);
else {
group_menu(hnd, mem_ctx, dom_hnd, openg.out.group_hnd);
if(!cac_SamFlush(hnd, mem_ctx, &flush)) {
printerr("Lost handle while flushing SAM.", hnd->status);
in[0] = 'q';
}
}
}
else {
printf("Unknown RID/Name.\n");
}
break;
case 'c':
case 'C':
create_menu(hnd, mem_ctx, dom_hnd);
if(!cac_SamFlush(hnd, mem_ctx, &flush)) {
printerr("Lost handle while flushing SAM.", hnd->status);
in[0] = 'q';
}
break;
case 'u':
case 'U':
ZERO_STRUCT(enumu);
enumu.in.dom_hnd = dom_hnd;
enumu.in.acb_mask = ACB_NORMAL;
printf("Users:\n");
while(cac_SamEnumUsers(hnd, mem_ctx, &enumu)) {
print_rid_list(enumu.out.rids, enumu.out.names, enumu.out.num_users);
}
if(CAC_OP_FAILED(hnd->status))
printerr("Error occured while enumerating users.", hnd->status);
break;
case 'g':
case 'G':
ZERO_STRUCT(enumg);
enumg.in.dom_hnd = dom_hnd;
while(cac_SamEnumGroups(hnd, mem_ctx, &enumg)) {
print_rid_list( enumg.out.rids, enumg.out.names, enumg.out.num_groups);
}
if(CAC_OP_FAILED(hnd->status))
printerr("Error occured while enumerating groups.", hnd->status);
break;
case 'm':
case 'M':
ZERO_STRUCT(enumu);
enumu.in.dom_hnd = dom_hnd;
enumu.in.acb_mask = ACB_WSTRUST;
printf("Users:\n");
while(cac_SamEnumUsers(hnd, mem_ctx, &enumu)) {
print_rid_list( enumu.out.rids, enumu.out.names, enumu.out.num_users);
}
if(CAC_OP_FAILED(hnd->status))
printerr("Error occured while enumerating accounts.", hnd->status);
break;
case 'q':
case 'Q':
break;
default:
printf("Invalid Command.\n");
}
}
}
int main(int argc, char **argv) {
CacServerHandle *hnd = NULL;
TALLOC_CTX *mem_ctx = NULL;
struct SamOpenDomain sod;
mem_ctx = talloc_init("cacusermgr");
if(!mem_ctx) {
printf("Could not initialize Talloc Context\n");
exit(-1);
}
hnd = cac_NewServerHandle(True);
if(!hnd) {
printf("Could not create server handle\n");
exit(-1);
}
if(!process_cmd_line(hnd, mem_ctx, argc, argv))
usage();
if(!cac_Connect(hnd, NULL)) {
printf("Could not connect to server %s. %s\n", hnd->server, nt_errstr(hnd->status));
exit(-1);
}
ZERO_STRUCT(sod);
sod.in.access = MAXIMUM_ALLOWED_ACCESS;
if(!cac_SamOpenDomain(hnd, mem_ctx, &sod)) {
printf("Could not open handle to domain SAM. %s\n", nt_errstr(hnd->status));
goto cleanup;
}
main_menu(hnd, mem_ctx, sod.out.dom_hnd);
cleanup:
if(sod.out.dom_hnd)
cac_SamClose(hnd, mem_ctx, sod.out.dom_hnd);
if(sod.out.sam)
cac_SamClose(hnd, mem_ctx, sod.out.sam);
cac_FreeHandle(hnd);
talloc_destroy(mem_ctx);
return 0;
}