summaryrefslogtreecommitdiff
path: root/examples/libmsrpc/test/reg/security.c
blob: 6808f8c1f3454313c1c2f980835fca55887bc089 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
/*tests cac_RegSetKeySecurity()*/

#include "libmsrpc.h"
#include "test_util.h"

int main(int argc, char **argv) {
   CacServerHandle *hnd = NULL;
   TALLOC_CTX *mem_ctx  = NULL;

   fstring tmp;
   
   mem_ctx = talloc_init("regsetval");

   hnd = cac_NewServerHandle(True);

   cac_SetAuthDataFn(hnd, cactest_GetAuthDataFn);

   cac_parse_cmd_line(argc, argv, hnd);

   if(!cac_Connect(hnd, NULL)) {
      fprintf(stderr, "Could not connect to server %s. Error: %s\n", hnd->server, nt_errstr(hnd->status));
      exit(-1);
   }

   struct RegOpenKey rok;
   ZERO_STRUCT(rok);

   printf("enter key to query: ");
   cactest_readline(stdin, tmp);

   rok.in.name = talloc_strdup(mem_ctx, tmp);
   rok.in.access = REG_KEY_ALL;

   if(!cac_RegOpenKey(hnd, mem_ctx, &rok)) {
      fprintf(stderr, "Could not open key %s. Error %s\n", rok.in.name, nt_errstr(hnd->status));
      exit(-1);
   }

   struct RegGetKeySecurity rks;
   ZERO_STRUCT(rks);

   rks.in.key = rok.out.key;
   rks.in.info_type = ALL_SECURITY_INFORMATION;

   if(!cac_RegGetKeySecurity(hnd, mem_ctx, &rks)) {
      fprintf(stderr, "Could not query security for %s.  Error: %s\n", rok.in.name, nt_errstr(hnd->status));
      goto done;
   }

   printf("resetting key security...\n");

   struct RegSetKeySecurity rss;
   ZERO_STRUCT(rss);

   rss.in.key = rok.out.key;
   rss.in.info_type = ALL_SECURITY_INFORMATION;
   rss.in.size = rks.out.size;
   rss.in.descriptor = rks.out.descriptor;

   if(!cac_RegSetKeySecurity(hnd, mem_ctx, &rss)) {
      fprintf(stderr, "Could not set security. Error %s\n", nt_errstr(hnd->status));
   }

done:
   cac_RegClose(hnd, mem_ctx, rok.out.key);
   
   cac_FreeHandle(hnd);

   talloc_destroy(mem_ctx);

   return 0;
}