htp_status_t htp_tx_res_set_header(htp_tx_t *tx, const char *name, size_t name_len,
const char *value, size_t value_len, enum htp_alloc_strategy_t alloc) {
if ((tx == NULL) || (name == NULL) || (value == NULL)) return HTP_ERROR;
htp_header_t *h = calloc(1, sizeof (htp_header_t));
if (h == NULL) return HTP_ERROR;
h->name = copy_or_wrap_mem(name, name_len, alloc);
if (h->name == NULL) {
free(h);
return HTP_ERROR;
}
h->value = copy_or_wrap_mem(value, value_len, alloc);
if (h->value == NULL) {
bstr_free(h->name);
free(h);
return HTP_ERROR;
}
if (htp_table_add(tx->response_headers, h->name, h) != HTP_OK) {
bstr_free(h->name);
bstr_free(h->value);
free(h);
return HTP_ERROR;
}
return HTP_OK;
}
htp_status_t htp_tx_req_set_protocol(htp_tx_t *tx, const char *protocol, size_t protocol_len, enum htp_alloc_strategy_t alloc) {
if ((tx == NULL) || (protocol == NULL)) return HTP_ERROR;
tx->request_protocol = copy_or_wrap_mem(protocol, protocol_len, alloc);
if (tx->request_protocol == NULL) return HTP_ERROR;
return HTP_OK;
}
htp_status_t htp_tx_req_set_uri(htp_tx_t *tx, const char *uri, size_t uri_len, enum htp_alloc_strategy_t alloc) {
if ((tx == NULL) || (uri == NULL)) return HTP_ERROR;
tx->request_uri = copy_or_wrap_mem(uri, uri_len, alloc);
if (tx->request_uri == NULL) return HTP_ERROR;
return HTP_OK;
}
htp_status_t htp_tx_req_set_method(htp_tx_t *tx, const char *method, size_t method_len, enum htp_alloc_strategy_t alloc) {
if ((tx == NULL) || (method == NULL)) return HTP_ERROR;
tx->request_method = copy_or_wrap_mem(method, method_len, alloc);
if (tx->request_method == NULL) return HTP_ERROR;
return HTP_OK;
}
htp_status_t htp_tx_req_set_query_string(htp_tx_t *tx, const char *qs, size_t qs_len, enum htp_alloc_strategy_t alloc) {
if ((tx->parsed_uri == NULL) || (qs == NULL)) return HTP_ERROR;
tx->parsed_uri->query = copy_or_wrap_mem(qs, qs_len, alloc);
if (tx->parsed_uri->query == NULL) return HTP_ERROR;
return HTP_OK;
}
htp_status_t htp_tx_res_set_status_line(htp_tx_t *tx, const char *line, size_t line_len, enum htp_alloc_strategy_t alloc) {
if ((tx == NULL) || (line == NULL) || (line_len == 0)) return HTP_ERROR;
tx->response_line = copy_or_wrap_mem(line, line_len, alloc);
if (tx->response_line == NULL) return HTP_ERROR;
if (tx->connp->cfg->parse_response_line(tx->connp) != HTP_OK) return HTP_ERROR;
return HTP_OK;
}
htp_status_t htp_tx_res_set_status_message(htp_tx_t *tx, const char *msg, size_t msg_len, enum htp_alloc_strategy_t alloc) {
if ((tx == NULL) || (msg == NULL)) return HTP_ERROR;
if (tx->response_message != NULL) {
bstr_free(tx->response_message);
}
tx->response_message = copy_or_wrap_mem(msg, msg_len, alloc);
if (tx->response_message == NULL) return HTP_ERROR;
return HTP_OK;
}
