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
107
108
109
110
111
112
113
|
/*
* Test suite for environment variables set by the server.
*
* Written by Russ Allbery <eagle@eyrie.org>
* Copyright 2015, 2018 Russ Allbery <eagle@eyrie.org>
* Copyright 2006, 2009-2010, 2012-2014
* The Board of Trustees of the Leland Stanford Junior University
*
* SPDX-License-Identifier: MIT
*/
#include <config.h>
#include <portable/system.h>
#include <client/internal.h>
#include <client/remctl.h>
#include <tests/tap/basic.h>
#include <tests/tap/kerberos.h>
#include <tests/tap/remctl.h>
#include <tests/tap/string.h>
/*
* Run the remote env command with the given variable and return the value
* from the server or NULL if there was an error.
*/
static char *
test_env(struct remctl *r, const char *variable)
{
struct remctl_output *output;
char *value = NULL;
const char *command[] = {"test", "env", NULL, NULL};
command[2] = variable;
if (!remctl_command(r, command)) {
diag("remctl error %s", remctl_error(r));
return NULL;
}
do {
output = remctl_output(r);
switch (output->type) {
case REMCTL_OUT_OUTPUT:
value = bstrndup(output->data, output->length);
break;
case REMCTL_OUT_STATUS:
if (output->status != 0) {
free(value);
diag("test env returned status %d", output->status);
return NULL;
}
if (value == NULL)
value = bstrdup("");
return value;
case REMCTL_OUT_ERROR:
free(value);
diag("test env returned error: %.*s", (int) output->length,
output->data);
return NULL;
case REMCTL_OUT_DONE:
free(value);
diag("unexpected done token");
return NULL;
}
} while (output->type == REMCTL_OUT_OUTPUT);
return value;
}
int
main(void)
{
struct kerberos_config *config;
char *expected, *value;
struct remctl *r;
time_t expires;
/* Unless we have Kerberos available, we can't really do anything. */
config = kerberos_setup(TAP_KRB_NEEDS_KEYTAB);
remctld_start(config, "data/conf-simple", NULL);
plan(5);
/* Run the tests. */
r = remctl_new();
if (!remctl_open(r, "localhost", 14373, config->principal))
bail("cannot contact remctld");
basprintf(&expected, "%s\n", config->principal);
value = test_env(r, "REMUSER");
is_string(expected, value, "value for REMUSER");
free(value);
value = test_env(r, "REMOTE_USER");
is_string(expected, value, "value for REMOTE_USER");
free(value);
value = test_env(r, "REMOTE_ADDR");
if (strcmp(value, "::1\n") == 0)
is_string("::1\n", value, "value for REMOTE_ADDR");
else
is_string("127.0.0.1\n", value, "value for REMOTE_ADDR");
free(value);
value = test_env(r, "REMOTE_HOST");
ok(strcmp(value, "\n") == 0 || strstr(value, "localhost") != NULL,
"value for REMOTE_HOST");
free(value);
value = test_env(r, "REMOTE_EXPIRES");
expires = strtol(value, NULL, 10);
free(value);
ok(expires > time(NULL), "REMOTE_EXPIRES is in the future (%lu)",
(unsigned long) expires);
remctl_close(r);
free(expected);
return 0;
}
|
