1 /***************************************************************************
3 * Project ___| | | | _ \| |
5 * | (__| |_| | _ <| |___
6 * \___|\___/|_| \_\_____|
8 * Copyright (C) 1998 - 2016, Daniel Stenberg, <daniel@haxx.se>, et al.
10 * This software is licensed as described in the file COPYING, which
11 * you should have received as part of this distribution. The terms
12 * are also available at https://curl.haxx.se/docs/copyright.html.
14 * You may opt to use, copy, modify, merge, publish, distribute and/or sell
15 * copies of the Software, and permit persons to whom the Software is
16 * furnished to do so, under the terms of the COPYING file.
18 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
19 * KIND, either express or implied.
21 ***************************************************************************/
22 #include "server_setup.h"
25 * curl's test suite Real Time Streaming Protocol (RTSP) server.
27 * This source file was started based on curl's HTTP test suite server.
33 #ifdef HAVE_NETINET_IN_H
34 #include <netinet/in.h>
36 #ifdef HAVE_ARPA_INET_H
37 #include <arpa/inet.h>
42 #ifdef HAVE_NETINET_TCP_H
43 #include <netinet/tcp.h> /* for TCP_NODELAY */
46 #define ENABLE_CURLX_PRINTF
47 /* make the curlx header define all printf() functions to use the curlx_*
49 #include "curlx.h" /* from the private lib dir */
52 #include "server_sockaddr.h"
54 /* include memdebug.h last */
59 #define EINTR 4 /* errno.h value */
61 #define ERANGE 34 /* errno.h value */
65 static bool use_ipv6 = FALSE;
67 static const char *ipv_inuse = "IPv4";
68 static int serverlogslocked = 0;
70 #define REQBUFSIZ 150000
71 #define REQBUFSIZ_TXT "149999"
73 static long prevtestno=-1; /* previous test number we served */
74 static long prevpartno=-1; /* previous part number we served */
75 static bool prevbounce=FALSE; /* instructs the server to increase the part
76 number for a test in case the identical
77 testno+partno request shows up again */
79 #define RCMD_NORMALREQ 0 /* default request, use the tests file normally */
80 #define RCMD_IDLE 1 /* told to sit idle */
81 #define RCMD_STREAM 2 /* told to stream */
89 #define SET_RTP_PKT_CHN(p,c) ((p)[1] = (unsigned char)((c) & 0xFF))
91 #define SET_RTP_PKT_LEN(p,l) (((p)[2] = (unsigned char)(((l) >> 8) & 0xFF)), \
92 ((p)[3] = (unsigned char)((l) & 0xFF)))
95 char reqbuf[REQBUFSIZ]; /* buffer area for the incoming request */
96 size_t checkindex; /* where to start checking of the request */
97 size_t offset; /* size of the incoming request */
98 long testno; /* test number found in the request */
99 long partno; /* part number found in the request */
100 bool open; /* keep connection open info, as found in the request */
101 bool auth_req; /* authentication required, don't wait for body unless
102 there's an Authorization header */
103 bool auth; /* Authorization header present in the incoming request */
104 size_t cl; /* Content-Length of the incoming request */
105 bool digest; /* Authorization digest header found */
106 bool ntlm; /* Authorization ntlm header found */
107 int pipe; /* if non-zero, expect this many requests to do a "piped"
109 int skip; /* if non-zero, the server is instructed to not read this
110 many bytes from a PUT/POST request. Ie the client sends N
111 bytes said in Content-Length, but the server only reads N
113 int rcmd; /* doing a special command, see defines above */
114 reqprot_t protocol; /* request protocol, HTTP or RTSP */
115 int prot_version; /* HTTP or RTSP version (major*10 + minor) */
116 bool pipelining; /* true if request is pipelined */
118 size_t rtp_buffersize;
121 static int ProcessRequest(struct httprequest *req);
122 static void storerequest(char *reqbuf, size_t totalsize);
124 #define DEFAULT_PORT 8999
126 #ifndef DEFAULT_LOGFILE
127 #define DEFAULT_LOGFILE "log/rtspd.log"
130 const char *serverlogfile = DEFAULT_LOGFILE;
132 #define RTSPDVERSION "curl test suite RTSP server/0.1"
134 #define REQUEST_DUMP "log/server.input"
135 #define RESPONSE_DUMP "log/server.response"
137 /* very-big-path support */
138 #define MAXDOCNAMELEN 140000
139 #define MAXDOCNAMELEN_TXT "139999"
141 #define REQUEST_KEYWORD_SIZE 256
142 #define REQUEST_KEYWORD_SIZE_TXT "255"
144 #define CMD_AUTH_REQUIRED "auth_required"
146 /* 'idle' means that it will accept the request fine but never respond
147 any data. Just keep the connection alive. */
148 #define CMD_IDLE "idle"
150 /* 'stream' means to send a never-ending stream of data */
151 #define CMD_STREAM "stream"
153 #define END_OF_HEADERS "\r\n\r\n"
156 DOCNUMBER_NOTHING = -7,
158 DOCNUMBER_BADCONNECT = -5,
159 DOCNUMBER_INTERNAL= -4,
160 DOCNUMBER_CONNECT = -3,
161 DOCNUMBER_WERULEZ = -2,
166 /* sent as reply to a QUIT */
167 static const char *docquit =
168 "HTTP/1.1 200 Goodbye" END_OF_HEADERS;
170 /* sent as reply to a CONNECT */
171 static const char *docconnect =
172 "HTTP/1.1 200 Mighty fine indeed" END_OF_HEADERS;
174 /* sent as reply to a "bad" CONNECT */
175 static const char *docbadconnect =
176 "HTTP/1.1 501 Forbidden you fool" END_OF_HEADERS;
178 /* send back this on HTTP 404 file not found */
179 static const char *doc404_HTTP = "HTTP/1.1 404 Not Found\r\n"
180 "Server: " RTSPDVERSION "\r\n"
181 "Connection: close\r\n"
182 "Content-Type: text/html"
184 "<!DOCTYPE HTML PUBLIC \"-//IETF//DTD HTML 2.0//EN\">\n"
186 "<TITLE>404 Not Found</TITLE>\n"
188 "<H1>Not Found</H1>\n"
189 "The requested URL was not found on this server.\n"
190 "<P><HR><ADDRESS>" RTSPDVERSION "</ADDRESS>\n" "</BODY></HTML>\n";
192 /* send back this on RTSP 404 file not found */
193 static const char *doc404_RTSP = "RTSP/1.0 404 Not Found\r\n"
194 "Server: " RTSPDVERSION
197 /* Default size to send away fake RTP data */
198 #define RTP_DATA_SIZE 12
199 static const char *RTP_DATA = "$_1234\n\0asdf";
201 /* do-nothing macro replacement for systems which lack siginterrupt() */
203 #ifndef HAVE_SIGINTERRUPT
204 #define siginterrupt(x,y) do {} while(0)
207 /* vars used to keep around previous signal handlers */
209 typedef RETSIGTYPE (*SIGHANDLER_T)(int);
212 static SIGHANDLER_T old_sighup_handler = SIG_ERR;
216 static SIGHANDLER_T old_sigpipe_handler = SIG_ERR;
220 static SIGHANDLER_T old_sigalrm_handler = SIG_ERR;
224 static SIGHANDLER_T old_sigint_handler = SIG_ERR;
228 static SIGHANDLER_T old_sigterm_handler = SIG_ERR;
231 #if defined(SIGBREAK) && defined(WIN32)
232 static SIGHANDLER_T old_sigbreak_handler = SIG_ERR;
235 /* var which if set indicates that the program should finish execution */
237 SIG_ATOMIC_T got_exit_signal = 0;
239 /* if next is set indicates the first signal handled in exit_signal_handler */
241 static volatile int exit_signal = 0;
243 /* signal handler that will be triggered to indicate that the program
244 should finish its execution in a controlled manner as soon as possible.
245 The first time this is called it will set got_exit_signal to one and
246 store in exit_signal the signal that triggered its execution. */
248 static RETSIGTYPE exit_signal_handler(int signum)
250 int old_errno = errno;
251 if(got_exit_signal == 0) {
253 exit_signal = signum;
255 (void)signal(signum, exit_signal_handler);
259 static void install_signal_handlers(void)
262 /* ignore SIGHUP signal */
263 old_sighup_handler = signal(SIGHUP, SIG_IGN);
264 if(old_sighup_handler == SIG_ERR)
265 logmsg("cannot install SIGHUP handler: %s", strerror(errno));
268 /* ignore SIGPIPE signal */
269 old_sigpipe_handler = signal(SIGPIPE, SIG_IGN);
270 if(old_sigpipe_handler == SIG_ERR)
271 logmsg("cannot install SIGPIPE handler: %s", strerror(errno));
274 /* ignore SIGALRM signal */
275 old_sigalrm_handler = signal(SIGALRM, SIG_IGN);
276 if(old_sigalrm_handler == SIG_ERR)
277 logmsg("cannot install SIGALRM handler: %s", strerror(errno));
280 /* handle SIGINT signal with our exit_signal_handler */
281 old_sigint_handler = signal(SIGINT, exit_signal_handler);
282 if(old_sigint_handler == SIG_ERR)
283 logmsg("cannot install SIGINT handler: %s", strerror(errno));
285 siginterrupt(SIGINT, 1);
288 /* handle SIGTERM signal with our exit_signal_handler */
289 old_sigterm_handler = signal(SIGTERM, exit_signal_handler);
290 if(old_sigterm_handler == SIG_ERR)
291 logmsg("cannot install SIGTERM handler: %s", strerror(errno));
293 siginterrupt(SIGTERM, 1);
295 #if defined(SIGBREAK) && defined(WIN32)
296 /* handle SIGBREAK signal with our exit_signal_handler */
297 old_sigbreak_handler = signal(SIGBREAK, exit_signal_handler);
298 if(old_sigbreak_handler == SIG_ERR)
299 logmsg("cannot install SIGBREAK handler: %s", strerror(errno));
301 siginterrupt(SIGBREAK, 1);
305 static void restore_signal_handlers(void)
308 if(SIG_ERR != old_sighup_handler)
309 (void)signal(SIGHUP, old_sighup_handler);
312 if(SIG_ERR != old_sigpipe_handler)
313 (void)signal(SIGPIPE, old_sigpipe_handler);
316 if(SIG_ERR != old_sigalrm_handler)
317 (void)signal(SIGALRM, old_sigalrm_handler);
320 if(SIG_ERR != old_sigint_handler)
321 (void)signal(SIGINT, old_sigint_handler);
324 if(SIG_ERR != old_sigterm_handler)
325 (void)signal(SIGTERM, old_sigterm_handler);
327 #if defined(SIGBREAK) && defined(WIN32)
328 if(SIG_ERR != old_sigbreak_handler)
329 (void)signal(SIGBREAK, old_sigbreak_handler);
333 static int ProcessRequest(struct httprequest *req)
335 char *line=&req->reqbuf[req->checkindex];
336 bool chunked = FALSE;
337 static char request[REQUEST_KEYWORD_SIZE];
338 static char doc[MAXDOCNAMELEN];
339 static char prot_str[5];
341 int prot_major, prot_minor;
344 end = strstr(line, END_OF_HEADERS);
346 logmsg("ProcessRequest() called with testno %ld and line [%s]",
349 /* try to figure out the request characteristics as soon as possible, but
351 if((req->testno == DOCNUMBER_NOTHING) &&
353 "%" REQUEST_KEYWORD_SIZE_TXT"s %" MAXDOCNAMELEN_TXT "s %4s/%d.%d",
361 if(!strcmp(prot_str, "HTTP")) {
362 req->protocol = RPROT_HTTP;
364 else if(!strcmp(prot_str, "RTSP")) {
365 req->protocol = RPROT_RTSP;
368 req->protocol = RPROT_NONE;
369 logmsg("got unknown protocol %s", prot_str);
373 req->prot_version = prot_major*10 + prot_minor;
375 /* find the last slash */
376 ptr = strrchr(doc, '/');
378 /* get the number after it */
383 if((strlen(doc) + strlen(request)) < 200)
384 snprintf(logbuf, sizeof(logbuf), "Got request: %s %s %s/%d.%d",
385 request, doc, prot_str, prot_major, prot_minor);
387 snprintf(logbuf, sizeof(logbuf), "Got a *HUGE* request %s/%d.%d",
388 prot_str, prot_major, prot_minor);
389 logmsg("%s", logbuf);
391 if(!strncmp("/verifiedserver", ptr, 15)) {
392 logmsg("Are-we-friendly question received");
393 req->testno = DOCNUMBER_WERULEZ;
397 if(!strncmp("/quit", ptr, 5)) {
398 logmsg("Request-to-quit received");
399 req->testno = DOCNUMBER_QUIT;
403 ptr++; /* skip the slash */
405 /* skip all non-numericals following the slash */
406 while(*ptr && !ISDIGIT(*ptr))
409 req->testno = strtol(ptr, &ptr, 10);
411 if(req->testno > 10000) {
412 req->partno = req->testno % 10000;
413 req->testno /= 10000;
418 snprintf(logbuf, sizeof(logbuf), "Requested test number %ld part %ld",
419 req->testno, req->partno);
420 logmsg("%s", logbuf);
422 filename = test2file(req->testno);
424 stream=fopen(filename, "rb");
427 logmsg("fopen() failed with error: %d %s", error, strerror(error));
428 logmsg("Error opening file: %s", filename);
429 logmsg("Couldn't open test file %ld", req->testno);
430 req->open = FALSE; /* closes connection */
442 char *rtp_scratch = NULL;
444 /* get the custom server control "commands" */
445 error = getpart(&cmd, &cmdsize, "reply", "servercmd", stream);
448 logmsg("getpart() failed with error: %d", error);
449 req->open = FALSE; /* closes connection */
455 logmsg("Found a reply-servercmd section!");
457 if(!strncmp(CMD_AUTH_REQUIRED, ptr, strlen(CMD_AUTH_REQUIRED))) {
458 logmsg("instructed to require authorization header");
459 req->auth_req = TRUE;
461 else if(!strncmp(CMD_IDLE, ptr, strlen(CMD_IDLE))) {
462 logmsg("instructed to idle");
463 req->rcmd = RCMD_IDLE;
466 else if(!strncmp(CMD_STREAM, ptr, strlen(CMD_STREAM))) {
467 logmsg("instructed to stream");
468 req->rcmd = RCMD_STREAM;
470 else if(1 == sscanf(ptr, "pipe: %d", &num)) {
471 logmsg("instructed to allow a pipe size of %d", num);
473 logmsg("negative pipe size ignored");
475 req->pipe = num-1; /* decrease by one since we don't count the
476 first request in this number */
478 else if(1 == sscanf(ptr, "skip: %d", &num)) {
479 logmsg("instructed to skip this number of bytes %d", num);
482 else if(3 == sscanf(ptr, "rtp: part %d channel %d size %d",
483 &rtp_partno, &rtp_channel, &rtp_size)) {
485 if(rtp_partno == req->partno) {
486 logmsg("RTP: part %d channel %d size %d",
487 rtp_partno, rtp_channel, rtp_size);
489 /* Make our scratch buffer enough to fit all the
490 * desired data and one for padding */
491 rtp_scratch = malloc(rtp_size + 4 + RTP_DATA_SIZE);
493 /* RTP is signalled with a $ */
494 rtp_scratch[0] = '$';
496 /* The channel follows and is one byte */
497 SET_RTP_PKT_CHN(rtp_scratch, rtp_channel);
499 /* Length follows and is a two byte short in network order */
500 SET_RTP_PKT_LEN(rtp_scratch, rtp_size);
502 /* Fill it with junk data */
503 for(i = 0; i < rtp_size; i+= RTP_DATA_SIZE) {
504 memcpy(rtp_scratch + 4 + i, RTP_DATA, RTP_DATA_SIZE);
507 if(req->rtp_buffer == NULL) {
508 req->rtp_buffer = rtp_scratch;
509 req->rtp_buffersize = rtp_size + 4;
512 req->rtp_buffer = realloc(req->rtp_buffer,
513 req->rtp_buffersize +
515 memcpy(req->rtp_buffer + req->rtp_buffersize, rtp_scratch,
517 req->rtp_buffersize += rtp_size + 4;
520 logmsg("rtp_buffersize is %zu, rtp_size is %d.",
521 req->rtp_buffersize, rtp_size);
525 logmsg("funny instruction found: %s", ptr);
528 ptr = strchr(ptr, '\n');
533 } while(ptr && *ptr);
534 logmsg("Done parsing server commands");
540 if(sscanf(req->reqbuf, "CONNECT %" MAXDOCNAMELEN_TXT "s HTTP/%d.%d",
541 doc, &prot_major, &prot_minor) == 3) {
542 snprintf(logbuf, sizeof(logbuf),
543 "Received a CONNECT %s HTTP/%d.%d request",
544 doc, prot_major, prot_minor);
545 logmsg("%s", logbuf);
547 if(req->prot_version == 10)
548 req->open = FALSE; /* HTTP 1.0 closes connection by default */
550 if(!strncmp(doc, "bad", 3))
551 /* if the host name starts with bad, we fake an error here */
552 req->testno = DOCNUMBER_BADCONNECT;
553 else if(!strncmp(doc, "test", 4)) {
554 /* if the host name starts with test, the port number used in the
555 CONNECT line will be used as test number! */
556 char *portp = strchr(doc, ':');
557 if(portp && (*(portp+1) != '\0') && ISDIGIT(*(portp+1)))
558 req->testno = strtol(portp+1, NULL, 10);
560 req->testno = DOCNUMBER_CONNECT;
563 req->testno = DOCNUMBER_CONNECT;
566 logmsg("Did not find test number in PATH");
567 req->testno = DOCNUMBER_404;
573 /* we don't have a complete request yet! */
574 logmsg("ProcessRequest returned without a complete request");
575 return 0; /* not complete yet */
577 logmsg("ProcessRequest found a complete request");
580 /* we do have a full set, advance the checkindex to after the end of the
581 headers, for the pipelining case mostly */
582 req->checkindex += (end - line) + strlen(END_OF_HEADERS);
584 /* **** Persistence ****
586 * If the request is a HTTP/1.0 one, we close the connection unconditionally
589 * If the request is a HTTP/1.1 one, we MUST check for a "Connection:"
590 * header that might say "close". If it does, we close a connection when
591 * this request is processed. Otherwise, we keep the connection alive for X
599 if((req->cl==0) && strncasecompare("Content-Length:", line, 15)) {
600 /* If we don't ignore content-length, we read it and we read the whole
601 request including the body before we return. If we've been told to
602 ignore the content-length, we will return as soon as all headers
603 have been received */
605 char *ptr = line + 15;
606 unsigned long clen = 0;
607 while(*ptr && ISSPACE(*ptr))
611 clen = strtoul(ptr, &endptr, 10);
612 if((ptr == endptr) || !ISSPACE(*endptr) || (ERANGE == errno)) {
613 /* this assumes that a zero Content-Length is valid */
614 logmsg("Found invalid Content-Length: (%s) in the request", ptr);
615 req->open = FALSE; /* closes connection */
618 req->cl = clen - req->skip;
620 logmsg("Found Content-Length: %lu in the request", clen);
622 logmsg("... but will abort after %zu bytes", req->cl);
625 else if(strncasecompare("Transfer-Encoding: chunked", line,
626 strlen("Transfer-Encoding: chunked"))) {
627 /* chunked data coming in */
632 if(strstr(req->reqbuf, "\r\n0\r\n\r\n"))
633 /* end of chunks reached */
636 return 0; /* not done */
639 line = strchr(line, '\n');
645 if(!req->auth && strstr(req->reqbuf, "Authorization:")) {
646 req->auth = TRUE; /* Authorization: header present! */
648 logmsg("Authorization header found, as required");
651 if(!req->digest && strstr(req->reqbuf, "Authorization: Digest")) {
652 /* If the client is passing this Digest-header, we set the part number
653 to 1000. Not only to spice up the complexity of this, but to make
654 Digest stuff to work in the test suite. */
656 req->digest = TRUE; /* header found */
657 logmsg("Received Digest request, sending back data %ld", req->partno);
659 else if(!req->ntlm &&
660 strstr(req->reqbuf, "Authorization: NTLM TlRMTVNTUAAD")) {
661 /* If the client is passing this type-3 NTLM header */
663 req->ntlm = TRUE; /* NTLM found */
664 logmsg("Received NTLM type-3, sending back data %ld", req->partno);
666 logmsg(" Expecting %zu POSTed bytes", req->cl);
669 else if(!req->ntlm &&
670 strstr(req->reqbuf, "Authorization: NTLM TlRMTVNTUAAB")) {
671 /* If the client is passing this type-1 NTLM header */
673 req->ntlm = TRUE; /* NTLM found */
674 logmsg("Received NTLM type-1, sending back data %ld", req->partno);
676 else if((req->partno >= 1000) &&
677 strstr(req->reqbuf, "Authorization: Basic")) {
678 /* If the client is passing this Basic-header and the part number is
679 already >=1000, we add 1 to the part number. This allows simple Basic
680 authentication negotiation to work in the test suite. */
682 logmsg("Received Basic request, sending back data %ld", req->partno);
684 if(strstr(req->reqbuf, "Connection: close"))
685 req->open = FALSE; /* close connection after this request */
689 req->prot_version >= 11 &&
691 req->reqbuf + req->offset > end + strlen(END_OF_HEADERS) &&
692 (!strncmp(req->reqbuf, "GET", strlen("GET")) ||
693 !strncmp(req->reqbuf, "HEAD", strlen("HEAD")))) {
694 /* If we have a persistent connection, HTTP version >= 1.1
695 and GET/HEAD request, enable pipelining. */
696 req->checkindex = (end - req->reqbuf) + strlen(END_OF_HEADERS);
697 req->pipelining = TRUE;
703 /* scan for more header ends within this chunk */
704 line = &req->reqbuf[req->checkindex];
705 end = strstr(line, END_OF_HEADERS);
708 req->checkindex += (end - line) + strlen(END_OF_HEADERS);
712 /* If authentication is required and no auth was provided, end now. This
713 makes the server NOT wait for PUT/POST data and you can then make the
714 test case send a rejection before any such data has been sent. Test case
716 if(req->auth_req && !req->auth)
720 if(req->cl <= req->offset - (end - req->reqbuf) - strlen(END_OF_HEADERS))
723 return 0; /* not complete yet */
729 /* store the entire request in a file */
730 static void storerequest(char *reqbuf, size_t totalsize)
744 dump = fopen(REQUEST_DUMP, "ab");
745 } while((dump == NULL) && ((error = errno) == EINTR));
747 logmsg("Error opening file %s error: %d %s",
748 REQUEST_DUMP, error, strerror(error));
749 logmsg("Failed to write request input to " REQUEST_DUMP);
753 writeleft = totalsize;
755 written = fwrite(&reqbuf[totalsize-writeleft],
758 goto storerequest_cleanup;
760 writeleft -= written;
761 } while((writeleft > 0) && ((error = errno) == EINTR));
764 logmsg("Wrote request (%zu bytes) input to " REQUEST_DUMP, totalsize);
765 else if(writeleft > 0) {
766 logmsg("Error writing file %s error: %d %s",
767 REQUEST_DUMP, error, strerror(error));
768 logmsg("Wrote only (%zu bytes) of (%zu bytes) request input to %s",
769 totalsize-writeleft, totalsize, REQUEST_DUMP);
772 storerequest_cleanup:
776 } while(res && ((error = errno) == EINTR));
778 logmsg("Error closing file %s error: %d %s",
779 REQUEST_DUMP, error, strerror(error));
782 /* return 0 on success, non-zero on failure */
783 static int get_request(curl_socket_t sock, struct httprequest *req)
787 int done_processing = 0;
788 char *reqbuf = req->reqbuf;
791 char *pipereq = NULL;
792 size_t pipereq_length = 0;
794 if(req->pipelining) {
795 pipereq = reqbuf + req->checkindex;
796 pipereq_length = req->offset - req->checkindex;
799 /*** Init the httprequest structure properly for the upcoming request ***/
803 req->testno = DOCNUMBER_NOTHING;
806 req->auth_req = FALSE;
813 req->rcmd = RCMD_NORMALREQ;
814 req->protocol = RPROT_NONE;
815 req->prot_version = 0;
816 req->pipelining = FALSE;
817 req->rtp_buffer = NULL;
818 req->rtp_buffersize = 0;
820 /*** end of httprequest init ***/
822 while(!done_processing && (req->offset < REQBUFSIZ-1)) {
823 if(pipereq_length && pipereq) {
824 memmove(reqbuf, pipereq, pipereq_length);
825 got = curlx_uztosz(pipereq_length);
830 /* we are instructed to not read the entire thing, so we make sure to
831 only read what we're supposed to and NOT read the enire thing the
832 client wants to send! */
833 got = sread(sock, reqbuf + req->offset, req->cl);
835 got = sread(sock, reqbuf + req->offset, REQBUFSIZ-1 - req->offset);
840 logmsg("Connection closed by client");
845 logmsg("recv() returned error: (%d) %s", error, strerror(error));
849 /* dump the request received so far to the external file */
850 reqbuf[req->offset] = '\0';
851 storerequest(reqbuf, req->offset);
855 logmsg("Read %zd bytes", got);
857 req->offset += (size_t)got;
858 reqbuf[req->offset] = '\0';
860 done_processing = ProcessRequest(req);
863 if(done_processing && req->pipe) {
864 logmsg("Waiting for another piped request");
870 if((req->offset == REQBUFSIZ-1) && (got > 0)) {
871 logmsg("Request would overflow buffer, closing connection");
872 /* dump request received so far to external file anyway */
873 reqbuf[REQBUFSIZ-1] = '\0';
876 else if(req->offset > REQBUFSIZ-1) {
877 logmsg("Request buffer overflow, closing connection");
878 /* dump request received so far to external file anyway */
879 reqbuf[REQBUFSIZ-1] = '\0';
883 reqbuf[req->offset] = '\0';
885 /* dump the request to an external file */
886 storerequest(reqbuf, req->pipelining ? req->checkindex : req->offset);
890 return fail; /* return 0 on success */
893 /* returns -1 on failure */
894 static int send_doc(curl_socket_t sock, struct httprequest *req)
904 bool persistant = TRUE;
905 bool sendfailure = FALSE;
910 static char weare[256];
912 char partbuf[80]="data";
914 logmsg("Send response number %ld part %ld", req->testno, req->partno);
919 break; /* continue with business as usual */
921 #define STREAMTHIS "a string to stream 01234567890\n"
922 count = strlen(STREAMTHIS);
924 written = swrite(sock, STREAMTHIS, count);
927 if(written != (ssize_t)count) {
928 logmsg("Stopped streaming");
934 /* Do nothing. Sit idle. Pretend it rains. */
940 if(req->testno < 0) {
944 switch(req->testno) {
946 logmsg("Replying to QUIT");
949 case DOCNUMBER_WERULEZ:
950 /* we got a "friends?" question, reply back that we sure are */
951 logmsg("Identifying ourselves as friends");
952 snprintf(msgbuf, sizeof(msgbuf), "RTSP_SERVER WE ROOLZ: %ld\r\n",
954 msglen = strlen(msgbuf);
955 snprintf(weare, sizeof(weare),
956 "HTTP/1.1 200 OK\r\nContent-Length: %zu\r\n\r\n%s",
960 case DOCNUMBER_INTERNAL:
961 logmsg("Bailing out due to internal error");
963 case DOCNUMBER_CONNECT:
964 logmsg("Replying to CONNECT");
967 case DOCNUMBER_BADCONNECT:
968 logmsg("Replying to a bad CONNECT");
969 buffer = docbadconnect;
973 logmsg("Replying to with a 404");
974 if(req->protocol == RPROT_HTTP) {
975 buffer = doc404_HTTP;
978 buffer = doc404_RTSP;
983 count = strlen(buffer);
986 char *filename = test2file(req->testno);
989 snprintf(partbuf, sizeof(partbuf), "data%ld", req->partno);
991 stream=fopen(filename, "rb");
994 logmsg("fopen() failed with error: %d %s", error, strerror(error));
995 logmsg("Error opening file: %s", filename);
996 logmsg("Couldn't open test file");
1000 error = getpart(&ptr, &count, "reply", partbuf, stream);
1003 logmsg("getpart() failed with error: %d", error);
1009 if(got_exit_signal) {
1014 /* re-open the same file again */
1015 stream=fopen(filename, "rb");
1018 logmsg("fopen() failed with error: %d %s", error, strerror(error));
1019 logmsg("Error opening file: %s", filename);
1020 logmsg("Couldn't open test file");
1025 /* get the custom server control "commands" */
1026 error = getpart(&cmd, &cmdsize, "reply", "postcmd", stream);
1029 logmsg("getpart() failed with error: %d", error);
1036 if(got_exit_signal) {
1042 /* If the word 'swsclose' is present anywhere in the reply chunk, the
1043 connection will be closed after the data has been sent to the requesting
1045 if(strstr(buffer, "swsclose") || !count) {
1047 logmsg("connection close instruction \"swsclose\" found in response");
1049 if(strstr(buffer, "swsbounce")) {
1051 logmsg("enable \"swsbounce\" in the next request");
1056 dump = fopen(RESPONSE_DUMP, "ab");
1059 logmsg("fopen() failed with error: %d %s", error, strerror(error));
1060 logmsg("Error opening file: %s", RESPONSE_DUMP);
1061 logmsg("couldn't create logfile: " RESPONSE_DUMP);
1067 responsesize = count;
1069 /* Ok, we send no more than 200 bytes at a time, just to make sure that
1070 larger chunks are split up so that the client will need to do multiple
1071 recv() calls to get it and thus we exercise that code better */
1075 written = swrite(sock, buffer, num);
1081 logmsg("Sent off %zd bytes", written);
1083 /* write to file as well */
1084 fwrite(buffer, 1, (size_t)written, dump);
1092 /* Send out any RTP data */
1093 if(req->rtp_buffer) {
1094 logmsg("About to write %zu RTP bytes", req->rtp_buffersize);
1095 count = req->rtp_buffersize;
1100 written = swrite(sock, req->rtp_buffer + (req->rtp_buffersize - count),
1109 free(req->rtp_buffer);
1110 req->rtp_buffersize = 0;
1115 } while(res && ((error = errno) == EINTR));
1117 logmsg("Error closing file %s error: %d %s",
1118 RESPONSE_DUMP, error, strerror(error));
1120 if(got_exit_signal) {
1127 logmsg("Sending response failed. Only (%zu bytes) of "
1128 "(%zu bytes) were sent",
1129 responsesize-count, responsesize);
1135 logmsg("Response sent (%zu bytes) and written to " RESPONSE_DUMP,
1145 if(2 == sscanf(ptr, "%31s %d", command, &num)) {
1146 if(!strcmp("wait", command)) {
1147 logmsg("Told to sleep for %d seconds", num);
1149 while(quarters > 0) {
1155 /* should not happen */
1157 logmsg("wait_ms() failed with error: (%d) %s",
1158 error, strerror(error));
1163 logmsg("Continuing after sleeping %d seconds", num);
1166 logmsg("Unknown command in reply command section");
1168 ptr = strchr(ptr, '\n');
1173 } while(ptr && *ptr);
1176 req->open = persistant;
1178 prevtestno = req->testno;
1179 prevpartno = req->partno;
1185 int main(int argc, char *argv[])
1187 srvr_sockaddr_union_t me;
1188 curl_socket_t sock = CURL_SOCKET_BAD;
1189 curl_socket_t msgsock = CURL_SOCKET_BAD;
1190 int wrotepidfile = 0;
1192 unsigned short port = DEFAULT_PORT;
1193 char *pidname= (char *)".rtsp.pid";
1194 struct httprequest req;
1201 if(!strcmp("--version", argv[arg])) {
1202 printf("rtspd IPv4%s"
1213 else if(!strcmp("--pidfile", argv[arg])) {
1216 pidname = argv[arg++];
1218 else if(!strcmp("--logfile", argv[arg])) {
1221 serverlogfile = argv[arg++];
1223 else if(!strcmp("--ipv4", argv[arg])) {
1230 else if(!strcmp("--ipv6", argv[arg])) {
1237 else if(!strcmp("--port", argv[arg])) {
1241 unsigned long ulnum = strtoul(argv[arg], &endptr, 10);
1242 if((endptr != argv[arg] + strlen(argv[arg])) ||
1243 (ulnum < 1025UL) || (ulnum > 65535UL)) {
1244 fprintf(stderr, "rtspd: invalid --port argument (%s)\n",
1248 port = curlx_ultous(ulnum);
1252 else if(!strcmp("--srcdir", argv[arg])) {
1260 puts("Usage: rtspd [option]\n"
1262 " --logfile [file]\n"
1263 " --pidfile [file]\n"
1267 " --srcdir [path]");
1274 atexit(win32_cleanup);
1277 install_signal_handlers();
1279 pid = (long)getpid();
1284 sock = socket(AF_INET, SOCK_STREAM, 0);
1287 sock = socket(AF_INET6, SOCK_STREAM, 0);
1290 if(CURL_SOCKET_BAD == sock) {
1292 logmsg("Error creating socket: (%d) %s",
1293 error, strerror(error));
1294 goto server_cleanup;
1298 if(0 != setsockopt(sock, SOL_SOCKET, SO_REUSEADDR,
1299 (void *)&flag, sizeof(flag))) {
1301 logmsg("setsockopt(SO_REUSEADDR) failed with error: (%d) %s",
1302 error, strerror(error));
1303 goto server_cleanup;
1309 memset(&me.sa4, 0, sizeof(me.sa4));
1310 me.sa4.sin_family = AF_INET;
1311 me.sa4.sin_addr.s_addr = INADDR_ANY;
1312 me.sa4.sin_port = htons(port);
1313 rc = bind(sock, &me.sa, sizeof(me.sa4));
1317 memset(&me.sa6, 0, sizeof(me.sa6));
1318 me.sa6.sin6_family = AF_INET6;
1319 me.sa6.sin6_addr = in6addr_any;
1320 me.sa6.sin6_port = htons(port);
1321 rc = bind(sock, &me.sa, sizeof(me.sa6));
1323 #endif /* ENABLE_IPV6 */
1326 logmsg("Error binding socket on port %hu: (%d) %s",
1327 port, error, strerror(error));
1328 goto server_cleanup;
1331 logmsg("Running %s version on port %d", ipv_inuse, (int)port);
1333 /* start accepting connections */
1334 rc = listen(sock, 5);
1337 logmsg("listen() failed with error: (%d) %s",
1338 error, strerror(error));
1339 goto server_cleanup;
1343 ** As soon as this server writes its pid file the test harness will
1344 ** attempt to connect to this server and initiate its verification.
1347 wrotepidfile = write_pidfile(pidname);
1349 goto server_cleanup;
1352 msgsock = accept(sock, NULL, NULL);
1356 if(CURL_SOCKET_BAD == msgsock) {
1358 logmsg("MAJOR ERROR: accept() failed with error: (%d) %s",
1359 error, strerror(error));
1364 ** As soon as this server acepts a connection from the test harness it
1365 ** must set the server logs advisor read lock to indicate that server
1366 ** logs should not be read until this lock is removed by this server.
1369 set_advisor_read_lock(SERVERLOGS_LOCK);
1370 serverlogslocked = 1;
1372 logmsg("====> Client connect");
1376 * Disable the Nagle algorithm to make it easier to send out a large
1377 * response in many small segments to torture the clients more.
1380 if(setsockopt(msgsock, IPPROTO_TCP, TCP_NODELAY,
1381 (void *)&flag, sizeof(flag)) == -1) {
1382 logmsg("====> TCP_NODELAY failed");
1386 /* initialization of httprequest struct is done in get_request(), but due
1387 to pipelining treatment the pipelining struct field must be initialized
1388 previously to FALSE every time a new connection arrives. */
1390 req.pipelining = FALSE;
1396 if(get_request(msgsock, &req))
1397 /* non-zero means error, break out of loop */
1401 /* bounce treatment requested */
1402 if((req.testno == prevtestno) &&
1403 (req.partno == prevpartno)) {
1405 logmsg("BOUNCE part number to %ld", req.partno);
1414 send_doc(msgsock, &req);
1418 if((req.testno < 0) && (req.testno != DOCNUMBER_CONNECT)) {
1419 logmsg("special request received, no persistency");
1423 logmsg("instructed to close connection after server-reply");
1428 logmsg("=> persistant connection request ended, awaits new request");
1429 /* if we got a CONNECT, loop and get another request as well! */
1430 } while(req.open || (req.testno == DOCNUMBER_CONNECT));
1435 logmsg("====> Client disconnect");
1437 msgsock = CURL_SOCKET_BAD;
1439 if(serverlogslocked) {
1440 serverlogslocked = 0;
1441 clear_advisor_read_lock(SERVERLOGS_LOCK);
1444 if(req.testno == DOCNUMBER_QUIT)
1450 if((msgsock != sock) && (msgsock != CURL_SOCKET_BAD))
1453 if(sock != CURL_SOCKET_BAD)
1457 logmsg("signalled to die");
1462 if(serverlogslocked) {
1463 serverlogslocked = 0;
1464 clear_advisor_read_lock(SERVERLOGS_LOCK);
1467 restore_signal_handlers();
1469 if(got_exit_signal) {
1470 logmsg("========> %s rtspd (port: %d pid: %ld) exits with signal (%d)",
1471 ipv_inuse, (int)port, pid, exit_signal);
1473 * To properly set the return status of the process we
1474 * must raise the same signal SIGINT or SIGTERM that we
1475 * caught and let the old handler take care of it.
1480 logmsg("========> rtspd quits");