http_server.c 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388
  1. /*
  2. * Copyright (C) 2011, 2012, 2013, 2014 Citrix Systems
  3. *
  4. * All rights reserved.
  5. *
  6. * Redistribution and use in source and binary forms, with or without
  7. * modification, are permitted provided that the following conditions
  8. * are met:
  9. * 1. Redistributions of source code must retain the above copyright
  10. * notice, this list of conditions and the following disclaimer.
  11. * 2. Redistributions in binary form must reproduce the above copyright
  12. * notice, this list of conditions and the following disclaimer in the
  13. * documentation and/or other materials provided with the distribution.
  14. * 3. Neither the name of the project nor the names of its contributors
  15. * may be used to endorse or promote products derived from this software
  16. * without specific prior written permission.
  17. *
  18. * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
  19. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  21. * ARE DISCLAIMED. IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
  22. * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  23. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  24. * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  25. * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  26. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  27. * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  28. * SUCH DAMAGE.
  29. */
  30. #include "ns_ioalib_impl.h"
  31. #include "http_server.h"
  32. #include <event2/http.h>
  33. #include <event2/keyvalq_struct.h>
  34. #include <time.h>
  35. //////////////////////////////////////
  36. struct headers_list {
  37. size_t n;
  38. char **keys;
  39. char **values;
  40. };
  41. struct http_headers {
  42. struct evkeyvalq *uri_headers;
  43. struct headers_list *post_headers;
  44. };
  45. //////////////////////////////////////
  46. static void write_http_echo(ioa_socket_handle s)
  47. {
  48. if(s && !ioa_socket_tobeclosed(s)) {
  49. SOCKET_APP_TYPE sat = get_ioa_socket_app_type(s);
  50. if((sat == HTTP_CLIENT_SOCKET) || (sat == HTTPS_CLIENT_SOCKET)) {
  51. ioa_network_buffer_handle nbh_http = ioa_network_buffer_allocate(s->e);
  52. size_t len_http = ioa_network_buffer_get_size(nbh_http);
  53. uint8_t *data = ioa_network_buffer_data(nbh_http);
  54. char data_http[1025];
  55. char content_http[1025];
  56. const char* title = "TURN Server";
  57. snprintf(content_http,sizeof(content_http)-1,"<!DOCTYPE html>\r\n<html>\r\n <head>\r\n <title>%s</title>\r\n </head>\r\n <body>\r\n <b>%s</b> <br> <b><i>use https connection for the admin session</i></b>\r\n </body>\r\n</html>\r\n",title,title);
  58. snprintf(data_http,sizeof(data_http)-1,"HTTP/1.0 200 OK\r\nServer: %s\r\nContent-Type: text/html; charset=UTF-8\r\nContent-Length: %d\r\n\r\n%.906s",TURN_SOFTWARE,(int)strlen(content_http),content_http);
  59. len_http = strlen(data_http);
  60. bcopy(data_http,data,len_http);
  61. ioa_network_buffer_set_size(nbh_http,len_http);
  62. send_data_from_ioa_socket_nbh(s, NULL, nbh_http, TTL_IGNORE, TOS_IGNORE,NULL);
  63. }
  64. }
  65. }
  66. void handle_http_echo(ioa_socket_handle s) {
  67. write_http_echo(s);
  68. }
  69. const char* get_http_date_header()
  70. {
  71. static char buffer_date[256];
  72. static char buffer_header[1025];
  73. static const char* wds[]={"Sun","Mon","Tue","Wed","Thu","Fri","Sat"};
  74. static const char* mons[]={"Jan","Feb","Mar","Apr","May","Jun","Jul","Aug","Sep","Oct","Nov","Dec"};
  75. time_t now = time(NULL);
  76. struct tm *gmtm = gmtime(&now);
  77. buffer_header[0]=0;
  78. buffer_date[0]=0;
  79. if(gmtm) {
  80. snprintf(buffer_date,sizeof(buffer_date)-1,"%s, %d %s %d %d:%d:%d GMT",wds[gmtm->tm_wday], gmtm->tm_mday, mons[gmtm->tm_mon], gmtm->tm_year+1900, gmtm->tm_hour, gmtm->tm_min, gmtm->tm_sec);
  81. buffer_date[sizeof(buffer_date)-1]=0;
  82. snprintf(buffer_header,sizeof(buffer_header)-1,"Date: %s\r\n",buffer_date);
  83. buffer_header[sizeof(buffer_header)-1]=0;
  84. }
  85. return buffer_header;
  86. }
  87. ///////////////////////////////////////////////
  88. static struct headers_list * post_parse(char *data, size_t data_len)
  89. {
  90. while((*data=='\r')||(*data=='\n')) { ++data; --data_len; }
  91. if (data_len) {
  92. char *post_data = (char*)calloc(data_len + 1, sizeof(char));
  93. if (post_data != NULL) {
  94. memcpy(post_data, data, data_len);
  95. char *fmarker = NULL;
  96. char *fsplit = strtok_r(post_data, "&", &fmarker);
  97. struct headers_list *list = (struct headers_list*)malloc(sizeof(struct headers_list));
  98. bzero(list,sizeof(struct headers_list));
  99. while (fsplit != NULL) {
  100. char *vmarker = NULL;
  101. char *key = strtok_r(fsplit, "=", &vmarker);
  102. if (key == NULL)
  103. break;
  104. else {
  105. char *value = strtok_r(NULL, "=", &vmarker);
  106. char empty[1];
  107. empty[0]=0;
  108. value = value ? value : empty;
  109. value = evhttp_decode_uri(value);
  110. char *p = value;
  111. while (*p) {
  112. if (*p == '+')
  113. *p = ' ';
  114. p++;
  115. }
  116. list->keys = (char**)realloc(list->keys,sizeof(char*)*(list->n+1));
  117. list->keys[list->n] = strdup(key);
  118. list->values = (char**)realloc(list->values,sizeof(char*)*(list->n+1));
  119. list->values[list->n] = value;
  120. ++(list->n);
  121. fsplit = strtok_r(NULL, "&", &fmarker);
  122. }
  123. }
  124. free(post_data);
  125. return list;
  126. }
  127. }
  128. return NULL;
  129. }
  130. static struct http_request* parse_http_request_1(struct http_request* ret, char* request, int parse_post)
  131. {
  132. if(ret && request) {
  133. char* s = strstr(request," HTTP/");
  134. if(!s) {
  135. free(ret);
  136. ret = NULL;
  137. } else {
  138. *s = 0;
  139. struct evhttp_uri *uri = evhttp_uri_parse(request);
  140. if(!uri) {
  141. free(ret);
  142. ret = NULL;
  143. } else {
  144. const char *query = evhttp_uri_get_query(uri);
  145. if(query) {
  146. struct evkeyvalq* kv = (struct evkeyvalq*)malloc(sizeof(struct evkeyvalq));
  147. bzero(kv,sizeof(struct evkeyvalq));
  148. if(evhttp_parse_query_str(query, kv)<0) {
  149. free(ret);
  150. ret = NULL;
  151. } else {
  152. ret->headers = (struct http_headers*)malloc(sizeof(struct http_headers));
  153. bzero(ret->headers,sizeof(struct http_headers));
  154. ret->headers->uri_headers = kv;
  155. }
  156. }
  157. const char *path = evhttp_uri_get_path(uri);
  158. if(path && ret)
  159. ret->path = strdup(path);
  160. evhttp_uri_free(uri);
  161. if(parse_post && ret) {
  162. char *body = strstr(s+1,"\r\n\r\n");
  163. if(body && body[0]) {
  164. if(!ret->headers) {
  165. ret->headers = (struct http_headers*)malloc(sizeof(struct http_headers));
  166. bzero(ret->headers,sizeof(struct http_headers));
  167. }
  168. ret->headers->post_headers = post_parse(body,strlen(body));
  169. }
  170. }
  171. }
  172. *s = ' ';
  173. }
  174. }
  175. return ret;
  176. }
  177. struct http_request* parse_http_request(char* request) {
  178. struct http_request* ret = NULL;
  179. if(request) {
  180. ret = (struct http_request*)malloc(sizeof(struct http_request));
  181. bzero(ret,sizeof(struct http_request));
  182. if(strstr(request,"GET ") == request) {
  183. ret->rtype = HRT_GET;
  184. ret = parse_http_request_1(ret,request+4,0);
  185. } else if(strstr(request,"HEAD ") == request) {
  186. ret->rtype = HRT_HEAD;
  187. ret = parse_http_request_1(ret,request+5,0);
  188. } else if(strstr(request,"POST ") == request) {
  189. ret->rtype = HRT_POST;
  190. ret = parse_http_request_1(ret,request+5,1);
  191. } else if(strstr(request,"PUT ") == request) {
  192. ret->rtype = HRT_PUT;
  193. ret = parse_http_request_1(ret,request+4,1);
  194. } else if(strstr(request,"DELETE ") == request) {
  195. ret->rtype = HRT_DELETE;
  196. ret = parse_http_request_1(ret,request+7,1);
  197. } else {
  198. free(ret);
  199. ret = NULL;
  200. }
  201. }
  202. return ret;
  203. }
  204. static const char * get_headers_list_value(struct headers_list *h, const char* key) {
  205. const char* ret = NULL;
  206. if(h && h->keys && h->values && key && key[0]) {
  207. size_t i = 0;
  208. for(i=0;i<h->n;++i) {
  209. if(h->keys[i] && !strcmp(key,h->keys[i]) && h->values[i]) {
  210. ret = h->values[i];
  211. break;
  212. }
  213. }
  214. }
  215. return ret;
  216. }
  217. static void free_headers_list(struct headers_list *h) {
  218. if(h) {
  219. if(h->keys) {
  220. size_t i = 0;
  221. for(i=0;i<h->n;++i) {
  222. if(h->keys[i]) {
  223. free(h->keys[i]);
  224. h->keys[i]=NULL;
  225. }
  226. }
  227. free(h->keys);
  228. h->keys = NULL;
  229. }
  230. if(h->values) {
  231. size_t i = 0;
  232. for(i=0;i<h->n;++i) {
  233. if(h->values[i]) {
  234. free(h->values[i]);
  235. h->values[i]=NULL;
  236. }
  237. }
  238. free(h->values);
  239. h->values = NULL;
  240. }
  241. h->n = 0;
  242. free(h);
  243. }
  244. }
  245. const char *get_http_header_value(const struct http_request *request, const char* key, const char* default_value) {
  246. const char *ret = NULL;
  247. if(key && key[0] && request && request->headers) {
  248. if(request->headers->uri_headers) {
  249. ret = evhttp_find_header(request->headers->uri_headers,key);
  250. }
  251. if(!ret && request->headers->post_headers) {
  252. ret = get_headers_list_value(request->headers->post_headers,key);
  253. }
  254. }
  255. if(!ret) {
  256. ret = default_value;
  257. }
  258. return ret;
  259. }
  260. void free_http_request(struct http_request *request) {
  261. if(request) {
  262. if(request->path) {
  263. free(request->path);
  264. request->path = NULL;
  265. }
  266. if(request->headers) {
  267. if(request->headers->uri_headers) {
  268. evhttp_clear_headers(request->headers->uri_headers);
  269. free(request->headers->uri_headers);
  270. request->headers->uri_headers = NULL;
  271. }
  272. if(request->headers->post_headers) {
  273. free_headers_list(request->headers->post_headers);
  274. request->headers->post_headers = NULL;
  275. }
  276. free(request->headers);
  277. request->headers = NULL;
  278. }
  279. free(request);
  280. }
  281. }
  282. ////////////////////////////////////////////
  283. struct str_buffer {
  284. size_t capacity;
  285. size_t sz;
  286. char* buffer;
  287. };
  288. struct str_buffer* str_buffer_new(void)
  289. {
  290. struct str_buffer* ret = (struct str_buffer*)malloc(sizeof(struct str_buffer));
  291. bzero(ret,sizeof(struct str_buffer));
  292. ret->buffer = (char*)malloc(1);
  293. ret->buffer[0] = 0;
  294. ret->capacity = 1;
  295. return ret;
  296. }
  297. void str_buffer_append(struct str_buffer* sb, const char* str)
  298. {
  299. if(sb && str && str[0]) {
  300. size_t len = strlen(str);
  301. while(sb->sz + len + 1 > sb->capacity) {
  302. sb->capacity += len + 1024;
  303. sb->buffer = (char*)realloc(sb->buffer,sb->capacity);
  304. }
  305. bcopy(str,sb->buffer+sb->sz,len+1);
  306. sb->sz += len;
  307. }
  308. }
  309. void str_buffer_append_sz(struct str_buffer* sb, size_t sz)
  310. {
  311. char ssz[129];
  312. snprintf(ssz,sizeof(ssz)-1,"%lu",(unsigned long)sz);
  313. str_buffer_append(sb,ssz);
  314. }
  315. void str_buffer_append_sid(struct str_buffer* sb, turnsession_id sid)
  316. {
  317. char ssz[129];
  318. snprintf(ssz,sizeof(ssz)-1,"%018llu",(unsigned long long)sid);
  319. str_buffer_append(sb,ssz);
  320. }
  321. const char* str_buffer_get_str(const struct str_buffer *sb)
  322. {
  323. if(sb) {
  324. return sb->buffer;
  325. }
  326. return NULL;
  327. }
  328. size_t str_buffer_get_str_len(const struct str_buffer *sb)
  329. {
  330. if(sb) {
  331. return sb->sz;
  332. }
  333. return 0;
  334. }
  335. void str_buffer_free(struct str_buffer *sb)
  336. {
  337. if(sb) {
  338. free(sb->buffer);
  339. free(sb);
  340. }
  341. }
  342. ///////////////////////////////////////////////