e64e30058d2a253be2a1ae95e48f2c99006ef6df
[openafs.git] / tests / volser / vos-t.c
1 #include <afsconfig.h>
2 #include <afs/param.h>
3
4 #include <roken.h>
5
6 #include <sys/wait.h>
7
8 #include <rx/rx.h>
9 #include <ubik.h>
10
11 #include <afs/com_err.h>
12 #include <afs/vldbint.h>
13 #include <afs/cellconfig.h>
14
15 #include <tap/basic.h>
16
17 #include "common.h"
18
19 /* This checks for a bug in vos where it would fail to allocate additional
20  * space for the results of multi homed VL_GetAddrsU, and so would segfault
21  * if a host with a small number of addresses was immediately followed by
22  * a host with a large number of addresses.
23  */
24 void
25 TestListAddrs(struct ubik_client *client, char *dirname)
26 {
27     int code;
28     bulkaddrs addrs;
29     pid_t pid;
30     int outpipe[2];
31     int status;
32     char *buffer;
33     size_t len;
34
35     afs_uint32 addrsA[] = {0x0a000000};
36     afs_uint32 addrsB[] = {0x0a000001, 0x0a000002, 0x0a000003, 0x0a000004,
37                            0x0a000005, 0x0a000006, 0x0a000007, 0x0a000008,
38                            0x0a000009, 0x0a00000a, 0x0a00000b, 0x0a00000c,
39                            0x0a00000d, 0x0a00000e, 0x0a00000f};
40     char uuidA[] = {0x4F, 0x44, 0x94, 0x47, 0x76, 0xBA, 0x47, 0x2C, 0x97, 0x1A,
41                     0x86, 0x6B, 0xC0, 0x10, 0x1A, 0x4B};
42     char uuidB[] = {0x5D, 0x2A, 0x39, 0x36, 0x94, 0xB2, 0x48, 0x90, 0xA8, 0xD2,
43                     0x7F, 0xBC, 0x1B, 0x29, 0xDA, 0x9B};
44     char expecting[] = "10.0.0.0\n10.0.0.1\n10.0.0.2\n10.0.0.3\n10.0.0.4\n"
45                        "10.0.0.5\n10.0.0.6\n10.0.0.7\n10.0.0.8\n10.0.0.9\n"
46                        "10.0.0.10\n10.0.0.11\n10.0.0.12\n10.0.0.13\n"
47                        "10.0.0.14\n10.0.0.15\n";
48
49     addrs.bulkaddrs_len = 1;
50     addrs.bulkaddrs_val = addrsA;
51     code = ubik_VL_RegisterAddrs(client, 0, (afsUUID *)uuidA, 0, &addrs);
52     is_int(0, code, "First address registration succeeds");
53
54     addrs.bulkaddrs_len = 15;
55     addrs.bulkaddrs_val = addrsB;
56     code = ubik_VL_RegisterAddrs(client, 0, (afsUUID *)uuidB, 0, &addrs);
57     is_int(0, code, "Second address registration succeeds");
58
59     /* Now we need to run vos ListAddrs and see what happens ... */
60     pipe(outpipe);
61     pid = fork();
62     if (pid == 0) {
63         dup2(outpipe[1], STDOUT_FILENO); /* Redirect stdout into pipe */
64         close(outpipe[0]);
65         close(outpipe[1]);
66
67         execl("../../src/volser/vos", "vos",
68               "listaddrs", "-config", dirname, "-noauth", NULL);
69     }
70     close(outpipe[1]);
71     buffer = malloc(4096);
72     len = read(outpipe[0], buffer, 4096);
73     waitpid(pid, &status, 0);
74     buffer[len]='\0';
75     is_string(expecting, buffer, "vos output matches");
76     free(buffer);
77 }
78
79 int
80 main(int argc, char **argv)
81 {
82     char *dirname;
83     struct afsconf_dir *dir;
84     struct afsconf_cell info;
85     int code;
86     int i;
87     pid_t serverPid;
88     struct rx_securityClass *secClass;
89     struct rx_connection *serverconns[MAXSERVERS];
90     struct ubik_client *ubikClient = NULL;
91     int secIndex;
92
93     plan(6);
94
95     code = rx_Init(0);
96
97     dirname = afstest_BuildTestConfig();
98
99     dir = afsconf_Open(dirname);
100
101     code = afstest_AddDESKeyFile(dir);
102     if (code) {
103         afs_com_err("vos-t", code, "while adding test DES keyfile");
104         exit(1);
105     }
106
107     code = afstest_StartVLServer(dirname, &serverPid);
108     if (code) {
109         afs_com_err("vos-t", code, "while starting the vlserver");
110         exit(1);
111     }
112
113     /* Let it figure itself out ... */
114     sleep(5);
115
116     code = afsconf_ClientAuthSecure(dir, &secClass, &secIndex);
117     if (code) {
118         afs_com_err("vos-t", code, "while getting a fake token");
119         exit(1);
120     }
121
122     code = afsconf_GetCellInfo(dir, NULL, AFSCONF_VLDBSERVICE, &info);
123     if (code) {
124         afs_com_err("vos-t", code, " while getting addresses from cellservdb");
125         exit(1);
126     }
127     ok(info.numServers < MAXSERVERS, "Number of servers is within range");
128
129     for (i = 0; i < info.numServers; i++)
130         serverconns[i] = rx_NewConnection(info.hostAddr[i].sin_addr.s_addr,
131                                           info.hostAddr[i].sin_port,
132                                           USER_SERVICE_ID,
133                                           secClass, secIndex);
134
135     code = ubik_ClientInit(serverconns, &ubikClient);
136
137     TestListAddrs(ubikClient, dirname);
138
139     code = afstest_StopVLServer(serverPid);
140     is_int(0, code, "Server exited cleanly");
141
142     return 0;
143 }