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
114
115
116
117
118
119
120
121
|
/*
* Test suite for server command logging.
*
* Written by Russ Allbery <eagle@eyrie.org>
* Copyright 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 <portable/uio.h>
#include <server/internal.h>
#include <tests/tap/basic.h>
#include <tests/tap/messages.h>
int
main(void)
{
struct rule rule = {NULL, 0, NULL, NULL, NULL, NULL, NULL, 0,
NULL, NULL, 0, 0, NULL, NULL, NULL};
struct iovec **command;
int i;
plan(8);
/* Command without subcommand. */
command = bcalloc(5, sizeof(struct iovec *));
command[0] = bmalloc(sizeof(struct iovec));
command[0]->iov_base = bstrdup("foo");
command[0]->iov_len = strlen("foo");
command[1] = NULL;
errors_capture();
server_log_command(command, &rule, "test@EXAMPLE.ORG");
is_string("COMMAND from test@EXAMPLE.ORG: foo\n", errors,
"command without subcommand logging");
/* Filtering of non-printable characters. */
command[1] = bmalloc(sizeof(struct iovec));
command[1]->iov_base = bstrdup("f\1o\33o\37o\177");
command[1]->iov_len = strlen("f\1o\33o\37o\177");
command[2] = NULL;
errors_capture();
server_log_command(command, &rule, "test");
is_string("COMMAND from test: foo f.o.o.o.\n", errors,
"logging of unprintable characters");
/* Simple command. */
for (i = 2; i < 4; i++)
command[i] = bmalloc(sizeof(struct iovec));
free(command[1]->iov_base);
command[1]->iov_base = bstrdup("bar");
command[1]->iov_len = strlen("bar");
command[2]->iov_base = bstrdup("arg1");
command[2]->iov_len = strlen("arg1");
command[3]->iov_base = bstrdup("arg2");
command[3]->iov_len = strlen("arg2");
command[4] = NULL;
errors_capture();
server_log_command(command, &rule, "test@EXAMPLE.ORG");
is_string("COMMAND from test@EXAMPLE.ORG: foo bar arg1 arg2\n", errors,
"simple command logging");
/* Command with stdin numeric argument. */
rule.stdin_arg = 2;
errors_capture();
server_log_command(command, &rule, "test");
is_string("COMMAND from test: foo bar **DATA** arg2\n", errors,
"stdin argument");
/* Command with stdin set to "last". */
rule.stdin_arg = -1;
errors_capture();
server_log_command(command, &rule, "test");
is_string("COMMAND from test: foo bar arg1 **DATA**\n", errors,
"stdin last argument");
/* Logmask of a single argument. */
rule.stdin_arg = 0;
rule.logmask = bcalloc(2, sizeof(unsigned int));
rule.logmask[0] = 2;
rule.logmask[1] = 0;
errors_capture();
server_log_command(command, &rule, "test");
is_string("COMMAND from test: foo bar **MASKED** arg2\n", errors,
"one masked parameter");
/* Logmask that doesn't apply to this command. */
rule.logmask[0] = 4;
errors_capture();
server_log_command(command, &rule, "test@EXAMPLE.ORG");
is_string("COMMAND from test@EXAMPLE.ORG: foo bar arg1 arg2\n", errors,
"mask that doesn't apply");
/* Logmask of the subcommand and multiple masks. */
free(rule.logmask);
rule.logmask = bcalloc(4, sizeof(unsigned int));
rule.logmask[0] = 4;
rule.logmask[1] = 1;
rule.logmask[2] = 3;
rule.logmask[3] = 0;
errors_capture();
server_log_command(command, &rule, "test");
is_string("COMMAND from test: foo **MASKED** arg1 **MASKED**\n", errors,
"two masked parameters");
errors_uncapture();
free(errors);
errors = NULL;
free(rule.logmask);
for (i = 0; i < 4; i++) {
if (command[i] != NULL)
free(command[i]->iov_base);
free(command[i]);
}
free(command);
return 0;
}
|
