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

h2load: Use Headers instead of std::vector<std::pair<>> to store custom headers

This commit is contained in:
Tatsuhiro Tsujikawa 2014-10-22 22:53:54 +09:00
parent a9ecdca08a
commit 6933e0ef54
2 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/h2load.cc
View File

@ -833,7 +833,7 @@ int main(int argc, char **argv)
// Note that there is no processing currently to handle multiple
// message-header fields with the same field name
config.custom_headers.emplace_back(header, value);
util::inp_strlower(config.custom_headers.back().first);
util::inp_strlower(config.custom_headers.back().name);
break;
}
case 'i': {
@ -982,18 +982,18 @@ int main(int argc, char **argv)
};
for(auto& kv : config.custom_headers) {
if(std::find(std::begin(override_hdrs), std::end(override_hdrs), kv.first)
if(std::find(std::begin(override_hdrs), std::end(override_hdrs), kv.name)
!= std::end(override_hdrs)) {
// override header
for(auto& nv : shared_nva) {
if((nv.name == ":authority" && kv.first == ":host")
|| (nv.name == kv.first) ) {
nv.value = kv.second;
if((nv.name == ":authority" && kv.name == ":host")
|| (nv.name == kv.name) ) {
nv.value = kv.value;
}
}
} else {
// add additional headers
shared_nva.emplace_back(kv.first, kv.second);
shared_nva.push_back(kv);
}
}

4
src/h2load.h
View File

@ -43,6 +43,8 @@
#include <openssl/ssl.h>
#include "http2.h"
namespace h2load {
class Session;
@ -50,7 +52,7 @@ class Session;
struct Config {
std::vector<std::vector<nghttp2_nv>> nva;
std::vector<std::vector<const char*>> nv;
std::vector<std::pair<std::string, std::string>> custom_headers;
nghttp2::Headers custom_headers;
std::string scheme;
std::string host;
std::string ifile;