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
122
123
124
125
126
127
128
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include "splunk-starter.h"
#include <dirent.h>
#include <sys/stat.h>
#include <vespa/log/log.h>
LOG_SETUP(".splunk-starter");
SplunkStarter::SplunkStarter() = default;
SplunkStarter::~SplunkStarter() = default;
namespace {
vespalib::string fixDir(const vespalib::string &parent, const vespalib::string &subdir) {
auto dirname = parent + "/" + subdir;
DIR *dp = opendir(dirname.c_str());
if (dp == NULL) {
if (errno != ENOENT || mkdir(dirname.c_str(), 0755) != 0) {
LOG(warning, "Could not create directory '%s'", dirname.c_str());
perror(dirname.c_str());
}
} else {
closedir(dp);
}
return dirname;
}
vespalib::string
cfFilePath(const vespalib::string &parent, const vespalib::string &filename) {
vespalib::string path = parent;
path = fixDir(path, "etc");
path = fixDir(path, "system");
path = fixDir(path, "local");
return path + "/" + filename;
}
void appendFile(FILE *target, const vespalib::string &filename) {
FILE *fp = fopen(filename.c_str(), "r");
if (fp != NULL) {
int c;
while (EOF != (c = fgetc(fp))) {
fputc(c, target);
}
fclose(fp);
}
}
} // namespace <unnamed>
void SplunkStarter::gotConfig(const LogforwarderConfig& config) {
vespalib::string path = cfFilePath(config.splunkHome, "deploymentclient.conf");
LOG(debug, "got config, writing %s", path.c_str());
vespalib::string tmpPath = path + ".new";
FILE *fp = fopen(tmpPath.c_str(), "w");
if (fp == NULL) {
LOG(warning, "could not open '%s' for write", tmpPath.c_str());
return;
}
fprintf(fp, "[deployment-client]\n");
fprintf(fp, "clientName = %s\n", config.clientName.c_str());
fprintf(fp, "phoneHomeIntervalInSecs = %i\n", config.phoneHomeInterval);
fprintf(fp, "\n");
fprintf(fp, "[target-broker:deploymentServer]\n");
fprintf(fp, "targetUri = %s\n", config.deploymentServer.c_str());
fclose(fp);
rename(tmpPath.c_str(), path.c_str());
if (getenv("VESPA_HOSTNAME") != NULL &&
getenv("VESPA_TENANT") != NULL &&
getenv("VESPA_APPLICATION")!= NULL &&
getenv("VESPA_INSTANCE") != NULL &&
getenv("VESPA_ENVIRONMENT") != NULL &&
getenv("VESPA_REGION") != NULL)
{
path = cfFilePath(config.splunkHome, "inputs.conf");
tmpPath = path + ".new";
fp = fopen(tmpPath.c_str(), "w");
if (fp != NULL) {
fprintf(fp, "[default]\n");
fprintf(fp, "host = %s\n", getenv("VESPA_HOSTNAME"));
fprintf(fp, "_meta = vespa_tenant::%s vespa_app::%s.%s vespa_zone::%s.%s\n",
getenv("VESPA_TENANT"),
getenv("VESPA_APPLICATION"),
getenv("VESPA_INSTANCE"),
getenv("VESPA_ENVIRONMENT"),
getenv("VESPA_REGION"));
fclose(fp);
rename(tmpPath.c_str(), path.c_str());
}
}
vespalib::string clientCert = clientCertFile();
vespalib::string clientKey = clientKeyFile();
if (!clientCert.empty() && !clientKey.empty()) {
vespalib::string certPath = cfFilePath(config.splunkHome, "clientcert.pem");
tmpPath = certPath + ".new";
fp = fopen(tmpPath.c_str(), "w");
appendFile(fp, clientCert);
appendFile(fp, clientKey);
appendFile(fp, "/etc/ssl/certs/ca-bundle.crt");
fclose(fp);
rename(tmpPath.c_str(), certPath.c_str());
path = cfFilePath(config.splunkHome, "outputs.conf");
tmpPath = path + ".new";
fp = fopen(tmpPath.c_str(), "w");
if (fp != NULL) {
fprintf(fp, "[tcpout]\n");
fprintf(fp, "clientCert = %s\n", certPath.c_str());
fclose(fp);
rename(tmpPath.c_str(), path.c_str());
}
}
if (config.clientName.size() == 0 ||
config.deploymentServer.size() == 0)
{
_childHandler.stopChild();
} else {
_childHandler.startChild(config.splunkHome);
}
}
void SplunkStarter::stop() {
_childHandler.stopChild();
}
|