unbound/testcode/petal.c

   1 /*
   2  * petal.c - https daemon that is small and beautiful.
   3  *
   4  * Copyright (c) 2010, NLnet Labs. All rights reserved.
   5  *
   6  * This software is open source.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  *
  12  * Redistributions of source code must retain the above copyright notice,
  13  * this list of conditions and the following disclaimer.
  14  *
  15  * Redistributions in binary form must reproduce the above copyright notice,
  16  * this list of conditions and the following disclaimer in the documentation
  17  * and/or other materials provided with the distribution.
  18  *
  19  * Neither the name of the NLNET LABS nor the names of its contributors may
  20  * be used to endorse or promote products derived from this software without
  21  * specific prior written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  24  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  25  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  26  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  27  * HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  28  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
  29  * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  30  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  31  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  32  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  33  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  34  */
  35
  36 /**
  37  * \file
  38  *
  39  * HTTP1.1/SSL server.
  40  */
  41
  42 #include "config.h"
  43 #ifdef HAVE_GETOPT_H
  44 #include <getopt.h>
  45 #endif
  46 #ifdef HAVE_OPENSSL_SSL_H
  47 #include <openssl/ssl.h>
  48 #endif
  49 #ifdef HAVE_OPENSSL_ERR_H
  50 #include <openssl/err.h>
  51 #endif
  52 #ifdef HAVE_OPENSSL_RAND_H
  53 #include <openssl/rand.h>
  54 #endif
  55 #include <openssl/x509.h>
  56 #include <openssl/pem.h>
  57 #include <ctype.h>
  58 #include <signal.h>
  59 #if defined(UNBOUND_ALLOC_LITE) || defined(UNBOUND_ALLOC_STATS)
  60 #ifdef malloc
  61 #undef malloc
  62 #endif
  63 #ifdef free
  64 #undef free
  65 #endif
  66 #endif /* alloc lite or alloc stats */
  67
  68 /** verbosity for this application */
  69 static int verb = 0;
  70
  71 /** Give petal usage, and exit (1). */
  72 static void
  73 usage()
  74 {
  75         printf("Usage:  petal [opts]\n");
  76         printf("        https daemon serves files from ./'host'/filename\n");
  77         printf("        (no hostname: from the 'default' directory)\n");
  78         printf("-a addr         bind to this address, 127.0.0.1\n");
  79         printf("-p port         port number, default 443\n");
  80         printf("-k keyfile      SSL private key file (PEM), petal.key\n");
  81         printf("-c certfile     SSL certificate file (PEM), petal.pem\n");
  82         printf("-v              more verbose\n");
  83         printf("-h              show this usage help\n");
  84         printf("Version %s\n", PACKAGE_VERSION);
  85         printf("BSD licensed, see LICENSE in source package for details.\n");
  86         printf("Report bugs to %s\n", PACKAGE_BUGREPORT);
  87         exit(1);
  88 }
  89
  90 /** fatal exit */
  91 static void print_exit(const char* str) {printf("error %s\n", str); exit(1);}
  92 /** print errno */
  93 static void log_errno(const char* str)
  94 {printf("error %s: %s\n", str, strerror(errno));}
  95
  96 /** parse a text IP address into a sockaddr */
  97 static int
  98 parse_ip_addr(char* str, int port, struct sockaddr_storage* ret, socklen_t* l)
  99 {
 100         socklen_t len = 0;
 101         struct sockaddr_storage* addr = NULL;
 102         struct sockaddr_in6 a6;
 103         struct sockaddr_in a;
 104         uint16_t p = (uint16_t)port;
 105         int fam = 0;
 106         memset(&a6, 0, sizeof(a6));
 107         memset(&a, 0, sizeof(a));
 108
 109         if(inet_pton(AF_INET6, str, &a6.sin6_addr) > 0) {
 110                 /* it is an IPv6 */
 111                 fam = AF_INET6;
 112                 a6.sin6_family = AF_INET6;
 113                 a6.sin6_port = (in_port_t)htons(p);
 114                 addr = (struct sockaddr_storage*)&a6;
 115                 len = (socklen_t)sizeof(struct sockaddr_in6);
 116         }
 117         if(inet_pton(AF_INET, str, &a.sin_addr) > 0) {
 118                 /* it is an IPv4 */
 119                 fam = AF_INET;
 120                 a.sin_family = AF_INET;
 121                 a.sin_port = (in_port_t)htons(p);
 122                 addr = (struct sockaddr_storage*)&a;
 123                 len = (socklen_t)sizeof(struct sockaddr_in);
 124         }
 125         if(!len) print_exit("cannot parse addr");
 126         *l = len;
 127         memmove(ret, addr, len);
 128         return fam;
 129 }
 130
 131 /** close the fd */
 132 static void
 133 fd_close(int fd)
 134 {
 135 #ifndef USE_WINSOCK
 136         close(fd);
 137 #else
 138         closesocket(fd);
 139 #endif
 140 }
 141
 142 /**
 143  * Read one line from SSL
 144  * zero terminates.
 145  * skips "\r\n" (but not copied to buf).
 146  * @param ssl: the SSL connection to read from (blocking).
 147  * @param buf: buffer to return line in.
 148  * @param len: size of the buffer.
 149  * @return 0 on error, 1 on success.
 150  */
 151 static int
 152 read_ssl_line(SSL* ssl, char* buf, size_t len)
 153 {
 154         size_t n = 0;
 155         int r;
 156         int endnl = 0;
 157         while(1) {
 158                 if(n >= len) {
 159                         if(verb) printf("line too long\n");
 160                         return 0;
 161                 }
 162                 if((r = SSL_read(ssl, buf+n, 1)) <= 0) {
 163                         if(SSL_get_error(ssl, r) == SSL_ERROR_ZERO_RETURN) {
 164                                 /* EOF */
 165                                 break;
 166                         }
 167                         if(verb) printf("could not SSL_read\n");
 168                         return 0;
 169                 }
 170                 if(endnl && buf[n] == '\n') {
 171                         break;
 172                 } else if(endnl) {
 173                         /* bad data */
 174                         if(verb) printf("error: stray linefeeds\n");
 175                         return 0;
 176                 } else if(buf[n] == '\r') {
 177                         /* skip \r, and also \n on the wire */
 178                         endnl = 1;
 179                         continue;
 180                 } else if(buf[n] == '\n') {
 181                         /* skip the \n, we are done */
 182                         break;
 183                 } else n++;
 184         }
 185         buf[n] = 0;
 186         return 1;
 187 }
 188
 189 /** process one http header */
 190 static int
 191 process_one_header(char* buf, char* file, size_t flen, char* host, size_t hlen,
 192         int* vs)
 193 {
 194         if(strncasecmp(buf, "GET ", 4) == 0) {
 195                 char* e = strstr(buf, " HTTP/1.1");
 196                 if(!e) e = strstr(buf, " http/1.1");
 197                 if(!e) {
 198                         e = strstr(buf, " HTTP/1.0");
 199                         if(!e) e = strstr(buf, " http/1.0");
 200                         if(!e) e = strrchr(buf, ' ');
 201                         if(!e) e = strrchr(buf, '\t');
 202                         if(e) *vs = 10;
 203                 }
 204                 if(e) *e = 0;
 205                 if(strlen(buf) < 4) return 0;
 206                 (void)strlcpy(file, buf+4, flen);
 207         } else if(strncasecmp(buf, "Host: ", 6) == 0) {
 208                 (void)strlcpy(host, buf+6, hlen);
 209         }
 210         return 1;
 211 }
 212
 213 /** read http headers and process them */
 214 static int
 215 read_http_headers(SSL* ssl, char* file, size_t flen, char* host, size_t hlen,
 216         int* vs)
 217 {
 218         char buf[1024];
 219         file[0] = 0;
 220         host[0] = 0;
 221         while(read_ssl_line(ssl, buf, sizeof(buf))) {
 222                 if(verb>=2) printf("read: %s\n", buf);
 223                 if(buf[0] == 0)
 224                         return 1;
 225                 if(!process_one_header(buf, file, flen, host, hlen, vs))
 226                         return 0;
 227         }
 228         return 0;
 229 }
 230
 231 /** setup SSL context */
 232 static SSL_CTX*
 233 setup_ctx(char* key, char* cert)
 234 {
 235         SSL_CTX* ctx = SSL_CTX_new(SSLv23_server_method());
 236         if(!ctx) print_exit("out of memory");
 237         (void)SSL_CTX_set_options(ctx, SSL_OP_NO_SSLv2);
 238         (void)SSL_CTX_set_options(ctx, SSL_OP_NO_SSLv3);
 239         if(!SSL_CTX_use_certificate_file(ctx, cert, SSL_FILETYPE_PEM))
 240                 print_exit("cannot read cert");
 241         if(!SSL_CTX_use_PrivateKey_file(ctx, key, SSL_FILETYPE_PEM))
 242                 print_exit("cannot read key");
 243         if(!SSL_CTX_check_private_key(ctx))
 244                 print_exit("private key is not correct");
 245         if(!SSL_CTX_load_verify_locations(ctx, cert, NULL))
 246                 print_exit("cannot load cert verify locations");
 247         return ctx;
 248 }
 249
 250 /** setup listening TCP */
 251 static int
 252 setup_fd(char* addr, int port)
 253 {
 254         struct sockaddr_storage ad;
 255         socklen_t len;
 256         int fd;
 257         int c = 1;
 258         int fam = parse_ip_addr(addr, port, &ad, &len);
 259         fd = socket(fam, SOCK_STREAM, 0);
 260         if(fd == -1) {
 261                 log_errno("socket");
 262                 return -1;
 263         }
 264         if(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR,
 265                 (void*)&c, (socklen_t) sizeof(int)) < 0) {
 266                 log_errno("setsockopt(SOL_SOCKET, SO_REUSEADDR)");
 267         }
 268         if(bind(fd, (struct sockaddr*)&ad, len) == -1) {
 269                 log_errno("bind");
 270                 fd_close(fd);
 271                 return -1;
 272         }
 273         if(listen(fd, 5) == -1) {
 274                 log_errno("listen");
 275                 fd_close(fd);
 276                 return -1;
 277         }
 278         return fd;
 279 }
 280
 281 /** setup SSL connection to the client */
 282 static SSL*
 283 setup_ssl(int s, SSL_CTX* ctx)
 284 {
 285         SSL* ssl = SSL_new(ctx);
 286         if(!ssl) return NULL;
 287         SSL_set_accept_state(ssl);
 288         (void)SSL_set_mode(ssl, SSL_MODE_AUTO_RETRY);
 289         if(!SSL_set_fd(ssl, s)) {
 290                 SSL_free(ssl);
 291                 return NULL;
 292         }
 293         return ssl;
 294 }
 295
 296 /** check a file name for safety */
 297 static int
 298 file_name_is_safe(char* s)
 299 {
 300         size_t l = strlen(s);
 301         if(s[0] != '/')
 302                 return 0; /* must start with / */
 303         if(strstr(s, "/../"))
 304                 return 0; /* no updirs in URL */
 305         if(l>=3 && s[l-1]=='.' && s[l-2]=='.' && s[l-3]=='/')
 306                 return 0; /* ends with /.. */
 307         return 1;
 308 }
 309
 310 /** adjust host and filename */
 311 static void
 312 adjust_host_file(char* host, char* file)
 313 {
 314         size_t i, len;
 315         /* remove a port number if present */
 316         if(strrchr(host, ':'))
 317                 *strrchr(host, ':') = 0;
 318         /* lowercase */
 319         len = strlen(host);
 320         for(i=0; i<len; i++)
 321                 host[i] = tolower((unsigned char)host[i]);
 322         len = strlen(file);
 323         for(i=0; i<len; i++)
 324                 file[i] = tolower((unsigned char)file[i]);
 325 }
 326
 327 /** check a host name for safety */
 328 static int
 329 host_name_is_safe(char* s)
 330 {
 331         if(strchr(s, '/'))
 332                 return 0;
 333         if(strcmp(s, "..") == 0)
 334                 return 0;
 335         if(strcmp(s, ".") == 0)
 336                 return 0;
 337         return 1;
 338 }
 339
 340 /** provide file in whole transfer */
 341 static void
 342 provide_file_10(SSL* ssl, char* fname)
 343 {
 344         char* buf, *at;
 345         size_t len, avail, header_reserve=1024;
 346         FILE* in = fopen(fname,
 347 #ifndef USE_WINSOCK
 348                 "r"
 349 #else
 350                 "rb"
 351 #endif
 352                 );
 353         size_t r;
 354         const char* rcode = "200 OK";
 355         if(!in) {
 356                 char hdr[1024];
 357                 rcode = "404 File not found";
 358                 snprintf(hdr, sizeof(hdr), "HTTP/1.1 %s\r\n\r\n", rcode);
 359                 r = strlen(hdr);
 360                 if(SSL_write(ssl, hdr, (int)r) <= 0) {
 361                         /* write failure */
 362                 }
 363                 return;
 364         }
 365         fseek(in, 0, SEEK_END);
 366         len = (size_t)ftell(in);
 367         fseek(in, 0, SEEK_SET);
 368         /* plus some space for the header */
 369         buf = (char*)malloc(len+header_reserve);
 370         if(!buf) {
 371                 fclose(in);
 372                 return;
 373         }
 374         avail = len+header_reserve;
 375         at = buf;
 376         snprintf(at, avail, "HTTP/1.1 %s\r\n", rcode);
 377         r = strlen(at);
 378         at += r;
 379         avail -= r;
 380         snprintf(at, avail, "Server: petal/%s\r\n", PACKAGE_VERSION);
 381         r = strlen(at);
 382         at += r;
 383         avail -= r;
 384         snprintf(at, avail, "Content-Length: %u\r\n", (unsigned)len);
 385         r = strlen(at);
 386         at += r;
 387         avail -= r;
 388         snprintf(at, avail, "\r\n");
 389         r = strlen(at);
 390         at += r;
 391         avail -= r;
 392         if(avail < len) { /* robust */
 393                 free(buf);
 394                 fclose(in);
 395                 return;
 396         }
 397         if(fread(at, 1, len, in) != len) {
 398                 free(buf);
 399                 fclose(in);
 400                 return;
 401         }
 402         fclose(in);
 403         at += len;
 404         avail -= len;
 405         if(SSL_write(ssl, buf, at-buf) <= 0) {
 406                 /* write failure */
 407         }
 408         free(buf);
 409 }
 410
 411 /** provide file over SSL, chunked encoding */
 412 static void
 413 provide_file_chunked(SSL* ssl, char* fname)
 414 {
 415         char buf[16384];
 416         char* at = buf;
 417         size_t avail = sizeof(buf);
 418         size_t r;
 419         FILE* in = fopen(fname,
 420 #ifndef USE_WINSOCK
 421                 "r"
 422 #else
 423                 "rb"
 424 #endif
 425                 );
 426         const char* rcode = "200 OK";
 427         if(!in) {
 428                 rcode = "404 File not found";
 429         }
 430
 431         /* print headers */
 432         snprintf(at, avail, "HTTP/1.1 %s\r\n", rcode);
 433         r = strlen(at);
 434         at += r;
 435         avail -= r;
 436         snprintf(at, avail, "Server: petal/%s\r\n", PACKAGE_VERSION);
 437         r = strlen(at);
 438         at += r;
 439         avail -= r;
 440         snprintf(at, avail, "Transfer-Encoding: chunked\r\n");
 441         r = strlen(at);
 442         at += r;
 443         avail -= r;
 444         snprintf(at, avail, "Connection: close\r\n");
 445         r = strlen(at);
 446         at += r;
 447         avail -= r;
 448         snprintf(at, avail, "\r\n");
 449         r = strlen(at);
 450         at += r;
 451         avail -= r;
 452         if(avail < 16) { /* robust */
 453                 if(in) fclose(in);
 454                 return;
 455         }
 456
 457         do {
 458                 char tmpbuf[sizeof(buf)];
 459                 /* read chunk; space-16 for xxxxCRLF..CRLF0CRLFCRLF (3 spare)*/
 460                 size_t red = in?fread(tmpbuf, 1, avail-16, in):0;
 461                 /* prepare chunk */
 462                 snprintf(at, avail, "%x\r\n", (unsigned)red);
 463                 r = strlen(at);
 464                 if(verb >= 3)
 465                 {printf("chunk len %x\n", (unsigned)red); fflush(stdout);}
 466                 at += r;
 467                 avail -= r;
 468                 if(red != 0) {
 469                         if(red > avail) break; /* robust */
 470                         memmove(at, tmpbuf, red);
 471                         at += red;
 472                         avail -= red;
 473                         snprintf(at, avail, "\r\n");
 474                         r = strlen(at);
 475                         at += r;
 476                         avail -= r;
 477                 }
 478                 if(in && feof(in) && red != 0) {
 479                         snprintf(at, avail, "0\r\n");
 480                         r = strlen(at);
 481                         at += r;
 482                         avail -= r;
 483                 }
 484                 if(!in || feof(in)) {
 485                         snprintf(at, avail, "\r\n");
 486                         r = strlen(at);
 487                         at += r;
 488                         avail -= r;
 489                 }
 490                 /* send chunk */
 491                 if(SSL_write(ssl, buf, at-buf) <= 0) {
 492                         /* SSL error */
 493                         break;
 494                 }
 495
 496                 /* setup for next chunk */
 497                 at = buf;
 498                 avail = sizeof(buf);
 499         } while(in && !feof(in) && !ferror(in));
 500
 501         if(in) fclose(in);
 502 }
 503
 504 /** provide service to the ssl descriptor */
 505 static void
 506 service_ssl(SSL* ssl, struct sockaddr_storage* from, socklen_t falen)
 507 {
 508         char file[1024];
 509         char host[1024];
 510         char combined[2048];
 511         int vs = 11;
 512         if(!read_http_headers(ssl, file, sizeof(file), host, sizeof(host),
 513                 &vs))
 514                 return;
 515         adjust_host_file(host, file);
 516         if(host[0] == 0 || !host_name_is_safe(host))
 517                 (void)strlcpy(host, "default", sizeof(host));
 518         if(!file_name_is_safe(file)) {
 519                 return;
 520         }
 521         snprintf(combined, sizeof(combined), "%s%s", host, file);
 522         if(verb) {
 523                 char out[100];
 524                 void* a = &((struct sockaddr_in*)from)->sin_addr;
 525                 if(falen != (socklen_t)sizeof(struct sockaddr_in))
 526                         a = &((struct sockaddr_in6*)from)->sin6_addr;
 527                 out[0]=0;
 528                 (void)inet_ntop((int)((struct sockaddr_in*)from)->sin_family,
 529                         a, out, (socklen_t)sizeof(out));
 530                 printf("%s requests %s\n", out, combined);
 531                 fflush(stdout);
 532         }
 533         if(vs == 10)
 534                 provide_file_10(ssl, combined);
 535         else    provide_file_chunked(ssl, combined);
 536 }
 537
 538 /** provide ssl service */
 539 static void
 540 do_service(char* addr, int port, char* key, char* cert)
 541 {
 542         SSL_CTX* sslctx = setup_ctx(key, cert);
 543         int fd = setup_fd(addr, port);
 544         int go = 1;
 545         if(fd == -1) print_exit("could not setup sockets");
 546         if(verb) {printf("petal start\n"); fflush(stdout);}
 547         while(go) {
 548                 struct sockaddr_storage from;
 549                 socklen_t flen = (socklen_t)sizeof(from);
 550                 int s = accept(fd, (struct sockaddr*)&from, &flen);
 551                 if(verb) fflush(stdout);
 552                 if(s != -1) {
 553                         SSL* ssl = setup_ssl(s, sslctx);
 554                         if(verb) fflush(stdout);
 555                         if(ssl) {
 556                                 service_ssl(ssl, &from, flen);
 557                                 if(verb) fflush(stdout);
 558                                 SSL_shutdown(ssl);
 559                                 SSL_free(ssl);
 560                         }
 561                         fd_close(s);
 562                 } else if (verb >=2) log_errno("accept");
 563                 if(verb) fflush(stdout);
 564         }
 565         /* if we get a kill signal, the process dies and the OS reaps us */
 566         if(verb) printf("petal end\n");
 567         fd_close(fd);
 568         SSL_CTX_free(sslctx);
 569 }
 570
 571 /** getopt global, in case header files fail to declare it. */
 572 extern int optind;
 573 /** getopt global, in case header files fail to declare it. */
 574 extern char* optarg;
 575
 576 /** Main routine for petal */
 577 int main(int argc, char* argv[])
 578 {
 579         int c;
 580         int port = 443;
 581         char* addr = "127.0.0.1", *key = "petal.key", *cert = "petal.pem";
 582 #ifdef USE_WINSOCK
 583         WSADATA wsa_data;
 584         if((c=WSAStartup(MAKEWORD(2,2), &wsa_data)) != 0)
 585         {       printf("WSAStartup failed\n"); exit(1); }
 586         atexit((void (*)(void))WSACleanup);
 587 #endif
 588
 589         /* parse the options */
 590         while( (c=getopt(argc, argv, "a:c:k:hp:v")) != -1) {
 591                 switch(c) {
 592                 case 'a':
 593                         addr = optarg;
 594                         break;
 595                 case 'c':
 596                         cert = optarg;
 597                         break;
 598                 case 'k':
 599                         key = optarg;
 600                         break;
 601                 case 'p':
 602                         port = atoi(optarg);
 603                         break;
 604                 case 'v':
 605                         verb++;
 606                         break;
 607                 case '?':
 608                 case 'h':
 609                 default:
 610                         usage();
 611                 }
 612         }
 613         argc -= optind;
 614         argv += optind;
 615         if(argc != 0)
 616                 usage();
 617
 618 #ifdef SIGPIPE
 619         (void)signal(SIGPIPE, SIG_IGN);
 620 #endif
 621         ERR_load_crypto_strings();
 622         ERR_load_SSL_strings();
 623         OpenSSL_add_all_algorithms();
 624         (void)SSL_library_init();
 625
 626         do_service(addr, port, key, cert);
 627
 628         CRYPTO_cleanup_all_ex_data();
 629         ERR_remove_state(0);
 630         ERR_free_strings();
 631         RAND_cleanup();
 632         return 0;
 633 }