summaryrefslogtreecommitdiffstats
path: root/src/lib/kadm5/unit-test/lock-test.c
blob: 85049a7e7fbc742e73ea3c5af992cdf720441be5 (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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#include <stdio.h>
#include <krb5.h>
#include <kadm5/admin.h>
#include <kdb.h>
#include <string.h>

char *whoami;

static void usage()
{
     fprintf(stderr,
	     "Usage: %s {shared|exclusive|permanent|release|"
	     "get name|wait} ...\n", whoami);
     exit(1);
}

int main(int argc, char **argv)
{
     krb5_error_code ret;
     osa_policy_ent_t entry;
     krb5_context context;
     kadm5_config_params params;
     krb5_error_code kret;

     whoami = argv[0];

     kret = kadm5_init_krb5_context(&context);
     if (kret) {
	 com_err(whoami, kret, "while initializing krb5");
	 exit(1);
     }

     params.mask = 0;
     ret = kadm5_get_config_params(context, 1, &params, &params);
     if (ret) {
	  com_err(whoami, ret, "while retrieving configuration parameters");
	  exit(1);
     }
     if (! (params.mask & KADM5_CONFIG_ADBNAME)) {
	  com_err(whoami, KADM5_BAD_SERVER_PARAMS,
		  "while retrieving configuration parameters");
	  exit(1);
     }

     ret = krb5_db_open( context, NULL, KRB5_KDB_OPEN_RW);
     if (ret) {
	  com_err(whoami, ret, "while opening database");
	  exit(1);
     }

     argc--; argv++;
     while (argc) {
	  if (strcmp(*argv, "shared") == 0) {
	       ret = krb5_db_lock(context, KRB5_DB_LOCKMODE_SHARED);
	       if (ret)
		    com_err(whoami, ret, "while getting shared lock");
	       else
		    printf("shared\n");
	  } else if (strcmp(*argv, "exclusive") == 0) {
	       ret = krb5_db_lock(context, KRB5_DB_LOCKMODE_EXCLUSIVE );
	       if (ret)
		    com_err(whoami, ret, "while getting exclusive lock");
	       else
		    printf("exclusive\n");
	  } else if (strcmp(*argv, "permanent") == 0) {
	       ret = krb5_db_lock(context, KRB5_DB_LOCKMODE_EXCLUSIVE );
	       if (ret)
		    com_err(whoami, ret, "while getting permanent lock");
	       else
		    printf("permanent\n");
	  } else if (strcmp(*argv, "release") == 0) {
	       ret = krb5_db_unlock(context);
	       if (ret)
		    com_err(whoami, ret, "while releasing lock");
	       else
		    printf("released\n");
	  } else if (strcmp(*argv, "get") == 0) {
	       int cnt = 1;
	       argc--; argv++;
	       if (!argc) usage();
	       if ((ret = krb5_db_get_policy(context, *argv,
					     &entry, &cnt)) ) {
		    com_err(whoami, ret, "while getting policy");
	       } else {
		    printf("retrieved\n");
		    krb5_db_free_policy(context, entry);
	       }
	  } else if (strcmp(*argv, "wait") == 0) {
	       getchar();
	  } else {
	       fprintf(stderr, "%s: Invalid argument \"%s\"\n",
		       whoami, *argv);
	       usage();
	  }

	  argc--; argv++;
     }

     ret = krb5_db_fini(context);
     if (ret) {
	  com_err(whoami, ret, "while closing database");
	  exit(1);
     }

     return 0;
}