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
|
/* $Id$
**
** remctl interface for the wallet client.
**
** Written by Russ Allbery <rra@stanford.edu>
** Copyright 2007 Board of Trustees, Leland Stanford Jr. University
**
** See LICENSE for licensing terms.
*/
#include <config.h>
#include <system.h>
#include <remctl.h>
#include <client/internal.h>
#include <util/util.h>
/*
** Given a remctl connection and a command, run the command.
**
** If data is non-NULL, save the output in it and return the length in
** length. Otherwise, send any output to stdout. Either way, send error
** output to stderr, and return the exit status (or 255 if there is an
** error).
*/
int
run_command(struct remctl *r, const char **command, char **data,
size_t *length)
{
struct remctl_output *output;
int status = 255;
if (data != NULL)
*data = NULL;
if (length != NULL)
*length = 0;
if (!remctl_command(r, command)) {
warn("%s", remctl_error(r));
return 255;
}
do {
output = remctl_output(r);
switch (output->type) {
case REMCTL_OUT_OUTPUT:
if (output->stream == 1) {
if (data != NULL) {
*data = xrealloc(*data, *length + output->length);
memcpy(*data + *length, output->data, output->length);
*length += output->length;
} else {
fwrite(output->data, 1, output->length, stdout);
}
} else {
fprintf(stderr, "wallet: ");
fwrite(output->data, 1, output->length, stderr);
}
break;
case REMCTL_OUT_STATUS:
status = output->status;
break;
case REMCTL_OUT_ERROR:
fprintf(stderr, "wallet: ");
fwrite(output->data, 1, output->length, stderr);
fputc('\n', stderr);
status = 255;
break;
case REMCTL_OUT_DONE:
break;
}
} while (output->type != REMCTL_OUT_DONE);
return status;
}
/*
** Check whether an object exists using the exists wallet interface. Returns
** true if it does, false if it doesn't, and dies on remctl errors.
*/
int
object_exists(struct remctl *r, const char *prefix, const char *type,
const char *name)
{
const char *command[5];
char *data = NULL;
size_t length;
command[0] = prefix;
command[1] = "exists";
command[2] = type;
command[3] = name;
command[4] = NULL;
if (run_command(r, command, &data, &length) != 0)
exit(1);
if (length == 4 && strncmp(data, "yes\n", 4) == 0)
return 1;
else
return 0;
}
/*
** Attempt autocreation of an object. Dies if autocreation fails.
*/
void
object_autocreate(struct remctl *r, const char *prefix, const char *type,
const char *name)
{
const char *command[5];
command[0] = prefix;
command[1] = "autocreate";
command[2] = type;
command[3] = name;
command[4] = NULL;
if (run_command(r, command, NULL, NULL) != 0)
exit(1);
}
|
