TEST(BstrTest, AddCNoex) {
bstr *p1;
bstr *p2;
p1 = bstr_alloc(10);
p1 = bstr_add_c(p1, "12345");
p2 = bstr_add_c_noex(p1,"abcdefghijk");
EXPECT_EQ(p1,p2);
EXPECT_EQ(0,bstr_cmp_c(p2,"12345abcde"));
bstr_free(p1);
}
/**
* \brief Generates the normalized uri.
*
* Libhtp doesn't recreate the whole normalized uri and save it.
* That duty has now been passed to us. A lot of this code has been
* copied from libhtp.
*
* Keep an eye out on the tx->parsed_uri struct and how the parameters
* in it are generated, just in case some modifications are made to
* them in the future.
*
* \param uri_include_all boolean to indicate if scheme, username/password,
hostname and port should be part of the buffer
*/
bstr *SCHTPGenerateNormalizedUri(htp_tx_t *tx, htp_uri_t *uri, int uri_include_all)
{
if (uri == NULL)
return NULL;
// On the first pass determine the length of the final string
size_t len = 0;
if (uri_include_all) {
if (uri->scheme != NULL) {
len += bstr_len(uri->scheme);
len += 3; // "://"
}
if ((uri->username != NULL) || (uri->password != NULL)) {
if (uri->username != NULL) {
len += bstr_len(uri->username);
}
len += 1; // ":"
if (uri->password != NULL) {
len += bstr_len(uri->password);
}
len += 1; // "@"
}
if (uri->hostname != NULL) {
len += bstr_len(uri->hostname);
}
if (uri->port != NULL) {
len += 1; // ":"
len += bstr_len(uri->port);
}
}
if (uri->path != NULL) {
len += bstr_len(uri->path);
}
if (uri->query != NULL) {
len += 1; // "?"
len += bstr_len(uri->query);
}
if (uri->fragment != NULL) {
len += 1; // "#"
len += bstr_len(uri->fragment);
}
// On the second pass construct the string
/* FIXME in memcap */
bstr *r = bstr_alloc(len);
if (r == NULL) {
return NULL;
}
if (uri_include_all) {
if (uri->scheme != NULL) {
bstr_add_noex(r, uri->scheme);
bstr_add_c_noex(r, "://");
}
if ((uri->username != NULL) || (uri->password != NULL)) {
if (uri->username != NULL) {
bstr_add_noex(r, uri->username);
}
bstr_add_c_noex(r, ":");
if (uri->password != NULL) {
bstr_add_noex(r, uri->password);
}
bstr_add_c_noex(r, "@");
}
if (uri->hostname != NULL) {
bstr_add_noex(r, uri->hostname);
}
if (uri->port != NULL) {
bstr_add_c_noex(r, ":");
bstr_add_noex(r, uri->port);
}
}
if (uri->path != NULL) {
bstr_add_noex(r, uri->path);
}
if (uri->query != NULL) {
bstr *query = bstr_dup(uri->query);
if (query) {
uint64_t flags = 0;
htp_urldecode_inplace(tx->cfg, HTP_DECODER_URLENCODED, query, &flags);
bstr_add_c_noex(r, "?");
bstr_add_noex(r, query);
bstr_free(query);
}
}
if (uri->fragment != NULL) {
bstr_add_c_noex(r, "#");
bstr_add_noex(r, uri->fragment);
}
return r;
}
