IHTTPRequestHandler.h

/*
 *  Copyright (C) 2011-2018 Team Kodi
 *  This file is part of Kodi - https://kodi.tv
 *
 *  SPDX-License-Identifier: GPL-2.0-or-later
 *  See LICENSES/README.md for more information.
 */

#pragma once

#include "utils/HttpRangeUtils.h"

#include <map>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <string>

#include <microhttpd.h>
#include <sys/select.h>
#include <sys/socket.h>
#include <sys/types.h>

#if MHD_VERSION >= 0x00097002
using MHD_RESULT = MHD_Result;
#else
using MHD_RESULT = int;
#endif

class CDateTime;
class CWebServer;

enum HTTPMethod
{
  UNKNOWN,
  POST,
  GET,
  HEAD
};

HTTPMethod GetHTTPMethod(const char *method);
std::string GetHTTPMethod(HTTPMethod method);

typedef enum HTTPResponseType
{
  HTTPNone,
  // creates and returns a HTTP error
  HTTPError,
  // creates and returns a HTTP redirect response
  HTTPRedirect,
  // creates a HTTP response with the content from a file
  HTTPFileDownload,
  // creates a HTTP response from a buffer without copying or freeing the buffer
  HTTPMemoryDownloadNoFreeNoCopy,
  // creates a HTTP response from a buffer by copying but not freeing the buffer
  HTTPMemoryDownloadNoFreeCopy,
  // creates a HTTP response from a buffer without copying followed by freeing the buffer
  // the buffer must have been malloc'ed and not new'ed
  HTTPMemoryDownloadFreeNoCopy,
  // creates a HTTP response from a buffer by copying followed by freeing the buffer
  // the buffer must have been malloc'ed and not new'ed
  HTTPMemoryDownloadFreeCopy
} HTTPResponseType;

typedef struct HTTPRequest
{
  CWebServer *webserver;
  struct MHD_Connection *connection;
  std::string pathUrlFull;
  std::string pathUrl;
  HTTPMethod method;
  std::string version;
  CHttpRanges ranges;
} HTTPRequest;

typedef struct HTTPResponseDetails {
  HTTPResponseType type;
  int status;
  std::multimap<std::string, std::string> headers;
  std::string contentType;
  uint64_t totalLength;
} HTTPResponseDetails;

class IHTTPRequestHandler
{
public:
  virtual ~IHTTPRequestHandler() = default;

  virtual IHTTPRequestHandler* Create(const HTTPRequest &request) const = 0;

  virtual int GetPriority() const { return 0; }

  virtual bool CanHandleRequest(const HTTPRequest &request) const = 0;

  virtual MHD_RESULT HandleRequest() = 0;

  virtual bool CanHandleRanges() const { return false; }

  virtual bool CanBeCached() const { return false; }

  virtual int GetMaximumAgeForCaching() const { return 0; }

  virtual bool GetLastModifiedDate(CDateTime &lastModified) const { return false; }

  virtual HttpResponseRanges GetResponseData() const { return HttpResponseRanges(); }

  virtual std::string GetRedirectUrl() const { return ""; }

  virtual std::string GetResponseFile() const { return ""; }

  const HTTPRequest& GetRequest() const { return m_request; }

  bool IsRequestRanged() const { return m_ranged; }

  void SetRequestRanged(bool ranged) { m_ranged = ranged; }

  void SetResponseStatus(int status) { m_response.status = status; }

  bool HasResponseHeader(const std::string &field) const;

  bool AddResponseHeader(const std::string &field, const std::string &value, bool allowMultiple = false);

  const HTTPResponseDetails& GetResponseDetails() const { return m_response; }

  void AddPostField(const std::string &key, const std::string &value);
  bool AddPostData(const char *data, size_t size);

protected:
  IHTTPRequestHandler();
  explicit IHTTPRequestHandler(const HTTPRequest &request);

  virtual bool appendPostData(const char *data, size_t size)
  { return true; }

  bool GetRequestedRanges(uint64_t totalLength);
  bool GetHostnameAndPort(std::string& hostname, uint16_t &port);

  HTTPRequest m_request;
  HTTPResponseDetails m_response;

  std::map<std::string, std::string> m_postFields;

private:
  bool m_ranged = false;
};