client/Windows/cli/wfreerdp.c

 
#include <freerdp/config.h>
 
#include <winpr/windows.h>
 
#include <winpr/crt.h>
 
#include <freerdp/freerdp.h>
#include <freerdp/constants.h>
 
#include <freerdp/client/file.h>
#include <freerdp/client/cmdline.h>
#include <freerdp/client/channels.h>
#include <freerdp/channels/channels.h>
 
#include "../resource/resource.h"
 
#include <wf_client.h>
#include <wf_defaults.h>
 
#include <shellapi.h>
 
INT WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
{
  int status;
  HANDLE thread;
  wfContext* wfc;
  DWORD dwExitCode;
  rdpContext* context;
  rdpSettings* settings;
  LPWSTR cmd;
  char** argv = NULL;
  RDP_CLIENT_ENTRY_POINTS clientEntryPoints = { 0 };
  int ret = 1;
  int argc = 0;
  LPWSTR* args = NULL;
 
  WINPR_UNUSED(hInstance);
  WINPR_UNUSED(hPrevInstance);
  WINPR_UNUSED(lpCmdLine);
  WINPR_UNUSED(nCmdShow);
 
  RdpClientEntry(&clientEntryPoints);
  context = freerdp_client_context_new(&clientEntryPoints);
 
  if (!context)
    return -1;
 
  cmd = GetCommandLineW();
 
  if (!cmd)
    goto out;
 
  args = CommandLineToArgvW(cmd, &argc);
 
  if (!args || (argc <= 0))
    goto out;
 
  argv = calloc((size_t)argc, sizeof(char*));
 
  if (!argv)
    goto out;
 
  for (int i = 0; i < argc; i++)
  {
    int size = WideCharToMultiByte(CP_UTF8, 0, args[i], -1, NULL, 0, NULL, NULL);
    if (size <= 0)
      goto out;
    argv[i] = calloc((size_t)size, sizeof(char));
 
    if (!argv[i])
      goto out;
 
    if (WideCharToMultiByte(CP_UTF8, 0, args[i], -1, argv[i], size, NULL, NULL) != size)
      goto out;
  }
 
  freerdp_client_warn_deprecated(argc, argv);
 
  settings = context->settings;
  wfc = (wfContext*)context;
 
  if (!settings || !wfc)
    goto out;
 
  status = freerdp_client_settings_parse_command_line(settings, argc, argv, FALSE);
  if (status)
  {
    ret = freerdp_client_settings_command_line_status_print(settings, status, argc, argv);
    goto out;
  }
 
  AddDefaultSettings(settings);
 
  if (freerdp_client_start(context) != 0)
    goto out;
 
  thread = freerdp_client_get_thread(context);
 
  if (thread)
  {
    if (WaitForSingleObject(thread, INFINITE) == WAIT_OBJECT_0)
    {
      GetExitCodeThread(thread, &dwExitCode);
      ret = (int)dwExitCode;
    }
  }
 
  if (freerdp_client_stop(context) != 0)
    goto out;
 
out:
  freerdp_client_context_free(context);
 
  if (argv)
  {
    for (int i = 0; i < argc; i++)
      free(argv[i]);
 
    free(argv);
  }
 
  LocalFree(args);
  return ret;
}
 
#ifdef WITH_WIN_CONSOLE
int main()
{
  return WinMain(NULL, NULL, NULL, 0);
}
#endif