/*
 * uhttpd - Tiny single-threaded httpd - Utility header
 *
 *   Copyright (C) 2010-2012 Jo-Philipp Wich <xm@subsignal.org>
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

#ifndef _UHTTPD_UTILS_

#include <fcntl.h>
#include <pwd.h>
#include <stdarg.h>
#include <sys/stat.h>

#include "libubox/uloop.h"

#ifdef __APPLE__
static inline void clearenv(void) {
  extern char **environ;
  environ = NULL;
}
#endif

#ifdef HAVE_SHADOW
#include <shadow.h>
#endif

#define min(x, y) (((x) < (y)) ? (x) : (y))
#define max(x, y) (((x) > (y)) ? (x) : (y))

#define array_size(x) (sizeof(x) / sizeof(x[0]))

#define foreach_header(i, h)                                                   \
  for (i = 0; (i + 1) < (sizeof(h) / sizeof(h[0])) && h[i]; i += 2)

#define fd_cloexec(fd) fcntl(fd, F_SETFD, fcntl(fd, F_GETFD) | FD_CLOEXEC)

#define fd_nonblock(fd) fcntl(fd, F_SETFL, fcntl(fd, F_GETFL) | O_NONBLOCK)

#define ensure_out(x)                                                          \
  do {                                                                         \
    if ((x) < 0)                                                               \
      goto out;                                                                \
  } while (0)

#define ensure_ret(x)                                                          \
  do {                                                                         \
    if ((x) < 0)                                                               \
      return -1;                                                               \
  } while (0)

struct path_info {
  char *root;
  char *phys;
  char *name;
  char *info;
  char *query;
  int redirected;
  struct stat stat;
};

const char *sa_straddr(void *sa);
const char *sa_strport(void *sa);
int sa_port(void *sa);
int sa_rfc1918(void *sa);

char *strfind(char *haystack, int hslen, const char *needle, int ndlen);

bool uh_socket_wait(int fd, int sec, bool write);

int uh_raw_send(int fd, const char *buf, int len, int seconds);
int uh_raw_recv(int fd, char *buf, int len, int seconds);
int uh_tcp_send(struct client *cl, const char *buf, int len);
int uh_tcp_send_lowlevel(struct client *cl, const char *buf, int len);
int uh_tcp_recv(struct client *cl, char *buf, int len);
int uh_tcp_recv_lowlevel(struct client *cl, char *buf, int len);

int uh_http_sendhf(struct client *cl, int code, const char *summary,
                   const char *fmt, ...);

#define uh_http_response(cl, code, message)                                    \
  uh_http_sendhf(cl, code, message, message)

int uh_http_sendc(struct client *cl, const char *data, int len);

int uh_http_sendf(struct client *cl, struct http_request *req, const char *fmt,
                  ...);

int uh_http_send(struct client *cl, struct http_request *req, const char *buf,
                 int len);

int uh_urldecode(char *buf, int blen, const char *src, int slen);
int uh_urlencode(char *buf, int blen, const char *src, int slen);
int uh_b64decode(char *buf, int blen, const unsigned char *src, int slen);

struct auth_realm *uh_auth_add(char *path, char *user, char *pass);

int uh_auth_check(struct client *cl, struct http_request *req,
                  struct path_info *pi);

extern struct index_file *uh_index_files;
struct index_file *uh_index_add(const char *filename);

struct path_info *uh_path_lookup(struct client *cl, const char *url);

struct listener *uh_listener_add(int sock, struct config *conf);
struct listener *uh_listener_lookup(int sock);

struct client *uh_client_add(int sock, struct listener *serv,
                             struct sockaddr_in6 *peer);

struct client *uh_client_lookup(int sock);

#define uh_client_error(cl, code, status, ...)                                 \
  do {                                                                         \
    uh_http_sendhf(cl, code, status, __VA_ARGS__);                             \
    uh_client_shutdown(cl);                                                    \
  } while (0)

void uh_client_shutdown(struct client *cl);
void uh_client_remove(struct client *cl);

void uh_ufd_add(struct uloop_fd *u, uloop_fd_handler h, unsigned int ev);
void uh_ufd_remove(struct uloop_fd *u);

#ifdef HAVE_CGI
struct interpreter *uh_interpreter_add(const char *extn, const char *path);
struct interpreter *uh_interpreter_lookup(const char *path);
#endif

#endif