freerdp_proxy.c

 
#include <winpr/collections.h>
 
#include <freerdp/version.h>
#include <freerdp/freerdp.h>
 
#include <freerdp/server/proxy/proxy_server.h>
#include <freerdp/server/proxy/proxy_log.h>
 
#include <stdlib.h>
#include <signal.h>
 
#define TAG PROXY_TAG("server")
 
static proxyServer* server = nullptr;
 
#if defined(_WIN32)
WINPR_ATTR_NODISCARD
static const char* strsignal(int signum)
{
  switch (signum)
  {
    case SIGINT:
      return "SIGINT";
    case SIGTERM:
      return "SIGTERM";
    default:
      return "UNKNOWN";
  }
}
#endif
 
// NOLINTBEGIN(bugprone-signal-handler,cert-msc54-cpp,cert-sig30-c)
static void cleanup_handler(int signum)
{
  // NOLINTNEXTLINE(concurrency-mt-unsafe)
  WLog_INFO(TAG, "caught signal %s [%d], starting cleanup...", strsignal(signum), signum);
 
  WLog_INFO(TAG, "stopping all connections.");
  pf_server_stop(server);
}
// NOLINTEND(bugprone-signal-handler,cert-msc54-cpp,cert-sig30-c)
 
static void pf_server_register_signal_handlers(void)
{
  (void)signal(SIGINT, cleanup_handler);
  (void)signal(SIGTERM, cleanup_handler);
#ifndef _WIN32
  (void)signal(SIGQUIT, cleanup_handler);
  (void)signal(SIGKILL, cleanup_handler);
#endif
}
 
static int usage(const char* app)
{
  printf("Usage:\n");
  printf("%s -h                               Display this help text.\n", app);
  printf("%s --help                           Display this help text.\n", app);
  printf("%s --buildconfig                    Print the build configuration.\n", app);
  printf("%s <config ini file>                Start the proxy with <config.ini>\n", app);
  printf("%s --dump-config <config ini file>  Create a template <config.ini>\n", app);
  printf("%s -v                               Print out binary version.\n", app);
  printf("%s --version                        Print out binary version.\n", app);
  return 0;
}
 
WINPR_ATTR_NODISCARD
static int version(const char* app)
{
  printf("%s version %s", app, freerdp_get_version_string());
  return 0;
}
 
WINPR_ATTR_NODISCARD
static int buildconfig(WINPR_ATTR_UNUSED const char* app)
{
  printf("This is FreeRDP version %s (%s)\n", FREERDP_VERSION_FULL, FREERDP_GIT_REVISION);
  printf("%s", freerdp_get_build_config());
  return 0;
}
 
int main(int argc, char* argv[])
{
  int status = -1;
 
  pf_server_register_signal_handlers();
 
  WLog_INFO(TAG, "freerdp-proxy version info:");
  WLog_INFO(TAG, "\tFreeRDP version: %s", FREERDP_VERSION_FULL);
  WLog_INFO(TAG, "\tGit commit: %s", FREERDP_GIT_REVISION);
  WLog_DBG(TAG, "\tBuild config: %s", freerdp_get_build_config());
 
  if (argc < 2)
  {
    status = usage(argv[0]);
    goto fail;
  }
 
  {
    const char* arg = argv[1];
    if (_stricmp(arg, "-h") == 0)
    {
      status = usage(argv[0]);
      goto fail;
    }
    else if (_stricmp(arg, "--help") == 0)
    {
      status = usage(argv[0]);
      goto fail;
    }
    else if (_stricmp(arg, "--buildconfig") == 0)
    {
      status = buildconfig(argv[0]);
      goto fail;
    }
    else if (_stricmp(arg, "--dump-config") == 0)
    {
      if (argc != 3)
      {
        status = usage(argv[0]);
        goto fail;
      }
      status = pf_server_config_dump(argv[2]) ? 0 : -1;
      goto fail;
    }
    else if (_stricmp(arg, "-v") == 0)
    {
      status = version(argv[0]);
      goto fail;
    }
    else if (_stricmp(arg, "--version") == 0)
    {
      status = version(argv[0]);
      goto fail;
    }
  }
 
  {
    const char* config_path = argv[1];
    if (argc != 2)
    {
      status = usage(argv[0]);
      goto fail;
    }
 
    {
      proxyConfig* config = pf_server_config_load_file(config_path);
      if (!config)
        goto fail;
 
      pf_server_config_print(config);
 
      server = pf_server_new(config);
      pf_server_config_free(config);
    }
  }
 
  if (!server)
    goto fail;
 
  if (!pf_server_start(server))
    goto fail;
 
  if (!pf_server_run(server))
    goto fail;
 
  status = 0;
 
fail:
  pf_server_free(server);
 
  return status;
}