walkero
/
nghttp2
1
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

nghttpx: Use std::unique_ptr<char[]> instead of raw char pointer

This commit is contained in:
Tatsuhiro Tsujikawa 2015-07-20 21:37:23 +09:00
parent ca3444c34c
commit dd8ce1e9d2
6 changed files with 90 additions and 94 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

74
src/shrpx_config.cc
View File

@ -261,35 +261,37 @@ std::string read_passwd_from_file(const char *filename) {
return line;
}
std::vector<char *> parse_config_str_list(const char *s, char delim) {
std::vector<Range<const char *>> split_config_str_list(const char *s,
char delim) {
size_t len = 1;
for (const char *first = s, *p = nullptr; (p = strchr(first, delim));
++len, first = p + 1)
auto last = s + strlen(s);
for (const char *first = s, *d = nullptr;
(d = std::find(first, last, delim)) != last; ++len, first = d + 1)
;
auto list = std::vector<char *>(len);
auto first = strdup(s);
auto list = std::vector<Range<const char *>>(len);
len = 0;
for (;;) {
auto p = strchr(first, delim);
if (p == nullptr) {
for (auto first = s;; ++len) {
auto stop = std::find(first, last, delim);
list[len] = {first, stop};
if (stop == last) {
break;
}
list[len++] = first;
*p = '\0';
first = p + 1;
first = stop + 1;
}
list[len++] = first;
return list;
}
void clear_config_str_list(std::vector<char *> &list) {
if (list.empty()) {
return;
std::vector<std::unique_ptr<char[]>> parse_config_str_list(const char *s,
char delim) {
auto ranges = split_config_str_list(s, delim);
auto res = std::vector<std::unique_ptr<char[]>>();
res.reserve(ranges.size());
for (const auto &range : ranges) {
res.push_back(strcopy(range.first, range.second));
}
free(list[0]);
list.clear();
return res;
}
std::pair<std::string, std::string> parse_header(const char *optarg) {
@ -590,22 +592,21 @@ namespace {
void parse_mapping(const DownstreamAddr &addr, const char *src) {
// This returns at least 1 element (it could be empty string). We
// will append '/' to all patterns, so it becomes catch-all pattern.
auto mapping = parse_config_str_list(src, ':');
auto mapping = split_config_str_list(src, ':');
assert(!mapping.empty());
for (auto raw_pattern : mapping) {
for (const auto &raw_pattern : mapping) {
auto done = false;
std::string pattern;
auto slash = strchr(raw_pattern, '/');
if (slash == nullptr) {
auto slash = std::find(raw_pattern.first, raw_pattern.second, '/');
if (slash == raw_pattern.second) {
// This effectively makes empty pattern to "/".
pattern = raw_pattern;
pattern.assign(raw_pattern.first, raw_pattern.second);
util::inp_strlower(pattern);
pattern += "/";
} else {
pattern.assign(raw_pattern, slash);
pattern.assign(raw_pattern.first, slash);
util::inp_strlower(pattern);
pattern +=
http2::normalize_path(slash, raw_pattern + strlen(raw_pattern));
pattern += http2::normalize_path(slash, raw_pattern.second);
}
for (auto &g : mod_config()->downstream_addr_groups) {
if (g.pattern == pattern) {
@ -621,7 +622,6 @@ void parse_mapping(const DownstreamAddr &addr, const char *src) {
g.addrs.push_back(addr);
mod_config()->downstream_addr_groups.push_back(std::move(g));
}
clear_config_str_list(mapping);
}
} // namespace
@ -1629,12 +1629,10 @@ int parse_config(const char *opt, const char *optarg,
LOG(WARN) << opt << ": not implemented yet";
return parse_uint_with_unit(&mod_config()->worker_write_burst, opt, optarg);
case SHRPX_OPTID_NPN_LIST:
clear_config_str_list(mod_config()->npn_list);
mod_config()->npn_list = parse_config_str_list(optarg);
return 0;
case SHRPX_OPTID_TLS_PROTO_LIST:
clear_config_str_list(mod_config()->tls_proto_list);
mod_config()->tls_proto_list = parse_config_str_list(optarg);
return 0;
@ -1689,13 +1687,13 @@ int parse_config(const char *opt, const char *optarg,
int port;
if (parse_uint(&port, opt, tokens[1]) != 0) {
if (parse_uint(&port, opt, tokens[1].get()) != 0) {
return -1;
}
if (port < 1 ||
port > static_cast<int>(std::numeric_limits<uint16_t>::max())) {
LOG(ERROR) << opt << ": port is invalid: " << tokens[1];
LOG(ERROR) << opt << ": port is invalid: " << tokens[1].get();
return -1;
}
@ -1703,16 +1701,16 @@ int parse_config(const char *opt, const char *optarg,
altsvc.port = port;
altsvc.protocol_id = tokens[0];
altsvc.protocol_id_len = strlen(altsvc.protocol_id);
altsvc.protocol_id = std::move(tokens[0]);
altsvc.protocol_id_len = strlen(altsvc.protocol_id.get());
if (tokens.size() > 2) {
altsvc.host = tokens[2];
altsvc.host_len = strlen(altsvc.host);
altsvc.host = std::move(tokens[2]);
altsvc.host_len = strlen(altsvc.host.get());
if (tokens.size() > 3) {
altsvc.origin = tokens[3];
altsvc.origin_len = strlen(altsvc.origin);
altsvc.origin = std::move(tokens[3]);
altsvc.origin_len = strlen(altsvc.origin.get());
}
}

39
src/shrpx_config.h
View File

@ -184,13 +184,11 @@ union sockaddr_union {
enum shrpx_proto { PROTO_HTTP2, PROTO_HTTP };
struct AltSvc {
AltSvc()
: protocol_id(nullptr), host(nullptr), origin(nullptr),
protocol_id_len(0), host_len(0), origin_len(0), port(0) {}
AltSvc() : protocol_id_len(0), host_len(0), origin_len(0), port(0) {}
char *protocol_id;
char *host;
char *origin;
std::unique_ptr<char[]> protocol_id;
std::unique_ptr<char[]> host;
std::unique_ptr<char[]> origin;
size_t protocol_id_len;
size_t host_len;
@ -291,10 +289,10 @@ struct Config {
// list of supported NPN/ALPN protocol strings in the order of
// preference. The each element of this list is a NULL-terminated
// string.
std::vector<char *> npn_list;
std::vector<std::unique_ptr<char[]>> npn_list;
// list of supported SSL/TLS protocol strings. The each element of
// this list is a NULL-terminated string.
std::vector<char *> tls_proto_list;
std::vector<std::unique_ptr<char[]>> tls_proto_list;
// Path to file containing CA certificate solely used for client
// certificate validation
std::unique_ptr<char[]> verify_client_cacert;
@ -410,19 +408,20 @@ int load_config(const char *filename, std::set<std::string> &include_set);
// Read passwd from |filename|
std::string read_passwd_from_file(const char *filename);
// Parses delimited strings in |s| and returns the array of pointers,
// each element points to the each substring in |s|. The delimiter is
// given by |delim. The |s| must be comma delimited list of strings.
// The strings must be delimited by a single comma and any white
// spaces around it are treated as a part of protocol strings. This
// function copies |s| and first element in the return value points to
// it. It is caller's responsibility to deallocate its memory.
std::vector<char *> parse_config_str_list(const char *s, char delim = ',');
template <typename T> using Range = std::pair<T, T>;
// Clears all elements of |list|, which is returned by
// parse_config_str_list(). If list is not empty, list[0] is freed by
// free(2). After this call, list.empty() must be true.
void clear_config_str_list(std::vector<char *> &list);
// Parses delimited strings in |s| and returns the array of substring,
// delimited by |delim|. The any white spaces around substring are
// treated as a part of substring.
std::vector<std::unique_ptr<char[]>> parse_config_str_list(const char *s,
char delim = ',');
// Parses delimited strings in |s| and returns the array of pointers,
// each element points to the beginning and one beyond last of
// substring in |s|. The delimiter is given by |delim|. The any
// white spaces around substring are treated as a part of substring.
std::vector<Range<const char *>> split_config_str_list(const char *s,
char delim);
// Parses header field in |optarg|. We expect header field is formed
// like "NAME: VALUE". We require that NAME is non empty string. ":"

27
src/shrpx_config_test.cc
View File

@ -39,34 +39,29 @@ namespace shrpx {
void test_shrpx_config_parse_config_str_list(void) {
auto res = parse_config_str_list("a");
CU_ASSERT(1 == res.size());
CU_ASSERT(0 == strcmp("a", res[0]));
clear_config_str_list(res);
CU_ASSERT(0 == strcmp("a", res[0].get()));
res = parse_config_str_list("a,");
CU_ASSERT(2 == res.size());
CU_ASSERT(0 == strcmp("a", res[0]));
CU_ASSERT(0 == strcmp("", res[1]));
clear_config_str_list(res);
CU_ASSERT(0 == strcmp("a", res[0].get()));
CU_ASSERT(0 == strcmp("", res[1].get()));
res = parse_config_str_list(":a::", ':');
CU_ASSERT(4 == res.size());
CU_ASSERT(0 == strcmp("", res[0]));
CU_ASSERT(0 == strcmp("a", res[1]));
CU_ASSERT(0 == strcmp("", res[2]));
CU_ASSERT(0 == strcmp("", res[3]));
clear_config_str_list(res);
CU_ASSERT(0 == strcmp("", res[0].get()));
CU_ASSERT(0 == strcmp("a", res[1].get()));
CU_ASSERT(0 == strcmp("", res[2].get()));
CU_ASSERT(0 == strcmp("", res[3].get()));
res = parse_config_str_list("");
CU_ASSERT(1 == res.size());
CU_ASSERT(0 == strcmp("", res[0]));
clear_config_str_list(res);
CU_ASSERT(0 == strcmp("", res[0].get()));
res = parse_config_str_list("alpha,bravo,charlie");
CU_ASSERT(3 == res.size());
CU_ASSERT(0 == strcmp("alpha", res[0]));
CU_ASSERT(0 == strcmp("bravo", res[1]));
CU_ASSERT(0 == strcmp("charlie", res[2]));
clear_config_str_list(res);
CU_ASSERT(0 == strcmp("alpha", res[0].get()));
CU_ASSERT(0 == strcmp("bravo", res[1].get()));
CU_ASSERT(0 == strcmp("charlie", res[2].get()));
}
void test_shrpx_config_parse_header(void) {

5
src/shrpx_https_upstream.cc
View File

@ -845,9 +845,10 @@ int HttpsUpstream::on_downstream_header_complete(Downstream *downstream) {
hdrs += "Alt-Svc: ";
for (auto &altsvc : get_config()->altsvcs) {
hdrs += util::percent_encode_token(altsvc.protocol_id);
hdrs += util::percent_encode_token(altsvc.protocol_id.get());
hdrs += "=\"";
hdrs += util::quote_string(std::string(altsvc.host, altsvc.host_len));
hdrs +=
util::quote_string(std::string(altsvc.host.get(), altsvc.host_len));
hdrs += ":";
hdrs += util::utos(altsvc.port);
hdrs += "\", ";

31
src/shrpx_ssl.cc
View File

@ -86,11 +86,12 @@ int verify_callback(int preverify_ok, X509_STORE_CTX *ctx) {
}
} // namespace
std::vector<unsigned char> set_alpn_prefs(const std::vector<char *> &protos) {
std::vector<unsigned char>
set_alpn_prefs(const std::vector<std::unique_ptr<char[]>> &protos) {
size_t len = 0;
for (auto proto : protos) {
auto n = strlen(proto);
for (auto &proto : protos) {
auto n = strlen(proto.get());
if (n > 255) {
LOG(FATAL) << "Too long ALPN identifier: " << n;
@ -108,11 +109,11 @@ std::vector<unsigned char> set_alpn_prefs(const std::vector<char *> &protos) {
auto out = std::vector<unsigned char>(len);
auto ptr = out.data();
for (auto proto : protos) {
auto proto_len = strlen(proto);
for (auto &proto : protos) {
auto proto_len = strlen(proto.get());
*ptr++ = proto_len;
memcpy(ptr, proto, proto_len);
memcpy(ptr, proto.get(), proto_len);
ptr += proto_len;
}
@ -281,16 +282,15 @@ int alpn_select_proto_cb(SSL *ssl, const unsigned char **out,
// We assume that get_config()->npn_list contains ALPN protocol
// identifier sorted by preference order. So we just break when we
// found the first overlap.
for (auto target_proto_id : get_config()->npn_list) {
auto target_proto_len =
strlen(reinterpret_cast<const char *>(target_proto_id));
for (auto &target_proto_id : get_config()->npn_list) {
auto target_proto_len = strlen(target_proto_id.get());
for (auto p = in, end = in + inlen; p < end;) {
auto proto_id = p + 1;
auto proto_len = *p;
if (proto_id + proto_len <= end && target_proto_len == proto_len &&
memcmp(target_proto_id, proto_id, proto_len) == 0) {
memcmp(target_proto_id.get(), proto_id, proto_len) == 0) {
*out = reinterpret_cast<const unsigned char *>(proto_id);
*outlen = proto_len;
@ -314,13 +314,14 @@ constexpr long int tls_masks[] = {SSL_OP_NO_TLSv1_2, SSL_OP_NO_TLSv1_1,
SSL_OP_NO_TLSv1};
} // namespace
long int create_tls_proto_mask(const std::vector<char *> &tls_proto_list) {
long int create_tls_proto_mask(
const std::vector<std::unique_ptr<char[]>> &tls_proto_list) {
long int res = 0;
for (size_t i = 0; i < tls_namelen; ++i) {
size_t j;
for (j = 0; j < tls_proto_list.size(); ++j) {
if (util::strieq(tls_names[i], tls_proto_list[j])) {
if (util::strieq(tls_names[i], tls_proto_list[j].get())) {
break;
}
}
@ -949,10 +950,10 @@ int cert_lookup_tree_add_cert_from_file(CertLookupTree *lt, SSL_CTX *ssl_ctx,
return 0;
}
bool in_proto_list(const std::vector<char *> &protos,
bool in_proto_list(const std::vector<std::unique_ptr<char[]>> &protos,
const unsigned char *needle, size_t len) {
for (auto proto : protos) {
if (strlen(proto) == len && memcmp(proto, needle, len) == 0) {
for (auto &proto : protos) {
if (strlen(proto.get()) == len && memcmp(proto.get(), needle, len) == 0) {
return true;
}
}

8
src/shrpx_ssl.h
View File

@ -140,7 +140,7 @@ int cert_lookup_tree_add_cert_from_file(CertLookupTree *lt, SSL_CTX *ssl_ctx,
// Returns true if |needle| which has |len| bytes is included in the
// protocol list |protos|.
bool in_proto_list(const std::vector<char *> &protos,
bool in_proto_list(const std::vector<std::unique_ptr<char[]>> &protos,
const unsigned char *needle, size_t len);
// Returns true if security requirement for HTTP/2 is fulfilled.
@ -149,9 +149,11 @@ bool check_http2_requirement(SSL *ssl);
// Returns SSL/TLS option mask to disable SSL/TLS protocol version not
// included in |tls_proto_list|. The returned mask can be directly
// passed to SSL_CTX_set_options().
long int create_tls_proto_mask(const std::vector<char *> &tls_proto_list);
long int create_tls_proto_mask(
const std::vector<std::unique_ptr<char[]>> &tls_proto_list);
std::vector<unsigned char> set_alpn_prefs(const std::vector<char *> &protos);
std::vector<unsigned char>
set_alpn_prefs(const std::vector<std::unique_ptr<char[]>> &protos);
// Setups server side SSL_CTX. This function inspects get_config()
// and if upstream_no_tls is true, returns nullptr. Otherwise