https_client.c 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864
  1. // SPDX-License-Identifier: GPL-3.0-or-later
  2. #include "libnetdata/libnetdata.h"
  3. #include "https_client.h"
  4. #include "mqtt_websockets/c-rbuf/include/ringbuffer.h"
  5. #include "aclk_util.h"
  6. #include "daemon/global_statistics.h"
  7. #define DEFAULT_CHUNKED_RESPONSE_BUFFER_SIZE (4096)
  8. enum http_parse_state {
  9. HTTP_PARSE_INITIAL = 0,
  10. HTTP_PARSE_HEADERS,
  11. HTTP_PARSE_CONTENT
  12. };
  13. static const char *http_req_type_to_str(http_req_type_t req) {
  14. switch (req) {
  15. case HTTP_REQ_GET:
  16. return "GET";
  17. case HTTP_REQ_POST:
  18. return "POST";
  19. case HTTP_REQ_CONNECT:
  20. return "CONNECT";
  21. default:
  22. return "unknown";
  23. }
  24. }
  25. #define TRANSFER_ENCODING_CHUNKED (-2)
  26. typedef struct {
  27. enum http_parse_state state;
  28. int content_length;
  29. int http_code;
  30. // for chunked data only
  31. char *chunked_response;
  32. size_t chunked_response_size;
  33. size_t chunked_response_written;
  34. enum chunked_content_state {
  35. CHUNKED_CONTENT_CHUNK_SIZE = 0,
  36. CHUNKED_CONTENT_CHUNK_DATA,
  37. CHUNKED_CONTENT_CHUNK_END_CRLF,
  38. CHUNKED_CONTENT_FINAL_CRLF
  39. } chunked_content_state;
  40. size_t chunk_size;
  41. size_t chunk_got;
  42. } http_parse_ctx;
  43. #define HTTP_PARSE_CTX_INITIALIZER { .state = HTTP_PARSE_INITIAL, .content_length = -1, .http_code = 0 }
  44. static inline void http_parse_ctx_clear(http_parse_ctx *ctx) {
  45. ctx->state = HTTP_PARSE_INITIAL;
  46. ctx->content_length = -1;
  47. ctx->http_code = 0;
  48. }
  49. #define POLL_TO_MS 100
  50. #define NEED_MORE_DATA 0
  51. #define PARSE_SUCCESS 1
  52. #define PARSE_ERROR -1
  53. #define HTTP_LINE_TERM "\x0D\x0A"
  54. #define RESP_PROTO "HTTP/1.1 "
  55. #define HTTP_KEYVAL_SEPARATOR ": "
  56. #define HTTP_HDR_BUFFER_SIZE 1024
  57. #define PORT_STR_MAX_BYTES 12
  58. static int process_http_hdr(http_parse_ctx *parse_ctx, const char *key, const char *val)
  59. {
  60. // currently we care only about specific headers
  61. // we can skip the rest
  62. if (!strcmp("content-length", key)) {
  63. if (parse_ctx->content_length == TRANSFER_ENCODING_CHUNKED) {
  64. netdata_log_error("Content-length and transfer-encoding: chunked headers are mutually exclusive");
  65. return 1;
  66. }
  67. if (parse_ctx->content_length != -1) {
  68. netdata_log_error("Duplicate content-length header");
  69. return 1;
  70. }
  71. parse_ctx->content_length = atoi(val);
  72. if (parse_ctx->content_length < 0) {
  73. netdata_log_error("Invalid content-length %d", parse_ctx->content_length);
  74. return 1;
  75. }
  76. return 0;
  77. }
  78. if (!strcmp("transfer-encoding", key)) {
  79. if (!strcmp("chunked", val)) {
  80. if (parse_ctx->content_length != -1) {
  81. netdata_log_error("Content-length and transfer-encoding: chunked headers are mutually exclusive");
  82. return 1;
  83. }
  84. parse_ctx->content_length = TRANSFER_ENCODING_CHUNKED;
  85. }
  86. return 0;
  87. }
  88. return 0;
  89. }
  90. static int parse_http_hdr(rbuf_t buf, http_parse_ctx *parse_ctx)
  91. {
  92. int idx, idx_end;
  93. char buf_key[HTTP_HDR_BUFFER_SIZE];
  94. char buf_val[HTTP_HDR_BUFFER_SIZE];
  95. char *ptr = buf_key;
  96. if (!rbuf_find_bytes(buf, HTTP_LINE_TERM, strlen(HTTP_LINE_TERM), &idx_end)) {
  97. netdata_log_error("CRLF expected");
  98. return 1;
  99. }
  100. char *separator = rbuf_find_bytes(buf, HTTP_KEYVAL_SEPARATOR, strlen(HTTP_KEYVAL_SEPARATOR), &idx);
  101. if (!separator) {
  102. netdata_log_error("Missing Key/Value separator");
  103. return 1;
  104. }
  105. if (idx >= HTTP_HDR_BUFFER_SIZE) {
  106. netdata_log_error("Key name is too long");
  107. return 1;
  108. }
  109. rbuf_pop(buf, buf_key, idx);
  110. buf_key[idx] = 0;
  111. rbuf_bump_tail(buf, strlen(HTTP_KEYVAL_SEPARATOR));
  112. idx_end -= strlen(HTTP_KEYVAL_SEPARATOR) + idx;
  113. if (idx_end >= HTTP_HDR_BUFFER_SIZE) {
  114. netdata_log_error("Value of key \"%s\" too long", buf_key);
  115. return 1;
  116. }
  117. rbuf_pop(buf, buf_val, idx_end);
  118. buf_val[idx_end] = 0;
  119. for (ptr = buf_key; *ptr; ptr++)
  120. *ptr = tolower(*ptr);
  121. if (process_http_hdr(parse_ctx, buf_key, buf_val))
  122. return 1;
  123. return 0;
  124. }
  125. static inline void chunked_response_buffer_grow_by(http_parse_ctx *parse_ctx, size_t size)
  126. {
  127. if (unlikely(parse_ctx->chunked_response_size == 0)) {
  128. parse_ctx->chunked_response = mallocz(size);
  129. parse_ctx->chunked_response_size = size;
  130. return;
  131. }
  132. parse_ctx->chunked_response = reallocz((void *)parse_ctx->chunked_response, parse_ctx->chunked_response_size + size);
  133. parse_ctx->chunked_response_size += size;
  134. }
  135. static int process_chunked_content(rbuf_t buf, http_parse_ctx *parse_ctx)
  136. {
  137. int idx;
  138. size_t bytes_to_copy;
  139. do {
  140. switch (parse_ctx->chunked_content_state) {
  141. case CHUNKED_CONTENT_CHUNK_SIZE:
  142. if (!rbuf_find_bytes(buf, HTTP_LINE_TERM, strlen(HTTP_LINE_TERM), &idx)) {
  143. if (rbuf_bytes_available(buf) >= rbuf_get_capacity(buf))
  144. return PARSE_ERROR;
  145. return NEED_MORE_DATA;
  146. }
  147. if (idx == 0) {
  148. parse_ctx->chunked_content_state = CHUNKED_CONTENT_FINAL_CRLF;
  149. continue;
  150. }
  151. if (idx >= HTTP_HDR_BUFFER_SIZE) {
  152. netdata_log_error("Chunk size is too long");
  153. return PARSE_ERROR;
  154. }
  155. char buf_size[HTTP_HDR_BUFFER_SIZE];
  156. rbuf_pop(buf, buf_size, idx);
  157. buf_size[idx] = 0;
  158. long chunk_size = strtol(buf_size, NULL, 16);
  159. if (chunk_size < 0 || chunk_size == LONG_MAX) {
  160. netdata_log_error("Chunk size out of range");
  161. return PARSE_ERROR;
  162. }
  163. parse_ctx->chunk_size = chunk_size;
  164. if (parse_ctx->chunk_size == 0) {
  165. if (errno == EINVAL) {
  166. netdata_log_error("Invalid chunk size");
  167. return PARSE_ERROR;
  168. }
  169. parse_ctx->chunked_content_state = CHUNKED_CONTENT_CHUNK_END_CRLF;
  170. continue;
  171. }
  172. parse_ctx->chunk_got = 0;
  173. chunked_response_buffer_grow_by(parse_ctx, parse_ctx->chunk_size);
  174. rbuf_bump_tail(buf, strlen(HTTP_LINE_TERM));
  175. parse_ctx->chunked_content_state = CHUNKED_CONTENT_CHUNK_DATA;
  176. // fallthrough
  177. case CHUNKED_CONTENT_CHUNK_DATA:
  178. if (!(bytes_to_copy = rbuf_bytes_available(buf)))
  179. return NEED_MORE_DATA;
  180. if (bytes_to_copy > parse_ctx->chunk_size - parse_ctx->chunk_got)
  181. bytes_to_copy = parse_ctx->chunk_size - parse_ctx->chunk_got;
  182. rbuf_pop(buf, parse_ctx->chunked_response + parse_ctx->chunked_response_written, bytes_to_copy);
  183. parse_ctx->chunk_got += bytes_to_copy;
  184. parse_ctx->chunked_response_written += bytes_to_copy;
  185. if (parse_ctx->chunk_got != parse_ctx->chunk_size)
  186. continue;
  187. parse_ctx->chunked_content_state = CHUNKED_CONTENT_CHUNK_END_CRLF;
  188. // fallthrough
  189. case CHUNKED_CONTENT_FINAL_CRLF:
  190. case CHUNKED_CONTENT_CHUNK_END_CRLF:
  191. if (rbuf_bytes_available(buf) < strlen(HTTP_LINE_TERM))
  192. return NEED_MORE_DATA;
  193. char buf_crlf[strlen(HTTP_LINE_TERM)];
  194. rbuf_pop(buf, buf_crlf, strlen(HTTP_LINE_TERM));
  195. if (memcmp(buf_crlf, HTTP_LINE_TERM, strlen(HTTP_LINE_TERM))) {
  196. netdata_log_error("CRLF expected");
  197. return PARSE_ERROR;
  198. }
  199. if (parse_ctx->chunked_content_state == CHUNKED_CONTENT_FINAL_CRLF) {
  200. if (parse_ctx->chunked_response_size != parse_ctx->chunked_response_written)
  201. netdata_log_error("Chunked response size mismatch");
  202. chunked_response_buffer_grow_by(parse_ctx, 1);
  203. parse_ctx->chunked_response[parse_ctx->chunked_response_written] = 0;
  204. return PARSE_SUCCESS;
  205. }
  206. if (parse_ctx->chunk_size == 0) {
  207. parse_ctx->chunked_content_state = CHUNKED_CONTENT_FINAL_CRLF;
  208. continue;
  209. }
  210. parse_ctx->chunked_content_state = CHUNKED_CONTENT_CHUNK_SIZE;
  211. continue;
  212. }
  213. } while(1);
  214. }
  215. static int parse_http_response(rbuf_t buf, http_parse_ctx *parse_ctx)
  216. {
  217. int idx;
  218. char rc[4];
  219. do {
  220. if (parse_ctx->state != HTTP_PARSE_CONTENT && !rbuf_find_bytes(buf, HTTP_LINE_TERM, strlen(HTTP_LINE_TERM), &idx))
  221. return NEED_MORE_DATA;
  222. switch (parse_ctx->state) {
  223. case HTTP_PARSE_INITIAL:
  224. if (rbuf_memcmp_n(buf, RESP_PROTO, strlen(RESP_PROTO))) {
  225. netdata_log_error("Expected response to start with \"%s\"", RESP_PROTO);
  226. return PARSE_ERROR;
  227. }
  228. rbuf_bump_tail(buf, strlen(RESP_PROTO));
  229. if (rbuf_pop(buf, rc, 4) != 4) {
  230. netdata_log_error("Expected HTTP status code");
  231. return PARSE_ERROR;
  232. }
  233. if (rc[3] != ' ') {
  234. netdata_log_error("Expected space after HTTP return code");
  235. return PARSE_ERROR;
  236. }
  237. rc[3] = 0;
  238. parse_ctx->http_code = atoi(rc);
  239. if (parse_ctx->http_code < 100 || parse_ctx->http_code >= 600) {
  240. netdata_log_error("HTTP code not in range 100 to 599");
  241. return PARSE_ERROR;
  242. }
  243. rbuf_find_bytes(buf, HTTP_LINE_TERM, strlen(HTTP_LINE_TERM), &idx);
  244. rbuf_bump_tail(buf, idx + strlen(HTTP_LINE_TERM));
  245. parse_ctx->state = HTTP_PARSE_HEADERS;
  246. break;
  247. case HTTP_PARSE_HEADERS:
  248. if (!idx) {
  249. parse_ctx->state = HTTP_PARSE_CONTENT;
  250. rbuf_bump_tail(buf, strlen(HTTP_LINE_TERM));
  251. break;
  252. }
  253. if (parse_http_hdr(buf, parse_ctx))
  254. return PARSE_ERROR;
  255. rbuf_find_bytes(buf, HTTP_LINE_TERM, strlen(HTTP_LINE_TERM), &idx);
  256. rbuf_bump_tail(buf, idx + strlen(HTTP_LINE_TERM));
  257. break;
  258. case HTTP_PARSE_CONTENT:
  259. // replies like CONNECT etc. do not have content
  260. if (parse_ctx->content_length == TRANSFER_ENCODING_CHUNKED)
  261. return process_chunked_content(buf, parse_ctx);
  262. if (parse_ctx->content_length < 0)
  263. return PARSE_SUCCESS;
  264. if (rbuf_bytes_available(buf) >= (size_t)parse_ctx->content_length)
  265. return PARSE_SUCCESS;
  266. return NEED_MORE_DATA;
  267. }
  268. } while(1);
  269. }
  270. typedef struct https_req_ctx {
  271. https_req_t *request;
  272. int sock;
  273. rbuf_t buf_rx;
  274. struct pollfd poll_fd;
  275. SSL_CTX *ssl_ctx;
  276. SSL *ssl;
  277. size_t written;
  278. int self_signed_allowed;
  279. http_parse_ctx parse_ctx;
  280. time_t req_start_time;
  281. } https_req_ctx_t;
  282. static int https_req_check_timedout(https_req_ctx_t *ctx) {
  283. if (now_realtime_sec() > ctx->req_start_time + ctx->request->timeout_s) {
  284. netdata_log_error("request timed out");
  285. return 1;
  286. }
  287. return 0;
  288. }
  289. static char *_ssl_err_tos(int err)
  290. {
  291. switch(err){
  292. case SSL_ERROR_SSL:
  293. return "SSL_ERROR_SSL";
  294. case SSL_ERROR_WANT_READ:
  295. return "SSL_ERROR_WANT_READ";
  296. case SSL_ERROR_WANT_WRITE:
  297. return "SSL_ERROR_WANT_WRITE";
  298. case SSL_ERROR_NONE:
  299. return "SSL_ERROR_NONE";
  300. case SSL_ERROR_ZERO_RETURN:
  301. return "SSL_ERROR_ZERO_RETURN";
  302. case SSL_ERROR_WANT_CONNECT:
  303. return "SSL_ERROR_WANT_CONNECT";
  304. case SSL_ERROR_WANT_ACCEPT:
  305. return "SSL_ERROR_WANT_ACCEPT";
  306. }
  307. return "Unknown!!!";
  308. }
  309. static int socket_write_all(https_req_ctx_t *ctx, char *data, size_t data_len) {
  310. ctx->written = 0;
  311. ctx->poll_fd.events = POLLOUT;
  312. do {
  313. int ret = poll(&ctx->poll_fd, 1, POLL_TO_MS);
  314. if (ret < 0) {
  315. netdata_log_error("poll error");
  316. return 1;
  317. }
  318. if (ret == 0) {
  319. if (https_req_check_timedout(ctx)) {
  320. netdata_log_error("Poll timed out");
  321. return 2;
  322. }
  323. continue;
  324. }
  325. ret = write(ctx->sock, &data[ctx->written], data_len - ctx->written);
  326. if (ret > 0) {
  327. ctx->written += ret;
  328. } else if (errno != EAGAIN && errno != EWOULDBLOCK) {
  329. netdata_log_error("Error writing to socket");
  330. return 3;
  331. }
  332. } while (ctx->written < data_len);
  333. return 0;
  334. }
  335. static int ssl_write_all(https_req_ctx_t *ctx, char *data, size_t data_len) {
  336. ctx->written = 0;
  337. ctx->poll_fd.events |= POLLOUT;
  338. do {
  339. int ret = poll(&ctx->poll_fd, 1, POLL_TO_MS);
  340. if (ret < 0) {
  341. netdata_log_error("poll error");
  342. return 1;
  343. }
  344. if (ret == 0) {
  345. if (https_req_check_timedout(ctx)) {
  346. netdata_log_error("Poll timed out");
  347. return 2;
  348. }
  349. continue;
  350. }
  351. ctx->poll_fd.events = 0;
  352. ret = SSL_write(ctx->ssl, &data[ctx->written], data_len - ctx->written);
  353. if (ret > 0) {
  354. ctx->written += ret;
  355. } else {
  356. ret = SSL_get_error(ctx->ssl, ret);
  357. switch (ret) {
  358. case SSL_ERROR_WANT_READ:
  359. ctx->poll_fd.events |= POLLIN;
  360. break;
  361. case SSL_ERROR_WANT_WRITE:
  362. ctx->poll_fd.events |= POLLOUT;
  363. break;
  364. default:
  365. netdata_log_error("SSL_write Err: %s", _ssl_err_tos(ret));
  366. return 3;
  367. }
  368. }
  369. } while (ctx->written < data_len);
  370. return 0;
  371. }
  372. static inline int https_client_write_all(https_req_ctx_t *ctx, char *data, size_t data_len) {
  373. if (ctx->ssl_ctx)
  374. return ssl_write_all(ctx, data, data_len);
  375. return socket_write_all(ctx, data, data_len);
  376. }
  377. static int read_parse_response(https_req_ctx_t *ctx) {
  378. int ret;
  379. char *ptr;
  380. size_t size;
  381. ctx->poll_fd.events = POLLIN;
  382. do {
  383. ret = poll(&ctx->poll_fd, 1, POLL_TO_MS);
  384. if (ret < 0) {
  385. netdata_log_error("poll error");
  386. return 1;
  387. }
  388. if (ret == 0) {
  389. if (https_req_check_timedout(ctx)) {
  390. netdata_log_error("Poll timed out");
  391. return 2;
  392. }
  393. if (!ctx->ssl_ctx)
  394. continue;
  395. }
  396. ctx->poll_fd.events = 0;
  397. do {
  398. ptr = rbuf_get_linear_insert_range(ctx->buf_rx, &size);
  399. if (ctx->ssl_ctx)
  400. ret = SSL_read(ctx->ssl, ptr, size);
  401. else
  402. ret = read(ctx->sock, ptr, size);
  403. if (ret > 0) {
  404. rbuf_bump_head(ctx->buf_rx, ret);
  405. } else {
  406. if (ctx->ssl_ctx) {
  407. ret = SSL_get_error(ctx->ssl, ret);
  408. switch (ret) {
  409. case SSL_ERROR_WANT_READ:
  410. ctx->poll_fd.events |= POLLIN;
  411. break;
  412. case SSL_ERROR_WANT_WRITE:
  413. ctx->poll_fd.events |= POLLOUT;
  414. break;
  415. default:
  416. netdata_log_error("SSL_read Err: %s", _ssl_err_tos(ret));
  417. return 3;
  418. }
  419. } else {
  420. if (errno != EAGAIN && errno != EWOULDBLOCK) {
  421. netdata_log_error("write error");
  422. return 3;
  423. }
  424. ctx->poll_fd.events |= POLLIN;
  425. }
  426. }
  427. } while (ctx->poll_fd.events == 0 && rbuf_bytes_free(ctx->buf_rx) > 0);
  428. } while (!(ret = parse_http_response(ctx->buf_rx, &ctx->parse_ctx)));
  429. if (ret != PARSE_SUCCESS) {
  430. netdata_log_error("Error parsing HTTP response");
  431. return 1;
  432. }
  433. return 0;
  434. }
  435. #define TX_BUFFER_SIZE 8192
  436. #define RX_BUFFER_SIZE (TX_BUFFER_SIZE*2)
  437. static int handle_http_request(https_req_ctx_t *ctx) {
  438. BUFFER *hdr = buffer_create(TX_BUFFER_SIZE, &netdata_buffers_statistics.buffers_aclk);
  439. int rc = 0;
  440. http_parse_ctx_clear(&ctx->parse_ctx);
  441. // Prepare data to send
  442. switch (ctx->request->request_type) {
  443. case HTTP_REQ_CONNECT:
  444. buffer_strcat(hdr, "CONNECT ");
  445. break;
  446. case HTTP_REQ_GET:
  447. buffer_strcat(hdr, "GET ");
  448. break;
  449. case HTTP_REQ_POST:
  450. buffer_strcat(hdr, "POST ");
  451. break;
  452. default:
  453. netdata_log_error("Unknown HTTPS request type!");
  454. rc = 1;
  455. goto err_exit;
  456. }
  457. if (ctx->request->request_type == HTTP_REQ_CONNECT) {
  458. buffer_strcat(hdr, ctx->request->host);
  459. buffer_sprintf(hdr, ":%d", ctx->request->port);
  460. } else {
  461. buffer_strcat(hdr, ctx->request->url);
  462. }
  463. buffer_strcat(hdr, " HTTP/1.1\x0D\x0A");
  464. //TODO Headers!
  465. if (ctx->request->request_type != HTTP_REQ_CONNECT) {
  466. buffer_sprintf(hdr, "Host: %s\x0D\x0A", ctx->request->host);
  467. }
  468. buffer_strcat(hdr, "User-Agent: Netdata/rocks newhttpclient\x0D\x0A");
  469. if (ctx->request->request_type == HTTP_REQ_POST && ctx->request->payload && ctx->request->payload_size) {
  470. buffer_sprintf(hdr, "Content-Length: %zu\x0D\x0A", ctx->request->payload_size);
  471. }
  472. if (ctx->request->proxy_username) {
  473. size_t creds_plain_len = strlen(ctx->request->proxy_username) + strlen(ctx->request->proxy_password) + 1 /* ':' */;
  474. char *creds_plain = callocz(1, creds_plain_len + 1);
  475. char *ptr = creds_plain;
  476. strcpy(ptr, ctx->request->proxy_username);
  477. ptr += strlen(ctx->request->proxy_username);
  478. *ptr++ = ':';
  479. strcpy(ptr, ctx->request->proxy_password);
  480. int creds_base64_len = (((4 * creds_plain_len / 3) + 3) & ~3);
  481. // OpenSSL encoder puts newline every 64 output bytes
  482. // we remove those but during encoding we need that space in the buffer
  483. creds_base64_len += (1+(creds_base64_len/64)) * strlen("\n");
  484. char *creds_base64 = callocz(1, creds_base64_len + 1);
  485. base64_encode_helper((unsigned char*)creds_base64, &creds_base64_len, (unsigned char*)creds_plain, creds_plain_len);
  486. buffer_sprintf(hdr, "Proxy-Authorization: Basic %s\x0D\x0A", creds_base64);
  487. freez(creds_plain);
  488. }
  489. buffer_strcat(hdr, "\x0D\x0A");
  490. // Send the request
  491. if (https_client_write_all(ctx, hdr->buffer, hdr->len)) {
  492. netdata_log_error("Couldn't write HTTP request header into SSL connection");
  493. rc = 2;
  494. goto err_exit;
  495. }
  496. if (ctx->request->request_type == HTTP_REQ_POST && ctx->request->payload && ctx->request->payload_size) {
  497. if (https_client_write_all(ctx, ctx->request->payload, ctx->request->payload_size)) {
  498. netdata_log_error("Couldn't write payload into SSL connection");
  499. rc = 3;
  500. goto err_exit;
  501. }
  502. }
  503. // Read The Response
  504. if (read_parse_response(ctx)) {
  505. netdata_log_error("Error reading or parsing response from server");
  506. if (ctx->parse_ctx.chunked_response)
  507. freez(ctx->parse_ctx.chunked_response);
  508. rc = 4;
  509. goto err_exit;
  510. }
  511. err_exit:
  512. buffer_free(hdr);
  513. return rc;
  514. }
  515. static int cert_verify_callback(int preverify_ok, X509_STORE_CTX *ctx)
  516. {
  517. X509 *err_cert;
  518. int err, depth;
  519. char *err_str;
  520. if (!preverify_ok) {
  521. err = X509_STORE_CTX_get_error(ctx);
  522. depth = X509_STORE_CTX_get_error_depth(ctx);
  523. err_cert = X509_STORE_CTX_get_current_cert(ctx);
  524. err_str = X509_NAME_oneline(X509_get_subject_name(err_cert), NULL, 0);
  525. netdata_log_error("Cert Chain verify error:num=%d:%s:depth=%d:%s", err,
  526. X509_verify_cert_error_string(err), depth, err_str);
  527. free(err_str);
  528. }
  529. #ifdef ACLK_SSL_ALLOW_SELF_SIGNED
  530. if (!preverify_ok && err == X509_V_ERR_DEPTH_ZERO_SELF_SIGNED_CERT)
  531. {
  532. preverify_ok = 1;
  533. netdata_log_error("Self Signed Certificate Accepted as the agent was built with ACLK_SSL_ALLOW_SELF_SIGNED");
  534. }
  535. #endif
  536. return preverify_ok;
  537. }
  538. int https_request(https_req_t *request, https_req_response_t *response) {
  539. int rc = 1, ret;
  540. char connect_port_str[PORT_STR_MAX_BYTES];
  541. const char *connect_host = request->proxy_host ? request->proxy_host : request->host;
  542. int connect_port = request->proxy_host ? request->proxy_port : request->port;
  543. struct timeval timeout = { .tv_sec = request->timeout_s, .tv_usec = 0 };
  544. https_req_ctx_t *ctx = callocz(1, sizeof(https_req_ctx_t));
  545. ctx->req_start_time = now_realtime_sec();
  546. ctx->buf_rx = rbuf_create(RX_BUFFER_SIZE);
  547. if (!ctx->buf_rx) {
  548. netdata_log_error("Couldn't allocate buffer for RX data");
  549. goto exit_req_ctx;
  550. }
  551. snprintfz(connect_port_str, PORT_STR_MAX_BYTES, "%d", connect_port);
  552. ctx->sock = connect_to_this_ip46(IPPROTO_TCP, SOCK_STREAM, connect_host, 0, connect_port_str, &timeout);
  553. if (ctx->sock < 0) {
  554. netdata_log_error("Error connecting TCP socket to \"%s\"", connect_host);
  555. goto exit_buf_rx;
  556. }
  557. if (fcntl(ctx->sock, F_SETFL, fcntl(ctx->sock, F_GETFL, 0) | O_NONBLOCK) == -1) {
  558. netdata_log_error("Error setting O_NONBLOCK to TCP socket.");
  559. goto exit_sock;
  560. }
  561. ctx->poll_fd.fd = ctx->sock;
  562. // Do the CONNECT if proxy is used
  563. if (request->proxy_host) {
  564. https_req_t req = HTTPS_REQ_T_INITIALIZER;
  565. req.request_type = HTTP_REQ_CONNECT;
  566. req.timeout_s = request->timeout_s;
  567. req.host = request->host;
  568. req.port = request->port;
  569. req.url = request->url;
  570. req.proxy_username = request->proxy_username;
  571. req.proxy_password = request->proxy_password;
  572. ctx->request = &req;
  573. if (handle_http_request(ctx)) {
  574. netdata_log_error("Failed to CONNECT with proxy");
  575. goto exit_sock;
  576. }
  577. if (ctx->parse_ctx.http_code != 200) {
  578. netdata_log_error("Proxy didn't return 200 OK (got %d)", ctx->parse_ctx.http_code);
  579. goto exit_sock;
  580. }
  581. netdata_log_info("Proxy accepted CONNECT upgrade");
  582. }
  583. ctx->request = request;
  584. ctx->ssl_ctx = netdata_ssl_create_client_ctx(0);
  585. if (ctx->ssl_ctx==NULL) {
  586. netdata_log_error("Cannot allocate SSL context");
  587. goto exit_sock;
  588. }
  589. if (!SSL_CTX_set_default_verify_paths(ctx->ssl_ctx)) {
  590. netdata_log_error("Error setting default verify paths");
  591. goto exit_CTX;
  592. }
  593. SSL_CTX_set_verify(ctx->ssl_ctx, SSL_VERIFY_PEER | SSL_VERIFY_CLIENT_ONCE, cert_verify_callback);
  594. ctx->ssl = SSL_new(ctx->ssl_ctx);
  595. if (ctx->ssl==NULL) {
  596. netdata_log_error("Cannot allocate SSL");
  597. goto exit_CTX;
  598. }
  599. if (!SSL_set_tlsext_host_name(ctx->ssl, connect_host)) {
  600. netdata_log_error("Error setting TLS SNI host");
  601. goto exit_CTX;
  602. }
  603. SSL_set_fd(ctx->ssl, ctx->sock);
  604. ret = SSL_connect(ctx->ssl);
  605. if (ret != -1 && ret != 1) {
  606. netdata_log_error("SSL could not connect");
  607. goto exit_SSL;
  608. }
  609. if (ret == -1) {
  610. // expected as underlying socket is non blocking!
  611. // consult SSL_connect documentation for details
  612. int ec = SSL_get_error(ctx->ssl, ret);
  613. if (ec != SSL_ERROR_WANT_READ && ec != SSL_ERROR_WANT_WRITE) {
  614. netdata_log_error("Failed to start SSL connection");
  615. goto exit_SSL;
  616. }
  617. }
  618. // The actual request here
  619. if (handle_http_request(ctx)) {
  620. netdata_log_error("Couldn't process request");
  621. goto exit_SSL;
  622. }
  623. response->http_code = ctx->parse_ctx.http_code;
  624. if (ctx->parse_ctx.content_length == TRANSFER_ENCODING_CHUNKED) {
  625. response->payload_size = ctx->parse_ctx.chunked_response_size;
  626. response->payload = ctx->parse_ctx.chunked_response;
  627. }
  628. if (ctx->parse_ctx.content_length > 0) {
  629. response->payload_size = ctx->parse_ctx.content_length;
  630. response->payload = mallocz(response->payload_size + 1);
  631. ret = rbuf_pop(ctx->buf_rx, response->payload, response->payload_size);
  632. if (ret != (int)response->payload_size) {
  633. netdata_log_error("Payload size doesn't match remaining data on the buffer!");
  634. response->payload_size = ret;
  635. }
  636. // normally we take payload as it is and copy it
  637. // but for convenience in cases where payload is sth. like
  638. // json we add terminating zero so that user of the data
  639. // doesn't have to convert to C string (0 terminated)
  640. // other uses still have correct payload_size and can copy
  641. // only exact data without affixed 0x00
  642. ((char*)response->payload)[response->payload_size] = 0; // mallocz(response->payload_size + 1);
  643. }
  644. netdata_log_info("HTTPS \"%s\" request to \"%s\" finished with HTTP code: %d", http_req_type_to_str(ctx->request->request_type), ctx->request->host, response->http_code);
  645. rc = 0;
  646. exit_SSL:
  647. SSL_free(ctx->ssl);
  648. exit_CTX:
  649. SSL_CTX_free(ctx->ssl_ctx);
  650. exit_sock:
  651. close(ctx->sock);
  652. exit_buf_rx:
  653. rbuf_free(ctx->buf_rx);
  654. exit_req_ctx:
  655. freez(ctx);
  656. return rc;
  657. }
  658. void https_req_response_free(https_req_response_t *res) {
  659. freez(res->payload);
  660. }
  661. void https_req_response_init(https_req_response_t *res) {
  662. res->http_code = 0;
  663. res->payload = NULL;
  664. res->payload_size = 0;
  665. }
  666. static inline char *UNUSED_FUNCTION(min_non_null)(char *a, char *b) {
  667. if (!a)
  668. return b;
  669. if (!b)
  670. return a;
  671. return (a < b ? a : b);
  672. }
  673. #define URI_PROTO_SEPARATOR "://"
  674. #define URL_PARSER_LOG_PREFIX "url_parser "
  675. static int parse_host_port(url_t *url) {
  676. char *ptr = strrchr(url->host, ':');
  677. if (ptr) {
  678. size_t port_len = strlen(ptr + 1);
  679. if (!port_len) {
  680. netdata_log_error(URL_PARSER_LOG_PREFIX ": specified but no port number");
  681. return 1;
  682. }
  683. if (port_len > 5 /* MAX port length is 5digit long in decimal */) {
  684. netdata_log_error(URL_PARSER_LOG_PREFIX "port # is too long");
  685. return 1;
  686. }
  687. *ptr = 0;
  688. if (!strlen(url->host)) {
  689. netdata_log_error(URL_PARSER_LOG_PREFIX "host empty after removing port");
  690. return 1;
  691. }
  692. url->port = atoi (ptr + 1);
  693. }
  694. return 0;
  695. }
  696. static inline void port_by_proto(url_t *url) {
  697. if (url->port)
  698. return;
  699. if (!url->proto)
  700. return;
  701. if (!strcmp(url->proto, "http")) {
  702. url->port = 80;
  703. return;
  704. }
  705. if (!strcmp(url->proto, "https")) {
  706. url->port = 443;
  707. return;
  708. }
  709. }
  710. #define STRDUPZ_2PTR(dest, start, end) \
  711. { \
  712. dest = mallocz(1 + end - start); \
  713. memcpy(dest, start, end - start); \
  714. dest[end - start] = 0; \
  715. }
  716. int url_parse(const char *url, url_t *parsed) {
  717. const char *start = url;
  718. const char *end = strstr(url, URI_PROTO_SEPARATOR);
  719. if (end) {
  720. if (end == start) {
  721. netdata_log_error(URL_PARSER_LOG_PREFIX "found " URI_PROTO_SEPARATOR " without protocol specified");
  722. return 1;
  723. }
  724. STRDUPZ_2PTR(parsed->proto, start, end)
  725. start = end + strlen(URI_PROTO_SEPARATOR);
  726. }
  727. end = strchr(start, '/');
  728. if (!end)
  729. end = start + strlen(start);
  730. if (start == end) {
  731. netdata_log_error(URL_PARSER_LOG_PREFIX "Host empty");
  732. return 1;
  733. }
  734. STRDUPZ_2PTR(parsed->host, start, end);
  735. if (parse_host_port(parsed))
  736. return 1;
  737. if (!*end) {
  738. parsed->path = strdupz("/");
  739. port_by_proto(parsed);
  740. return 0;
  741. }
  742. parsed->path = strdupz(end);
  743. port_by_proto(parsed);
  744. return 0;
  745. }
  746. void url_t_destroy(url_t *url) {
  747. freez(url->host);
  748. freez(url->path);
  749. freez(url->proto);
  750. }