test-server/test-server-http.c

   1 /*
   2  * libwebsockets-test-server - libwebsockets test implementation
   3  *
   4  * Copyright (C) 2010-2015 Andy Green <andy@warmcat.com>
   5  *
   6  *  This library is free software; you can redistribute it and/or
   7  *  modify it under the terms of the GNU Lesser General Public
   8  *  License as published by the Free Software Foundation:
   9  *  version 2.1 of the License.
  10  *
  11  *  This library is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  *  Lesser General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU Lesser General Public
  17  *  License along with this library; if not, write to the Free Software
  18  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
  19  *  MA  02110-1301  USA
  20  */
  21 #include "test-server.h"
  22
  23 /*
  24  * This demo server shows how to use libwebsockets for one or more
  25  * websocket protocols in the same server
  26  *
  27  * It defines the following websocket protocols:
  28  *
  29  *  dumb-increment-protocol:  once the socket is opened, an incrementing
  30  *                              ascii string is sent down it every 50ms.
  31  *                              If you send "reset\n" on the websocket, then
  32  *                              the incrementing number is reset to 0.
  33  *
  34  *  lws-mirror-protocol: copies any received packet to every connection also
  35  *                              using this protocol, including the sender
  36  */
  37
  38 enum demo_protocols {
  39         /* always first */
  40         PROTOCOL_HTTP = 0,
  41
  42         PROTOCOL_DUMB_INCREMENT,
  43         PROTOCOL_LWS_MIRROR,
  44
  45         /* always last */
  46         DEMO_PROTOCOL_COUNT
  47 };
  48
  49 /*
  50  * We take a strict whitelist approach to stop ../ attacks
  51  */
  52 struct serveable {
  53         const char *urlpath;
  54         const char *mimetype;
  55 };
  56
  57 /*
  58  * this is just an example of parsing handshake headers, you don't need this
  59  * in your code unless you will filter allowing connections by the header
  60  * content
  61  */
  62 void
  63 dump_handshake_info(struct lws *wsi)
  64 {
  65         int n = 0;
  66         char buf[256];
  67         const unsigned char *c;
  68
  69         do {
  70                 c = lws_token_to_string(n);
  71                 if (!c) {
  72                         n++;
  73                         continue;
  74                 }
  75
  76                 if (!lws_hdr_total_length(wsi, n)) {
  77                         n++;
  78                         continue;
  79                 }
  80
  81                 lws_hdr_copy(wsi, buf, sizeof buf, n);
  82
  83                 fprintf(stderr, "    %s = %s\n", (char *)c, buf);
  84                 n++;
  85         } while (c);
  86 }
  87
  88 const char * get_mimetype(const char *file)
  89 {
  90         int n = strlen(file);
  91
  92         if (n < 5)
  93                 return NULL;
  94
  95         if (!strcmp(&file[n - 4], ".ico"))
  96                 return "image/x-icon";
  97
  98         if (!strcmp(&file[n - 4], ".png"))
  99                 return "image/png";
 100
 101         if (!strcmp(&file[n - 5], ".html"))
 102                 return "text/html";
 103
 104         return NULL;
 105 }
 106
 107 /* this protocol server (always the first one) handles HTTP,
 108  *
 109  * Some misc callbacks that aren't associated with a protocol also turn up only
 110  * here on the first protocol server.
 111  */
 112
 113 int callback_http(struct lws_context *context, struct lws *wsi,
 114                   enum lws_callback_reasons reason, void *user,
 115                   void *in, size_t len)
 116 {
 117         struct per_session_data__http *pss =
 118                         (struct per_session_data__http *)user;
 119         static unsigned char buffer[4096];
 120         unsigned long amount, file_len;
 121         char leaf_path[1024];
 122         const char *mimetype;
 123         char *other_headers;
 124         unsigned char *end;
 125         struct timeval tv;
 126         unsigned char *p;
 127         char buf[256];
 128         char b64[64];
 129         int n, m;
 130
 131 #ifdef EXTERNAL_POLL
 132         struct lws_pollargs *pa = (struct lws_pollargs *)in;
 133 #endif
 134
 135         switch (reason) {
 136         case LWS_CALLBACK_HTTP:
 137
 138                 dump_handshake_info(wsi);
 139
 140                 if (len < 1) {
 141                         lws_return_http_status(context, wsi,
 142                                                 HTTP_STATUS_BAD_REQUEST, NULL);
 143                         goto try_to_reuse;
 144                 }
 145
 146                 /* this example server has no concept of directories */
 147                 if (strchr((const char *)in + 1, '/')) {
 148                         lws_return_http_status(context, wsi,
 149                                                 HTTP_STATUS_FORBIDDEN, NULL);
 150                         goto try_to_reuse;
 151                 }
 152
 153                 /* if a legal POST URL, let it continue and accept data */
 154                 if (lws_hdr_total_length(wsi, WSI_TOKEN_POST_URI))
 155                         return 0;
 156
 157                 /* check for the "send a big file by hand" example case */
 158
 159                 if (!strcmp((const char *)in, "/leaf.jpg")) {
 160                         if (strlen(resource_path) > sizeof(leaf_path) - 10)
 161                                 return -1;
 162                         sprintf(leaf_path, "%s/leaf.jpg", resource_path);
 163
 164                         /* well, let's demonstrate how to send the hard way */
 165
 166                         p = buffer + LWS_SEND_BUFFER_PRE_PADDING;
 167                         end = p + sizeof(buffer) - LWS_SEND_BUFFER_PRE_PADDING;
 168
 169                         pss->fd = lws_plat_file_open(lws_get_fops(context),
 170                                                      leaf_path, &file_len,
 171                                                      O_RDONLY);
 172
 173                         if (pss->fd == LWS_INVALID_FILE)
 174                                 return -1;
 175
 176                         /*
 177                          * we will send a big jpeg file, but it could be
 178                          * anything.  Set the Content-Type: appropriately
 179                          * so the browser knows what to do with it.
 180                          *
 181                          * Notice we use the APIs to build the header, which
 182                          * will do the right thing for HTTP 1/1.1 and HTTP2
 183                          * depending on what connection it happens to be working
 184                          * on
 185                          */
 186                         if (lws_add_http_header_status(context, wsi, 200, &p, end))
 187                                 return 1;
 188                         if (lws_add_http_header_by_token(context, wsi,
 189                                         WSI_TOKEN_HTTP_SERVER,
 190                                         (unsigned char *)"libwebsockets",
 191                                         13, &p, end))
 192                                 return 1;
 193                         if (lws_add_http_header_by_token(context, wsi,
 194                                         WSI_TOKEN_HTTP_CONTENT_TYPE,
 195                                         (unsigned char *)"image/jpeg",
 196                                         10, &p, end))
 197                                 return 1;
 198                         if (lws_add_http_header_content_length(context, wsi,
 199                                                                file_len, &p,
 200                                                                end))
 201                                 return 1;
 202                         if (lws_finalize_http_header(context, wsi, &p, end))
 203                                 return 1;
 204
 205                         /*
 206                          * send the http headers...
 207                          * this won't block since it's the first payload sent
 208                          * on the connection since it was established
 209                          * (too small for partial)
 210                          *
 211                          * Notice they are sent using LWS_WRITE_HTTP_HEADERS
 212                          * which also means you can't send body too in one step,
 213                          * this is mandated by changes in HTTP2
 214                          */
 215
 216                         n = lws_write(wsi, buffer + LWS_SEND_BUFFER_PRE_PADDING,
 217                                       p - (buffer + LWS_SEND_BUFFER_PRE_PADDING),
 218                                       LWS_WRITE_HTTP_HEADERS);
 219
 220                         if (n < 0) {
 221                                 lws_plat_file_close(lws_get_fops(context),
 222                                                     pss->fd);
 223                                 return -1;
 224                         }
 225                         /*
 226                          * book us a LWS_CALLBACK_HTTP_WRITEABLE callback
 227                          */
 228                         lws_callback_on_writable(context, wsi);
 229                         break;
 230                 }
 231
 232                 /* if not, send a file the easy way */
 233                 strcpy(buf, resource_path);
 234                 if (strcmp(in, "/")) {
 235                         if (*((const char *)in) != '/')
 236                                 strcat(buf, "/");
 237                         strncat(buf, in, sizeof(buf) - strlen(resource_path));
 238                 } else /* default file to serve */
 239                         strcat(buf, "/test.html");
 240                 buf[sizeof(buf) - 1] = '\0';
 241
 242                 /* refuse to serve files we don't understand */
 243                 mimetype = get_mimetype(buf);
 244                 if (!mimetype) {
 245                         lwsl_err("Unknown mimetype for %s\n", buf);
 246                         lws_return_http_status(context, wsi,
 247                                       HTTP_STATUS_UNSUPPORTED_MEDIA_TYPE, NULL);
 248                         return -1;
 249                 }
 250
 251                 /* demostrates how to set a cookie on / */
 252
 253                 other_headers = NULL;
 254                 n = 0;
 255                 if (!strcmp((const char *)in, "/") &&
 256                            !lws_hdr_total_length(wsi, WSI_TOKEN_HTTP_COOKIE)) {
 257                         /* this isn't very unguessable but it'll do for us */
 258                         gettimeofday(&tv, NULL);
 259                         n = sprintf(b64, "test=LWS_%u_%u_COOKIE;Max-Age=360000",
 260                                 (unsigned int)tv.tv_sec,
 261                                 (unsigned int)tv.tv_usec);
 262
 263                         p = (unsigned char *)leaf_path;
 264
 265                         if (lws_add_http_header_by_name(context, wsi,
 266                                 (unsigned char *)"set-cookie:",
 267                                 (unsigned char *)b64, n, &p,
 268                                 (unsigned char *)leaf_path + sizeof(leaf_path)))
 269                                 return 1;
 270                         n = (char *)p - leaf_path;
 271                         other_headers = leaf_path;
 272                 }
 273
 274                 n = lws_serve_http_file(context, wsi, buf,
 275                                                 mimetype, other_headers, n);
 276                 if (n < 0 || ((n > 0) && lws_http_transaction_completed(wsi)))
 277                         return -1; /* error or can't reuse connection: close the socket */
 278
 279                 /*
 280                  * notice that the sending of the file completes asynchronously,
 281                  * we'll get a LWS_CALLBACK_HTTP_FILE_COMPLETION callback when
 282                  * it's done
 283                  */
 284
 285                 break;
 286
 287         case LWS_CALLBACK_HTTP_BODY:
 288                 strncpy(buf, in, 20);
 289                 buf[20] = '\0';
 290                 if (len < 20)
 291                         buf[len] = '\0';
 292
 293                 lwsl_notice("LWS_CALLBACK_HTTP_BODY: %s... len %d\n",
 294                                 (const char *)buf, (int)len);
 295
 296                 break;
 297
 298         case LWS_CALLBACK_HTTP_BODY_COMPLETION:
 299                 lwsl_notice("LWS_CALLBACK_HTTP_BODY_COMPLETION\n");
 300                 /* the whole of the sent body arrived, close or reuse the connection */
 301                 lws_return_http_status(context, wsi,
 302                                                 HTTP_STATUS_OK, NULL);
 303                 goto try_to_reuse;
 304
 305         case LWS_CALLBACK_HTTP_FILE_COMPLETION:
 306                 goto try_to_reuse;
 307
 308         case LWS_CALLBACK_HTTP_WRITEABLE:
 309                 /*
 310                  * we can send more of whatever it is we were sending
 311                  */
 312                 do {
 313                         /* we'd like the send this much */
 314                         n = sizeof(buffer) - LWS_SEND_BUFFER_PRE_PADDING;
 315
 316                         /* but if the peer told us he wants less, we can adapt */
 317                         m = lws_get_peer_write_allowance(wsi);
 318
 319                         /* -1 means not using a protocol that has this info */
 320                         if (m == 0)
 321                                 /* right now, peer can't handle anything */
 322                                 goto later;
 323
 324                         if (m != -1 && m < n)
 325                                 /* he couldn't handle that much */
 326                                 n = m;
 327
 328                         n = lws_plat_file_read(lws_get_fops(context), pss->fd,
 329                                                &amount, buffer +
 330                                                 LWS_SEND_BUFFER_PRE_PADDING, n);
 331                         /* problem reading, close conn */
 332                         if (n < 0)
 333                                 goto bail;
 334                         n = (int)amount;
 335                         /* sent it all, close conn */
 336                         if (n == 0)
 337                                 goto flush_bail;
 338                         /*
 339                          * To support HTTP2, must take care about preamble space
 340                          *
 341                          * identification of when we send the last payload frame
 342                          * is handled by the library itself if you sent a
 343                          * content-length header
 344                          */
 345                         m = lws_write(wsi, buffer + LWS_SEND_BUFFER_PRE_PADDING,
 346                                       n, LWS_WRITE_HTTP);
 347                         if (m < 0)
 348                                 /* write failed, close conn */
 349                                 goto bail;
 350
 351                         /*
 352                          * http2 won't do this
 353                          */
 354                         if (m != n)
 355                                 /* partial write, adjust */
 356                                 if (lws_plat_file_seek_cur(lws_get_fops(context),
 357                                                            pss->fd, m - n) ==
 358                                                              (unsigned long)-1)
 359                                         goto bail;
 360
 361                         if (m) /* while still active, extend timeout */
 362                                 lws_set_timeout(wsi,
 363                                                 PENDING_TIMEOUT_HTTP_CONTENT, 5);
 364
 365                         /* if we have indigestion, let him clear it
 366                          * before eating more */
 367                         if (lws_partial_buffered(wsi))
 368                                 break;
 369
 370                 } while (!lws_send_pipe_choked(wsi));
 371
 372 later:
 373                 lws_callback_on_writable(context, wsi);
 374                 break;
 375 flush_bail:
 376                 /* true if still partial pending */
 377                 if (lws_partial_buffered(wsi)) {
 378                         lws_callback_on_writable(context, wsi);
 379                         break;
 380                 }
 381                 lws_plat_file_close(lws_get_fops(context), pss->fd);
 382                 goto try_to_reuse;
 383
 384 bail:
 385                 lws_plat_file_close(lws_get_fops(context), pss->fd);
 386                 return -1;
 387
 388         /*
 389          * callback for confirming to continue with client IP appear in
 390          * protocol 0 callback since no websocket protocol has been agreed
 391          * yet.  You can just ignore this if you won't filter on client IP
 392          * since the default uhandled callback return is 0 meaning let the
 393          * connection continue.
 394          */
 395         case LWS_CALLBACK_FILTER_NETWORK_CONNECTION:
 396
 397                 /* if we returned non-zero from here, we kill the connection */
 398                 break;
 399
 400         /*
 401          * callbacks for managing the external poll() array appear in
 402          * protocol 0 callback
 403          */
 404
 405         case LWS_CALLBACK_LOCK_POLL:
 406                 /*
 407                  * lock mutex to protect pollfd state
 408                  * called before any other POLL related callback
 409                  * if protecting wsi lifecycle change, len == 1
 410                  */
 411                 test_server_lock(len);
 412                 break;
 413
 414         case LWS_CALLBACK_UNLOCK_POLL:
 415                 /*
 416                  * unlock mutex to protect pollfd state when
 417                  * called after any other POLL related callback
 418                  * if protecting wsi lifecycle change, len == 1
 419                  */
 420                 test_server_unlock(len);
 421                 break;
 422
 423 #ifdef EXTERNAL_POLL
 424         case LWS_CALLBACK_ADD_POLL_FD:
 425
 426                 if (count_pollfds >= max_poll_elements) {
 427                         lwsl_err("LWS_CALLBACK_ADD_POLL_FD: too many sockets to track\n");
 428                         return 1;
 429                 }
 430
 431                 fd_lookup[pa->fd] = count_pollfds;
 432                 pollfds[count_pollfds].fd = pa->fd;
 433                 pollfds[count_pollfds].events = pa->events;
 434                 pollfds[count_pollfds++].revents = 0;
 435                 break;
 436
 437         case LWS_CALLBACK_DEL_POLL_FD:
 438                 if (!--count_pollfds)
 439                         break;
 440                 m = fd_lookup[pa->fd];
 441                 /* have the last guy take up the vacant slot */
 442                 pollfds[m] = pollfds[count_pollfds];
 443                 fd_lookup[pollfds[count_pollfds].fd] = m;
 444                 break;
 445
 446         case LWS_CALLBACK_CHANGE_MODE_POLL_FD:
 447                 pollfds[fd_lookup[pa->fd]].events = pa->events;
 448                 break;
 449 #endif
 450
 451         case LWS_CALLBACK_GET_THREAD_ID:
 452                 /*
 453                  * if you will call "lws_callback_on_writable"
 454                  * from a different thread, return the caller thread ID
 455                  * here so lws can use this information to work out if it
 456                  * should signal the poll() loop to exit and restart early
 457                  */
 458
 459                 /* return pthread_getthreadid_np(); */
 460
 461                 break;
 462
 463         default:
 464                 break;
 465         }
 466
 467         return 0;
 468
 469         /* if we're on HTTP1.1 or 2.0, will keep the idle connection alive */
 470 try_to_reuse:
 471         if (lws_http_transaction_completed(wsi))
 472                 return -1;
 473
 474         return 0;
 475 }